"use client";
import { useMemo, useState } from "react";
import { ChevronDown, Star, Eye, Wrench, Braces, Brain, Zap, Gauge, Coins, Sparkles, Check, Search } from "lucide-react";
import { useModels, PROVIDER_LABEL, shortName } from "@/lib/client/use-models";
import type { AIModel, RoutingMode } from "@/lib/client/types";
import { api } from "@/lib/client/fetcher";
import { cn, Dialog, Badge } from "../ui";

const MODES: { id: RoutingMode; label: string; desc: string; icon: React.ComponentType<{ className?: string }> }[] = [
  { id: "auto", label: "Automatique", desc: "Classe la tâche et choisit le meilleur compromis", icon: Sparkles },
  { id: "fast", label: "Rapide", desc: "Latence la plus basse (≥ 89 % au benchmark)", icon: Zap },
  { id: "quality", label: "Qualité", desc: "Score maximal, palier le plus élevé", icon: Gauge },
  { id: "economy", label: "Économie", desc: "Prix le plus bas (score 100 %)", icon: Coins },
];

export function ModelPicker({ value, mode, onChange, compact }: { value: string | null; mode: RoutingMode; onChange: (v: { modelId: string | null; mode: RoutingMode }) => void; compact?: boolean }) {
  const { chat, byId, mutate } = useModels(); const [open, setOpen] = useState(false); const [q, setQ] = useState("");
  const current = byId(value);
  const label = mode !== "manual" ? MODES.find(m => m.id === mode)!.label : shortName(current, value);
  const groups = useMemo(() => {
    const ql = q.toLowerCase();
    const list = chat.filter(m => !ql || m.id.includes(ql) || m.displayName.toLowerCase().includes(ql));
    const fav = list.filter(m => m.favorite);
    const g: Record<string, AIModel[]> = {}; for (const m of list) (g[m.provider] ??= []).push(m);
    return { fav, g };
  }, [chat, q]);
  async function toggleFav(e: React.MouseEvent, m: AIModel) { e.stopPropagation(); await api("/api/models", { method: "PATCH", json: { id: m.id, favorite: !m.favorite } }); void mutate(); }
  return (
    <>
      <button type="button" data-testid="model-picker" onClick={() => setOpen(true)} className={cn("inline-flex max-w-[52vw] items-center gap-1.5 rounded-lg border border-border bg-surface px-2.5 text-[13px] font-medium text-fg hover:border-border-strong transition-colors", compact ? "h-8" : "h-9")}>
        {mode !== "manual" ? <Sparkles className="h-3.5 w-3.5 text-accent-2" /> : <span className={cn("h-2 w-2 rounded-full", current?.provider === "anthropic" ? "bg-accent-2" : "bg-accent")} />}
        <span className="truncate">{label}</span><ChevronDown className="h-3.5 w-3.5 text-muted" />
      </button>
      <Dialog open={open} onClose={() => setOpen(false)} width="max-w-2xl" title="Choisir un modèle">
        <div className="p-3">
          <div className="mb-3 grid grid-cols-2 gap-2 md:grid-cols-4">
            {MODES.map(m => <button key={m.id} onClick={() => { onChange({ modelId: value, mode: m.id }); setOpen(false); }} className={cn("flex flex-col items-start gap-1 rounded-xl border p-3 text-left transition-colors", mode === m.id ? "border-accent bg-accent/10" : "border-border hover:border-border-strong hover:bg-surface")}><span className="flex items-center gap-1.5 text-[13px] font-medium"><m.icon className="h-3.5 w-3.5 text-accent-2" />{m.label}</span><span className="text-[11px] leading-snug text-muted">{m.desc}</span></button>)}
          </div>
          <div className="relative mb-2"><Search className="pointer-events-none absolute left-2.5 top-2.5 h-4 w-4 text-muted" /><input value={q} onChange={e => setQ(e.target.value)} placeholder="Filtrer les modèles vérifiés…" className="input !pl-8" autoFocus /></div>
          <div className="max-h-[52dvh] overflow-y-auto pr-1">
            {groups.fav.length > 0 && !q && <Group title="Favoris">{groups.fav.map(m => <Row key={m.id} m={m} selected={mode === "manual" && m.id === value} onSelect={() => { onChange({ modelId: m.id, mode: "manual" }); setOpen(false); }} onFav={e => toggleFav(e, m)} />)}</Group>}
            {Object.entries(groups.g).map(([prov, ms]) => <Group key={prov} title={PROVIDER_LABEL[prov] ?? prov} count={ms.length}>{ms.map(m => <Row key={m.id} m={m} selected={mode === "manual" && m.id === value} onSelect={() => { onChange({ modelId: m.id, mode: "manual" }); setOpen(false); }} onFav={e => toggleFav(e, m)} />)}</Group>)}
            <Group title="MacLustr local"><div className="px-3 py-2 text-[12px] text-muted">Intégration future de l'inférence locale sur le cluster — aucun modèle vérifié pour l'instant.</div></Group>
          </div>
          <div className="mt-2 flex items-center justify-between px-1 text-[11px] text-muted"><span>Seuls les modèles vérifiés par requête réelle apparaissent ici.</span><a href="/models" className="text-accent hover:underline">Laboratoire de modèles →</a></div>
        </div>
      </Dialog>
    </>
  );
}

