next: ai part 3

2026-07-24 01:18:10 -07:00 · 2026-07-22 16:50:32 +02:00
parent 52e4db5ea6
commit f0871d895a
29 changed files with 3207 additions and 143 deletions
@@ -0,0 +1,91 @@
+const WORKER_URL = import.meta.resolve("./worker.js");
+
+export class AskSource {
+  /** @type {Worker | undefined} */
+  #worker;
+
+  /** @type {Map<string, { resolve: (value: any) => void, reject: (error: Error) => void, onProgress?: (progress: { loaded: number, total: number }) => void }>} */
+  #pending = new Map();
+
+  /**
+   * @param {string} query
+   * @param {string | undefined} path
+   * @param {(progress: { loaded: number, total: number }) => void} onProgress
+   */
+  search(query, path, onProgress) {
+    return this.#request("search", { query, path }, onProgress);
+  }
+
+  /**
+   * @param {string} question
+   * @param {(progress: { loaded: number, total: number }) => void} onProgress
+   */
+  explain(question, onProgress) {
+    return this.#request("explain", { question }, onProgress);
+  }
+
+  /**
+   * @param {string} path
+   * @param {number} startLine
+   * @param {number} endLine
+   */
+  read(path, startLine, endLine) {
+    return this.#request("read", { path, startLine, endLine });
+  }
+
+  /**
+   * @param {"search" | "read" | "explain"} type
+   * @param {Record<string, unknown>} data
+   * @param {((progress: { loaded: number, total: number }) => void) | undefined} [onProgress]
+   */
+  #request(type, data, onProgress) {
+    this.#ensureWorker();
+    const id = crypto.randomUUID();
+
+    return new Promise((resolve, reject) => {
+      this.#pending.set(id, { resolve, reject, onProgress });
+      this.#worker?.postMessage({ id, type, data });
+    });
+  }
+
+  #ensureWorker() {
+    if (this.#worker) return;
+
+    this.#worker = new Worker(WORKER_URL, { type: "module" });
+    this.#worker.addEventListener("message", this.#handleMessage);
+    this.#worker.addEventListener("error", this.#handleWorkerError);
+  }
+
+  terminate() {
+    const error = new Error("Source search stopped");
+    for (const request of this.#pending.values()) request.reject(error);
+    this.#pending.clear();
+    this.#worker?.terminate();
+    this.#worker = undefined;
+  }
+
+  /** @param {MessageEvent} event */
+  #handleMessage = (event) => {
+    const message = event.data;
+    const request = this.#pending.get(message.id);
+    if (!request) return;
+
+    if (message.status === "progress") {
+      request.onProgress?.({ loaded: message.loaded, total: message.total });
+      return;
+    }
+
+    this.#pending.delete(message.id);
+    if (message.status === "complete") request.resolve(message.data);
+    else request.reject(new Error(message.data));
+  };
+
+  /** @param {ErrorEvent} event */
+  #handleWorkerError = (event) => {
+    const error = new Error(event.message || "The source worker failed");
+    for (const request of this.#pending.values()) request.reject(error);
+    this.#pending.clear();
+    this.#worker?.terminate();
+    this.#worker = undefined;
+  };
+}