@arbidocs/react 0.5.62 → 0.5.64

package/dist/{chunk-ER4LFZG3.js → chunk-QGOYIYLI.js}

@@ -6,6 +6,130 @@ import { getUserEncryptionKeypair, sealedBoxDecrypt, sealedBoxEncrypt } from '@a
 import { useSyncExternalStore, useCallback, useMemo, useRef, useEffect } from 'react';
 import { useQuery, useMutation, useQueryClient, useInfiniteQuery, useQueries } from '@tanstack/react-query';
 
+// src/session/thread-tabs.ts
+var threadTabId = (tab) => tab.conversationId ?? "new";
+var MAX_THREAD_TABS = 12;
+var withView = (threads, tabId, view) => threads.map((tab) => tabId !== null && threadTabId(tab) === tabId ? { ...tab, view } : tab);
+function selectThreadTabIn(tabs, tabId, leaving) {
+  const target = tabs.threads.find((tab) => threadTabId(tab) === tabId);
+  if (!target) return { tabs, show: null };
+  const isActive = tabs.activeThread === tabId;
+  if (isActive && leaving.conversationId === target.conversationId) return { tabs, show: null };
+  return {
+    tabs: {
+      ...tabs,
+      // Re-opening the active tab saves nothing: its live composer is the cleared one.
+      threads: isActive ? tabs.threads : withView(tabs.threads, tabs.activeThread, leaving.view),
+      activeThread: tabId,
+      threadScrollRestore: target.view?.scrollTop ?? null
+    },
+    show: { conversationId: target.conversationId, view: target.view }
+  };
+}
+function openThreadTabIn(tabs, conversationId, leaving) {
+  const tabId = conversationId ?? "new";
+  const existing = tabs.threads.find((tab) => threadTabId(tab) === tabId);
+  if (existing) {
+    const threads2 = tabs.threads.map(
+      (tab) => threadTabId(tab) === tabId && tab.view ? { ...tab, view: { ...tab.view, scrollTop: null, cursor: null } } : tab
+    );
+    return { tabId, ...selectThreadTabIn({ ...tabs, threads: threads2 }, tabId, leaving) };
+  }
+  const fresh = { conversationId, view: void 0 };
+  const onlyNewChat = tabs.threads.length === 1 && tabs.threads[0].conversationId === null;
+  if (conversationId && onlyNewChat)
+    return {
+      tabId,
+      tabs: { ...tabs, threads: [{ conversationId }], activeThread: tabId },
+      show: fresh
+    };
+  let threads = [...withView(tabs.threads, tabs.activeThread, leaving.view), { conversationId }];
+  if (threads.length > MAX_THREAD_TABS) {
+    const evict = threads.findIndex(
+      (tab, i) => i < threads.length - 1 && threadTabId(tab) !== tabs.activeThread
+    );
+    if (evict !== -1) threads = threads.filter((_, i) => i !== evict);
+  }
+  return { tabId, tabs: { ...tabs, threads, activeThread: tabId }, show: fresh };
+}
+function closeThreadTabIn(tabs, tabId) {
+  const threads = tabs.threads.filter((tab) => threadTabId(tab) !== tabId);
+  if (threads.length === 0)
+    return { tabs: { ...tabs, threads: [], activeThread: null }, show: "list" };
+  if (tabs.activeThread !== tabId) return { tabs: { ...tabs, threads }, show: null };
+  const index = tabs.threads.findIndex((tab) => threadTabId(tab) === tabId);
+  const next = threads[Math.min(index, threads.length - 1)];
+  return {
+    tabs: { ...tabs, threads, activeThread: threadTabId(next) },
+    show: { conversationId: next.conversationId, view: next.view }
+  };
+}
+function showThreadListIn(tabs, leaving) {
+  return {
+    ...tabs,
+    activeThread: null,
+    threads: tabs.threads.map(
+      (tab) => threadTabId(tab) === tabs.activeThread ? { ...tab, view: { ...leaving.view, scrollTop: null } } : tab.view ? { ...tab, view: { ...tab.view, scrollTop: null } } : tab
+    )
+  };
+}
+function renameThreadTabIn(tabs, fromId, toId) {
+  if (fromId === toId || !tabs.threads.some((tab) => threadTabId(tab) === fromId)) return tabs;
+  const taken = tabs.threads.some((tab) => threadTabId(tab) === toId);
+  return {
+    ...tabs,
+    activeThread: tabs.activeThread === fromId ? toId : tabs.activeThread,
+    threads: taken ? tabs.threads.filter((tab) => threadTabId(tab) !== fromId) : tabs.threads.map(
+      (tab) => threadTabId(tab) === fromId ? { ...tab, conversationId: toId } : tab
+    )
+  };
+}
+function withThreadDrafts(threads, drafts) {
+  let changed = false;
+  const next = threads.map((tab) => {
+    const draft = drafts[threadTabId(tab)];
+    if (!draft || !tab.view || tab.view.draft) return tab;
+    changed = true;
+    return { ...tab, view: { ...tab.view, draft } };
+  });
+  return changed ? next : threads;
+}
+var isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+var stringOrNull = (v) => typeof v === "string" ? v : null;
+var stringOrUndefined = (v) => typeof v === "string" ? v : void 0;
+function parseView(value) {
+  if (!isRecord(value)) return void 0;
+  return {
+    cursor: stringOrNull(value.cursor),
+    draft: "",
+    editingMessageId: stringOrNull(value.editingMessageId),
+    preEditDraft: null,
+    preEditCursor: stringOrNull(value.preEditCursor),
+    chatModel: stringOrUndefined(value.chatModel),
+    chatModelName: stringOrUndefined(value.chatModelName),
+    scrollTop: typeof value.scrollTop === "number" && Number.isFinite(value.scrollTop) ? value.scrollTop : null
+  };
+}
+function parseSavedThreadTabs(threads, activeThread) {
+  const tabs = [];
+  for (const entry of Array.isArray(threads) ? threads : []) {
+    if (!isRecord(entry)) continue;
+    const { conversationId } = entry;
+    if (conversationId !== null && typeof conversationId !== "string") continue;
+    const view = parseView(entry.view);
+    const tab = view ? { conversationId, view } : { conversationId };
+    if (!tabs.some((t) => threadTabId(t) === threadTabId(tab))) tabs.push(tab);
+  }
+  const active = typeof activeThread === "string" ? activeThread : null;
+  return {
+    threads: tabs,
+    activeThread: active && tabs.some((tab) => threadTabId(tab) === active) ? active : null
+  };
+}
+var savedThreadTabs = (threads) => threads.map(
+  (tab) => tab.view ? { ...tab, view: { ...tab.view, draft: "", preEditDraft: null } } : tab
+);
+
 // src/session/layout.ts
 var PANEL_MODES = ["main", "docked", "overlay", "popout"];
 var defaultPinned = () => typeof window === "undefined" || window.innerWidth >= 640;
