"use client";

import { Suspense } from "react";
import { useEffect, useState } from "react";
import { useSearchParams } from "next/navigation";
import { BrainCircuit, KeyRound, Sparkles } from "lucide-react";
import { AIChat } from "@/components/ai/chat";
import { Button } from "@/components/ui/button";
import { Textarea } from "@/components/ui/textarea";
import { Skeleton } from "@/components/ui/skeleton";
import { useWorkspaceStore } from "@/stores/workspace-store";
import { cn } from "@/lib/utils";

function AIPageContent() {
  const searchParams = useSearchParams();
  const activeTeamId = useWorkspaceStore((state) => state.activeTeamId);
  const [settings, setSettings] = useState<{ model: string } | null>(null);
  const [draftPrompt, setDraftPrompt] = useState("");
  const [memory, setMemory] = useState("");
  const [memoryLoading, setMemoryLoading] = useState(true);
  const [memorySaving, setMemorySaving] = useState(false);
  const [memoryStatus, setMemoryStatus] = useState<string | null>(null);
  const [memoryOpen, setMemoryOpen] = useState(false);

  const initialPrompt = searchParams.get("q") || searchParams.get("prompt") || "";

  useEffect(() => {
    if (initialPrompt) setDraftPrompt(initialPrompt);
  }, [initialPrompt]);

  useEffect(() => {
    fetch("/api/ai/settings")
      .then((r) => r.ok ? r.json() : null)
      .then((d) => d && setSettings(d));
  }, []);

  useEffect(() => {
    const params = new URLSearchParams();
    if (activeTeamId) params.set("teamId", activeTeamId);
    fetch(`/api/ai/memory${params.toString() ? `?${params}` : ""}`)
      .then((r) => r.ok ? r.json() : null)
      .then((d) => {
        if (d) setMemory(d.memory || "");
        setMemoryLoading(false);
      });
  }, [activeTeamId]);

  async function handleSaveMemory() {
    setMemorySaving(true);
    setMemoryStatus(null);
    try {
      const res = await fetch("/api/ai/memory", {
        method: "POST",
        headers: { "Content-Type": "application/json" },
        body: JSON.stringify({ teamId: activeTeamId, memory, source: "manual_ui" }),
      });
      if (!res.ok) throw new Error();
      setMemoryStatus(memory.trim() ? "Gespeichert." : "Gelöscht.");
    } catch {
      setMemoryStatus("Fehler.");
    } finally {
      setMemorySaving(false);
    }
  }

  return (
    <div className="flex flex-col gap-3" style={{ height: "calc(100dvh - 72px)" }}>

      {/* ── Minimal header ─────────────────────────────────────────── */}
      <div className="flex items-center justify-between shrink-0">
        <div className="flex items-center gap-2.5">
          <div className="flex h-7 w-7 items-center justify-center rounded-lg bg-indigo-500/10 ring-1 ring-indigo-500/15">
            <Sparkles className="h-3.5 w-3.5 text-indigo-400" />
          </div>
          <span className="text-[15px] font-semibold tracking-tight">AI</span>
          {settings?.model && (
            <span className="rounded-full border border-border/40 px-2 py-0.5 text-[10px] text-muted-foreground">
              {settings.model}
            </span>
          )}
        </div>

        {/* Memory toggle */}
        <button
          onClick={() => setMemoryOpen((v) => !v)}
          className={cn(
            "flex items-center gap-1.5 rounded-lg border px-2.5 py-1.5 text-[11px] font-medium transition-all",
            memoryOpen
              ? "border-indigo-500/25 bg-indigo-500/[0.08] text-indigo-300"
              : "border-border/40 text-muted-foreground hover:text-foreground"
          )}
        >
          <BrainCircuit className="h-3.5 w-3.5" />
          Memory
        </button>
      </div>

      {/* ── Memory panel (collapsible) ──────────────────────────────── */}
      {memoryOpen && (
        <div className="shrink-0 rounded-xl border border-border/40 bg-card px-4 py-3 space-y-2">
          <p className="text-[11px] text-muted-foreground">
            Was soll die AI über dich wissen? Kontext, Vorlieben, Ziele — wird bei jedem Chat mitgeschickt.
          </p>
          {memoryLoading ? (
            <Skeleton className="h-20 rounded-lg" />
          ) : (
            <>
              <Textarea
                value={memory}
                onChange={(e) => setMemory(e.target.value)}
                rows={3}
                placeholder="z.B.: Ich bevorzuge direkte Antworten auf Deutsch, freitags möchte ich eine Wochenübersicht."
                className="text-[12px] resize-none"
              />
              <div className="flex items-center justify-between">
                {memoryStatus && <p className="text-[11px] text-muted-foreground">{memoryStatus}</p>}
                <Button
                  size="sm"
                  onClick={() => void handleSaveMemory()}
                  disabled={memorySaving}
                  className="ml-auto h-7 rounded-lg text-[11px]"
                >
                  <KeyRound className="h-3 w-3" />
                  {memorySaving ? "Speichern…" : "Speichern"}
                </Button>
              </div>
            </>
          )}
        </div>
      )}

      {/* ── Chat — takes all remaining space ───────────────────────── */}
      <div className="flex-1 min-h-0">
        <AIChat
          initialPrompt={initialPrompt || undefined}
          draftPrompt={draftPrompt || undefined}
          height="100%"
        />
      </div>

    </div>
  );
}

export default function AIAssistantPage() {
  return (
    <Suspense>
      <AIPageContent />
    </Suspense>
  );
}