From 5f80250a7fa2f72207415d5b0ffeb41a1b8a433c Mon Sep 17 00:00:00 2001
From: sdohuajia <63113882+sdohuajia@users.noreply.github.com>
Date: Sat, 27 Jul 2024 17:50:32 +0800
Subject: [PATCH] Update Waku.sh

---
 Waku.sh | 35 ++++++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/Waku.sh b/Waku.sh
index 467eb0e..2c36ae6 100644
--- a/Waku.sh
+++ b/Waku.sh
@@ -14,7 +14,8 @@ function main_menu() {
         echo "1. 安装节点"
         echo "2. 修复错误"
         echo "3. 编辑 .env 文件"
-        echo "4. 退出"
+        echo "4. 更新旧版本"
+        echo "5. 退出"
         read -rp "请输入操作选项：" choice
 
         case $choice in
@@ -28,6 +29,9 @@ function main_menu() {
                 edit_env_file
                 ;;
             4)
+                update_old_version
+                ;;
+            5)
                 echo "退出脚本，谢谢使用！"
                 exit 0
                 ;;
@@ -140,6 +144,35 @@ function edit_env_file() {
     read -rp "按 Enter 返回菜单。"
 }
 
+# 更新旧版本函数
+function update_old_version() {
+    echo "正在更新旧版本..."
+
+    # 停止 Docker Compose 服务
+    docker-compose down
+
+    # 执行 git stash 和 git pull 操作
+    git stash push --include-untracked
+    git pull https://github.com/waku-org/nwaku-compose.git
+
+    # 删除 keystore 和 rln_tree 目录
+    rm -rf keystore rln_tree
+
+    # 从 origin/master 分支拉取最新代码
+    git pull origin master
+
+    # 编辑 .env 文件
+    nano .env  # 请修改 ETH_CLIENT_ADDRESS 为 RLN_RELAY_ETH_CLIENT_ADDRESS
+
+    # 注册节点
+    ./register_rln.sh || { echo "注册节点失败，请检查错误信息。"; exit 1; }
+
+    # 启动 Docker Compose
+    docker-compose up -d || { echo "启动 Docker Compose 失败，请检查错误信息。"; exit 1; }
+
+    echo "旧版本更新完成。"
+    read -rp "按 Enter 返回菜单。"
+}
 
 # 主程序开始
 main_menu