Update main.py

main.py

@@ -171,23 +171,66 @@ remotePort = 8188
 
 
 def install_frp():
-    frp_tar_path = os.path.join(
-        REPO_PATH,
-        "frp_0.61.0_linux_amd64.tar.gz"
-    )
+    frp_tar_path = os.path.join(REPO_PATH, "frp_0.61.0_linux_amd64.tar.gz")
+    frp_install_path = os.path.join(REPO_PATH, "frp_0.61.0_linux_amd64")
 
-    frp_install_path = os.path.join(
-        REPO_PATH,
-        "frp_0.61.0_linux_amd64"
-    )
+    frp_download_urls = [
+        "https://github.com/fatedier/frp/releases/download/v0.61.0/frp_0.61.0_linux_amd64.tar.gz",
+        "https://gh.llkk.cc/https://github.com/fatedier/frp/releases/download/v0.61.0/frp_0.61.0_linux_amd64.tar.gz",
+        "https://gh-proxy.com/https://github.com/fatedier/frp/releases/download/v0.61.0/frp_0.61.0_linux_amd64.tar.gz"
+    ]
 
     if os.path.exists(frp_install_path):
         print(f"frp 已安装: {frp_install_path}")
         return True
 
-    if not os.path.exists(frp_tar_path):
-        print(f"未找到 frp 安装包: {frp_tar_path}")
-        return False
+    def is_valid_tar_gz(path):
+        if not os.path.exists(path):
+            return False
+
+        try:
+            result = subprocess.run(
+                ["tar", "-tzf", path],
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True
+            )
+            return result.returncode == 0
+        except Exception:
+            return False
+
+    if not is_valid_tar_gz(frp_tar_path):
+        print("frp 压缩包不存在或不是有效 gzip 文件，正在重新下载...")
+
+        if os.path.exists(frp_tar_path):
+            os.remove(frp_tar_path)
+
+        download_success = False
+
+        for url in frp_download_urls:
+            try:
+                print(f"尝试下载: {url}")
+
+                result = subprocess.run(
+                    ["wget", "-O", frp_tar_path, url],
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.PIPE,
+                    text=True
+                )
+
+                if result.returncode == 0 and is_valid_tar_gz(frp_tar_path):
+                    print("frp 下载成功。")
+                    download_success = True
+                    break
+
+                print("下载失败或文件无效，尝试下一个地址。")
+
+            except Exception as e:
+                print(f"下载出错: {e}")
+
+        if not download_success:
+            print("frp 下载失败，请检查网络。")
+            return False
 
     try:
         print(f"正在解压 frp 安装包到 {REPO_PATH}...")
@@ -197,14 +240,18 @@ def install_frp():
             check=True
         )
 
+        frpc_path = os.path.join(frp_install_path, "frpc")
+
+        if os.path.exists(frpc_path):
+            subprocess.run(["chmod", "+x", frpc_path], check=False)
+
         print("frp 安装成功！")
         return True
 
     except subprocess.CalledProcessError as e:
         print(f"解压 frp 时出错: {e}")
         return False
-
-
+        
 def start_jupyterlab_process():
     jupyter_script = os.path.join(REPO_PATH, "start_jupyterlab.py")