Epicure/apps/web/components/recipe/cooking-assistant-panel.tsx

"use client";

import { useState, useRef, useEffect } from "react";
import { useTranslations } from "next-intl";
import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input";
import { Sheet, SheetContent, SheetHeader, SheetTitle } from "@/components/ui/sheet";
import { MessageCircle, Send, Bot, User, Maximize2, Minimize2 } from "lucide-react";
import ReactMarkdown from "react-markdown";
import { cn } from "@/lib/utils";
import { ChatHistorySearch } from "./chat-history-search";
import { ConversationMenu, type Conversation } from "./conversation-menu";

type Message = {
  role: "user" | "assistant";
  content: string;
};

type HistoryEntry = { role: "user" | "assistant"; content: string };

export function CookingAssistantPanel() {
  const t = useTranslations("cookingChat");
  const tCommon = useTranslations("common");
  const [open, setOpen] = useState(false);
  const [fullscreen, setFullscreen] = useState(false);
  const [messages, setMessages] = useState<Message[]>([]);
  const [input, setInput] = useState("");
  const [loading, setLoading] = useState(false);
  const [conversationId, setConversationId] = useState<string | null>(null);
  const bottomRef = useRef<HTMLDivElement>(null);
  const historyLoadedRef = useRef(false);

  async function loadMessages(id: string) {
    try {
      const res = await fetch(`/api/v1/ai/chat-history?conversationId=${id}&limit=50`);
      const json = res.ok ? (await res.json()) as { data?: HistoryEntry[] } : null;
      setMessages(json?.data?.length ? json.data.map((m) => ({ role: m.role, content: m.content })) : []);
    } catch {
      setMessages([]);
    }
  }

  function handleSelectConversation(id: string) {
    setConversationId(id);
    void loadMessages(id);
  }

  function handleConversationCreated(conversation: Conversation) {
    setConversationId(conversation.id);
    setMessages([]);
  }

  function handleActiveConversationDeleted() {
    // Fall back to a fresh conversation rather than leaving the panel
    // pointed at an id that no longer exists.
    fetch("/api/v1/ai/conversations", { method: "POST" })
      .then((res) => (res.ok ? res.json() : null))
      .then((created: Conversation | null) => {
        if (created) {
          setConversationId(created.id);
          setMessages([]);
        }
      })
      .catch(() => {});
  }

  useEffect(() => {
    if (!open || historyLoadedRef.current) return;
    historyLoadedRef.current = true;
    (async () => {
      const listRes = await fetch("/api/v1/ai/conversations");
      const list = listRes.ok ? ((await listRes.json()) as { data: Conversation[] }).data : [];
      const id = list.length > 0
        ? list[0]!.id
        : await fetch("/api/v1/ai/conversations", { method: "POST" })
            .then((res) => (res.ok ? res.json() : null))
            .then((created: Conversation | null) => created?.id ?? null);
      if (!id) return;
      setConversationId(id);
      await loadMessages(id);
    })().catch(() => {});
  }, [open]);

  useEffect(() => {
    if (open && bottomRef.current) {
      bottomRef.current.scrollIntoView({ behavior: "smooth" });
    }
  }, [messages, open]);

  const handleSubmit = async (e: React.FormEvent) => {
    e.preventDefault();
    const question = input.trim();
    if (!question || loading) return;

    setInput("");
    setMessages((prev) => [...prev, { role: "user", content: question }]);
    setLoading(true);

    try {
      const res = await fetch("/api/v1/ai/cooking-chat", {
        method: "POST",
        headers: { "Content-Type": "application/json" },
        body: JSON.stringify({ question, conversationId: conversationId ?? undefined }),
      });
      const data = await res.json() as { answer?: string; error?: string };
      setMessages((prev) => [
        ...prev,
        { role: "assistant", content: data.answer ?? t("sorry") },
      ]);
    } catch {
      setMessages((prev) => [
        ...prev,
        { role: "assistant", content: t("error") },
      ]);
    } finally {
      setLoading(false);
    }
  };

  const suggestions = [
    t("suggestion1"),
    t("suggestion2"),
    t("suggestion3"),
    t("suggestion4"),
  ];

  return (
    <>
      <Button
        variant="default"
        size="icon"
        // bottom-16 (not bottom-6, like the per-recipe chat button) clears the
        // recipe list's floating bulk-action bar, which occupies the bottom-4
        // to bottom-8 band while select mode is active.
        className="fixed bottom-16 right-6 h-12 w-12 rounded-full shadow-lg z-40"
        onClick={() => setOpen(true)}
        aria-label={t("ariaLabel")}
      >
        <MessageCircle className="h-5 w-5" />
      </Button>

      <Sheet open={open} onOpenChange={(v) => { setOpen(v); if (!v) setFullscreen(false); }}>
        <SheetContent
          side="right"
          className={cn(
            "flex flex-col p-0",
            fullscreen ? "!w-screen !max-w-none !h-screen !inset-0" : "w-full sm:w-[420px]"
          )}
        >
          <SheetHeader className="px-4 py-3 pr-10 border-b shrink-0">
            <div className="flex items-center justify-between gap-2">
              <SheetTitle className="text-base flex items-center gap-2">
                <Bot className="h-4 w-4 text-primary" />
                {t("title")}
              </SheetTitle>
              <div className="flex items-center gap-0.5">
                <button
                  type="button"
                  onClick={() => setFullscreen((v) => !v)}
                  aria-label={fullscreen ? tCommon("collapse") : tCommon("expand")}
                  className="p-1.5 rounded-md text-muted-foreground hover:text-foreground hover:bg-accent transition-colors"
                >
                  {fullscreen ? <Minimize2 className="h-4 w-4" /> : <Maximize2 className="h-4 w-4" />}
                </button>
                <ChatHistorySearch />
                <ConversationMenu
                  activeId={conversationId}
                  onSelect={handleSelectConversation}
                  onCreated={handleConversationCreated}
                  onDeletedActive={handleActiveConversationDeleted}
                />
              </div>
            </div>
            <p className="text-xs text-muted-foreground text-left">{t("subtitle")}</p>
          </SheetHeader>

          <div className="flex-1 overflow-y-auto px-4 py-4 space-y-4 min-h-0">
            {messages.length === 0 && (
              <div className="space-y-3">
                <p className="text-sm text-muted-foreground text-center py-4">
                  {t("intro")}
                </p>
                <div className="space-y-2">
                  {suggestions.map((s) => (
                    <button
                      key={s}
                      onClick={() => setInput(s)}
                      className="w-full text-left text-sm px-3 py-2 rounded-lg border bg-muted/50 hover:bg-muted transition-colors"
                    >
                      {s}
                    </button>
                  ))}
                </div>
              </div>
            )}

            {messages.map((msg, i) => (
              <div
                key={i}
                className={cn(
                  "flex gap-2 items-start",
                  msg.role === "user" && "flex-row-reverse"
                )}
              >
                <div className={cn(
                  "h-7 w-7 shrink-0 rounded-full flex items-center justify-center",
                  msg.role === "user"
                    ? "bg-primary text-primary-foreground"
                    : "bg-muted text-muted-foreground"
                )}>
                  {msg.role === "user" ? <User className="h-3.5 w-3.5" /> : <Bot className="h-3.5 w-3.5" />}
                </div>
                <div className={cn(
                  "rounded-xl px-3 py-2 text-sm max-w-[80%] leading-relaxed",
                  msg.role === "user"
                    ? "bg-primary text-primary-foreground whitespace-pre-wrap"
                    : "bg-muted space-y-2 [&_ul]:list-disc [&_ol]:list-decimal [&_ul]:pl-4 [&_ol]:pl-4 [&_strong]:font-semibold [&_li]:my-0.5"
                )}>
                  {msg.role === "assistant" ? (
                    <ReactMarkdown>{msg.content}</ReactMarkdown>
                  ) : (
                    msg.content
                  )}
                </div>
              </div>
            ))}

            {loading && (
              <div className="flex gap-2 items-start">
                <div className="h-7 w-7 shrink-0 rounded-full flex items-center justify-center bg-muted text-muted-foreground">
                  <Bot className="h-3.5 w-3.5" />
                </div>
                <div className="rounded-xl px-3 py-2 bg-muted">
                  <span className="flex gap-1">
                    <span className="w-1.5 h-1.5 rounded-full bg-muted-foreground animate-bounce [animation-delay:0ms]" />
                    <span className="w-1.5 h-1.5 rounded-full bg-muted-foreground animate-bounce [animation-delay:150ms]" />
                    <span className="w-1.5 h-1.5 rounded-full bg-muted-foreground animate-bounce [animation-delay:300ms]" />
                  </span>
                </div>
              </div>
            )}

            <div ref={bottomRef} />
          </div>

          <form onSubmit={handleSubmit} className="px-4 py-3 border-t shrink-0 flex gap-2">
            <Input
              value={input}
              onChange={(e) => setInput(e.target.value)}
              placeholder={t("inputPlaceholder")}
              disabled={loading}
              className="flex-1"
              autoComplete="off"
            />
            <Button type="submit" size="icon" disabled={loading || !input.trim()} aria-label={t("send")}>
              <Send className="h-4 w-4" />
            </Button>
          </form>
        </SheetContent>
      </Sheet>
    </>
  );
}