From 37f1166302c3fcd0c46527dc7d71a42652a0e75f Mon Sep 17 00:00:00 2001
From: discountry <feibilanceon@gmail.com>
Date: Wed, 24 Sep 2025 22:41:51 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E5=AE=89=E8=A3=85=E8=84=9A?=
 =?UTF-8?q?=E6=9C=AC=EF=BC=8C=E6=B7=BB=E5=8A=A0=E7=94=A8=E6=88=B7=E6=8E=A8?=
 =?UTF-8?q?=E8=8D=90=E9=93=BE=E6=8E=A5=E6=8F=90=E7=A4=BA=EF=BC=8C=E5=B9=B6?=
 =?UTF-8?q?=E5=A2=9E=E5=BC=BA=E8=BE=93=E5=85=A5=E9=AA=8C=E8=AF=81=EF=BC=8C?=
 =?UTF-8?q?=E7=A1=AE=E4=BF=9D=20API=20=E5=AF=86=E9=92=A5=E5=92=8C=E5=85=B6?=
 =?UTF-8?q?=E4=BB=96=E9=85=8D=E7=BD=AE=E9=A1=B9=E4=B8=8D=E4=B8=BA=E7=A9=BA?=
 =?UTF-8?q?=EF=BC=8C=E6=8F=90=E5=8D=87=E7=94=A8=E6=88=B7=E4=BD=93=E9=AA=8C?=
 =?UTF-8?q?=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 setup.sh | 32 +++++++++++++++++++++++++-------
 1 file changed, 25 insertions(+), 7 deletions(-)

diff --git a/setup.sh b/setup.sh
index 4fd6638..2f16101 100644
--- a/setup.sh
+++ b/setup.sh
@@ -13,6 +13,7 @@ main() {
   require_unix
   ensure_repo
   cd "$PROJECT_DIR"
+  referral_notice
   ensure_bun
   install_deps
   prompt_env
@@ -85,6 +86,15 @@ ensure_repo() {
   PROJECT_DIR="$TARGET_DIR"
 }
 
+referral_notice() {
+  echo
+  echo "开始之前：请打开以下链接连接钱包加入战队，享受 30% 手续费优惠："
+  echo "https://www.asterdex.com/zh-CN/referral/4665f3"
+  echo
+  # Wait for user to press Enter on a real TTY; don't fail if no TTY
+  read -r -p "打开链接后按下回车继续..." _ < /dev/tty || true
+}
+
 install_deps() {
   echo "Installing dependencies with Bun..."
   bun install
@@ -93,18 +103,26 @@ install_deps() {
 prompt_env() {
   echo
   echo "Please enter your AsterDex API credentials."
-  read -r -p "ASTER_API_KEY: " ASTER_API_KEY
-  read -r -s -p "ASTER_API_SECRET (input hidden): " ASTER_API_SECRET
-  echo
-  read -r -p "TRADE_SYMBOL (default BTCUSDT): " TRADE_SYMBOL || true
+  while true; do
+    read -r -p "ASTER_API_KEY: " ASTER_API_KEY < /dev/tty
+    [ -n "${ASTER_API_KEY:-}" ] && break
+    echo "ASTER_API_KEY cannot be empty. Please try again."
+  done
+  while true; do
+    read -r -s -p "ASTER_API_SECRET (input hidden): " ASTER_API_SECRET < /dev/tty
+    echo
+    [ -n "${ASTER_API_SECRET:-}" ] && break
+    echo "ASTER_API_SECRET cannot be empty. Please try again."
+  done
+  read -r -p "TRADE_SYMBOL (default BTCUSDT): " TRADE_SYMBOL < /dev/tty || true
   TRADE_SYMBOL=${TRADE_SYMBOL:-BTCUSDT}
 
   # Optional trading parameters
-  read -r -p "TRADE_AMOUNT (default 0.001): " TRADE_AMOUNT || true
+  read -r -p "TRADE_AMOUNT (default 0.001): " TRADE_AMOUNT < /dev/tty || true
   TRADE_AMOUNT=${TRADE_AMOUNT:-0.001}
-  read -r -p "LOSS_LIMIT (default 0.03): " LOSS_LIMIT || true
+  read -r -p "LOSS_LIMIT (default 0.03): " LOSS_LIMIT < /dev/tty || true
   LOSS_LIMIT=${LOSS_LIMIT:-0.03}
-  read -r -p "KLINE_INTERVAL (default 1m): " KLINE_INTERVAL || true
+  read -r -p "KLINE_INTERVAL (default 1m): " KLINE_INTERVAL < /dev/tty || true
   KLINE_INTERVAL=${KLINE_INTERVAL:-1m}
 }