"""
Lightweight wallet label lookup endpoint for CF Worker enrichment.
Called by Workers to annotate tool responses with label data.
"""

import json
import logging
import os

from fastapi import APIRouter, HTTPException
from pydantic import BaseModel, Field

logger = logging.getLogger("labels.lookup")

router = APIRouter(prefix="/api/v1/labels", tags=["wallet-labels"])


class LabelLookupRequest(BaseModel):
    addresses: list[str] = Field(..., min_items=1, max_items=10)
    chain: str = Field(default="ethereum")


@router.post("/lookup")
async def lookup_labels(req: LabelLookupRequest):
    """Batch lookup wallet labels for enrichment.

    Called by CF Workers to annotate tool responses with wallet intelligence.
    Returns labels from Redis cache + ClickHouse, with 5min Redis caching.
    """
    try:
        addresses = [a.strip() for a in req.addresses if a.strip()]
        labels: dict[str, list] = {a: [] for a in addresses}

        # Try ClickHouse first
        try:
            from clickhouse_driver import Client

            ch = Client(
                host=os.getenv("CH_HOST", "rmi-clickhouse"),
                port=int(os.getenv("CH_PORT", "9000")),
                user=os.getenv("CH_USER", "default"),
                password=os.getenv("CH_PASSWORD", "") or None,
                settings={"max_execution_time": 3},
            )
            placeholders = ", ".join(["%s"] * len(addresses))
            rows = ch.execute(
                f"SELECT address, label_name, label_category, source, is_sanctioned "
                f"FROM wallet_memory.wallet_labels "
                f"WHERE address IN ({placeholders}) "
                f"ORDER BY loaded_at DESC LIMIT 50",
                addresses,
            )
            for row in rows:
                addr = row[0]
                labels[addr].append(
                    {
                        "name": row[1] or "",
                        "category": row[2] or "",
                        "source": row[3] or "",
                        "sanctioned": bool(row[4]),
                    }
                )
        except Exception:
            # Fallback to Redis
            try:
                import redis as _redis

                r = _redis.Redis(
                    host=os.getenv("REDIS_HOST", "rmi-redis"),
                    port=int(os.getenv("REDIS_PORT", "6379")),
                    password=os.getenv("REDIS_PASSWORD", ""),
                    decode_responses=True,
                    socket_connect_timeout=2,
                )
                for addr in addresses:
                    cached = r.get(f"x402:enrich:{addr}")
                    if cached:
                        data = json.loads(cached)
                        labels[addr] = data.get("labels", [])
            except Exception:
                pass

        # Count labeled addresses
        labeled = sum(1 for v in labels.values() if v)

        return {
            "labels": labels,
            "addresses_checked": len(addresses),
            "addresses_labeled": labeled,
            "total_labels": sum(len(v) for v in labels.values()),
        }

    except Exception as e:
        logger.error(f"Label lookup failed: {e}")
        raise HTTPException(status_code=500, detail=str(e))