auto port-forward and teardown

Author: willcl-ark

src/warnet/control.py

@@ -15,6 +15,7 @@
 from rich.table import Table
 
 from .constants import COMMANDER_CHART
+from .deploy import _port_stop_internal
 from .k8s import (
     get_default_namespace,
     get_mission,
@@ -162,6 +163,8 @@ def delete_pod(pod_name, namespace):
         for future in as_completed(futures):
             console.print(f"[yellow]{future.result()}[/yellow]")
 
+    # Shutdown any port forwarding
+    _port_stop_internal()
     console.print("[bold yellow]Teardown process initiated for all components.[/bold yellow]")
     console.print("[bold yellow]Note: Some processes may continue in the background.[/bold yellow]")
     console.print("[bold green]Warnet teardown process completed.[/bold green]")

src/warnet/deploy.py

@@ -1,3 +1,6 @@
+import os
+import subprocess
+import sys
 import tempfile
 from pathlib import Path
 
@@ -16,7 +19,7 @@
     NAMESPACES_FILE,
     NETWORK_FILE,
 )
-from .k8s import get_default_namespace
+from .k8s import get_default_namespace, wait_for_caddy_ready
 from .process import stream_command
 
 
@@ -102,14 +105,18 @@ def deploy_caddy(directory: Path, debug: bool):
         return
 
     namespace = get_default_namespace()
-    cmd = f"{HELM_COMMAND} 'caddy' {CADDY_CHART} --namespace {namespace}"
+    name = "caddy"
+    cmd = f"{HELM_COMMAND} {name} {CADDY_CHART} --namespace {namespace}"
     if debug:
         cmd += " --debug"
 
     if not stream_command(cmd):
         click.echo(f"Failed to run Helm command: {cmd}")
         return
 
+    wait_for_caddy_ready(name, namespace)
+    _port_start_internal()
+
 
 def deploy_fork_observer(directory: Path, debug: bool):
     network_file_path = directory / NETWORK_FILE
@@ -242,3 +249,43 @@ def deploy_namespaces(directory: Path):
         finally:
             if temp_override_file_path.exists():
                 temp_override_file_path.unlink()
+
+
+def is_windows():
+    return sys.platform.startswith("win")
+
+
+def run_detached_process(command):
+    if is_windows():
+        # For Windows, use CREATE_NEW_PROCESS_GROUP and DETACHED_PROCESS
+        subprocess.Popen(
+            command,
+            shell=True,
+            stdin=None,
+            stdout=None,
+            stderr=None,
+            close_fds=True,
+            creationflags=subprocess.CREATE_NEW_PROCESS_GROUP | subprocess.DETACHED_PROCESS,
+        )
+    else:
+        # For Unix-like systems, use nohup and redirect output
+        command = f"nohup {command} > /dev/null 2>&1 &"
+        subprocess.Popen(command, shell=True, stdin=None, stdout=None, stderr=None, close_fds=True)
+
+    print(f"Started detached process: {command}")
+
+
+def _port_start_internal():
+    command = "kubectl port-forward service/caddy 2019:80"
+    run_detached_process(command)
+    click.echo(
+        "Port forwarding on port 2019 started in the background. To access landing page visit localhost:2019."
+    )
+
+
+def _port_stop_internal():
+    if is_windows():
+        os.system("taskkill /F /IM kubectl.exe")
+    else:
+        os.system("pkill -f 'kubectl port-forward service/caddy-service 2019:80'")
+    click.echo("Port forwarding stopped.")

src/warnet/k8s.py

@@ -4,7 +4,7 @@
 from pathlib import Path
 
 import yaml
-from kubernetes import client, config
+from kubernetes import client, config, watch
 from kubernetes.client.models import CoreV1Event, V1PodList
 from kubernetes.dynamic import DynamicClient
 from kubernetes.stream import stream
@@ -227,3 +227,21 @@ def snapshot_bitcoin_datadir(
 
     except Exception as e:
         print(f"An error occurred: {str(e)}")
+
+
+def wait_for_caddy_ready(name, namespace, timeout=300):
+    sclient = get_static_client()
+    w = watch.Watch()
+    for event in w.stream(
+        sclient.list_namespaced_pod, namespace=namespace, timeout_seconds=timeout
+    ):
+        pod = event["object"]
+        if pod.metadata.name == name and pod.status.phase == "Running":
+            conditions = pod.status.conditions or []
+            ready_condition = next((c for c in conditions if c.type == "Ready"), None)
+            if ready_condition and ready_condition.status == "True":
+                print(f"Caddy pod {name} is ready.")
+                w.stop()
+                return True
+    print(f"Timeout waiting for Caddy pod {name} to be ready.")
+    return False