From d66d8938f1f845e92966835f70bc2f97ebb8246b Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 2 Aug 2024 23:26:49 +0800
Subject: feat: prompt confirmation for fs write operations outside cwd (#90)

---
 tools/fs_rm.sh | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

(limited to 'tools/fs_rm.sh')

diff --git a/tools/fs_rm.sh b/tools/fs_rm.sh
index 1d40295..c07c17d 100755
--- a/tools/fs_rm.sh
+++ b/tools/fs_rm.sh
@@ -7,9 +7,25 @@ set -e
 # @option --path! The path of the file or directory to remove
 
 main() {
-    path="$FS_BASE_DIR/$argc_path"
-    rm -rf "$path"
-    echo "Path removed: $path" >> "$LLM_OUTPUT"
+    if [[ -f "$argc_path" ]]; then
+        _guard_path "$argc_path" Remove
+        rm -rf "$argc_path"
+    fi
+    echo "Path removed: $argc_path" >> "$LLM_OUTPUT"
+}
+
+_guard_path() {
+    path="$(realpath "$1")"
+    action="$2"
+    if [[ ! "$path" == "$(pwd)"* ]]; then
+        if [ -t 1 ]; then
+            read -r -p "$action $path? [Y/n] " ans
+            if [[ "$ans" == "N" || "$ans" == "n" ]]; then
+                echo "Aborted!"
+                exit 1
+            fi
+        fi
+    fi
 }
 
 eval "$(argc --argc-eval "$0" "$@")"
-- 
cgit v1.2.3