@schandlergarcia/sf-web-components 1.8.0 → 1.9.1

package/src/components/library/chat/useChatState.jsx

@@ -1,130 +0,0 @@
-import { useState, useCallback, useRef } from "react";
-
-let _nextId = 1;
-function uid() {
-  return `msg-${Date.now()}-${_nextId++}`;
-}
-
-/**
- * Core state management hook for AI chat.
- *
- * @param {Object} options
- * @param {Array}  options.initialMessages — seed the conversation
- * @param {Function} options.onSend — async (userMessage, allMessages) => assistantMessage | void
- *   Return a message object to add it, or use appendChunk / addMessage for streaming.
- *
- * @returns {Object}
- *
- * @example
- * const chat = useChatState({
- *   onSend: async (msg, history) => {
- *     const res = await fetch("/api/chat", { method: "POST", body: JSON.stringify({ messages: history }) });
- *     const data = await res.json();
- *     return { role: "assistant", content: data.reply, components: data.components };
- *   },
- * });
- */
-export default function useChatState({ initialMessages = [], onSend } = {}) {
-  const [messages, setMessages] = useState(() =>
-    initialMessages.map((m) => ({ id: uid(), timestamp: new Date().toISOString(), ...m }))
-  );
-  const [isLoading, setIsLoading] = useState(false);
-  const [isStreaming, setIsStreaming] = useState(false);
-  const [error, setError] = useState(null);
-  const onSendRef = useRef(onSend);
-  onSendRef.current = onSend;
-
-  const addMessage = useCallback((msg) => {
-    const full = { id: uid(), timestamp: new Date().toISOString(), ...msg };
-    setMessages((prev) => [...prev, full]);
-    return full.id;
-  }, []);
-
-  const updateMessage = useCallback((id, updates) => {
-    setMessages((prev) =>
-      prev.map((m) => (m.id === id ? { ...m, ...updates } : m))
-    );
-  }, []);
-
-  const appendChunk = useCallback((id, chunk) => {
-    setMessages((prev) =>
-      prev.map((m) =>
-        m.id === id ? { ...m, content: (m.content ?? "") + chunk } : m
-      )
-    );
-  }, []);
-
-  const removeMessage = useCallback((id) => {
-    setMessages((prev) => prev.filter((m) => m.id !== id));
-  }, []);
-
-  const clearMessages = useCallback(() => {
-    setMessages([]);
-    setError(null);
-  }, []);
-
-  const sendMessage = useCallback(
-    async (content, extra = {}) => {
-      if (!content?.trim()) return;
-      setError(null);
-
-      const userMsg = { role: "user", content: content.trim(), ...extra };
-      const userId = uid();
-      const fullUser = { ...userMsg, id: userId, timestamp: new Date().toISOString() };
-
-      setMessages((prev) => [...prev, fullUser]);
-      setIsLoading(true);
-
-      try {
-        const allMsgs = [...messages, fullUser];
-        const result = await onSendRef.current?.(fullUser, allMsgs, {
-          addMessage,
-          updateMessage,
-          appendChunk,
-          setStreaming: setIsStreaming,
-        });
-
-        if (result && typeof result === "object" && result.role) {
-          addMessage(result);
-        }
-      } catch (err) {
-        setError(err?.message ?? "Failed to send message");
-        addMessage({
-          role: "system",
-          content: err?.message ?? "Something went wrong. Please try again.",
-          isError: true,
-        });
-      } finally {
-        setIsLoading(false);
-        setIsStreaming(false);
-      }
-    },
-    [messages, addMessage, updateMessage, appendChunk]
-  );
-
-  const retryLast = useCallback(() => {
-    const lastUser = [...messages].reverse().find((m) => m.role === "user");
-    if (!lastUser) return;
-
-    const idx = messages.lastIndexOf(lastUser);
-    setMessages(messages.slice(0, idx));
-    setError(null);
-
-    sendMessage(lastUser.content);
-  }, [messages, sendMessage]);
-
-  return {
-    messages,
-    isLoading,
-    isStreaming,
-    error,
-    sendMessage,
-    addMessage,
-    updateMessage,
-    appendChunk,
-    removeMessage,
-    clearMessages,
-    retryLast,
-    setError,
-  };
-}