@@ -17,26 +141,36 @@ function defaultLayoutState() {
       panels: { main: null, docked: null, overlay: null, popout: null, modes: {}, sizes: {} },
       focusedArea: "main"
     },
-    tabs: { dm: [], activeDm: null, dmMessageId: null, projects: [], activeProject: null },
+    tabs: {
+      threads: [],
+      activeThread: null,
+      threadScrollRestore: null,
+      dm: [],
+      activeDm: null,
+      dmMessageId: null,
+      projects: [],
+      activeProject: null
+    },
     views: { tabs: {}, filters: {}, values: {}, transient: {} },
     grid: { states: {}, rowHeights: {}, labelWidths: {}, savedViews: {} }
   };
 }
-var isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
-var recordOf = (value, keep) => isRecord(value) ? Object.fromEntries(Object.entries(value).filter(([, v]) => keep(v))) : {};
+var isRecord2 = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+var recordOf = (value, keep) => isRecord2(value) ? Object.fromEntries(Object.entries(value).filter(([, v]) => keep(v))) : {};
 var isString = (v) => typeof v === "string";
 var isNumber = (v) => typeof v === "number" && Number.isFinite(v);
 var isAny = (v) => v !== void 0;
 var isMode = (v) => PANEL_MODES.includes(v);
-var isSize = (v) => isRecord(v) && isNumber(v.width) && isNumber(v.height);
+var isSize = (v) => isRecord2(v) && isNumber(v.width) && isNumber(v.height);
 var isList = (v) => Array.isArray(v);
