@athenaintel/react 0.9.14 → 0.9.15

package/dist/index.d.ts

@@ -7,6 +7,7 @@ import { EmptyMessagePartComponent } from '@assistant-ui/react';
 import { FC } from 'react';
 import { ForwardRefExoticComponent } from 'react';
 import { JSX } from 'react/jsx-runtime';
+import { PersistOptions } from 'zustand/middleware';
 import * as React_2 from 'react';
 import { ReactNode } from 'react';
 import { ReasoningMessagePartComponent } from '@assistant-ui/react';
@@ -1025,7 +1026,19 @@ export declare interface UseAssetEmbedResult {
     error: string | null;
 }
 
-export declare const useAssetPanelStore: UseBoundStore<StoreApi<AssetPanelState>>;
+export declare const useAssetPanelStore: UseBoundStore<Omit<StoreApi<AssetPanelState>, "setState" | "persist"> & {
+setState(partial: AssetPanelState | Partial<AssetPanelState> | ((state: AssetPanelState) => AssetPanelState | Partial<AssetPanelState>), replace?: false | undefined): unknown;
+setState(state: AssetPanelState | ((state: AssetPanelState) => AssetPanelState), replace: true): unknown;
+persist: {
+setOptions: (options: Partial<PersistOptions<AssetPanelState, unknown, unknown>>) => void;
+clearStorage: () => void;
+rehydrate: () => Promise<void> | void;
+hasHydrated: () => boolean;
+onHydrate: (fn: (state: AssetPanelState) => void) => () => void;
+onFinishHydration: (fn: (state: AssetPanelState) => void) => () => void;
+getOptions: () => Partial<PersistOptions<AssetPanelState, unknown, unknown>>;
+};
+}>;
 
 export declare const useAthenaCitationLinkHandler: () => AthenaCitationLinkHandler;
 

package/dist/index.js

@@ -16483,7 +16483,7 @@ const createUrlWithFallback = (url, fallbackUrl) => {
     return new URL(`${normalizeBaseUrl(fallbackUrl)}/`);
   }
 };
