dev/0.0.2 #1
@@ -1,4 +1,5 @@
|
||||
certs
|
||||
test
|
||||
|
||||
# Byte-compiled / optimized / DLL files
|
||||
__pycache__/
|
||||
|
||||
25
.env.example
25
.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 — основные ссылки и конфигурации, которые будут возвращаться пользователю
|
||||
|
||||
1
.gitignore
vendored
1
.gitignore
vendored
@@ -1,4 +1,5 @@
|
||||
certs
|
||||
test
|
||||
|
||||
# Byte-compiled / optimized / DLL files
|
||||
__pycache__/
|
||||
|
||||
13
config.py
13
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', '')
|
||||
|
||||
@@ -27,4 +27,10 @@ services:
|
||||
# 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 файле
|
||||
|
||||
148
main.py
148
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)
|
||||
decoded = base64.b64decode(sub_response.text).decode('utf-8')
|
||||
res = decoded.split('\n')
|
||||
res = list(filter(None, res))
|
||||
return res
|
||||
except:
|
||||
return []
|
||||
sub_response = requests.get(link, timeout=10)
|
||||
userinfo = sub_response.headers.get('Subscription-Userinfo') or \
|
||||
sub_response.headers.get('subscription-userinfo')
|
||||
|
||||
decoded = base64.b64decode(sub_response.text).decode('utf-8')
|
||||
nodes = [line for line in decoded.split('\n') if line.strip()]
|
||||
|
||||
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)
|
||||
res += subs
|
||||
else:
|
||||
res.append(url)
|
||||
return res
|
||||
nodes, userinfo = get_subs_from_server(url)
|
||||
all_nodes += nodes
|
||||
if userinfo:
|
||||
userinfos.append(userinfo)
|
||||
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)
|
||||
urls_text = '\n'.join(urls)
|
||||
res = base64.b64encode(bytes(urls_text, 'utf-8'))
|
||||
return res
|
||||
nodes, upstream_userinfos = format_urls(
|
||||
c.URLS.get(user, []) + c.URLS.get('all', []), user
|
||||
)
|
||||
|
||||
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__':
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user