-var stringOrNull = (v) => isString(v) ? v : null;
+var stringOrNull2 = (v) => isString(v) ? v : null;
 function parseSavedLayout(saved) {
-  if (!isRecord(saved) || saved.version !== 1) return null;
-  const shell = isRecord(saved.shell) ? saved.shell : {};
-  const panels = isRecord(shell.panels) ? shell.panels : {};
-  const views = isRecord(saved.views) ? saved.views : {};
-  const grid = isRecord(saved.grid) ? saved.grid : {};
+  if (!isRecord2(saved) || saved.version !== 1) return null;
+  const shell = isRecord2(saved.shell) ? saved.shell : {};
+  const panels = isRecord2(shell.panels) ? shell.panels : {};
+  const tabs = isRecord2(saved.tabs) ? saved.tabs : {};
+  const views = isRecord2(saved.views) ? saved.views : {};
+  const grid = isRecord2(saved.grid) ? saved.grid : {};
   const defaults = defaultLayoutState();
   return {
     version: 1,
@@ -45,19 +179,20 @@ function parseSavedLayout(saved) {
       page: isString(shell.page) ? shell.page : defaults.shell.page,
       sidebarPinned: typeof shell.sidebarPinned === "boolean" ? shell.sidebarPinned : defaults.shell.sidebar.pinned,
       panels: {
-        main: stringOrNull(panels.main),
-        docked: stringOrNull(panels.docked),
+        main: stringOrNull2(panels.main),
+        docked: stringOrNull2(panels.docked),
         modes: recordOf(panels.modes, isMode),
         sizes: recordOf(panels.sizes, isSize)
       }
     },
+    tabs: parseSavedThreadTabs(tabs.threads, tabs.activeThread),
     views: {
       tabs: recordOf(views.tabs, isString),
       filters: recordOf(views.filters, isAny),
       values: recordOf(views.values, isAny)
     },
     grid: {
-      states: recordOf(grid.states, isRecord),
+      states: recordOf(grid.states, isRecord2),
       rowHeights: recordOf(grid.rowHeights, isNumber),
       labelWidths: recordOf(grid.labelWidths, isNumber),
       savedViews: recordOf(grid.savedViews, isList)
@@ -65,7 +200,7 @@ function parseSavedLayout(saved) {
   };
 }
 function savedLayoutOf(layout, migrated) {
-  const { shell, views, grid } = layout;
+  const { shell, tabs, views, grid } = layout;
   return {
     version: 1,
     migrated,
@@ -80,6 +215,7 @@ function savedLayoutOf(layout, migrated) {
         sizes: shell.panels.sizes
       }
     },
+    tabs: { threads: savedThreadTabs(tabs.threads), activeThread: tabs.activeThread },
     views: { tabs: views.tabs, filters: views.filters, values: views.values },
     grid
   };
@@ -93,6 +229,7 @@ function restoreLayout(current, saved) {
       sidebar: { ...current.shell.sidebar, pinned: saved.shell.sidebarPinned },
       panels: { ...current.shell.panels, ...saved.shell.panels }
     },
+    tabs: { ...current.tabs, ...saved.tabs },
     views: { ...saved.views, transient: current.views.transient },
     grid: saved.grid
   };
@@ -126,8 +263,11 @@ function panelPlace(panels, panelId) {
 }
 var isPanelOpen = (state, panelId) => panelPlace(state.shell.panels, panelId) !== null;
 var dmTabId = (tab) => tab.recipientId ?? "new";
-function createArbiLayoutStore({ persist } = {}) {
-  let layout = defaultLayoutState();
+function createArbiLayoutStore({
+  persist,
+  initial
+} = {}) {
+  let layout = { ...defaultLayoutState(), ...initial };
   let migrated = [];
   let isHydrated = !persist;
   const listeners = /* @__PURE__ */ new Set();
@@ -136,7 +276,7 @@ function createArbiLayoutStore({ persist } = {}) {
   };
   const update = (next) => {
     if (next === layout) return;
-    const persisted = next.views.tabs !== layout.views.tabs || next.views.filters !== layout.views.filters || next.views.values !== layout.views.values || next.grid !== layout.grid || next.shell.page !== layout.shell.page || next.shell.sidebar.pinned !== layout.shell.sidebar.pinned || next.shell.panels !== layout.shell.panels;
+    const persisted = next.tabs.threads !== layout.tabs.threads || next.tabs.activeThread !== layout.tabs.activeThread || next.views.tabs !== layout.views.tabs || next.views.filters !== layout.views.filters || next.views.values !== layout.views.values || next.grid !== layout.grid || next.shell.page !== layout.shell.page || next.shell.sidebar.pinned !== layout.shell.sidebar.pinned || next.shell.panels !== layout.shell.panels;
     layout = next;
     listeners.forEach((listener) => listener());
     if (persisted) save();
@@ -169,6 +309,41 @@ function createArbiLayoutStore({ persist } = {}) {
     },
     hydrated: () => hydration,
     isHydrated: () => isHydrated,
+    threads: {
+      open: (conversationId, leaving) => {
+        const change = openThreadTabIn(layout.tabs, conversationId, leaving);
+        tabs(change.tabs);
+        return { tabId: change.tabId, show: change.show };
+      },
+      close: (tabId) => {
+        const change = closeThreadTabIn(layout.tabs, tabId);
+        tabs(change.tabs);
+        return change.show;
+      },
+      select: (tabId, leaving) => {
+        const change = selectThreadTabIn(layout.tabs, tabId, leaving);
+        if (change.tabs !== layout.tabs) tabs(change.tabs);
+        return change.show;
+      },
+      showList: (leaving) => tabs(showThreadListIn(layout.tabs, leaving)),
+      rename: (fromId, toId) => {
+        const next = renameThreadTabIn(layout.tabs, fromId, toId);
+        if (next !== layout.tabs) tabs(next);
+      },
+      clear: () => {
+        if (layout.tabs.threads.length > 0 || layout.tabs.activeThread !== null)
+          tabs({ threads: [], activeThread: null });
+      },
+      restoreDrafts: (drafts) => {
+        const threads = withThreadDrafts(layout.tabs.threads, drafts);
+        if (threads !== layout.tabs.threads) tabs({ threads });
+      },
+      consumeScrollRestore: () => {
+        const value = layout.tabs.threadScrollRestore;
+        if (value !== null) tabs({ threadScrollRestore: null });
+        return value;
+      }
+    },
     setPage: (page) => {
       if (layout.shell.page !== page || layout.shell.focusedArea !== "main")
         shell({ page, focusedArea: "main" });
@@ -435,8 +610,6 @@ var agentControl = {
 };
 
 // src/session/store.ts
-var threadTabId = (tab) => tab.conversationId ?? "new";
-var MAX_THREAD_TABS = 12;
 var DEFAULT_DOCUMENT_VIEWER_STATE = {
   tab: "pdf",
   compareId: null,
@@ -469,8 +642,6 @@ var EMPTY_SESSION_STATE = {
   activeCitationKey: null,
   chatModel: void 0,
   chatModelName: void 0,
-  threadTabs: [],
-  activeThreadTabId: null,
   newConversationAssistantMode: "always",
   pendingTools: null,
   autoSendDraft: false,
@@ -500,6 +671,7 @@ function layoutActionsOf(layout) {
   delete actions.hydrated;
   delete actions.isHydrated;
   delete actions.reset;
+  delete actions.threads;
   return actions;
 }
 function withoutNew(record) {
@@ -511,12 +683,18 @@ function createArbiSessionStore(initial = {}, {
   persist,
   workspace = createArbiWorkspaceStore({ workspaceId: initial.workspaceId ?? null })
 } = {}) {
-  const layout = createArbiLayoutStore({ persist });
+  const { shell, tabs, views, grid, ...rest } = initial;
+  const layout = createArbiLayoutStore({
+    persist,
+    initial: Object.fromEntries(
+      Object.entries({ shell, tabs, views, grid }).filter(([, part]) => part !== void 0)
+    )
+  });
   if (initial.workspaceStatus) workspace.setWorkspaceStatus(initial.workspaceStatus);
   let state = {
     ...EMPTY_SESSION_STATE,
+    ...rest,
     ...layout.getLayout(),
-    ...initial,
     ...workspace.getState()
   };
   const listeners = /* @__PURE__ */ new Set();
@@ -568,6 +746,35 @@ function createArbiSessionStore(initial = {}, {
     else delete drafts[key];
     setDrafts(drafts);
   };
+  const composer = () => ({
+    conversationId: state.conversationId,
+    view: {
+      cursor: state.cursor,
+      draft: selectDraft(state),
+      editingMessageId: state.editingMessageId,
+      preEditDraft: state.editingMessageId ? preEdit.draft : null,
+      preEditCursor: state.editingMessageId ? preEdit.cursor : null,
+      chatModel: state.chatModel,
+      chatModelName: state.chatModelName,
+      scrollTop: null
+    }
+  });
+  const show = (next) => {
+    if (next === null) return;
+    if (next === "list") {
+      preEdit = { draft: "", cursor: null };
+      return update({ conversationId: null, cursor: null, editingMessageId: null });
+    }
+    const view = next.view;
+    preEdit = { draft: view?.preEditDraft ?? "", cursor: view?.preEditCursor ?? null };
+    update({
+      conversationId: next.conversationId,
+      cursor: view?.cursor ?? null,
+      editingMessageId: view?.editingMessageId ?? null,
+      chatModel: view?.chatModel,
+      chatModelName: view?.chatModelName
+    });
+  };
   return {
     ...layoutActionsOf(layout),
     layoutHydrated: () => layout.hydrated(),
@@ -598,10 +805,9 @@ function createArbiSessionStore(initial = {}, {
         pinnedDocumentId: null,
         newAgentChat: null,
         newChatDocumentId: null,
-        messageTimers: {},
-        threadTabs: [],
-        activeThreadTabId: null
+        messageTimers: {}
       });
+      layout.threads.clear();
     },
     setConversationId: (conversationId) => {
       if (state.conversationId !== conversationId)
@@ -669,13 +875,9 @@ function createArbiSessionStore(initial = {}, {
         conversationDocuments: move(state.conversationDocuments),
         conversationDocumentAutoAdd: move(state.conversationDocumentAutoAdd),
         conversationFacts: move(state.conversationFacts),
-        conversationId: state.conversationId ?? toId,
-        // The new chat's tab is now the conversation's.
-        threadTabs: state.threadTabs.map(
-          (tab) => threadTabId(tab) === fromId ? { conversationId: toId } : tab
-        ),
-        activeThreadTabId: state.activeThreadTabId === fromId ? toId : state.activeThreadTabId
+        conversationId: state.conversationId ?? toId
       });
+      layout.threads.rename(fromId, toId);
       if (fromId in state.drafts) setDrafts(move(state.drafts));
     },
     setConversationFacts: (conversationId, factIds) => update({ conversationFacts: { ...state.conversationFacts, [conversationId]: factIds } }),
@@ -702,37 +904,17 @@ function createArbiSessionStore(initial = {}, {
       if (state.chatModel !== chatModel) update({ chatModel });
     },
     openThreadTab: (conversationId) => {
-      const id = conversationId ?? "new";
-      const others = state.threadTabs.filter((tab) => threadTabId(tab) !== id);
-      const threadTabs = [...others, { conversationId }].slice(-MAX_THREAD_TABS);
-      update({
-        threadTabs,
-        activeThreadTabId: id,
-        ...conversationId === null ? { newChatDocumentId: null } : {}
-      });
-      if (state.conversationId !== conversationId)
-        update({ conversationId, cursor: null, editingMessageId: null });
-      return id;
-    },
-    closeThreadTab: (tabId) => {
-      const threadTabs = state.threadTabs.filter((tab) => threadTabId(tab) !== tabId);
-      if (threadTabs.length === state.threadTabs.length) return;
-      if (state.activeThreadTabId !== tabId) return update({ threadTabs });
-      const next = threadTabs.at(-1);
-      update({ threadTabs, activeThreadTabId: next ? threadTabId(next) : null });
-      if (next && state.conversationId !== next.conversationId)
-        update({ conversationId: next.conversationId, cursor: null, editingMessageId: null });
+      if (conversationId === null) update({ newAgentChat: null, newChatDocumentId: null });
+      else if (state.newAgentChat) update({ newAgentChat: null });
+      const { tabId, show: next } = layout.threads.open(conversationId, composer());
+      show(next);
+      return tabId;
     },
+    closeThreadTab: (tabId) => show(layout.threads.close(tabId)),
     selectThreadTab: (tabId) => {
-      if (tabId === null) {
-        if (state.activeThreadTabId !== null) update({ activeThreadTabId: null });
-        return;
-      }
-      const tab = state.threadTabs.find((t) => threadTabId(t) === tabId);
-      if (!tab) return;
-      update({ activeThreadTabId: tabId });
-      if (state.conversationId !== tab.conversationId)
-        update({ conversationId: tab.conversationId, cursor: null, editingMessageId: null });
+      if (tabId !== null) return show(layout.threads.select(tabId, composer()));
+      layout.threads.showList(composer());
+      show("list");
     },
     setChatModelName: (chatModelName) => {
       if (state.chatModelName !== chatModelName) update({ chatModelName });
@@ -1482,5 +1664,5 @@ function useUserUsageToday(options) {
 }
 
 export { DEFAULT_DOCUMENT_VIEWER_STATE, EMPTY_SESSION_STATE, MAX_THREAD_TABS, MIN_QUEUED_MESSAGE_VISIBILITY_MS, PROJECT_CREATION_DENIED_REASON, WORKSPACE_LAST_OPENED_KEY, WORKSPACE_STORAGE_KEY, agentControl, browserWorkspaceStorage, clearSealedDrafts, closePanelIn, conversationKey, createArbiLayoutStore, createArbiSessionStore, createArbiWorkspaceStore, createDebouncedWriter, createSealedDraftStorage, createStoragePersist, defaultLayoutState, dmTabId, flushOnPageExit, isPanelOpen, isUsageMetered, layoutActionsOf, nextViewingDocument, openPanelIn, panelPlace, parseSavedLayout, queuedMessages, readWorkspaceLastOpened, restoreLayout, savedLayoutOf, selectDraft, selectIsGenerating, selectViewingDocumentId, threadTabId, useActivity, useAgents, useArbiSession, useArbiSessionState, useArbiSessionStore, useAssistantQuery, useAvailableModels, useBoxRole, useBuiltinCommands, useCanCreateProjects, useConfig, useConfigs, useConversationThreads, useConversations, useCurrentUser, useDeleteAgents, useDocumentVersionList, useFirmPool, useHealth, useMessageDetails, useParsedDocument, useProjectInvoices, useProjectUsage, useProjects, useProjectsUsage, useSessionPageTab, useSessionView, useSignedInSession, useTags, useThumbnails, useTopUpFirmPool, useTransientView, useUserUsageToday, useWorkspaceActivity, useWorkspaces };
-//# sourceMappingURL=chunk-ER4LFZG3.js.map
-//# sourceMappingURL=chunk-ER4LFZG3.js.map
+//# sourceMappingURL=chunk-QGOYIYLI.js.map
+//# sourceMappingURL=chunk-QGOYIYLI.js.map