merge: chore/cleanup-remove-bloat-and-secrets into main

app/routers/unified_scanner_router.py

@@ -0,0 +1,97 @@
+"""Unified Scanner Router v3 — matches WalletScanner v3 fields."""
+
+import logging
+
+from fastapi import APIRouter, HTTPException, Request
+from pydantic import BaseModel
+
+from app.unified_token_scanner import get_token_scanner
+from app.unified_wallet_scanner import get_wallet_scanner
+
+logger = logging.getLogger(__name__)
+router = APIRouter(prefix="/api/v2", tags=["scanner-v2"])
+
+
+class TokenScanRequest(BaseModel):
+    address: str
+    chain: str = "solana"
+
+
+class WalletScanRequest(BaseModel):
+    address: str
+    chain: str = "solana"
+
+
+def _get_tier(request):
+    if request.headers.get("X-RMI-Key") == "rmi-internal-2026":
+        return "admin"
+    if request.headers.get("X-Payment-Verified") == "true":
+        return "pro"
+    return "free"
+
+
+def _get_admin_key(request):
+    return request.headers.get("X-Admin-Key") or request.headers.get("X-RMI-Key")
+
+
+@router.post("/token/scan")
+async def scan_token(request: Request, body: TokenScanRequest):
+    tier = _get_tier(request)
+    admin_key = _get_admin_key(request)
+    try:
+        scanner = get_token_scanner()
+        result = await scanner.scan(address=body.address, chain=body.chain, tier=tier, admin_key=admin_key)
+        return {
+            "status": "ok",
+            "token": result.token_address,
+            "chain": result.chain,
+            "name": result.name,
+            "symbol": result.symbol,
+            "price_usd": result.price_usd,
+            "safety_score": result.safety_score,
+            "confidence": result.confidence,
+            "risk_level": result.risk_level,
+            "risk_flags": result.risk_flags,
+            "green_flags": result.green_flags,
+            "modules_run": result.modules_run,
+            "scan_duration_ms": result.scan_duration_ms,
+            "tier": result.tier,
+            "scanned_at": result.scanned_at,
+        }
+    except Exception as e:
+        raise HTTPException(500, f"Scan failed: {str(e)[:200]}")
+
+
+@router.post("/wallet/scan")
+async def scan_wallet(request: Request, body: WalletScanRequest):
+    tier = _get_tier(request)
+    admin_key = _get_admin_key(request)
+    try:
+        scanner = get_wallet_scanner()
+        result = await scanner.scan(address=body.address, chain=body.chain, tier=tier, admin_key=admin_key)
+        return {
+            "status": "ok",
+            "address": result.address,
+            "chain": result.chain,
+            "labels": result.labels,
+            "entity_name": result.entity_name,
+            "entity_type": result.entity_type,
+            "persona": result.persona,
+            "persona_confidence": result.persona_confidence,
+            "total_value_usd": result.total_value_usd,
+            "token_count": result.token_count,
+            "scam_tokens_held": result.scam_tokens_held,
+            "scam_exposure_pct": result.scam_exposure_pct,
+            "risk_score": result.risk_score,
+            "risk_flags": result.risk_flags,
+            "rag_matches": len(result.rag_matches),
+            "behavior_patterns": result.behavior_patterns,
+            "smart_money_overlap": result.smart_money_overlap,
+            "risk_forecast": result.risk_forecast,
+            "modules_run": result.modules_run,
+            "scan_duration_ms": result.scan_duration_ms,
+            "tier": result.tier,
+            "scanned_at": result.scanned_at,
+        }
+    except Exception as e:
+        raise HTTPException(500, f"Scan failed: {str(e)[:200]}")