fix: harden speech and document interactions

src/main/knowledge/document-parser.test.ts

@@ -2,6 +2,33 @@ import { strToU8, zipSync } from 'fflate'
 import { describe, expect, it } from 'vitest'
 import { chunkDocument, parseDocument } from './document-parser'
 
+function createPdfFixture(text: string): Buffer {
+  const stream = `BT /F1 18 Tf 50 100 Td (${text}) Tj ET`
+  const objects = [
+    '<< /Type /Catalog /Pages 2 0 R >>',
+    '<< /Type /Pages /Kids [3 0 R] /Count 1 >>',
+    '<< /Type /Page /Parent 2 0 R /MediaBox [0 0 300 200] /Resources << /Font << /F1 4 0 R >> >> /Contents 5 0 R >>',
+    '<< /Type /Font /Subtype /Type1 /BaseFont /Helvetica >>',
+    `<< /Length ${Buffer.byteLength(stream)} >>\nstream\n${stream}\nendstream`
+  ]
+  let content = '%PDF-1.4\n'
+  const offsets = [0]
+  for (const [index, object] of objects.entries()) {
+    offsets.push(Buffer.byteLength(content))
+    content += `${index + 1} 0 obj\n${object}\nendobj\n`
+  }
+  const xrefOffset = Buffer.byteLength(content)
+  content += `xref\n0 ${objects.length + 1}\n`
+  content += '0000000000 65535 f \n'
+  content += offsets
+    .slice(1)
+    .map((offset) => `${String(offset).padStart(10, '0')} 00000 n \n`)
+    .join('')
+  content += `trailer\n<< /Size ${objects.length + 1} /Root 1 0 R >>\n`
+  content += `startxref\n${xrefOffset}\n%%EOF\n`
+  return Buffer.from(content)
+}
+
 describe('document parser', () => {
   it('parses text and creates overlapping bounded chunks', async () => {
     const parsed = await parseDocument(
@@ -66,6 +93,21 @@ describe('document parser', () => {
     }
   })
 
+  it('extracts page text and locators from PDF files', async () => {
+    const parsed = await parseDocument(
+      'sample.pdf',
+      createPdfFixture('PDF body text')
+    )
+
+    expect(parsed.content).toContain('PDF body text')
+    expect(parsed.sections).toEqual([
+      {
+        locator: '第 1 页',
+        content: 'PDF body text'
+      }
+    ])
+  })
+
   it('rejects unsupported or oversized content', async () => {
     await expect(
       parseDocument('archive.zip', Buffer.from('not supported'))