#!/bin/sh

#shopt -s expand_aliases
dmenu="$(command -v cdmenu)"

getmount() { \
    [ -z "$chosen" ] && exit 1
    # shellcheck disable=SC2086
    mp="$(find $1 2>/dev/null | "$dmenu" -l 16 -i -p "Type in mount point.")" || exit 1
    [ "$mp" = "" ] && exit 1
    if [ ! -d "$mp" ]; then
        mkdiryn=$(printf "No\nYes" | "$dmenu" -i -p "$mp does not exist. Create it?") || exit 1
        [ "$mkdiryn" = "Yes" ] && (mkdir -p "$mp" || sudo -A mkdir -p "$mp")
    fi
}

mountusb() { \
    # try to mount based on fstab, hoping "user" option is there
    mount "$1" 2>/dev/null && notify-send "$1 mounted" "" && exit 0
    # if no "user" option - retry with sudo
    sudo -A mount "$1" 2>/dev/null && notify-send "$1 mounted" "" && exit 0
    # if failed - ask for a dir to mount to
    # this creates argument for "find"
    alreadymounted=$(lsblk -nrpo "name,type,mountpoint" | awk '$3!~/\/boot|\/home$|SWAP/&&length($3)>1{printf "-not ( -path *%s -prune ) ",$3}')

    getmount "/mnt /media -maxdepth 1 -type d $alreadymounted"
    partitiontype="$(lsblk -no "fstype" "$1")"
    case "$partitiontype" in
        "vfat") sudo -A mount -t vfat "$1" "$mp" -o rw,umask=0000;;
        "exfat") sudo -A mount "$1" "$mp" -o uid="$(id -u)",gid="$(id -g)";;
        *crypto_LUKS*)
            name="$(LC_ALL=C tr -dc A-Za-z0-9 </dev/urandom | head -c 16)"
            mkfifo fifo
            dpass > fifo &
            sudo -A cryptsetup open --type luks "$1" "$name" --key-file fifo
            rm fifo
            device="/dev/mapper/$name"
            sudo -A mount "$device" "$mp"
            ;;
        *) sudo -A mount "$1" "$mp"; user="$(whoami)"; ug="$(groups | awk '{print $1}')"; sudo chown "$user":"$ug" "$mp";;
    esac
    notify-send "$1 mounted to $mp" ""
}

mountandroid() { \
    getmount "/mnt -maxdepth 1 -type d"
    jmtpfs -device="$1" "$mp"
    #echo "OK" | "$dmenu" -i -p "Tap Allow on your phone if it asks for permission and then press enter" || exit 1
    #jmtpfs -device="$1" "$mp"
    notify-send "Android mounted to $mp" ""
}


unmountdev() { \
    result="$(sudo umount "$1" 2>&1)"
    if [ -z "$result" ]; then
        notify-send "$1 unmounted" ""
        return
    else
        processes=$(lsof "$1" | tail -n +2 | awk '{print $1 " " $2}' | uniq)
        pids="$(echo "$processes" | awk '{print $2}')"
        #notify-send -u critical "Unmount device" "$result\nPreparing list of processes blocking $1..."
        notify-send "Unmount $1" "$result\nBlocking processes:\n$processes"
        action=$(printf 'nothing\nterminate\nkill' | "$dmenu" -i -p "What to do to processes blocking $1" -l 3)
        case $action in
            terminate) echo "$pids" | while IFS="$(printf '\n')" read -r pid ; do kill "$pid"; done ;;
            kill) echo "$pids" | while IFS="$(printf '\n')" read -r pid ; do kill -9 "$pid"; done ;;
            nothing)
                notify-send "Unmount of $1 aborted" ""
                return ;;
        esac
        unmountdev "$1"
    fi
}

selectusb() { \
    chosen=$(lsblk -lpo "vendor,model,name,label,size,mountpoint" | "$dmenu" -i -l 16 | xargs | cut -d' ' -f 1)
    # if mounted
    if grep -q "$chosen" "/etc/mtab" ; then
        choice=$(printf 'No\nYes' | "$dmenu" -i -l 2 -p "Partition is mounted. Unmount?")
        case $choice in
            Yes) unmountdev "$chosen" ;;
            No) exit ;;
        esac
    else
        mountusb "$chosen"
    fi
}


