@athenaintel/react 0.7.2 → 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) => {
@@ -24491,70 +24527,6 @@ const themes = {
     radius: "0.625rem"
   }
 };
-const QuoteCtx = React.createContext(null);
-function QuoteProvider({ children }) {
-  const [quote, setQuote] = React.useState(null);
-  const clearQuote = React.useCallback(() => setQuote(null), []);
-  const value = React.useMemo(() => ({ quote, setQuote, clearQuote }), [quote, setQuote, clearQuote]);
-  return /* @__PURE__ */ jsxRuntime.jsx(QuoteCtx.Provider, { value, children });
-}
-function useQuote() {
-  const ctx = React.useContext(QuoteCtx);
-  if (!ctx) {
-    throw new Error("[AthenaSDK] useQuote must be used within <QuoteProvider>");
-  }
-  return ctx;
-}
-function useQuoteFromPostMessage() {
-  const { setQuote } = useQuote();
-  React.useEffect(() => {
-    const handleKeydown = (e) => {
-      var _a2;
-      if ((e.metaKey || e.ctrlKey) && e.key.toLowerCase() === "l") {
-        const sel = window.getSelection();
-        const selection = (_a2 = sel == null ? void 0 : sel.toString()) == null ? void 0 : _a2.trim();
-        if (selection && selection.length > 0) {
-          e.preventDefault();
-          let messageId;
-          let role;
-          let node = sel == null ? void 0 : sel.anchorNode;
-          while (node && node !== document.body) {
-            if (node instanceof HTMLElement) {
-              const dataRole = node.getAttribute("data-role");
-              if (dataRole === "assistant" || dataRole === "user") {
-                role = dataRole;
-                messageId = node.getAttribute("data-message-id") ?? void 0;
-                break;
-              }
-            }
-            node = node.parentElement;
-          }
-          setQuote({ text: selection, sourceRole: role, sourceMessageId: messageId });
-        }
-      }
-    };
-    const handleMessage = (event) => {
-      const data = event.data;
-      if (data && typeof data === "object" && data.type === "athena-add-to-composer" && typeof data.text === "string" && data.text.trim().length > 0) {
-        setQuote({
-          text: data.text,
-          sourceTitle: data.sourceTitle,
-          sourceAssetId: data.sourceAssetId
-        });
-      }
-    };
-    document.addEventListener("keydown", handleKeydown, true);
-    window.addEventListener("message", handleMessage);
-    return () => {
-      document.removeEventListener("keydown", handleKeydown, true);
-      window.removeEventListener("message", handleMessage);
-    };
-  }, [setQuote]);
-}
-function QuotePostMessageBridge() {
-  useQuoteFromPostMessage();
-  return null;
-}
 function AthenaStandalone({
   children,
   apiUrl,
@@ -24591,10 +24563,7 @@ function AthenaStandalone({
     () => ({ backendUrl, apiKey, token }),
     [backendUrl, apiKey, token]
   );
-  return /* @__PURE__ */ jsxRuntime.jsx(AssistantRuntimeProvider, { aui, runtime, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsxRuntime.jsxs(QuoteProvider, { children: [
-    /* @__PURE__ */ jsxRuntime.jsx(QuotePostMessageBridge, {}),
-    /* @__PURE__ */ jsxRuntime.jsx(TooltipProvider, { children })
-  ] }) }) });
+  return /* @__PURE__ */ jsxRuntime.jsx(AssistantRuntimeProvider, { aui, runtime, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsxRuntime.jsx(TooltipProvider, { children }) }) });
 }
 function useAthenaRuntimeHook(config2) {
   const remoteId = useAthenaThreadId();
@@ -24673,10 +24642,7 @@ function AthenaWithThreadList({
     () => ({ backendUrl, apiKey, token }),
     [backendUrl, apiKey, token]
   );
-  return /* @__PURE__ */ jsxRuntime.jsx(AssistantRuntimeProvider, { aui, runtime, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsxRuntime.jsxs(QuoteProvider, { children: [
-    /* @__PURE__ */ jsxRuntime.jsx(QuotePostMessageBridge, {}),
-    /* @__PURE__ */ jsxRuntime.jsx(TooltipProvider, { children })
-  ] }) }) });
+  return /* @__PURE__ */ jsxRuntime.jsx(AssistantRuntimeProvider, { aui, runtime, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsxRuntime.jsx(TooltipProvider, { children }) }) });
 }
 function AthenaProvider({
   children,
@@ -24697,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,
@@ -24724,7 +24694,7 @@ function AthenaProvider({
     inner = /* @__PURE__ */ jsxRuntime.jsx(
       AthenaStandalone,
       {
-        apiUrl,
+        apiUrl: effectiveApiUrl,
         backendUrl: effectiveBackendUrl,
         apiKey,
         token: effectiveToken,
@@ -60512,7 +60482,7 @@ function AttachmentProvider({ children }) {
   const clearAttachments = React.useCallback(() => setAttachments([]), []);
   const value = React.useMemo(
     () => ({ attachments, addAttachments, removeAttachment, clearAttachments, isUploading, setIsUploading }),
-    [attachments, addAttachments, removeAttachment, clearAttachments, isUploading]
+    [attachments, addAttachments, removeAttachment, clearAttachments, isUploading, setIsUploading]
   );
   return /* @__PURE__ */ jsxRuntime.jsx(AttachmentCtx.Provider, { value, children });
 }
@@ -60527,6 +60497,66 @@ function buildAttachmentMarkdown(attachments) {
   if (attachments.length === 0) return "";
   return attachments.map((a) => `[@${a.title}](https://app.athenaintel.com/dashboard/spaces?asset_id=${a.id})`).join("\n");
 }
+const QuoteCtx = React.createContext(null);
+function QuoteProvider({ children }) {
+  const [quote, setQuote] = React.useState(null);
+  const clearQuote = React.useCallback(() => setQuote(null), []);
+  const value = React.useMemo(() => ({ quote, setQuote, clearQuote }), [quote, setQuote, clearQuote]);
+  return /* @__PURE__ */ jsxRuntime.jsx(QuoteCtx.Provider, { value, children });
+}
+function useQuote() {
+  const ctx = React.useContext(QuoteCtx);
+  if (!ctx) {
+    throw new Error("[AthenaSDK] useQuote must be used within <QuoteProvider>");
+  }
+  return ctx;
+}
+function useQuoteFromPostMessage() {
+  const { setQuote } = useQuote();
+  React.useEffect(() => {
+    const handleKeydown = (e) => {
+      var _a2;
+      if ((e.metaKey || e.ctrlKey) && e.key.toLowerCase() === "l") {
+        const sel = window.getSelection();
+        const selection = (_a2 = sel == null ? void 0 : sel.toString()) == null ? void 0 : _a2.trim();
+        if (selection && selection.length > 0) {
+          e.preventDefault();
+          let messageId;
+          let role;
+          let node = sel == null ? void 0 : sel.anchorNode;
+          while (node && node !== document.body) {
+            if (node instanceof HTMLElement) {
+              const dataRole = node.getAttribute("data-role");
+              if (dataRole === "assistant" || dataRole === "user") {
+                role = dataRole;
+                messageId = node.getAttribute("data-message-id") ?? void 0;
+                break;
+              }
+            }
+            node = node.parentElement;
+          }
+          setQuote({ text: selection, sourceRole: role, sourceMessageId: messageId });
+        }
+      }
+    };
+    const handleMessage = (event) => {
+      const data = event.data;
+      if (data && typeof data === "object" && data.type === "athena-add-to-composer" && typeof data.text === "string" && data.text.trim().length > 0) {
+        setQuote({
+          text: data.text,
+          sourceTitle: data.sourceTitle,
+          sourceAssetId: data.sourceAssetId
+        });
+      }
+    };
+    document.addEventListener("keydown", handleKeydown, true);
+    window.addEventListener("message", handleMessage);
+    return () => {
+      document.removeEventListener("keydown", handleKeydown, true);
+      window.removeEventListener("message", handleMessage);
+    };
+  }, [setQuote]);
+}
 function buildComposedMessage(opts) {
   var _a2;
   const parts = [];
@@ -60579,15 +60609,17 @@ const TiptapComposer = ({ tools = [] }) => {
           role: "user",
           content: [{ type: "text", text: fullMessage }]
         });
+        clearAttachments();
+        clearQuote();
       }
-      clearAttachments();
-      clearQuote();
     } else {
       composerRuntime.setText(markdown);
       composerRuntime.send();
     }
     editor2.commands.clearContent();
   }, [aui, composerRuntime, clearAttachments, clearQuote]);
+  const handleSubmitRef = React.useRef(handleSubmit);
+  handleSubmitRef.current = handleSubmit;
   const editor = useEditor({
     immediatelyRender: true,
     extensions: [
@@ -60618,7 +60650,7 @@ const TiptapComposer = ({ tools = [] }) => {
         store: mentionStore
       }),
       ComposerKeybinds.configure({
-        onSubmit: () => handleSubmit()
+        onSubmit: () => handleSubmitRef.current()
       })
     ],
     editorProps: {
@@ -63598,8 +63630,8 @@ const ComposerDropZone = ({
   onUpload
 }) => {
   const [isDragOver, setIsDragOver] = React.useState(false);
-  const { upload, isUploading } = useFileUpload();
-  const { addAttachments, setIsUploading } = useAttachments();
+  const { upload } = useFileUpload();
+  const { addAttachments, setIsUploading, isUploading } = useAttachments();
   const handleDragOver = React.useCallback((e) => {
     e.preventDefault();
     e.stopPropagation();
@@ -63652,6 +63684,10 @@ const ComposerDropZone = ({
   );
 };
 const EMPTY_MENTION_TOOLS = [];
+function QuotePostMessageBridge() {
+  useQuoteFromPostMessage();
+  return null;
+}
 const DEFAULT_SUGGESTIONS = [
   {
     icon: Search,
@@ -63750,12 +63786,15 @@ const AthenaChat = ({
               ),
               /* @__PURE__ */ jsxRuntime.jsxs(ThreadPrimitiveViewportFooter, { className: "aui-thread-viewport-footer sticky bottom-0 mx-auto mt-auto flex w-full max-w-(--thread-max-width) flex-col gap-4 overflow-visible rounded-t-3xl bg-background pb-4 md:pb-6", children: [
                 /* @__PURE__ */ jsxRuntime.jsx(ThreadScrollToBottom, {}),
-                /* @__PURE__ */ jsxRuntime.jsx(AttachmentProvider, { children: /* @__PURE__ */ jsxRuntime.jsx(ComposerEditorProvider, { children: /* @__PURE__ */ jsxRuntime.jsx(ComposerDropZone, { children: /* @__PURE__ */ jsxRuntime.jsxs(ComposerPrimitiveRoot, { className: "aui-composer-root relative flex w-full flex-col rounded-2xl border border-input bg-background px-1 outline-none transition-shadow focus-within:border-ring focus-within:ring-2 focus-within:ring-ring/20", children: [
-                  /* @__PURE__ */ jsxRuntime.jsx(ComposerQuotePreview, {}),
-                  /* @__PURE__ */ jsxRuntime.jsx(ComposerAttachmentPreview, {}),
-                  /* @__PURE__ */ jsxRuntime.jsx(TiptapComposer, { tools }),
-                  /* @__PURE__ */ jsxRuntime.jsx(ComposerAction, {})
-                ] }) }) }) })
+                /* @__PURE__ */ jsxRuntime.jsx(AttachmentProvider, { children: /* @__PURE__ */ jsxRuntime.jsx(ComposerEditorProvider, { children: /* @__PURE__ */ jsxRuntime.jsxs(QuoteProvider, { children: [
+                  /* @__PURE__ */ jsxRuntime.jsx(QuotePostMessageBridge, {}),
+                  /* @__PURE__ */ jsxRuntime.jsx(ComposerDropZone, { children: /* @__PURE__ */ jsxRuntime.jsxs(ComposerPrimitiveRoot, { className: "aui-composer-root relative flex w-full flex-col rounded-2xl border border-input bg-background px-1 pt-2 outline-none transition-shadow focus-within:border-ring focus-within:ring-2 focus-within:ring-ring/20", children: [
+                    /* @__PURE__ */ jsxRuntime.jsx(ComposerQuotePreview, {}),
+                    /* @__PURE__ */ jsxRuntime.jsx(ComposerAttachmentPreview, {}),
+                    /* @__PURE__ */ jsxRuntime.jsx(TiptapComposer, { tools }),
+                    /* @__PURE__ */ jsxRuntime.jsx(ComposerAction, {})
+                  ] }) })
+                ] }) }) })
               ] })
             ]
           }
@@ -63812,6 +63851,7 @@ const ComposerSendWithQuote = () => {
     var _a2;
     if (isUploading) return;
     const editor = editorRef.current;
+    if (!editor && !quote && attachments.length === 0) return;
     const userText = ((_a2 = editor == null ? void 0 : editor.getMarkdown()) == null ? void 0 : _a2.trim()) ?? "";
     const fullMessage = buildComposedMessage({
       attachments,
@@ -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