1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
| | import React, {
| | createContext,
| | useCallback,
| | useContext,
| | useEffect,
| | useRef,
| | useState,
| | } from "react";
| | import { Message, WebSocketMessage } from "../types";
| | import { useConnection } from "./ConnectionContext";
| | import { playAudio } from "../services/audio";
| |
| | function generateId(): string {
| | return Date.now().toString(36) + Math.random().toString(36).slice(2);
| | }
| |
| | interface ChatContextValue {
| | messages: Message[];
| | sendTextMessage: (text: string) => void;
| | sendVoiceMessage: (audioUri: string, durationMs?: number) => void;
| | clearMessages: () => void;
| | }
| |
| | const ChatContext = createContext<ChatContextValue | null>(null);
| |
| | export function ChatProvider({ children }: { children: React.ReactNode }) {
| | const [messages, setMessages] = useState<Message[]>([]);
| | const { sendTextMessage: wsSend, sendVoiceMessage: wsVoice, onMessageReceived } = useConnection();
| |
| | const addMessage = useCallback((msg: Message) => {
| | setMessages((prev) => [...prev, msg]);
| | }, []);
| |
| | const updateMessageStatus = useCallback(
| | (id: string, status: Message["status"]) => {
| | setMessages((prev) =>
| | prev.map((m) => (m.id === id ? { ...m, status } : m))
| | );
| | },
| | []
| | );
| |
| | // Handle incoming WebSocket messages
| | useEffect(() => {
| | onMessageReceived.current = (data: WebSocketMessage) => {
| | if (data.type === "text") {
| | const msg: Message = {
| | id: generateId(),
| | role: "assistant",
| | type: "text",
| | content: data.content,
| | timestamp: Date.now(),
| | status: "sent",
| | };
| | setMessages((prev) => [...prev, msg]);
| | } else if (data.type === "voice") {
| | const msg: Message = {
| | id: generateId(),
| | role: "assistant",
| | type: "voice",
| | content: data.content ?? "",
| | audioUri: data.audioBase64
| | ? `data:audio/mp4;base64,${data.audioBase64}`
| | : undefined,
| | timestamp: Date.now(),
| | status: "sent",
| | };
| | setMessages((prev) => [...prev, msg]);
| |
| | // Auto-play incoming voice messages
| | if (msg.audioUri) {
| | playAudio(msg.audioUri).catch(() => {});
| | }
| | }
| | };
| |
| | return () => {
| | onMessageReceived.current = null;
| | };
| | }, [onMessageReceived]);
| |
| | const sendTextMessage = useCallback(
| | (text: string) => {
| | const id = generateId();
| | const msg: Message = {
| | id,
| | role: "user",
| | type: "text",
| | content: text,
| | timestamp: Date.now(),
| | status: "sending",
| | };
| | addMessage(msg);
| |
| | const sent = wsSend(text);
| | updateMessageStatus(id, sent ? "sent" : "error");
| | },
| | [wsSend, addMessage, updateMessageStatus]
| | );
| |
| | const sendVoiceMessage = useCallback(
| | (audioUri: string, durationMs?: number) => {
| | const id = generateId();
| | const msg: Message = {
| | id,
| | role: "user",
| | type: "voice",
| | content: "",
| | audioUri,
| | timestamp: Date.now(),
| | status: "sending",
| | duration: durationMs,
| | };
| | addMessage(msg);
| |
| | // For now, send with empty base64 since we'd need expo-file-system to encode
| | const sent = wsVoice("", "Voice message");
| | updateMessageStatus(id, sent ? "sent" : "error");
| | },
| | [wsVoice, addMessage, updateMessageStatus]
| | );
| |
| | const clearMessages = useCallback(() => {
| | setMessages([]);
| | }, []);
| |
| | return (
| | <ChatContext.Provider
| | value={{ messages, sendTextMessage, sendVoiceMessage, clearMessages }}
| | >
| | {children}
| | </ChatContext.Provider>
| | );
| | }
| |
| | export function useChat() {
| | const ctx = useContext(ChatContext);
| | if (!ctx) throw new Error("useChat must be used within ChatProvider");
| | return ctx;
| | }
|
|
