From 63b6fcdac06e243ffbbaad42d4d2d0cd05e941c0 Mon Sep 17 00:00:00 2001
From: Roman Dobosz <gryf73@gmail.com>
Date: Sat, 11 Apr 2015 17:51:32 +0200
Subject: [PATCH] Added sharedfolder and setextradata commands

---
 README     |   2 +
 VBoxManage | 132 ++++++++++++++++++++++++++++++++++++++---------------
 2 files changed, 98 insertions(+), 36 deletions(-)

diff --git a/README b/README
index aee9be6..8a9f63d 100644
--- a/README
+++ b/README
@@ -23,6 +23,8 @@ Current version of script was written and tested against VBoxManage in version
 - extpack
 - getextradata
 - list
+- setextradata
+- sharedfolder
 - showhdinfo
 - showvminfo
 - snapshot
diff --git a/VBoxManage b/VBoxManage
index b307682..32968f5 100644
--- a/VBoxManage
+++ b/VBoxManage
@@ -16,13 +16,44 @@
 # [ ] modifyvm
 # [ ] natnetwork
 # [ ] registervm
-# [ ] setextradata
 # [ ] setproperty
-# [ ] sharedfolder
 
 _VBoxManage() {
     local cur prev opts cmd subcommand tmp items name index result
 
+    # Check the COMP_WORDS looking for name of the vm. If name contain space or
+    # is enclosed in quotes, glue name together in variable name. Variable index
+    # will hold the last index of COMP_WORDS array which contain the end of the
+    # name.
+    _find_item_name() {
+        local idx=$1
+        name=""
+
+        while true
+        do
+            name="${name}${COMP_WORDS[$idx]}"
+            [[ ${COMP_WORDS[$idx]} = *'"' ]] && break
+
+            if [[ ${COMP_WORDS[$idx]} = '"'* || ${COMP_WORDS[$idx]} = *'\ ' ]]
+            then
+                idx=$((++idx))
+                continue
+            fi
+            break
+        done
+        index=$idx
+    }
+
+    _get_excluded_items() {
+        local i
+
+        result=""
+        for i in $@; do
+            [[ " ${COMP_WORDS[@]} " == *" $i "* ]] && continue
+            result="$result $i"
+        done
+    }
+
     # Generate registered hard disk files.
     # NOTE: This function may introduce some quirks, if there is a space or
     # other characters which usually are treated as IFS - like space. Pipe
@@ -217,39 +248,6 @@ _VBoxManage() {
         done
     }
 
-    # Check the COMP_WORDS looking for name of the vm. If name contain space or
-    # is enclosed in quotes, glue name together in variable name. Variable index
-    # will hold the last index of COMP_WORDS array which contain the end of the
-    # name.
-    _find_item_name() {
-        local idx=$1
-        name=""
-
-        while true
-        do
-            name="${name}${COMP_WORDS[$idx]}"
-            [[ ${COMP_WORDS[$idx]} = *'"' ]] && break
-
-            if [[ ${COMP_WORDS[$idx]} = '"'* || ${COMP_WORDS[$idx]} = *'\ ' ]]
-            then
-                idx=$((++idx))
-                continue
-            fi
-            break
-        done
-        index=$idx
-    }
-
-    _get_excluded_items() {
-        local i
-
-        result=""
-        for i in $@; do
-            [[ " ${COMP_WORDS[@]} " == *" $i "* ]] && continue
-            result="$result $i"
-        done
-    }
-
     _bandwidthctl_comp() {
         local rules cur=$1
         local item
@@ -338,6 +336,28 @@ _VBoxManage() {
         COMPREPLY=( $(compgen -W "$list" -- ${cur}) )
     }
 
+    _sharedfolder_comp() {
+        local vm=$1
+        local cur=$2
+        local folders
+        local item
+
+        folders=$(VBoxManage showvminfo ${vm} --machinereadable | \
+            grep SharedFolderName | \
+            awk -F= '{print $2}' | \
+            sed 's/\s/\\ /g'| \
+            tr '\n' '|' | \
+            sed 's/|$//' | \
+            sed 's/"//g')
+        IFS='|' read -ra folders <<< "$folders"
+
+        for item in "${folders[@]}"
+        do
+            [[ ${item^^} == ${cur^^}* ]] && COMPREPLY+=("$item")
+        done
+    }
+
+
     COMP_WORDBREAKS=${COMP_WORDBREAKS//|/}  # remove pipe from comp word breaks
     COMPREPLY=()
 
@@ -856,13 +876,53 @@ _VBoxManage() {
         registervm)
             ;;
         setextradata)
+            if [[ ${prev} == ${cmd} ]]; then
+                COMPREPLY=( $(compgen -W "global" -- ${cur}) )
+                _vms_comp vms ${cur}
+            fi
             ;;
         setproperty)
             ;;
         sharedfolder)
+            items=(add remove)
+            subcommand=${COMP_WORDS[2]}
+            case "${prev}" in
+                add|remove)
+                    _vms_comp vms ${cur}
+                    ;;
+                --hostpath)
+                    COMPREPLY=( $(compgen -o dirnames -- ${cur}) )
+                    ;;
+                --name)
+                    if [[ ${subcommand} == "remove" ]]; then
+                        _find_item_name 3
+                        _sharedfolder_comp "${name}" "${cur}"
+                    fi
+                    ;;
+            esac
+            if [[ ${#COMPREPLY[@]} -eq 0 ]]; then
+                case "${subcommand}" in
+                    add)
+                        items=(--name --hostpath --transient --readonly
+                        --automount)
+                        _get_excluded_items "${items[@]}"
+                        COMPREPLY=( $(compgen -W "$result" -- ${cur}) )
+                        ;;
+                    remove)
+                        items=(--name --transient)
+                        _get_excluded_items "${items[@]}"
+                        COMPREPLY=( $(compgen -W "$result" -- ${cur}) )
+                        ;;
+                esac
+            fi
+
+            [[ ${#COMPREPLY[@]} -eq 0 ]] && \
+                COMPREPLY=( $(compgen -W "${items[*]}" -- ${cur}) )
             ;;
         showhdinfo)
-            _hdd_comp ${cur}
+            if [[ ${prev} == ${cmd} ]]; then
+                _hdd_comp ${cur}
+            fi
             ;;
         showvminfo)
             if [[ ${prev} == ${cmd} ]]; then