add Subscription-Userinfo support

.dockerignore

@@ -1,3 +1,6 @@
+certs
+test
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[codz]

.env.example

@@ -7,6 +7,31 @@ URI_PATH=/sub/               # Базовый путь URL для подписк
                              # Должен начинаться и заканчиваться на '/'.
                              # Пример: https://sub.example.com/sub/username
 
+# Время кеширования внешних подписок (в секундах). По умолчанию 1 час.
+SUBSCRIPTION_CACHE_TTL=3600
+
+# ГЛОБАЛЬНЫЕ МЕТАДАННЫЕ ПОДПИСКИ
+# Ссылка на поддержку (отображается в клиентах)
+SUPPORT_URL=https://t.me/your_support
+
+# Ссылка на профиль / панель управления
+PROFILE_WEB_PAGE_URL=https://panel.example.com
+
+# Объявление / важное сообщение (показывается в некоторых клиентах)
+ANNOUNCE=Это объявление
+
+# Как часто клиенты должны обновлять подписку (в часах)
+UPDATE_INTERVAL=12
+
+# Информация об использовании трафика по умолчанию
+# Используются при отсутствии настоящих данных
+# Формат: upload=XXX; download=YYY; total=ZZZ; expire=UNIX_TIMESTAMP
+SUBSCRIPTION_USERINFO=upload=0; download=0; total=0; expire=0
+
+# Ссылка на импорт маршрутизации для клиента Happ (happ://routing/add/...)
+# Скопируй полную ссылку из Happ или сгенерируй самостоятельно
+HAPP_ROUTING_LINK=happ://routing/add/eyJibG9ja2lwIjpbXSwiYmxvY2tzaXRlcyI6W10sImRpcmVjdGlwIjpbIjEwLjAuMC4wLzgiLCIxNzIuMTYuMC4wLzEyIiwiMTkyLjE2OC4wLjAvMTYiLCIxNjkuMjU0LjAuMC8xNiIsIjIyNC4wLjAuMC80IiwiMjU1LjI1NS4yNTUuMjU1IiwiZ2VvaXA6cnUiXSwiZGlyZWN0c2l0ZXMiOlsiZ2Vvc2l0ZTpjYXRlZ29yeS1ydSIsIioubG9jYWwiXSwiZG5zaG9zdHMiOnsiY2xvdWRmbGFyZS1kbnMuY29tIjoiMS4xLjEuMSIsImRucy5nb29nbGUiOiI4LjguOC44In0sImRvbWFpbnN0cmF0ZWd5IjoiSVBJZk5vbk1hdGNoIiwiZG9tZXN0aWNkbnNkb21haW4iOiJodHRwczovL2Rucy5nb29nbGUvZG5zLXF1ZXJ5IiwiZG9tZXN0aWNkbnNpcCI6IjguOC44LjgiLCJkb21lc3RpY2Ruc3R5cGUiOiJEb0giLCJmYWtlZG5zIjpmYWxzZSwiZ2VvaXB1cmwiOiJodHRwczovL2dpdGh1Yi5jb20vTG95YWxzb2xkaWVyL3YycmF5LXJ1bGVzLWRhdGEvcmVsZWFzZXMvbGF0ZXN0L2Rvd25sb2FkL2dlb2lwLmRhdCIsImdlb3NpdGV1cmwiOiJodHRwczovL2dpdGh1Yi5jb20vTG95YWxzb2xkaWVyL3YycmF5LXJ1bGVzLWRhdGEvcmVsZWFzZXMvbGF0ZXN0L2Rvd25sb2FkL2dlb3NpdGUuZGF0IiwiZ2xvYmFscHJveHkiOnRydWUsIm5hbWUiOiJydSwgbG9jYWwiLCJwcm94eWlwIjpbXSwicHJveHlzaXRlcyI6W10sInJlbW90ZWRuc2RvbWFpbiI6Imh0dHBzOi8vY2xvdWRmbGFyZS1kbnMuY29tL2Rucy1xdWVyeSIsInJlbW90ZWRuc2lwIjoiMS4xLjEuMSIsInJlbW90ZWRuc3R5cGUiOiJEb0giLCJyb3V0ZW9yZGVyIjoiYmxvY2stZGlyZWN0LXByb3h5In0=
+
 # =====================================================================
 
 # URLS — основные ссылки и конфигурации, которые будут возвращаться пользователю

.gitignore

@@ -1,3 +1,6 @@
+certs
+test
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[codz]

config.py

