添加了远程 Mixin 的功能

app.py

@@ -1,8 +1,10 @@
 import os
 import re
+import ast
 import yaml
 import httpx
 import base64
+import inspect
 from dotenv import load_dotenv
 from fastapi import FastAPI, Response, HTTPException, Request
 
@@ -23,8 +25,63 @@ app = FastAPI()
 
 # 从环境变量中获取密钥
 load_dotenv()
-API_KEY = os.environ.get('API_KEY')
-SUBSCRIBE_URLS = os.environ.get('SUBSCRIBE_URLS')  # 存储真实URL的变量
+API_KEY         = os.environ.get('API_KEY')
+SUBSCRIBE_URLS  = os.environ.get('SUBSCRIBE_URLS')  # 存储真实URL的变量
+INJECT_SCRIPT   = os.environ.get('INJECT_SCRIPT')
+
+def validate_function_signature(code_string):
+  """ 使用 ast 验证函数签名是否符合要求。
+  """
+  try:
+    module = ast.parse(code_string)
+    if not module.body:
+      print("错误：代码为空。")
+      return False
+    if len(module.body) != 1 or not isinstance(module.body[0], ast.FunctionDef):
+      print("错误：代码必须包含单个函数定义。")
+      return False
+    function_def = module.body[0]
+    # Step 01. 检查函数名
+    if function_def.name != "handle_mixin":
+      print(f"错误：函数名必须为 'handle_mixin'，而不是 '{function_def.name}'。")
+      return False
+    # Step 02. 检查参数个数
+    args = function_def.args
+    if args.defaults or args.kw_defaults or args.vararg or args.kwarg or args.posonlyargs:
+      print("错误：函数不能有默认参数、可变位置参数、可变关键字参数、或者仅位置参数")
+      return False
+    if len(args.args) != 1:
+      print(f"错误：函数必须恰好接受一个参数，而不是 {len(args.args)} 个。")
+      return False
+    # Step 03. 检查参数名称
+    if args.args[0].arg != "content":
+      print(f"错误：参数名称必须为 'content'，而不是 '{args.args[0].arg}'。")
+      return False
+    return True
+  except SyntaxError as e:
+    print(f"语法错误：{e}")
+    return False
+  except Exception as e:
+    print(f"发生错误：{e}")
+    return False
+
+def load_mixin_function(code_string) -> callable:
+  if not validate_function_signature(code_string):
+    return
+  try:
+    code_obj = compile(code_string, '<string>', 'exec')
+    local_namespace = {}
+    exec(code_obj, local_namespace)
+    handle_mixin = local_namespace['handle_mixin']
+    return handle_mixin
+  except Exception as e:
+    print(f"解析函数时出错：{e}")
+
+try:
+    handle_mixin = load_mixin_function(INJECT_SCRIPT)
+except Exception as e:
+    handle_mixin = lambda x:x
+    print(f"无法加载 MINIX 函数, 错误信息: {e}")
 
 def subscribe_mixin(content: str) -> str:
     """输入 YAML 字符串，输出转换后的 YAML 字符串
@@ -76,6 +133,13 @@ def subscribe_mixin(content: str) -> str:
             d["proxy-groups"] = [new_proxy_group]
 
         d.pop('socks-port', None)
+
+        # 在此处尝试调用 mixin 函数
+        try:
+            d = handle_mixin(d)
+        except Exception as e:
+            print(f"执行 Minix 函数时出错!")
+
         modified_yaml = yaml.dump(d, allow_unicode=True, indent=2)
 
         return modified_yaml