增加语音对话功能

src/modules/lecture/infrastructure/storage/repository.ts

@@ -0,0 +1,85 @@
+import type { PaperCache } from "../../domain/types";
+import { buildCacheKey } from "../pdf/fileHash";
+import {
+  ensureLectureDirs,
+  lectureIndexPath,
+} from "./paths";
+
+interface IndexFile {
+  schemaVersion: number;
+  /** keyed by cacheKey (attachmentId|fileHash) */
+  papers: Record<string, PaperCache>;
+}
+
+const SCHEMA_VERSION = 2;
+
+function migratePapers(
+  papers: Record<string, PaperCache>,
+): Record<string, PaperCache> {
+  const out: Record<string, PaperCache> = {};
+  for (const [key, paper] of Object.entries(papers)) {
+    const cacheKey =
+      key.includes("|") && key.includes(paper.fileHash)
+        ? key
+        : buildCacheKey(paper.attachmentId, paper.fileHash);
+    out[cacheKey] = paper;
+  }
+  return out;
+}
+
+async function readIndex(): Promise<IndexFile> {
+  await ensureLectureDirs();
+  const path = lectureIndexPath();
+  try {
+    if (!(await IOUtils.exists(path))) {
+      return { schemaVersion: SCHEMA_VERSION, papers: {} };
+    }
+    const raw = await IOUtils.readUTF8(path);
+    const data = JSON.parse(raw) as IndexFile;
+    const papers = migratePapers(data.papers ?? {});
+    return {
+      schemaVersion: SCHEMA_VERSION,
+      papers,
+    };
+  } catch (e) {
+    ztoolkit.log("[ChatPapers:Lecture] readIndex failed", e);
+    return { schemaVersion: SCHEMA_VERSION, papers: {} };
+  }
+}
+
+async function writeIndex(index: IndexFile): Promise<void> {
+  await ensureLectureDirs();
+  const path = lectureIndexPath();
+  await IOUtils.writeUTF8(
+    path,
+    JSON.stringify({ ...index, schemaVersion: SCHEMA_VERSION }, null, 2),
+  );
+}
+
+export class PaperRepository {
+  async getByCacheKey(cacheKey: string): Promise<PaperCache | undefined> {
+    const index = await readIndex();
+    return index.papers[cacheKey];
+  }
+
+  async getByAttachment(
+    attachmentId: string,
+    fileHash: string,
+  ): Promise<PaperCache | undefined> {
+    const cacheKey = `${attachmentId}|${fileHash}`;
+    return this.getByCacheKey(cacheKey);
+  }
+
+  async upsert(cacheKey: string, paper: PaperCache): Promise<void> {
+    const index = await readIndex();
+    index.papers[cacheKey] = paper;
+    await writeIndex(index);
+  }
+
+  async listAll(): Promise<PaperCache[]> {
+    const index = await readIndex();
+    return Object.values(index.papers);
+  }
+}
+
+export const paperRepository = new PaperRepository();