feat: add document OCR and offline model archives

src/main/document-parsing-settings-store.test.ts

@@ -0,0 +1,120 @@
+import {
+  mkdtemp,
+  readFile,
+  readdir,
+  rm,
+  writeFile
+} from 'node:fs/promises'
+import { tmpdir } from 'node:os'
+import { join } from 'node:path'
+import { afterEach, describe, expect, it } from 'vitest'
+import {
+  defaultDocumentParsingSettings,
+  DocumentParsingSettingsStore
+} from './document-parsing-settings-store'
+
+const temporaryDirectories: string[] = []
+
+async function createStore(): Promise<{
+  directory: string
+  filePath: string
+  store: DocumentParsingSettingsStore
+}> {
+  const directory = await mkdtemp(
+    join(tmpdir(), 'goodbuddy-document-parsing-settings-')
+  )
+  temporaryDirectories.push(directory)
+  const filePath = join(directory, 'document-parsing-settings.json')
+  return {
+    directory,
+    filePath,
+    store: new DocumentParsingSettingsStore(filePath)
+  }
+}
+
+afterEach(async () => {
+  await Promise.all(
+    temporaryDirectories.splice(0).map((directory) =>
+      rm(directory, { recursive: true, force: true })
+    )
+  )
+})
+
+describe('DocumentParsingSettingsStore', () => {
+  it('returns local-first defaults without creating a file', async () => {
+    const { directory, store } = await createStore()
+
+    await expect(store.get()).resolves.toEqual(
+      defaultDocumentParsingSettings
+    )
+    await expect(readdir(directory)).resolves.toEqual([])
+  })
+
+  it('persists a complete versioned settings document', async () => {
+    const { filePath, store } = await createStore()
+    const settings = {
+      ...defaultDocumentParsingSettings,
+      chatWorkflow: 'fast-text' as const,
+      maximumPages: 42
+    }
+
+    await expect(store.update(settings)).resolves.toEqual(settings)
+    expect(JSON.parse(await readFile(filePath, 'utf8'))).toEqual({
+      version: 2,
+      ...settings
+    })
+    await expect(
+      new DocumentParsingSettingsStore(filePath).get()
+    ).resolves.toEqual(settings)
+  })
+
+  it('migrates legacy cloud permissions to the local OCR provider', async () => {
+    const { filePath, store } = await createStore()
+    const {
+      ocrProvider: _ocrProvider,
+      ...legacySettings
+    } = defaultDocumentParsingSettings
+    void _ocrProvider
+    await writeFile(
+      filePath,
+      JSON.stringify({
+        version: 1,
+        ...legacySettings,
+        chatCloudPermission: 'always',
+        knowledgeCloudPermission: 'never'
+      }),
+      'utf8'
+    )
+
+    await expect(store.get()).resolves.toEqual(
+      defaultDocumentParsingSettings
+    )
+  })
+
+  it('rejects incomplete or out-of-range settings', async () => {
+    const { directory, store } = await createStore()
+
+    await expect(store.update({})).rejects.toThrow()
+    await expect(
+      store.update({
+        ...defaultDocumentParsingSettings,
+        maximumPages: 0
+      })
+    ).rejects.toThrow()
+    await expect(readdir(directory)).resolves.toEqual([])
+  })
+
+  it('isolates corrupt settings and restores defaults', async () => {
+    const { directory, filePath, store } = await createStore()
+    await writeFile(filePath, '{not-json', 'utf8')
+
+    await expect(store.get()).resolves.toEqual(
+      defaultDocumentParsingSettings
+    )
+    const entries = await readdir(directory)
+    expect(entries).toHaveLength(1)
+    expect(entries[0]).toMatch(
+      /^document-parsing-settings\.json\.corrupt-\d+-[a-f0-9]{12}$/u
+    )
+  })
+})