add a small script to pre-generate the mitmproxy certs

octotail/main.py

@@ -4,6 +4,7 @@
 import dataclasses
 import multiprocessing as mp
 import sys
+import time
 from pathlib import Path
 from threading import Event
 from typing import Dict, Union
@@ -18,7 +19,7 @@
 from octotail.cli import Opts, entrypoint
 from octotail.fmt import Formatter
 from octotail.gh import JobDone, RunWatcher, WorkflowDone, get_active_run, guess_repo
-from octotail.mitm import ProxyWatcher, WsSub
+from octotail.mitm import MITM_CONFIG_DIR, ProxyWatcher, WsSub
 from octotail.streamer import OutputItem, run_streamer
 from octotail.utils import debug, find_free_port, log
 
@@ -134,5 +135,22 @@ def _main(opts: Opts) -> None:
     ActorRegistry.stop_all()
 
 
+def generate_cert() -> None:
+    # start the proxy_watcher actor and wait until a cert is generated
+    port = find_free_port()
+    mitm = ProxyWatcher.start(None, port)
+    cert_file = MITM_CONFIG_DIR / "mitmproxy-ca-cert.cer"
+    tries = 0
+    while (not cert_file.exists()) or (not cert_file.stat().st_size):
+        if tries > 25:
+            debug("giving up waiting for mitmproxy to generate a certificate")
+            sys.exit(1)
+        time.sleep(0.2)
+        tries += 1
+
+    log(f"{cert_file}", skip_prefix=True)
+    mitm.stop()
+
+
 if __name__ == "__main__":
     _main()

octotail/mitm.py

@@ -46,14 +46,16 @@ class ProxyWatcher(ThreadingActor):
     _proxy_ps: multiprocessing.Process
     _q: multiprocessing.Queue
 
-    def __init__(self, mgr: ActorRef, port: int):
+    def __init__(self, mgr: ActorRef | None, port: int):
         super().__init__()
-        self.mgr = mgr
+        if mgr is not None:
+            self.mgr = mgr
+            self._stop_event = mgr.proxy().stop_event.get()
         self.port = port
-        self._stop_event = mgr.proxy().stop_event.get()
 
     def on_start(self) -> None:
         self._q = multiprocessing.Queue()
+        MITM_CONFIG_DIR.mkdir(exist_ok=True, parents=True)
         self._proxy_ps = multiprocessing.Process(
             target=_mitmdump_wrapper, args=(self._q, self.port)
         )

pyproject.toml

@@ -31,6 +31,7 @@ dependencies = [
 
 [project.scripts]
 octotail = "octotail.main:_main"
+generate-cert = "octotail.main:generate_cert"
 
 [project.urls]
 Homepage = "https://github.com/rarescosma/octotail"