Upload 12 files

colab_tunnel/__init__.py

@@ -25,7 +25,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
 
 from ._config import WORK_FOLDER
 from ._logger import logger
-from ._registry import proxies_functions, tunnel_provider
+from ._registry import proxies_functions, tunnel_provider, list_providers, enable_provider, disable_provider
 from ._utils import run, download
 
 # Импорт модулей регистрирует провайдеров через @tunnel_provider
@@ -63,6 +63,9 @@ __all__ = [
     'benchmark',
     'TunnelBenchmarkResult',
     # Реестр (для расширения)
+    'disable_provider',
+    'enable_provider',
+    'list_providers',
     'proxies_functions',
     'tunnel_provider',
     # Провайдеры (прямой доступ)

colab_tunnel/_diagnostics.py

@@ -177,46 +177,51 @@ def _print_report(results: list[TunnelBenchmarkResult]) -> None:
 def benchmark(
     providers: list[str] | None = None,
     timeout_per_provider: float = 30.0,
-    latency_timeout: float = 15.0,
+    latency_timeout: float = 10.0,
     print_report: bool = True,
 ) -> list[TunnelBenchmarkResult]:
     """
     Последовательно тестирует провайдеров туннелей и собирает метрики.
 
-    Схема работы:
-    1. Находит свободный порт, запускает тестовый HTTP-сервер (бинарник
-       из папки test_server/).
-    2. Поочерёдно вызывает каждого провайдера, замеряя время запуска.
-    3. Проверяет доступность полученной ссылки, измеряет RTT.
-    4. Гарантированно останавливает тестовый сервер (блок finally).
-    5. Выводит форматированный отчёт.
+    По умолчанию тестирует только активных провайдеров (из proxies_functions).
+    Передать список имён явно — включая отключённых — можно через параметр providers.
 
-    Последовательный (не параллельный) порядок важен для точности замеров:
-    параллельный запуск исказил бы время за счёт конкуренции за сеть и CPU.
+    Схема работы:
+    1. Запускает тестовый HTTP-сервер.
+    2. Последовательно запускает каждого провайдера, измеряя время старта.
+    3. Сразу проверяет RTT полученного URL.
+    4. После всех провайдеров — второй проход RTT для тех, кто не ответил
+       с первого раза (cloudflared, tunnelite: маршрут распространяется 10–15с).
+    5. Останавливает тестовый сервер (блок finally).
+    6. Выводит форматированный отчёт.
 
     Args:
-        providers:            Имена провайдеров для теста. None — все доступные.
+        providers:            Имена провайдеров. None — все активные.
         timeout_per_provider: Таймаут ожидания ссылки от провайдера (сек).
-        latency_timeout:      Таймаут проверки доступности URL (сек).
-        print_report:         Вывести форматированную таблицу в конце.
+        latency_timeout:      Таймаут проверки RTT (сек).
+        print_report:         Вывести таблицу результатов.
 
     Returns:
-        Список TunnelBenchmarkResult для каждого протестированного провайдера.
-
-    Raises:
-        FileNotFoundError: Бинарник тестового сервера не найден.
-        ValueError:        Указан несуществующий провайдер.
+        Список TunnelBenchmarkResult.
     """
     server_bin = _find_test_server_bin()
     port       = find_free_port()
 
-    names = providers or list(proxies_functions.keys())
-    unknown = [n for n in names if n not in proxies_functions]
-    if unknown:
-        raise ValueError(
-            f'Неизвестные провайдеры: {unknown}. '
-            f'Доступные: {list(proxies_functions.keys())}'
-        )
+    # Если провайдеры переданы явно — ищем в _all_providers (включая отключённых).
+    # Если нет — берём только активных из proxies_functions.
+    if providers is not None:
+        from ._registry import _all_providers
+        unknown = [n for n in providers if n not in _all_providers]
+        if unknown:
+            raise ValueError(
+                f'Неизвестные провайдеры: {unknown}. '
+                f'Все зарегистрированные: {list(_all_providers)}'
+            )
+        names = providers
+        func_map = _all_providers
+    else:
+        names    = list(proxies_functions.keys())
+        func_map = proxies_functions
 
     print(f'Бенчмарк: {len(names)} провайдеров, порт {port}.')
     _start_test_server(server_bin, port)
@@ -225,8 +230,9 @@ def benchmark(
     results: list[TunnelBenchmarkResult] = []
 
     try:
+        # ── Основной проход: запуск + первая проверка RTT ────────────────
         for name in names:
-            func = proxies_functions[name]
+            func = func_map[name]
             print(f'[{name:<12}] запуск...', end=' ', flush=True)
             result = TunnelBenchmarkResult(name=name)
             t_start = time.time()
@@ -252,6 +258,21 @@ def benchmark(
 
             results.append(result)
 
+        # ── Второй проход: RTT для провайдеров с warmup-задержкой ────────
+        # К этому моменту прошло достаточно времени (пока тестировались
+        # остальные провайдеры), чтобы маршруты cloudflared/tunnelite
+        # успели распространиться по сети провайдера.
+        needs_recheck = [
+            r for r in results
+            if r.success and r.latency is None and r.name != 'native'
+        ]
+        if needs_recheck:
+            print(f'\nПовторная проверка RTT ({len(needs_recheck)} провайдеров)...')
+            for r in needs_recheck:
+                r.latency = _check_url_latency(r.url, timeout=latency_timeout)  # type: ignore[arg-type]
+                status = f'RTT {r.latency * 1000:.0f}мс' if r.latency else 'всё ещё недоступен'
+                print(f'  [{r.name:<12}] {status}')
+
     finally:
         print()
         _stop_test_server(server_bin)
@@ -261,3 +282,4 @@ def benchmark(
         _print_report(results)
 
     return results
+

colab_tunnel/_registry.py

@@ -1,25 +1,87 @@
 """
 Реестр провайдеров туннелей.
 
-Новый провайдер регистрируется декоратором:
+Регистрация нового провайдера:
+    @tunnel_provider('myprovider')
+    def get_myprovider_url(port: int) -> str: ...
 
-    from colab_tunnel._registry import tunnel_provider
+Отключить провайдера (сервис временно недоступен):
+    @tunnel_provider('tunwg', enabled=False)
+    # или в рантайме:
+    disable_provider('tunwg')
 
-    @tunnel_provider('myprovider')
-    def get_myprovider_url(port: int) -> str:
-        ...
+Включить обратно:
+    enable_provider('tunwg')
 
-После этого провайдер автоматически доступен через get_share_link() и try_all().
+Посмотреть статус всех:
+    list_providers()
 """
 from typing import Callable
 
-# Заполняется при импорте модулей _tunnels и _ssh через декоратор @tunnel_provider
+# Активные провайдеры — используются get_share_link(), try_all(), benchmark()
 proxies_functions: dict[str, Callable[[int], str]] = {}
 
+# Все зарегистрированные провайдеры (активные + отключённые)
+_all_providers: dict[str, Callable[[int], str]] = {}
+
+# Имена отключённых провайдеров
+_disabled_providers: set[str] = set()
+
 
-def tunnel_provider(name: str) -> Callable:
-    """Декоратор регистрации провайдера туннеля по имени."""
+def tunnel_provider(name: str, *, enabled: bool = True) -> Callable:
+    """
+    Декоратор регистрации провайдера туннеля.
+
+    Args:
+        name:    Уникальное имя провайдера.
+        enabled: False — провайдер зарегистрирован, но не активен.
+                 Включить позднее через enable_provider(name).
+    """
     def decorator(func: Callable[[int], str]) -> Callable[[int], str]:
-        proxies_functions[name] = func
+        _all_providers[name] = func
+        if enabled:
+            proxies_functions[name] = func
+        else:
+            _disabled_providers.add(name)
         return func
     return decorator
+
+
+def disable_provider(name: str) -> None:
+    """
+    Деактивирует провайдера: убирает из активного реестра.
+    Функция сохраняется — включить обратно через enable_provider().
+
+    Raises:
+        KeyError: Провайдер с таким именем не зарегистрирован.
+    """
+    if name not in _all_providers:
+        raise KeyError(f'Провайдер "{name}" не зарегистрирован. Доступные: {list(_all_providers)}')
+    proxies_functions.pop(name, None)
+    _disabled_providers.add(name)
+
+
+def enable_provider(name: str) -> None:
+    """
+    Активирует ранее отключённого провайдера.
+
+    Raises:
+        KeyError: Провайдер с таким именем не зарегистрирован.
+    """
+    if name not in _all_providers:
+        raise KeyError(f'Провайдер "{name}" не зарегистрирован. Доступные: {list(_all_providers)}')
+    _disabled_providers.discard(name)
+    proxies_functions[name] = _all_providers[name]
+
+
+def list_providers() -> dict[str, bool]:
+    """
+    Возвращает статус всех зарегистрированных провайдеров.
+
+    Returns:
+        Словарь {имя: активен}, в порядке регистрации.
+
+    Пример вывода:
+        {'cloudflared': True, 'tunwg': False, 'bore': True, ...}
+    """
+    return {name: (name not in _disabled_providers) for name in _all_providers}

colab_tunnel/_tunnels.py

@@ -169,7 +169,7 @@ def get_tmole_url(port: int) -> str:
     )
 
 
-@tunnel_provider('tunwg')
+@tunnel_provider('tunwg', enabled=False)  # Сервис недоступен с 2026-06-16, включить: enable_provider('tunwg')
 def get_tunwg_url(port: int) -> str:
     return get_revproxy_url(
         bin_url='https://github.com/ntnj/tunwg/releases/latest/download/tunwg',