@elqnt/chat 3.3.0 → 3.5.0

package/dist/hooks/index.mjs

@@ -4,6 +4,18 @@
 import { useCallback, useEffect, useRef, useState } from "react";
 
 // transport/types.ts
+async function resolveTransportToken(cfg) {
+  if (!cfg) return void 0;
+  if (cfg.getToken) {
+    try {
+      const t = await cfg.getToken();
+      return t ?? void 0;
+    } catch {
+      return void 0;
+    }
+  }
+  return cfg.token ?? void 0;
+}
 function createLogger(debug = false) {
   return {
     debug: debug ? console.log.bind(console, "[chat]") : () => {
@@ -84,9 +96,12 @@ function createSSETransport(options = {}) {
     }
     const url = `${config.baseUrl}/${endpoint}`;
     logger.debug(`POST ${endpoint}`, body);
+    const token = await resolveTransportToken(config);
+    const headers = { "Content-Type": "application/json" };
+    if (token) headers["Authorization"] = `Bearer ${token}`;
     const response = await fetch(url, {
       method: "POST",
-      headers: { "Content-Type": "application/json" },
+      headers,
       body: JSON.stringify(body)
     });
     if (!response.ok) {
@@ -213,6 +228,7 @@ function createSSETransport(options = {}) {
         reconnectTimeout = void 0;
       }
       state = retryCount > 0 ? "reconnecting" : "connecting";
+      const connectToken = await resolveTransportToken(cfg);
       return new Promise((resolve, reject) => {
         const connectionStart = Date.now();
         const streamParams = new URLSearchParams({
@@ -221,6 +237,7 @@ function createSSETransport(options = {}) {
           clientType: cfg.clientType
         });
         if (cfg.chatKey) streamParams.set("chatId", cfg.chatKey);
+        if (connectToken) streamParams.set("token", connectToken);
         const url = `${cfg.baseUrl}/stream?${streamParams.toString()}`;
         logger.debug("Connecting to:", url);
         const es = new EventSource(url);
@@ -474,9 +491,12 @@ function createFetchSSETransport(options = {}) {
     }
     const url = `${config.baseUrl}/${endpoint}`;
     logger.debug(`POST ${endpoint}`, body);
+    const token = await resolveTransportToken(config);
+    const headers = { "Content-Type": "application/json" };
+    if (token) headers["Authorization"] = `Bearer ${token}`;
     const response = await customFetch(url, {
       method: "POST",
-      headers: { "Content-Type": "application/json" },
+      headers,
       body: JSON.stringify(body)
     });
     if (!response.ok) {
@@ -519,15 +539,19 @@ function createFetchSSETransport(options = {}) {
       clientType: cfg.clientType
     });
     if (cfg.chatKey) streamParams.set("chatId", cfg.chatKey);
+    const token = await resolveTransportToken(cfg);
+    if (token) streamParams.set("token", token);
     const url = `${cfg.baseUrl}/stream?${streamParams.toString()}`;
     logger.debug("Connecting to:", url);
     abortController = new AbortController();
+    const streamHeaders = {
+      Accept: "text/event-stream",
+      "Cache-Control": "no-cache"
+    };
+    if (token) streamHeaders["Authorization"] = `Bearer ${token}`;
     const response = await customFetch(url, {
       method: "GET",
-      headers: {
-        Accept: "text/event-stream",
-        "Cache-Control": "no-cache"
-      },
+      headers: streamHeaders,
       signal: abortController.signal
     });
     if (!response.ok) {
@@ -841,7 +865,9 @@ function useChat(options) {
     onConnectionChange,
     autoConnect = false,
     retryConfig,
-    debug = false
+    debug = false,
+    token,
+    getToken
   } = options;
   const [connectionState, setConnectionState] = useState("disconnected");
   const [currentChat, setCurrentChat] = useState(null);
@@ -860,10 +886,14 @@ function useChat(options) {
   const onMessageRef = useRef(onMessage);
   const onErrorRef = useRef(onError);
   const typingTimeoutRef = useRef(null);
+  const getTokenRef = useRef(getToken);
+  const tokenRef = useRef(token);
   useEffect(() => {
     onMessageRef.current = onMessage;
     onErrorRef.current = onError;
-  }, [onMessage, onError]);
+    getTokenRef.current = getToken;
+    tokenRef.current = token;
+  }, [onMessage, onError, getToken, token]);
   useEffect(() => {
     if (typeof transportOption === "object") {
       transportRef.current = transportOption;
@@ -946,7 +976,10 @@ function useChat(options) {
         userId,
         clientType,
         chatKey: chatKey || void 0,
-        debug
+        debug,
+        // Always resolve via the ref so a long-lived/reconnecting stream picks
+        // up a freshly-minted token rather than a stale captured one.
+        getToken: () => getTokenRef.current ? getTokenRef.current() : tokenRef.current
       });
       setConnectionState("connected");
       setError(null);