@athenaintel/react 0.10.35 → 0.10.37

package/dist/index.d.ts

@@ -27,7 +27,7 @@ export { AppendDocumentToolUI } from './tools/append-document-tool-ui';
 export { getAssetInfo, tryParseJson } from './tools/asset-utils';
 export { AthenaAssetEmbed } from './assets/AthenaAssetEmbed';
 export type { AthenaAssetEmbedProps } from './assets/AthenaAssetEmbed';
-export type { AthenaAssetCitationBehavior, AthenaAssetCitationClickEvent, AthenaAssetCitationReference, } from './assets/asset-embed-messages';
+export type { AthenaAssetCitationAction, AthenaAssetCitationActionEvent, AthenaAssetCitationBehavior, AthenaAssetCitationClickEvent, AthenaAssetCitationReference, } from './assets/asset-embed-messages';
 export { AssetPanel } from './assets/AssetPanel';
 export type { AssetPanelProps } from './assets/AssetPanel';
 export { AthenaLayout } from './assets/AthenaLayout';

package/dist/index.js

@@ -56215,12 +56215,30 @@ const AthenaUserMessage = ({
 const ATHENA_ASSET_EMBED_READY_MESSAGE = "athena:asset-embed-ready";
 const ATHENA_ASSET_EMBED_CONFIG_MESSAGE = "athena:asset-embed-config";
 const ATHENA_ASSET_CITATION_CLICK_MESSAGE = "athena:asset-citation-click";
+const ATHENA_ASSET_CITATION_ACTION_MESSAGE = "athena:asset-citation-action";
+function resolveAthenaAssetCitationBehavior({
+  citationBehavior,
+  hasCitationClickHandler
+}) {
+  if (citationBehavior === "host" && !hasCitationClickHandler) return "browser";
+  return citationBehavior ?? (hasCitationClickHandler ? "host" : "browser");
+}
+function resolveAthenaAssetCitationActionBehavior({
+  citationActionBehavior,
+  hasCitationActionHandler
+}) {
+  if (citationActionBehavior === "host" && !hasCitationActionHandler) return "browser";
+  return citationActionBehavior ?? (hasCitationActionHandler ? "host" : "browser");
+}
 function isAthenaAssetEmbedReadyEvent(value) {
   return typeof value === "object" && value !== null && "type" in value && value.type === ATHENA_ASSET_EMBED_READY_MESSAGE && "version" in value && value.version === 1;
 }
 function isAthenaAssetCitationClickEvent(value) {
   return typeof value === "object" && value !== null && "type" in value && value.type === ATHENA_ASSET_CITATION_CLICK_MESSAGE && "version" in value && value.version === 1 && "reference" in value && typeof value.reference === "object" && value.reference !== null && "id" in value.reference && typeof value.reference.id === "string";
 }
+function isAthenaAssetCitationActionEvent(value) {
+  return typeof value === "object" && value !== null && "type" in value && value.type === ATHENA_ASSET_CITATION_ACTION_MESSAGE && "version" in value && value.version === 1 && "action" in value && value.action === "open-in-spaces" && "reference" in value && typeof value.reference === "object" && value.reference !== null && "id" in value.reference && typeof value.reference.id === "string";
+}
 function buildAssetEmbedRequestBody({
   assetId,
   readOnly,
@@ -56317,7 +56335,11 @@ function resolveAssetEmbedUrl({
     if (!resolvedEmbedUrl.pathname.startsWith("/embed/")) {
       return embedUrl;
     }
-    const embedAppUrl = resolveEmbedAppUrl({ appUrl, backendUrl, currentOrigin });
+    const embedAppUrl = resolveEmbedAppUrl({
+      appUrl,
+      backendUrl,
+      currentOrigin
+    });
     if (!embedAppUrl) {
       return embedUrl;
     }
@@ -56329,6 +56351,41 @@ function resolveAssetEmbedUrl({
     return embedUrl;
   }
 }
+function buildSdkAssetEmbedUrl(embedUrl) {
+  try {
+    const url = new URL(embedUrl);
+    url.searchParams.set("athena_sdk_bridge", "1");
+    return url.toString();
+  } catch {
+    return embedUrl;
+  }
+}
+const getJwtPayload = (token) => {
+  try {
+    const payloadSegment = token.split(".")[1];
+    if (!payloadSegment || typeof atob !== "function") return null;
+    const normalized = payloadSegment.replace(/-/g, "+").replace(/_/g, "/");
+    const payload = JSON.parse(
+      atob(normalized.padEnd(normalized.length + (4 - normalized.length % 4) % 4, "="))
+    );
+    return payload && typeof payload === "object" ? payload : null;
+  } catch {
+    return null;
+  }
+};
+function getAssetEmbedAuthContext({ token, apiKey }) {
+  if (!token) return apiKey ? `api-key:${apiKey}` : "anonymous";
+  const payload = getJwtPayload(token);
+  const userId = (payload == null ? void 0 : payload.user_id) ?? (payload == null ? void 0 : payload.sub);
+  if (typeof userId !== "string" || !userId) return `opaque-token:${token}`;
+  return `jwt:${JSON.stringify({
+    issuer: (payload == null ? void 0 : payload.iss) ?? null,
+    userId,
+    activeOrgId: (payload == null ? void 0 : payload.active_org_id) ?? (payload == null ? void 0 : payload.org_id) ?? null,
+    workspaceId: (payload == null ? void 0 : payload.workspace_id) ?? null,
+    orgMemberships: (payload == null ? void 0 : payload.org_id_to_org_member_info) ?? null
+  })}`;
+}
 function useAssetEmbed(assetId, options = {
   backendUrl: ""
 }) {
@@ -56347,7 +56404,7 @@ function useAssetEmbed(assetId, options = {
   const abortRef = useRef(null);
   const tokenRef = useRef(token);
   tokenRef.current = token;
-  const hasToken = !!token;
+  const authContext = getAssetEmbedAuthContext({ token, apiKey });
   useEffect(() => {
     var _a2;
     if (!assetId || !backendUrl) {
@@ -56356,7 +56413,6 @@ function useAssetEmbed(assetId, options = {
       setError(null);
       return;
     }
-    const authContext = hasToken ? "token" : apiKey ?? "anon";
     const cacheKey = `${assetId}:${readOnly}:${displayMode}:${authContext}:${backendUrl}:${appUrl ?? ""}`;
     const cached2 = embedCache.get(cacheKey);
     if (cached2 && cached2.expiresAt > Date.now() / 1e3) {
@@ -56372,10 +56428,12 @@ function useAssetEmbed(assetId, options = {
     abortRef.current = controller;
     setIsLoading(true);
     setError(null);
-    const headers = { "Content-Type": "application/json" };
+    const headers = {
+      "Content-Type": "application/json"
+    };
     const currentToken = tokenRef.current;
     if (currentToken) {
-      headers["Authorization"] = `Bearer ${currentToken}`;
+      headers.Authorization = `Bearer ${currentToken}`;
     } else if (apiKey) {
       headers["X-API-KEY"] = apiKey;
     }
@@ -56383,7 +56441,12 @@ function useAssetEmbed(assetId, options = {
       method: "POST",
       headers,
       body: JSON.stringify(
-        buildAssetEmbedRequestBody({ assetId, readOnly, expiresInSeconds, displayMode })
+        buildAssetEmbedRequestBody({
+          assetId,
+          readOnly,
+          expiresInSeconds,
+          displayMode
+        })
       ),
       signal: controller.signal
     }).then(async (res) => {
@@ -56398,7 +56461,10 @@ function useAssetEmbed(assetId, options = {
         appUrl,
         backendUrl
       });
-      embedCache.set(cacheKey, { url: resolvedEmbedUrl, expiresAt: data.expires_at });
+      embedCache.set(cacheKey, {
+        url: resolvedEmbedUrl,
+        expiresAt: data.expires_at
+      });
       setEmbedUrl(resolvedEmbedUrl);
       setIsLoading(false);
     }).catch((err) => {
@@ -56407,7 +56473,7 @@ function useAssetEmbed(assetId, options = {
       setIsLoading(false);
     });
     return () => controller.abort();
-  }, [assetId, readOnly, expiresInSeconds, displayMode, backendUrl, appUrl, apiKey, hasToken]);
+  }, [assetId, readOnly, expiresInSeconds, displayMode, backendUrl, appUrl, authContext, apiKey]);
   return { embedUrl, isLoading, error: error2 };
 }
 const DEFAULT_STYLE = {
@@ -56423,6 +56489,8 @@ const AthenaAssetEmbed = ({
   expiresInSeconds,
   citationBehavior,
   onCitationClick,
+  citationActionBehavior,
+  onCitationAction,
   onLoad,
   onError,
   loadingFallback,
@@ -56453,7 +56521,15 @@ const AthenaAssetEmbed = ({
       return null;
     }
   }, [embedUrl]);
-  const effectiveCitationBehavior = citationBehavior ?? (onCitationClick ? "host" : "browser");
+  const iframeSrc = useMemo(() => embedUrl ? buildSdkAssetEmbedUrl(embedUrl) : null, [embedUrl]);
+  const effectiveCitationBehavior = resolveAthenaAssetCitationBehavior({
+    citationBehavior,
+    hasCitationClickHandler: !!onCitationClick
+  });
+  const effectiveCitationActionBehavior = resolveAthenaAssetCitationActionBehavior({
+    citationActionBehavior,
+    hasCitationActionHandler: !!onCitationAction
+  });
   const sendConfig = useCallback(() => {
     var _a2, _b;
     if (!targetOrigin) return;
@@ -56462,28 +56538,32 @@ const AthenaAssetEmbed = ({
         type: ATHENA_ASSET_EMBED_CONFIG_MESSAGE,
         version: 1,
         ...athenaConfig.token ? { token: athenaConfig.token } : {},
-        citationBehavior: effectiveCitationBehavior
+        citationBehavior: effectiveCitationBehavior,
+        citationActionBehavior: effectiveCitationActionBehavior
       },
       targetOrigin
     );
-  }, [athenaConfig, effectiveCitationBehavior, targetOrigin]);
+  }, [athenaConfig, effectiveCitationActionBehavior, effectiveCitationBehavior, targetOrigin]);
   useEffect(() => {
     if (!embedUrl || !targetOrigin) return;
     const handleMessage = (event) => {
       var _a2;
-      if (event.origin !== targetOrigin || event.source !== ((_a2 = iframeRef.current) == null ? void 0 : _a2.contentWindow))
-        return;
+      if (event.origin !== targetOrigin || event.source !== ((_a2 = iframeRef.current) == null ? void 0 : _a2.contentWindow)) return;
       if (isAthenaAssetEmbedReadyEvent(event.data)) {
         sendConfig();
         return;
       }
       if (isAthenaAssetCitationClickEvent(event.data)) {
         onCitationClick == null ? void 0 : onCitationClick(event.data);
+        return;
+      }
+      if (isAthenaAssetCitationActionEvent(event.data)) {
+        onCitationAction == null ? void 0 : onCitationAction(event.data);
       }
     };
     window.addEventListener("message", handleMessage);
     return () => window.removeEventListener("message", handleMessage);
-  }, [embedUrl, onCitationClick, sendConfig, targetOrigin]);
+  }, [embedUrl, onCitationAction, onCitationClick, sendConfig, targetOrigin]);
   useEffect(() => {
     if (!embedUrl) return;
     sendConfig();
@@ -56520,7 +56600,7 @@ const AthenaAssetEmbed = ({
     {
       ...iframeProps,
       ref: iframeRef,
-      src: embedUrl,
+      src: iframeSrc ?? void 0,
       title: title ?? `Athena asset ${assetId}`,
       allow,
       className: cn("block h-full w-full", className),