mountsamba() { \

    mounted=$(mount -t cifs)
    if [ -z "$mounted" ]; then
        actions="mount\nunmount"
        action=$(printf '%s' "$actions" | "$dmenu" -i -p "Samba action:")

        if [ "$action" = "unmount" ]; then
            mount=$(printf '%s' "$mounted" | "$dmenu" -i -p "Share to unmount:")
            printf 'Selected: %s\n' "$mount"
            mount_dir=$(printf '%s' "$mount" | cut -d' ' -f 3)
            printf '%s\n' "$mount_dir"
            sudo -A umount "$mount_dir"
            exit 0
        fi
        # proceed to mounting
    fi

    # get IP of this device
    IP=$(ip addr | grep 'state UP' -A2 | tail -n1 | awk '{print $2}' | cut -f1  -d'/')
    echo "IP: $IP"
    # prepare subnet choices: choose IP or run a scan if a wildcard specified
    subnets="$IP\n$(echo "$IP" | cut -d. -f1-3).1\n$(echo "$IP" | cut -d. -f1-3).*\n$(echo "$IP" | cut -d. -f1-2).*.*"
    printf 'Subnets available:\n%s\n' "$subnets"
    subnet=$(printf '%s' "$subnets" | "$dmenu" -i -l 16 -p "Input samba scan range")
    printf 'Subnet chosen: %s' "$subnet"

    # if there's a * in subnet
    if [ "${subnet#*\*}" != "$subnet" ]; then
        notify-send "Scanning network for samba shares..." ""
        servers="$(nmap "$subnet" -p 445 --open -oG -)"
        echo "$servers"

        IPs=$(echo "$servers" | grep Scan | cut -d' ' -f2-)
        printf 'Server IPs:\n%s\n' "$IPs"

        IP=$(printf '%s' "$IPs" | "$dmenu" -i -l 16 -p "Choose samba server IP")
        echo "Selected IP: $IP"
    else
        IP="$subnet"
    fi

    user=$(printf '%s\nanonymous' "$(whoami)" | "$dmenu" -p "Samba username:")
    echo "$user"
    if [ "$user" != "anonymous" ]; then
        password=$(pass-askpass "$user's password ")
        shares="""$(smbclient -L "$IP" --user "$user" --password "$password")"""
    else
        notify-send "Attempting anonymous login..." ""
        shares="""$(smbclient -L "$IP" -N)"""
    fi
    printf 'Raw shares:\n%s\n' "$shares"

    #shares=$(printf "$shares" | sed -e 's/^[[:space:]]*//')
    #printf "Cleaned shares:\n$shares\n"

    share=$(printf '%s' "$shares" | "$dmenu" -i -l 16 -p "Which samba share to mount?")
    echo "$share"

    # leave only the share name, assuming no whitespace in share name
    chosen=$(printf '%s' "$share" | cut -d' ' -f 1)
    echo "Chosen: $chosen"

    share="//$IP/$chosen"

    getmount "/mnt -maxdepth 1 -type d"

    if ! sudo -A mount -t cifs -o user="$user",password="$password",iocharset=utf8,uid="$(id -u)",gid="$(id -g)",file_mode=0644,dir_mode=0755 "$share" "$mp"; then
        notify-send "Samba mount failed" "$subnet to $mp as $user"
        exit 1
    fi
    notify-send "Samba share succeeded" "$subnet mounted to $mp as $user"
}

selectandroid() { \
    unmounted=$(jmtpfs -l | sed -e '1,2d')
    mounted=$(grep "jmtpfs" "/etc/mtab")

    list="Mounted:\n"
    if [ -n "$mounted" ]; then
        list="$list$mounted"
    fi
    list="$list\nUnmounted:\n"
    if [ -n "$unmounted" ]; then
        list="$list$unmounted"
    fi

    chosen=$(echo "$list" "$mounted" | "$dmenu" -i -l 16)
    #echo "CHOSEN" $chosen
    case $chosen in
        *jmtpfs*) unmountdev "$(echo "$chosen" | cut -d' ' -f 2)" ;;
        *) mountandroid "$(echo "$chosen" | awk '{print $1 $2}' | cut -d, -f1-2)" ;;
    esac
}

choice="$(printf 'drive\nSamba\nAndroid' | "$dmenu" -i -p "(Un-)mount a")" || exit 1
case $choice in
    drive) selectusb ;;
    Samba) mountsamba ;;
    Android) selectandroid ;;
esac