From 86aa9106090a0219bac30bc12f5a5bd91949afd9 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 16 Nov 2024 11:09:40 +0800
Subject: refactor: improve bash code (#125)

* refactor: extract guard_path to utils/guard_path.sh

* add utils/guard_operation.sh
---
 tools/execute_sql_code.sh | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

(limited to 'tools/execute_sql_code.sh')

diff --git a/tools/execute_sql_code.sh b/tools/execute_sql_code.sh
index df57dea..34cd948 100755
--- a/tools/execute_sql_code.sh
+++ b/tools/execute_sql_code.sh
@@ -9,15 +9,11 @@ set -e
 # @env USQL_DSN! The database url, e.g. pgsql://user:pass@host/dbname
 # @env LLM_OUTPUT=/dev/stdout The output path
 
+ROOT_DIR="${LLM_ROOT_DIR:-$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)}"
+
 main() {
     if ! grep -qi '^select' <<<"$argc_code"; then
-        if [ -t 1 ]; then
-            read -r -p "Are you sure you want to continue? [Y/n] " ans
-            if [[ "$ans" == "N" || "$ans" == "n" ]]; then
-                echo "Aborted!"
-                exit 1
-            fi
-        fi
+        "$ROOT_DIR/utils/guard_operation.sh"
     fi
     usql -c "$argc_code" "$USQL_DSN" >> "$LLM_OUTPUT"
 }
-- 
cgit v1.2.3