Upload patch_websocket_hub.py

patch_websocket_hub.py

@@ -0,0 +1,155 @@
+#!/usr/bin/env python3
+"""
+patch_websocket_hub.py
+──────────────────────
+Injects TradeLogParser (from hub_dashboard_service.py) and four API routes
+into /app/websocket_hub.py so that port 7860 serves:
+
+  GET /api/trades          → full open + closed state + stats
+  GET /api/trades/open     → open trades only
+  GET /api/trades/closed   → recent closed trades + stats  (?limit=N, default 50)
+  GET /api/health          → service health including trade counts
+
+Usage:
+    python3 patch_websocket_hub.py [--target /app/websocket_hub.py] [--dry-run]
+"""
+
+import argparse
+import shutil
+import sys
+from datetime import datetime
+from pathlib import Path
+
+# ── Snippet 1: Parser instantiation block ─────────────────────────────────────
+# Inserted BEFORE  `_START_TIME = time.time()`
+PARSER_BLOCK = '''
+# ── Trade log parser — injected so /api/trades is served on port 7860 ────────
+import sys as _sys, os as _os
+_sys.path.insert(0, '/app')
+from hub_dashboard_service import TradeLogParser as _TradeLogParser
+_trade_parser = _TradeLogParser(log_dir=_os.environ.get("RANKER_LOG_DIR", "/app/ranker_logs"))
+_trade_parser.start_background()
+'''
+
+# ── Snippet 2: FastAPI route definitions ──────────────────────────────────────
+# Appended AFTER  `_START_TIME = time.time()`
+TRADE_ROUTES = '''
+
+# ── /api/trades routes — injected by patch_websocket_hub.py ──────────────────
+@app.get("/api/trades")
+async def api_trades():
+    """Full trade state: open trades, recent closed trades, summary stats."""
+    return JSONResponse(_trade_parser.get_state())
+
+
+@app.get("/api/trades/open")
+async def api_trades_open():
+    """Open trades only."""
+    state = _trade_parser.get_state()
+    return JSONResponse({"open": state["open"]})
+
+
+@app.get("/api/trades/closed")
+async def api_trades_closed(limit: int = 50):
+    """Recent closed trades (newest first) + cumulative stats."""
+    state = _trade_parser.get_state()
+    return JSONResponse({
+        "closed": state["closed"][:limit],
+        "stats":  state["stats"],
+    })
+
+
+@app.get("/api/health")
+async def api_health():
+    """Service health check — includes live trade counts and log-file inventory."""
+    import glob as _glob
+    return JSONResponse({
+        "service":      "websocket_hub",
+        "version":      "v2.0",
+        "status":       "running",
+        "log_files":    len(_glob.glob("/app/ranker_logs/*.log")),
+        "trade_open":   len(_trade_parser.get_state()["open"]),
+        "trade_closed": len(_trade_parser.get_state()["closed"]),
+    })
+'''
+
+ANCHOR = "_START_TIME = time.time()"
+
+
+def patch(source: str) -> str:
+    """Apply both substitutions and return the patched source."""
+    if ANCHOR not in source:
+        raise ValueError(
+            f"Anchor '{ANCHOR}' not found in source — "
+            "is this the right file?"
+        )
+
+    # Count occurrences so we can give a clear warning if there are multiples.
+    count = source.count(ANCHOR)
+    if count > 1:
+        print(
+            f"[WARNING] Anchor appears {count} times; "
+            "only the first occurrence will be modified.",
+            file=sys.stderr,
+        )
+
+    # ── Step 1: prepend the parser block before the anchor ────────────────────
+    src = source.replace(ANCHOR, PARSER_BLOCK + ANCHOR, 1)
+
+    # ── Step 2: append trade routes after the (now-unique) anchor ────────────
+    #   After step 1, ANCHOR still appears exactly once (at the end of the
+    #   inserted block), so a second replace(…, 1) is safe.
+    src = src.replace(ANCHOR, ANCHOR + TRADE_ROUTES, 1)
+
+    return src
+
+
+def main() -> None:
+    ap = argparse.ArgumentParser(description="Patch websocket_hub.py with trade routes.")
+    ap.add_argument(
+        "--target",
+        default="/app/websocket_hub.py",
+        help="Path to websocket_hub.py (default: /app/websocket_hub.py)",
+    )
+    ap.add_argument(
+        "--dry-run",
+        action="store_true",
+        help="Print the patched source to stdout instead of writing to disk.",
+    )
+    args = ap.parse_args()
+
+    target = Path(args.target)
+
+    if not target.exists():
+        print(f"[ERROR] Target file not found: {target}", file=sys.stderr)
+        sys.exit(1)
+
+    original = target.read_text(encoding="utf-8")
+
+    # Guard: don't double-patch
+    if "_trade_parser" in original:
+        print(
+            "[SKIP] Patch already applied (_trade_parser already present in file).",
+            file=sys.stderr,
+        )
+        sys.exit(0)
+
+    patched = patch(original)
+
+    if args.dry_run:
+        print(patched)
+        return
+
+    # ── Backup the original ───────────────────────────────────────────────────
+    ts = datetime.now().strftime("%Y%m%d_%H%M%S")
+    backup = target.with_suffix(f".bak_{ts}.py")
+    shutil.copy2(target, backup)
+    print(f"[INFO] Backup written → {backup}")
+
+    target.write_text(patched, encoding="utf-8")
+    print(f"[OK] Patch applied → {target}")
+    print(f"     Routes added:  /api/trades  /api/trades/open  /api/trades/closed  /api/health")
+
+
+if __name__ == "__main__":
+    main()