Extending blocknote

electron/main/filesystem/filesystem.ts

@@ -112,19 +112,26 @@ export function GetFilesInfoListForListOfPaths(paths: string[]): FileInfo[] {
   return fileInfoListWithoutDuplicates
 }
 
-export function createFileRecursive(filePath: string, content: string, charset?: BufferEncoding): void {
+export function createFileRecursive(filePath: string, content: string, charset?: BufferEncoding): FileInfo | null {
   const dirname = path.dirname(filePath)
 
   if (!fs.existsSync(dirname)) {
     fs.mkdirSync(dirname, { recursive: true })
   }
 
   if (fs.existsSync(filePath)) {
-    return
+    return null
   }
   const filePathWithExtension = addExtensionToFilenameIfNoExtensionPresent(filePath, markdownExtensions, '.md')
 
   fs.writeFileSync(filePathWithExtension, content, charset)
+  return {
+    name: path.basename(filePathWithExtension),
+    path: filePathWithExtension,
+    relativePath: path.relative(dirname, filePathWithExtension),
+    dateModified: new Date(),
+    dateCreated: new Date(),
+  }
 }
 
 export function updateFileListForRenderer(win: BrowserWindow, directory: string): void {
@@ -194,3 +201,26 @@ export function splitDirectoryPathIntoBaseAndRepo(fullPath: string) {
 
   return { localModelPath, repoName }
 }
+
+export function findFileRecursive(dir: string, filename: string): string | null {
+  const files = fs.readdirSync(dir)
+  const basename = path.parse(filename).name
+
+  for (const file of files) {
+    const fullPath = path.resolve(dir, file)
+    const stat = fs.statSync(fullPath)
+
+    if (stat.isDirectory()) {
+      const result = findFileRecursive(fullPath, filename)
+      if (result) return result
+    } else {
+      // Check if files match
+      const fileName = path.parse(file).name
+      if (fileName === basename) {
+        return fullPath
+      }
+    }
+  }
+
+  return null
+}

electron/main/filesystem/ipcHandlers.ts

@@ -9,7 +9,7 @@ import { StoreSchema } from '../electron-store/storeConfig'
 import { handleFileRename, updateFileInTable } from '../vector-database/tableHelperFunctions'
 
 import { GetFilesInfoTree, createFileRecursive, isHidden, GetFilesInfoListForListOfPaths } from './filesystem'
-import { FileInfoTree, WriteFileProps, RenameFileProps, FileInfoWithContent } from './types'
+import { FileInfoTree, WriteFileProps, RenameFileProps, FileInfoWithContent, FileInfo } from './types'
 import ImageStorage from './storage/ImageStore'
 import VideoStorage from './storage/VideoStore'
 
@@ -145,8 +145,10 @@ const registerFileHandlers = (store: Store<StoreSchema>, _windowsManager: Window
     await updateFileInTable(windowInfo.dbTableClient, filePath)
   })
 
-  ipcMain.handle('create-file', async (event, filePath: string, content: string): Promise<void> => {
-    createFileRecursive(filePath, content, 'utf-8')
+  ipcMain.handle('create-file', async (event, filePath: string, content: string): Promise<FileInfo | undefined> => {
+    const fileObject = createFileRecursive(filePath, content, 'utf-8')
+    if (fileObject) return fileObject
+    return undefined
   })
 
   ipcMain.handle('create-directory', async (event, dirPath: string): Promise<void> => {
@@ -217,6 +219,17 @@ const registerFileHandlers = (store: Store<StoreSchema>, _windowsManager: Window
     })
     return result.filePaths
   })
+
+  ipcMain.handle('get-file-info', (event, absolutePath: string, parentRelativePath: string): FileInfo => {
+    const fileInfo = fs.statSync(absolutePath)
+    return {
+      name: path.basename(absolutePath),
+      path: absolutePath,
+      relativePath: parentRelativePath,
+      dateModified: fileInfo.mtime,
+      dateCreated: fileInfo.birthtime, // Add the birthtime property here
+    }
+  })
 }
 
 export default registerFileHandlers

electron/main/path/ipcHandlers.ts

@@ -24,6 +24,11 @@ const registerPathHandlers = () => {
   )
 
   ipcMain.handle('path-ext-name', (event, pathString: string) => path.extname(pathString))
+
+  ipcMain.handle('find-absolute-path', (event, filePath: string) => {
+    const absolutePath = path.resolve(filePath)
+    return absolutePath
+  })
 }
 
 export default registerPathHandlers

electron/main/vector-database/lanceTableWrapper.ts

@@ -1,5 +1,4 @@
 import { Connection, Table as LanceDBTable, MetricType, makeArrowTable } from 'vectordb'
-
 import { EmbeddingModelConfig } from '../electron-store/storeConfig'
 
 import { EnhancedEmbeddingFunction, createEmbeddingFunction } from './embeddings'
