@elqnt/chat 2.0.7 → 3.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (52) hide show
  1. package/README.md +386 -0
  2. package/dist/api/index.d.mts +308 -0
  3. package/dist/api/index.d.ts +308 -0
  4. package/dist/api/index.js +220 -0
  5. package/dist/api/index.js.map +1 -0
  6. package/dist/api/index.mjs +183 -0
  7. package/dist/api/index.mjs.map +1 -0
  8. package/dist/hooks/index.d.mts +78 -0
  9. package/dist/hooks/index.d.ts +78 -0
  10. package/dist/hooks/index.js +709 -0
  11. package/dist/hooks/index.js.map +1 -0
  12. package/dist/hooks/index.mjs +683 -0
  13. package/dist/hooks/index.mjs.map +1 -0
  14. package/dist/index.d.mts +4 -109
  15. package/dist/index.d.ts +4 -109
  16. package/dist/index.js +699 -3607
  17. package/dist/index.js.map +1 -1
  18. package/dist/index.mjs +690 -3584
  19. package/dist/index.mjs.map +1 -1
  20. package/dist/models/index.d.mts +76 -6
  21. package/dist/models/index.d.ts +76 -6
  22. package/dist/models/index.js +21 -0
  23. package/dist/models/index.js.map +1 -1
  24. package/dist/models/index.mjs +14 -0
  25. package/dist/models/index.mjs.map +1 -1
  26. package/dist/transport/index.d.mts +243 -0
  27. package/dist/transport/index.d.ts +243 -0
  28. package/dist/transport/index.js +875 -0
  29. package/dist/transport/index.js.map +1 -0
  30. package/dist/transport/index.mjs +843 -0
  31. package/dist/transport/index.mjs.map +1 -0
  32. package/dist/types-BB5nRdZs.d.mts +222 -0
  33. package/dist/types-CNvuxtcv.d.ts +222 -0
  34. package/package.json +32 -40
  35. package/dist/hooks/use-websocket-chat-admin.d.mts +0 -17
  36. package/dist/hooks/use-websocket-chat-admin.d.ts +0 -17
  37. package/dist/hooks/use-websocket-chat-admin.js +0 -1196
  38. package/dist/hooks/use-websocket-chat-admin.js.map +0 -1
  39. package/dist/hooks/use-websocket-chat-admin.mjs +0 -1172
  40. package/dist/hooks/use-websocket-chat-admin.mjs.map +0 -1
  41. package/dist/hooks/use-websocket-chat-base.d.mts +0 -81
  42. package/dist/hooks/use-websocket-chat-base.d.ts +0 -81
  43. package/dist/hooks/use-websocket-chat-base.js +0 -1025
  44. package/dist/hooks/use-websocket-chat-base.js.map +0 -1
  45. package/dist/hooks/use-websocket-chat-base.mjs +0 -1001
  46. package/dist/hooks/use-websocket-chat-base.mjs.map +0 -1
  47. package/dist/hooks/use-websocket-chat-customer.d.mts +0 -24
  48. package/dist/hooks/use-websocket-chat-customer.d.ts +0 -24
  49. package/dist/hooks/use-websocket-chat-customer.js +0 -1092
  50. package/dist/hooks/use-websocket-chat-customer.js.map +0 -1
  51. package/dist/hooks/use-websocket-chat-customer.mjs +0 -1068
  52. package/dist/hooks/use-websocket-chat-customer.mjs.map +0 -1
