PLUS: проверка API понимает заглушку провайдера, переоткрытий не больше 5

plus/serve_muscriptor.py

@@ -22,6 +22,7 @@ import subprocess
 import sys
 import threading
 import time
+import urllib.error
 import urllib.request
 
 LINKS_FILE = "muscriptor_links.txt"
@@ -280,15 +281,31 @@ BROWSER_UA = ("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 "
               "(KHTML, like Gecko) Chrome/128.0 Safari/537.36")
 
 
+BYPASS_HEADERS = {"serveo-skip-browser-warning": "true"}
+PAGE_MARKS = ("continue to", "continue to site", "proceed", "phishing",
+              "browser warning", "serveo")
+
+
 def probe_api(url: str, timeout: float = 10.0) -> tuple[bool, str]:
-    """API через туннель: должен вернуть JSON со списком инструментов."""
+    """API через туннель: JSON со списком инструментов, заглушка — не поломка."""
+    request = urllib.request.Request(url + "/instruments", headers=BYPASS_HEADERS)
     try:
-        with urllib.request.urlopen(url + "/instruments", timeout=timeout) as answer:
-            if answer.status != 200:
-                return False, "API код %d" % answer.status
-            data = json.loads(answer.read().decode("utf-8", "ignore"))
+        with urllib.request.urlopen(request, timeout=timeout) as answer:
+            status = answer.status
+            body = answer.read().decode("utf-8", "ignore")
+    except urllib.error.HTTPError as problem:
+        return False, "API код %d" % problem.code
     except Exception as problem:
         return False, "API не ответил (%s)" % type(problem).__name__
+    if status != 200:
+        return False, "API код %d" % status
+    try:
+        data = json.loads(body)
+    except Exception:
+        low = body.lower()
+        if any(mark in low for mark in PAGE_MARKS):
+            return True, "API за страницей-предупреждением провайдера"
+        return False, "API отдал не JSON"
     names = data.get("instruments") if isinstance(data, dict) else None
     if not names:
         return False, "API отдал пустой список инструментов"
@@ -612,6 +629,8 @@ def main(argv: list[str] | None = None) -> int:
         if tunnel.url and SUBDOMAIN and SUBDOMAIN not in tunnel.url:
             say("      провайдер выдал другой адрес (свой поддомен занят): %s" % tunnel.url)
         if tunnel.url:
+            api_ok, api_note = probe_api(tunnel.url)
+            say("      API снаружи: %s" % api_note)
             ttfb, total, size = speed_probe(tunnel.url)
             if ttfb >= 0:
                 say("      скорость ссылки: первый байт %.1f с, страница %.1f с"
@@ -624,6 +643,7 @@ def main(argv: list[str] | None = None) -> int:
 
     started, beat = time.time(), time.time()
     public_fails, checked = 0, time.time()
+    reopens = 0
     try:
         while True:
             time.sleep(10)
@@ -655,14 +675,19 @@ def main(argv: list[str] | None = None) -> int:
                 else:
                     public_fails += 1
                     say("      ссылка снаружи: %s (%d подряд)" % (note, public_fails))
-                    if public_fails >= 2:
-                        say("      переоткрываю туннель: снаружи не отвечает")
+                    if public_fails >= 3 and reopens < 5:
+                        reopens += 1
+                        say("      переоткрываю туннель: снаружи не отвечает"
+                            " (раз %d из 5)" % reopens)
                         tunnel.kill()
                         tunnel = open_tunnel(proxy_port, order)
                         write_link(tunnel.url or "", args.port, proxy_port)
                         if tunnel.url:
                             say(">>> OPEN THIS LINK: " + tunnel.url)
                         public_fails = 0
+                    elif public_fails >= 3:
+                        say("      переоткрытий уже 5, оставляю этот туннель:"
+                            " проблема не в нём")
             if tunnel is not None and not tunnel.alive():
                 say("туннель умер, открываю новый")
                 tunnel.kill()