Tidy

templates/dropshell-agent/shared/_autocommands.sh

@@ -3,49 +3,77 @@
 
 _autocommandrun_volume() {
     command="$1"
-    value="$2"
-    backup_file="$3"
-    temp_path="$4"
+    volume_name="$2"
+
+    case "$command" in
+        create)
+            echo "Creating volume ${volume_name}"
+            docker volume create ${volume_name}
+            ;;
+        nuke)
+            echo "Nuking volume ${volume_name}"
+            docker volume rm ${volume_name}
+            ;;
+        backup)
+            local backup_file="$3"
+            echo "Backing up volume ${volume_name}"
+            docker run --rm -v ${volume_name}:/volume -v ${temp_path}:/backup alpine tar -czvf /backup/volume.tar.gz -C /volume .
+            ;;
+        restore)
+            local backup_file="$3"
+            echo "Restoring volume ${volume_name}"
+            docker volume rm ${volume_name}
+            docker volume create ${volume_name}
+            docker run --rm -v ${volume_name}:/volume -v ${temp_path}:/backup alpine tar -xzvf /backup/volume.tar.gz -C /volume --strip-components=1
+            ;;
+    esac
 }   
 
 _autocommandrun_path() {
     command="$1"
-    value="$2"
-    backup_file="$3"
-    temp_path="$4"
+    path="$2"
+
+    case "$command" in
+        create)
+            echo "Creating path ${path}"
+            mkdir -p ${path}
+            ;;
+        nuke)
+            echo "Nuking path ${path}"
+            rm -rf ${path}
+            ;;
+        backup)
+            local backup_file="$3"
+            echo "Backing up path ${path}"
+            tar -czvf ${backup_file} -C ${path} .
+            ;;
+        restore)
+            local backup_file="$3"
+            echo "Restoring path ${path}"
+            tar -xzvf ${backup_file} -C ${path} --strip-components=1
+            ;;
+    esac
 }
 
 _autocommandrun_file() {
     command="$1"
     value="$2"
-    backup_file="$3"
-    temp_path="$4"
-}
 
-_autocommandrun() {
-    command="$1"
-    key="$2"
-    value="$3"
-
-    # only passed through if command is backup or restore.
-    backup_file="$4"
-    temp_path="$5"
-    
-    case "$key" in
-        volume)
-            echo "Volume: $value"
-            _autocommandrun_volume "$command" "$value" "$backup_file" "$temp_path"
+    case "$command" in
+        create)
             ;;
-        path)
-            echo "Path: $value"
-            _autocommandrun_path "$command" "$value" "$backup_file" "$temp_path"
+        nuke)
+            rm -f ${value}
             ;;
-        file)
-            echo "File: $value"
-            _autocommandrun_file "$command" "$value" "$backup_file" "$temp_path"
+        backup)
+            local backup_file="$3"
+            echo "Backing up file ${value}"
+            cp ${value} ${backup_file}
             ;;
-        *)
-            _die "Unknown key $key passed to auto${command}. We only support volume, path and file."
+        restore)
+            local backup_file="$3"
+            echo "Restoring file ${value}"
+            cp ${backup_file} ${value}
             ;;
     esac
 }
@@ -59,19 +87,20 @@ _autocommandparse() {
     # value is the path or volume name.
 
     # we iterate over the key=value arguments, and for each we call:
-    #    autorun <command> <key> <value> <backup_file> <temp_path>
+    #    autorun <command> <backupfile> <key> <value>
 
     local command="$1"
     shift
 
+    local temp_path="$2"
+    shift
+
     # Extract the backup file and temp path (last two arguments)
     local args=("$@")
     local arg_count=${#args[@]}
-    local backup_file="${args[$arg_count-2]}"
-    local temp_path="${args[$arg_count-1]}"
     
     # Process all key=value pairs
-    for ((i=0; i<$arg_count-2; i++)); do
+    for ((i=0; i<$arg_count; i++)); do
         local pair="${args[$i]}"
         
         # Skip if not in key=value format
@@ -81,30 +110,71 @@ _autocommandparse() {
         
         local key="${pair%%=*}"
         local value="${pair#*=}"
-        
+
+        local bfile="${temp_path}/${key}_${value}.tgz"
+
         # Key must be one of volume, path or file
-        _autocommandrun "$command" "$key" "$value" "$backup_file" "$temp_path"
+        case "$key" in
+            volume)
+                _autocommandrun_volume "$command" "$value" "$bfile"
+                ;;
+            path)
+                _autocommandrun_path "$command" "$value" "$bfile"
+                ;;
+            file)
+                _autocommandrun_file "$command" "$value" "$bfile"
+                ;;
+            *)
+                _die "Unknown key $key passed to auto${command}. We only support volume, path and file."
+                ;;
+        esac
     done
 }
 
 
 autocreate() {
-    _autocommandparse create "$@" "-" "-"
+    _autocommandparse create "-" "$@"
 }
 
 
 autonuke() {
-    _autocommandparse nuke "$@" "-" "-"
+    _autocommandparse nuke "-" "$@"
 }
 
 
 autobackup() {
-    _autocommandparse backup "$@"
+    local backup_file="$1"
+    shift
+    local temp_path="$1"    
+    shift
+
+    [ -f "$backup_file" ] || _die "Backup file $backup_file does not exist"
+    [ -d "$temp_path" ] || _die "Temp path $temp_path does not exist"
+
+    local backup_temp_path="$temp_path/backup"
+
+    mkdir -p "$backup_temp_path"
+    _autocommandparse backup "$backup_temp_path" "$@"
+
+    tar zcvf "$backup_file" -C "$backup_temp_path" .
 }
 
 
 autorestore() {
-    _autocommandparse restore "$@"
+    local backup_file="$1"
+    shift
+    local temp_path="$1"
+    shift
+
+    [ -f "$backup_file" ] || _die "Backup file $backup_file does not exist"
+    [ -d "$temp_path" ] || _die "Temp path $temp_path does not exist"
+
+    local restore_temp_path="$temp_path/restore"
+
+    mkdir -p "$restore_temp_path"
+    tar zxvf "$backup_file" -C "$restore_temp_path" --strip-components=1
+
+    _autocommandparse restore "$restore_temp_path" "$@"
 }