function Group({ title, count, children }: { title: string; count?: number; children: React.ReactNode }) {
  return <div className="mb-2"><div className="sticky top-0 z-10 bg-bg-elev px-2 py-1 text-[11px] font-semibold uppercase tracking-wider text-muted">{title}{count != null && <span className="ml-1 font-normal">· {count}</span>}</div>{children}</div>;
}

export function CapIcons({ m, className }: { m: AIModel; className?: string }) {
  const c = m.capabilities;
  return <span className={cn("inline-flex items-center gap-1 text-muted", className)}>
    {c.vision === true && <Eye className="h-3 w-3" aria-label="vision" />}{c.tools === true && <Wrench className="h-3 w-3" aria-label="outils" />}{c.structuredOutput === true && <Braces className="h-3 w-3" aria-label="JSON" />}{c.reasoning === true && <Brain className="h-3 w-3" aria-label="raisonnement" />}
  </span>;
}

function Row({ m, selected, onSelect, onFav }: { m: AIModel; selected: boolean; onSelect: () => void; onFav: (e: React.MouseEvent) => void }) {
  const p = m.pricing; const b = m.benchmark;
  return (
    <div role="button" tabIndex={0} onClick={onSelect} onKeyDown={e => { if (e.key === "Enter" || e.key === " ") { e.preventDefault(); onSelect(); } }} className={cn("flex w-full cursor-pointer items-center gap-3 rounded-lg px-3 py-2 text-left transition-colors", selected ? "bg-accent/10" : "hover:bg-surface")}>
      <span className={cn("h-2 w-2 shrink-0 rounded-full", m.provider === "anthropic" ? "bg-accent-2" : "bg-accent")} />
      <div className="min-w-0 flex-1">
        <div className="flex items-center gap-2"><span className="truncate text-[13px] font-medium">{m.displayName}</span>{m.kind === "chat-pro" && <Badge tone="amber">pro · lent</Badge>}{m.kind === "search" && <Badge tone="accent">recherche web</Badge>}<CapIcons m={m} /></div>
        <div className="mt-0.5 flex flex-wrap items-center gap-x-3 gap-y-0.5 text-[11px] text-muted">
          {b?.qualityScore != null && <span className="inline-flex items-center gap-1"><span className="inline-block h-1.5 w-12 overflow-hidden rounded-full bg-surface-2"><span className={cn("block h-full", b.qualityScore >= 100 ? "bg-ok" : b.qualityScore >= 89 ? "bg-accent" : "bg-warn")} style={{ width: `${b.qualityScore}%` }} /></span>{b.qualityScore} %</span>}
          {b?.p50LatencyMs != null && <span>{b.p50LatencyMs} ms</span>}
          {p?.input != null && p?.output != null ? <span>{p.input} $ / {p.output} $ par M</span> : <span className="italic">prix non vérifié</span>}
          {m.contextWindow && <span>{Math.round(m.contextWindow / 1000)}k ctx</span>}
        </div>
      </div>
      <button onClick={onFav} className={cn("rounded p-1 hover:bg-surface-2", m.favorite ? "text-accent-2" : "text-muted/60 hover:text-fg")} aria-label="Favori"><Star className={cn("h-3.5 w-3.5", m.favorite && "fill-current")} /></button>
      {selected && <Check className="h-4 w-4 text-accent" />}
    </div>
  );
}