refactor: v2.0 全量重构 — TypeScript + React + Zustand + IndexedDB

技术栈升级： - JavaScript → TypeScript 5.6（全量类型安全） - 原生 DOM → React 18（函数组件 + Hooks） - 全局变量 → Zustand 5（轻量状态管理） - localStorage → IndexedDB / Dexie.js 4（大容量、异步、索引） - marked.js → unified / remark / rehype（插件化渲染管线） - 无打包 → electron-vite 3（HMR 热更新） - 纯 CSS → CSS Variables + CSS Modules 新增功能： - 标签页状态 IndexedDB 持久化（关闭后可恢复） - 最近打开文件列表 - 大文件虚拟化行号（>2000 行） - 搜索高亮二分查找优化 O(log N) - rehype-sanitize HTML 安全过滤文件结构： - 7 个源文件 → 51 个模块化文件 - src/main/ 主进程（6 文件） - src/preload/ 预加载（1 文件） - src/renderer/ 渲染进程（42 文件：组件/hooks/stores/lib/db/types/styles） - src/shared/ 共享类型（2 文件）构建验证： - TypeScript 检查零错误 - electron-vite build 成功 - 产物：main 15kB + preload 2kB + renderer 1.5MB
2026-05-27 20:29:23 +08:00
parent c2cdba546b
commit 5e1c89d280
59 changed files with 4674 additions and 314 deletions
@@ -0,0 +1,75 @@
+import { readFile, stat, writeFile, readdir } from 'fs/promises'
+import { join, extname } from 'path'
+
+const MAX_FILE_SIZE = 20 * 1024 * 1024 // 20MB
+const ALLOWED_EXTENSIONS = new Set(['.md', '.markdown', '.txt'])
+const SKIP_DIRS = new Set([
+  'node_modules', '.git', '.svn', '.hg', 'dist', 'out',
+  '.next', '.nuxt', '__pycache__', '.DS_Store'
+])
+
+export interface FileNode {
+  name: string
+  path: string
+  type: 'file' | 'dir'
+  children?: FileNode[]
+}
+
+export async function readFileContent(filePath: string): Promise<{ success: boolean; content?: string; error?: string }> {
+  try {
+    const fileStat = await stat(filePath)
+    if (fileStat.size > MAX_FILE_SIZE) {
+      return { success: false, error: `文件过大（${(fileStat.size / 1024 / 1024).toFixed(1)} MB），暂不支持超过 20MB 的文件` }
+    }
+    let content = await readFile(filePath, 'utf-8')
+    // 剥离 UTF-8 BOM
+    if (content.charCodeAt(0) === 0xFEFF) content = content.slice(1)
+    return { success: true, content }
+  } catch (err) {
+    return { success: false, error: (err as Error).message }
+  }
+}
+
+export async function saveFileContent(filePath: string, content: string): Promise<{ success: boolean; filePath?: string; error?: string }> {
+  try {
+    await writeFile(filePath, content, 'utf-8')
+    return { success: true, filePath }
+  } catch (err) {
+    return { success: false, error: (err as Error).message }
+  }
+}
+
+export async function buildDirTree(dirPath: string): Promise<FileNode[]> {
+  const entries = await readdir(dirPath, { withFileTypes: true })
+  entries.sort((a, b) => {
+    if (a.isDirectory() && !b.isDirectory()) return -1
+    if (!a.isDirectory() && b.isDirectory()) return 1
+    return a.name.localeCompare(b.name)
+  })
+
+  const children: FileNode[] = []
+  for (const entry of entries) {
+    if (SKIP_DIRS.has(entry.name)) continue
+    if (entry.name.startsWith('.')) continue
+
+    const childPath = join(dirPath, entry.name)
+    if (entry.isDirectory()) {
+      const subChildren = await buildDirTree(childPath)
+      if (subChildren.length > 0) {
+        children.push({ name: entry.name, path: childPath, type: 'dir', children: subChildren })
+      }
+    } else {
+      const ext = extname(entry.name).toLowerCase()
+      if (ALLOWED_EXTENSIONS.has(ext)) {
+        children.push({ name: entry.name, path: childPath, type: 'file' })
+      }
+    }
+  }
+  return children
+}
+
+export function formatBytes(bytes: number): string {
+  if (bytes < 1024) return bytes + ' B'
+  if (bytes < 1024 * 1024) return (bytes / 1024).toFixed(1) + ' KB'
+  return (bytes / (1024 * 1024)).toFixed(1) + ' MB'
+}
@@ -0,0 +1,83 @@
+import * as fs from 'fs'
+import { BrowserWindow } from 'electron'
+
+export class FileWatcher {
+  private watcher: fs.FSWatcher | null = null
+  private currentPath: string | null = null
+  private isSelfWriting = false
+
+  constructor(private getMainWindow: () => BrowserWindow | null) {}
+
+  start(filePath: string): void {
+    this.stop()
+    if (!filePath) return
+    try {
+      this.currentPath = filePath
+      this.watcher = fs.watch(filePath, (eventType) => {
+        if (eventType === 'change') {
+          if (this.isSelfWriting) return
+          const win = this.getMainWindow()
+          if (win && !win.isDestroyed()) {
+            win.webContents.send('file:externallyModified', filePath)
+          }
+        }
+      })
+    } catch {
+      // silently ignore
+    }
+  }
+
+  stop(): void {
+    if (this.watcher) {
+      this.watcher.close()
+      this.watcher = null
+    }
+  }
+
+  getCurrentPath(): string | null {
+    return this.currentPath
+  }
+
+  setSelfWriting(value: boolean): void {
+    this.isSelfWriting = value
+  }
+}
+
+export class SidebarWatcher {
+  private watcher: fs.FSWatcher | null = null
+  private watchPath: string | null = null
+  private refreshTimer: NodeJS.Timeout | null = null
+
+  constructor(private getMainWindow: () => BrowserWindow | null) {}
+
+  start(dirPath: string): void {
+    this.stop()
+    if (!dirPath) return
+    try {
+      this.watchPath = dirPath
+      this.watcher = fs.watch(dirPath, { recursive: true }, () => {
+        const win = this.getMainWindow()
+        if (win && !win.isDestroyed()) {
+          if (this.refreshTimer) clearTimeout(this.refreshTimer)
+          this.refreshTimer = setTimeout(() => {
+            win.webContents.send('sidebar:dirChanged')
+          }, 300)
+        }
+      })
+    } catch {
+      // silently ignore
+    }
+  }
+
+  stop(): void {
+    if (this.watcher) {
+      this.watcher.close()
+      this.watcher = null
+    }
+    if (this.refreshTimer) {
+      clearTimeout(this.refreshTimer)
+      this.refreshTimer = null
+    }
+    this.watchPath = null
+  }
+}
@@ -0,0 +1,129 @@
+import { app, BrowserWindow } from 'electron'
+import { join } from 'path'
+import { createWindow, setupSingleInstanceLock, getFilePathFromArgs } from './window-manager'
+import { FileWatcher, SidebarWatcher } from './file-watcher'
+import { registerIpcHandlers } from './ipc-handlers'
+import { readFileContent } from './file-system'
+
+let mainWindow: BrowserWindow | null = null
+const state = {
+  activeFilePath: null as string | null,
+  pendingFilePath: null as string | null,
+  isClosing: false,
+  closeTimeout: null as NodeJS.Timeout | null
+}
+
+const fileWatcher = new FileWatcher(() => mainWindow)
+const sidebarWatcher = new SidebarWatcher(() => mainWindow)
+
+function openFileInTab(filePath: string): void {
+  if (!mainWindow || mainWindow.isDestroyed()) return
+  readFileContent(filePath).then(result => {
+    if (result.success) {
+      state.activeFilePath = filePath
+      fileWatcher.start(filePath)
+      mainWindow!.setTitle(`MarkLite - ${filePath.split(/[/\\]/).pop()}`)
+      mainWindow!.webContents.send('file:openInTab', { filePath, content: result.content })
+    }
+  })
+}
+
+// 单实例锁
+const lockOk = setupSingleInstanceLock((filePath) => {
+  if (mainWindow && !mainWindow.isDestroyed()) {
+    if (mainWindow.isMinimized()) mainWindow.restore()
+    mainWindow.focus()
+    if (filePath) openFileInTab(filePath)
+  }
+})
+
+if (!lockOk) {
+  app.quit()
+} else {
+
+  function setupCloseHandler(): void {
+    if (!mainWindow) return
+    mainWindow.on('close', (e) => {
+      if (state.isClosing) return
+      state.isClosing = true
+      e.preventDefault()
+      try {
+        if (mainWindow && !mainWindow.webContents.isDestroyed()) {
+          mainWindow.webContents.send('window:confirmClose')
+        } else {
+          mainWindow?.removeAllListeners('close')
+          mainWindow?.close()
+          return
+        }
+      } catch {
+        mainWindow?.removeAllListeners('close')
+        mainWindow?.close()
+        return
+      }
+      state.closeTimeout = setTimeout(() => {
+        state.closeTimeout = null
+        if (mainWindow && !mainWindow.isDestroyed()) {
+          mainWindow.removeAllListeners('close')
+          mainWindow.close()
+        }
+      }, 5000)
+    })
+  }
+
+  app.whenReady().then(() => {
+    app.on('open-file', (event, filePath) => {
+      event.preventDefault()
+      if (mainWindow && !mainWindow.isDestroyed() && mainWindow.webContents.isLoading()) {
+        state.pendingFilePath = filePath
+      } else if (mainWindow && !mainWindow.isDestroyed()) {
+        openFileInTab(filePath)
+      } else {
+        state.pendingFilePath = filePath
+      }
+    })
+
+    mainWindow = createWindow()
+
+    // 注册 IPC 处理器
+    registerIpcHandlers(() => mainWindow, fileWatcher, sidebarWatcher, state)
+
+    // 加载渲染进程页面
+    if (process.env.ELECTRON_RENDERER_URL) {
+      mainWindow.loadURL(process.env.ELECTRON_RENDERER_URL)
+    } else {
+      mainWindow.loadFile(join(__dirname, '../renderer/index.html'))
+    }
+
+    mainWindow.webContents.on('did-finish-load', () => {
+      if (state.pendingFilePath) {
+        openFileInTab(state.pendingFilePath)
+        state.pendingFilePath = null
+      }
+    })
+
+    setupCloseHandler()
+
+    mainWindow.on('closed', () => {
+      fileWatcher.stop()
+      sidebarWatcher.stop()
+      mainWindow = null
+    })
+
+    // 命令行文件
+    const cmdFile = getFilePathFromArgs(process.argv)
+    if (cmdFile) {
+      state.pendingFilePath = cmdFile
+    }
+  })
+
+  app.on('window-all-closed', () => {
+    if (process.platform !== 'darwin') app.quit()
+  })
+
+  app.on('activate', () => {
+    if (BrowserWindow.getAllWindows().length === 0) {
+      mainWindow = createWindow()
+      setupCloseHandler()
+    }
+  })
+}
@@ -0,0 +1,23 @@
+export const IPC_CHANNELS = {
+  DIALOG_OPEN_FILE: 'dialog:openFile',
+  FILE_READ: 'file:read',
+  FILE_SAVE: 'file:save',
+  FILE_SAVE_AS: 'file:saveAs',
+  FILE_GET_CURRENT_PATH: 'file:getCurrentPath',
+  FILE_STATS: 'file:stats',
+  FILE_RELOAD: 'file:reload',
+  TAB_SWITCHED: 'tab:switched',
+  WINDOW_FORCE_CLOSE: 'window:forceClose',
+  WINDOW_CANCEL_CLOSE: 'window:cancelClose',
+  DIR_READ_TREE: 'dir:readTree',
+  DIR_OPEN_DIALOG: 'dir:openDialog',
+  DIR_WATCH: 'dir:watch',
+  DIR_UNWATCH: 'dir:unwatch',
+  FILE_OPEN_IN_TAB: 'file:openInTab',
+  FILE_EXTERNALLY_MODIFIED: 'file:externallyModified',
+  MENU_SAVE: 'menu:save',
+  MENU_SAVE_AS: 'menu:saveAs',
+  MENU_VIEW_MODE: 'menu:viewMode',
+  WINDOW_CONFIRM_CLOSE: 'window:confirmClose',
+  SIDEBAR_DIR_CHANGED: 'sidebar:dirChanged'
+} as const
@@ -0,0 +1,183 @@
+import { ipcMain, dialog, BrowserWindow } from 'electron'
+import { readFileContent, saveFileContent, buildDirTree } from './file-system'
+import { FileWatcher, SidebarWatcher } from './file-watcher'
+import { IPC_CHANNELS } from './ipc-channels'
+import { stat } from 'fs/promises'
+import { join, basename } from 'path'
+
+export function registerIpcHandlers(
+  getMainWindow: () => BrowserWindow | null,
+  fileWatcher: FileWatcher,
+  sidebarWatcher: SidebarWatcher,
+  state: { activeFilePath: string | null; pendingFilePath: string | null }
+): void {
+  // 打开文件对话框
+  ipcMain.handle(IPC_CHANNELS.DIALOG_OPEN_FILE, async () => {
+    const win = getMainWindow()
+    if (!win) return null
+    try {
+      const result = await dialog.showOpenDialog(win, {
+        properties: ['openFile'],
+        filters: [{ name: 'Markdown 文件', extensions: ['md', 'markdown', 'txt'] }]
+      })
+      if (!result.canceled && result.filePaths.length > 0) {
+        const filePath = result.filePaths[0]
+        const fileResult = await readFileContent(filePath)
+        if (fileResult.success) {
+          state.activeFilePath = filePath
+          fileWatcher.start(filePath)
+          win.setTitle(`MarkLite - ${basename(filePath)}`)
+          return { filePath, content: fileResult.content }
+        }
+        return { error: fileResult.error }
+      }
+      return null
+    } catch (err) {
+      return { error: (err as Error).message }
+    }
+  })
+
+  // 读取文件
+  ipcMain.handle(IPC_CHANNELS.FILE_READ, async (_event: unknown, filePath: string) => {
+    return readFileContent(filePath)
+  })
+
+  // 保存文件
+  ipcMain.handle(IPC_CHANNELS.FILE_SAVE, async (_event: unknown, data: { filePath: string | null; content: string }) => {
+    const win = getMainWindow()
+    try {
+      if (data.filePath) {
+        fileWatcher.stop()
+        fileWatcher.setSelfWriting(true)
+        const result = await saveFileContent(data.filePath, data.content)
+        fileWatcher.setSelfWriting(false)
+        state.activeFilePath = data.filePath
+        setTimeout(() => {
+          if (state.activeFilePath === data.filePath && data.filePath) {
+            fileWatcher.start(data.filePath)
+          }
+        }, 300)
+        if (win && !win.isDestroyed()) {
+          win.setTitle(`MarkLite - ${basename(data.filePath)}`)
+        }
+        return result
+      } else {
+        if (!win) return { success: false, error: '窗口不可用' }
+        const saveResult = await dialog.showSaveDialog(win, {
+          filters: [{ name: 'Markdown 文件', extensions: ['md'] }]
+        })
+        if (!saveResult.canceled) {
+          const result = await saveFileContent(saveResult.filePath, data.content)
+          if (result.success) {
+            state.activeFilePath = saveResult.filePath
+            fileWatcher.start(saveResult.filePath)
+            win.setTitle(`MarkLite - ${basename(saveResult.filePath)}`)
+          }
+          return result
+        }
+        return { success: false, canceled: true }
+      }
+    } catch (err) {
+      fileWatcher.setSelfWriting(false)
+      fileWatcher.start(state.activeFilePath || '')
+      return { success: false, error: (err as Error).message }
+    }
+  })
+
+  // 另存为
+  ipcMain.handle(IPC_CHANNELS.FILE_SAVE_AS, async (_event: unknown, data: { content: string }) => {
+    const win = getMainWindow()
+    if (!win) return { success: false, error: '窗口不可用' }
+    try {
+      const result = await dialog.showSaveDialog(win, {
+        filters: [{ name: 'Markdown 文件', extensions: ['md'] }]
+      })
+      if (!result.canceled) {
+        const saveResult = await saveFileContent(result.filePath, data.content)
+        if (saveResult.success) {
+          state.activeFilePath = result.filePath
+          fileWatcher.start(result.filePath)
+          win.setTitle(`MarkLite - ${basename(result.filePath)}`)
+        }
+        return saveResult
+      }
+      return { success: false, canceled: true }
+    } catch (err) {
+      return { success: false, error: (err as Error).message }
+    }
+  })
+
+  // 获取当前路径
+  ipcMain.handle(IPC_CHANNELS.FILE_GET_CURRENT_PATH, () => state.activeFilePath)
+
+  // 文件统计
+  ipcMain.handle(IPC_CHANNELS.FILE_STATS, async (_event: unknown, filePath: string) => {
+    try {
+      const fileStat = await stat(filePath)
+      return { success: true, size: fileStat.size, mtime: fileStat.mtime.toISOString() }
+    } catch (err) {
+      return { success: false, error: (err as Error).message }
+    }
+  })
+
+  // 重新加载
+  ipcMain.handle(IPC_CHANNELS.FILE_RELOAD, async () => {
+    if (!state.activeFilePath) return { success: false, error: '没有打开的文件' }
+    const result = await readFileContent(state.activeFilePath)
+    return { ...result, filePath: state.activeFilePath }
+  })
+
+  // 目录树
+  ipcMain.handle(IPC_CHANNELS.DIR_READ_TREE, async (_event: unknown, dirPath: string) => {
+    try {
+      const tree = await buildDirTree(dirPath)
+      return { success: true, tree, rootPath: dirPath }
+    } catch (err) {
+      return { success: false, error: (err as Error).message }
+    }
+  })
+
+  // 打开文件夹对话框
+  ipcMain.handle(IPC_CHANNELS.DIR_OPEN_DIALOG, async () => {
+    const win = getMainWindow()
+    if (!win) return null
+    const result = await dialog.showOpenDialog(win, { properties: ['openDirectory'] })
+    if (!result.canceled && result.filePaths.length > 0) {
+      return result.filePaths[0]
+    }
+    return null
+  })
+
+  // 目录监听
+  ipcMain.handle(IPC_CHANNELS.DIR_WATCH, (_event: unknown, dirPath: string) => {
+    sidebarWatcher.start(dirPath)
+  })
+
+  ipcMain.handle(IPC_CHANNELS.DIR_UNWATCH, () => {
+    sidebarWatcher.stop()
+  })
+
+  // 标签切换
+  ipcMain.handle(IPC_CHANNELS.TAB_SWITCHED, (_event: unknown, filePath: string | null) => {
+    state.activeFilePath = filePath || null
+    fileWatcher.start(filePath || '')
+    const win = getMainWindow()
+    if (win && !win.isDestroyed()) {
+      win.setTitle(filePath ? `MarkLite - ${basename(filePath)}` : 'MarkLite')
+    }
+  })
+
+  // 窗口控制
+  ipcMain.handle(IPC_CHANNELS.WINDOW_FORCE_CLOSE, () => {
+    const win = getMainWindow()
+    if (win && !win.isDestroyed()) {
+      fileWatcher.stop()
+      win.removeAllListeners('close')
+      win.close()
+    }
+  })
+
+  ipcMain.handle(IPC_CHANNELS.WINDOW_CANCEL_CLOSE, () => {
+    // 重置关闭状态
+  })
+}
@@ -0,0 +1,59 @@
+import { BrowserWindow, app } from 'electron'
+import { join } from 'path'
+
+export function createWindow(): BrowserWindow {
+  const mainWindow = new BrowserWindow({
+    width: 1200,
+    height: 800,
+    minWidth: 800,
+    minHeight: 600,
+    icon: join(__dirname, '../assets/icon.ico'),
+    webPreferences: {
+      preload: join(__dirname, '../preload/index.js'),
+      contextIsolation: true,
+      nodeIntegration: false
+    },
+    titleBarStyle: 'default',
+    show: false
+  })
+
+  mainWindow.setMenu(null)
+
+  mainWindow.once('ready-to-show', () => {
+    mainWindow.show()
+  })
+
+  return mainWindow
+}
+
+export function setupSingleInstanceLock(
+  onSecondInstance: (filePath: string | null) => void
+): boolean {
+  const gotTheLock = app.requestSingleInstanceLock()
+  if (!gotTheLock) {
+    app.quit()
+    return false
+  }
+
+  app.on('second-instance', (_event, commandLine) => {
+    const filePath = getFilePathFromArgs(commandLine)
+    onSecondInstance(filePath)
+  })
+
+  return true
+}
+
+export function getFilePathFromArgs(args: string[]): string | null {
+  for (let i = 1; i < args.length; i++) {
+    const arg = args[i]
+    if (!arg.startsWith('--') && !arg.startsWith('-')) {
+      try {
+        const fs = require('fs')
+        if (fs.existsSync(arg)) return arg
+      } catch {
+        // ignore
+      }
+    }
+  }
+  return null
+}