47 lines
1.3 KiB
TypeScript
47 lines
1.3 KiB
TypeScript
|
import { useState, useEffect, useCallback, useRef } from 'react';
|
||
|
|
import { chatSocket } from '@/lib/websocket';
|
||
|
|
import type { Message } from '@/types';
|
||
|
|
|
||
|
|
export function useChat(conversationId: string | null) {
|
||
|
|
const [messages, setMessages] = useState<Message[]>([]);
|
||
|
|
const [connected, setConnected] = useState(false);
|
||
|
|
const pendingRef = useRef(false);
|
||
|
|
|
||
|
|
useEffect(() => {
|
||
|
|
chatSocket.connect();
|
||
|
|
|
||
|
|
chatSocket.on('connected', () => setConnected(true));
|
||
|
|
chatSocket.on('history', (data) => {
|
||
|
|
setMessages((data.messages as Message[]) || []);
|
||
|
|
});
|
||
|
|
chatSocket.on('message', (data) => {
|
||
|
|
setMessages((prev) => [...prev, data as unknown as Message]);
|
||
|
|
});
|
||
|
|
|
||
|
|
return () => {
|
||
|
|
chatSocket.disconnect();
|
||
|
|
};
|
||
|
|
}, []);
|
||
|
|
|
||
|
|
useEffect(() => {
|
||
|
|
if (conversationId && connected && !pendingRef.current) {
|
||
|
|
pendingRef.current = true;
|
||
|
|
chatSocket.send('init', { conversationId });
|
||
|
|
}
|
||
|
|
if (!conversationId) {
|
||
|
|
pendingRef.current = false;
|
||
|
|
setMessages([]);
|
||
|
|
}
|
||
|
|
}, [conversationId, connected]);
|
||
|
|
|
||
|
|
const send = useCallback((content: string) => {
|
||
|
|
chatSocket.send('chat', { content });
|
||
|
|
}, []);
|
||
|
|
|
||
|
|
const createConversation = useCallback((title: string, accountId?: string) => {
|
||
|
|
chatSocket.send('create_conversation', { title, accountId });
|
||
|
|
}, []);
|
||
|
|
|
||
|
|
return { messages, connected, send, createConversation };
|
||
|
|
}
|