40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
"""meshnet-node CLI entry point."""
|
|
|
|
import argparse
|
|
import sys
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(
|
|
prog="meshnet-node",
|
|
description="Distributed Inference Network node client",
|
|
)
|
|
subparsers = parser.add_subparsers(dest="command")
|
|
|
|
start_cmd = subparsers.add_parser("start", help="Start the node server")
|
|
start_cmd.add_argument("--tracker", default="http://localhost:8080", help="Tracker URL")
|
|
start_cmd.add_argument("--port", type=int, default=7000, help="Port to listen on")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.command == "start":
|
|
from meshnet_node.server import StubNodeServer
|
|
|
|
node = StubNodeServer(port=args.port)
|
|
port = node.start()
|
|
print(f"meshnet-node listening on port {port}")
|
|
print(f"Tracker URL: {args.tracker}")
|
|
try:
|
|
import time
|
|
while True:
|
|
time.sleep(1)
|
|
except KeyboardInterrupt:
|
|
node.stop()
|
|
sys.exit(0)
|
|
else:
|
|
parser.print_help()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|