f

cmdnat/zerotier.py

@@ -0,0 +1,151 @@
+from __future__ import annotations
+
+import os
+from typing import Any, Dict, List, Optional, Sequence, Tuple
+
+from cmdlet._shared import Cmdlet, CmdletArg
+from SYS.result_table import ResultTable
+from SYS.config import load_config
+from API import zerotier as zt
+
+def exec(pipe: Sequence[Any], args: Sequence[str], options: Dict[str, Any]) -> ResultTable:
+    table = ResultTable(title="ZeroTier Status", max_columns=10)
+    
+    cfg = load_config()
+    
+    # 1. Local Node Status
+    node_id = "unknown"
+    try:
+        # Best effort to get local node ID
+        st = zt._run_cli_json("status")
+        if isinstance(st, dict):
+            node_id = st.get("address") or node_id
+    except: pass
+    
+    # 2. Hosting Status
+    zt_net = cfg.get("networking", {}).get("zerotier", {})
+    serve_target = zt_net.get("serve")
+    if serve_target:
+        port = zt_net.get("port") or 999
+        net_id = zt_net.get("network_id") or "all"
+        
+        status = "OFFLINE"
+        detail = ""
+        
+        # Try to find the local ZT address for this network
+        zt_addrs = []
+        if net_id and net_id != "all":
+            zt_addrs = zt.get_assigned_addresses(net_id)
+        
+        # We probe localhost for hosting status, but show ZT IP in the table
+        display_addr = zt_addrs[0] if zt_addrs else "localhost"
+        
+        # Try probes
+        # Using 127.0.0.1 is often more reliable than 'localhost' on Windows
+        probe_targets = [f"http://127.0.0.1:{port}/health"]
+        if zt_addrs:
+            probe_targets.insert(0, f"http://{zt_addrs[0]}:{port}/health")
+
+        from API.HTTP import HTTPClient
+        with HTTPClient(timeout=1.0, retries=0) as client:
+            for url in probe_targets:
+                try:
+                    resp = client.get(url)
+                    if resp.status_code == 200:
+                        status = "ONLINE"
+                        payload = resp.json()
+                        detail = f"Serving {payload.get('name') or serve_target}"
+                        break
+                    else:
+                        status = f"HTTP {resp.status_code}"
+                except Exception as exc:
+                    if not detail: # Keep the first failure reason if all fail
+                         detail = f"Probe failed: {exc}"
+
+        if status == "OFFLINE" and not zt_addrs:
+            detail = "No ZeroTier IP assigned yet. Check 'zerotier-cli listnetworks'."
+
+        row = table.add_row()
+        row.add_column("TYPE", "HOST")
+        row.add_column("NAME", serve_target)
+        row.add_column("ID", net_id)
+        row.add_column("ADDRESS", f"{display_addr}:{port}")
+        row.add_column("STATUS", status)
+        row.add_column("DETAIL", detail)
+
+    # 3. Connections (Remote Stores)
+    zt_stores = cfg.get("store", {}).get("zerotier", {})
+    if zt_stores:
+        for name, sconf in zt_stores.items():
+            net_id = sconf.get("NETWORK_ID") or sconf.get("network_id") or ""
+            host = sconf.get("HOST") or sconf.get("host") or ""
+            port = sconf.get("PORT") or sconf.get("port") or 999
+            svc = sconf.get("SERVICE") or sconf.get("service") or "remote"
+            
+            status = "probing..."
+            detail = ""
+            
+            if not host:
+                status = "MISCONFIGURED"
+                detail = "No host IP"
+            else:
+                try:
+                    from API.HTTP import HTTPClient
+                    with HTTPClient(timeout=2.0) as client:
+                        # Paths depend on service type
+                        path = "/api_version" if svc == "hydrus" else "/health"
+                        resp = client.get(f"http://{host}:{port}{path}")
+                        if resp.status_code == 200:
+                            status = "ONLINE"
+                            if svc == "remote":
+                                p = resp.json()
+                                detail = f"Remote store: {p.get('name', 'unknown')}"
+                            else:
+                                detail = "Hydrus API"
+                        else:
+                            status = f"HTTP {resp.status_code}"
+                except Exception as exc:
+                    status = "OFFLINE"
+                    detail = str(exc)
+
+            row = table.add_row()
+            row.add_column("TYPE", "REMOTE")
+            row.add_column("NAME", name)
+            row.add_column("ID", net_id)
+            row.add_column("ADDRESS", f"{host}:{port}")
+            row.add_column("STATUS", status)
+            row.add_column("DETAIL", detail)
+
+    # 4. Networking Networks (Raw ZT status)
+    try:
+        nets = zt.list_networks()
+        if not nets:
+            row = table.add_row()
+            row.add_column("TYPE", "INFO")
+            row.add_column("NAME", "ZeroTier CLI")
+            row.add_column("STATUS", "No networks found or CLI error")
+            row.add_column("DETAIL", f"CLI Path: {zt._get_cli_path() or 'Not found'}")
+        
+        for n in nets:
+            row = table.add_row()
+            row.add_column("TYPE", "NETWORK")
+            row.add_column("NAME", n.name)
+            row.add_column("ID", n.id)
+            row.add_column("ADDRESS", ", ".join(n.assigned_addresses))
+            row.add_column("STATUS", n.status)
+            row.add_column("DETAIL", "")
+    except Exception as exc:
+        row = table.add_row()
+        row.add_column("TYPE", "ERROR")
+        row.add_column("NAME", "ZeroTier CLI")
+        row.add_column("STATUS", "EXCEPTION")
+        row.add_column("DETAIL", str(exc))
+
+    return table
+
+CMDLET = Cmdlet(
+    name=".zerotier",
+    summary="Check ZeroTier hosting and connection status",
+    usage=".zerotier",
+    exec=exec
+)