@@ -11,3 +11,16 @@ LISTEN_HOST = os.getenv('LISTEN_HOST', '0.0.0.0')
 LISTEN_PORT = int(os.getenv('LISTEN_PORT', '2096'))
 URI_PATH = os.getenv('URI_PATH', '/sub/')
 URLS = json.loads(os.getenv('URLS', '{}'))
+
+# TTL кеша внешних подписок в секундах (по умолчанию 1 час)
+SUBSCRIPTION_CACHE_TTL = int(os.getenv('SUBSCRIPTION_CACHE_TTL', '3600'))
+
+# метаданные подписки
+SUPPORT_URL = os.getenv('SUPPORT_URL', '')
+PROFILE_WEB_PAGE_URL = os.getenv('PROFILE_WEB_PAGE_URL', '')
+ANNOUNCE = os.getenv('ANNOUNCE', '')
+UPDATE_INTERVAL = int(os.getenv('UPDATE_INTERVAL', '12'))          # в часах
+HAPP_ROUTING_LINK = os.getenv('HAPP_ROUTING_LINK', '')             # полный happ://routing/add/...
+
+# Fallback значение Subscription-Userinfo, если ни одна внешняя подписка не вернула свои данные
+SUBSCRIPTION_USERINFO = os.getenv('SUBSCRIPTION_USERINFO', '')

docker-compose.yml

@@ -0,0 +1,36 @@
+version: '3.8'
+
+services:
+  xray-sub-server:
+    image: crutoboy/xray_sub_server:latest
+    container_name: xray-sub-server
+    restart: unless-stopped
+    ports:
+      - "2096:2096"
+    volumes:
+      # Монтируем папку с сертификатами
+      # Измените на свои. например,
+      # - /etc/letsencrypt/live/example.com:/certs:ro
+      - ./certs:/certs:ro
+    # Запуск через gunicorn с SSL
+    command: >
+      gunicorn
+      --workers=4
+      --bind=0.0.0.0:2096
+      --keyfile=/certs/privkey.pem
+      --certfile=/certs/fullchain.pem
+      wsgi:app
+    env_file:
+      - .env
+    # Пример с явным указанием переменных (альтернатива env_file):
+    # environment:
+    #   LISTEN_HOST: 0.0.0.0
+    #   LISTEN_PORT: 2096
+    #   URI_PATH: /sub/
+    #   SUBSCRIPTION_CACHE_TTL: 3600
+    #   SUPPORT_URL: https://t.me/your_support
+    #   PROFILE_WEB_PAGE_URL: https://panel.example.com
+    #   ANNOUNCE: Это объявление
+    #   UPDATE_INTERVAL: 12
+    #   HAPP_ROUTING_LINK: happ://routing/add/...
+    #   URLS: '{"all": [...]}'   # большой JSON лучше хранить в .env файле

main.py

@@ -2,7 +2,9 @@ from typing import List
 import base64
 
 import flask
+from flask import make_response
 import requests
+from cachetools import TTLCache, cached
 
 import config as c
 
@@ -15,33 +17,141 @@ def index():
     ...
 
 
-def get_subs_from_server(link: str) -> List[str]:
+# Кеш внешних подписок с TTL (по умолчанию 1 час)
+_subs_cache = TTLCache(maxsize=256, ttl=c.SUBSCRIPTION_CACHE_TTL)
+@cached(_subs_cache)
+def get_subs_from_server(link: str):
+    """
+    Получает внешнюю подписку.
+    Возвращает кортеж: (список нод, Subscription-Userinfo из заголовка или None)
+    """
     try:
-        sub_response = requests.get(link)
+        sub_response = requests.get(link, timeout=10)
-        decoded = base64.b64decode(sub_response.text).decode('utf-8')
+        userinfo = sub_response.headers.get('Subscription-Userinfo') or \
-        res = decoded.split('\n')
+                   sub_response.headers.get('subscription-userinfo')
-        res = list(filter(None, res))
+
-        return res
+        decoded = base64.b64decode(sub_response.text).decode('utf-8')
-    except:
+        nodes = [line for line in decoded.split('\n') if line.strip()]
-        return []
+
+        return nodes, userinfo
+    except Exception:
+        # При ошибке не кешируем и пробуем заново в следующий раз
+        _subs_cache.pop(link, None)
+        return [], None
+
+def format_urls(urls: List[str], user: str):
+    """
+    Возвращает (список всех нод, список найденных Subscription-Userinfo из внешних подписок)
+    """
+    all_nodes = []
+    userinfos = []
 
