@athenaintel/react 0.9.19 → 0.9.20

package/dist/index.cjs

@@ -5,6 +5,7 @@ var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "sy
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const jsxRuntime = require("react/jsx-runtime");
 const React = require("react");
+const AthenaAuthContext = require("./AthenaAuthContext-B3AwLA5Z.cjs");
 const ReactDOM = require("react-dom");
 function _interopNamespaceDefault(e) {
   const n = Object.create(null, { [Symbol.toStringTag]: { value: "Module" } });
@@ -16583,17 +16584,50 @@ function createAthenaSpacesUrl({
   }
   return url.toString();
 }
-function isTrustedOrigin(origin) {
+function isTrustedOrigin({
+  origin,
+  trustedOrigins
+}) {
   try {
-    const { hostname } = new URL(origin);
+    const normalizedOrigin = new URL(origin).origin;
+    if (trustedOrigins.includes(normalizedOrigin)) {
+      return true;
+    }
+    const { hostname } = new URL(normalizedOrigin);
     return hostname === "athenaintel.com" || hostname.endsWith(".athenaintel.com") || hostname === "localhost";
   } catch {
     return false;
   }
 }
 const BRIDGE_TIMEOUT_MS = 2e3;
-function useParentBridge() {
+const normalizeOrigin = (value) => {
+  try {
+    return new URL(value).origin;
+  } catch {
+    return null;
+  }
+};
+function useParentBridge({
+  trustedOrigins = []
+} = {}) {
   const isInIframe = typeof window !== "undefined" && window.parent !== window;
+  const runtimeTrustedOrigins = React.useMemo(() => {
+    const origins = /* @__PURE__ */ new Set();
+    if (typeof window !== "undefined") {
+      origins.add(window.location.origin);
+      const referrerOrigin = normalizeOrigin(document.referrer);
+      if (referrerOrigin) {
+        origins.add(referrerOrigin);
+      }
+    }
+    for (const trustedOrigin of trustedOrigins) {
+      const normalizedOrigin = normalizeOrigin(trustedOrigin);
+      if (normalizedOrigin) {
+        origins.add(normalizedOrigin);
+      }
+    }
+    return [...origins];
+  }, [trustedOrigins]);
   const [state, setState] = React.useState({
     token: null,
     apiUrl: null,
@@ -16607,7 +16641,9 @@ function useParentBridge() {
   React.useEffect(() => {
     if (!isInIframe) return;
     const handler = (event) => {
-      if (!isTrustedOrigin(event.origin)) return;
+      if (!isTrustedOrigin({ origin: event.origin, trustedOrigins: runtimeTrustedOrigins })) {
+        return;
+      }
       if (!event.data || typeof event.data !== "object") return;
       if (event.data.type === "athena-config") {
         configReceived.current = true;
@@ -16642,7 +16678,7 @@ function useParentBridge() {
       window.removeEventListener("message", handler);
       clearTimeout(timer);
     };
-  }, [isInIframe]);
+  }, [isInIframe, runtimeTrustedOrigins]);
   return state;
 }
 function useParentAuth() {
@@ -24990,8 +25026,13 @@ function AthenaProvider({
   const configuredApiUrl = (config2 == null ? void 0 : config2.apiUrl) ?? apiUrl;
   const configuredBackendUrl = (config2 == null ? void 0 : config2.backendUrl) ?? backendUrl;
   const configuredAppUrl = (config2 == null ? void 0 : config2.appUrl) ?? appUrl;
-  const bridge = useParentBridge();
-  const effectiveToken = configuredToken !== void 0 ? configuredToken : bridge.token;
+  const configuredTrustedParentOrigins = config2 == null ? void 0 : config2.trustedParentOrigins;
+  const bridge = useParentBridge({
+    trustedOrigins: configuredTrustedParentOrigins
+  });
+  const authContext = React.useContext(AthenaAuthContext.AthenaAuthContext);
+  const authProviderToken = (authContext == null ? void 0 : authContext.accessToken) ?? null;
+  const effectiveToken = configuredToken !== void 0 ? configuredToken : authProviderToken ?? bridge.token;
   const effectiveApiUrl = configuredApiUrl ?? bridge.apiUrl ?? environmentUrls.apiUrl;
   const effectiveBackendUrl = configuredBackendUrl ?? bridge.backendUrl ?? environmentUrls.backendUrl;
   const effectiveAppUrl = configuredAppUrl ?? bridge.appUrl ?? deriveAthenaAppUrl({ apiUrl: effectiveApiUrl, backendUrl: effectiveBackendUrl }) ?? environmentUrls.appUrl;
@@ -62374,7 +62415,9 @@ const TOOL_META = {
   create_email_draft: { displayName: "Drafting email", icon: Mail },
   unified_email_create_draft: { displayName: "Drafting email", icon: Mail },
   edit_email_draft: { displayName: "Editing email draft", icon: Mail },
-  unified_edit_email_draft: { displayName: "Editing email draft", icon: Mail },
+  unified_email_edit_draft: { displayName: "Editing email draft", icon: Mail },
+  unified_email_fetch_attachments: { displayName: "Fetching email attachments", icon: Mail },
+  unified_email_forward: { displayName: "Forwarding email", icon: Mail },
   // Documents
   read_full_asset: { displayName: "Reading document", icon: BookOpen },
   create_new_document: { displayName: "Creating document", icon: FileText },