Upload 12 files

colab_tunnel/_diagnostics.py

@@ -81,23 +81,39 @@ def _stop_test_server(server_bin: Path) -> None:
     logger.debug('Тестовый сервер остановлен.')
 
 
-def _check_url_latency(url: str, timeout: float = 15.0) -> Optional[float]:
+def _check_url_latency(
+    url: str,
+    timeout: float = 10.0,
+    warmup_retries: int = 2,
+    warmup_delay: float = 5.0,
+) -> Optional[float]:
     """
-    Делает GET-запрос к публичному URL и возвращает RTT.
+    Проверяет доступность публичного URL и возвращает RTT.
+
+    Делает повторные попытки с паузой — некоторые провайдеры (cloudflared,
+    tunnelite) анонсируют URL раньше, чем маршрут фактически распространяется
+    по их сети. Cloudflared обычно требует 5–10 секунд после старта.
 
     Returns:
-        Время ответа в секундах, или None если URL недоступен.
+        RTT в секундах, или None если URL недоступен после всех попыток.
     """
-    # Убираем строку с IPv4-паролем (если провайдер вернул её как вторую строку)
+    # Убираем строку с IPv4-паролем, если провайдер вернул её второй строкой
     clean_url = url.split('\n')[0].strip()
-    try:
-        t = time.time()
-        resp = get_url(clean_url, timeout=timeout, allow_redirects=True)
-        elapsed = round(time.time() - t, 3)
-        if 200 <= resp.status_code < 400:
-            return elapsed
-    except Exception as e:
-        logger.debug(f'URL недоступен ({clean_url}): {e}')
+
+    for attempt in range(warmup_retries):
+        if attempt > 0:
+            logger.debug(f'Повтор проверки доступности (попытка {attempt + 1}): {clean_url}')
+            time.sleep(warmup_delay)
+        try:
+            t = time.time()
+            resp = get_url(clean_url, timeout=timeout, allow_redirects=True)
+            elapsed = round(time.time() - t, 3)
+            if 200 <= resp.status_code < 400:
+                return elapsed
+            logger.debug(f'HTTP {resp.status_code} от {clean_url}')
+        except Exception as e:
+            logger.debug(f'Попытка {attempt + 1}/{warmup_retries} не удалась: {e}')
+
     return None
 
 
@@ -145,6 +161,12 @@ def _print_report(results: list[TunnelBenchmarkResult]) -> None:
         if with_lat:
             lowest = min(with_lat, key=lambda r: r.latency)  # type: ignore[arg-type]
             print(f'Минимальная задержка: {lowest.name} ({lowest.latency * 1000:.0f}мс)')  # type: ignore[operator]
+
+    # Пояснение для native — его URL требует сессионные куки Google Colab
+    native_result = next((r for r in results if r.name == 'native'), None)
+    if native_result and native_result.success and not native_result.latency:
+        print('\n  ℹ native: URL работает, но требует Google-авторизации —')
+        print('    внешняя проверка без куки браузера всегда будет недоступна.')
     print()
 
 

colab_tunnel/_tunnels.py

@@ -36,6 +36,7 @@ def get_revproxy_url(
     timeout: float = 20.0,
     write_link: bool = False,
     stdin_input: str | None = None,
+    extra_env: dict[str, str] | None = None,
 ) -> str:
     """
     Универсальный запуск бинарного туннельного провайдера.
@@ -53,6 +54,8 @@ def get_revproxy_url(
         timeout:          Максимальное время ожидания URL (сек).
         write_link:       Сохранять ли ссылку в links.txt.
         stdin_input:      Строка для отправки в stdin процесса.
+        extra_env: Дополнительные переменные окружения для процесса.
+                   Накладываются поверх os.environ (не заменяют его).
 
     Returns:
         Публичный URL туннеля.
@@ -90,11 +93,18 @@ def get_revproxy_url(
 
     # ── Завершение предыдущего экземпляра ───────────────────────────────
     kill_process_by_name(bin_path.name)
-
+    # ── Подготовка окружения ─────────────────────────────────────────────
+    import os as _os
+    proc_env = _os.environ.copy()
+    if extra_env:
+        proc_env.update(extra_env)
     # ── Запуск процесса ──────────────────────────────────────────────────
     stdin_flag = PIPE if stdin_input is not None else None
-    process = Popen(start_commands, stdout=PIPE, stderr=PIPE, stdin=stdin_flag)
-
+    process = Popen(
+        start_commands,
+        stdout=PIPE, stderr=PIPE, stdin=stdin_flag,
+        env=proc_env,  # ← передаём окружение
+    )
     if stdin_input is not None and process.stdin:
         try:
             process.stdin.write(stdin_input.encode())
@@ -325,6 +335,9 @@ def get_bore_url(port: int) -> str:
         read_from_stderr=True,  # Rust логи идут в stderr
         url_pattern=r'bore\.pub:\d+',
         timeout=15.0,
+        # Bore использует Rust-крейт `tracing`. Без RUST_LOG=info строка
+        # "listening at bore.pub:XXXXX" (INFO-уровень) не выводится совсем.
+        extra_env={'RUST_LOG': 'info'},
     )
     return f'http://{raw}/'
 

pyproject.toml

@@ -10,7 +10,7 @@ readme = "README.md"
 requires-python = ">=3.10"
 license = { text = "MIT" }
 dependencies = [
-    "requests",
+    "requests>=2.32.4",
 ]
 
 [project.optional-dependencies]