From e7f8a9f984547b0045fd637dc94aa5c62cc07ab0 Mon Sep 17 00:00:00 2001
From: Jensfrank <everett7623@gmail.com>
Date: Fri, 12 Jul 2024 23:46:41 +0800
Subject: [PATCH] Update Nlbench.sh

---
 Nlbench.sh | 49 +++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 43 insertions(+), 6 deletions(-)

diff --git a/Nlbench.sh b/Nlbench.sh
index 750421c..5640bfb 100644
--- a/Nlbench.sh
+++ b/Nlbench.sh
@@ -19,14 +19,16 @@ if [ "$(id -u)" != "0" ]; then
     echo "已获取 sudo 权限。"
 fi
 
-# 检查并安装依赖
+# 更新系统并安装依赖
 install_dependencies() {
     echo -e "${YELLOW}正在检查并安装必要的依赖项...${NC}"
     
-    # 更新包列表
-    if ! sudo apt-get update; then
-        echo -e "${RED}无法更新包列表。请检查您的网络连接和系统设置。${NC}"
-        exit 1
+    # 更新系统
+    echo -e "${YELLOW}正在更新系统...${NC}"
+    if update_system; then
+        echo -e "${GREEN}系统更新完成。${NC}"
+    else
+        echo -e "${RED}系统更新失败。继续安装依赖项。${NC}"
     fi
     
     # 安装依赖
@@ -39,7 +41,7 @@ install_dependencies() {
     for dep in "${dependencies[@]}"; do
         if ! command -v "$dep" &> /dev/null; then
             echo -e "${YELLOW}正在安装 $dep...${NC}"
-            if ! sudo apt-get install -y "$dep"; then
+            if ! install_package "$dep"; then
                 echo -e "${RED}无法安装 $dep。请手动安装此依赖项。${NC}"
             fi
         else
@@ -51,6 +53,41 @@ install_dependencies() {
     clear
 }
 
+# 更新系统
+update_system() {
+    if command -v apt &>/dev/null; then
+        sudo apt-get update && sudo apt-get upgrade -y
+    elif command -v dnf &>/dev/null; then
+        sudo dnf check-update && sudo dnf upgrade -y
+    elif command -v yum &>/dev/null; then
+        sudo yum check-update && sudo yum upgrade -y
+    elif command -v apk &>/dev/null; then
+        sudo apk update && sudo apk upgrade
+    else
+        echo -e "${RED}不支持的Linux发行版${NC}"
+        return 1
+    fi
+    return 0
+}
+
+# 安装包
+install_package() {
+    local package=$1
+    if command -v apt &>/dev/null; then
+        sudo apt-get install -y "$package"
+    elif command -v dnf &>/dev/null; then
+        sudo dnf install -y "$package"
+    elif command -v yum &>/dev/null; then
+        sudo yum install -y "$package"
+    elif command -v apk &>/dev/null; then
+        sudo apk add "$package"
+    else
+        echo -e "${RED}不支持的Linux发行版${NC}"
+        return 1
+    fi
+    return 0
+}
+
 # 获取IP地址和ISP信息
 ip_address_and_isp() {
     ipv4_address=$(curl -s --max-time 5 ipv4.ip.sb)