mitra-interactions-sdk 1.0.63 → 1.0.64-rc.0

package/dist/index.js

@@ -1377,6 +1377,7 @@ async function agentHttpPost(path, body) {
   if (typeof fetchFn !== "function") {
     throw new Error("Agent Chat (HTTP): fetch indispon\xEDvel. Use Node 18+ ou um browser.");
   }
+  if (isTokenExpiring()) await tryRefreshToken();
   const resp = await fetchFn(`${getAgentHttpBase()}${path}`, {
     method: "POST",
     headers: {
@@ -1406,16 +1407,16 @@ async function httpSendPrompt(payload, taskId) {
     routeMessage({ type: "stream_end", taskId: resolvedTaskId });
   }
 }
-function connect() {
-  if (ws && ws.readyState === WebSocket.OPEN) {
-    return Promise.resolve(ws);
-  }
-  if (connectPromise) return connectPromise;
+function isChannelError(err) {
+  const msg = err instanceof Error ? err.message : String(err != null ? err : "");
+  return /conexão fechada|timeout ao conectar|erro ao conectar/i.test(msg);
+}
+function openSocket() {
   const baseUrl = getWsUrl();
   const token = getRawToken();
   const separator = baseUrl.includes("?") ? "&" : "?";
   const url = `${baseUrl}${separator}token=${encodeURIComponent(token)}`;
-  connectPromise = new Promise((resolve, reject) => {
+  return new Promise((resolve, reject) => {
     const socket = new WebSocket(url);
     let opened = false;
     const timer = setTimeout(() => {
@@ -1424,7 +1425,6 @@ function connect() {
           socket.close();
         } catch (e) {
         }
-        connectPromise = null;
         reject(new Error("Agent Chat: timeout ao conectar no WebSocket"));
       }
     }, 15e3);
@@ -1432,18 +1432,20 @@ function connect() {
       opened = true;
       clearTimeout(timer);
       ws = socket;
-      connectPromise = null;
       resolve(socket);
     };
     socket.onerror = () => {
       if (!opened) {
         clearTimeout(timer);
-        connectPromise = null;
         reject(new Error("Agent Chat: erro ao conectar no WebSocket"));
       }
     };
     socket.onclose = () => {
       if (ws === socket) ws = null;
+      if (!opened) {
+        clearTimeout(timer);
+        reject(new Error("Agent Chat: conex\xE3o fechada"));
+      }
       for (const [id, pending] of pendingRequests) {
         pending.reject(new Error("Agent Chat: conex\xE3o fechada"));
         pendingRequests.delete(id);
@@ -1452,7 +1454,29 @@ function connect() {
     };
     socket.onmessage = (event) => routeMessage(event.data);
   });
-  return connectPromise;
+}
+function connect() {
+  if (ws && ws.readyState === WebSocket.OPEN) {
+    return Promise.resolve(ws);
+  }
+  if (connectPromise) return connectPromise;
+  const attempt = (async () => {
+    if (isTokenExpiring()) await tryRefreshToken();
+    try {
+      return await openSocket();
+    } catch (err) {
+      if (!isChannelError(err)) throw err;
+      const refreshed = await tryRefreshToken();
+      if (!refreshed) throw err;
+      return openSocket();
+    }
+  })();
+  connectPromise = attempt;
+  const clear = () => {
+    if (connectPromise === attempt) connectPromise = null;
+  };
+  attempt.then(clear, clear);
+  return attempt;
 }
 function routeMessage(raw) {
   var _a, _b;
@@ -1492,6 +1516,21 @@ function routeMessage(raw) {
     for (const handler of pendingHandlers.values()) handler(msg);
   }
 }
+function requestOverSocket(type, payload) {
+  return connect().then((socket) => new Promise((resolve, reject) => {
+    const requestId = nextRequestId();
+    pendingRequests.set(requestId, {
+      resolve: (data) => resolve(data),
+      reject
+    });
+    try {
+      socket.send(JSON.stringify({ type, requestId, payload }));
+    } catch (err) {
+      pendingRequests.delete(requestId);
+      reject(err instanceof Error ? err : new Error(String(err)));
+    }
+  }));
+}
 var transport = {
   async ensureConnected() {
     if (isHttpEnabled()) return;
@@ -1504,20 +1543,14 @@ var transport = {
       if (data && data.ok === false) throw new Error(data.error || "Erro desconhecido");
       return (_a = data == null ? void 0 : data.data) != null ? _a : data;
     }
-    const socket = await connect();
-    const requestId = nextRequestId();
-    return new Promise((resolve, reject) => {
-      pendingRequests.set(requestId, {
-        resolve: (data) => resolve(data),
-        reject
-      });
-      try {
-        socket.send(JSON.stringify({ type, requestId, payload }));
-      } catch (err) {
-        pendingRequests.delete(requestId);
-        reject(err instanceof Error ? err : new Error(String(err)));
-      }
-    });
+    try {
+      return await requestOverSocket(type, payload);
+    } catch (err) {
+      if (!isChannelError(err)) throw err;
+      const refreshed = await tryRefreshToken();
+      if (!refreshed) throw err;
+      return requestOverSocket(type, payload);
+    }
   },
   send(type, payload, taskId) {
     if (isHttpEnabled()) {
@@ -1771,6 +1804,7 @@ function createMitraInstance(initialConfig) {
             ...((_b = data.merge) == null ? void 0 : _b.integrationURL) ? { integrationURL: data.merge.integrationURL } : {}
           };
           _config = { ..._config, token: stripBearer(session.token) };
+          updateGlobalToken(session.token);
           if (_config.onTokenRefresh) _config.onTokenRefresh(session);
           return true;
         }
@@ -1781,6 +1815,7 @@ function createMitraInstance(initialConfig) {
     try {
       const session = await refreshTokenSilently(_config.authUrl, _config.projectId);
       _config = { ..._config, token: stripBearer(session.token) };
+      updateGlobalToken(session.token);
       if (_config.onTokenRefresh) _config.onTokenRefresh(session);
       return true;
     } catch (e) {
@@ -2089,6 +2124,26 @@ function stripBearer(token) {
   }
   return t;
 }
+function decodeJwtPayload(token) {
+  try {
+    const part = stripBearer(token).split(".")[1];
+    if (!part) return null;
+    const b64 = part.replace(/-/g, "+").replace(/_/g, "/");
+    const padded = b64 + "=".repeat((4 - b64.length % 4) % 4);
+    const json = typeof atob === "function" ? atob(padded) : Buffer.from(padded, "base64").toString("utf8");
+    return JSON.parse(json);
+  } catch (e) {
+    return null;
+  }
+}
+function isTokenExpiring(token, skewMs = 6e4) {
+  const t = globalConfig == null ? void 0 : globalConfig.token;
+  if (!t) return false;
+  const payload = decodeJwtPayload(t);
+  const exp = payload == null ? void 0 : payload.exp;
+  if (typeof exp !== "number") return false;
+  return exp * 1e3 - Date.now() < skewMs;
+}
 function readInjectedEnv() {
   if (typeof window === "undefined") return {};
   const env = window.__mitraEnv;