Upload 17 files

app/main.py

@@ -7,11 +7,16 @@ Simple Clash Relay - Flask 应用入口
 
 import os
 import logging
-from flask import Flask, request, jsonify, Response
+import time
+import signal
+import sys
+import json
+import requests
+from flask import Flask, request, jsonify, Response, send_from_directory
+from werkzeug.exceptions import HTTPException
 from .clash_manager import ClashManager
 from .sub_manager import SubscriptionManager
 from .auth import authenticate
-import requests
 
 # 配置日志
 logging.basicConfig(
@@ -421,6 +426,87 @@ def index():
     </html>
     """
 
+# --- Clash API 反向代理 --- 
+@app.route('/ui/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE', 'OPTIONS'])
+def proxy_clash_api(path):
+    """将 /ui/ 下的请求转发给内部的 Clash API (127.0.0.1:9090)"""
+    if not clash_manager:
+        return jsonify({"error": "Clash Manager 未初始化"}), 503
+        
+    clash_api_base_url = f"http://127.0.0.1:{clash_manager.api_port}/"
+    target_url = clash_api_base_url + path
+    
+    # 准备转发请求头，移除 Host 头，可能需要添加原始 Host 或 Forwarded 头
+    headers = {key: value for key, value in request.headers if key.lower() != 'host'}
+    # 确保传递原始的 X-Real-IP 或 X-Forwarded-For (如果存在)
+    # headers['X-Forwarded-For'] = request.remote_addr
+    
+    # 处理 Websocket 连接 (Clash API 日志/流量推送可能使用)
+    if request.headers.get('Upgrade', '').lower() == 'websocket':
+        logger.info(f"检测到 WebSocket 请求: {path}")
+        # Flask 本身不直接支持代理 WebSocket，需要额外库或特殊处理
+        # 这里返回错误，表明此代理不支持 WebSocket
+        # 可以考虑使用 flask_sock 或其他库来支持
+        logger.warning("当前反向代理不支持 WebSocket 连接")
+        return jsonify({"error": "WebSocket proxying not supported"}), 501 # Not Implemented
+    
+    try:
+        resp = requests.request(
+            method=request.method,
+            url=target_url,
+            headers=headers,
+            data=request.get_data(),
+            params=request.args,
+            stream=True,  # 使用流式传输以处理大数据和流式API (如日志)
+            timeout=30 # 设置超时
+        )
+
+        # 准备转发响应头
+        response_headers = []
+        for name, value in resp.raw.headers.items():
+            # 避免转发某些头 (例如 'Transfer-Encoding': 'chunked' 可能导致问题)
+            if name.lower() not in ['content-encoding', 'content-length', 'transfer-encoding', 'connection']:
+                response_headers.append((name, value))
+        
+        # 创建Flask响应对象
+        # 使用 resp.content 而不是迭代器，对于非流式响应更简单可靠
+        response = Response(resp.content, status=resp.status_code, headers=response_headers)
+        return response
+
+    except requests.exceptions.ConnectionError:
+        logger.error(f"无法连接到内部Clash API: {target_url}")
+        return jsonify({"error": "无法连接到内部 Clash API"}), 502 # Bad Gateway
+    except requests.exceptions.Timeout:
+        logger.error(f"连接到内部Clash API超时: {target_url}")
+        return jsonify({"error": "内部 Clash API 请求超时"}), 504 # Gateway Timeout
+    except Exception as e:
+        logger.error(f"代理到 Clash API 时出错: {str(e)}")
+        return jsonify({"error": f"代理请求失败: {str(e)}"}), 500
+
+# --- 错误处理 ---
+@app.errorhandler(HTTPException)
+def handle_exception(e):
+    """返回JSON格式的HTTP错误"""
+    response = e.get_response()
+    response.data = json.dumps({
+        "code": e.code,
+        "name": e.name,
+        "description": e.description,
+    })
+    response.content_type = "application/json"
+    return response
+
+@app.errorhandler(Exception)
+def handle_generic_exception(e):
+    """处理未捕获的异常"""
+    logger.error(f"未捕获的异常: {str(e)}", exc_info=True)
+    # 对于非HTTP异常，返回500 Internal Server Error
+    return jsonify({
+        "code": 500,
+        "name": "Internal Server Error",
+        "description": f"发生内部错误: {str(e)}"
+    }), 500
+
 if __name__ == "__main__":
     # 如果直接运行此文件，将初始化应用并启动Flask服务器
     initialize()

requirements.txt

@@ -4,4 +4,7 @@ requests==2.28.1
 Werkzeug==2.0.1
 PyYAML==6.0.1
 # pyyaml==6.0 # 已通过 apk 安装 
-# 添加与 Flask==2.0.1 兼容的版本 
+# 添加与 Flask==2.0.1 兼容的版本 
+
+# 确保 requests 库被包含，因为反向代理需要它
+# (requests 已在上面，无需重复添加)