From 5d0bc5509333faedb69bfb1c1838293013cb3ceb Mon Sep 17 00:00:00 2001
From: zclaiqcc <zhao.chang.dut@gmail.com>
Date: Fri, 5 May 2023 15:48:19 +0800
Subject: [PATCH] feat: auto add SAFELINE_DIR

---
 upgrade.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/upgrade.sh b/upgrade.sh
index 7b59ee2..4762b96 100755
--- a/upgrade.sh
+++ b/upgrade.sh
@@ -35,5 +35,7 @@ wget https://waf-ce.chaitin.cn/release/latest/compose.yaml --no-check-certificat
 ENV_FILE=".env"
 sed -i "s/IMAGE_TAG=.*/IMAGE_TAG=latest/g" ${ENV_FILE}
 
+grep "SAFELINE_DIR" ${ENV_FILE} > /dev/null || echo "SAFELINE_DIR=$(pwd)" >> ${ENV_FILE}
+
 $compose_command down && $compose_command pull && $compose_command up -d
 echo "Upgrade success!"