auto mixin

app.py

@@ -1,5 +1,7 @@
 import os
+import re
 import sys
+import yaml
 import requests
 from flask import Flask, Response, request
 from urllib.parse import urlencode
@@ -11,6 +13,67 @@ app = Flask(__name__)
 API_KEY = os.environ.get('API_KEY')
 SUBSCRIBE_URLS = os.environ.get('SUBSCRIBE_URLS')  # 存储真实URL的变量
 
+def subscribe_mixin(content: str) -> str:
+    """输入 YAML 字符串，输出转换后的 YAML 字符串
+    """
+    try:
+        d = yaml.safe_load(content)
+
+        my_auto_group_name = "AI Unrestrict"
+        regex_list = [
+            re.compile(r"美国", re.IGNORECASE),     # 美国 (不区分大小写)
+            re.compile(r"america", re.IGNORECASE),  # america (不区分大小写)
+            re.compile(r"us", re.IGNORECASE),       # us (不区分大小写)
+            re.compile(r"新加坡", re.IGNORECASE),   # 新加坡 (不区分大小写)
+            re.compile(r"singapore", re.IGNORECASE),  # singapore (不区分大小写)
+            re.compile(r"sg", re.IGNORECASE),       # sg (不区分大小写)
+            re.compile(r"加拿大", re.IGNORECASE),   # 加拿大 (不区分大小写)
+            re.compile(r"canada", re.IGNORECASE),   # canada (不区分大小写)
+            re.compile(r"ca", re.IGNORECASE),       # ca (不区分大小写)
+        ]
+        matching_proxies = []  # 用于存储匹配的代理名称
+
+        # 1. 查找并保存符合正则表达式的 proxy name
+        if "proxies" in d and isinstance(d["proxies"], list):
+            for proxy in d["proxies"]:
+                if "name" in proxy:
+                    for regex in regex_list:
+                        if regex.search(proxy["name"]):  # 使用 re.search
+                            matching_proxies.append(proxy["name"])
+                            break  # 匹配到一个 regex 就跳出循环，避免重复添加
+
+        # 2. 创建新的 proxy-group 对象
+        new_proxy_group = {
+            "name": my_auto_group_name,
+            "type": "url-test",
+            "proxies": matching_proxies,
+            "url": "http://www.gstatic.com/generate_204",
+            "interval": 60  # interval 是整数
+        }
+
+        # 3. 将新的 proxy-group 添加到 proxy-groups 数组
+        if "proxy-groups" in d and isinstance(d["proxy-groups"], list):
+            d["proxy-groups"].append(new_proxy_group)
+
+            # 4. 将 myAutoGroupName 添加到第一个 proxy-group 的 "proxies" 列表的最前面
+            if d["proxy-groups"] and len(d["proxy-groups"]) > 0 and \
+               "proxies" in d["proxy-groups"][0] and isinstance(d["proxy-groups"][0]["proxies"], list):
+                d["proxy-groups"][0]["proxies"].insert(0, my_auto_group_name)  # 使用 insert(0, ...)
+        else:
+            d["proxy-groups"] = [new_proxy_group]  # 如果 proxy-groups 不存在，则创建
+
+        # 将修改后的字典转换回 YAML 字符串
+        modified_yaml = yaml.dump(d, allow_unicode=True, indent=2)  # 使用 yaml.dump
+
+        return modified_yaml
+
+    except yaml.YAMLError as e:
+        print(f"YAML 解析错误：{e}")
+        return ""  # 或者抛出异常，根据你的需求
+    except Exception as e:
+        print(f"其他错误：{e}")
+        return ""
+
 @app.get("/getsub")
 def read_subscribe():
     # 验证API Key
@@ -36,6 +99,7 @@ def read_subscribe():
         resp = requests.get(target_url)
         resp.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
         data = resp.text
+        data = subscribe_mixin(data)
         return Response(data, mimetype='text/yaml')
     except requests.exceptions.RequestException as e:
         return {"error": str(e)}, 500  # Handle request errors and return an error response

requirements.txt

@@ -1,5 +1,6 @@
 
 flask
-requests
+pyyaml
 asgiref
+requests
 uvicorn[standard]==0.17.*