Merge branch 'master' of https://git.d-popov.com/popov/neuron-tai

packages/tracker/meshnet_tracker/cli.py

@@ -8,41 +8,49 @@ from .server import TrackerServer
 
 
 def main() -> None:
-    parser = argparse.ArgumentParser(
-        prog="meshnet-tracker",
-        description="Distributed Inference Network node registry and route selection",
-    )
-    subparsers = parser.add_subparsers(dest="command")
-
-    start_cmd = subparsers.add_parser("start", help="Start the tracker server")
-    start_cmd.add_argument("--host", default="127.0.0.1", help="Host interface to listen on")
-    start_cmd.add_argument("--port", type=int, default=8080, help="Port to listen on")
-    start_cmd.add_argument(
+    common = argparse.ArgumentParser(add_help=False)
+    common.add_argument("--host", default="0.0.0.0", help="Host interface to listen on")
+    common.add_argument("--port", type=int, default=8080, help="Port to listen on")
+    common.add_argument(
         "--heartbeat-timeout",
         type=float,
         default=30.0,
         help="Seconds before a node is removed from the registry after missed heartbeat",
     )
-    start_cmd.add_argument(
+    common.add_argument(
         "--cluster-peers",
         default="",
         help="Comma-separated URLs of peer tracker nodes (enables Raft cluster mode)",
     )
-    start_cmd.add_argument(
+    common.add_argument(
         "--self-url",
         default=None,
         help="This tracker's own URL as seen by peers (auto-derived from --host/--port if omitted)",
     )
-    start_cmd.add_argument(
+    common.add_argument(
         "--stats-db",
         default=None,
         metavar="PATH",
         help="SQLite database path for persistent model usage statistics",
     )
+    common.add_argument(
+        "--relay-url",
+        default=None,
+        help="Public ws(s):// relay URL advertised to nodes, for example wss://ai.neuron.d-popov.com/ws",
+    )
+
+    parser = argparse.ArgumentParser(
+        prog="meshnet-tracker",
+        description="Distributed Inference Network node registry and route selection",
+        parents=[common],
+    )
+    subparsers = parser.add_subparsers(dest="command")
+
+    subparsers.add_parser("start", help="Start the tracker server", parents=[common])
 
     args = parser.parse_args()
 
-    if args.command == "start":
+    if args.command in {None, "start"}:
         cluster_peers = [u.strip() for u in args.cluster_peers.split(",") if u.strip()]
         server = TrackerServer(
             host=args.host,
@@ -51,6 +59,7 @@ def main() -> None:
             cluster_peers=cluster_peers or None,
             cluster_self_url=args.self_url,
             stats_db=getattr(args, "stats_db", None),
+            relay_url=args.relay_url,
         )
         port = server.start()
         print(f"meshnet-tracker listening on http://{args.host}:{port}", flush=True)