feat: prompt confirmation for fs write operations outside cwd (#90)

This commit is contained in:
sigoden
2024-08-02 23:26:49 +08:00
committed by GitHub
parent f5c67b3eff
commit d66d8938f1
6 changed files with 63 additions and 28 deletions
+22 -10
View File
@@ -1,15 +1,13 @@
#!/usr/bin/env bash #!/usr/bin/env bash
set -e set -e
# @env FS_BASE_DIR=. The base dir
# @cmd Create a new file at the specified path with content. # @cmd Create a new file at the specified path with content.
# @option --path! The path where the file should be created # @option --path! The path where the file should be created
# @option --content! The content of the file # @option --content! The content of the file
fs_create() { fs_create() {
path="$FS_BASE_DIR/$argc_path" _guard_path "$argc_path" Create
printf "%s" "$argc_content" > "$path" printf "%s" "$argc_content" > "$argc_path"
echo "File created: $path" >> "$LLM_OUTPUT" echo "File created: $argc_path" >> "$LLM_OUTPUT"
} }
# @cmd Apply changes to a file. Use this when you need to edit an existing file. # @cmd Apply changes to a file. Use this when you need to edit an existing file.
@@ -20,10 +18,10 @@ fs_create() {
# @option --content! The new content to apply to the file # @option --content! The new content to apply to the file
# @meta require-tools git # @meta require-tools git
fs_edit() { fs_edit() {
path="$FS_BASE_DIR/$argc_path" if [[ -f "$argc_path" ]]; then
if [[ -f "$path" ]]; then _guard_path "$argc_path" Edit
changed=0 changed=0
printf "%s" "$argc_content" | git diff --no-index "$path" - || { printf "%s" "$argc_content" | git diff --no-index "$argc_path" - || {
changed=1 changed=1
} }
if [[ "$changed" -eq 0 ]]; then if [[ "$changed" -eq 0 ]]; then
@@ -37,11 +35,25 @@ fs_edit() {
exit 1 exit 1
fi fi
fi fi
printf "%s" "$argc_content" > "$path" printf "%s" "$argc_content" > "$argc_path"
echo "Applied changes" >> "$LLM_OUTPUT" echo "Applied changes" >> "$LLM_OUTPUT"
fi fi
else else
echo "Not found file: $path" >> "$LLM_OUTPUT" echo "Not found file: $argc_path" >> "$LLM_OUTPUT"
fi
}
_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 fi
} }
+1 -3
View File
@@ -4,12 +4,10 @@ set -e
# @describe Read the contents of a file at the specified path. # @describe Read the contents of a file at the specified path.
# Use this when you need to examine the contents of an existing file. # Use this when you need to examine the contents of an existing file.
# @env FS_BASE_DIR=. The base dir
# @option --path! The path of the file to read # @option --path! The path of the file to read
main() { main() {
path="$FS_BASE_DIR/$argc_path" cat "$argc_path" >> "$LLM_OUTPUT"
cat "$path" >> "$LLM_OUTPUT"
} }
eval "$(argc --argc-eval "$0" "$@")" eval "$(argc --argc-eval "$0" "$@")"
+1 -3
View File
@@ -3,12 +3,10 @@ set -e
# @describe List all files and directories at the specified path. # @describe List all files and directories at the specified path.
# @env FS_BASE_DIR=. The base dir
# @option --path! The path of the directory to list # @option --path! The path of the directory to list
main() { main() {
path="$FS_BASE_DIR/$argc_path" ls -1 "$argc_path" >> "$LLM_OUTPUT"
ls -1 "$path" >> "$LLM_OUTPUT"
} }
eval "$(argc --argc-eval "$0" "$@")" eval "$(argc --argc-eval "$0" "$@")"
+2 -4
View File
@@ -3,13 +3,11 @@ set -e
# @describe Create a new directory at the specified path. # @describe Create a new directory at the specified path.
# @env FS_BASE_DIR=. The base dir
# @option --path! The path of the directory to create # @option --path! The path of the directory to create
main() { main() {
path="$FS_BASE_DIR/$argc_path" mkdir -p "$argc_path"
mkdir -p "$path" echo "Directory created: $argc_path" >> "$LLM_OUTPUT"
echo "Directory created: $path" >> "$LLM_OUTPUT"
} }
eval "$(argc --argc-eval "$0" "$@")" eval "$(argc --argc-eval "$0" "$@")"
+19 -3
View File
@@ -7,9 +7,25 @@ set -e
# @option --path! The path of the file or directory to remove # @option --path! The path of the file or directory to remove
main() { main() {
path="$FS_BASE_DIR/$argc_path" if [[ -f "$argc_path" ]]; then
rm -rf "$path" _guard_path "$argc_path" Remove
echo "Path removed: $path" >> "$LLM_OUTPUT" 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" "$@")" eval "$(argc --argc-eval "$0" "$@")"
+18 -5
View File
@@ -6,15 +6,28 @@ set -e
# If the file doesn't exist, it will be created. # If the file doesn't exist, it will be created.
# Always provide the full intended contents of the file. # Always provide the full intended contents of the file.
# @env FS_BASE_DIR=. The base dir
# @option --path! The path of the file to write to # @option --path! The path of the file to write to
# @option --contents! The full contents to write to the file # @option --contents! The full contents to write to the file
main() { main() {
path="$FS_BASE_DIR/$argc_path" _guard_path "$argc_path" Write
mkdir -p "$(dirname "$path")" mkdir -p "$(dirname "$argc_path")"
printf "%s" "$argc_contents" > "$path" printf "%s" "$argc_contents" > "$argc_path"
echo "The contents written to: $path" >> "$LLM_OUTPUT" echo "The contents written to: $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" "$@")" eval "$(argc --argc-eval "$0" "$@")"