feat: add persistent desktop assistant workspace

Co-authored-by: factory-droid[bot] <138933559+factory-droid[bot]@users.noreply.github.com>
2026-07-31 22:33:03 +08:00
parent 698a15ad14
commit 6ef1795b81
101 changed files with 31866 additions and 1176 deletions
@@ -0,0 +1,147 @@
+import {
+  access,
+  mkdtemp,
+  mkdir,
+  rm,
+  writeFile
+} from 'node:fs/promises'
+import { tmpdir } from 'node:os'
+import { join } from 'node:path'
+import { afterEach, describe, expect, it } from 'vitest'
+import { KnowledgeService } from './knowledge-service'
+import { UrlImporter } from './url-importer'
+
+const temporaryDirectories: string[] = []
+const services: KnowledgeService[] = []
+
+async function createService(
+  urlImporter?: UrlImporter
+): Promise<{ directory: string; service: KnowledgeService }> {
+  const directory = await mkdtemp(join(tmpdir(), 'goodbuddy-knowledge-service-'))
+  temporaryDirectories.push(directory)
+  const service = new KnowledgeService({
+    databasePath: join(directory, 'knowledge.sqlite'),
+    managedRoot: join(directory, 'managed'),
+    urlImporter
+  })
+  await service.initialize()
+  services.push(service)
+  return { directory, service }
+}
+
+afterEach(async () => {
+  await Promise.all(services.splice(0).map((service) => service.dispose()))
+  await Promise.all(
+    temporaryDirectories.splice(0).map((directory) =>
+      rm(directory, { recursive: true, force: true })
+    )
+  )
+})
+
+describe('KnowledgeService', () => {
+  it('indexes referenced files and returns cited search results', async () => {
+    const { directory, service } = await createService()
+    const sourcePath = join(directory, '产品说明.md')
+    await writeFile(sourcePath, '# GoodBuddy\n跨平台桌面智能助手', 'utf8')
+    const library = service.createLibrary({
+      name: '产品知识',
+      storageMode: 'reference',
+      graphEnabled: false,
+      graphStrategy: 'rules'
+    })
+
+    await service.importPaths(library.id, [sourcePath])
+    const snapshot = service.snapshot(library.id)
+    const results = service.search(library.id, '跨平台桌面')
+
+    expect(snapshot.sources).toHaveLength(1)
+    expect(snapshot.documents).toHaveLength(1)
+    expect(snapshot.documents[0]?.status).toBe('ready')
+    expect(results[0]?.document.title).toBe('产品说明')
+    expect(results[0]?.source.location).toBe(sourcePath)
+    await service.dispose()
+  })
+
+  it('copies managed directories and never deletes the original source', async () => {
+    const { directory, service } = await createService()
+    const original = join(directory, 'original')
+    await mkdir(original)
+    await writeFile(join(original, 'notes.txt'), '托管目录知识', 'utf8')
+    const library = service.createLibrary({
+      name: '托管知识',
+      storageMode: 'managed',
+      graphEnabled: false,
+      graphStrategy: 'rules'
+    })
+
+    await service.importPaths(library.id, [original])
+    const [source] = service.snapshot(library.id).sources
+    expect(source?.location).not.toBe(original)
+    if (!source) {
+      throw new Error('Managed source was not created')
+    }
+    await access(join(source.location, 'notes.txt'))
+
+    await service.removeSource(source.id)
+    await access(join(original, 'notes.txt'))
+    await expect(access(source.location)).rejects.toThrow()
+    await service.dispose()
+  })
+
+  it('imports safe URLs through the validated importer', async () => {
+    const importer = new UrlImporter({
+      lookup: async () => [{ address: '93.184.216.34', family: 4 }],
+      transport: async () => ({
+        status: 200,
+        headers: { 'content-type': 'text/html' },
+        body: Buffer.from(
+          '<html><title>帮助中心</title><main>安装与配置说明</main></html>'
+        )
+      })
+    })
+    const { service } = await createService(importer)
+    const library = service.createLibrary({
+      name: '网页知识',
+      storageMode: 'managed',
+      graphEnabled: false,
+      graphStrategy: 'rules'
+    })
+
+    await service.importUrl(
+      library.id,
+      'https://example.com/help',
+      new AbortController().signal
+    )
+
+    expect(service.snapshot(library.id).sources[0]).toMatchObject({
+      type: 'url',
+      status: 'ready',
+      displayName: '帮助中心'
+    })
+    expect(service.search(library.id, '安装配置')).not.toHaveLength(0)
+    await service.dispose()
+  })
+
+  it('extracts an optional local rule graph with evidence', async () => {
+    const { directory, service } = await createService()
+    const sourcePath = join(directory, 'architecture.md')
+    await writeFile(
+      sourcePath,
+      'GoodBuddy（产品）依赖 Electron（框架）。',
+      'utf8'
+    )
+    const library = service.createLibrary({
+      name: '架构图谱',
+      storageMode: 'reference',
+      graphEnabled: true,
+      graphStrategy: 'rules'
+    })
+
+    await service.importPaths(library.id, [sourcePath])
+    const snapshot = service.snapshot(library.id)
+
+    expect(snapshot.entities.length).toBeGreaterThan(0)
+    expect(snapshot.evidence.length).toBeGreaterThan(0)
+    await service.dispose()
+  })
+})