From 2fe58482db1f0433b38a66805a0f5da87a9e8a85 Mon Sep 17 00:00:00 2001
From: Paul Pan <i@0x7f.app>
Date: Fri, 5 Jan 2024 15:03:04 +0800
Subject: [PATCH] chore: import.sh: read config from env

---
 resource/runner/scripts/import.sh | 39 +++++++++++++++++++++----------
 1 file changed, 27 insertions(+), 12 deletions(-)

diff --git a/resource/runner/scripts/import.sh b/resource/runner/scripts/import.sh
index 4037a4f..4b79de4 100755
--- a/resource/runner/scripts/import.sh
+++ b/resource/runner/scripts/import.sh
@@ -3,22 +3,37 @@
 WORKSPACE=$(cd "$(dirname "$0")"/.. && pwd)
 . "$WORKSPACE"/scripts/common.sh
 
-read -p "Enter HTTP API Endpoint: " -r endpoint
-if [ -z "$endpoint" ]; then
-    log_error "[-] HTTP API Endpoint cannot be empty"
-    exit 1
+if [ -z "$WOJ_ENDPOINT" ]; then
+    read -p "Enter HTTP API Endpoint: " -r endpoint
+    if [ -z "$endpoint" ]; then
+        log_error "[-] HTTP API Endpoint cannot be empty"
+        exit 1
+    fi
+else
+    endpoint="$WOJ_ENDPOINT"
+    log_info "[*] Using HTTP API Endpoint: $endpoint"
 fi
 
-read -p "Enter Token: " -r token
-if [ -z "$token" ]; then
-    log_error "[-] Token cannot be empty"
-    exit 1
+if [ -z "$WOJ_TOKEN" ]; then
+    read -p "Enter Token: " -r token
+    if [ -z "$token" ]; then
+        log_error "[-] Token cannot be empty"
+        exit 1
+    fi
+else
+    token="$WOJ_TOKEN"
+    log_info "[*] Using Token: $token"
 fi
 
-read -p "Enter Directory: " -r directory
-if [ ! -d "$directory" ]; then
-    log_error "[-] Not a Directory"
-    exit 1
+if [ -z "$WOJ_DIR" ]; then
+    read -p "Enter Directory: " -r directory
+    if [ ! -d "$directory" ]; then
+        log_error "[-] Not a Directory"
+        exit 1
+    fi
+else
+    directory="$WOJ_DIR"
+    log_info "[*] Using Directory: $directory"
 fi
 
 for problem in "$directory/"*; do