更新 'quick_consolidate.sh'

quick_consolidate.sh

@@ -7,7 +7,37 @@ set -e
 
 # 默认配置
 DEFAULT_FEE=1
-DEFAULT_MASTER_PUBKEY="33zmSubiizyAVzeSfKbVtLMKifm5u6VaTt9wae5K4CT6h5PLXovecjYwxwDXuzfKVsuefe1wh2VJVHAh4nbA53zBiXKf2Nc3fN3sVmZpmPWneUYGbWcXKzhmN714DVbzaS9U"
+KEY_FILE="key.txt"
+
+# 读取或创建主公钥
+load_master_pubkey() {
+    if [[ -f "$KEY_FILE" ]]; then
+        DEFAULT_MASTER_PUBKEY=$(cat "$KEY_FILE" | tr -d '\n\r' | sed 's/^[[:space:]]*//;s/[[:space:]]*$//')
+        if [[ -z "$DEFAULT_MASTER_PUBKEY" ]]; then
+            print_error "key.txt 文件为空，请重新输入主公钥"
+            ask_and_save_master_pubkey
+        else
+            print_info "从 $KEY_FILE 读取主公钥: $DEFAULT_MASTER_PUBKEY"
+        fi
+    else
+        print_info "未找到 $KEY_FILE 文件"
+        ask_and_save_master_pubkey
+    fi
+}
+
+ask_and_save_master_pubkey() {
+    echo -n "请输入主公钥地址: "
+    read -r DEFAULT_MASTER_PUBKEY
+    DEFAULT_MASTER_PUBKEY=$(echo "$DEFAULT_MASTER_PUBKEY" | tr -d '\n\r' | sed 's/^[[:space:]]*//;s/[[:space:]]*$//')
+    
+    if [[ -z "$DEFAULT_MASTER_PUBKEY" ]]; then
+        print_error "主公钥不能为空"
+        exit 1
+    fi
+    
+    echo "$DEFAULT_MASTER_PUBKEY" > "$KEY_FILE"
+    print_success "主公钥已保存到 $KEY_FILE"
+}
 
 # 颜色输出
 GREEN='\033[0;32m'
@@ -76,15 +106,18 @@ while [[ $# -gt 0 ]]; do
     esac
 done
 
+# 加载主公钥
+load_master_pubkey
+
 if [[ -z "$RECIPIENT" ]]; then
-    RECIPIENT="$MASTER_PUBKEY"
+    RECIPIENT="$DEFAULT_MASTER_PUBKEY"
 fi
 
 print_info "开始快速归集流程..."
 
 # 1. 扫描余额
 print_info "扫描UTXO余额..."
-nockchain-wallet scan --master-pubkey "$MASTER_PUBKEY"
+nockchain-wallet scan --master-pubkey "$DEFAULT_MASTER_PUBKEY"
 
 # 2. 获取所有notes
 print_info "获取notes列表..."