From 8a1c2c90c337733e475be9614736f19e07900700 Mon Sep 17 00:00:00 2001
From: xaizek <xaizek@posteo.net>
Date: Wed, 24 Apr 2024 15:12:50 +0300
Subject: Support MTP via simple-mtpfs in vifm-media

Thanks to FlyCat (a.k.a. yanwh0311) and Alexandre Viau.

Closes #623 on GitHub.
---
 ChangeLog       |   3 ++
 THANKS          |   1 +
 data/vifm-media | 101 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 3 files changed, 103 insertions(+), 2 deletions(-)

diff --git a/ChangeLog b/ChangeLog
index 5afe8be40..df5855e9b 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -109,6 +109,9 @@
 	Make vifm-media not offer partitioned drives as they aren't mountable as a
 	whole.
 
+	Extended vifm-media script to support MTP devices if simple-mtpfs tool is
+	installed.  Thanks to FlyCat (a.k.a. yanwh0311) and Alexandre Viau.
+
 	Fixed line number column not including padding to the left of it.
 
 	Fixed local options not being loaded on Ctrl-W x.
diff --git a/THANKS b/THANKS
index 5db1ee7da..5fb5edb93 100644
--- a/THANKS
+++ b/THANKS
@@ -99,6 +99,7 @@ Fang (peromage)
 Flaviu Tamas (flaviut)
 Florian Baumann (derflob)
 flux242
+FlyCat (yanwh0311)
 fogine
 fohrums
 gammaray
diff --git a/data/vifm-media b/data/vifm-media
index c3b0a660e..81e503a5f 100755
--- a/data/vifm-media
+++ b/data/vifm-media
@@ -15,6 +15,15 @@
 #  - udisks2: second vesion of disk management D-Bus daemon
 #    requirements: `udisksctl` and either `python2` or `python3` with "dbus"
 #                  module
+#
+# Support for additional protocols (used in addition to standard drives, but
+# support depends on installed dependencies):
+#  - MTP: /dev/libmtp-* devices
+#    requirements: requires `simple-mtpfs` command to obtain device's name or
+#                  mount/unmount it
+#    limitations: can list and unmount only if this script was used for mounting
+#                 because source field of a FUSE mount is mounter's name rather
+#                 than the mounted device
 
 function usage_error() {
     echo "Usage: vifm-media list | mount <device> | unmount <path>"
@@ -173,24 +182,112 @@ else
     exit 1
 fi
 
+function have_simple_mtpfs() {
+    type simple-mtpfs > /dev/null 2>&1
+}
+
+function make_mtp_mount_path() {
+    local dev=$1
+
+    echo "/tmp/vifm-media$(echo -n "$dev" | tr -c 'a-zA-Z0-9-' _)"
+}
+
+function is_mtp_mount_path() {
+    local path=$1
+
+    case "$path" in
+        /tmp/vifm-media*) return 0 ;;
+        *)                return 1 ;;
+    esac
+}
+
+function list_mtp() {
+    for dev in /dev/libmtp-*; do
+        echo device="$dev"
+
+        if have_simple_mtpfs; then
+            local label=$(simple-mtpfs --list-devices "$dev" | cut -f2- -d' ')
+            echo "label=$label"
+        fi
+
+        echo -n /dev/libmtp-1-9 | tr -c 'a-zA-Z0-9-' _
+
+        local mount_path=$(make_mtp_mount_path "$dev")
+        if findmnt "$mount_path"; then
+            echo "mount-point=$mount_path"
+        fi
+    done
+}
+
+function is_mtp() {
+    local dev=$1
+
+    case "$dev" in
+        /dev/libmtp-*) return 0 ;;
+        *)             return 1 ;;
+    esac
+}
+
+function mount_mtp() {
+    local dev=$1
+
+    if ! have_simple_mtpfs; then
+        echo "vifm-media: need simple-mtpfs command to mount over MTP" 1>&2
+        exit 1
+    fi
+
+    local mount_path=$(make_mtp_mount_path "$dev")
+    if [ -e "$mount_path" ]; then
+        if ! rmdir "$mount_path"; then
+            exit 1
+        fi
+    fi
+
+    if ! ( umask 077 && mkdir "$mount_path" ); then
+        exit 1
+    fi
+
+    if ! simple-mtpfs "$dev" "$mount_path"; then
+        rmdir "$mount_path"
+        exit 1
+    fi
+}
+
+function unmount_mtp() {
+    local path=$1
+
+    umount "$path" && rmdir "$path"
+}
+
 case "$1" in
     list)
         if [ $# -ne 1 ]; then
             usage_error
         fi
         list
+        list_mtp
         ;;
     mount)
         if [ $# -ne 2 ]; then
             usage_error
         fi
-        mount "$2"
+
+        if is_mtp "$2"; then
+            mount_mtp "$2"
+        else
+            mount "$2"
+        fi
         ;;
     unmount)
         if [ $# -ne 2 ]; then
             usage_error
         fi
-        unmount "$2"
+
+        if is_mtp_mount_path "$2"; then
+            unmount_mtp "$2"
+        else
+            unmount "$2"
+        fi
         ;;
 
     *)
-- 
cgit v1.2.3