aboutsummaryrefslogtreecommitdiff
path: root/.local/bin/mounter
blob: 5828cd9d6299fa608fc24fd13e3c0170f6bcfc10 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#!/bin/bash

IFS='
'
# Function for escaping cell-phone names.
escape(){ echo "$@" | iconv -cf UTF-8 -t ASCII//TRANSLIT | tr -d '[:punct:]' | tr '[:upper:]' '[:lower:]' | tr ' ' '-' | sed "s/-\+/-/g;s/\(^-\|-\$\)//g" ;}

# Check for phones.
phones="$(simple-mtpfs -l 2>/dev/null | sed "s/^/📱/")"
mountedphones="$(rg "simple-mtpfs" /etc/mtab)"
# If there are already mounted phones, remove them from the list of mountables.
[ -n "$mountedphones" ] && phones="$(for phone in $phones; do
  for mounted in $mountedphones; do
    escphone="$(escape "$phone")"
    [[ "$mounted" =~ "$escphone" ]] && break 1
  done && continue 1
  echo "$phone"
done)"

# Check for drives.
lsblkoutput="$(doas lsblk -rpo "uuid,name,type,size,label,mountpoint,fstype")"
# Get all LUKS drives
allluks="$(echo "$lsblkoutput" | rg crypto_LUKS)"
# Get a list of the LUKS drive UUIDs already decrypted.
decrypted="$(find /dev/disk/by-id/dm-uuid-CRYPT-LUKS2-* | sed "s|.*LUKS2-||;s|-.*||")"
# Functioning for formatting drives correctly for rofi -dmenu:
filter() { sed "s/ /:/g" | awk -F':' '$7==""{printf "%s%s (%s) %s\n",$1,$3,$5,$6}' ; }

# Get only LUKS drives that are not decrypted.
unopenedluks="$(for drive in $allluks; do
  uuid="${drive%% *}"
  uuid="${uuid//-}"	# This is a bashism.
  [ -n "$decrypted" ] && for open in $decrypted; do
    [ "$uuid" = "$open" ] && break 1
  done && continue 1
  echo "🔒 $drive"
done | filter)"

# Get all normal, non-encrypted or decrypted partitions that are not mounted.
normalparts="$(echo "$lsblkoutput"| rg -v crypto_LUKS | rg 'part|rom|crypt' | sed "s/^/💾 /" | filter )"

# Add all to one variable. If no mountable drives found, exit.
alldrives="$(echo "$phones
$unopenedluks
$normalparts" | sed "/^$/d;s/ *$//")"

# Quit the script if a sequential command fails.
set -e

test -n "$alldrives"

# Feed all found drives to rofi -dmenu and get user choice.
chosen="$(echo "$alldrives" | rofi -dmenu -p "Mount which drive?" -i)"

# Function for prompting user for a mountpoint.
getmount(){
  mp="$(find /mnt /media /mount /home -maxdepth 1 -type d 2>/dev/null | rofi -dmenu -i -p "Mount this drive where?")"
  test -n "$mp"
  if [ ! -d "$mp" ]; then
    mkdiryn=$(printf "No\\nYes" | rofi -dmenu -i -p "$mp does not exist. Create it?")
    [ "$mkdiryn" = "Yes" ] && (mkdir -p "$mp" || doas_askpass mkdir -p "$mp")
  fi
}

attemptmount(){
  # Attempt to mount without a mountpoint, to see if drive is in fstab.
  doas mount "$chosen" || return 1
  notify-send "💾Drive Mounted." "$chosen mounted."
  exit
}

case "$chosen" in
  💾*)
    chosen="${chosen%% *}"
    chosen="${chosen:1}"	# This is a bashism.
    parttype="$(echo "$lsblkoutput" | rg "$chosen" | awk '{print $NF}' | tr -d '[:space:]\r\n')"
    attemptmount || getmount
    case "${parttype##* }" in
      vfat) doas mount -t vfat "$chosen" "$mp" -o rw,umask=0000 ;;
      btrfs) doas mount "$chosen" "$mp" ;;
      ntfs) doas mount -t ntfs3 "$chosen" "$mp" ;;
      *) doas mount "$chosen" "$mp" -o uid="$(id -u)",gid="$(id -g)" ;;
    esac
    notify-send "💾Drive Mounted." "$chosen mounted to $mp."
    ;;

  🔒*)
    chosen="${chosen%% *}"
    chosen="${chosen:1}"	# This is a bashism.
    # Number the drive.
    while true; do
      [ -f "/dev/mapper/usb$num" ] || break
      num="$(printf "%02d" "$((num +1))")"
    done

    # Decrypt in a terminal window
    ${TERMINAL:-st} --app-id termfloat -e doas cryptsetup open "$chosen" "usb$num"
    # Check if now decrypted.
    test -b "/dev/mapper/usb$num"

    attemptmount || getmount
    doas mount "/dev/mapper/usb$num" "$mp"
    notify-send "🔓Decrypted drive Mounted." "$chosen decrypted and mounted to $mp."
		;;

  📱*)
    notify-send "❗Note" "Remember to allow file access on your phone now."
    getmount
    number="${chosen%%:*}"
    number="${chosen:1}"	# This is a bashism.
    doas_askpass simple-mtpfs -o allow_other -o fsname="simple-mtpfs-$(escape "$chosen")" --device "$number" "$mp"
    notify-send "🤖 Android Mounted." "Android device mounted to $mp."
    ;;
esac