@@ -102,7 +101,7 @@ class LanceDBTableWrapper {
   }
 
   async search(query: string, limit: number, filter?: string): Promise<DBQueryResult[]> {
-    const lanceQuery = await this.lanceTable.search(query).metricType(MetricType.Cosine).limit(limit)
+    const lanceQuery = this.lanceTable.search(query).metricType(MetricType.Cosine).limit(limit)
 
     if (filter) {
       lanceQuery.prefilter(true)

electron/main/vector-database/schema.ts

@@ -1,6 +1,7 @@
 import { Schema, Field, Utf8, FixedSizeList, Float32, Float64, DateUnit, Date_ as ArrowDate } from 'apache-arrow'
 
 export interface DBEntry {
+  name: string
   notepath: string
   content: string
   subnoteindex: number
@@ -15,6 +16,7 @@ export interface DBQueryResult extends DBEntry {
 export const chunksize = 500
 
 export enum DatabaseFields {
+  NAME = 'name',
   NOTE_PATH = 'notepath',
   VECTOR = 'vector',
   CONTENT = 'content',
@@ -27,6 +29,7 @@ export enum DatabaseFields {
 
 const CreateDatabaseSchema = (vectorDim: number): Schema => {
   const schemaFields = [
+    new Field(DatabaseFields.NAME, new Utf8(), false),
     new Field(DatabaseFields.NOTE_PATH, new Utf8(), false),
     new Field(DatabaseFields.VECTOR, new FixedSizeList(vectorDim, new Field('item', new Float32())), false),
     new Field(DatabaseFields.CONTENT, new Utf8(), false),

electron/main/vector-database/tableHelperFunctions.ts

@@ -1,5 +1,6 @@
 import * as fs from 'fs'
 import * as fsPromises from 'fs/promises'
+import path from 'path'
 
 import { BrowserWindow } from 'electron'
 import { chunkMarkdownByHeadingsAndByCharsIfBig } from '../common/chunking'
@@ -20,6 +21,7 @@ const convertFileTypeToDBType = async (file: FileInfo): Promise<DBEntry[]> => {
   const fileContent = readFile(file.path)
   const chunks = await chunkMarkdownByHeadingsAndByCharsIfBig(fileContent)
   const entries = chunks.map((content, index) => ({
+    name: file.name,
     notepath: file.path,
     content,
     subnoteindex: index,
@@ -169,8 +171,10 @@ export const updateFileInTable = async (dbTable: LanceDBTableWrapper, filePath:
   await dbTable.deleteDBItemsByFilePaths([filePath])
   const content = readFile(filePath)
   const chunkedContentList = await chunkMarkdownByHeadingsAndByCharsIfBig(content)
+  const fileName = path.basename(filePath).split('.')[0]
   const stats = fs.statSync(filePath)
   const dbEntries = chunkedContentList.map((_content, index) => ({
+    name: fileName,
     notepath: filePath,
     content: _content,
     subnoteindex: index,

electron/preload/index.ts

@@ -9,7 +9,13 @@ import {
   TamaguiThemeTypes,
 } from 'electron/main/electron-store/storeConfig'
 import { SearchProps } from 'electron/main/electron-store/types'
-import { FileInfoTree, FileInfoWithContent, RenameFileProps, WriteFileProps } from 'electron/main/filesystem/types'
+import {
+  FileInfo,
+  FileInfoTree,
+  FileInfoWithContent,
+  RenameFileProps,
+  WriteFileProps,
+} from 'electron/main/filesystem/types'
 import { DBQueryResult } from 'electron/main/vector-database/schema'
 
 import { AgentConfig, ChatMetadata, Chat } from '@/lib/llm/types'
@@ -102,12 +108,14 @@ const fileSystem = {
   getVideo: createIPCHandler<(fileName: string) => Promise<string | null>>('get-video'),
   isDirectory: createIPCHandler<(filePath: string) => Promise<boolean>>('is-directory'),
   renameFile: createIPCHandler<(renameFileProps: RenameFileProps) => Promise<void>>('rename-file'),
-  createFile: createIPCHandler<(filePath: string, content: string) => Promise<void>>('create-file'),
+  createFile: createIPCHandler<(filePath: string, content: string) => Promise<FileInfo | undefined>>('create-file'),
   createDirectory: createIPCHandler<(dirPath: string) => Promise<void>>('create-directory'),
   checkFileExists: createIPCHandler<(filePath: string) => Promise<boolean>>('check-file-exists'),
   deleteFile: createIPCHandler<(filePath: string) => Promise<void>>('delete-file'),
   getAllFilenamesInDirectory: createIPCHandler<(dirName: string) => Promise<string[]>>('get-files-in-directory'),
   getFiles: createIPCHandler<(filePaths: string[]) => Promise<FileInfoWithContent[]>>('get-files'),
+  getFileInfo:
+    createIPCHandler<(absolutePath: string, parentRelativePath: string) => Promise<FileInfo>>('get-file-info'),
 }
 
 const path = {

package.json

@@ -168,7 +168,8 @@
     "vaul": "^1.0.0",
     "vectordb": "0.4.10",
     "yjs": "^13.6.23",
-    "zod": "^3.23.8"
+    "zod": "^3.23.8",
+    "zustand": "^5.0.3"
   },
   "devDependencies": {
     "@electron/notarize": "2.3.2",