Refactor server state and add health checks

scripts/smoke_test.py

@@ -0,0 +1,81 @@
+import json
+import os
+import subprocess
+import sys
+import time
+import urllib.request
+
+
+PORT = os.environ.get("TELEGRAM_SCRAPER_SMOKE_PORT", "18080")
+BASE_URL = f"http://127.0.0.1:{PORT}"
+ENDPOINTS = [
+    "/",
+    "/viewer",
+    "/swagger",
+    "/openapi.json",
+    "/health",
+    "/health/continuous",
+    "/api/dashboard",
+    "/api/continuous",
+    "/api/channels",
+]
+
+
+def fetch(path: str) -> tuple[int, bytes]:
+    with urllib.request.urlopen(BASE_URL + path, timeout=3) as response:
+        return response.status, response.read()
+
+
+def wait_for_server(process: subprocess.Popen) -> None:
+    deadline = time.time() + 15
+    last_error = None
+    while time.time() < deadline:
+        if process.poll() is not None:
+            raise RuntimeError(f"server exited with code {process.returncode}")
+        try:
+            fetch("/health")
+            return
+        except Exception as exc:
+            last_error = exc
+            time.sleep(0.3)
+    raise RuntimeError(f"server did not become ready: {last_error}")
+
+
+def main() -> int:
+    env = {
+        **os.environ,
+        "TELEGRAM_SCRAPER_PORT": PORT,
+        "TELEGRAM_SCRAPER_START_CONTINUOUS": "0",
+    }
+    process = subprocess.Popen(
+        [sys.executable, "main.py"],
+        env=env,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.STDOUT,
+        text=True,
+    )
+    try:
+        wait_for_server(process)
+        for endpoint in ENDPOINTS:
+            status, body = fetch(endpoint)
+            if status != 200:
+                raise RuntimeError(f"{endpoint} returned HTTP {status}")
+            if endpoint.endswith(".json") or endpoint.startswith("/api") or endpoint.startswith("/health"):
+                json.loads(body.decode("utf-8"))
+            print(f"ok {endpoint}")
+        return 0
+    except Exception as exc:
+        print(f"smoke test failed: {exc}", file=sys.stderr)
+        if process.stdout:
+            print(process.stdout.read(), file=sys.stderr)
+        return 1
+    finally:
+        process.terminate()
+        try:
+            process.wait(timeout=5)
+        except subprocess.TimeoutExpired:
+            process.kill()
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())