Update app.py

app.py

@@ -1,115 +1,19 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-
-"""
-服务器端说明：
-1. 启动一个服务器监听指定端口（本例中为8765），同时支持普通的 HTTP 请求和 WebSocket 请求。
-2. 当请求为 HTTP 请求时，返回简单文本“网站运行中”。
-3. 当请求为 WebSocket 请求时，则按照原来的逻辑：
-   - 等待客户端发送连接请求（JSON格式），其中包含目标地址和端口；
-   - 尝试建立到目标地址的 TCP 连接；
-   - 回复连接状态，并进行双向数据转发（WebSocket <--> 目标TCP连接）。
-"""
-
-import asyncio
-import websockets
-import json
-
-# ------------------------------------------------
-# 处理普通 HTTP 请求的函数
-# ------------------------------------------------
-async def process_request(path, request_headers):
-    """
-    process_request 用于处理非 WebSocket 升级请求。
-    如果请求头中没有 Upgrade:websocket，则认为是普通 HTTP 请求，
-    返回状态码200及简单文本“网站运行中”。
-    如果是 WebSocket 升级请求，则返回 None 由后续 WebSocket 处理。
-    """
-    # 检查是否为 WebSocket 升级请求（Upgrade 头部存在且值为 websocket）
-    if request_headers.get("Upgrade", "").lower() == "websocket":
-        return None  # 表示继续执行 WebSocket 握手
-    # 构造 HTTP 响应内容
-    body = "网站运行中".encode("utf-8")
-    headers = [
-        ("Content-Type", "text/plain; charset=utf-8"),
-        ("Content-Length", str(len(body)))
-    ]
-    # 返回 HTTP 响应：状态码、头部、响应体
-    return 200, headers, body
-
-# ------------------------------------------------
-# 处理 WebSocket 连接的函数（原代码逻辑）
-# ------------------------------------------------
-async def handle_ws(websocket: websockets.WebSocketServerProtocol, path: str):
-    try:
-        # --- 1. 等待客户端发送连接请求信息 ---
-        msg = await websocket.recv()
-        req = json.loads(msg)
-        if req.get("cmd") != "connect":
-            await websocket.send(json.dumps({"cmd": "connect", "status": "error", "error": "invalid command"}))
-            return
-
-        dest_addr = req.get("address")
-        dest_port = req.get("port")
-        print(f"收到连接请求：目标 {dest_addr}:{dest_port}")
-
-        # --- 2. 尝试建立到目标的TCP连接 ---
-        try:
-            reader, writer = await asyncio.open_connection(dest_addr, dest_port)
-        except Exception as e:
-            error_msg = f"连接目标失败：{e}"
-            print(error_msg)
-            await websocket.send(json.dumps({"cmd": "connect", "status": "error", "error": str(e)}))
-            return
-
-        # --- 3. 回复客户端连接成功 ---
-        await websocket.send(json.dumps({"cmd": "connect", "status": "ok"}))
-
-        # --- 4. 双向转发数据（WebSocket <--> 目标TCP连接） ---
-        async def ws_to_remote():
-            try:
-                async for message in websocket:
-                    # 如果收到的是文本消息，则转为二进制数据发送给目标
-                    if isinstance(message, str):
-                        data = message.encode()
-                    else:
-                        data = message
-                    writer.write(data)
-                    await writer.drain()
-            except Exception as e:
-                print("ws_to_remote 出现异常：", e)
-            writer.close()
-
-        async def remote_to_ws():
-            try:
-                while True:
-                    data = await reader.read(1024)
-                    if not data:
-                        break
-                    await websocket.send(data)
-            except Exception as e:
-                print("remote_to_ws 出现异常：", e)
-            await websocket.close()
-
-        # 同时运行两个数据转发任务，实现全双工通信
-        await asyncio.gather(ws_to_remote(), remote_to_ws())
-
-    except Exception as e:
-        print("handle_ws 出现异常：", e)
-
-# ------------------------------------------------
-# 主函数，启动服务器
-# ------------------------------------------------
-async def main():
-    # 使用 websockets.serve 启动服务器，绑定 process_request 参数处理 HTTP 请求
-    ws_server = await websockets.serve(
-        handle_ws,            # WebSocket 连接处理函数
-        "0.0.0.0",            # 监听所有网卡
-        7860,                 # 监听端口8765（可自行修改）
-        process_request=process_request  # 用于处理非 WebSocket 请求
-    )
-    print("服务器已启动，监听端口8765，支持 HTTP 和 WebSocket 请求")
-    await ws_server.wait_closed()
-
-if __name__ == '__main__':
-    asyncio.run(main())
+# app.py
+import subprocess
+import os
+import time
+
+if __name__ == "__main__":
+    # 第一步：给可执行文件赋予执行权限
+    os.system("chmod +x socks5_http")
+
+    # 第二步：以指定端口7860启动 .NET 7 自包含应用
+    # 如果想兼容Hugging Face自动映射端口，可改为：
+    # port = os.environ.get("PORT", "7860")
+    # p = subprocess.Popen(["./socks5_http", "--urls", f"http://0.0.0.0:{port}"])
+    p = subprocess.Popen(["./socks5_http", "--urls", "http://0.0.0.0:7860"])
+
+    # 第三步：阻塞当前脚本，让进程保持运行状态
+    # 如果脚本结束，Hugging Face容器会退出
+    while True:
+        time.sleep(60)