diff --git a/python/app.py b/python/app.py
index 49d2661..c9755ad 100644
--- a/python/app.py
+++ b/python/app.py
@@ -1,56 +1,271 @@
-import sys
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
import os
+import sys
+import signal
+import time
+import stat
import subprocess
+import urllib.request
+import urllib.error
import http.server
import socketserver
import threading
+import platform
+from pathlib import Path
-PORT = int(os.environ.get('PORT') or 3000) # http port
+# 环境变量配置
+PORT = int(os.environ.get('PORT', 3000)) # http服务端口
+SUB_PATH = os.environ.get('SUB_PATH', 'sub') # 订阅token
+config = {
+ 'UUID': os.environ.get('UUID', 'b5c445d1-8e59-465f-af0c-1f4193d15693'), # 节点UUID，使用哪吒v1时在不不同的平台部署需要修改，否则agent会覆盖
+ 'NEZHA_SERVER': os.environ.get('NEZHA_SERVER', ''), # 哪吒面板地址，v1格式: nezha.xxx.com:8008 v0格式： nezha.xxx.com
+ 'NEZHA_PORT': os.environ.get('NEZHA_PORT', ''), # 哪吒v1请留空，哪吒v0 agent端口
+ 'NEZHA_KEY': os.environ.get('NEZHA_KEY', ''), # 哪吒v1的NZ_CLIENT_SECRET或哪吒v0-agent密钥
+ 'ARGO_DOMAIN': os.environ.get('ARGO_DOMAIN', ''), # 固定隧道域名,留空即启用临时隧道
+ 'ARGO_AUTH': os.environ.get('ARGO_AUTH', ''), # 固定隧道token或json,留空即启用临时隧道,json获取:https://json.zone.id
+ 'ARGO_PORT': os.environ.get('ARGO_PORT', '8001'), # argo端口 使用固定隧道token,cloudflare后台设置的端口需和这里对应
+ 'CFIP': os.environ.get('CFIP', 'saas.sin.fan'), # 优选域名或优选ip
+ 'CFPORT': os.environ.get('CFPORT', '443'), # 优选域名或优选ip对应端口
+ 'NAME': os.environ.get('NAME', ''), # 节点备注
+ 'S5_PORT': os.environ.get('S5_PORT', ''), # socks5端口,支持多端口玩具可填写，否则不动
+ 'HY2_PORT': os.environ.get('HY2_PORT', ''), # Hy2 端口，支持多端口玩具可填写，否则不动
+ 'TUIC_PORT': os.environ.get('TUIC_PORT', ''), # Tuic 端口，支持多端口玩具可填写，否则不动
+ 'ANYTLS_PORT': os.environ.get('ANYTLS_PORT', ''), # AnyTLS 端口,支持多端口玩具可填写，否则不动
+ 'REALITY_PORT': os.environ.get('REALITY_PORT', ''), # Reality 端口,支持多端口玩具可填写，否则不动
+ 'ANYREALITY_PORT': os.environ.get('ANYREALITY_PORT', ''), # AnyReality 端口,支持多端口玩具可填写，否则不动
+ 'CHAT_ID': os.environ.get('CHAT_ID', ''), # TG chat_id，可在https://t.me/laowang_serv00_bot 获取
+ 'BOT_TOKEN': os.environ.get('BOT_TOKEN', ''), # TG bot_token, 使用自己的bot需要填写,使用上方的bot不用填写,不会给别人发送
+ 'UPLOAD_URL': os.environ.get('UPLOAD_URL', ''), # 节点上传地址，需部署merge-sub订阅器项目，例如：https://merge.xxx.com
+ 'FILE_PATH': os.environ.get('FILE_PATH', '.cache'), # sub,.txt节点存放目录
+ 'DISABLE_ARGO': os.environ.get('DISABLE_ARGO', 'false'), # 是否禁用argo, true为禁用,false为不禁用,默认开启
+}
-class MyHandler(http.server.SimpleHTTPRequestHandler):
+def sleep(ms):
+ time.sleep(ms / 1000)
- def log_message(self, format, *args):
- pass
+def get_architecture():
+ """获取系统架构"""
+ arch = platform.machine().lower()
+ system = platform.system().lower()
+
+ if system in ['linux', 'darwin']:
+ if arch in ['x86_64', 'amd64']:
+ return 'amd64'
+ elif arch in ['aarch64', 'arm64']:
+ return 'arm64'
+
+ raise Exception(f"Unsupported architecture: {system} {arch}")
+def download_file(url, dest_path):
+ print(f"Downloading from: {url}")
+ opener = urllib.request.build_opener()
+ opener.addheaders = [
+ ('User-Agent', 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'),
+ ('Accept', '*/*'),
+ ('Connection', 'keep-alive')
+ ]
+
+ urllib.request.install_opener(opener)
+
+ try:
+ # 下载文件
+ urllib.request.urlretrieve(url, dest_path)
+ # print("Download completed!")
+ except urllib.error.HTTPError as e:
+ if os.path.exists(dest_path):
+ os.unlink(dest_path)
+ raise Exception(f"Download failed (HTTP {e.code}): {e.reason}")
+ except urllib.error.URLError as e:
+ if os.path.exists(dest_path):
+ os.unlink(dest_path)
+ raise Exception(f"Download failed (URL error): {str(e)}")
+ except Exception as e:
+ if os.path.exists(dest_path):
+ os.unlink(dest_path)
+ raise Exception(f"Download failed: {str(e)}")
+
+def set_executable(file_path):
+ """设置文件可执行权限"""
+ try:
+ current_permissions = os.stat(file_path).st_mode
+ os.chmod(file_path, current_permissions | stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH)
+ except Exception as e:
+ raise Exception(f"Failed to set executable permission: {str(e)}")
+
+def delete_file(file_path):
+ """删除文件"""
+ try:
+ if os.path.exists(file_path):
+ os.unlink(file_path)
+ except Exception as e:
+ print(f"Failed to delete file {file_path}: {str(e)}")
+
+class CustomHTTPRequestHandler(http.server.SimpleHTTPRequestHandler):
+ """自定义HTTP请求处理器"""
+
def do_GET(self):
if self.path == '/':
- self.send_response(200)
- self.end_headers()
- self.wfile.write(b'Hello, world')
- elif self.path == '/sub':
- try:
- with open("./sub.txt", 'rb') as file:
- content = file.read()
- self.send_response(200)
- self.send_header('Content-Type', 'text/plain; charset=utf-8')
- self.end_headers()
- self.wfile.write(content)
- except FileNotFoundError:
- self.send_response(500)
- self.end_headers()
- self.wfile.write(b'Error reading file')
+ self.handle_root()
+ elif self.path == f'/{SUB_PATH}':
+ self.handle_sub()
+ elif self.path == '/ps':
+ self.handle_ps()
else:
- self.send_response(404)
+ self.send_error(404, '404 Not Found')
+
+ def handle_root(self):
+ """处理根路径请求"""
+ try:
+ html_path = os.path.join(os.path.dirname(__file__), 'index.html')
+ if os.path.exists(html_path):
+ with open(html_path, 'r', encoding='utf-8') as f:
+ data = f.read()
+ self.send_response(200)
+ self.send_header('Content-Type', 'text/html')
+ self.end_headers()
+ self.wfile.write(data.encode('utf-8'))
+ else:
+ self.send_response(200)
+ self.send_header('Content-Type', 'text/html; charset=utf-8')
+ self.end_headers()
+ self.wfile.write("Hello world!
You can access /{SUB_PATH}(Default: /sub) to get your nodes!".encode('utf-8'))
+ except Exception as e:
+ self.send_error(500, str(e))
+
+ def handle_sub(self):
+ """处理订阅路径请求"""
+ sub_file_path = os.path.join(config['FILE_PATH'], 'sub.txt')
+ try:
+ if os.path.exists(sub_file_path):
+ with open(sub_file_path, 'r', encoding='utf-8') as f:
+ data = f.read()
+ self.send_response(200)
+ self.send_header('Content-Type', 'text/plain')
+ self.end_headers()
+ self.wfile.write(data.encode('utf-8'))
+ else:
+ self.send_error(404, f'Sub file not found at: {sub_file_path}')
+ except Exception as e:
+ self.send_error(500, str(e))
+
+ def handle_ps(self):
+ """处理进程列表请求"""
+ try:
+ result = subprocess.run(['ps', 'aux'], capture_output=True, text=True, check=True)
+ self.send_response(200)
+ self.send_header('Content-Type', 'text/plain')
self.end_headers()
- self.wfile.write(b'Not found')
-httpd = socketserver.TCPServer(('', PORT), MyHandler)
-server_thread = threading.Thread(target=httpd.serve_forever)
-server_thread.daemon = True
-server_thread.start()
+ self.wfile.write(result.stdout.encode('utf-8'))
+ except subprocess.CalledProcessError as e:
+ self.send_error(500, f'Error executing ps command: {str(e)}')
+
+ def log_message(self, format, *args):
+ """覆盖日志方法，不输出访问日志"""
+ pass
-shell_command = "chmod +x start.sh && ./start.sh"
+def start_http_server():
+ """启动HTTP服务器"""
+ handler = CustomHTTPRequestHandler
+
+ # 创建服务器，允许地址重用
+ socketserver.TCPServer.allow_reuse_address = True
+ httpd = socketserver.TCPServer(('0.0.0.0', PORT), handler)
+
+ server_thread = threading.Thread(target=httpd.serve_forever)
+ server_thread.daemon = True
+ server_thread.start()
+
+ return httpd
-try:
- completed_process = subprocess.run(['bash', '-c', shell_command], stdout=sys.stdout, stderr=subprocess.PIPE, text=True, check=True)
+def cleanup(binary_path):
+ """清理二进制文件"""
+ delete_file(binary_path)
- print("App is running")
+def main():
+ """主函数"""
+ binary_path = None
+ httpd = None
+ process = None
+
+ try:
+ # 启动HTTP服务器
+ httpd = start_http_server()
+
+ # 获取架构并下载
+ arch = get_architecture()
+ download_url = 'https://amd64.eooce.com/sbsh' if arch == 'amd64' else 'https://arm64.eooce.com/sbsh'
+
+ # print(f"Using download link: {download_url}")
+ binary_path = os.path.join(os.getcwd(), 'sbsh')
+
+ download_file(download_url, binary_path)
+ if not os.path.exists(binary_path):
+ raise Exception('Download failed, file does not exist at the specified path')
+
+ set_executable(binary_path)
+
+ # 准备环境变量
+ env = os.environ.copy()
+ env.update({k: str(v) for k, v in config.items()})
+
+ # 启动子进程
+ process = subprocess.Popen(
+ [binary_path],
+ env=env,
+ stdout=subprocess.PIPE,
+ stderr=subprocess.STDOUT,
+ universal_newlines=True,
+ bufsize=1
+ )
+
+ # 实时输出子进程日志
+ def log_output():
+ for line in process.stdout:
+ print(line, end='')
+
+ log_thread = threading.Thread(target=log_output)
+ log_thread.daemon = True
+ log_thread.start()
+
+ sleep(18000)
+ print('\nLogs will be deleted in 90 seconds, you can copy the above nodes!')
+
+ sleep(90000)
+
+ cleanup(binary_path)
+
+ # 清除控制台
+ os.system('clear' if os.name == 'posix' else 'cls')
+
+ print('✅ App is running')
+ print(f'🌐 HTTP server is running on {PORT}')
+
+ # 保持主线程运行
+ try:
+ while True:
+ time.sleep(1)
+ except KeyboardInterrupt:
+ print("\nShutting down...")
+
+ except Exception as e:
+ print(f"\n❌ An error occurred: {str(e)}")
+ if binary_path:
+ cleanup(binary_path)
+ sys.exit(1)
+ finally:
+ if process and process.poll() is None:
+ process.terminate()
+ if httpd:
+ httpd.shutdown()
-except subprocess.CalledProcessError as e:
- print(f"Error: {e.returncode}")
- print("Standard Output:")
- print(e.stdout)
- print("Standard Error:")
- print(e.stderr)
- sys.exit(1)
+def signal_handler(signum, frame):
+ print("\nReceived signal to terminate")
+ sys.exit(0)
-server_thread.join()
+if __name__ == "__main__":
+ signal.signal(signal.SIGINT, signal_handler)
+ signal.signal(signal.SIGTERM, signal_handler)
+
+ main()