delete: improve

1 files changed, 19 insertions, 1 deletions

diff --git a/delete-files b/delete-files
index 1354016..778b27c 100755
--- a/delete-files
+++ b/delete-files
@@ -5,6 +5,7 @@ set -eu -o pipefail
 dev="$1"
 dir="$2"
 img="$3"
+cmd="$4"
 
 echo
 echo "unmount $dev"
@@ -22,7 +23,24 @@ sudo mount "$dev" "$dir"
 
 echo
 echo "delete all files under $dir"
-find "$dir" -mindepth 2 ! -delete
+files="$dir/files"
+case "$cmd" in
+    find)
+	find "$files" -delete
+	;;
+    rm)
+	rm -rf "$files"
+	;;
+    rsync)
+	tmp="$(mktemp -d)"
+	rsync -r --del "$tmp/." "$files/."
+	rmdir "$files"
+	;;
+    *)
+	echo "I don't know $cmd" 1>&2 ;
+	exit 1
+	;;
+esac
 
 echo
 echo "unmount $dev"