From 0e145847bf7b4741bfa3fd4f1d8d3ae6a2e45f42 Mon Sep 17 00:00:00 2001
From: Jensfrank <everett7623@gmail.com>
Date: Fri, 5 Jul 2024 14:15:34 +0800
Subject: [PATCH] Update Nlbench.sh

---
 Nlbench.sh | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/Nlbench.sh b/Nlbench.sh
index e07eb66..ccaf410 100644
--- a/Nlbench.sh
+++ b/Nlbench.sh
@@ -19,6 +19,53 @@ if [ "$(id -u)" != "0" ]; then
     echo "已获取 sudo 权限。"
 fi
 
+# 更新脚本
+update_scripts() {
+    CURRENT_VERSION="2024-07-05 v1.0.0"  # 当前版本号
+    SCRIPT_URL="https://raw.githubusercontent.com/everett7623/main/Nlbencha.sh"
+    VERSION_URL="https://raw.githubusercontent.com/everett7623/main/version.sh"
+    
+    # 获取远程版本号
+    REMOTE_VERSION=$(curl -s $VERSION_URL)
+    if [ -z "$REMOTE_VERSION" ]; then
+        echo -e "${RED}无法获取远程版本信息。请检查您的网络连接。${NC}"
+        sleep 2
+        return 1
+    fi
+
+    # 对比版本号
+    if [ "$REMOTE_VERSION" != "$CURRENT_VERSION" ]; then
+        echo -e "${BLUE}发现新版本 $REMOTE_VERSION，当前版本 $CURRENT_VERSION${NC}"
+        echo -e "${BLUE}正在更新...${NC}"
+        
+        # 下载新的脚本文件
+        if curl -s -o /tmp/Nlbencha.sh $SCRIPT_URL; then
+            NEW_VERSION=$(grep '^VERSION=' /tmp/Nlbencha.sh | cut -d'"' -f2)
+            sed -i "s/^CURRENT_VERSION=.*/CURRENT_VERSION=\"$NEW_VERSION\"/" "$0"
+            
+            # 替换脚本文件
+            if mv /tmp/Nlbench_beta.sh "$0"; then
+                chmod +x "$0"
+                echo -e "${GREEN}脚本更新成功！新版本: $NEW_VERSION${NC}"
+                echo -e "${YELLOW}请等待 3 秒...${NC}"
+                sleep 3
+                exec bash "$0"
+            else
+                echo -e "${RED}无法替换脚本文件。请检查权限。${NC}"
+                sleep 2
+                return 1
+            fi
+        else
+            echo -e "${RED}下载新版本失败。请稍后重试。${NC}"
+            sleep 2
+            return 1
+        fi
+    else
+        echo -e "${GREEN}脚本已是最新版本 $CURRENT_VERSION。${NC}"
+        sleep 2
+    fi
+}
+
 # 检查并安装依赖
 install_dependencies() {
     echo -e "${YELLOW}正在检查并安装必要的依赖项...${NC}"
@@ -479,6 +526,9 @@ show_welcome() {
 main() {
     # 检查并安装依赖
     install_dependencies
+
+    # 更新脚本
+    update_scripts
     
     # 获取统计数据
     sum_run_times