@@ -1,1068 +0,0 @@
1
- "use client";
2
- "use client";
3
-
4
- // hooks/use-websocket-chat-customer.ts
5
- import { useCallback as useCallback2, useState as useState2 } from "react";
6
-
7
- // hooks/use-websocket-chat-base.ts
8
- import { useCallback, useEffect, useRef, useState } from "react";
9
- var createDefaultLogger = (debug) => ({
10
- debug: debug ? console.log : () => {
11
- },
12
- info: console.info,
13
- warn: console.warn,
14
- error: console.error
15
- });
16
- var DEFAULT_RETRY_CONFIG = {
17
- maxRetries: 10,
18
- intervals: [1e3, 2e3, 5e3],
19
- backoffMultiplier: 1.5,
20
- maxBackoffTime: 3e4
21
- };
22
- var DEFAULT_QUEUE_CONFIG = {
23
- maxSize: 100,
24
- dropStrategy: "oldest"
25
- };
26
- var DEFAULT_HEARTBEAT_INTERVAL = 3e4;
27
- var DEFAULT_HEARTBEAT_TIMEOUT = 5e3;
28
- var DEFAULT_TRANSPORT = "websocket";
29
- function isChatEvent(data) {
30
- return data && typeof data === "object" && (typeof data.type === "string" || data.message);
31
- }
32
- var useWebSocketChatBase = ({
33
- serverBaseUrl,
34
- orgId,
35
- clientType,
36
- product,
37
- onMessage,
38
- retryConfig = DEFAULT_RETRY_CONFIG,
39
- queueConfig = DEFAULT_QUEUE_CONFIG,
40
- debug = false,
41
- logger = createDefaultLogger(debug),
42
- heartbeatInterval = DEFAULT_HEARTBEAT_INTERVAL,
43
- heartbeatTimeout = DEFAULT_HEARTBEAT_TIMEOUT,
44
- transport = DEFAULT_TRANSPORT
45
- }) => {
46
- const [connectionState, setConnectionState] = useState("disconnected");
47
- const [error, setError] = useState(void 0);
48
- const [startTime, setStartTime] = useState(void 0);
49
- const [metrics, setMetrics] = useState({
50
- latency: 0,
51
- messagesSent: 0,
52
- messagesReceived: 0,
53
- messagesQueued: 0,
54
- reconnectCount: 0,
55
- transportType: transport
56
- });
57
- const wsRef = useRef(void 0);
58
- const sseRef = useRef(void 0);
59
- const transportRef = useRef(transport);
60
- useEffect(() => {
61
- transportRef.current = transport;
62
- }, [transport]);
63
- const reconnectTimeoutRef = useRef(void 0);
64
- const retryCountRef = useRef(0);
65
- const messageQueueRef = useRef([]);
66
- const mountedRef = useRef(false);
67
- const currentChatKeyRef = useRef(void 0);
68
- const currentUserIdRef = useRef(void 0);
69
- const intentionalDisconnectRef = useRef(false);
70
- const eventHandlersRef = useRef(
71
- /* @__PURE__ */ new Map()
72
- );
73
- const onMessageRef = useRef(onMessage);
74
- const pendingMessagesRef = useRef(/* @__PURE__ */ new Map());
75
- const heartbeatIntervalRef = useRef(void 0);
76
- const heartbeatTimeoutRef = useRef(void 0);
77
- const lastPongRef = useRef(Date.now());
78
- const chatCreationPromiseRef = useRef(null);
79
- const loadChatRetryMapRef = useRef(/* @__PURE__ */ new Map());
80
- useEffect(() => {
81
- onMessageRef.current = onMessage;
82
- }, [onMessage]);
83
- const isConnected = connectionState === "connected";
84
- const on = useCallback((eventType, handler) => {
85
- if (!eventHandlersRef.current.has(eventType)) {
86
- eventHandlersRef.current.set(eventType, /* @__PURE__ */ new Set());
87
- }
88
- eventHandlersRef.current.get(eventType).add(handler);
89
- return () => off(eventType, handler);
90
- }, []);
91
- const off = useCallback((eventType, handler) => {
92
- const handlers = eventHandlersRef.current.get(eventType);
93
- if (handlers) {
94
- handlers.delete(handler);
95
- if (handlers.size === 0) {
96
- eventHandlersRef.current.delete(eventType);
97
- }
98
- }
99
- }, []);
100
- const emit = useCallback(
101
- (eventType, data) => {
102
- const handlers = eventHandlersRef.current.get(eventType);
103
- if (handlers) {
104
- handlers.forEach((handler) => {
105
- try {
106
- handler(data);
107
- } catch (error2) {
108
- logger.error(`Error in event handler for ${eventType}:`, error2);
109
- }
110
- });
111
- }
112
- },
113
- [logger]
114
- );
115
- const updateMetrics = useCallback((updates) => {
116
- setMetrics((prev) => ({ ...prev, ...updates }));
117
- }, []);
118
- const addToQueue = useCallback(
119
- (event) => {
120
- const currentQueueSize = messageQueueRef.current.length;
121
- const maxSize = queueConfig.maxSize || DEFAULT_QUEUE_CONFIG.maxSize;
122
- if (currentQueueSize >= maxSize) {
123
- if (queueConfig.dropStrategy === "newest") {
124
- logger.warn("Message queue full, dropping new message");
125
- return false;
126
- } else {
127
- const dropped = messageQueueRef.current.shift();
128
- logger.warn("Message queue full, dropped oldest message", dropped);
129
- }
130
- }
131
- messageQueueRef.current.push(event);
132
- updateMetrics({ messagesQueued: messageQueueRef.current.length });
133
- return true;
134
- },
135
- [queueConfig, logger, updateMetrics]
136
- );
137
- const calculateRetryInterval = useCallback(
138
- (retryCount) => {
139
- const config = { ...DEFAULT_RETRY_CONFIG, ...retryConfig };
140
- const {
141
- intervals = [],
142
- backoffMultiplier = 1.5,
143
- maxBackoffTime = 3e4
144
- } = config;
145
- if (retryCount < intervals.length) {
146
- return intervals[retryCount];
147
- }
148
- const baseInterval = intervals[intervals.length - 1] || 5e3;
149
- const backoffTime = baseInterval * Math.pow(backoffMultiplier, retryCount - intervals.length + 1);
150
- return Math.min(backoffTime, maxBackoffTime);
151
- },
152
- [retryConfig]
153
- );
154
- const startHeartbeat = useCallback(() => {
155
- if (!heartbeatInterval || heartbeatInterval <= 0) return;
156
- const sendPing = () => {
157
- if (wsRef.current?.readyState === WebSocket.OPEN) {
158
- const pingTime = Date.now();
159
- wsRef.current.send(
160
- JSON.stringify({ type: "ping", timestamp: pingTime })
161
- );
162
- heartbeatTimeoutRef.current = setTimeout(() => {
163
- logger.warn("Heartbeat timeout - no pong received");
164
- if (wsRef.current) {
165
- wsRef.current.close(4e3, "Heartbeat timeout");
166
- }
167
- }, heartbeatTimeout);
168
- }
169
- };
170
- if (heartbeatIntervalRef.current) {
171
- clearInterval(heartbeatIntervalRef.current);
172
- }
173
- heartbeatIntervalRef.current = setInterval(sendPing, heartbeatInterval);
174
- logger.debug("Heartbeat started");
175
- }, [heartbeatInterval, heartbeatTimeout, logger]);
176
- const stopHeartbeat = useCallback(() => {
177
- if (heartbeatIntervalRef.current) {
178
- clearInterval(heartbeatIntervalRef.current);
179
- heartbeatIntervalRef.current = void 0;
180
- }
181
- if (heartbeatTimeoutRef.current) {
182
- clearTimeout(heartbeatTimeoutRef.current);
183
- heartbeatTimeoutRef.current = void 0;
184
- }
185
- logger.debug("Heartbeat stopped");
186
- }, [logger]);
187
- const cleanup = useCallback(() => {
188
- if (wsRef.current && wsRef.current.readyState === WebSocket.OPEN) {
189
- wsRef.current.close(1e3, "Cleanup");
190
- }
191
- wsRef.current = void 0;
192
- if (sseRef.current) {
193
- sseRef.current.close();
194
- sseRef.current = void 0;
195
- }
196
- if (reconnectTimeoutRef.current) {
197
- clearTimeout(reconnectTimeoutRef.current);
198
- reconnectTimeoutRef.current = void 0;
199
- }
200
- stopHeartbeat();
201
- pendingMessagesRef.current.forEach(({ reject, timeout }) => {
202
- clearTimeout(timeout);
203
- reject(new Error("Connection closed"));
204
- });
205
- pendingMessagesRef.current.clear();
206
- loadChatRetryMapRef.current.forEach((retryState) => {
207
- if (retryState.timeoutId) {
208
- clearTimeout(retryState.timeoutId);
209
- }
210
- });
211
- loadChatRetryMapRef.current.clear();
212
- }, [stopHeartbeat]);
213
- const getRestApiUrl = useCallback((endpoint) => {
214
- return `${serverBaseUrl}/${endpoint}`;
215
- }, [serverBaseUrl]);
216
- const sendRestMessage = useCallback(
217
- async (endpoint, body) => {
218
- const url = getRestApiUrl(endpoint);
219
- logger.debug(`SSE REST API call: POST ${endpoint}`, body);
220
- try {
221
- const response = await fetch(url, {
222
- method: "POST",
223
- headers: {
224
- "Content-Type": "application/json"
225
- },
226
- body: JSON.stringify(body)
227
- });
228
- if (!response.ok) {
229
- const errorText = await response.text();
230
- throw new Error(`REST API error: ${response.status} - ${errorText}`);
231
- }
232
- const data = await response.json();
233
- return data;
234
- } catch (error2) {
235
- logger.error(`SSE REST API error for ${endpoint}:`, error2);
236
- throw error2;
237
- }
238
- },
239
- [getRestApiUrl, logger]
240
- );
241
- const connect = useCallback(
242
- async (userId) => {
243
- if (!mountedRef.current) {
244
- mountedRef.current = true;
245
- }
246
- if (!orgId) {
247
- const error2 = {
248
- code: "CONNECTION_FAILED",
249
- message: "Cannot connect: orgId is undefined",
250
- retryable: false,
251
- timestamp: Date.now()
252
- };
253
- logger.error("Cannot connect: orgId is undefined");
254
- setError(error2);
255
- return Promise.reject(error2);
256
- }
257
- if (wsRef.current?.readyState === WebSocket.OPEN) {
258
- logger.debug("Already connected (WebSocket)");
259
- return Promise.resolve();
260
- }
261
- if (sseRef.current?.readyState === EventSource.OPEN) {
262
- logger.debug("Already connected (SSE)");
263
- return Promise.resolve();
264
- }
265
- if (connectionState === "connecting" || connectionState === "reconnecting") {
266
- logger.debug("Connection already in progress");
267
- return Promise.resolve();
268
- }
269
- const maxRetries = retryConfig.maxRetries ?? DEFAULT_RETRY_CONFIG.maxRetries;
270
- if (retryCountRef.current >= maxRetries && !intentionalDisconnectRef.current) {
271
- const error2 = {
272
- code: "CONNECTION_FAILED",
273
- message: `Max retries (${maxRetries}) exceeded`,
274
- retryable: false,
275
- timestamp: Date.now()
276
- };
277
- setError(error2);
278
- updateMetrics({ lastError: error2 });
279
- return Promise.reject(error2);
280
- }
281
- cleanup();
282
- setConnectionState(
283
- retryCountRef.current > 0 ? "reconnecting" : "connecting"
284
- );
285
- intentionalDisconnectRef.current = false;
286
- return new Promise((resolve, reject) => {
287
- try {
288
- const connectionStartTime = Date.now();
289
- logger.info(`\u{1F504} Connecting with transport: ${transportRef.current}`);
290
- if (transportRef.current === "sse") {
291
- const sseUrl = getRestApiUrl(`stream?orgId=${orgId}&userId=${userId}&clientType=${clientType}&chatId=${currentChatKeyRef.current || ""}`);
292
- logger.debug("Connecting to SSE:", sseUrl);
293
- console.log(`\u23F3 Initiating SSE connection to ${sseUrl}...`);
294
- const eventSource = new EventSource(sseUrl);
295
- eventSource.onopen = () => {
296
- if (!mountedRef.current) {
297
- eventSource.close();
298
- reject(new Error("Component unmounted"));
299
- return;
300
- }
301
- const connectionTimeMs = Date.now() - connectionStartTime;
302
- const connectionTimeSec = (connectionTimeMs / 1e3).toFixed(2);
303
- logger.info("\u2705 SSE connected", {
304
- userId,
305
- retryCount: retryCountRef.current,
306
- connectionTime: `${connectionTimeSec}s (${connectionTimeMs}ms)`
307
- });
308
- console.log(`\u{1F50C} SSE connection established in ${connectionTimeSec} seconds`);
309
- setConnectionState("connected");
310
- setError(void 0);
311
- const wasReconnecting = retryCountRef.current > 0;
312
- retryCountRef.current = 0;
313
- updateMetrics({
314
- connectedAt: Date.now(),
315
- latency: connectionTimeMs,
316
- transportType: "sse",
317
- reconnectCount: wasReconnecting ? metrics.reconnectCount + 1 : metrics.reconnectCount
318
- });
319
- currentUserIdRef.current = userId;
320
- if (currentChatKeyRef.current) {
321
- logger.info("Loading chat after SSE reconnection:", currentChatKeyRef.current);
322
- sendRestMessage("load", {
323
- orgId,
324
- chatKey: currentChatKeyRef.current,
325
- userId
326
- }).then((response) => {
327
- if (response.success && response.data?.chat) {
328
- const chatEvent = {
329
- type: "load_chat_response",
330
- orgId,
331
- chatKey: currentChatKeyRef.current,
332
- userId,
333
- timestamp: Date.now(),
334
- data: response.data
335
- };
336
- emit("load_chat_response", chatEvent);
337
- if (onMessageRef.current) {
338
- onMessageRef.current(chatEvent);
339
- }
340
- }
341
- }).catch((err) => {
342
- logger.error("Failed to load chat after SSE reconnection:", err);
343
- });
344
- }
345
- emit("connected", { userId, wasReconnecting, transport: "sse" });
346
- resolve();
347
- };
348
- const handleSSEMessage = (event) => {
349
- if (!mountedRef.current) return;
350
- if (!event.data || event.data === "") {
351
- return;
352
- }
353
- try {
354
- const data = JSON.parse(event.data);
355
- if (!isChatEvent(data)) {
356
- logger.warn("Received invalid SSE message format:", data);
357
- return;
358
- }
359
- const chatEvent = data;
360
- logger.debug("SSE message received:", chatEvent.type);
361
- updateMetrics({
362
- messagesReceived: metrics.messagesReceived + 1,
363
- lastMessageAt: Date.now()
364
- });
365
- switch (chatEvent.type) {
366
- case "new_chat_created":
367
- const newChatKey = chatEvent.data?.chatKey;
368
- if (newChatKey) {
369
- logger.info("New chat created with key:", newChatKey);
370
- currentChatKeyRef.current = newChatKey;
371
- if (chatCreationPromiseRef.current) {
372
- chatCreationPromiseRef.current.resolve(newChatKey);
373
- chatCreationPromiseRef.current = null;
374
- }
375
- }
376
- break;
377
- case "load_chat_response":
378
- const chat = chatEvent.data?.chat;
379
- if (chat && chat.key) {
380
- logger.info("Chat loaded with key:", chat.key);
381
- currentChatKeyRef.current = chat.key;
382
- }
383
- break;
384
- case "chat_ended":
385
- logger.info("Chat ended, clearing key");
386
- currentChatKeyRef.current = void 0;
387
- break;
388
- }
389
- emit(chatEvent.type || "message", chatEvent);
390
- if (onMessageRef.current) {
391
- onMessageRef.current(chatEvent);
392
- }
393
- } catch (error2) {
394
- logger.error("Failed to parse SSE message:", error2);
395
- }
396
- };
397
- eventSource.addEventListener("message", handleSSEMessage);
398
- eventSource.addEventListener("reconnected", handleSSEMessage);
399
- eventSource.addEventListener("typing", handleSSEMessage);
400
- eventSource.addEventListener("stopped_typing", handleSSEMessage);
401
- eventSource.addEventListener("waiting", handleSSEMessage);
402
- eventSource.addEventListener("waiting_for_agent", handleSSEMessage);
403
- eventSource.addEventListener("human_agent_joined", handleSSEMessage);
404
- eventSource.addEventListener("human_agent_left", handleSSEMessage);
405
- eventSource.addEventListener("chat_ended", handleSSEMessage);
406
- eventSource.addEventListener("chat_updated", handleSSEMessage);
407
- eventSource.addEventListener("load_chat_response", handleSSEMessage);
408
- eventSource.addEventListener("new_chat_created", handleSSEMessage);
409
- eventSource.addEventListener("show_csat_survey", handleSSEMessage);
410
- eventSource.addEventListener("csat_response", handleSSEMessage);
411
- eventSource.addEventListener("user_suggested_actions", handleSSEMessage);
412
- eventSource.addEventListener("agent_execution_started", handleSSEMessage);
413
- eventSource.addEventListener("agent_execution_ended", handleSSEMessage);
414
- eventSource.addEventListener("agent_context_update", handleSSEMessage);
415
- eventSource.addEventListener("plan_pending_approval", handleSSEMessage);
416
- eventSource.addEventListener("step_started", handleSSEMessage);
417
- eventSource.addEventListener("step_completed", handleSSEMessage);
418
- eventSource.addEventListener("step_failed", handleSSEMessage);
419
- eventSource.addEventListener("plan_completed", handleSSEMessage);
420
- eventSource.addEventListener("skills_changed", handleSSEMessage);
421
- eventSource.addEventListener("summary_update", handleSSEMessage);
422
- eventSource.onerror = (error2) => {
423
- logger.error("SSE error:", error2);
424
- if (!mountedRef.current) return;
425
- if (eventSource.readyState === EventSource.CLOSED) {
426
- const sseError = {
427
- code: "CONNECTION_FAILED",
428
- message: "SSE connection failed",
429
- retryable: true,
430
- timestamp: Date.now()
431
- };
432
- setError(sseError);
433
- updateMetrics({ lastError: sseError });
434
- setConnectionState("disconnected");
435
- emit("disconnected", { reason: "SSE error" });
436
- if (!intentionalDisconnectRef.current && mountedRef.current) {
437
- const retryInterval = calculateRetryInterval(retryCountRef.current);
438
- retryCountRef.current++;
439
- logger.info(`SSE reconnecting in ${retryInterval}ms (attempt ${retryCountRef.current})`);
440
- if (reconnectTimeoutRef.current) {
441
- clearTimeout(reconnectTimeoutRef.current);
442
- }
443
- reconnectTimeoutRef.current = setTimeout(() => {
444
- connect(userId);
445
- }, retryInterval);
446
- }
447
- }
448
- };
449
- sseRef.current = eventSource;
450
- return;
451
- }
452
- const wsUrl = `${serverBaseUrl}?orgId=${orgId}&userId=${userId}&clientType=${clientType}&product=${product}`;
453
- logger.debug("Connecting to WebSocket:", wsUrl);
454
- console.log(`\u23F3 Initiating WebSocket connection to ${serverBaseUrl}...`);
455
- const ws = new WebSocket(wsUrl);
456
- ws.onopen = () => {
457
- if (!mountedRef.current) {
458
- ws.close(1e3, "Component unmounted");
459
- reject(new Error("Component unmounted"));
460
- return;
461
- }
462
- const connectionTimeMs = Date.now() - connectionStartTime;
463
- const connectionTimeSec = (connectionTimeMs / 1e3).toFixed(2);
464
- logger.info("\u2705 WebSocket connected", {
465
- userId,
466
- retryCount: retryCountRef.current,
467
- connectionTime: `${connectionTimeSec}s (${connectionTimeMs}ms)`
468
- });
469
- console.log(`\u{1F50C} WebSocket connection established in ${connectionTimeSec} seconds`);
470
- setConnectionState("connected");
471
- setError(void 0);
472
- const wasReconnecting = retryCountRef.current > 0;
473
- retryCountRef.current = 0;
474
- updateMetrics({
475
- connectedAt: Date.now(),
476
- latency: connectionTimeMs,
477
- reconnectCount: wasReconnecting ? metrics.reconnectCount + 1 : metrics.reconnectCount
478
- });
479
- while (messageQueueRef.current.length > 0 && ws.readyState === WebSocket.OPEN) {
480
- const event = messageQueueRef.current.shift();
481
- if (event) {
482
- ws.send(JSON.stringify({ ...event, timestamp: Date.now() }));
483
- updateMetrics({
484
- messagesSent: metrics.messagesSent + 1,
485
- messagesQueued: messageQueueRef.current.length
486
- });
487
- }
488
- }
489
- currentUserIdRef.current = userId;
490
- if (currentChatKeyRef.current) {
491
- if (!orgId) {
492
- logger.error("Cannot resubscribe to chat: orgId is undefined");
493
- } else {
494
- logger.info(
495
- "Resubscribing to chat after reconnection:",
496
- currentChatKeyRef.current
497
- );
498
- const resubscribeEvent = {
499
- type: "load_chat",
500
- orgId,
501
- chatKey: currentChatKeyRef.current,
502
- userId,
503
- timestamp: Date.now(),
504
- data: {}
505
- };
506
- ws.send(JSON.stringify(resubscribeEvent));
507
- }
508
- }
509
- startHeartbeat();
510
- emit("connected", { userId, wasReconnecting });
511
- resolve();
512
- };
513
- ws.onmessage = (event) => {
514
- if (!mountedRef.current) return;
515
- try {
516
- const data = JSON.parse(event.data);
517
- if (!isChatEvent(data)) {
518
- logger.warn("Received invalid message format:", data);
519
- return;
520
- }
521
- const chatEvent = data;
522
- logger.debug("Message received:", chatEvent.type);
523
- updateMetrics({
524
- messagesReceived: metrics.messagesReceived + 1,
525
- lastMessageAt: Date.now()
526
- });
527
- if (chatEvent.type === "pong") {
528
- if (heartbeatTimeoutRef.current) {
529
- clearTimeout(heartbeatTimeoutRef.current);
530
- heartbeatTimeoutRef.current = void 0;
531
- }
532
- const latency = Date.now() - (chatEvent.timestamp || Date.now());
533
- lastPongRef.current = Date.now();
534
- updateMetrics({ latency });
535
- return;
536
- }
537
- switch (chatEvent.type) {
538
- case "new_chat_created":
539
- const newChatKey = chatEvent.data?.chatKey;
540
- if (newChatKey) {
541
- logger.info("New chat created with key:", newChatKey);
542
- currentChatKeyRef.current = newChatKey;
543
- if (chatCreationPromiseRef.current) {
544
- chatCreationPromiseRef.current.resolve(newChatKey);
545
- chatCreationPromiseRef.current = null;
546
- }
547
- if (!orgId) {
548
- logger.error("Cannot send load_chat: orgId is undefined");
549
- return;
550
- }
551
- const loadChatEvent = {
552
- type: "load_chat",
553
- orgId,
554
- chatKey: newChatKey,
555
- userId: currentUserIdRef.current || userId,
556
- timestamp: Date.now(),
557
- data: {}
558
- };
559
- ws.send(JSON.stringify(loadChatEvent));
560
- }
561
- break;
562
- case "load_chat_response":
563
- const chat = chatEvent.data?.chat;
564
- if (chat && chat.key) {
565
- logger.info("Chat loaded with key:", chat.key);
566
- currentChatKeyRef.current = chat.key;
567
- const retryState = loadChatRetryMapRef.current.get(chat.key);
568
- if (retryState) {
569
- if (retryState.timeoutId) {
570
- clearTimeout(retryState.timeoutId);
571
- }
572
- loadChatRetryMapRef.current.delete(chat.key);
573
- }
574
- } else if (!chat) {
575
- logger.warn("Chat load failed, clearing key");
576
- currentChatKeyRef.current = void 0;
577
- }
578
- break;
579
- case "chat_ended":
580
- logger.info("Chat ended, clearing key");
581
- currentChatKeyRef.current = void 0;
582
- break;
583
- case "error":
584
- const errorMessage = chatEvent.data?.message || "Unknown error";
585
- logger.error("Received error from server:", errorMessage);
586
- if (errorMessage.includes("nats: key not found") || errorMessage.includes("Failed to load chat")) {
587
- const chatKeyFromError = currentChatKeyRef.current;
588
- if (chatKeyFromError) {
589
- const maxRetries2 = 5;
590
- let retryState = loadChatRetryMapRef.current.get(chatKeyFromError);
591
- if (!retryState) {
592
- retryState = { retryCount: 0, timeoutId: null };
593
- loadChatRetryMapRef.current.set(chatKeyFromError, retryState);
594
- }
595
- if (retryState.retryCount < maxRetries2) {
596
- const delay = 200 * Math.pow(2, retryState.retryCount);
597
- retryState.retryCount++;
598
- logger.info(
599
- `Chat load failed, retrying (${retryState.retryCount}/${maxRetries2}) in ${delay}ms...`,
600
- chatKeyFromError
601
- );
602
- retryState.timeoutId = setTimeout(() => {
603
- if (!wsRef.current || !mountedRef.current) return;
604
- if (!orgId) {
605
- logger.error("Cannot retry load_chat: orgId is undefined", chatKeyFromError);
606
- loadChatRetryMapRef.current.delete(chatKeyFromError);
607
- return;
608
- }
609
- logger.debug("Retrying load_chat:", chatKeyFromError);
610
- const retryLoadEvent = {
611
- type: "load_chat",
612
- orgId,
613
- chatKey: chatKeyFromError,
614
- userId: currentUserIdRef.current || "",
615
- timestamp: Date.now(),
616
- data: {}
617
- };
618
- ws.send(JSON.stringify(retryLoadEvent));
619
- }, delay);
620
- } else {
621
- logger.error("Max retries reached for loading chat:", chatKeyFromError);
622
- loadChatRetryMapRef.current.delete(chatKeyFromError);
623
- }
624
- }
625
- }
626
- const wsError = {
627
- code: "NETWORK_ERROR",
628
- message: errorMessage,
629
- retryable: true,
630
- timestamp: Date.now()
631
- };
632
- setError(wsError);
633
- updateMetrics({ lastError: wsError });
634
- break;
635
- }
636
- emit(chatEvent.type || "message", chatEvent);
637
- if (onMessageRef.current) {
638
- onMessageRef.current(chatEvent);
639
- }
640
- } catch (error2) {
641
- logger.error("Failed to parse WebSocket message:", error2);
642
- const parseError = {
643
- code: "PARSE_ERROR",
644
- message: "Failed to parse message",
645
- retryable: false,
646
- timestamp: Date.now()
647
- };
648
- setError(parseError);
649
- updateMetrics({ lastError: parseError });
650
- }
651
- };
652
- ws.onclose = (event) => {
653
- if (!mountedRef.current) return;
654
- logger.info("WebSocket closed", {
655
- code: event.code,
656
- reason: event.reason
657
- });
658
- setConnectionState("disconnected");
659
- wsRef.current = void 0;
660
- stopHeartbeat();
661
- emit("disconnected", { code: event.code, reason: event.reason });
662
- if (event.code !== 1e3 && !intentionalDisconnectRef.current && mountedRef.current) {
663
- const retryInterval = calculateRetryInterval(
664
- retryCountRef.current
665
- );
666
- retryCountRef.current++;
667
- logger.info(
668
- `Reconnecting in ${retryInterval}ms (attempt ${retryCountRef.current})`
669
- );
670
- if (reconnectTimeoutRef.current) {
671
- clearTimeout(reconnectTimeoutRef.current);
672
- }
673
- reconnectTimeoutRef.current = setTimeout(() => {
674
- connect(userId);
675
- }, retryInterval);
676
- }
677
- };
678
- ws.onerror = (error2) => {
679
- logger.error("WebSocket error:", error2);
680
- if (!mountedRef.current) return;
681
- const wsError = {
682
- code: "CONNECTION_FAILED",
683
- message: "WebSocket connection failed",
684
- retryable: true,
685
- timestamp: Date.now()
686
- };
687
- setError(wsError);
688
- updateMetrics({ lastError: wsError });
689
- reject(wsError);
690
- };
691
- wsRef.current = ws;
692
- } catch (error2) {
693
- logger.error("Failed to create WebSocket:", error2);
694
- const wsError = {
695
- code: "CONNECTION_FAILED",
696
- message: error2 instanceof Error ? error2.message : "Failed to create connection",
697
- retryable: true,
698
- timestamp: Date.now()
699
- };
700
- setError(wsError);
701
- updateMetrics({ lastError: wsError });
702
- reject(wsError);
703
- }
704
- });
705
- },
706
- [
707
- serverBaseUrl,
708
- orgId,
709
- clientType,
710
- product,
711
- connectionState,
712
- logger,
713
- retryConfig,
714
- metrics,
715
- updateMetrics,
716
- cleanup,
717
- calculateRetryInterval,
718
- startHeartbeat,
719
- emit
720
- ]
721
- );
722
- const sendMessage = useCallback(
723
- (event, overrideUserId) => {
724
- return new Promise(async (resolve, reject) => {
725
- if (!mountedRef.current) {
726
- reject(new Error("Component not mounted"));
727
- return;
728
- }
729
- const fullEvent = {
730
- ...event,
731
- timestamp: Date.now()
732
- };
733
- logger.debug("Sending message:", fullEvent.type);
734
- if (transportRef.current === "sse") {
735
- if (!sseRef.current || sseRef.current.readyState !== EventSource.OPEN) {
736
- logger.debug("SSE not connected, attempting to connect");
737
- if (connectionState === "disconnected" && overrideUserId) {
738
- try {
739
- await connect(overrideUserId);
740
- } catch (error2) {
741
- reject(error2);
742
- return;
743
- }
744
- } else {
745
- reject(new Error("SSE not connected"));
746
- return;
747
- }
748
- }
749
- try {
750
- switch (fullEvent.type) {
751
- case "message":
752
- await sendRestMessage("send", {
753
- orgId: fullEvent.orgId || orgId,
754
- chatKey: fullEvent.chatKey || currentChatKeyRef.current,
755
- userId: fullEvent.userId || currentUserIdRef.current,
756
- message: fullEvent.message
757
- });
758
- break;
759
- case "typing":
760
- await sendRestMessage("typing", {
761
- orgId: fullEvent.orgId || orgId,
762
- chatKey: fullEvent.chatKey || currentChatKeyRef.current,
763
- userId: fullEvent.userId || currentUserIdRef.current,
764
- typing: true
765
- });
766
- break;
767
- case "stopped_typing":
768
- await sendRestMessage("typing", {
769
- orgId: fullEvent.orgId || orgId,
770
- chatKey: fullEvent.chatKey || currentChatKeyRef.current,
771
- userId: fullEvent.userId || currentUserIdRef.current,
772
- typing: false
773
- });
774
- break;
775
- case "load_chat":
776
- const loadResponse = await sendRestMessage("load", {
777
- orgId: fullEvent.orgId || orgId,
778
- chatKey: fullEvent.chatKey,
779
- userId: fullEvent.userId || currentUserIdRef.current
780
- });
781
- if (loadResponse.success && loadResponse.data?.chat) {
782
- currentChatKeyRef.current = loadResponse.data.chat.key;
783
- const chatEvent = {
784
- type: "load_chat_response",
785
- orgId: fullEvent.orgId,
786
- chatKey: loadResponse.data.chat.key,
787
- userId: fullEvent.userId,
788
- timestamp: Date.now(),
789
- data: loadResponse.data
790
- };
791
- emit("load_chat_response", chatEvent);
792
- if (onMessageRef.current) {
793
- onMessageRef.current(chatEvent);
794
- }
795
- }
796
- break;
797
- case "new_chat":
798
- const createResponse = await sendRestMessage("create", {
799
- orgId: fullEvent.orgId || orgId,
800
- userId: fullEvent.userId || currentUserIdRef.current,
801
- metadata: fullEvent.data
802
- });
803
- if (createResponse.success && createResponse.data?.chatKey) {
804
- currentChatKeyRef.current = createResponse.data.chatKey;
805
- const newChatEvent = {
806
- type: "new_chat_created",
807
- orgId: fullEvent.orgId,
808
- chatKey: createResponse.data.chatKey,
809
- userId: fullEvent.userId,
810
- timestamp: Date.now(),
811
- data: { chatKey: createResponse.data.chatKey }
812
- };
813
- emit("new_chat_created", newChatEvent);
814
- if (onMessageRef.current) {
815
- onMessageRef.current(newChatEvent);
816
- }
817
- if (chatCreationPromiseRef.current) {
818
- chatCreationPromiseRef.current.resolve(createResponse.data.chatKey);
819
- chatCreationPromiseRef.current = null;
820
- }
821
- }
822
- break;
823
- case "end_chat":
824
- await sendRestMessage("end", {
825
- orgId: fullEvent.orgId || orgId,
826
- chatKey: fullEvent.chatKey || currentChatKeyRef.current,
827
- userId: fullEvent.userId || currentUserIdRef.current,
828
- data: fullEvent.data
829
- });
830
- break;
831
- case "human_agent_joined":
832
- await sendRestMessage("agent-join", {
833
- orgId: fullEvent.orgId || orgId,
834
- chatKey: fullEvent.chatKey || currentChatKeyRef.current,
835
- user: fullEvent.data?.user
836
- });
837
- break;
838
- case "human_agent_left":
839
- await sendRestMessage("agent-leave", {
840
- orgId: fullEvent.orgId || orgId,
841
- chatKey: fullEvent.chatKey || currentChatKeyRef.current,
842
- user: fullEvent.data?.user
843
- });
844
- break;
845
- // Event types that use the generic /event endpoint
846
- case "load_agent_context":
847
- case "skill_activate":
848
- case "skill_deactivate":
849
- case "sync_metadata":
850
- case "sync_user_session":
851
- case "csat_response":
852
- case "plan_approved":
853
- case "plan_rejected":
854
- await sendRestMessage("event", {
855
- type: fullEvent.type,
856
- orgId: fullEvent.orgId || orgId,
857
- chatKey: fullEvent.chatKey || currentChatKeyRef.current,
858
- userId: fullEvent.userId || currentUserIdRef.current,
859
- data: fullEvent.data
860
- });
861
- break;
862
- default:
863
- logger.warn("Sending unrecognized event type via generic endpoint:", fullEvent.type);
864
- await sendRestMessage("event", {
865
- type: fullEvent.type,
866
- orgId: fullEvent.orgId || orgId,
867
- chatKey: fullEvent.chatKey || currentChatKeyRef.current,
868
- userId: fullEvent.userId || currentUserIdRef.current,
869
- data: fullEvent.data
870
- });
871
- break;
872
- }
873
- updateMetrics({ messagesSent: metrics.messagesSent + 1 });
874
- logger.debug("SSE REST message sent successfully");
875
- resolve();
876
- } catch (error2) {
877
- logger.error("Failed to send SSE REST message:", error2);
878
- const sendError = {
879
- code: "SEND_FAILED",
880
- message: error2 instanceof Error ? error2.message : "Failed to send message",
881
- retryable: true,
882
- timestamp: Date.now()
883
- };
884
- setError(sendError);
885
- updateMetrics({ lastError: sendError });
886
- reject(sendError);
887
- }
888
- return;
889
- }
890
- if (!wsRef.current || wsRef.current.readyState !== WebSocket.OPEN) {
891
- if (addToQueue(fullEvent)) {
892
- logger.debug("Message queued, attempting to connect");
893
- if (connectionState === "disconnected" && overrideUserId) {
894
- connect(overrideUserId).then(() => resolve()).catch(reject);
895
- } else {
896
- resolve();
897
- }
898
- } else {
899
- reject(new Error("Message queue full"));
900
- }
901
- return;
902
- }
903
- try {
904
- wsRef.current.send(JSON.stringify(fullEvent));
905
- updateMetrics({ messagesSent: metrics.messagesSent + 1 });
906
- logger.debug("Message sent successfully");
907
- resolve();
908
- } catch (error2) {
909
- logger.error("Failed to send message:", error2);
910
- if (addToQueue(fullEvent)) {
911
- resolve();
912
- } else {
913
- const sendError = {
914
- code: "SEND_FAILED",
915
- message: error2 instanceof Error ? error2.message : "Failed to send message",
916
- retryable: true,
917
- timestamp: Date.now()
918
- };
919
- setError(sendError);
920
- updateMetrics({ lastError: sendError });
921
- reject(sendError);
922
- }
923
- }
924
- });
925
- },
926
- [connectionState, connect, addToQueue, logger, metrics, updateMetrics, sendRestMessage, emit, orgId]
927
- );
928
- const startNewChat = useCallback(
929
- (userId, data) => {
930
- return new Promise((resolve, reject) => {
931
- if (!userId) {
932
- reject(new Error("User ID is required"));
933
- return;
934
- }
935
- logger.info("Requesting new chat from server with userId:", userId);
936
- setStartTime(/* @__PURE__ */ new Date());
937
- chatCreationPromiseRef.current = { resolve, reject };
938
- sendMessage(
939
- {
940
- type: "new_chat",
941
- orgId,
942
- chatKey: "",
943
- // Server will generate
944
- userId,
945
- data: data ?? {}
946
- },
947
- userId
948
- ).catch((error2) => {
949
- chatCreationPromiseRef.current = null;
950
- reject(error2);
951
- });
952
- setTimeout(() => {
953
- if (chatCreationPromiseRef.current) {
954
- chatCreationPromiseRef.current = null;
955
- reject(new Error("Chat creation timed out"));
956
- }
957
- }, 3e4);
958
- });
959
- },
960
- [sendMessage, orgId, logger]
961
- );
962
- const disconnect = useCallback(
963
- (intentional = true) => {
964
- logger.info("Disconnecting", { intentional, transport: transportRef.current });
965
- intentionalDisconnectRef.current = intentional;
966
- cleanup();
967
- setConnectionState("disconnected");
968
- messageQueueRef.current = [];
969
- retryCountRef.current = 0;
970
- mountedRef.current = false;
971
- currentChatKeyRef.current = void 0;
972
- currentUserIdRef.current = void 0;
973
- },
974
- [cleanup, logger]
975
- );
976
- const clearError = useCallback(() => {
977
- setError(void 0);
978
- }, []);
979
- useEffect(() => {
980
- mountedRef.current = true;
981
- return () => {
982
- mountedRef.current = false;
983
- disconnect(true);
984
- };
985
- }, []);
986
- return {
987
- connectionState,
988
- isConnected,
989
- sendMessage,
990
- error,
991
- connect,
992
- startNewChat,
993
- startTime,
994
- disconnect,
995
- metrics,
996
- on,
997
- off,
998
- clearError
999
- };
1000
- };
1001
-
1002
- // hooks/use-websocket-chat-customer.ts
1003
- var useWebSocketChatCustomer = ({
1004
- serverBaseUrl,
1005
- orgId,
1006
- chatKey,
1007
- product
1008
- }) => {
1009
- const [currentChat, setCurrentChat] = useState2(void 0);
1010
- const handleMessage = useCallback2((chatEvent) => {
1011
- console.log("Received event:", chatEvent.type);
1012
- switch (chatEvent.type) {
1013
- case "message":
1014
- if (!chatEvent.message) return;
1015
- console.log(
1016
- "got message:",
1017
- chatEvent.message.role,
1018
- ":",
1019
- chatEvent.message.content
1020
- );
1021
- setCurrentChat((prev) => {
1022
- if (!prev) return prev;
1023
- return {
1024
- ...prev,
1025
- messages: [...prev.messages, chatEvent.message]
1026
- };
1027
- });
1028
- break;
1029
- case "chat_updated":
1030
- const chat = chatEvent.data?.chat?.value;
1031
- if (chat) {
1032
- setCurrentChat(chat);
1033
- }
1034
- break;
1035
- case "load_chat":
1036
- const history = chatEvent.data?.chat;
1037
- if (!history) return;
1038
- setCurrentChat(history);
1039
- break;
1040
- default:
1041
- break;
1042
- }
1043
- }, []);
1044
- const base = useWebSocketChatBase({
1045
- serverBaseUrl,
1046
- orgId,
1047
- clientType: "customer",
1048
- onMessage: handleMessage,
1049
- product
1050
- });
1051
- return {
1052
- ...base,
1053
- chatKey,
1054
- title: currentChat?.title,
1055
- messages: currentChat?.messages ?? [],
1056
- users: currentChat?.users ?? [],
1057
- isWaiting: currentChat?.isWaiting ?? false,
1058
- isWaitingForAgent: currentChat?.isWaitingForAgent ?? false,
1059
- aiEngaged: currentChat?.aiEngaged ?? false,
1060
- humanAgentEngaged: currentChat?.humanAgentEngaged ?? false,
1061
- metadata: currentChat?.metadata ?? {},
1062
- status: currentChat?.status
1063
- };
1064
- };
1065
- export {
1066
- useWebSocketChatCustomer
1067
- };
1068
- //# sourceMappingURL=use-websocket-chat-customer.mjs.map