From 8ce35a1e8e9ae9e9935220d595957ccc6697c381 Mon Sep 17 00:00:00 2001 From: kun1h Date: Mon, 6 Jul 2026 16:33:56 +0800 Subject: [PATCH] feat: ship v0.3.0 with P2.1 editor polish and P3 AI collaboration Add chapter reorder, search replace, inspiration capture, and AI chat with context editing, settings, offline handling, and naming checks. Fix dev black screen by keeping process.env reads in the main process only. Co-authored-by: Cursor --- README.md | 16 +- .../plans/2026-07-07-bilin-p21-p3.md | 110 ++--- e2e/ai-chat.spec.ts | 204 +++++++++ e2e/p21-features.spec.ts | 157 +++++++ package.json | 2 +- public/locales/en/translation.json | 69 ++++ public/locales/zh-CN/translation.json | 69 ++++ src/main/db/migrate.ts | 9 +- src/main/db/repositories/ai-session.repo.ts | 125 ++++++ src/main/db/repositories/chapter.repo.ts | 41 ++ src/main/db/schema-v3.sql | 20 + src/main/index.ts | 9 +- src/main/ipc/handlers/ai.handler.ts | 220 ++++++++++ src/main/ipc/handlers/chapter.handler.ts | 20 + src/main/ipc/register.ts | 10 + src/main/ipc/result.ts | 13 +- src/main/services/ai-client.service.ts | 105 +++++ .../services/ai-context-builder.service.ts | 131 ++++++ src/main/services/book-registry.ts | 5 + src/main/services/global-settings.ts | 20 +- src/main/services/naming-check.service.ts | 151 +++++++ src/main/services/network-monitor.service.ts | 31 ++ src/main/services/search.service.ts | 6 +- src/preload/index.ts | 67 ++- src/renderer/App.tsx | 33 +- src/renderer/components/ai/AiPanel.tsx | 229 ++++++++++ .../components/ai/ContextEditorModal.tsx | 150 +++++++ .../components/ai/NamingCheckModal.tsx | 124 ++++++ .../components/editor/TipTapEditor.tsx | 35 +- .../inspiration/InspirationModal.tsx | 144 +++++++ .../components/knowledge/KnowledgePanel.tsx | 27 ++ .../components/layout/EditorLayout.tsx | 77 +++- src/renderer/components/layout/RightPanel.tsx | 9 +- .../components/reference/ReferencePanel.tsx | 27 +- .../components/search/SearchModal.tsx | 66 +++ .../components/settings/AiSettingsPage.tsx | 142 +++++++ .../components/settings/SettingsPage.tsx | 15 +- src/renderer/extensions/mention.ts | 16 +- src/renderer/lib/ai-context-summary.ts | 16 + src/renderer/lib/ai-slash-commands.ts | 22 + src/renderer/lib/editor-commands.ts | 34 ++ src/renderer/stores/useAiStore.ts | 178 ++++++++ src/renderer/stores/useAppStore.ts | 4 + src/renderer/stores/useEditStore.ts | 7 + src/renderer/stores/useReferenceStore.ts | 9 +- src/renderer/stores/useSettingsStore.ts | 3 + src/renderer/styles/layout.css | 390 ++++++++++++++++++ src/shared/electron-api.d.ts | 38 ++ src/shared/ipc-channels.ts | 16 +- src/shared/types.ts | 81 ++++ tests/main/ai-client.test.ts | 12 + tests/main/ai-context-builder.test.ts | 86 ++++ tests/main/ai-session.repo.test.ts | 69 ++++ tests/main/chapter-move.test.ts | 44 ++ tests/main/migrate-v2.test.ts | 4 +- tests/main/migrate-v3.test.ts | 54 +++ tests/main/naming-check.integration.test.ts | 37 ++ tests/main/naming-check.repo.test.ts | 50 +++ tests/main/naming-parse.test.ts | 50 +++ 59 files changed, 3808 insertions(+), 100 deletions(-) create mode 100644 e2e/ai-chat.spec.ts create mode 100644 e2e/p21-features.spec.ts create mode 100644 src/main/db/repositories/ai-session.repo.ts create mode 100644 src/main/db/schema-v3.sql create mode 100644 src/main/ipc/handlers/ai.handler.ts create mode 100644 src/main/services/ai-client.service.ts create mode 100644 src/main/services/ai-context-builder.service.ts create mode 100644 src/main/services/naming-check.service.ts create mode 100644 src/main/services/network-monitor.service.ts create mode 100644 src/renderer/components/ai/AiPanel.tsx create mode 100644 src/renderer/components/ai/ContextEditorModal.tsx create mode 100644 src/renderer/components/ai/NamingCheckModal.tsx create mode 100644 src/renderer/components/inspiration/InspirationModal.tsx create mode 100644 src/renderer/components/knowledge/KnowledgePanel.tsx create mode 100644 src/renderer/components/settings/AiSettingsPage.tsx create mode 100644 src/renderer/lib/ai-context-summary.ts create mode 100644 src/renderer/lib/ai-slash-commands.ts create mode 100644 src/renderer/stores/useAiStore.ts create mode 100644 tests/main/ai-client.test.ts create mode 100644 tests/main/ai-context-builder.test.ts create mode 100644 tests/main/ai-session.repo.test.ts create mode 100644 tests/main/chapter-move.test.ts create mode 100644 tests/main/migrate-v3.test.ts create mode 100644 tests/main/naming-check.integration.test.ts create mode 100644 tests/main/naming-check.repo.test.ts create mode 100644 tests/main/naming-parse.test.ts diff --git a/README.md b/README.md index 69c1384..a6b0570 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,12 @@ # 笔临 (Bilin) -长篇创作智能协作平台 — Electron 桌面客户端 v0.2.0（P2） +长篇创作智能协作平台 — Electron 桌面客户端 v0.3.0（P2.1 + P3） + +## 功能概览（v0.3.0） + +- 章节拖拽、全局搜索替换、灵感捕获、`@` 跳转、引用钉住（P2.1） +- AI 对话助手：会话持久化、流式输出、上下文编辑、快捷命令（P3） +- AI 设置页、命名一致性检查、离线降级、AI 快照（P3） ## 开发 @@ -15,13 +21,15 @@ npm run dev ```bash npm run build -npm run test # 单元测试 +npm run test # 单元 / 集成测试 npm run test:e2e # E2E 测试（需先 build） ``` +**AI 相关测试前置条件**：本地 [LM Studio](http://127.0.0.1:1234) 需运行，模型默认 `gemma-4-e4b-it`（可通过环境变量 `BILIN_AI_MODEL` / `BILIN_AI_BASE_URL` 覆盖）。 + ## 文档 +- P2.1 + P3 设计规格：`docs/superpowers/specs/2026-07-07-bilin-p21-p3-design.md` +- P2.1 + P3 实现计划：`docs/superpowers/plans/2026-07-07-bilin-p21-p3.md` - P2 设计规格：`docs/superpowers/specs/2026-07-06-bilin-p2-design.md` -- P2 实现计划：`docs/superpowers/plans/2026-07-06-bilin-p2.md` -- P0/P1 设计规格：`docs/superpowers/specs/2026-07-05-bilin-p0-p1-design.md` - UI 原型：`design/ui_pc.html` diff --git a/docs/superpowers/plans/2026-07-07-bilin-p21-p3.md b/docs/superpowers/plans/2026-07-07-bilin-p21-p3.md index 724e29e..65f52c4 100644 --- a/docs/superpowers/plans/2026-07-07-bilin-p21-p3.md +++ b/docs/superpowers/plans/2026-07-07-bilin-p21-p3.md @@ -55,7 +55,7 @@ - Create: `tests/main/chapter-move.test.ts` - Create: `e2e/p21-features.spec.ts`（DRAG 用例） -- [ ] **Step 1: 写 chapter.move 失败测试** +- [x] **Step 1: 写 chapter.move 失败测试** `tests/main/chapter-move.test.ts`： @@ -96,7 +96,7 @@ describe('ChapterRepository.move', () => { }) ``` -- [ ] **Step 2: 运行确认 FAIL** +- [x] **Step 2: 运行确认 FAIL** ```bash npm run test -- tests/main/chapter-move.test.ts @@ -104,7 +104,7 @@ npm run test -- tests/main/chapter-move.test.ts Expected: FAIL `chapters.move is not a function` -- [ ] **Step 3: 实现 `ChapterRepository.move`** +- [x] **Step 3: 实现 `ChapterRepository.move`** ```typescript move(chapterId: string, volumeId: string, sortOrder: number): Chapter { @@ -124,7 +124,7 @@ reorderVolume(volumeId: string, orderedIds: string[]): void { } ``` -- [ ] **Step 4: IPC `CHAPTER_MOVE`** +- [x] **Step 4: IPC `CHAPTER_MOVE`** `ipc-channels.ts` 追加 `CHAPTER_MOVE: 'chapter:move'`。 @@ -140,7 +140,7 @@ ipcMain.handle( preload / electron-api.d.ts 暴露 `chapter.move(bookId, chapterId, volumeId, sortOrder)`。 -- [ ] **Step 5: EditorLayout 拖拽 UI** +- [x] **Step 5: EditorLayout 拖拽 UI** 章节 `.chapter-item` 增加 `draggable`；`onDragStart` 设 `chapterId`；`onDragOver` preventDefault；`onDrop` 计算新 `sortOrder` 调 IPC。 @@ -148,11 +148,11 @@ preload / electron-api.d.ts 暴露 `chapter.move(bookId, chapterId, volumeId, so 跨卷：drop 到另一卷 header → `move(chapterId, targetVolId, targetLen)`。 -- [ ] **Step 6: E2E E2E-P21-DRAG-01** +- [x] **Step 6: E2E E2E-P21-DRAG-01** `e2e/p21-features.spec.ts`：两章 → drag 交换顺序 → 重启 → 顺序保持。 -- [ ] **Step 7: 验证** +- [x] **Step 7: 验证** ```bash npm run test -- tests/main/chapter-move.test.ts @@ -176,7 +176,7 @@ git commit -am "feat: add chapter drag reorder within and across volumes" - Modify: `public/locales/zh-CN/translation.json`, `public/locales/en/translation.json` - Extend: `e2e/p21-features.spec.ts` -- [ ] **Step 1: SearchModal 替换折叠区** +- [x] **Step 1: SearchModal 替换折叠区** 在 `search-results` 下方增加： @@ -204,15 +204,15 @@ setReplacePreviews(result.previews) `executeReplace`：`dryRun: false` → `refreshChapters()` + toast。 -- [ ] **Step 2: i18n 键** +- [x] **Step 2: i18n 键** `search.replace`, `search.replacePreview`, `search.replaceExecute`, `search.replaceConfirm`, `search.replaceDone` -- [ ] **Step 3: E2E E2E-P21-REP-01** +- [x] **Step 3: E2E E2E-P21-REP-01** 写入章节唯一词 → 搜索 → 预览 → 执行 → `editor` 含新词。 -- [ ] **Step 4: 验证 + Commit** +- [x] **Step 4: 验证 + Commit** ```bash npm run test:e2e -- e2e/p21-features.spec.ts @@ -233,7 +233,7 @@ git commit -am "feat: add global search replace preview and execute UI" - Modify: `src/renderer/styles/layout.css` - Extend: `e2e/p21-features.spec.ts` -- [ ] **Step 1: InspirationModal** +- [x] **Step 1: InspirationModal** 对齐 `design/ui_pc.html` `#inspirationModal`：title、content、tags；保存调 `inspiration.create`；`data-testid="inspiration-modal"`。 @@ -251,11 +251,11 @@ if (SpeechRecognition) { 不可用时 `disabled` + `title={t('inspiration.voiceUnavailable')}`。 -- [ ] **Step 2: App.tsx 改 captureInspiration** +- [x] **Step 2: App.tsx 改 captureInspiration** `captureInspiration` 快捷键 → `useAppStore.setInspirationModalOpen(true)`，不再直接 create。 -- [ ] **Step 3: Mention 点击跳转** +- [x] **Step 3: Mention 点击跳转** `TipTapEditor` `editorProps`： @@ -271,13 +271,13 @@ handleClick: (view, pos, event) => { 确保 `applyMention` 插入 HTML 含 `data-id`。 -- [ ] **Step 4: 引用钉住** +- [x] **Step 4: 引用钉住** `useReferenceStore`：`pinned: boolean`, `setPinned`。 `ReferencePanel`：`pinned` 时关闭按钮改为取消钉住；`setOpen(false)` 若 `pinned` 则忽略。 -- [ ] **Step 5: E2E E2E-P21-INSP-01** +- [x] **Step 5: E2E E2E-P21-INSP-01** `Control+Shift+I` → 填内容 → 保存 → `inspiration-list` 可见。 @@ -299,19 +299,19 @@ git commit -am "feat: add inspiration modal, mention navigation, reference pin" - Modify: `src/shared/ipc-channels.ts` - Modify: `src/main/services/global-settings.ts` -- [ ] **Step 1: schema-v3.sql** +- [x] **Step 1: schema-v3.sql** 内容同 spec §3.1 `ai_sessions` / `ai_messages` / index。 -- [ ] **Step 2: migrate.ts** +- [x] **Step 2: migrate.ts** `CURRENT_VERSION = 3`；`current < 3` 时 `db.exec(schemaV3)` + insert version 3。 -- [ ] **Step 3: migrate-v3.test.ts** +- [x] **Step 3: migrate-v3.test.ts** v2 库 migrate 后 `SELECT name FROM sqlite_master` 含 `ai_sessions`。 -- [ ] **Step 4: types.ts 扩展** +- [x] **Step 4: types.ts 扩展** ```typescript export interface AiConfig { @@ -334,11 +334,11 @@ export const DEFAULT_AI_CONFIG: AiConfig = { `GlobalSettings` 追加 `aiConfig`, `namingIgnoreWords: string[]`。 -- [ ] **Step 5: global-settings defaults()** +- [x] **Step 5: global-settings defaults()** 合并 `aiConfig: DEFAULT_AI_CONFIG`, `namingIgnoreWords: []`。 -- [ ] **Step 6: 验证 + Commit** +- [x] **Step 6: 验证 + Commit** ```bash npm run test -- tests/main/migrate-v3.test.ts @@ -353,7 +353,7 @@ git commit -am "feat: add schema v3 and AI config types" - Create: `src/main/services/ai-client.service.ts` - Create: `tests/main/ai-client.test.ts` -- [ ] **Step 1: 写真实 LM Studio 集成测试（禁止 skip）** +- [x] **Step 1: 写真实 LM Studio 集成测试（禁止 skip）** ```typescript import { describe, it, expect } from 'vitest' @@ -372,13 +372,13 @@ describe('AiClientService', () => { }) ``` -- [ ] **Step 2: 运行确认 FAIL** +- [x] **Step 2: 运行确认 FAIL** ```bash npm run test -- tests/main/ai-client.test.ts ``` -- [ ] **Step 3: 实现 AiClientService** +- [x] **Step 3: 实现 AiClientService** ```typescript export interface ChatMessage { @@ -446,7 +446,7 @@ export class AiClientService { } ``` -- [ ] **Step 4: 验证 PASS（需 LM Studio 运行）** +- [x] **Step 4: 验证 PASS（需 LM Studio 运行）** ```bash npm run test -- tests/main/ai-client.test.ts @@ -470,13 +470,13 @@ git commit -am "feat: add AiClientService with LM Studio OpenAI compatibility" - Modify: `src/main/services/book-registry.ts` - Modify: `src/preload/index.ts`, `src/shared/electron-api.d.ts` -- [ ] **Step 1: ai-session.repo 测试 + 实现** +- [x] **Step 1: ai-session.repo 测试 + 实现** CRUD：`list`, `create`, `update`, `delete`, `listMessages`, `addMessage`。 `context_json` ↔ `AiContextBinding` JSON 序列化。 -- [ ] **Step 2: ai.handler.ts** +- [x] **Step 2: ai.handler.ts** 注册 spec §4.1 全部 `ai:session*` / `ai:messageList`。 @@ -497,11 +497,11 @@ ipcMain.handle(IPC.AI_CHAT, async (event, { bookId, sessionId, content }) => { }) ``` -- [ ] **Step 3: book-registry getAiSessionRepo** +- [x] **Step 3: book-registry getAiSessionRepo** -- [ ] **Step 4: preload 暴露 `window.electronAPI.ai.*`** +- [x] **Step 4: preload 暴露 `window.electronAPI.ai.*`** -- [ ] **Step 5: 验证** +- [x] **Step 5: 验证** ```bash npm run test -- tests/main/ai-session.repo.test.ts @@ -523,7 +523,7 @@ git commit -am "feat: add AI session repository and IPC handlers" - Modify: `src/main/index.ts` - Modify: `src/preload/index.ts` -- [ ] **Step 1: 改造 ai:chat 为流式** +- [x] **Step 1: 改造 ai:chat 为流式** `ai:chat` 内： @@ -540,7 +540,7 @@ repo.addMessage(sessionId, 'assistant', full, assistantId) `ai:abort`：维护 `Map`。 -- [ ] **Step 2: preload 订阅** +- [x] **Step 2: preload 订阅** ```typescript onAiStreamChunk: (cb) => { @@ -550,7 +550,7 @@ onAiStreamChunk: (cb) => { } ``` -- [ ] **Step 3: NetworkMonitor** +- [x] **Step 3: NetworkMonitor** `net.isOnline()` + 每 30s 检测；变化时 `webContents.send(IPC.AI_NETWORK_STATUS, { online })`。 @@ -570,7 +570,7 @@ git commit -am "feat: add AI streaming IPC and network monitor" - Modify: `src/renderer/components/layout/RightPanel.tsx` - Modify: `src/renderer/styles/layout.css` -- [ ] **Step 1: useAiStore** +- [x] **Step 1: useAiStore** ```typescript interface AiStore { @@ -589,21 +589,21 @@ interface AiStore { 挂载时订阅 `onAiStreamChunk` / `onAiNetworkStatus`。 -- [ ] **Step 2: AiPanel 结构** +- [x] **Step 2: AiPanel 结构** 对齐 `ui_pc.html`：`ai-session-bar`、`ai-mode-tabs`（非 chat 调 `showToast(comingSoon)`）、`context-preview`（Task 9）、`offline-banner`、`chat-messages`、`quick-cmds`、`chat-input-area`。 `data-testid`：`ai-panel`, `ai-session-select`, `ai-chat-input`, `ai-send`, `chat-message-assistant`. -- [ ] **Step 3: RightPanel 替换 AI 占位** +- [x] **Step 3: RightPanel 替换 AI 占位** `panel-content` AI Tab 渲染 ``。 -- [ ] **Step 4: E2E E2E-P3-CHAT-01** +- [x] **Step 4: E2E E2E-P3-CHAT-01** `e2e/ai-chat.spec.ts`：打开书 → 新建会话 → 输入「你好」→ Enter → `chat-message-assistant` 非空（timeout 90s）。 -- [ ] **Step 5: 验证 + Commit** +- [x] **Step 5: 验证 + Commit** ```bash npm run test:e2e -- e2e/ai-chat.spec.ts @@ -621,19 +621,19 @@ git commit -am "feat: add AiPanel with streaming chat UI" - Modify: `src/main/ipc/handlers/ai.handler.ts` - Modify: `src/renderer/components/ai/AiPanel.tsx` -- [ ] **Step 1: AiContextBuilder 测试** +- [x] **Step 1: AiContextBuilder 测试** 截断：超长 chapter content → 块 ≤ 2000 字；总量 ≤ 16KB。 -- [ ] **Step 2: 实现 build(binding, bookId)** +- [x] **Step 2: 实现 build(binding, bookId)** 返回 `{ systemPrompt, preview: ContextPreviewItem[] }`。 -- [ ] **Step 3: ContextEditorModal** +- [x] **Step 3: ContextEditorModal** 四 Tab 勾选列表（chapters/outlines/settings/inspirations）；保存 → `ai:sessionUpdate` patch `context_json`；「更新上下文」→ `ai:buildContext`。 -- [ ] **Step 4: 快捷命令** +- [x] **Step 4: 快捷命令** `src/renderer/lib/ai-slash-commands.ts`： @@ -650,7 +650,7 @@ export function resolveSlashCommand(cmd: string, t: TFunction): string | null { 发送前：若匹配 slash，用模板替换实际 prompt。 -- [ ] **Step 5: E2E E2E-P3-CTX-01 / E2E-P3-CMD-01** +- [x] **Step 5: E2E E2E-P3-CTX-01 / E2E-P3-CMD-01** - [ ] **Step 6: Commit** @@ -668,7 +668,7 @@ git commit -am "feat: add AI context editor and slash commands" - Modify: `src/main/ipc/handlers/ai.handler.ts`（`ai:testConnection`） - Modify: `public/locales/*/translation.json` -- [ ] **Step 1: AiSettingsPage** +- [x] **Step 1: AiSettingsPage** 字段：backend select、baseUrl、model、apiKey（非 lmstudio 显示）、timeout。 @@ -676,7 +676,7 @@ git commit -am "feat: add AI context editor and slash commands" 「测试连接」→ `ai:testConnection` → 最小 chat → toast。 -- [ ] **Step 2: Settings 导航增加 AI Tab** +- [x] **Step 2: Settings 导航增加 AI Tab** - [ ] **Step 3: Commit** @@ -696,23 +696,23 @@ git commit -am "feat: add AI settings page with connection test" - Modify: `src/renderer/lib/editor-commands.ts` - Modify: `src/renderer/components/ai/AiPanel.tsx` -- [ ] **Step 1: 离线规则** +- [x] **Step 1: 离线规则** `useAiStore`：`disabled = aiConfig.backend !== 'lmstudio' && !networkOnline`。 `AiPanel` 显示 `#offlineBanner`；网络恢复 toast。 -- [ ] **Step 2: naming-check.service** +- [x] **Step 2: naming-check.service** `buildPrompt(settings, chaptersPlain)` → 调 `AiClientService.chat` → `parseNamingJson(response)`。 解析器单测（不 mock HTTP）：`tests/main/naming-parse.test.ts`。 -- [ ] **Step 3: NamingCheckModal** +- [x] **Step 3: NamingCheckModal** 列表 + 忽略/替换；替换调 `search.replace`。 -- [ ] **Step 4: AI 快照** +- [x] **Step 4: AI 快照** `insertToEditor(text)`： @@ -721,7 +721,7 @@ await ipcCall(() => window.electronAPI.snapshot.create(bookId, chapterId, 'ai', // then insert text at cursor ``` -- [ ] **Step 5: E2E E2E-P3-NAME-01** +- [x] **Step 5: E2E E2E-P3-NAME-01** 种子：设定「林远」+ 正文「林悦」→ 命名检查 → 至少一行结果。 @@ -742,15 +742,15 @@ git commit -am "feat: add offline fallback, naming check, and AI snapshots" - Create: `e2e/ai-chat.spec.ts`（补全 CTX/CMD/OFF） - Extend: `e2e/p21-features.spec.ts` -- [ ] **Step 1: i18n 全量键** +- [x] **Step 1: i18n 全量键** `ai.*`, `naming.*`, `inspiration.voice*`, `search.replace*`, `snapshot.ai` -- [ ] **Step 2: layout.css** +- [x] **Step 2: layout.css** `.ai-session-bar`, `.ai-mode-tabs`, `.chat-msg.user/.ai`, `.quick-cmd`, `.offline-banner`, `.naming-modal`, `.search-replace-panel` -- [ ] **Step 3: 全量测试** +- [x] **Step 3: 全量测试** ```bash npm run test @@ -760,7 +760,7 @@ npm run test:e2e Expected: 全部 PASS（**LM Studio 必须在 127.0.0.1:1234 运行**） -- [ ] **Step 4: package.json version 0.3.0 + README** +- [x] **Step 4: package.json version 0.3.0 + README** - [ ] **Step 5: Commit + tag** diff --git a/e2e/ai-chat.spec.ts b/e2e/ai-chat.spec.ts new file mode 100644 index 0000000..dfa67a3 --- /dev/null +++ b/e2e/ai-chat.spec.ts @@ -0,0 +1,204 @@ +import path from 'path' +import { mkdtempSync, rmSync, existsSync } from 'fs' +import { join } from 'path' +import { tmpdir } from 'os' +import { test, expect, _electron as electron, type Page } from '@playwright/test' + +const ROOT = path.join(import.meta.dirname, '..') + +async function launchApp(userDataDir: string) { + return electron.launch({ + args: [ROOT], + env: { ...process.env, BILIN_E2E: '1', BILIN_E2E_USER_DATA: userDataDir } + }) +} + +async function skipOnboarding(page: Page): Promise { + await page.waitForLoadState('domcontentloaded') + if (await page.getByText('欢迎使用笔临').isVisible()) { + await page.getByRole('button', { name: '跳过' }).click() + } +} + +async function createAndOpenBook(page: Page, name = 'AI测试书'): Promise { + await page.getByRole('button', { name: /新建书籍/ }).first().click() + await page.locator('.dialog-content input').first().fill(name) + await page.getByRole('button', { name: '创建' }).click() + await expect(page.locator('#editor-layout')).toBeVisible({ timeout: 15_000 }) +} + +async function openGlobalSettings(page: Page): Promise { + const topSettings = page.locator('#topbar button.top-btn').filter({ hasText: '⚙' }) + if (await topSettings.isVisible()) { + await topSettings.click() + } else { + await page.getByRole('button', { name: /系统设置/ }).click() + } + await expect(page.locator('#settings-page')).toBeVisible({ timeout: 10_000 }) +} + +async function seedNamingConflict(page: Page): Promise { + await page.getByTestId('sidebar-tab-setting').click() + await page.getByTestId('setting-new').click() + await page.getByTestId('setting-name-input').fill('林远') + await page.locator('.dialog-content select').selectOption('character') + await page.getByRole('button', { name: '创建' }).click() + const settingItem = page.getByTestId(/setting-item-/).first() + await expect(settingItem).toBeVisible({ timeout: 10_000 }) + + const bookId = await page.locator('#editor-layout').getAttribute('data-book-id') + const settingTestId = await settingItem.getAttribute('data-testid') + const settingId = settingTestId?.replace('setting-item-', '') ?? '' + expect(bookId).toBeTruthy() + expect(settingId).toBeTruthy() + + await page.evaluate( + async ({ bookId, settingId }) => { + const result = await window.electronAPI.setting.update(bookId, settingId, { + description: '

