UI tab backup

electron/main/electron-store/ipcHandlers.ts

@@ -2,16 +2,17 @@ import path from 'path'
 
 import { ipcMain } from 'electron'
 import Store from 'electron-store'
-
-import WindowsManager from '../common/windowManager'
-
 import {
+  Tab,
   EmbeddingModelConfig,
   EmbeddingModelWithLocalPath,
   EmbeddingModelWithRepo,
   StoreKeys,
   StoreSchema,
 } from './storeConfig'
+
+import WindowsManager from '../common/windowManager'
+
 import { initializeAndMaybeMigrateStore } from './storeSchemaMigrator'
 import { ChatHistory } from '@/components/Chat/chatUtils'
 
@@ -114,6 +115,13 @@ export const registerStoreHandlers = (store: Store<StoreSchema>, windowsManager:
 
   ipcMain.handle('get-sb-compact', () => store.get(StoreKeys.IsSBCompact))
 
+  ipcMain.handle('get-editor-flex-center', () => store.get(StoreKeys.EditorFlexCenter))
+
+  ipcMain.handle('set-editor-flex-center', (event, setEditorFlexCenter) => {
+    store.set(StoreKeys.EditorFlexCenter, setEditorFlexCenter)
+    event.sender.send('editor-flex-center-changed', setEditorFlexCenter)
+  })
+
   ipcMain.handle('set-analytics-mode', (event, isAnalytics) => {
     store.set(StoreKeys.Analytics, isAnalytics)
   })
@@ -191,9 +199,68 @@ export const registerStoreHandlers = (store: Store<StoreSchema>, windowsManager:
     const chatHistoriesMap = store.get(StoreKeys.ChatHistories)
     const allChatHistories = chatHistoriesMap[vaultDir] || []
     const filteredChatHistories = allChatHistories.filter((item) => item.id !== chatID)
+
     chatHistoriesMap[vaultDir] = filteredChatHistories.reverse()
     store.set(StoreKeys.ChatHistories, chatHistoriesMap)
   })