-def format_urls(urls: List[str], user: str) -> List[str]:
-    res = []
     for url in urls:
         url = url.strip().format(user)
         if url.startswith('https://'):
-            subs = get_subs_from_server(url)
+            nodes, userinfo = get_subs_from_server(url)
-            res += subs
+            all_nodes += nodes
-        else: 
+            if userinfo:
-            res.append(url)
+                userinfos.append(userinfo)
-    return res 
+        else:
+            all_nodes.append(url)
+
+    return all_nodes, userinfos
+
+def _parse_userinfo(s: str) -> dict:
+    """Парсит строку вида 'upload=123; download=456; total=789; expire=1234567890'"""
+    data = {}
+    for part in s.split(';'):
+        part = part.strip()
+        if '=' in part:
+            k, v = part.split('=', 1)
+            try:
+                data[k.strip().lower()] = int(v.strip())
+            except ValueError:
+                pass
+    return data
+
+
+def _merge_userinfo(infos: list[str]) -> str | None:
+    """Объединяет несколько Subscription-Userinfo (суммирует трафик, берёт минимальные лимиты)."""
+    if not infos:
+        return None
+
+    total_upload = 0
+    total_download = 0
+    min_total = None
+    min_expire = None
+
+    for info in infos:
+        parsed = _parse_userinfo(info)
+        total_upload += parsed.get('upload', 0)
+        total_download += parsed.get('download', 0)
+
+        t = parsed.get('total')
+        if t is not None and t > 0:
+            min_total = t if min_total is None else min(min_total, t)
+
+        e = parsed.get('expire')
+        if e is not None and e > 0:
+            min_expire = e if min_expire is None else min(min_expire, e)
+
+    parts = [
+        f"upload={total_upload}",
+        f"download={total_download}",
+    ]
+
+    if min_total is not None:
+        parts.append(f"total={min_total}")
+    else:
+        parts.append("total=0")
+
+    if min_expire is not None:
+        parts.append(f"expire={min_expire}")
+    else:
+        parts.append("expire=0")
+
+    return '; '.join(parts)
+
 
 @app.route(f'{c.URI_PATH}<user>')
 def get_subs(user: str):
-    urls = format_urls(c.URLS.get(user, []) + c.URLS.get('all', []), user)
+    nodes, upstream_userinfos = format_urls(
-    urls_text = '\n'.join(urls)
+        c.URLS.get(user, []) + c.URLS.get('all', []), user
-    res = base64.b64encode(bytes(urls_text, 'utf-8'))
+    )
-    return res
+
+    urls_text = '\n'.join(nodes)
+    encoded = base64.b64encode(bytes(urls_text, 'utf-8'))
+
+    resp = make_response(encoded)
+    resp.headers['Content-Type'] = 'text/plain; charset=utf-8'
+
+    # === Динамический Userinfo из внешних подписок ===
+    dynamic_userinfo = _merge_userinfo(upstream_userinfos)
+    if dynamic_userinfo:
+        resp.headers['Subscription-Userinfo'] = dynamic_userinfo
+    elif c.SUBSCRIPTION_USERINFO:  # fallback на статическое значение, если есть
+        resp.headers['Subscription-Userinfo'] = c.SUBSCRIPTION_USERINFO
+
+    if c.UPDATE_INTERVAL:
+        resp.headers['Profile-Update-Interval'] = str(c.UPDATE_INTERVAL)
+
+    if c.SUPPORT_URL:
+        resp.headers['Support-Url'] = c.SUPPORT_URL
+
+    if c.PROFILE_WEB_PAGE_URL:
+        resp.headers['Profile-Web-Page-Url'] = c.PROFILE_WEB_PAGE_URL
+
+    if c.ANNOUNCE:
+        announce_bytes = base64.b64encode(bytes(c.ANNOUNCE, "utf-8"))
+        announce_encode = announce_bytes.decode('ascii')
+        resp.headers['Announce'] = f'base64:{announce_encode}'
+
+    if c.HAPP_ROUTING_LINK:
+        resp.headers['Routing'] = c.HAPP_ROUTING_LINK
+        resp.headers['Routing-Enable'] = 'true'
+
+    return resp
+
 
 
 if __name__ == '__main__':

requirements.txt

@@ -11,3 +11,4 @@ python-dotenv==1.2.2
 requests==2.34.2
 urllib3==2.7.0
 Werkzeug==3.1.8
+cachetools==5.5.2