@athenaintel/react 0.10.35 → 0.10.37

package/README.md

@@ -160,6 +160,11 @@ function AssetView() {
           onCitationClick={({ reference, sourceAssetId }) => {
             console.log('Citation clicked', { reference, sourceAssetId });
           }}
+          onCitationAction={({ action, reference }) => {
+            if (action === 'open-in-spaces') {
+              openAssetInMyApp(reference);
+            }
+          }}
         />
       </div>
     </AthenaProvider>
@@ -167,7 +172,9 @@ function AssetView() {
 }
 ```
 
-When `onCitationClick` is provided, citation navigation is delegated to the host by default. Set `citationBehavior="browser"` to observe the event while preserving Athena's default new-tab navigation, or `citationBehavior="host"` to delegate navigation without providing a callback.
+When `onCitationClick` is provided, citation navigation is delegated to the host by default. Set `citationBehavior="browser"` to observe the event while preserving Athena's default new-tab navigation. Host mode requires `onCitationClick`; without a handler, the SDK safely falls back to browser navigation.
+
+`onCitationAction` handles actions initiated from a citation popover. It currently receives `action: 'open-in-spaces'` when the user clicks the (+) button. Providing the handler delegates that action to the host by default. Set `citationActionBehavior="browser"` to observe the event while preserving Athena's default navigation; requesting host mode without a handler safely falls back to browser behavior.
 
 ## Theming
 

package/dist/assets/AthenaAssetEmbed.d.ts

@@ -1,5 +1,5 @@
 import { type FC, type IframeHTMLAttributes } from "react";
-import { type AthenaAssetCitationBehavior, type AthenaAssetCitationClickEvent } from "./asset-embed-messages";
+import { type AthenaAssetCitationActionEvent, type AthenaAssetCitationBehavior, type AthenaAssetCitationClickEvent } from "./asset-embed-messages";
 export interface AthenaAssetEmbedProps extends Omit<IframeHTMLAttributes<HTMLIFrameElement>, "src" | "onError" | "onLoad"> {
     assetId: string;
     /** Defaults to the native full asset experience. */
@@ -9,6 +9,9 @@ export interface AthenaAssetEmbedProps extends Omit<IframeHTMLAttributes<HTMLIFr
     expiresInSeconds?: number;
     citationBehavior?: AthenaAssetCitationBehavior;
     onCitationClick?: (event: AthenaAssetCitationClickEvent) => void;
+    /** Controls the citation popover's actions, including the (+) Open in Spaces button. */
+    citationActionBehavior?: AthenaAssetCitationBehavior;
+    onCitationAction?: (event: AthenaAssetCitationActionEvent) => void;
     onLoad?: () => void;
     onError?: (error: Error) => void;
     loadingFallback?: React.ReactNode;

package/dist/assets/asset-embed-messages.d.ts

@@ -1,7 +1,9 @@
 export declare const ATHENA_ASSET_EMBED_READY_MESSAGE: "athena:asset-embed-ready";
 export declare const ATHENA_ASSET_EMBED_CONFIG_MESSAGE: "athena:asset-embed-config";
 export declare const ATHENA_ASSET_CITATION_CLICK_MESSAGE: "athena:asset-citation-click";
+export declare const ATHENA_ASSET_CITATION_ACTION_MESSAGE: "athena:asset-citation-action";
 export type AthenaAssetCitationBehavior = "browser" | "host";
+export type AthenaAssetCitationAction = "open-in-spaces";
 export interface AthenaAssetCitationReference {
     id: string;
     anchor?: unknown;
@@ -13,6 +15,12 @@ export interface AthenaAssetCitationClickEvent {
     sourceAssetId?: string;
     reference: AthenaAssetCitationReference;
 }
+export interface AthenaAssetCitationActionEvent {
+    type: typeof ATHENA_ASSET_CITATION_ACTION_MESSAGE;
+    version: 1;
+    action: AthenaAssetCitationAction;
+    reference: AthenaAssetCitationReference;
+}
 export interface AthenaAssetEmbedReadyEvent {
     type: typeof ATHENA_ASSET_EMBED_READY_MESSAGE;
     version: 1;
@@ -22,6 +30,16 @@ export interface AthenaAssetEmbedConfigEvent {
     version: 1;
     token?: string;
     citationBehavior: AthenaAssetCitationBehavior;
+    citationActionBehavior?: AthenaAssetCitationBehavior;
 }
+export declare function resolveAthenaAssetCitationBehavior({ citationBehavior, hasCitationClickHandler, }: {
+    citationBehavior?: AthenaAssetCitationBehavior;
+    hasCitationClickHandler: boolean;
+}): AthenaAssetCitationBehavior;
+export declare function resolveAthenaAssetCitationActionBehavior({ citationActionBehavior, hasCitationActionHandler, }: {
+    citationActionBehavior?: AthenaAssetCitationBehavior;
+    hasCitationActionHandler: boolean;
+}): AthenaAssetCitationBehavior;
 export declare function isAthenaAssetEmbedReadyEvent(value: unknown): value is AthenaAssetEmbedReadyEvent;
 export declare function isAthenaAssetCitationClickEvent(value: unknown): value is AthenaAssetCitationClickEvent;
+export declare function isAthenaAssetCitationActionEvent(value: unknown): value is AthenaAssetCitationActionEvent;

package/dist/assets/use-asset-embed.d.ts

@@ -26,6 +26,11 @@ export declare function resolveAssetEmbedUrl({ embedUrl, appUrl, backendUrl, cur
     backendUrl?: string | null;
     currentOrigin?: string | null;
 }): string;
+export declare function buildSdkAssetEmbedUrl(embedUrl: string): string;
+export declare function getAssetEmbedAuthContext({ token, apiKey }: {
+    token?: string | null;
+    apiKey?: string;
+}): string;
 /**
  * Hook to generate an embed URL for rendering an asset in an iframe.
  * Calls the Athena embed token endpoint.

package/dist/index.cjs

@@ -56254,12 +56254,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,
@@ -56356,7 +56374,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;
     }
@@ -56368,6 +56390,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: ""
 }) {
@@ -56386,7 +56443,7 @@ function useAssetEmbed(assetId, options = {
   const abortRef = React.useRef(null);
   const tokenRef = React.useRef(token);
   tokenRef.current = token;
-  const hasToken = !!token;
+  const authContext = getAssetEmbedAuthContext({ token, apiKey });
   React.useEffect(() => {
     var _a2;
     if (!assetId || !backendUrl) {
@@ -56395,7 +56452,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) {
@@ -56411,10 +56467,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;
     }
@@ -56422,7 +56480,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) => {
@@ -56437,7 +56500,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) => {
@@ -56446,7 +56512,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 = {
@@ -56462,6 +56528,8 @@ const AthenaAssetEmbed = ({
   expiresInSeconds,
   citationBehavior,
   onCitationClick,
+  citationActionBehavior,
+  onCitationAction,
   onLoad,
   onError,
   loadingFallback,
@@ -56492,7 +56560,15 @@ const AthenaAssetEmbed = ({
       return null;
     }
   }, [embedUrl]);
-  const effectiveCitationBehavior = citationBehavior ?? (onCitationClick ? "host" : "browser");
+  const iframeSrc = React.useMemo(() => embedUrl ? buildSdkAssetEmbedUrl(embedUrl) : null, [embedUrl]);
+  const effectiveCitationBehavior = resolveAthenaAssetCitationBehavior({
+    citationBehavior,
+    hasCitationClickHandler: !!onCitationClick
+  });
+  const effectiveCitationActionBehavior = resolveAthenaAssetCitationActionBehavior({
+    citationActionBehavior,
+    hasCitationActionHandler: !!onCitationAction
+  });
   const sendConfig = React.useCallback(() => {
     var _a2, _b;
     if (!targetOrigin) return;
@@ -56501,28 +56577,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]);
   React.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]);
   React.useEffect(() => {
     if (!embedUrl) return;
     sendConfig();
@@ -56559,7 +56639,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),