+
+  ipcMain.handle('get-current-open-files', () => store.get(StoreKeys.OpenTabs) || [])
+
+  ipcMain.handle('set-current-open-files', (event, action, args) => {
+    const openTabs: Tab[] = store.get(StoreKeys.OpenTabs) || []
+
+    const addTab = ({ tab }: { tab: Tab }) => {
+      if (tab === null) return
+      const existingTab = openTabs.findIndex((item) => item.filePath === tab.filePath)
+
+      /* If tab is already open, do not do anything */
+      if (existingTab !== -1) return
+
+      openTabs.push(tab)
+      store.set(StoreKeys.OpenTabs, openTabs)
+    }
+
+    const removeTab = ({ tabId, idx, newIndex }: { tabId: string; idx: number; newIndex: number }) => {
+      openTabs[idx].lastAccessed = false
+      openTabs[newIndex].lastAccessed = true
+      const updatedTabs = openTabs.filter((tab) => tab.id !== tabId)
+      store.set(StoreKeys.OpenTabs, updatedTabs)
+    }
+
+    const clearAllTabs = () => {
+      store.set(StoreKeys.OpenTabs, [])
+    }
+
+    const updateTab = ({ draggedIndex, targetIndex }: { draggedIndex: number; targetIndex: number }) => {
+      // Swap dragged and target
+      ;[openTabs[draggedIndex], openTabs[targetIndex]] = [openTabs[targetIndex], openTabs[draggedIndex]]
+      store.set(StoreKeys.OpenTabs, openTabs)
+    }
+
+    const selectTab = ({ tabs }: { tabs: Tab[] }) => {
+      store.set(StoreKeys.OpenTabs, tabs)
+    }
+
+    switch (action) {
+      case 'add':
+        addTab(args)
+        break
+      case 'remove':
+        removeTab(args)
+        break
+      case 'update':
+        updateTab(args)
+        break
+      case 'select':
+        selectTab(args)
+        break
+      case 'clear':
+        clearAllTabs()
+        break
+      default:
+        throw new Error('Unsupported action type')
+    }
+  })
 }
 
 export function getDefaultEmbeddingModelConfig(store: Store<StoreSchema>): EmbeddingModelConfig {

electron/main/electron-store/storeConfig.ts

@@ -48,6 +48,15 @@ export type HardwareConfig = {
   useVulkan: boolean
 }
 
+export type Tab = {
+  id: string // Unique ID for the tab, useful for operations
+  filePath: string // Path to the file open in the tab
+  title: string // Title of the tab
+  lastAccessed: boolean
+  // timeOpened: Date; // Timestamp to preserve order
+  // isDirty: boolean; // Flag to indicate unsaved changes
+}
+
 export interface StoreSchema {
   hasUserOpenedAppBefore: boolean
   schemaVersion: number
@@ -72,6 +81,8 @@ export interface StoreSchema {
   isSBCompact: boolean
   DisplayMarkdown: boolean
   spellCheck: string
+  EditorFlexCenter: boolean
+  OpenTabs: Tab[]
 }
 
 export enum StoreKeys {
@@ -91,4 +102,6 @@ export enum StoreKeys {
   IsSBCompact = 'isSBCompact',
   DisplayMarkdown = 'DisplayMarkdown',
   SpellCheck = 'spellCheck',
+  EditorFlexCenter = 'editorFlexCenter',
+  OpenTabs = 'OpenTabs',
 }

electron/main/electron-utils/ipcHandlers.ts

@@ -20,7 +20,7 @@ const electronUtilsHandlers = (
       new MenuItem({
         label: 'New Note',
         click: () => {
-          event.sender.send('add-new-note-listener')
+          event.sender.send('add-new-note-response')
         },
       }),
     )
@@ -29,7 +29,7 @@ const electronUtilsHandlers = (
       new MenuItem({
         label: 'New Directory',
         click: () => {
-          event.sender.send('add-new-directory-listener')
+          event.sender.send('add-new-directory-response')
         },
       }),
     )
@@ -49,7 +49,7 @@ const electronUtilsHandlers = (
         new MenuItem({
           label: 'New Note',
           click: () => {
-            event.sender.send('add-new-note-listener', file.relativePath)
+            event.sender.send('add-new-note-response', file.relativePath)
           },
         }),
       )
@@ -58,7 +58,7 @@ const electronUtilsHandlers = (
         new MenuItem({
           label: 'New Directory',
           click: () => {
-            event.sender.send('add-new-directory-listener', file.path)
+            event.sender.send('add-new-directory-response', file.path)
           },
         }),
       )
@@ -153,6 +153,16 @@ const electronUtilsHandlers = (
   })
 
   ipcMain.handle('get-reor-app-version', async () => app.getVersion())
+
+  // Used on EmptyPage.tsx to create a new file
+  ipcMain.handle('empty-new-note-listener', (event, relativePath) => {
+    event.sender.send('add-new-note-response', relativePath)
+  })
+
+  // Used on EmptyPage.tsx to create a new directory
+  ipcMain.handle('empty-new-directory-listener', (event, relativePath) => {
+    event.sender.send('add-new-directory-response', relativePath)
+  })
 }
 
 export default electronUtilsHandlers

electron/main/llm/ipcHandlers.ts

@@ -2,7 +2,7 @@ import { MessageStreamEvent } from '@anthropic-ai/sdk/resources'
 import { ipcMain, IpcMainInvokeEvent } from 'electron'
 import Store from 'electron-store'
 import { ProgressResponse } from 'ollama'
-import { ChatCompletionChunk } from 'openai/resources/chat/completions'
+import { ChatCompletionChunk, ChatCompletionMessageParam } from 'openai/resources/chat/completions'
 
 import { LLMConfig, StoreKeys, StoreSchema } from '../electron-store/storeConfig'
 
@@ -44,13 +44,18 @@ export const registerLLMSessionHandlers = (store: Store<StoreSchema>) => {
         event.sender.send('anthropicTokenStream', chatHistory.id, chunk)
       }
 
+      const transformedChatHistory: ChatCompletionMessageParam[] = chatHistory.displayableChatHistory.map((message) => {
+        const { messageType, context, visibleContent, ...rest } = message
+        return rest
+      })
+
       switch (llmConfig.type) {
         case LLMType.OpenAI:
           await openAISession.streamingResponse(
             llmName,
             llmConfig,
             isJSONMode,
-            chatHistory.displayableChatHistory,
+            transformedChatHistory,
             handleOpenAIChunk,
             store.get(StoreKeys.LLMGenerationParameters),
           )
@@ -60,7 +65,7 @@ export const registerLLMSessionHandlers = (store: Store<StoreSchema>) => {
             llmName,
             llmConfig,
             isJSONMode,
-            chatHistory.displayableChatHistory,
+            transformedChatHistory,
             handleAnthropicChunk,
             store.get(StoreKeys.LLMGenerationParameters),
           )

electron/preload/index.ts

@@ -6,6 +6,7 @@ import {
   HardwareConfig,
   LLMConfig,
   LLMGenerationParameters,
+  Tab,
 } from 'electron/main/electron-store/storeConfig'
 import {
   AugmentPromptWithFileProps,
@@ -56,6 +57,8 @@ const electronUtils = {
   showFileItemContextMenu: createIPCHandler<(file: FileInfoNode) => Promise<void>>('show-context-menu-file-item'),
   showMenuItemContext: createIPCHandler<() => Promise<void>>('show-context-menu-item'),
   showChatItemContext: createIPCHandler<(chatRow: ChatHistoryMetadata) => Promise<void>>('show-chat-menu-item'),
+  showCreateFileModal: createIPCHandler<(relativePath: string) => Promise<void>>('empty-new-note-listener'),
+  showCreateDirectoryModal: createIPCHandler<(relativePath: string) => Promise<void>>('empty-new-directory-listener'),
 }
 
 const electronStore = {
@@ -85,19 +88,22 @@ const electronStore = {
     createIPCHandler<(params: LLMGenerationParameters) => Promise<void>>('set-llm-generation-params'),
   getAnalyticsMode: createIPCHandler<() => Promise<boolean>>('get-analytics-mode'),
   setAnalyticsMode: createIPCHandler<(isAnalytics: boolean) => Promise<void>>('set-analytics-mode'),
-  getSpellCheckMode: createIPCHandler<() => Promise<string>>('get-spellcheck-mode'),
-  setSpellCheckMode: createIPCHandler<(isSpellCheck: string) => Promise<void>>('set-spellcheck-mode'),
+  getSpellCheckMode: createIPCHandler<() => Promise<boolean>>('get-spellcheck-mode'),
+  setSpellCheckMode: createIPCHandler<(isSpellCheck: boolean) => Promise<void>>('set-spellcheck-mode'),
   getHasUserOpenedAppBefore: createIPCHandler<() => Promise<boolean>>('has-user-opened-app-before'),
   setHasUserOpenedAppBefore: createIPCHandler<() => Promise<void>>('set-user-has-opened-app-before'),
   getAllChatHistories: createIPCHandler<() => Promise<ChatHistory[]>>('get-all-chat-histories'),
   updateChatHistory: createIPCHandler<(chatHistory: ChatHistory) => Promise<void>>('update-chat-history'),
   removeChatHistoryAtID: createIPCHandler<(chatID: string) => Promise<void>>('remove-chat-history-at-id'),
   getChatHistory: createIPCHandler<(chatID: string) => Promise<ChatHistory>>('get-chat-history'),
-
   getSBCompact: createIPCHandler<() => Promise<boolean>>('get-sb-compact'),
   setSBCompact: createIPCHandler<(isSBCompact: boolean) => Promise<void>>('set-sb-compact'),
   getDisplayMarkdown: createIPCHandler<() => Promise<boolean>>('get-display-markdown'),
   setDisplayMarkdown: createIPCHandler<(displayMarkdown: boolean) => Promise<void>>('set-display-markdown'),
+  getEditorFlexCenter: createIPCHandler<() => Promise<boolean>>('get-editor-flex-center'),
+  setEditorFlexCenter: createIPCHandler<(editorFlexCenter: boolean) => Promise<void>>('set-editor-flex-center'),
+  getCurrentOpenFiles: createIPCHandler<() => Promise<Tab[]>>('get-current-open-files'),
+  setCurrentOpenFiles: createIPCHandler<(action: string, args: any) => Promise<void>>('set-current-open-files'),
 }
 
 const fileSystem = {