feat: add p2p shard swarm

packages/node/meshnet_node/startup.py

@@ -11,7 +11,7 @@ import urllib.request
 from pathlib import Path
 from typing import Any
 
-from .downloader import download_shard
+from .downloader import compute_shard_checksum, download_shard
 from .hardware import detect_hardware
 from .server import StubNodeServer
 from .wallet import load_or_create_wallet
@@ -47,7 +47,7 @@ def run_startup(
       1. Detect GPU / hardware profile
       2. Load or generate Solana wallet keypair
       3. Query tracker for optimal shard assignment
-      4. Download (or stub) the assigned shard
+      4. Download (or stub) the assigned shard from peers, then HuggingFace
       5. Start local HTTP server
       6. Register with tracker
 
@@ -96,6 +96,7 @@ def run_startup(
     shard_end: int = assignment["shard_end"]
     assigned_model: str = assignment.get("model", model)
     hf_repo: str | None = assignment.get("hf_repo")
+    peers: list[dict] = assignment.get("peers", [])
     print(f"  Shard: layers {shard_start}-{shard_end} of {assigned_model}", flush=True)
 
     # 4. Download shard
@@ -105,7 +106,10 @@ def run_startup(
         dl_kwargs["cache_dir"] = cache_dir
     if hf_repo is not None:
         dl_kwargs["hf_repo"] = hf_repo
+    if peers:
+        dl_kwargs["peers"] = peers
     shard_path = download_shard(assigned_model, shard_start, shard_end, **dl_kwargs)
+    shard_checksum = compute_shard_checksum(shard_path)
     print(f"  Cached at: {shard_path}", flush=True)
 
     # 5. Start HTTP server
@@ -116,6 +120,8 @@ def run_startup(
         shard_start=shard_start,
         shard_end=shard_end,
         is_last_shard=is_last,
+        model=assigned_model,
+        shard_path=shard_path,
     )
     actual_port = node.start()
     public_host = advertise_host or (socket.getfqdn() if host == "0.0.0.0" else host)
@@ -128,8 +134,10 @@ def run_startup(
             f"{tracker_url}/v1/nodes/register",
             {
                 "endpoint": endpoint,
+                "model": assigned_model,
                 "shard_start": shard_start,
                 "shard_end": shard_end,
+                "shard_checksum": shard_checksum,
                 "hardware_profile": hw,
                 "wallet_address": address,
                 "score": 1.0,