"use client";

import Link from "next/link";
import { useState } from "react";
import { PARTY_IDS, partyVar, partyLabel, partyName } from "@/lib/parties";
import { Card } from "./ui";

interface Ask { query: string; terms: string[]; parties: Record<string, { id: number; proposal: string; quote: string; page: number | null; topic: string; confidence: number; document: { id: number; title: string | null; url: string | null } }[]>; note: string }

export function AskBox() {
  const [q, setQ] = useState("");
  const [res, setRes] = useState<Ask | null>(null);
  const [busy, setBusy] = useState(false);
  const run = async (query: string) => {
    if (query.trim().length < 3) return;
    setBusy(true);
    try { setRes(await fetch(`/api/ask?q=${encodeURIComponent(query)}`).then((r) => r.json())); } finally { setBusy(false); }
  };
  return (
    <Card className="card-lg">
      <div className="eyebrow mb-2">Recherche documentaire</div>
      <form onSubmit={(e) => { e.preventDefault(); run(q); }} className="flex gap-2">
        <input value={q} onChange={(e) => setQ(e.target.value)} placeholder="Que proposent les partis pour le logement ?" className="flex-1 card px-3.5 py-2.5 text-[15px] outline-none focus:border-accent" aria-label="Question" />
        <button className="rounded-[8px] bg-brand text-white px-4 py-2.5 text-[14px] font-semibold disabled:opacity-40" disabled={busy}>{busy ? "…" : "Chercher"}</button>
      </form>
      <div className="flex flex-wrap gap-1.5 mt-2 text-[12px]">{["logement", "immigration", "santé", "troisième lien", "français", "impôts", "souveraineté", "Hydro-Québec"].map((s) => <button key={s} onClick={() => { setQ(s); run(s); }} className="rounded-full border border-border px-2.5 py-1 hover:bg-surface-2">{s}</button>)}</div>
      {res && (
        <div className="mt-5 grid md:grid-cols-2 xl:grid-cols-5 gap-3">
          {PARTY_IDS.map((id) => (
            <div key={id} className="surface-2 rounded-[8px] p-3" style={{ borderTop: `3px solid ${partyVar(id)}` }}>
              <div className="font-semibold text-[13px] mb-2">{partyName(id)}</div>
              {(res.parties[id] ?? []).length === 0 ? <div className="text-[12px] text-ink-3">Aucune proposition trouvée dans les documents ingérés.</div> : (res.parties[id] ?? []).map((p) => (
                <div key={p.id} className="mb-2.5 text-[12.5px] leading-snug"><div>{p.proposal}</div><div className="text-[11.5px] text-ink-3 mt-0.5"><Link href={`/documents/${p.document.id}#p${p.id}`} className="link">{p.document.title ?? "Document"}</Link>{p.page ? ` · p. ${p.page}` : ""}</div></div>
              ))}
            </div>
          ))}
          <p className="md:col-span-2 xl:col-span-5 text-[11.5px] text-ink-3">{res.note} {PARTY_IDS.map(partyLabel).join(", ")} — classement lexical sur {res.terms.join(", ")}.</p>
        </div>
      )}
    </Card>
  );
}