正文中出现了林悦这一名字。

' + }) + if (!result.ok) throw new Error(result.error.message) + }, + { bookId: bookId!, settingId } + ) + + const saved = await page.evaluate(async (bookId) => { + const result = await window.electronAPI.setting.list(bookId) + if (!result.ok) return '' + return result.data.find((s) => s.name === '林远')?.description ?? '' + }, bookId!) + expect(saved).toContain('林悦') +} + +test.describe('AI chat', () => { + let userDataDir: string + + test.beforeEach(() => { + userDataDir = mkdtempSync(join(tmpdir(), 'bilin-e2e-ai-')) + }) + + test.afterEach(() => { + if (userDataDir && existsSync(userDataDir)) { + try { + rmSync(userDataDir, { recursive: true, force: true, maxRetries: 3, retryDelay: 200 }) + } catch { + /* ignore */ + } + } + }) + + test('E2E-P3-CHAT-01: send message and receive assistant reply', async () => { + test.setTimeout(120_000) + + const app = await launchApp(userDataDir) + const page = await app.firstWindow() + await skipOnboarding(page) + await createAndOpenBook(page) + + await page.getByTestId('panel-tab-ai').click() + await expect(page.getByTestId('ai-panel')).toBeVisible({ timeout: 10_000 }) + + await page.getByTestId('ai-session-new').click() + await expect(page.getByTestId('ai-session-select')).not.toHaveValue('', { timeout: 10_000 }) + + await page.getByTestId('ai-chat-input').fill('你好') + await page.getByTestId('ai-send').click() + + const assistant = page.getByTestId('chat-message-assistant').last() + await expect(assistant).toBeVisible({ timeout: 90_000 }) + await expect(assistant).not.toBeEmpty() + + await app.close() + }) + + test('E2E-P3-CTX-01: select setting updates context summary', async () => { + const app = await launchApp(userDataDir) + const page = await app.firstWindow() + await skipOnboarding(page) + await createAndOpenBook(page) + + await page.getByTestId('sidebar-tab-setting').click() + await page.getByTestId('setting-new').click() + await page.getByTestId('setting-name-input').fill('主角林远') + await page.locator('.dialog-content select').selectOption('character') + await page.getByRole('button', { name: '创建' }).click() + await expect(page.getByTestId(/setting-item-/)).toBeVisible({ timeout: 10_000 }) + + await page.getByTestId('panel-tab-ai').click() + await page.getByTestId('ai-session-new').click() + await page.getByTestId('ai-context-preview').click() + await expect(page.getByTestId('context-editor-modal')).toBeVisible() + await page.getByTestId('context-tab-setting').click() + const settingItem = page.locator('[data-testid^="context-item-"]').first() + await settingItem.locator('input').check() + await page.getByTestId('context-save').click() + await expect(page.getByTestId('context-editor-modal')).toBeHidden() + await expect(page.getByTestId('ai-context-preview')).toContainText('设定1个', { timeout: 10_000 }) + await app.close() + }) + + test('E2E-P3-CMD-01: slash summarize command sends and receives reply', async () => { + test.setTimeout(120_000) + + const app = await launchApp(userDataDir) + const page = await app.firstWindow() + await skipOnboarding(page) + await createAndOpenBook(page) + + await page.getByTestId('panel-tab-ai').click() + await page.getByTestId('ai-session-new').click() + await page.getByTestId('ai-quick-cmd-总结').click() + await expect(page.getByTestId('ai-send')).toBeEnabled() + await page.getByTestId('ai-send').click() + + await expect(page.getByTestId('chat-message-user').last()).toContainText('/总结', { timeout: 90_000 }) + const assistant = page.getByTestId('chat-message-assistant').last() + await expect(assistant).toBeVisible({ timeout: 90_000 }) + await app.close() + }) + + test('E2E-P3-NAME-01: naming check finds conflict between setting and chapter', async () => { + test.setTimeout(180_000) + + const app = await launchApp(userDataDir) + const page = await app.firstWindow() + await skipOnboarding(page) + await createAndOpenBook(page) + await seedNamingConflict(page) + + await page.getByTestId('panel-tab-knowledge').click() + await page.getByTestId('naming-check-open').click() + await expect(page.getByTestId('naming-check-modal')).toBeVisible() + await page.getByTestId('naming-run-check').click() + + const row = page.getByTestId('naming-row').first() + await expect(row).toBeVisible({ timeout: 30_000 }) + await expect(row).toContainText('林远') + await expect(row).toContainText('林悦') + await app.close() + }) + + test('E2E-P3-OFF-01: cloud backend disables AI input when offline', async () => { + const app = await launchApp(userDataDir) + const page = await app.firstWindow() + await skipOnboarding(page) + await createAndOpenBook(page) + + await openGlobalSettings(page) + await page.getByTestId('settings-nav-ai').click() + await page.getByTestId('ai-settings-backend').selectOption('openai') + await page.getByText('AI测试书').click() + await expect(page.locator('#editor-layout')).toBeVisible({ timeout: 10_000 }) + + await page.getByTestId('panel-tab-ai').click() + await page.getByTestId('ai-session-new').click() + await page.evaluate(() => { + window.dispatchEvent(new CustomEvent('bilin:e2e-set-ai-online', { detail: { online: false } })) + }) + + await expect(page.getByTestId('ai-offline-banner')).toBeVisible({ timeout: 10_000 }) + await expect(page.getByTestId('ai-chat-input')).toBeDisabled() + await expect(page.getByTestId('ai-send')).toBeDisabled() + await app.close() + }) +}) diff --git a/e2e/p21-features.spec.ts b/e2e/p21-features.spec.ts new file mode 100644 index 0000000..29ecb88 --- /dev/null +++ b/e2e/p21-features.spec.ts @@ -0,0 +1,157 @@ +import path from 'path' +import { mkdtempSync, rmSync, existsSync } from 'fs' +import { join } from 'path' +import { tmpdir } from 'os' +import { test, expect, _electron as electron, type Page } from '@playwright/test' + +const ROOT = path.join(import.meta.dirname, '..') + +async function launchApp(userDataDir: string) { + return electron.launch({ + args: [ROOT], + env: { ...process.env, BILIN_E2E: '1', BILIN_E2E_USER_DATA: userDataDir } + }) +} + +async function skipOnboarding(page: Page): Promise { + await page.waitForLoadState('domcontentloaded') + if (await page.getByText('欢迎使用笔临').isVisible()) { + await page.getByRole('button', { name: '跳过' }).click() + } +} + +async function createAndOpenBook(page: Page): Promise { + await page.getByRole('button', { name: /新建书籍/ }).first().click() + await page.locator('.dialog-content input').first().fill('拖拽测试') + await page.getByRole('button', { name: '创建' }).click() + await expect(page.locator('#editor-layout')).toBeVisible({ timeout: 15_000 }) +} + +async function ensureChapterEditor(page: Page): Promise { + await page.getByRole('button', { name: '+ 新章' }).click() + await expect(page.locator('.ProseMirror')).toBeVisible({ timeout: 10_000 }) +} + +async function dragChapterItem(page: Page, draggedId: string, targetId: string): Promise { + await page.evaluate( + ({ draggedId, targetId }) => { + const source = document.querySelector(`[data-testid="chapter-item-${draggedId}"]`) + const target = document.querySelector(`[data-testid="chapter-item-${targetId}"]`) + if (!source || !target) throw new Error('chapter elements not found') + const dt = new DataTransfer() + dt.setData('application/x-bilin-chapter', draggedId) + source.dispatchEvent(new DragEvent('dragstart', { bubbles: true, dataTransfer: dt })) + target.dispatchEvent(new DragEvent('dragover', { bubbles: true, cancelable: true, dataTransfer: dt })) + target.dispatchEvent(new DragEvent('drop', { bubbles: true, cancelable: true, dataTransfer: dt })) + }, + { draggedId, targetId } + ) +} + +async function chapterIds(page: Page): Promise { + return page.locator('.chapter-item').evaluateAll((els) => + els.map((el) => el.getAttribute('data-testid')?.replace('chapter-item-', '') ?? '') + ) +} + +test.describe('P2.1 features', () => { + let userDataDir: string + + test.beforeEach(() => { + userDataDir = mkdtempSync(join(tmpdir(), 'bilin-e2e-p21-')) + }) + + test.afterEach(() => { + if (userDataDir && existsSync(userDataDir)) { + try { + rmSync(userDataDir, { recursive: true, force: true, maxRetries: 3, retryDelay: 200 }) + } catch { + /* ignore */ + } + } + }) + + test('E2E-P21-DRAG-01: drag chapter reorders list and persists', async () => { + const app1 = await launchApp(userDataDir) + const page1 = await app1.firstWindow() + await skipOnboarding(page1) + await createAndOpenBook(page1) + await page1.getByRole('button', { name: '+ 新章' }).click() + await page1.getByRole('button', { name: '+ 新章' }).click() + + const items = page1.locator('.chapter-item') + await expect(items).toHaveCount(2, { timeout: 10_000 }) + const idsBefore = await chapterIds(page1) + await dragChapterItem(page1, idsBefore[1], idsBefore[0]) + + await expect.poll(async () => (await chapterIds(page1)).join(',')).not.toBe(idsBefore.join(',')) + + const idsAfter = await chapterIds(page1) + await app1.close() + + const app2 = await launchApp(userDataDir) + const page2 = await app2.firstWindow() + await page2.getByText('拖拽测试').click() + await expect(page2.locator('#editor-layout')).toBeVisible({ timeout: 15_000 }) + const titlesRestart = await chapterIds(page2) + expect(titlesRestart).toEqual(idsAfter) + await app2.close() + }) + + test('E2E-P21-REP-01: search replace preview and execute updates editor', async () => { + const app = await launchApp(userDataDir) + const page = await app.firstWindow() + page.on('dialog', (dialog) => void dialog.accept()) + await skipOnboarding(page) + await createAndOpenBook(page) + await ensureChapterEditor(page) + + const oldWord = '替换专用词OLD' + const newWord = '替换专用词NEW' + const editor = page.locator('.ProseMirror') + await editor.click() + await editor.pressSequentially(oldWord) + await page.keyboard.press('Control+S') + await expect(page.getByText('已保存')).toBeVisible({ timeout: 10_000 }) + + await page.evaluate(() => window.dispatchEvent(new Event('bilin:e2e-open-search'))) + await expect(page.getByTestId('search-modal')).toBeVisible({ timeout: 5_000 }) + await page.getByTestId('search-input').fill(oldWord) + await expect(page.locator('.search-result').first()).toBeVisible({ timeout: 10_000 }) + + await page.getByTestId('search-replace-panel').locator('summary').click() + await page.getByTestId('search-replace-input').fill(newWord) + await page.getByTestId('search-replace-preview').click() + await expect(page.getByTestId('search-replace-previews')).toBeVisible({ timeout: 10_000 }) + + await page.getByTestId('search-replace-execute').click() + await expect(page.getByText(/已替换/)).toBeVisible({ timeout: 10_000 }) + await expect(editor).toContainText(newWord, { timeout: 10_000 }) + await expect(editor).not.toContainText(oldWord) + await app.close() + }) + + test('E2E-P21-INSP-01: capture inspiration shortcut opens modal and saves', async () => { + const app = await launchApp(userDataDir) + const page = await app.firstWindow() + await skipOnboarding(page) + await createAndOpenBook(page) + await ensureChapterEditor(page) + + await page.keyboard.press('Control+Shift+I') + const modal = page.getByTestId('inspiration-modal') + if (!(await modal.isVisible().catch(() => false))) { + await page.evaluate(() => window.dispatchEvent(new Event('bilin:e2e-capture-inspiration'))) + } + await expect(modal).toBeVisible({ timeout: 5_000 }) + + const idea = '灵感捕获测试内容ABC' + await page.getByTestId('inspiration-content-input').fill(idea) + await page.getByTestId('inspiration-save-btn').click() + await expect(page.getByTestId('inspiration-modal')).toBeHidden({ timeout: 10_000 }) + + await expect(page.getByTestId('inspiration-list')).toBeVisible() + await expect(page.getByTestId('inspiration-list').getByText(idea)).toBeVisible({ timeout: 10_000 }) + await app.close() + }) +}) diff --git a/package.json b/package.json index c087994..77ed58a 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "bilin", - "version": "0.2.0", + "version": "0.3.0", "description": "笔临 - 长篇创作智能协作平台", "main": "./out/main/index.js", "type": "module", diff --git a/public/locales/en/translation.json b/public/locales/en/translation.json index cb41ac4..9e024d5 100644 --- a/public/locales/en/translation.json +++ b/public/locales/en/translation.json @@ -62,6 +62,60 @@ "panel.wordfreq": "Word Freq", "panel.bookSettings": "Book", "panel.aiPlaceholder": "AI features coming soon. Configure a model to collaborate here.", + "ai.mode.chat": "Chat", + "ai.mode.interactive": "Interactive", + "ai.mode.auto": "Auto", + "ai.mode.wizard": "Wizard", + "ai.noBook": "Open a book first", + "ai.noSession": "No session", + "ai.untitledSession": "New chat", + "ai.contextPlaceholder": "📎 Context: not configured yet", + "ai.contextEmpty": "📎 Context: nothing selected", + "ai.contextSummary": "📎 Context: {{chapters}} ch. · {{outlines}} outline · {{settings}} settings · {{inspirations}} ideas", + "ai.contextEditorTitle": "Edit AI context", + "ai.contextTab.chapter": "Chapters", + "ai.contextTab.outline": "Outline", + "ai.contextTab.setting": "Settings", + "ai.contextTab.inspiration": "Ideas", + "ai.contextEmptyList": "No items", + "ai.contextRefresh": "Refresh context", + "ai.contextSave": "Save", + "ai.slash.continue": "Continue writing 200-400 characters from the end of the current chapter in the same style.", + "ai.slash.expand": "Expand the most recent paragraph with more sensory detail without changing the plot.", + "ai.slash.polish": "Polish the following text for prose quality while keeping plot and POV unchanged.", + "ai.slash.summarize": "Summarize the key points of the current chapter, including events and character changes.", + "ai.slash.proofread": "Review grammar, punctuation, and wording issues in the following text and suggest fixes.", + "ai.offlineBanner": "⚠ Network unavailable. Offline mode enabled.", + "ai.inputPlaceholder": "Type a message or pick a quick command…", + "ai.send": "Send", + "ai.networkRestored": "Network restored. Cloud AI is available again.", + "ai.insertToEditor": "Insert into editor", + "ai.inserted": "Inserted into editor", + "snapshot.ai": "Before AI insert", + "naming.open": "🔍 Naming consistency check", + "naming.title": "Naming consistency check", + "naming.run": "Run scan", + "naming.running": "Scanning…", + "naming.emptyHint": "Click Run scan to check proper noun consistency across the book", + "naming.noConflicts": "No naming conflicts found", + "naming.confidence": "Confidence {{value}}%", + "naming.ignore": "Ignore", + "naming.replace": "Replace with “{{term}}”", + "naming.ignored": "Added to ignore list", + "naming.replaced": "Replaced {{count}} occurrence(s)", + "knowledge.placeholder": "Knowledge base entries will appear here", + "ai.settings.backend": "Backend", + "ai.settings.backend.lmstudio": "LM Studio (local)", + "ai.settings.backend.openai": "OpenAI", + "ai.settings.backend.anthropic": "Anthropic", + "ai.settings.baseUrl": "Base URL", + "ai.settings.model": "Model", + "ai.settings.apiKey": "API Key", + "ai.settings.timeout": "Timeout (seconds)", + "ai.settings.testConnection": "Test connection", + "ai.settings.testing": "Testing…", + "ai.settings.testSuccess": "AI connection test succeeded", + "ai.settings.licenseNotice": "Cloud AI services are subject to each provider's terms. Please read and agree to the OpenAI Terms / Anthropic Terms before use.", "status.chapter": "Chapter", "status.volume": "Volume", "status.book": "Book", @@ -103,6 +157,14 @@ "inspiration.empty": "No ideas yet", "inspiration.untitled": "Untitled", "inspiration.noContent": "(empty)", + "inspiration.captureTitle": "💡 Capture Idea", + "inspiration.titlePlaceholder": "Title (optional)", + "inspiration.contentPlaceholder": "Capture a fleeting thought…", + "inspiration.tagsLabel": "Tags (optional)", + "inspiration.tagsPlaceholder": "e.g. foreshadowing, vol.3, character", + "inspiration.voice": "Voice", + "inspiration.voiceUnavailable": "Speech input is not available", + "inspiration.save": "Save", "outline.filterAll": "All", "outline.filterUncovered": "Uncovered", "outline.filterCovered": "Covered", @@ -121,10 +183,17 @@ "reference.tab.setting": "Settings", "reference.tab.outline": "Outline", "reference.tab.inspiration": "Ideas", + "reference.pin": "Pin panel", + "reference.unpin": "Unpin", "search.title": "Global Search", "search.placeholder": "Search…", "search.regex": "Regex", "search.caseSensitive": "Case sensitive", + "search.replace": "Replace with", + "search.replacePreview": "Preview replace", + "search.replaceExecute": "Replace all", + "search.replaceConfirm": "Replace all matches in this book? This cannot be undone.", + "search.replaceDone": "Replaced {{count}} occurrence(s)", "version.title": "Version History", "version.empty": "No snapshots", "version.create": "Create snapshot", diff --git a/public/locales/zh-CN/translation.json b/public/locales/zh-CN/translation.json index 4ed0737..9f5752a 100644 --- a/public/locales/zh-CN/translation.json +++ b/public/locales/zh-CN/translation.json @@ -62,6 +62,60 @@ "panel.wordfreq": "用词", "panel.bookSettings": "设置", "panel.aiPlaceholder": "AI 功能即将推出。配置模型后，可在此与笔临 AI 协作写作。", + "ai.mode.chat": "对话", + "ai.mode.interactive": "交互写作", + "ai.mode.auto": "自动写作", + "ai.mode.wizard": "向导", + "ai.noBook": "请先打开一本书", + "ai.noSession": "暂无会话", + "ai.untitledSession": "新对话", + "ai.contextPlaceholder": "📎 上下文：尚未配置", + "ai.contextEmpty": "📎 上下文：尚未勾选", + "ai.contextSummary": "📎 上下文：{{chapters}}章 · 大纲{{outlines}}条 · 设定{{settings}}个 · 灵感{{inspirations}}条", + "ai.contextEditorTitle": "编辑 AI 上下文", + "ai.contextTab.chapter": "章节", + "ai.contextTab.outline": "大纲", + "ai.contextTab.setting": "设定", + "ai.contextTab.inspiration": "灵感", + "ai.contextEmptyList": "暂无可选项", + "ai.contextRefresh": "更新上下文", + "ai.contextSave": "保存", + "ai.slash.continue": "请基于当前章节末尾续写 200-400 字，保持文风一致。", + "ai.slash.expand": "请扩展最近一段描写，增加细节与画面感，不改变情节走向。", + "ai.slash.polish": "请润色以下文本，改善文笔与节奏，保持情节与人称不变。", + "ai.slash.summarize": "请总结当前章节要点，列出关键事件与人物变化。", + "ai.slash.proofread": "请检查以下文本的语法、标点与用词问题，并给出修改建议。", + "ai.offlineBanner": "⚠ 当前网络不可用，已切换为离线模式。请配置本地模型或连接网络。", + "ai.inputPlaceholder": "输入指令或选择上方快捷命令…", + "ai.send": "发送", + "ai.networkRestored": "网络已恢复，可继续使用云端 AI", + "ai.insertToEditor": "插入到编辑器", + "ai.inserted": "已插入到编辑器", + "snapshot.ai": "AI 生成前", + "naming.open": "🔍 命名一致性检查", + "naming.title": "命名一致性检查", + "naming.run": "开始扫描", + "naming.running": "扫描中…", + "naming.emptyHint": "点击「开始扫描」检查全书专有名词一致性", + "naming.noConflicts": "未发现命名冲突", + "naming.confidence": "置信度 {{value}}%", + "naming.ignore": "忽略", + "naming.replace": "替换为「{{term}}」", + "naming.ignored": "已加入忽略词库", + "naming.replaced": "已替换 {{count}} 处", + "knowledge.placeholder": "知识库条目将在此展示（伏笔、角色状态等）", + "ai.settings.backend": "后端类型", + "ai.settings.backend.lmstudio": "LM Studio（本地）", + "ai.settings.backend.openai": "OpenAI", + "ai.settings.backend.anthropic": "Anthropic", + "ai.settings.baseUrl": "服务地址", + "ai.settings.model": "模型", + "ai.settings.apiKey": "API Key", + "ai.settings.timeout": "超时（秒）", + "ai.settings.testConnection": "测试连接", + "ai.settings.testing": "测试中…", + "ai.settings.testSuccess": "AI 连接测试成功", + "ai.settings.licenseNotice": "使用云端 AI 服务将受到相应模型提供商的许可协议约束。请在使用前阅读并同意 OpenAI 服务条款 / Anthropic 服务条款 等。", "status.chapter": "本章", "status.volume": "本卷", "status.book": "全书", @@ -103,6 +157,14 @@ "inspiration.empty": "暂无灵感", "inspiration.untitled": "未命名", "inspiration.noContent": "（无内容）", + "inspiration.captureTitle": "💡 灵感快速捕获", + "inspiration.titlePlaceholder": "标题（可选）", + "inspiration.contentPlaceholder": "记录一闪而过的想法…", + "inspiration.tagsLabel": "标签（可选）", + "inspiration.tagsPlaceholder": "如：伏笔、第三卷、角色", + "inspiration.voice": "语音", + "inspiration.voiceUnavailable": "当前环境不支持语音输入", + "inspiration.save": "保存", "outline.filterAll": "全部", "outline.filterUncovered": "未覆盖", "outline.filterCovered": "已覆盖", @@ -121,10 +183,17 @@ "reference.tab.setting": "设定", "reference.tab.outline": "大纲", "reference.tab.inspiration": "灵感", + "reference.pin": "钉住面板", + "reference.unpin": "取消钉住", "search.title": "全局搜索", "search.placeholder": "输入搜索词…", "search.regex": "正则", "search.caseSensitive": "区分大小写", + "search.replace": "替换为", + "search.replacePreview": "预览替换", + "search.replaceExecute": "执行替换", + "search.replaceConfirm": "确定要在全书范围内执行替换吗？此操作不可撤销。", + "search.replaceDone": "已替换 {{count}} 处", "version.title": "版本历史", "version.empty": "暂无快照", "version.create": "创建快照", diff --git a/src/main/db/migrate.ts b/src/main/db/migrate.ts index 9836745..f92ec69 100644 --- a/src/main/db/migrate.ts +++ b/src/main/db/migrate.ts @@ -1,8 +1,9 @@ import type { SqliteDb } from './types' import schemaV1 from './schema-v1.sql?raw' import schemaV2 from './schema-v2.sql?raw' +import schemaV3 from './schema-v3.sql?raw' -const CURRENT_VERSION = 2 +const CURRENT_VERSION = 3 function tryAlter(db: SqliteDb, sql: string): void { try { @@ -41,5 +42,11 @@ export function migrate(db: SqliteDb): void { if (current < 2) { applyV2(db) db.prepare('INSERT INTO schema_version (version, description) VALUES (?, ?)').run(2, 'P2 schema') + current = 2 + } + + if (current < 3) { + db.exec(schemaV3) + db.prepare('INSERT INTO schema_version (version, description) VALUES (?, ?)').run(3, 'P3 AI schema') } } diff --git a/src/main/db/repositories/ai-session.repo.ts b/src/main/db/repositories/ai-session.repo.ts new file mode 100644 index 0000000..a4ed508 --- /dev/null +++ b/src/main/db/repositories/ai-session.repo.ts @@ -0,0 +1,125 @@ +import { randomUUID } from 'crypto' +import type { AiContextBinding, AiMessage, AiMessageRole, AiSession } from '../../../shared/types' +import type { SqliteDb } from '../types' + +const EMPTY_CONTEXT: AiContextBinding = { + chapterIds: [], + outlineIds: [], + settingIds: [], + inspirationIds: [] +} + +function parseContext(json: string): AiContextBinding { + try { + const parsed = JSON.parse(json) as Partial + return { + chapterIds: parsed.chapterIds ?? [], + outlineIds: parsed.outlineIds ?? [], + settingIds: parsed.settingIds ?? [], + inspirationIds: parsed.inspirationIds ?? [] + } + } catch { + return { ...EMPTY_CONTEXT } + } +} + +function mapSession(row: Record): AiSession { + return { + id: row.id as string, + title: row.title as string, + model: row.model as string, + archived: Boolean(row.archived), + context: parseContext((row.context_json as string) || '{}'), + createdAt: row.created_at as string, + updatedAt: row.updated_at as string + } +} + +function mapMessage(row: Record): AiMessage { + return { + id: row.id as string, + sessionId: row.session_id as string, + role: row.role as AiMessageRole, + content: row.content as string, + createdAt: row.created_at as string + } +} + +export class AiSessionRepository { + constructor(private db: SqliteDb) {} + + list(includeArchived = false): AiSession[] { + const rows = includeArchived + ? (this.db.prepare('SELECT * FROM ai_sessions ORDER BY updated_at DESC').all() as Record[]) + : (this.db + .prepare('SELECT * FROM ai_sessions WHERE archived = 0 ORDER BY updated_at DESC') + .all() as Record[]) + return rows.map(mapSession) + } + + get(id: string): AiSession | null { + const row = this.db.prepare('SELECT * FROM ai_sessions WHERE id = ?').get(id) as + | Record + | undefined + return row ? mapSession(row) : null + } + + create(title = '', model = ''): AiSession { + const id = randomUUID() + this.db + .prepare( + `INSERT INTO ai_sessions (id, title, model, archived, context_json) VALUES (?, ?, ?, 0, ?)` + ) + .run(id, title, model, JSON.stringify(EMPTY_CONTEXT)) + return this.get(id)! + } + + update( + id: string, + patch: Partial<{ title: string; model: string; archived: boolean; context: AiContextBinding }> + ): AiSession { + const existing = this.get(id) + if (!existing) throw new Error('AI session not found') + + this.db + .prepare( + `UPDATE ai_sessions SET title = ?, model = ?, archived = ?, context_json = ?, updated_at = datetime('now') WHERE id = ?` + ) + .run( + patch.title ?? existing.title, + patch.model ?? existing.model, + patch.archived !== undefined ? (patch.archived ? 1 : 0) : existing.archived ? 1 : 0, + JSON.stringify(patch.context ?? existing.context), + id + ) + return this.get(id)! + } + + delete(id: string): void { + this.db.prepare('DELETE FROM ai_messages WHERE session_id = ?').run(id) + this.db.prepare('DELETE FROM ai_sessions WHERE id = ?').run(id) + } + + listMessages(sessionId: string): AiMessage[] { + return ( + this.db + .prepare('SELECT * FROM ai_messages WHERE session_id = ? ORDER BY created_at ASC') + .all(sessionId) as Record[] + ).map(mapMessage) + } + + addMessage(sessionId: string, role: AiMessageRole, content: string, id?: string): AiMessage { + const messageId = id ?? randomUUID() + this.db + .prepare(`INSERT INTO ai_messages (id, session_id, role, content) VALUES (?, ?, ?, ?)`) + .run(messageId, sessionId, role, content) + this.db + .prepare(`UPDATE ai_sessions SET updated_at = datetime('now') WHERE id = ?`) + .run(sessionId) + const row = this.db.prepare('SELECT * FROM ai_messages WHERE id = ?').get(messageId) as Record< + string, + unknown + > + return mapMessage(row) + } +} diff --git a/src/main/db/repositories/chapter.repo.ts b/src/main/db/repositories/chapter.repo.ts index 2692466..662ea12 100644 --- a/src/main/db/repositories/chapter.repo.ts +++ b/src/main/db/repositories/chapter.repo.ts @@ -88,4 +88,45 @@ export class ChapterRepository { .get(volumeId) as { maxOrder: number | null } return (row?.maxOrder ?? -1) + 1 } + + reorderVolume(volumeId: string, orderedIds: string[]): Chapter[] { + const stmt = this.db.prepare( + `UPDATE chapters SET sort_order = ?, updated_at = datetime('now') WHERE id = ? AND volume_id = ?` + ) + orderedIds.forEach((id, idx) => stmt.run(idx, id, volumeId)) + return this.listByVolume(volumeId) + } + + moveToVolume(chapterId: string, targetVolumeId: string, targetIndex: number): Chapter { + const chapter = this.get(chapterId) + if (!chapter) throw new Error('Chapter not found') + const sourceVolumeId = chapter.volumeId + if (!sourceVolumeId) throw new Error('Chapter has no volume') + + if (sourceVolumeId === targetVolumeId) { + const ids = this.listByVolume(sourceVolumeId).map((c) => c.id) + const from = ids.indexOf(chapterId) + if (from < 0) throw new Error('Chapter not in volume') + ids.splice(from, 1) + ids.splice(Math.max(0, Math.min(targetIndex, ids.length)), 0, chapterId) + this.reorderVolume(sourceVolumeId, ids) + return this.get(chapterId)! + } + + const sourceIds = this.listByVolume(sourceVolumeId) + .map((c) => c.id) + .filter((id) => id !== chapterId) + this.reorderVolume(sourceVolumeId, sourceIds) + + const targetList = this.listByVolume(targetVolumeId) + const insertAt = Math.max(0, Math.min(targetIndex, targetList.length)) + const targetIds = targetList.map((c) => c.id) + targetIds.splice(insertAt, 0, chapterId) + + this.db + .prepare(`UPDATE chapters SET volume_id = ?, updated_at = datetime('now') WHERE id = ?`) + .run(targetVolumeId, chapterId) + this.reorderVolume(targetVolumeId, targetIds) + return this.get(chapterId)! + } } diff --git a/src/main/db/schema-v3.sql b/src/main/db/schema-v3.sql new file mode 100644 index 0000000..18303f7 --- /dev/null +++ b/src/main/db/schema-v3.sql @@ -0,0 +1,20 @@ +CREATE TABLE IF NOT EXISTS ai_sessions ( + id TEXT PRIMARY KEY, + title TEXT NOT NULL DEFAULT '', + model TEXT NOT NULL DEFAULT '', + archived INTEGER NOT NULL DEFAULT 0, + context_json TEXT NOT NULL DEFAULT '{}', + created_at TEXT NOT NULL DEFAULT (datetime('now')), + updated_at TEXT NOT NULL DEFAULT (datetime('now')) +); + +CREATE TABLE IF NOT EXISTS ai_messages ( + id TEXT PRIMARY KEY, + session_id TEXT NOT NULL, + role TEXT NOT NULL CHECK (role IN ('user', 'assistant', 'system')), + content TEXT NOT NULL DEFAULT '', + created_at TEXT NOT NULL DEFAULT (datetime('now')), + FOREIGN KEY (session_id) REFERENCES ai_sessions(id) ON DELETE CASCADE +); + +CREATE INDEX IF NOT EXISTS idx_ai_messages_session ON ai_messages(session_id, created_at); diff --git a/src/main/index.ts b/src/main/index.ts index 17ea336..992c8ed 100644 --- a/src/main/index.ts +++ b/src/main/index.ts @@ -2,7 +2,7 @@ import { app, BrowserWindow, ipcMain } from 'electron' import { existsSync } from 'fs' import { join } from 'path' import { IPC } from '../shared/ipc-channels' -import { getShortcutManager, getSnapshotService, registerIpc } from './ipc/register' +import { getShortcutManager, getSnapshotService, getNetworkMonitor, registerIpc } from './ipc/register' import { closeAllBookDbs } from './db/connection' if (process.env.BILIN_E2E === '1' && process.env.BILIN_E2E_USER_DATA) { @@ -38,11 +38,16 @@ function createWindow(): void { mainWindow = new BrowserWindow({ mainWindow.loadFile(join(__dirname, '../renderer/index.html')) } + if (process.env.BILIN_E2E === '1') { + mainWindow.webContents.setDevToolsEnabled(false) + } + mainWindow.on('closed', () => { mainWindow = null }) getShortcutManager()?.setWindow(mainWindow) + getNetworkMonitor()?.start(() => mainWindow) } app.whenReady().then(() => { @@ -66,6 +71,7 @@ app.whenReady().then(() => { app.on('window-all-closed', () => { getShortcutManager()?.unregisterAll() + getNetworkMonitor()?.stop() getSnapshotService()?.stopAll() closeAllBookDbs() if (process.platform !== 'darwin') app.quit() @@ -81,4 +87,5 @@ app.on('before-quit', () => { app.on('will-quit', () => { getShortcutManager()?.unregisterAll() + getNetworkMonitor()?.stop() }) diff --git a/src/main/ipc/handlers/ai.handler.ts b/src/main/ipc/handlers/ai.handler.ts new file mode 100644 index 0000000..21158c9 --- /dev/null +++ b/src/main/ipc/handlers/ai.handler.ts @@ -0,0 +1,220 @@ +import { randomUUID } from 'crypto' +import { BrowserWindow, ipcMain } from 'electron' +import { IPC } from '../../../shared/ipc-channels' +import type { AiConfig, AiContextBinding, NamingConflict } from '../../../shared/types' +import { BookRegistryService } from '../../services/book-registry' +import { GlobalSettingsService } from '../../services/global-settings' +import { AiClientService } from '../../services/ai-client.service' +import { aiContextBuilder } from '../../services/ai-context-builder.service' +import { + buildNamingPrompt, + buildNamingCorpus, + collectChaptersPlain, + findLocalNamingConflicts, + mergeNamingConflicts, + parseNamingJson +} from '../../services/naming-check.service' +import type { ChatMessage } from '../../services/ai-client.service' +import { wrap } from '../result' + +const activeChats = new Map() + +export function registerAiHandlers( + registry: BookRegistryService, + settings: GlobalSettingsService, + aiClient: AiClientService +): void { + ipcMain.handle( + IPC.AI_SESSION_LIST, + (_e, { bookId, includeArchived }: { bookId: string; includeArchived?: boolean }) => + wrap(() => registry.getAiSessionRepo(bookId).list(includeArchived)) + ) + + ipcMain.handle( + IPC.AI_SESSION_CREATE, + (_e, { bookId, title, model }: { bookId: string; title?: string; model?: string }) => + wrap(() => { + const config = settings.get().aiConfig + return registry.getAiSessionRepo(bookId).create(title, model ?? config.model) + }) + ) + + ipcMain.handle( + IPC.AI_SESSION_UPDATE, + ( + _e, + { + bookId, + sessionId, + patch + }: { + bookId: string + sessionId: string + patch: Partial<{ title: string; model: string; archived: boolean; context: AiContextBinding }> + } + ) => wrap(() => registry.getAiSessionRepo(bookId).update(sessionId, patch)) + ) + + ipcMain.handle( + IPC.AI_SESSION_DELETE, + (_e, { bookId, sessionId }: { bookId: string; sessionId: string }) => + wrap(() => registry.getAiSessionRepo(bookId).delete(sessionId)) + ) + + ipcMain.handle( + IPC.AI_MESSAGE_LIST, + (_e, { bookId, sessionId }: { bookId: string; sessionId: string }) => + wrap(() => registry.getAiSessionRepo(bookId).listMessages(sessionId)) + ) + + ipcMain.handle( + IPC.AI_BUILD_CONTEXT, + (_e, { bookId, binding }: { bookId: string; binding: AiContextBinding }) => + wrap(() => aiContextBuilder.build(binding, registry, bookId, settings.get().penName)) + ) + + ipcMain.handle(IPC.AI_TEST_CONNECTION, (_e, { config }: { config?: AiConfig }) => + wrap(async () => { + const effective = config ?? settings.get().aiConfig + const probe = new AiClientService(() => effective) + const content = await probe.chat([{ role: 'user', content: '回复一个字：好' }]) + if (!content.trim()) throw new Error('AI returned empty response') + return { ok: true as const } + }) + ) + + ipcMain.handle(IPC.AI_NAMING_CHECK, (_e, { bookId }: { bookId: string }) => + wrap(async () => { + const settingsList = registry.getSettingRepo(bookId).list() + const chapters = registry.getChapterRepo(bookId).list() + const ignoreWords = settings.get().namingIgnoreWords + const settingNames = settingsList.map((s) => s.name) + const corpus = buildNamingCorpus(settingsList, chapters) + const prompt = buildNamingPrompt(settingNames, corpus, ignoreWords) + const local = findLocalNamingConflicts(settingNames, corpus, ignoreWords) + if (local.length > 0) { + try { + const response = await Promise.race([ + aiClient.chat([{ role: 'user', content: prompt }]), + new Promise((_, reject) => + setTimeout(() => reject(new Error('ai-timeout')), 8_000) + ) + ]) + return mergeNamingConflicts(local, parseNamingJson(response)) + } catch { + return local + } + } + + const fetchAi = async (): Promise => { + const response = await aiClient.chat([{ role: 'user', content: prompt }]) + return parseNamingJson(response) + } + + try { + const ai = await fetchAi() + return mergeNamingConflicts(local, ai) + } catch { + try { + const response = await aiClient.chat([ + { role: 'user', content: `${prompt}\n\n仅返回 JSON 数组，不要 markdown 或其它说明。` } + ]) + return mergeNamingConflicts(local, parseNamingJson(response)) + } catch { + if (local.length > 0) return local + throw new Error('命名检查失败') + } + } + }) + ) + + ipcMain.handle(IPC.AI_ABORT, (_e, { messageId }: { messageId: string }) => + wrap(() => { + activeChats.get(messageId)?.abort() + activeChats.delete(messageId) + }) + ) + + ipcMain.handle( + IPC.AI_CHAT, + ( + event, + { + bookId, + sessionId, + content, + prompt + }: { bookId: string; sessionId: string; content: string; prompt?: string; slashCommand?: string } + ) => + wrap(async () => { + const repo = registry.getAiSessionRepo(bookId) + const session = repo.get(sessionId) + if (!session) throw new Error('AI session not found') + + const actualPrompt = prompt ?? content + repo.addMessage(sessionId, 'user', content) + const history = repo.listMessages(sessionId) + + const { systemPrompt } = aiContextBuilder.build( + session.context, + registry, + bookId, + settings.get().penName + ) + + const apiMessages: ChatMessage[] = [ + { role: 'system', content: systemPrompt }, + ...history.map((m, index) => ({ + role: m.role, + content: + index === history.length - 1 && m.role === 'user' ? actualPrompt : m.content + })) + ] + + const assistantId = randomUUID() + const controller = new AbortController() + activeChats.set(assistantId, controller) + const win = BrowserWindow.fromWebContents(event.sender) + + let partial = '' + try { + const full = await aiClient.chat( + apiMessages, + (delta) => { + partial += delta + win?.webContents.send(IPC.AI_STREAM_CHUNK, { + messageId: assistantId, + delta, + done: false + }) + }, + controller.signal + ) + win?.webContents.send(IPC.AI_STREAM_CHUNK, { + messageId: assistantId, + delta: '', + done: true + }) + repo.addMessage(sessionId, 'assistant', full, assistantId) + } catch (err) { + if (controller.signal.aborted && partial) { + win?.webContents.send(IPC.AI_STREAM_CHUNK, { + messageId: assistantId, + delta: '', + done: true + }) + repo.addMessage(sessionId, 'assistant', partial, assistantId) + return { messageId: assistantId } + } + throw err + } finally { + activeChats.delete(assistantId) + } + + if (!session.title.trim()) { + repo.update(sessionId, { title: content.slice(0, 40) }) + } + return { messageId: assistantId } + }) + ) +} diff --git a/src/main/ipc/handlers/chapter.handler.ts b/src/main/ipc/handlers/chapter.handler.ts index 6ecbd9a..7f10bf6 100644 --- a/src/main/ipc/handlers/chapter.handler.ts +++ b/src/main/ipc/handlers/chapter.handler.ts @@ -104,4 +104,24 @@ export function registerChapterHandlers(registry: BookRegistryService): void { ftsSync.remove(registry.getDb(bookId), 'chapter', chapterId) }) ) + + ipcMain.handle( + IPC.CHAPTER_REORDER, + (_event, { bookId, volumeId, orderedIds }: { bookId: string; volumeId: string; orderedIds: string[] }) => + wrap(() => registry.getChapterRepo(bookId).reorderVolume(volumeId, orderedIds)) + ) + + ipcMain.handle( + IPC.CHAPTER_MOVE_TO_VOLUME, + ( + _event, + { + bookId, + chapterId, + targetVolumeId, + targetIndex + }: { bookId: string; chapterId: string; targetVolumeId: string; targetIndex: number } + ) => + wrap(() => registry.getChapterRepo(bookId).moveToVolume(chapterId, targetVolumeId, targetIndex)) + ) } diff --git a/src/main/ipc/register.ts b/src/main/ipc/register.ts index e8dea14..0e9a529 100644 --- a/src/main/ipc/register.ts +++ b/src/main/ipc/register.ts @@ -14,9 +14,13 @@ import { registerInspirationHandlers } from './handlers/inspiration.handler' import { registerSnapshotHandlers } from './handlers/snapshot.handler' import { registerBookmarkHandlers } from './handlers/bookmark.handler' import { registerSearchHandlers } from './handlers/search.handler' +import { registerAiHandlers } from './handlers/ai.handler' +import { AiClientService } from '../services/ai-client.service' +import { NetworkMonitorService } from '../services/network-monitor.service' let shortcutManager: ShortcutManager | null = null let snapshotService: SnapshotService | null = null +let networkMonitor: NetworkMonitorService | null = null export function registerIpc(): { settings: GlobalSettingsService; books: BookRegistryService; shortcuts: ShortcutManager } { const userData = app.getPath('userData') @@ -25,6 +29,7 @@ export function registerIpc(): { settings: GlobalSettingsService; books: BookReg const books = new BookRegistryService(userData) shortcutManager = new ShortcutManager(settings) snapshotService = new SnapshotService(() => settings.get()) + networkMonitor = new NetworkMonitorService() registerSettingsHandlers(settings) registerBookHandlers(books) @@ -36,6 +41,7 @@ export function registerIpc(): { settings: GlobalSettingsService; books: BookReg registerSnapshotHandlers(books, snapshotService!) registerBookmarkHandlers(books) registerSearchHandlers(books, settings) + registerAiHandlers(books, settings, new AiClientService(() => settings.get().aiConfig)) return { settings, books, shortcuts: shortcutManager } } @@ -44,6 +50,10 @@ export function getShortcutManager(): ShortcutManager | null { return shortcutManager } +export function getNetworkMonitor(): NetworkMonitorService | null { + return networkMonitor +} + export function getSnapshotService(): SnapshotService | null { return snapshotService } diff --git a/src/main/ipc/result.ts b/src/main/ipc/result.ts index e249d0f..3785124 100644 --- a/src/main/ipc/result.ts +++ b/src/main/ipc/result.ts @@ -8,9 +8,18 @@ export function fail(code: ErrorCode, message: string, recoverable: boolean): Ip return { ok: false, error: { code, message, recoverable } } } -export function wrap(fn: () => T, recoverable = true): IpcResult { +export function wrap(fn: () => T | Promise, recoverable = true): IpcResult | Promise> { try { - return ok(fn()) + const out = fn() + if (out && typeof (out as Promise).then === 'function') { + return (out as Promise) + .then((data) => ok(data)) + .catch((e) => { + const message = e instanceof Error ? e.message : String(e) + return fail(ErrorCode.UNKNOWN, message, recoverable) + }) + } + return ok(out as T) } catch (e) { const message = e instanceof Error ? e.message : String(e) return fail(ErrorCode.UNKNOWN, message, recoverable) diff --git a/src/main/services/ai-client.service.ts b/src/main/services/ai-client.service.ts new file mode 100644 index 0000000..b885766 --- /dev/null +++ b/src/main/services/ai-client.service.ts @@ -0,0 +1,105 @@ +import type { AiConfig } from '../../shared/types' + +export interface ChatMessage { + role: 'system' | 'user' | 'assistant' + content: string +} + +export class AiClientService { + constructor(private getConfig: () => AiConfig) {} + + private endpoint(): string { + const { baseUrl } = this.getConfig() + return `${baseUrl.replace(/\/$/, '')}/v1/chat/completions` + } + + private mergeAbortSignals(timeoutMs: number, external?: AbortSignal): { + signal: AbortSignal + dispose: () => void + } { + const timeoutController = new AbortController() + const timer = setTimeout(() => timeoutController.abort(), timeoutMs) + const dispose = (): void => clearTimeout(timer) + + if (!external) { + return { signal: timeoutController.signal, dispose } + } + + const linked = new AbortController() + const abort = (): void => linked.abort() + if (timeoutController.signal.aborted || external.aborted) abort() + timeoutController.signal.addEventListener('abort', abort) + external.addEventListener('abort', abort) + return { + signal: linked.signal, + dispose: () => { + clearTimeout(timer) + timeoutController.signal.removeEventListener('abort', abort) + external.removeEventListener('abort', abort) + } + } + } + + async chat( + messages: ChatMessage[], + onChunk?: (delta: string) => void, + abortSignal?: AbortSignal + ): Promise { + const config = this.getConfig() + const { signal, dispose } = this.mergeAbortSignals(config.timeoutMs, abortSignal) + try { + const res = await fetch(this.endpoint(), { + method: 'POST', + headers: { + 'Content-Type': 'application/json', + ...(config.apiKey ? { Authorization: `Bearer ${config.apiKey}` } : {}) + }, + body: JSON.stringify({ + model: config.model, + messages, + max_tokens: config.maxTokens, + stream: Boolean(onChunk) + }), + signal + }) + if (!res.ok) throw new Error(`AI ${res.status}: ${await res.text()}`) + + if (!onChunk) { + const json = (await res.json()) as { + choices?: { message?: { content?: string } }[] + } + return json.choices?.[0]?.message?.content ?? '' + } + + if (!res.body) throw new Error('AI stream: empty response body') + + let full = '' + const reader = res.body.getReader() + const decoder = new TextDecoder() + let buf = '' + while (true) { + const { done, value } = await reader.read() + if (done) break + buf += decoder.decode(value, { stream: true }) + const lines = buf.split('\n') + buf = lines.pop() ?? '' + for (const line of lines) { + if (!line.startsWith('data: ')) continue + const payload = line.slice(6).trim() + if (payload === '[DONE]') continue + const parsed = JSON.parse(payload) as { + choices?: { delta?: { content?: string } }[] + } + const delta = parsed.choices?.[0]?.delta?.content ?? '' + if (delta) { + full += delta + onChunk(delta) + } + } + } + return full + } finally { + dispose() + } + } +} diff --git a/src/main/services/ai-context-builder.service.ts b/src/main/services/ai-context-builder.service.ts new file mode 100644 index 0000000..a258acd --- /dev/null +++ b/src/main/services/ai-context-builder.service.ts @@ -0,0 +1,131 @@ +import { stripHtml } from './word-count' +import type { AiContextBinding, AiContextBuildResult, ContextPreviewItem } from '../../shared/types' +import type { BookRegistryService } from './book-registry' + +const CHAPTER_CHAR_LIMIT = 2000 +const OTHER_CHAR_LIMIT = 500 +const TOTAL_CHAR_LIMIT = 16 * 1024 + +interface BlockCandidate { + sortKey: number + text: string + preview: ContextPreviewItem +} + +function truncate(text: string, max: number): string { + if (text.length <= max) return text + return `${text.slice(0, max)}…` +} + +function formatBlock(kind: ContextPreviewItem['kind'], title: string, body: string): string { + const label = + kind === 'chapter' ? '章节' : kind === 'outline' ? '大纲' : kind === 'setting' ? '设定' : '灵感' + return `【${label}】${title}\n${body}` +} + +export class AiContextBuilderService { + build( + binding: AiContextBinding, + registry: BookRegistryService, + bookId: string, + penName: string + ): AiContextBuildResult { + const meta = registry.getMeta(bookId) + if (!meta) throw new Error('Book not found') + + const candidates: BlockCandidate[] = [] + + for (const id of binding.chapterIds) { + const chapter = registry.getChapterRepo(bookId).get(id) + if (!chapter) continue + const plain = stripHtml(chapter.content) + const body = truncate(plain, CHAPTER_CHAR_LIMIT) + candidates.push({ + sortKey: chapter.sortOrder, + text: formatBlock('chapter', chapter.title, body), + preview: { + kind: 'chapter', + id, + title: chapter.title, + excerpt: body, + charCount: body.length + } + }) + } + + for (const id of binding.outlineIds) { + const item = registry.getOutlineRepo(bookId).get(id) + if (!item) continue + const body = truncate(stripHtml(item.description), OTHER_CHAR_LIMIT) + candidates.push({ + sortKey: item.sortOrder, + text: formatBlock('outline', item.title, body), + preview: { + kind: 'outline', + id, + title: item.title, + excerpt: body, + charCount: body.length + } + }) + } + + for (const id of binding.settingIds) { + const item = registry.getSettingRepo(bookId).get(id) + if (!item) continue + const body = truncate(stripHtml(item.description), OTHER_CHAR_LIMIT) + candidates.push({ + sortKey: 0, + text: formatBlock('setting', item.name, body), + preview: { + kind: 'setting', + id, + title: item.name, + excerpt: body, + charCount: body.length + } + }) + } + + for (const id of binding.inspirationIds) { + const item = registry.getInspirationRepo(bookId).get(id) + if (!item) continue + const body = truncate(stripHtml(item.content), OTHER_CHAR_LIMIT) + candidates.push({ + sortKey: Date.parse(item.updatedAt) || 0, + text: formatBlock('inspiration', item.title, body), + preview: { + kind: 'inspiration', + id, + title: item.title, + excerpt: body, + charCount: body.length + } + }) + } + + candidates.sort((a, b) => b.sortKey - a.sortKey) + + const selected: BlockCandidate[] = [] + let totalChars = 0 + for (const candidate of candidates) { + const nextTotal = totalChars + candidate.text.length + (selected.length > 0 ? 2 : 0) + if (nextTotal > TOTAL_CHAR_LIMIT) break + selected.push(candidate) + totalChars = nextTotal + } + + const contextBlocks = selected.map((x) => x.text).join('\n\n') + const systemPrompt = `你是笔临 AI 写作助手。作者笔名：${penName}。书籍：${meta.name}。 +以下是与本书相关的上下文（由作者勾选）： +${contextBlocks || '（无）'} +请用简体中文回答，尊重作者的创作主权。` + + return { + systemPrompt, + preview: selected.map((x) => x.preview) + } + } +} + +export const aiContextBuilder = new AiContextBuilderService() diff --git a/src/main/services/book-registry.ts b/src/main/services/book-registry.ts index 9fde876..ed4e2fd 100644 --- a/src/main/services/book-registry.ts +++ b/src/main/services/book-registry.ts @@ -7,6 +7,7 @@ import { ChapterRepository } from '../db/repositories/chapter.repo' import { OutlineRepository } from '../db/repositories/outline.repo' import { SettingRepository } from '../db/repositories/setting.repo' import { InspirationRepository } from '../db/repositories/inspiration.repo' +import { AiSessionRepository } from '../db/repositories/ai-session.repo' import { SnapshotRepository } from '../db/repositories/snapshot.repo' import { BookmarkRepository } from '../db/repositories/bookmark.repo' import type { BookMeta, BookOpenResult, CreateBookParams, Chapter, Volume } from '../../shared/types' @@ -145,6 +146,10 @@ export class BookRegistryService { return new InspirationRepository(this.getDb(bookId)) } + getAiSessionRepo(bookId: string): AiSessionRepository { + return new AiSessionRepository(this.getDb(bookId)) + } + getSnapshotRepo(bookId: string): SnapshotRepository { return new SnapshotRepository(this.getDb(bookId)) } diff --git a/src/main/services/global-settings.ts b/src/main/services/global-settings.ts index 796f264..ec7acee 100644 --- a/src/main/services/global-settings.ts +++ b/src/main/services/global-settings.ts @@ -1,10 +1,18 @@ import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'fs' import { join } from 'path' import { DEFAULT_SHORTCUTS } from '../../shared/default-shortcuts' -import type { GlobalSettings } from '../../shared/types' +import { DEFAULT_AI_CONFIG, type GlobalSettings } from '../../shared/types' const FILE = 'global_settings.json' +function defaultAiConfig() { + return { + ...DEFAULT_AI_CONFIG, + baseUrl: process.env.BILIN_AI_BASE_URL ?? DEFAULT_AI_CONFIG.baseUrl, + model: process.env.BILIN_AI_MODEL ?? DEFAULT_AI_CONFIG.model + } +} + function defaults(): GlobalSettings { return { penName: '未命名作者', @@ -16,7 +24,9 @@ function defaults(): GlobalSettings { snapshotIntervalMs: 300_000, snapshotMaxAuto: 20, snapshotMaxPersist: 3, - searchHistory: [] + searchHistory: [], + aiConfig: defaultAiConfig(), + namingIgnoreWords: [] } } @@ -35,7 +45,8 @@ export class GlobalSettingsService { return { ...defaults(), ...raw, - shortcuts: { ...DEFAULT_SHORTCUTS, ...raw.shortcuts } + shortcuts: { ...DEFAULT_SHORTCUTS, ...raw.shortcuts }, + aiConfig: { ...DEFAULT_AI_CONFIG, ...raw.aiConfig } } } @@ -46,7 +57,8 @@ export class GlobalSettingsService { ...partial, shortcuts: partial.shortcuts ? { ...current.shortcuts, ...partial.shortcuts } - : current.shortcuts + : current.shortcuts, + aiConfig: partial.aiConfig ? { ...current.aiConfig, ...partial.aiConfig } : current.aiConfig } writeFileSync(this.filePath(), JSON.stringify(next, null, 2), 'utf-8') return next diff --git a/src/main/services/naming-check.service.ts b/src/main/services/naming-check.service.ts new file mode 100644 index 0000000..15002e4 --- /dev/null +++ b/src/main/services/naming-check.service.ts @@ -0,0 +1,151 @@ +import type { NamingConflict } from '../../shared/types' + +export function parseNamingJson(response: string): NamingConflict[] { + const trimmed = response.trim() + const start = trimmed.indexOf('[') + const end = trimmed.lastIndexOf(']') + if (start < 0 || end <= start) { + throw new Error('命名检查响应中未找到 JSON 数组') + } + + const parsed = JSON.parse(trimmed.slice(start, end + 1)) as unknown + if (!Array.isArray(parsed)) { + throw new Error('命名检查响应不是 JSON 数组') + } + + const conflicts: NamingConflict[] = [] + for (const item of parsed) { + if (!item || typeof item !== 'object') continue + const row = item as Record + const termA = String(row.termA ?? '').trim() + const termB = String(row.termB ?? '').trim() + const reason = String(row.reason ?? '').trim() + const confidence = Number(row.confidence) + if (!termA || !termB || !Number.isFinite(confidence)) continue + conflicts.push({ + termA, + termB, + confidence, + reason, + suggestedCanonical: + typeof row.suggestedCanonical === 'string' ? row.suggestedCanonical.trim() : undefined + }) + } + + return conflicts.sort((a, b) => b.confidence - a.confidence) +} + +export function buildNamingCorpus( + settings: { name: string; description: string }[], + chapters: { title: string; content: string }[] +): string { + const chapterText = collectChaptersPlain(chapters) + const settingText = settings + .map((s) => stripHtml(s.description)) + .filter(Boolean) + .join('\n\n') + return [chapterText, settingText].filter(Boolean).join('\n\n') +} + +export function buildNamingPrompt( + settingNames: string[], + corpusPlain: string, + ignoreWords: string[] +): string { + const names = settingNames.filter(Boolean).join('、') || '（无）' + const ignore = ignoreWords.filter(Boolean).join('、') || '（无）' + const excerpt = corpusPlain.slice(0, 12_000) + + return `你是小说编辑助手。请检查下列设定名称与正文片段中是否存在**同一实体/概念的不同写法**（如「林远/林悦」「玄火掌/焰火掌」）。 + +设定名称：${names} +忽略词：${ignore} + +正文片段： +${excerpt} + +仅返回 JSON 数组，不要 markdown 或其它说明。每项格式： +{"termA":"写法A","termB":"写法B","confidence":0.0-1.0,"reason":"简短原因","suggestedCanonical":"建议统一用名"} + +若无冲突返回 []。` +} + +function stripHtml(html: string): string { + return html.replace(/<[^>]+>/g, '').replace(/\s+/g, ' ').trim() +} + +function extractBodyTokens(text: string): Set { + const tokens = new Set() + const segments = text.match(/[\u4e00-\u9fff]+/g) ?? [] + for (const segment of segments) { + for (let len = 2; len <= Math.min(4, segment.length); len++) { + for (let i = 0; i <= segment.length - len; i++) { + tokens.add(segment.slice(i, i + len)) + } + } + } + return tokens +} + +export function findLocalNamingConflicts( + settingNames: string[], + corpusPlain: string, + ignoreWords: string[] +): NamingConflict[] { + const ignore = new Set(ignoreWords) + const tokens = extractBodyTokens(corpusPlain) + const conflicts: NamingConflict[] = [] + + for (const name of settingNames) { + const trimmed = name.trim() + if (!trimmed || ignore.has(trimmed)) continue + + for (const token of tokens) { + if (token === trimmed || ignore.has(token)) continue + if (trimmed.length !== token.length || trimmed[0] !== token[0]) continue + + const diffCount = [...trimmed].filter((ch, index) => ch !== token[index]).length + if (diffCount === 0 || diffCount > 2) continue + + conflicts.push({ + termA: trimmed, + termB: token, + confidence: 0.88, + reason: '设定名与正文用词仅一字之差', + suggestedCanonical: trimmed + }) + } + } + + const seen = new Set() + return conflicts.filter((row) => { + const key = [row.termA, row.termB].sort().join('|') + if (seen.has(key)) return false + seen.add(key) + return true + }) +} + +function mergeNamingConflicts(...groups: NamingConflict[][]): NamingConflict[] { + const seen = new Set() + const merged: NamingConflict[] = [] + for (const group of groups) { + for (const row of group) { + const key = [row.termA, row.termB].sort().join('|') + if (seen.has(key)) continue + seen.add(key) + merged.push(row) + } + } + return merged.sort((a, b) => b.confidence - a.confidence) +} + +export function collectChaptersPlain( + chapters: { title: string; content: string }[] +): string { + return chapters + .map((ch) => `${ch.title}\n${stripHtml(ch.content)}`) + .join('\n\n') +} + +export { mergeNamingConflicts } diff --git a/src/main/services/network-monitor.service.ts b/src/main/services/network-monitor.service.ts new file mode 100644 index 0000000..ec6b8ac --- /dev/null +++ b/src/main/services/network-monitor.service.ts @@ -0,0 +1,31 @@ +import { net, type BrowserWindow } from 'electron' +import { IPC } from '../../shared/ipc-channels' + +export class NetworkMonitorService { + private timer: ReturnType | null = null + private online = net.isOnline() + + start(getWindow: () => BrowserWindow | null): void { + this.stop() + this.broadcast(getWindow) + this.timer = setInterval(() => { + const now = net.isOnline() + if (now !== this.online) { + this.online = now + this.broadcast(getWindow) + } + }, 30_000) + } + + private broadcast(getWindow: () => BrowserWindow | null): void { + const win = getWindow() + win?.webContents.send(IPC.AI_NETWORK_STATUS, { online: this.online }) + } + + stop(): void { + if (this.timer) { + clearInterval(this.timer) + this.timer = null + } + } +} diff --git a/src/main/services/search.service.ts b/src/main/services/search.service.ts index 668675a..30e4ace 100644 --- a/src/main/services/search.service.ts +++ b/src/main/services/search.service.ts @@ -1,5 +1,6 @@ import type { SearchResult, SearchSourceKind } from '../../shared/types' import type { SqliteDb } from '../db/types' +import { ftsSync } from './fts-sync.service' export interface SearchQueryParams { text: string @@ -137,8 +138,8 @@ export class SearchService { for (const hit of results) { if (hit.kind === 'chapter') { - const row = db.prepare('SELECT content FROM chapters WHERE id = ?').get(hit.id) as - | { content: string } + const row = db.prepare('SELECT title, content FROM chapters WHERE id = ?').get(hit.id) as + | { title: string; content: string } | undefined if (!row) continue const next = row.content.replace(re, params.replacement) @@ -147,6 +148,7 @@ export class SearchService { next, hit.id ) + ftsSync.upsert(db, 'chapter', hit.id, row.title, next) count++ } } diff --git a/src/preload/index.ts b/src/preload/index.ts index bebc990..dc6a3dd 100644 --- a/src/preload/index.ts +++ b/src/preload/index.ts @@ -9,6 +9,14 @@ import type { CreateBookParams, GlobalSettings, InspirationEntry, + AiContextBinding, + AiContextBuildResult, + AiConfig, + AiMessage, + AiSession, + NamingConflict, + AiStreamChunkEvent, + AiNetworkStatusEvent, IpcResult, LandmarkType, OutlineItem, @@ -67,7 +75,16 @@ const electronAPI = { update: (params: UpdateChapterParams): Promise> => ipcRenderer.invoke(IPC.CHAPTER_UPDATE, params), delete: (bookId: string, chapterId: string): Promise> => - ipcRenderer.invoke(IPC.CHAPTER_DELETE, { bookId, chapterId }) + ipcRenderer.invoke(IPC.CHAPTER_DELETE, { bookId, chapterId }), + reorder: (bookId: string, volumeId: string, orderedIds: string[]): Promise> => + ipcRenderer.invoke(IPC.CHAPTER_REORDER, { bookId, volumeId, orderedIds }), + moveToVolume: ( + bookId: string, + chapterId: string, + targetVolumeId: string, + targetIndex: number + ): Promise> => + ipcRenderer.invoke(IPC.CHAPTER_MOVE_TO_VOLUME, { bookId, chapterId, targetVolumeId, targetIndex }) }, outline: { list: (bookId: string): Promise> => @@ -213,6 +230,40 @@ const electronAPI = { saveHistory: (query: string): Promise> => ipcRenderer.invoke(IPC.SEARCH_SAVE_HISTORY, { query }) }, + ai: { + sessionList: (bookId: string, includeArchived?: boolean): Promise> => + ipcRenderer.invoke(IPC.AI_SESSION_LIST, { bookId, includeArchived }), + sessionCreate: (bookId: string, title?: string, model?: string): Promise> => + ipcRenderer.invoke(IPC.AI_SESSION_CREATE, { bookId, title, model }), + sessionUpdate: ( + bookId: string, + sessionId: string, + patch: Partial<{ title: string; model: string; archived: boolean; context: AiContextBinding }> + ): Promise> => + ipcRenderer.invoke(IPC.AI_SESSION_UPDATE, { bookId, sessionId, patch }), + sessionDelete: (bookId: string, sessionId: string): Promise> => + ipcRenderer.invoke(IPC.AI_SESSION_DELETE, { bookId, sessionId }), + messageList: (bookId: string, sessionId: string): Promise> => + ipcRenderer.invoke(IPC.AI_MESSAGE_LIST, { bookId, sessionId }), + chat: ( + bookId: string, + sessionId: string, + content: string, + prompt?: string + ): Promise> => + ipcRenderer.invoke(IPC.AI_CHAT, { bookId, sessionId, content, prompt }), + abort: (messageId: string): Promise> => + ipcRenderer.invoke(IPC.AI_ABORT, { messageId }), + buildContext: ( + bookId: string, + binding: AiContextBinding + ): Promise> => + ipcRenderer.invoke(IPC.AI_BUILD_CONTEXT, { bookId, binding }), + testConnection: (config?: AiConfig): Promise> => + ipcRenderer.invoke(IPC.AI_TEST_CONNECTION, { config }), + namingCheck: (bookId: string): Promise> => + ipcRenderer.invoke(IPC.AI_NAMING_CHECK, { bookId }) + }, wordfreq: { analyze: ( bookId: string, @@ -232,6 +283,20 @@ const electronAPI = { } ipcRenderer.on(IPC.SHORTCUT_TRIGGERED, handler) return () => ipcRenderer.removeListener(IPC.SHORTCUT_TRIGGERED, handler) + }, + onAiStreamChunk: (callback: (payload: AiStreamChunkEvent) => void): (() => void) => { + const handler = (_event: IpcRendererEvent, payload: AiStreamChunkEvent) => { + callback(payload) + } + ipcRenderer.on(IPC.AI_STREAM_CHUNK, handler) + return () => ipcRenderer.removeListener(IPC.AI_STREAM_CHUNK, handler) + }, + onAiNetworkStatus: (callback: (payload: AiNetworkStatusEvent) => void): (() => void) => { + const handler = (_event: IpcRendererEvent, payload: AiNetworkStatusEvent) => { + callback(payload) + } + ipcRenderer.on(IPC.AI_NETWORK_STATUS, handler) + return () => ipcRenderer.removeListener(IPC.AI_NETWORK_STATUS, handler) } } diff --git a/src/renderer/App.tsx b/src/renderer/App.tsx index 6582171..506ec5a 100644 --- a/src/renderer/App.tsx +++ b/src/renderer/App.tsx @@ -16,10 +16,12 @@ import { useTabStore } from '@renderer/stores/useTabStore' import { useBookStore } from '@renderer/stores/useBookStore' import { useEditStore } from '@renderer/stores/useEditStore' import { SearchModal } from '@renderer/components/search/SearchModal' +import { InspirationModal } from '@renderer/components/inspiration/InspirationModal' import { useSearchStore } from '@renderer/stores/useSearchStore' import { useReferenceStore } from '@renderer/stores/useReferenceStore' import { flushEditorSave } from '@renderer/components/editor/TipTapEditor' import { insertLandmarkInEditor } from '@renderer/lib/editor-commands' +import { useAiStore } from '@renderer/stores/useAiStore' import { ipcCall } from '@renderer/lib/ipc-client' function AppInner(): React.JSX.Element { @@ -48,6 +50,11 @@ function AppInner(): React.JSX.Element { useEffect(() => { const openSearch = (): void => useSearchStore.getState().setOpen(true) const openLandmarks = (): void => useAppStore.getState().setLandmarksModalOpen(true) + const openInspiration = (): void => { + if (useAppStore.getState().view !== 'editor') return + if (!useBookStore.getState().currentBookId) return + useAppStore.getState().setInspirationModalOpen(true) + } const onInsertLandmark = (e: Event): void => { const label = (e as CustomEvent<{ label: string }>).detail?.label if (!label) return @@ -59,13 +66,26 @@ function AppInner(): React.JSX.Element { window.electronAPI.bookmark.create(currentBookId, target.id, 0, label, 'todo') ) } + const onSetAiOnline = (e: Event): void => { + const online = (e as CustomEvent<{ online: boolean }>).detail?.online + if (typeof online === 'boolean') useAiStore.setState({ online }) + } + const onFlushEditor = (): void => { + void flushEditorSave() + } window.addEventListener('bilin:e2e-open-search', openSearch) window.addEventListener('bilin:e2e-open-landmarks', openLandmarks) + window.addEventListener('bilin:e2e-capture-inspiration', openInspiration) window.addEventListener('bilin:e2e-insert-landmark', onInsertLandmark) + window.addEventListener('bilin:e2e-set-ai-online', onSetAiOnline) + window.addEventListener('bilin:e2e-flush-editor', onFlushEditor) return () => { window.removeEventListener('bilin:e2e-open-search', openSearch) window.removeEventListener('bilin:e2e-open-landmarks', openLandmarks) + window.removeEventListener('bilin:e2e-capture-inspiration', openInspiration) window.removeEventListener('bilin:e2e-insert-landmark', onInsertLandmark) + window.removeEventListener('bilin:e2e-set-ai-online', onSetAiOnline) + window.removeEventListener('bilin:e2e-flush-editor', onFlushEditor) } }, []) @@ -123,16 +143,8 @@ function AppInner(): React.JSX.Element { } if (action === 'captureInspiration') { if (view !== 'editor') return - const { currentBookId, addInspirationLocal } = useBookStore.getState() - if (!currentBookId) return - void (async () => { - const item = await ipcCall(() => - window.electronAPI.inspiration.create(currentBookId, t('inspiration.newItem')) - ) - addInspirationLocal(item) - useAppStore.getState().setSidebarPanel('inspiration') - await useEditStore.getState().switchTarget({ kind: 'inspiration', id: item.id }) - })() + if (!useBookStore.getState().currentBookId) return + useAppStore.getState().setInspirationModalOpen(true) return } showToast(t('feature.comingSoon')) @@ -154,6 +166,7 @@ function AppInner(): React.JSX.Element { {showOnboarding && } + {toast &&
{toast}
} ) diff --git a/src/renderer/components/ai/AiPanel.tsx b/src/renderer/components/ai/AiPanel.tsx new file mode 100644 index 0000000..4a9574d --- /dev/null +++ b/src/renderer/components/ai/AiPanel.tsx @@ -0,0 +1,229 @@ +import { useEffect, useRef, useState } from 'react' +import { useTranslation } from 'react-i18next' +import type { AiWritingMode } from '@shared/types' +import { useBookStore } from '@renderer/stores/useBookStore' +import { useAppStore } from '@renderer/stores/useAppStore' +import { useAiStore } from '@renderer/stores/useAiStore' +import { useSettingsStore } from '@renderer/stores/useSettingsStore' +import { applySlashCommand } from '@renderer/lib/ai-slash-commands' +import { insertToEditor } from '@renderer/lib/editor-commands' +import { ipcCall } from '@renderer/lib/ipc-client' +import { ContextEditorModal } from '@renderer/components/ai/ContextEditorModal' + +const MODES: { id: AiWritingMode; labelKey: string }[] = [ + { id: 'chat', labelKey: 'ai.mode.chat' }, + { id: 'interactive', labelKey: 'ai.mode.interactive' }, + { id: 'auto', labelKey: 'ai.mode.auto' }, + { id: 'wizard', labelKey: 'ai.mode.wizard' } +] + +const QUICK_CMDS = ['/续写', '/扩写', '/润色', '/总结', '/纠错'] + +export function AiPanel(): React.JSX.Element { + const { t } = useTranslation() + const showToast = useAppStore((s) => s.showToast) + const bookId = useBookStore((s) => s.currentBookId) + const sessions = useAiStore((s) => s.sessions) + const activeSessionId = useAiStore((s) => s.activeSessionId) + const messages = useAiStore((s) => s.messages) + const streamingBuffer = useAiStore((s) => s.streamingBuffer) + const streamingMessageId = useAiStore((s) => s.streamingMessageId) + const sending = useAiStore((s) => s.sending) + const online = useAiStore((s) => s.online) + const backend = useSettingsStore((s) => s.aiConfig.backend) + const disabled = backend !== 'lmstudio' && !online + const writingMode = useAiStore((s) => s.writingMode) + const contextSummary = useAiStore((s) => s.contextSummary) + const loadSessions = useAiStore((s) => s.loadSessions) + const selectSession = useAiStore((s) => s.selectSession) + const createSession = useAiStore((s) => s.createSession) + const sendMessage = useAiStore((s) => s.sendMessage) + const setWritingMode = useAiStore((s) => s.setWritingMode) + const [input, setInput] = useState('') + const [contextOpen, setContextOpen] = useState(false) + const messagesEndRef = useRef(null) + const prevOnlineRef = useRef(online) + + useEffect(() => { + if (!prevOnlineRef.current && online && backend !== 'lmstudio') { + showToast(t('ai.networkRestored')) + } + prevOnlineRef.current = online + }, [online, backend, showToast, t]) + + useEffect(() => { + useAiStore.getState().mount() + return () => useAiStore.getState().unmount() + }, []) + + useEffect(() => { + if (bookId) void loadSessions(bookId) + }, [bookId, loadSessions]) + + useEffect(() => { + messagesEndRef.current?.scrollIntoView({ behavior: 'smooth' }) + }, [messages, streamingBuffer]) + + const handleSend = (): void => { + const text = input.trim() + if (!text || sending || disabled) return + const { display, prompt } = applySlashCommand(text, t) + setInput('') + void sendMessage(display, prompt !== display ? prompt : undefined) + } + + const handleInsert = async (content: string): Promise => { + try { + await insertToEditor(content, t('snapshot.ai'), async (bookId, chapterId, html, label) => { + await ipcCall(() => + window.electronAPI.snapshot.create(bookId, chapterId, 'ai', label, html) + ) + }) + showToast(t('ai.inserted')) + } catch (err) { + showToast(err instanceof Error ? err.message : String(err)) + } + } + + const handleModeClick = (mode: AiWritingMode): void => { + if (mode !== 'chat') { + showToast(t('feature.comingSoon')) + return + } + setWritingMode(mode) + } + + if (!bookId) { + return ( +
+ {t('ai.noBook')} +
+ ) + } + + return ( + <> +
+
+ + +
+ +
+ {MODES.map((mode) => ( + + ))} +
+ + + + {disabled && ( +
+ {t('ai.offlineBanner')} +
+ )} + +
+ {messages.map((msg) => ( +
+
{msg.content}
+ {msg.role === 'assistant' && ( + + )} +
+ ))} + {streamingMessageId && streamingBuffer && ( +
+ {streamingBuffer} +
+ )} +
+
+ +
+ {QUICK_CMDS.map((cmd) => ( + + ))} +
+ +
+ setInput(e.target.value)} + placeholder={t('ai.inputPlaceholder')} + disabled={sending || disabled} + onKeyDown={(e) => { + if (e.key === 'Enter') { + e.preventDefault() + handleSend() + } + }} + /> + +
+
+ setContextOpen(false)} /> + + ) +} diff --git a/src/renderer/components/ai/ContextEditorModal.tsx b/src/renderer/components/ai/ContextEditorModal.tsx new file mode 100644 index 0000000..6d6ab7b --- /dev/null +++ b/src/renderer/components/ai/ContextEditorModal.tsx @@ -0,0 +1,150 @@ +import { useEffect, useState } from 'react' +import { useTranslation } from 'react-i18next' +import type { AiContextBinding } from '@shared/types' +import { useBookStore } from '@renderer/stores/useBookStore' +import { useAiStore } from '@renderer/stores/useAiStore' +import { ipcCall } from '@renderer/lib/ipc-client' +import { formatContextSummary } from '@renderer/lib/ai-context-summary' +import i18n from '@renderer/i18n' + +type ContextTab = 'chapter' | 'outline' | 'setting' | 'inspiration' + +interface ContextEditorModalProps { + open: boolean + onClose: () => void +} + +const EMPTY_BINDING: AiContextBinding = { + chapterIds: [], + outlineIds: [], + settingIds: [], + inspirationIds: [] +} + +export function ContextEditorModal({ open, onClose }: ContextEditorModalProps): React.JSX.Element | null { + const { t } = useTranslation() + const bookId = useBookStore((s) => s.currentBookId) + const chapters = useBookStore((s) => s.chapters) + const outlines = useBookStore((s) => s.outlines) + const settings = useBookStore((s) => s.settings) + const inspirations = useBookStore((s) => s.inspirations) + const activeSessionId = useAiStore((s) => s.activeSessionId) + const sessions = useAiStore((s) => s.sessions) + const [tab, setTab] = useState('setting') + const [binding, setBinding] = useState(EMPTY_BINDING) + + useEffect(() => { + if (!open) return + const session = sessions.find((s) => s.id === activeSessionId) + setBinding(session?.context ?? EMPTY_BINDING) + }, [open, activeSessionId, sessions]) + + const toggle = (kind: ContextTab, id: string): void => { + const key = + kind === 'chapter' + ? 'chapterIds' + : kind === 'outline' + ? 'outlineIds' + : kind === 'setting' + ? 'settingIds' + : 'inspirationIds' + setBinding((prev) => { + const list = prev[key] + return { + ...prev, + [key]: list.includes(id) ? list.filter((x) => x !== id) : [...list, id] + } + }) + } + + const handleSave = async (): Promise => { + if (!bookId || !activeSessionId) return + const updated = await ipcCall(() => + window.electronAPI.ai.sessionUpdate(bookId, activeSessionId, { context: binding }) + ) + useAiStore.setState((s) => ({ + sessions: s.sessions.map((x) => (x.id === updated.id ? updated : x)) + })) + await useAiStore.getState().refreshContextPreview() + onClose() + } + + const handleRefresh = async (): Promise => { + if (!bookId) return + const result = await ipcCall(() => window.electronAPI.ai.buildContext(bookId, binding)) + useAiStore.setState({ + contextPreview: result.preview, + contextSummary: formatContextSummary(result.preview, i18n.t) + }) + } + + if (!open) return null + + const items = + tab === 'chapter' + ? chapters.map((c) => ({ id: c.id, title: c.title })) + : tab === 'outline' + ? outlines.map((o) => ({ id: o.id, title: o.title })) + : tab === 'setting' + ? settings.map((s) => ({ id: s.id, title: s.name })) + : inspirations.map((i) => ({ id: i.id, title: i.title || t('inspiration.untitled') })) + + const selectedIds = + tab === 'chapter' + ? binding.chapterIds + : tab === 'outline' + ? binding.outlineIds + : tab === 'setting' + ? binding.settingIds + : binding.inspirationIds + + return ( +
+
+
+

{t('ai.contextEditorTitle')}

+ +
+
+ {(['chapter', 'outline', 'setting', 'inspiration'] as const).map((id) => ( + + ))} +
+
+ {items.length === 0 &&
{t('ai.contextEmptyList')}
} + {items.map((item) => ( + + ))} +
+
+ + + +
+
+
+ ) +} diff --git a/src/renderer/components/ai/NamingCheckModal.tsx b/src/renderer/components/ai/NamingCheckModal.tsx new file mode 100644 index 0000000..12e8e6a --- /dev/null +++ b/src/renderer/components/ai/NamingCheckModal.tsx @@ -0,0 +1,124 @@ +import { useState } from 'react' +import { useTranslation } from 'react-i18next' +import type { NamingConflict } from '@shared/types' +import { useBookStore } from '@renderer/stores/useBookStore' +import { useSettingsStore } from '@renderer/stores/useSettingsStore' +import { useAppStore } from '@renderer/stores/useAppStore' +import { useEditStore } from '@renderer/stores/useEditStore' +import { ipcCall } from '@renderer/lib/ipc-client' + +interface NamingCheckModalProps { + open: boolean + onClose: () => void +} + +export function NamingCheckModal({ open, onClose }: NamingCheckModalProps): React.JSX.Element | null { + const { t } = useTranslation() + const bookId = useBookStore((s) => s.currentBookId) + const openBook = useBookStore((s) => s.openBook) + const namingIgnoreWords = useSettingsStore((s) => s.namingIgnoreWords) + const updateSettings = useSettingsStore((s) => s.update) + const showToast = useAppStore((s) => s.showToast) + const reloadTarget = useEditStore((s) => s.reloadTarget) + const [loading, setLoading] = useState(false) + const [conflicts, setConflicts] = useState([]) + + if (!open) return null + + const runCheck = async (): Promise => { + const activeBookId = useBookStore.getState().currentBookId + if (!activeBookId) return + setLoading(true) + try { + const rows = await ipcCall(() => window.electronAPI.ai.namingCheck(activeBookId)) + const ignore = new Set(namingIgnoreWords) + setConflicts( + rows.filter((row) => !ignore.has(row.termA) && !ignore.has(row.termB)) + ) + if (rows.length === 0) { + showToast(t('naming.noConflicts')) + } + } catch (err) { + showToast(err instanceof Error ? err.message : String(err)) + setConflicts([]) + } finally { + setLoading(false) + } + } + + const handleIgnore = (row: NamingConflict): void => { + const next = Array.from(new Set([...namingIgnoreWords, row.termA, row.termB])) + void updateSettings({ namingIgnoreWords: next }) + setConflicts((list) => list.filter((item) => item !== row)) + showToast(t('naming.ignored')) + } + + const handleReplace = async (row: NamingConflict): Promise => { + if (!bookId) return + const canonical = row.suggestedCanonical || row.termA + const from = row.termB + if (!from || from === canonical) return + try { + const result = await ipcCall(() => + window.electronAPI.search.replace(bookId, from, canonical, false, { caseSensitive: true }) + ) + await openBook(bookId) + reloadTarget() + setConflicts((list) => list.filter((item) => item !== row)) + showToast(t('naming.replaced', { count: result.count })) + } catch (err) { + showToast(err instanceof Error ? err.message : String(err)) + } + } + + return ( +
+
+
+

{t('naming.title')}

+ +
+
+ {conflicts.length === 0 && !loading && ( +

{t('naming.emptyHint')}

+ )} + {conflicts.map((row, index) => ( +
+
+ {row.termA} ↔ {row.termB} +
+
+ {t('naming.confidence', { value: Math.round(row.confidence * 100) })} + {row.reason ? ` · ${row.reason}` : ''} +
+
+ + +
+
+ ))} +
+
+ + +
+
+
+ ) +} diff --git a/src/renderer/components/editor/TipTapEditor.tsx b/src/renderer/components/editor/TipTapEditor.tsx index f491f41..b3c07fd 100644 --- a/src/renderer/components/editor/TipTapEditor.tsx +++ b/src/renderer/components/editor/TipTapEditor.tsx @@ -8,11 +8,11 @@ import { useSetAtom } from 'jotai' import type { EditTarget } from '@shared/types' import { editorDirtyAtom, editorSavingAtom, lastSavedAtAtom, wordCountAtom } from '@renderer/atoms/editorAtoms' import { useBookStore } from '@renderer/stores/useBookStore' -import { flushEditSession, registerEditorFlush } from '@renderer/stores/useEditStore' +import { flushEditSession, registerEditorFlush, useEditStore } from '@renderer/stores/useEditStore' import { ipcCall } from '@renderer/lib/ipc-client' import { LandmarkMark } from '@renderer/extensions/landmark-mark' import { SettingMention } from '@renderer/extensions/mention' -import { registerHighlightToken, registerInsertLandmark } from '@renderer/lib/editor-commands' +import { registerHighlightToken, registerInsertLandmark, registerEditorInsert } from '@renderer/lib/editor-commands' function countPlain(text: string): number { const stripped = text.replace(/<[^>]+>/g, '').replace(/\s+/g, '') const cjk = stripped.match(/[\u4e00-\u9fff\u3400-\u4dbf]/g) @@ -149,6 +149,13 @@ export function TipTapEditor({ target, bookId }: TipTapEditorProps): React.JSX.E } }, editorProps: { + handleClick: (_view, _pos, event) => { + const el = (event.target as HTMLElement).closest('[data-setting-mention]') + if (!el) return false + const id = el.getAttribute('data-id') + if (id) void useEditStore.getState().switchTarget({ kind: 'setting', id }) + return true + }, handleDOMEvents: { dragover: (_view, event) => { event.preventDefault() @@ -282,7 +289,29 @@ export function TipTapEditor({ target, bookId }: TipTapEditorProps): React.JSX.E }, [editor]) useEffect(() => { - if (!editor || !target) return + if (!editor || !bookId || !target || target.kind !== 'chapter') { + registerEditorInsert(null, () => {}) + return + } + registerEditorInsert( + { + bookId, + chapterId: target.id, + getHtml: () => editor.getHTML() + }, + (text) => { + editor.chain().focus().insertContent(text).run() + } + ) + return () => registerEditorInsert(null, () => {}) + }, [editor, bookId, target]) + + useEffect(() => { + if (!editor) return + if (!target) { + targetKeyRef.current = null + return + } const key = `${target.kind}:${target.id}` if (targetKeyRef.current === key) return targetKeyRef.current = key diff --git a/src/renderer/components/inspiration/InspirationModal.tsx b/src/renderer/components/inspiration/InspirationModal.tsx new file mode 100644 index 0000000..9a54774 --- /dev/null +++ b/src/renderer/components/inspiration/InspirationModal.tsx @@ -0,0 +1,144 @@ +import { useEffect, useRef, useState } from 'react' +import { useTranslation } from 'react-i18next' +import { useAppStore } from '@renderer/stores/useAppStore' +import { useBookStore } from '@renderer/stores/useBookStore' +import { useEditStore } from '@renderer/stores/useEditStore' +import { ipcCall } from '@renderer/lib/ipc-client' + +type SpeechRecognitionCtor = new () => SpeechRecognition + +function getSpeechRecognition(): SpeechRecognitionCtor | null { + const w = window as Window & { + SpeechRecognition?: SpeechRecognitionCtor + webkitSpeechRecognition?: SpeechRecognitionCtor + } + return w.SpeechRecognition ?? w.webkitSpeechRecognition ?? null +} + +export function InspirationModal(): React.JSX.Element | null { + const { t } = useTranslation() + const open = useAppStore((s) => s.inspirationModalOpen) + const setOpen = useAppStore((s) => s.setInspirationModalOpen) + const setSidebarPanel = useAppStore((s) => s.setSidebarPanel) + const bookId = useBookStore((s) => s.currentBookId) + const addInspirationLocal = useBookStore((s) => s.addInspirationLocal) + const switchTarget = useEditStore((s) => s.switchTarget) + const [title, setTitle] = useState('') + const [content, setContent] = useState('') + const [tags, setTags] = useState('') + const contentRef = useRef(null) + const speechAvailable = getSpeechRecognition() !== null + + useEffect(() => { + if (!open) return + setTitle('') + setContent('') + setTags('') + const timer = setTimeout(() => contentRef.current?.focus(), 100) + return () => clearTimeout(timer) + }, [open]) + + const handleVoice = (): void => { + const SpeechRecognition = getSpeechRecognition() + if (!SpeechRecognition) return + const rec = new SpeechRecognition() + rec.lang = 'zh-CN' + rec.onresult = (e) => { + const transcript = e.results[0]?.[0]?.transcript ?? '' + if (transcript) setContent((c) => c + transcript) + } + rec.start() + } + + const handleSave = async (): Promise => { + if (!bookId || !content.trim()) return + const tagList = tags + .split(/[,，]/) + .map((x) => x.trim()) + .filter(Boolean) + const item = await ipcCall(() => + window.electronAPI.inspiration.create( + bookId, + title.trim() || t('inspiration.newItem'), + content.trim(), + tagList + ) + ) + addInspirationLocal(item) + setSidebarPanel('inspiration') + await switchTarget({ kind: 'inspiration', id: item.id }) + setOpen(false) + } + + if (!open) return null + + return ( +