xertica-ui 2.1.2 → 2.1.4

package/components/assistant/xertica-assistant/use-assistant.ts

@@ -0,0 +1,559 @@
+import { useState, useEffect, useRef, useCallback } from 'react';
+import { toast } from 'sonner';
+import { gerarResposta } from '../../shared/assistant-utils';
+import { useIsMobile } from '../../shared/use-mobile';
+import type {
+    Message,
+    Conversation,
+    Suggestion,
+    AssistantTab,
+    AssistantMode,
+    SearchResult,
+    MockResponse,
+} from './types';
+
+// ─────────────────────────────────────────────────────────────────────────────
+// Hook Props
+// ─────────────────────────────────────────────────────────────────────────────
+
+export interface UseAssistantProps {
+    /** Layout mode for the assistant panel */
+    mode?: AssistantMode;
+    /** Controlled expansion state */
+    isExpanded?: boolean;
+    /** Toggle callback for controlled expansion */
+    onToggle?: () => void;
+    /** Initially selected tab */
+    defaultTab?: AssistantTab;
+    /** Enables demo mode with mock AI responses */
+    demoMode?: boolean;
+    /** Custom mock responses for demo mode */
+    customResponses?: MockResponse[];
+    /** Pre-loaded messages to hydrate an existing conversation */
+    initialMessages?: Message[];
+    /** Previously saved conversations */
+    savedConversations?: Conversation[];
+    /** Suggested prompts shown in the empty state */
+    suggestions?: Suggestion[];
+    /** Callback fired when the user sends a message */
+    onSendMessage?: (message: string) => void;
+    /** Whether the assistant is currently processing a response */
+    isProcessing?: boolean;
+    /** Custom response generator — overrides demo mode defaults */
+    responseGenerator?: (message: string) => Promise<string | Partial<Message>> | string | Partial<Message>;
+    /** Extended suggestions shown when the user clicks "More suggestions" */
+    richSuggestions?: Suggestion[];
+    /** Callback fired when the user clicks a rich suggestion */
+    onRichAction?: (actionId: string, actionText: string) => void;
+    /** Callback fired when the user rates a message */
+    onEvaluation?: (messageId: string, type: 'like' | 'dislike', reason?: string) => void;
+    /** Negative feedback categories shown in the dislike dropdown */
+    feedbackOptions?: string[];
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// Hook Return Value
+// ─────────────────────────────────────────────────────────────────────────────
+
+export interface UseAssistantReturn {
+    // ── Layout state ──────────────────────────────────────────────────────────
+    isFullPage: boolean;
+    isExpanded: boolean;
+    isMobile: boolean;
+    abaSelecionada: AssistantTab;
+    setAbaSelecionada: (tab: AssistantTab) => void;
+
+    // ── Message state ─────────────────────────────────────────────────────────
+    mensagens: Message[];
+    setMensagens: React.Dispatch<React.SetStateAction<Message[]>>;
+    mensagem: string;
+    setMensagem: (value: string) => void;
+
+    // ── Conversation state ────────────────────────────────────────────────────
+    conversas: Conversation[];
+    conversaAtual: string | null;
+    conversasFiltradas: Conversation[];
+
+    // ── UI state ──────────────────────────────────────────────────────────────
+    copiedId: string | null;
+    generatingPodcastId: string | null;
+    executingCommand: string | null;
+    savedSearches: string[];
+    editingDocument: { content: string; title: string } | null;
+    setEditingDocument: (doc: { content: string; title: string } | null) => void;
+    showMoreSuggestions: boolean;
+    setShowMoreSuggestions: (show: boolean) => void;
+    evaluationState: {
+        isOpen: boolean;
+        messageId: string | null;
+        type: 'dislike' | null;
+        category?: string | null;
+        reason: string;
+    };
+    setEvaluationState: React.Dispatch<React.SetStateAction<UseAssistantReturn['evaluationState']>>;
+
+    // ── Suggestions ───────────────────────────────────────────────────────────
+    sugestoes: Suggestion[];
+
+    // ── Refs ──────────────────────────────────────────────────────────────────
+    messagesEndRef: React.RefObject<HTMLDivElement>;
+    fileInputRef: React.RefObject<HTMLInputElement>;
+    audioInputRef: React.RefObject<HTMLInputElement>;
+
+    // ── Handlers ──────────────────────────────────────────────────────────────
+    handleToggle: () => void;
+    handleExpandWithTab: (tab: AssistantTab) => void;
+    handleEnviarMensagem: (arg?: string | any) => Promise<void>;
+    handleToggleFavorite: (messageId: string) => void;
+    handleCopyMessage: (content: string, messageId: string) => Promise<void>;
+    handleGeneratePodcast: (messageId: string, content: string) => Promise<void>;
+    handleDownloadDocument: (content: string, fileName: string) => void;
+    handleDownloadPodcast: (audioUrl: string, fileName: string) => void;
+    handleEditDocument: (content: string, title: string) => void;
+    handleNovaConversa: () => void;
+    handleSelecionarConversa: (conversaId: string) => void;
+    handleToggleFavoritaConversa: (conversaId: string) => void;
+    handleOpenSearchResult: (result: SearchResult) => void;
+    handleExecuteSearchCommand: (commandId: string, searchTerm: string, results: SearchResult[], messageId: string) => Promise<void>;
+    handleRichSuggestionClick: (suggestion: Suggestion) => void;
+    handleEvaluationClick: (messageId: string, type: 'like' | 'dislike') => void;
+    openFeedbackDialog: (messageId: string, category: string | null) => void;
+    handleSubmitDislike: () => void;
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// Default suggestions
+// ─────────────────────────────────────────────────────────────────────────────
+
+const DEFAULT_SUGGESTIONS: Suggestion[] = [
+    { id: '1', text: 'Me ajude a criar um documento profissional' },
+    { id: '2', text: 'Buscar nos meus arquivos por "relatório"' },
+    { id: '3', text: 'Resuma as conversas importantes desta semana' },
+    { id: '4', text: 'Crie um podcast sobre o último projeto' },
+];
+
+// ─────────────────────────────────────────────────────────────────────────────
+// Hook
+// ─────────────────────────────────────────────────────────────────────────────
+
+/**
+ * `useAssistant` — Headless hook for the XerticaAssistant component.
+ *
+ * Encapsulates all state management and business logic for the AI assistant panel.
+ * Use this hook when you need to build a fully custom assistant UI while reusing
+ * the same state logic as the default `XerticaAssistant` component.
+ *
+ * @example
+ * ```tsx
+ * import { useAssistant } from 'xertica-ui/assistant';
+ *
+ * function MyCustomAssistant() {
+ *   const {
+ *     mensagens,
+ *     mensagem,
+ *     setMensagem,
+ *     handleEnviarMensagem,
+ *     isExpanded,
+ *     handleToggle,
+ *   } = useAssistant({ demoMode: true });
+ *
+ *   return (
+ *     <div>
+ *       {mensagens.map(msg => <p key={msg.id}>{msg.content}</p>)}
+ *       <input value={mensagem} onChange={e => setMensagem(e.target.value)} />
+ *       <button onClick={() => handleEnviarMensagem()}>Send</button>
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+export function useAssistant({
+    mode = 'expanded',
+    isExpanded: controlledIsExpanded,
+    onToggle,
+    defaultTab = 'chat',
+    demoMode = true,
+    customResponses = [],
+    initialMessages = [],
+    savedConversations = [],
+    suggestions: propSuggestions,
+    onSendMessage,
+    isProcessing = false,
+    responseGenerator,
+    richSuggestions = [],
+    onRichAction,
+    onEvaluation,
+}: UseAssistantProps = {}): UseAssistantReturn {
+
+    // ── Layout state ────────────────────────────────────────────────────────────
+    const isFullPage = mode === 'fullPage';
+    const [internalIsExpanded, setInternalIsExpanded] = useState(controlledIsExpanded ?? true);
+    const isExpanded = controlledIsExpanded ?? internalIsExpanded;
+    const isMobile = useIsMobile();
+
+    // ── Tab state ───────────────────────────────────────────────────────────────
+    const [abaSelecionada, setAbaSelecionada] = useState<AssistantTab>(defaultTab);
+
+    // ── Message state ───────────────────────────────────────────────────────────
+    const [mensagens, setMensagens] = useState<Message[]>(initialMessages);
+    const [mensagem, setMensagem] = useState('');
+
+    // ── Conversation state ──────────────────────────────────────────────────────
+    const [conversas, setConversas] = useState<Conversation[]>(savedConversations);
+    const [conversaAtual, setConversaAtual] = useState<string | null>(null);
+
+    // ── UI state ────────────────────────────────────────────────────────────────
+    const [copiedId, setCopiedId] = useState<string | null>(null);
+    const [generatingPodcastId, setGeneratingPodcastId] = useState<string | null>(null);
+    const [executingCommand, setExecutingCommand] = useState<string | null>(null);
+    const [savedSearches, setSavedSearches] = useState<string[]>([]);
+    const [editingDocument, setEditingDocument] = useState<{ content: string; title: string } | null>(null);
+    const [showMoreSuggestions, setShowMoreSuggestions] = useState(false);
+    const [evaluationState, setEvaluationState] = useState<UseAssistantReturn['evaluationState']>({
+        isOpen: false,
+        messageId: null,
+        type: null,
+        category: null,
+        reason: '',
+    });
+
+    // ── Refs ────────────────────────────────────────────────────────────────────
+    const messagesEndRef = useRef<HTMLDivElement>(null);
+    const fileInputRef = useRef<HTMLInputElement>(null);
+    const audioInputRef = useRef<HTMLInputElement>(null);
+
+    // ── Suggestions ─────────────────────────────────────────────────────────────
+    const sugestoes = propSuggestions ?? DEFAULT_SUGGESTIONS;
+
+    // ── Effects ─────────────────────────────────────────────────────────────────
+
+    // Auto-scroll on new messages
+    useEffect(() => {
+        if (messagesEndRef.current && abaSelecionada === 'chat') {
+            messagesEndRef.current.scrollIntoView({ behavior: 'smooth' });
+        }
+    }, [mensagens, abaSelecionada]);
+
+    // Sync initial messages when they change
+    useEffect(() => {
+        if (initialMessages && initialMessages.length > 0) {
+            setMensagens(initialMessages);
+        }
+    }, [initialMessages]);
+
+    // ── Computed ────────────────────────────────────────────────────────────────
+
+    const conversasFiltradas = conversas.filter(conversa => {
+        if (abaSelecionada === 'favoritos') return conversa.isFavorite;
+        return true;
+    });
+
+    // ── Handlers ────────────────────────────────────────────────────────────────
+
+    const handleToggle = useCallback(() => {
+        if (onToggle) {
+            onToggle();
+        } else {
+            setInternalIsExpanded(prev => !prev);
+        }
+    }, [onToggle]);
+
+    const handleExpandWithTab = useCallback((tab: AssistantTab) => {
+        setAbaSelecionada(tab);
+        if (onToggle) {
+            onToggle();
+        } else {
+            setInternalIsExpanded(true);
+        }
+    }, [onToggle]);
+
+    const handleEnviarMensagem = useCallback(async (arg?: string | any) => {
+        let msgToSend = mensagem;
+
+        if (typeof arg === 'string' && !['document', 'podcast', 'search'].includes(arg)) {
+            msgToSend = arg;
+        }
+
+        if (!msgToSend.trim() || isProcessing) return;
+
+        const novaMensagem: Message = {
+            id: `msg-${Date.now()}`,
+            type: 'user',
+            content: msgToSend,
+            timestamp: new Date(),
+            isFavorite: false,
+        };
+
+        setMensagens(prev => [...prev, novaMensagem]);
+
+        if (onSendMessage) {
+            onSendMessage(msgToSend);
+        }
+
+        if (demoMode || responseGenerator) {
+            const mensagemAtual = msgToSend;
+            setTimeout(async () => {
+                let resposta: string | Partial<Message>;
+
+                if (responseGenerator) {
+                    resposta = await responseGenerator(mensagemAtual);
+                } else {
+                    resposta = gerarResposta(mensagemAtual, customResponses);
+                }
+
+                let novaMensagemIA: Message = {
+                    id: `msg-${Date.now()}-ia`,
+                    type: 'assistant',
+                    content: '',
+                    timestamp: new Date(),
+                    isFavorite: false,
+                };
+
+                if (typeof resposta === 'string') {
+                    novaMensagemIA.content = resposta;
+                } else {
+                    novaMensagemIA = { ...novaMensagemIA, ...resposta };
+                }
+
+                setMensagens(prev => [...prev, novaMensagemIA]);
+            }, 1000 + Math.random() * 1000);
+        }
+
+        setMensagem('');
+    }, [mensagem, isProcessing, onSendMessage, demoMode, responseGenerator, customResponses]);
+
+    const handleToggleFavorite = useCallback((messageId: string) => {
+        setMensagens(prev =>
+            prev.map(msg =>
+                msg.id === messageId ? { ...msg, isFavorite: !msg.isFavorite } : msg
+            )
+        );
+    }, []);
+
+    const handleCopyMessage = useCallback(async (content: string, messageId: string) => {
+        try {
+            if (navigator.clipboard && navigator.clipboard.writeText) {
+                try {
+                    await navigator.clipboard.writeText(content);
+                    setCopiedId(messageId);
+                    setTimeout(() => setCopiedId(null), 2000);
+                } catch (clipboardError: any) {
+                    if (clipboardError.name === 'NotAllowedError' || clipboardError.message.includes('permissions policy')) {
+                        throw new Error('Clipboard permission denied, falling back');
+                    }
+                    throw clipboardError;
+                }
+            } else {
+                throw new Error('Clipboard API not available');
+            }
+        } catch {
+            try {
+                const textArea = document.createElement('textarea');
+                textArea.value = content;
+                textArea.style.position = 'fixed';
+                textArea.style.left = '-999999px';
+                textArea.style.top = '-999999px';
+                document.body.appendChild(textArea);
+                textArea.focus();
+                textArea.select();
+                const successful = document.execCommand('copy');
+                document.body.removeChild(textArea);
+                if (successful) {
+                    setCopiedId(messageId);
+                    setTimeout(() => setCopiedId(null), 2000);
+                }
+            } catch (fallbackErr) {
+                console.error('Falha ao copiar:', fallbackErr);
+            }
+        }
+    }, []);
+
+    const handleGeneratePodcast = useCallback(async (messageId: string, content: string) => {
+        setGeneratingPodcastId(messageId);
+        setTimeout(() => {
+            setMensagens(prev =>
+                prev.map(msg =>
+                    msg.id === messageId
+                        ? {
+                            ...msg,
+                            attachmentType: 'podcast' as const,
+                            attachmentName: 'Podcast - ' + content.substring(0, 30) + '...',
+                            audioUrl: 'data:audio/mpeg;base64,//uQx...',
+                        }
+                        : msg
+                )
+            );
+            setGeneratingPodcastId(null);
+        }, 2000);
+    }, []);
+
+    const handleDownloadDocument = useCallback((content: string, fileName: string) => {
+        const blob = new Blob([content], { type: 'text/markdown' });
+        const url = URL.createObjectURL(blob);
+        const a = document.createElement('a');
+        a.href = url;
+        a.download = fileName.endsWith('.md') ? fileName : fileName + '.md';
+        document.body.appendChild(a);
+        a.click();
+        document.body.removeChild(a);
+        URL.revokeObjectURL(url);
+    }, []);
+
+    const handleDownloadPodcast = useCallback((audioUrl: string, fileName: string) => {
+        const a = document.createElement('a');
+        a.href = audioUrl;
+        a.download = fileName.endsWith('.mp3') ? fileName : fileName + '.mp3';
+        document.body.appendChild(a);
+        a.click();
+        document.body.removeChild(a);
+    }, []);
+
+    const handleEditDocument = useCallback((content: string, title: string) => {
+        setEditingDocument({ content, title });
+    }, []);
+
+    const handleNovaConversa = useCallback(() => {
+        setMensagens([]);
+        setConversaAtual(null);
+        setAbaSelecionada('chat');
+    }, []);
+
+    const handleSelecionarConversa = useCallback((conversaId: string) => {
+        const conversa = conversas.find(c => c.id === conversaId);
+        if (conversa) {
+            setMensagens(conversa.messages);
+            setConversaAtual(conversaId);
+            setAbaSelecionada('chat');
+        }
+    }, [conversas]);
+
+    const handleToggleFavoritaConversa = useCallback((conversaId: string) => {
+        setConversas(prev =>
+            prev.map(conv =>
+                conv.id === conversaId ? { ...conv, isFavorite: !conv.isFavorite } : conv
+            )
+        );
+    }, []);
+
+    const handleOpenSearchResult = useCallback((result: SearchResult) => {
+        console.log('Abrir resultado:', result);
+    }, []);
+
+    const handleExecuteSearchCommand = useCallback(async (
+        commandId: string,
+        searchTerm: string,
+        results: SearchResult[],
+        messageId: string
+    ) => {
+        setExecutingCommand(commandId);
+        setTimeout(() => {
+            if (commandId === '5') {
+                setSavedSearches(prev => [...prev, messageId]);
+            }
+            setExecutingCommand(null);
+        }, 1500);
+    }, []);
+
+    const handleRichSuggestionClick = useCallback((suggestion: Suggestion) => {
+        if (onRichAction) {
+            onRichAction(suggestion.id, suggestion.text);
+        } else {
+            handleEnviarMensagem(suggestion.text);
+        }
+        setShowMoreSuggestions(false);
+    }, [onRichAction, handleEnviarMensagem]);
+
+    const handleEvaluationClick = useCallback((messageId: string, type: 'like' | 'dislike') => {
+        if (type === 'like') {
+            if (onEvaluation) onEvaluation(messageId, 'like');
+            setMensagens(prev => prev.map(m => m.id === messageId ? { ...m, evaluation: 'like' as const } : m));
+            toast.success("Obrigado por avaliar positivamente!");
+        }
+    }, [onEvaluation]);
+
+    const openFeedbackDialog = useCallback((messageId: string, category: string | null) => {
+        setEvaluationState({
+            isOpen: true,
+            messageId,
+            type: 'dislike',
+            category,
+            reason: '',
+        });
+    }, []);
+
+    const handleSubmitDislike = useCallback(() => {
+        if (evaluationState.messageId) {
+            const finalReason = evaluationState.category
+                ? (evaluationState.reason ? `${evaluationState.category}: ${evaluationState.reason}` : evaluationState.category)
+                : evaluationState.reason;
+
+            if (onEvaluation) {
+                onEvaluation(evaluationState.messageId, 'dislike', finalReason);
+            }
+
+            setMensagens(prev =>
+                prev.map(m =>
+                    m.id === evaluationState.messageId
+                        ? { ...m, evaluation: 'dislike' as const, evaluationReason: finalReason }
+                        : m
+                )
+            );
+            toast.success("Obrigado pelo seu feedback. Vamos melhorar.");
+        }
+        setEvaluationState({ isOpen: false, messageId: null, type: null, category: null, reason: '' });
+    }, [evaluationState, onEvaluation]);
+
+    return {
+        // Layout
+        isFullPage,
+        isExpanded,
+        isMobile,
+        abaSelecionada,
+        setAbaSelecionada,
+        // Messages
+        mensagens,
+        setMensagens,
+        mensagem,
+        setMensagem,
+        // Conversations
+        conversas,
+        conversaAtual,
+        conversasFiltradas,
+        // UI
+        copiedId,
+        generatingPodcastId,
+        executingCommand,
+        savedSearches,
+        editingDocument,
+        setEditingDocument,
+        showMoreSuggestions,
+        setShowMoreSuggestions,
+        evaluationState,
+        setEvaluationState,
+        // Suggestions
+        sugestoes,
+        // Refs
+        messagesEndRef,
+        fileInputRef,
+        audioInputRef,
+        // Handlers
+        handleToggle,
+        handleExpandWithTab,
+        handleEnviarMensagem,
+        handleToggleFavorite,
+        handleCopyMessage,
+        handleGeneratePodcast,
+        handleDownloadDocument,
+        handleDownloadPodcast,
+        handleEditDocument,
+        handleNovaConversa,
+        handleSelecionarConversa,
+        handleToggleFavoritaConversa,
+        handleOpenSearchResult,
+        handleExecuteSearchCommand,
+        handleRichSuggestionClick,
+        handleEvaluationClick,
+        openFeedbackDialog,
+        handleSubmitDislike,
+    };
+}