@athenaintel/react 0.7.3 → 0.8.0

package/dist/index.cjs

@@ -16483,24 +16483,60 @@ function isTrustedOrigin(origin) {
     return false;
   }
 }
-function useParentAuth() {
-  const [authToken, setAuthToken] = React.useState(null);
+const BRIDGE_TIMEOUT_MS = 2e3;
+function useParentBridge() {
+  const isInIframe = typeof window !== "undefined" && window.parent !== window;
+  const [state, setState] = React.useState({
+    token: null,
+    apiUrl: null,
+    backendUrl: null,
+    // If not in an iframe, we're ready immediately (standalone mode)
+    ready: !isInIframe
+  });
   const readySignalSent = React.useRef(false);
+  const configReceived = React.useRef(false);
   React.useEffect(() => {
+    if (!isInIframe) return;
     const handler = (event) => {
       if (!isTrustedOrigin(event.origin)) return;
-      if (event.data && typeof event.data === "object" && event.data.type === "athena-auth" && typeof event.data.token === "string") {
-        setAuthToken(event.data.token);
+      if (!event.data || typeof event.data !== "object") return;
+      if (event.data.type === "athena-config") {
+        configReceived.current = true;
+        setState((prev) => ({
+          ...prev,
+          apiUrl: typeof event.data.apiUrl === "string" ? event.data.apiUrl : prev.apiUrl,
+          backendUrl: typeof event.data.backendUrl === "string" ? event.data.backendUrl : prev.backendUrl,
+          ready: true
+        }));
+      }
+      if (event.data.type === "athena-auth" && typeof event.data.token === "string") {
+        setState((prev) => ({
+          ...prev,
+          token: event.data.token,
+          // If we got a token, we're ready even without config
+          ready: true
+        }));
       }
     };
     window.addEventListener("message", handler);
-    if (!readySignalSent.current && window.parent !== window) {
+    if (!readySignalSent.current) {
       window.parent.postMessage({ type: "athena-auth-ready" }, "*");
       readySignalSent.current = true;
     }
-    return () => window.removeEventListener("message", handler);
-  }, []);
-  return authToken;
+    const timer = setTimeout(() => {
+      if (!configReceived.current) {
+        setState((prev) => ({ ...prev, ready: true }));
+      }
+    }, BRIDGE_TIMEOUT_MS);
+    return () => {
+      window.removeEventListener("message", handler);
+      clearTimeout(timer);
+    };
+  }, [isInIframe]);
+  return state;
+}
+function useParentAuth() {
+  return useParentBridge().token;
 }
 const { fromThreadMessageLike, getAutoStatus } = INTERNAL;
 const joinExternalMessages = (messages) => {
@@ -24627,15 +24663,19 @@ function AthenaProvider({
 }) {
   const frontendToolNames = React.useMemo(() => Object.keys(frontendTools), [frontendTools]);
   const themeStyleVars = React.useMemo(() => theme ? themeToStyleVars(theme) : void 0, [theme]);
-  const parentAuthToken = useParentAuth();
-  const effectiveToken = tokenProp ?? parentAuthToken;
-  const effectiveBackendUrl = backendUrl ?? DEFAULT_BACKEND_URL;
+  const bridge = useParentBridge();
+  const effectiveToken = tokenProp ?? bridge.token;
+  const effectiveApiUrl = apiUrl ?? bridge.apiUrl ?? DEFAULT_API_URL;
+  const effectiveBackendUrl = backendUrl ?? bridge.backendUrl ?? DEFAULT_BACKEND_URL;
+  if (!bridge.ready) {
+    return null;
+  }
   let inner;
   if (enableThreadList) {
     inner = /* @__PURE__ */ jsxRuntime.jsx(
       AthenaWithThreadList,
       {
-        apiUrl,
+        apiUrl: effectiveApiUrl,
         backendUrl: effectiveBackendUrl,
         apiKey,
         token: effectiveToken,
@@ -24654,7 +24694,7 @@ function AthenaProvider({
     inner = /* @__PURE__ */ jsxRuntime.jsx(
       AthenaStandalone,
       {
-        apiUrl,
+        apiUrl: effectiveApiUrl,
         backendUrl: effectiveBackendUrl,
         apiKey,
         token: effectiveToken,
@@ -64405,6 +64445,7 @@ exports.CreateEmailDraftToolUI = CreateEmailDraftToolUI;
 exports.CreateNotebookToolUI = CreateNotebookToolUI;
 exports.CreatePresentationToolUI = CreatePresentationToolUI;
 exports.CreateSheetToolUI = CreateSheetToolUI;
+exports.DEFAULT_API_URL = DEFAULT_API_URL;
 exports.DEFAULT_BACKEND_URL = DEFAULT_BACKEND_URL;
 exports.EmailSearchToolUI = EmailSearchToolUI;
 exports.ExpandableSection = ExpandableSection;
@@ -64452,5 +64493,6 @@ exports.useComposerAttachment = useComposerAttachment;
 exports.useFileUpload = useFileUpload;
 exports.useMentionSuggestions = useMentionSuggestions;
 exports.useParentAuth = useParentAuth;
+exports.useParentBridge = useParentBridge;
 exports.useQuote = useQuote;
 //# sourceMappingURL=index.cjs.map