-const getHostname = (value) => {
+const getHostname$1 = (value) => {
   if (!value) return null;
   try {
     return new URL(value).hostname;
@@ -16524,7 +16524,7 @@ function deriveAthenaAppUrl({
       sourcePort: "8000"
     });
     if (workspaceAppUrl) return workspaceAppUrl;
-    const backendHostname = getHostname(backendUrl);
+    const backendHostname = getHostname$1(backendUrl);
     if (backendHostname === "api.athenaintel.com") {
       return ATHENA_ENVIRONMENT_URLS.production.appUrl;
     }
@@ -16538,7 +16538,7 @@ function deriveAthenaAppUrl({
       sourcePort: "8787"
     });
     if (workspaceAppUrl) return workspaceAppUrl;
-    const apiHostname = getHostname(apiUrl);
+    const apiHostname = getHostname$1(apiUrl);
     if (apiHostname === "sync.athenaintel.com") {
       return ATHENA_ENVIRONMENT_URLS.production.appUrl;
     }
@@ -58643,57 +58643,276 @@ const MentionExtension = Node3.create({
     };
   }
 });
-const useAssetPanelStore = create((set2, get2) => ({
-  isOpen: false,
-  tabs: [],
-  activeTabId: null,
-  viewMode: "tabs",
-  isFullscreen: false,
-  assetPanelHostCount: 0,
-  autoOpenGeneration: 0,
-  autoOpenedAssets: /* @__PURE__ */ new Map(),
-  registerAssetPanelHost: () => set2((state) => ({ assetPanelHostCount: state.assetPanelHostCount + 1 })),
-  unregisterAssetPanelHost: () => set2((state) => ({ assetPanelHostCount: Math.max(0, state.assetPanelHostCount - 1) })),
-  openAsset: (assetId, meta) => set2((s) => {
-    const existing = s.tabs.find((t) => t.id === assetId);
-    if (existing) {
-      const tabs = meta ? s.tabs.map(
-        (t) => t.id === assetId ? { ...t, name: meta.name ?? t.name, type: meta.type ?? t.type } : t
-      ) : s.tabs;
-      return { isOpen: true, tabs, activeTabId: assetId };
+function createJSONStorage(getStorage, options) {
+  let storage;
+  try {
+    storage = getStorage();
+  } catch (e) {
+    return;
+  }
+  const persistStorage = {
+    getItem: (name) => {
+      var _a2;
+      const parse2 = (str2) => {
+        if (str2 === null) {
+          return null;
+        }
+        return JSON.parse(str2, void 0);
+      };
+      const str = (_a2 = storage.getItem(name)) != null ? _a2 : null;
+      if (str instanceof Promise) {
+        return str.then(parse2);
+      }
+      return parse2(str);
+    },
+    setItem: (name, newValue) => storage.setItem(name, JSON.stringify(newValue, void 0)),
+    removeItem: (name) => storage.removeItem(name)
+  };
+  return persistStorage;
+}
+const toThenable = (fn) => (input) => {
+  try {
+    const result = fn(input);
+    if (result instanceof Promise) {
+      return result;
     }
-    const newTab = {
-      id: assetId,
-      name: (meta == null ? void 0 : meta.name) ?? null,
-      type: (meta == null ? void 0 : meta.type) ?? "unknown"
+    return {
+      then(onFulfilled) {
+        return toThenable(onFulfilled)(result);
+      },
+      catch(_onRejected) {
+        return this;
+      }
     };
-    return { isOpen: true, tabs: [...s.tabs, newTab], activeTabId: assetId };
-  }),
-  closeTab: (assetId) => set2((s) => {
-    var _a2;
-    const tabs = s.tabs.filter((t) => t.id !== assetId);
-    if (tabs.length === 0) {
-      return { isOpen: false, tabs: [], activeTabId: null, isFullscreen: false };
+  } catch (e) {
+    return {
+      then(_onFulfilled) {
+        return this;
+      },
+      catch(onRejected) {
+        return toThenable(onRejected)(e);
+      }
+    };
+  }
+};
+const persistImpl = (config2, baseOptions) => (set2, get2, api) => {
+  let options = {
+    storage: createJSONStorage(() => window.localStorage),
+    partialize: (state) => state,
+    version: 0,
+    merge: (persistedState, currentState) => ({
+      ...currentState,
+      ...persistedState
+    }),
+    ...baseOptions
+  };
+  let hasHydrated = false;
+  let hydrationVersion = 0;
+  const hydrationListeners = /* @__PURE__ */ new Set();
+  const finishHydrationListeners = /* @__PURE__ */ new Set();
+  let storage = options.storage;
+  if (!storage) {
+    return config2(
+      (...args) => {
+        console.warn(
+          `[zustand persist middleware] Unable to update item '${options.name}', the given storage is currently unavailable.`
+        );
+        set2(...args);
+      },
+      get2,
+      api
+    );
+  }
+  const setItem = () => {
+    const state = options.partialize({ ...get2() });
+    return storage.setItem(options.name, {
+      state,
+      version: options.version
+    });
+  };
+  const savedSetState = api.setState;
+  api.setState = (state, replace2) => {
+    savedSetState(state, replace2);
+    return setItem();
+  };
+  const configResult = config2(
+    (...args) => {
+      set2(...args);
+      return setItem();
+    },
+    get2,
+    api
+  );
+  api.getInitialState = () => configResult;
+  let stateFromStorage;
+  const hydrate = () => {
+    var _a2, _b;
+    if (!storage) return;
+    const currentVersion = ++hydrationVersion;
+    hasHydrated = false;
+    hydrationListeners.forEach((cb) => {
+      var _a22;
+      return cb((_a22 = get2()) != null ? _a22 : configResult);
+    });
+    const postRehydrationCallback = ((_b = options.onRehydrateStorage) == null ? void 0 : _b.call(options, (_a2 = get2()) != null ? _a2 : configResult)) || void 0;
+    return toThenable(storage.getItem.bind(storage))(options.name).then((deserializedStorageValue) => {
+      if (deserializedStorageValue) {
+        if (typeof deserializedStorageValue.version === "number" && deserializedStorageValue.version !== options.version) {
+          if (options.migrate) {
+            const migration = options.migrate(
+              deserializedStorageValue.state,
+              deserializedStorageValue.version
+            );
+            if (migration instanceof Promise) {
+              return migration.then((result) => [true, result]);
+            }
+            return [true, migration];
+          }
+          console.error(
+            `State loaded from storage couldn't be migrated since no migrate function was provided`
+          );
+        } else {
+          return [false, deserializedStorageValue.state];
+        }
+      }
+      return [false, void 0];
+    }).then((migrationResult) => {
+      var _a22;
+      if (currentVersion !== hydrationVersion) {
+        return;
+      }
+      const [migrated, migratedState] = migrationResult;
+      stateFromStorage = options.merge(
+        migratedState,
+        (_a22 = get2()) != null ? _a22 : configResult
+      );
+      set2(stateFromStorage, true);
+      if (migrated) {
+        return setItem();
+      }
+    }).then(() => {
+      if (currentVersion !== hydrationVersion) {
+        return;
+      }
+      postRehydrationCallback == null ? void 0 : postRehydrationCallback(stateFromStorage, void 0);
+      stateFromStorage = get2();
+      hasHydrated = true;
+      finishHydrationListeners.forEach((cb) => cb(stateFromStorage));
+    }).catch((e) => {
+      if (currentVersion !== hydrationVersion) {
+        return;
+      }
+      postRehydrationCallback == null ? void 0 : postRehydrationCallback(void 0, e);
+    });
+  };
+  api.persist = {
+    setOptions: (newOptions) => {
+      options = {
+        ...options,
+        ...newOptions
+      };
+      if (newOptions.storage) {
+        storage = newOptions.storage;
+      }
+    },
+    clearStorage: () => {
+      storage == null ? void 0 : storage.removeItem(options.name);
+    },
+    getOptions: () => options,
+    rehydrate: () => hydrate(),
+    hasHydrated: () => hasHydrated,
+    onHydrate: (cb) => {
+      hydrationListeners.add(cb);
+      return () => {
+        hydrationListeners.delete(cb);
+      };
+    },
+    onFinishHydration: (cb) => {
+      finishHydrationListeners.add(cb);
+      return () => {
+        finishHydrationListeners.delete(cb);
+      };
     }
-    const activeTabId = s.activeTabId === assetId ? ((_a2 = tabs[Math.min(s.tabs.findIndex((t) => t.id === assetId), tabs.length - 1)]) == null ? void 0 : _a2.id) ?? null : s.activeTabId;
-    return { tabs, activeTabId };
-  }),
-  setActiveTab: (assetId) => set2({ activeTabId: assetId, viewMode: "tabs" }),
-  setViewMode: (mode) => set2({ viewMode: mode }),
-  closePanel: () => set2({ isOpen: false, tabs: [], activeTabId: null, viewMode: "tabs", isFullscreen: false }),
-  toggleFullscreen: () => set2((s) => ({ isFullscreen: !s.isFullscreen })),
-  resetAutoOpen: () => set2((s) => ({ autoOpenGeneration: s.autoOpenGeneration + 1 })),
-  markAutoOpened: (assetId) => {
-    const state = get2();
-    if (state.autoOpenedAssets.get(assetId) === state.autoOpenGeneration) {
-      return false;
+  };
+  if (!options.skipHydration) {
+    hydrate();
+  }
+  return stateFromStorage || configResult;
+};
+const persist = persistImpl;
+const useAssetPanelStore = create()(
+  persist(
+    (set2, get2) => ({
+      isOpen: false,
+      tabs: [],
+      activeTabId: null,
+      viewMode: "tabs",
+      isFullscreen: false,
+      assetPanelHostCount: 0,
+      autoOpenGeneration: 0,
+      autoOpenedAssets: /* @__PURE__ */ new Map(),
+      registerAssetPanelHost: () => set2((state) => ({ assetPanelHostCount: state.assetPanelHostCount + 1 })),
+      unregisterAssetPanelHost: () => set2((state) => ({ assetPanelHostCount: Math.max(0, state.assetPanelHostCount - 1) })),
+      openAsset: (assetId, meta) => set2((s) => {
+        const existing = s.tabs.find((t) => t.id === assetId);
+        if (existing) {
+          const tabs = meta ? s.tabs.map(
+            (t) => t.id === assetId ? { ...t, name: meta.name ?? t.name, type: meta.type ?? t.type } : t
+          ) : s.tabs;
+          return { isOpen: true, tabs, activeTabId: assetId };
+        }
+        const newTab = {
+          id: assetId,
+          name: (meta == null ? void 0 : meta.name) ?? null,
+          type: (meta == null ? void 0 : meta.type) ?? "unknown"
+        };
+        return { isOpen: true, tabs: [...s.tabs, newTab], activeTabId: assetId };
+      }),
+      closeTab: (assetId) => set2((s) => {
+        var _a2;
+        const tabs = s.tabs.filter((t) => t.id !== assetId);
+        if (tabs.length === 0) {
+          return { isOpen: false, tabs: [], activeTabId: null, isFullscreen: false };
+        }
+        const activeTabId = s.activeTabId === assetId ? ((_a2 = tabs[Math.min(s.tabs.findIndex((t) => t.id === assetId), tabs.length - 1)]) == null ? void 0 : _a2.id) ?? null : s.activeTabId;
+        return { tabs, activeTabId };
+      }),
+      setActiveTab: (assetId) => set2({ activeTabId: assetId, viewMode: "tabs" }),
+      setViewMode: (mode) => set2({ viewMode: mode }),
+      closePanel: () => set2({ isOpen: false, tabs: [], activeTabId: null, viewMode: "tabs", isFullscreen: false }),
+      toggleFullscreen: () => set2((s) => ({ isFullscreen: !s.isFullscreen })),
+      resetAutoOpen: () => set2((s) => ({ autoOpenGeneration: s.autoOpenGeneration + 1 })),
+      markAutoOpened: (assetId) => {
+        const state = get2();
+        if (state.autoOpenedAssets.get(assetId) === state.autoOpenGeneration) {
+          return false;
+        }
+        state.autoOpenedAssets.set(assetId, state.autoOpenGeneration);
+        return true;
+      }
+    }),
+    {
+      name: "athena-react-asset-panel",
+      partialize: ({
+        isOpen,
+        tabs,
+        activeTabId,
+        viewMode,
+        isFullscreen
+      }) => ({
+        isOpen,
+        tabs,
+        activeTabId,
+        viewMode,
+        isFullscreen
+      }),
+      storage: createJSONStorage(() => sessionStorage)
     }
-    state.autoOpenedAssets.set(assetId, state.autoOpenGeneration);
-    return true;
-  }
-}));
+  )
+);
 const ATHENA_APP_HOSTNAMES = /* @__PURE__ */ new Set(["app.athenaintel.com", "staging-app.athenaintel.com"]);
 const ATHENA_PREVIEW_HOSTNAME_SUFFIX = ".previews.athenaintel.com";
+const ATHENA_WORKSPACE_HOSTNAME_SUFFIXES = [".app.athenaintel.com", ".staging-app.athenaintel.com"];
 const ASSET_TYPE_ALIASES = {
   document: "document",
   notebook: "notebook",
@@ -58723,6 +58942,20 @@ const getOrigin = (value) => {
     return null;
   }
 };
+const getHostname = (value) => {
+  if (!value) {
+    return null;
+  }
+  try {
+    return new URL(value).hostname.toLowerCase();
+  } catch {
+    return null;
+  }
+};
+const matchesHostnameOrSubdomain = ({
+  hostname,
+  candidateHostname
+}) => hostname === candidateHostname || hostname.endsWith(`.${candidateHostname}`);
 const getCurrentOrigin = () => {
   if (typeof window === "undefined") {
     return null;
@@ -58745,7 +58978,11 @@ const isAthenaAppUrl = ({
   appUrl
 }) => {
   const hostname = url.hostname.toLowerCase();
-  if (ATHENA_APP_HOSTNAMES.has(hostname) || hostname.endsWith(ATHENA_PREVIEW_HOSTNAME_SUFFIX)) {
+  if (ATHENA_APP_HOSTNAMES.has(hostname) || hostname.endsWith(ATHENA_PREVIEW_HOSTNAME_SUFFIX) || ATHENA_WORKSPACE_HOSTNAME_SUFFIXES.some((suffix) => hostname.endsWith(suffix))) {
+    return true;
+  }
+  const configuredHostname = getHostname(appUrl);
+  if (configuredHostname && matchesHostnameOrSubdomain({ hostname, candidateHostname: configuredHostname })) {
     return true;
   }
   const configuredOrigin = getOrigin(appUrl);