@moonshot-ai/kimi-code 0.43.0 → 2.0.0

package/dist/search-worker.mjs

@@ -12430,6 +12430,60 @@ function syncBudgetExhausted(budget) {
 function errorMessage(error) {
 	return error instanceof Error ? error.message : String(error);
 }
+async function sessionDirectoryIdentity(dir) {
+	try {
+		const info = await stat(dir, { bigint: true });
+		if (!info.isDirectory() || info.ino <= 0n || info.birthtimeNs <= 0n) return void 0;
+		return `${info.dev}:${info.ino}:${info.birthtimeNs}`;
+	} catch (error) {
+		const code = error.code;
+		if (code === "ENOENT" || code === "ENOTDIR") return void 0;
+		throw error;
+	}
+}
+const querySourceChecks = {
+	running: false,
+	waiters: /* @__PURE__ */ new Set()
+};
+async function queryDirectoryIdentity(dir, deadlineAt, deadline) {
+	while (querySourceChecks.running) {
+		let wake;
+		const available = new Promise((resolve) => {
+			wake = () => {
+				resolve(true);
+			};
+		});
+		querySourceChecks.waiters.add(wake);
+		try {
+			if (await Promise.race([available, deadline]) === null) return null;
+		} finally {
+			querySourceChecks.waiters.delete(wake);
+		}
+	}
+	if (Date.now() >= deadlineAt) return null;
+	querySourceChecks.running = true;
+	const identity = sessionDirectoryIdentity(dir);
+	const release = () => {
+		querySourceChecks.running = false;
+		for (const wake of querySourceChecks.waiters) wake();
+		querySourceChecks.waiters.clear();
+	};
+	identity.then(release, release);
+	return Promise.race([identity, deadline]);
+}
+async function sessionDirectoryTitle(dir, log) {
+	for (const scope of ["", "session-meta"]) try {
+		const meta = JSON.parse(await readFile(join(dir, scope, "state.json"), "utf8"));
+		if (typeof meta === "object" && meta !== null && "title" in meta && typeof meta.title === "string") return meta.title;
+		return "";
+	} catch (error) {
+		if (error.code !== "ENOENT") log.warn("search index: cannot read session title", {
+			dir,
+			error: errorMessage(error)
+		});
+	}
+	return "";
+}
 const INITIAL_TURN_STATE = {
 	next: 0,
 	hasTurn: false,
@@ -12780,6 +12834,7 @@ var SearchIndexCore = class {
 				break;
 			}
 			const result = await this.syncSession(db, summary, budget);
+			if (result === void 0) continue;
 			if (result.truncated) truncated = true;
 			if (result.failed) {
 				const count = (this.sessionSyncFailures.get(summary.id) ?? 0) + 1;
@@ -12811,6 +12866,7 @@ var SearchIndexCore = class {
 		}).length;
 		const stats = {
 			kind: "stats",
+			degraded: indexed < sessions.length ? `Skipped ${sessions.length - indexed} session(s) during indexing` : void 0,
 			sessions: indexed,
 			documents: db.size - metaCount,
 			lastIndexedAt: Date.now()
@@ -12861,6 +12917,34 @@ var SearchIndexCore = class {
 		await db.del(SESSION_META_PREFIX + sessionId);
 	}
 	async syncSession(db, summary, budget) {
+		let identity;
+		try {
+			identity = await sessionDirectoryIdentity(summary.dir);
+		} catch (error) {
+			this.log.warn("global search: failed to index session", {
+				sessionId: summary.id,
+				error: errorMessage(error)
+			});
+			return;
+		}
+		if (identity === void 0) {
+			await this.deleteSessionDocs(db, summary.id);
+			return;
+		}
+		const title = await sessionDirectoryTitle(summary.dir, this.log);
+		const metaKey = SESSION_META_PREFIX + summary.id;
+		const previous = db.get(metaKey);
+		if (previous?.kind !== "sessionMeta" || previous.identity !== identity || previous.dir !== summary.dir) {
+			await this.deleteSessionDocs(db, summary.id);
+			if (previous !== void 0) this.syncReplaced = true;
+		}
+		const meta = {
+			kind: "sessionMeta",
+			dir: summary.dir,
+			identity,
+			title
+		};
+		if (previous?.kind !== "sessionMeta" || previous.identity !== identity || previous.dir !== summary.dir || previous.title !== title) await db.set(metaKey, meta);
 		const wireFiles = await collectWireFiles(summary.dir);
 		const seenPaths = new Set(wireFiles.map((file) => file.path));
 		for (const row of db.query({ key: { prefix: fileMetaPrefixFor(summary.id) } })) {
@@ -12874,20 +12958,24 @@ var SearchIndexCore = class {
 		let failed = false;
 		let error;
 		for (const file of wireFiles) {
-			const result = await this.syncWireFile(db, summary, file, budget);
+			const result = await this.syncWireFile(db, {
+				...summary,
+				title,
+				sessionIdentity: identity
+			}, file, budget);
 			if (result.truncated) truncated = true;
 			if (result.failed) {
 				failed = true;
 				error ??= result.error;
 			}
 		}
-		const title = summary.title ?? "";
 		const titleKey = `${summary.id}/$title`;
 		const existing = db.get(titleKey);
 		if (title.length > 0) {
 			if (existing?.kind !== "title" || existing.text !== title) {
 				const doc = {
 					kind: "title",
+					sessionIdentity: identity,
 					sessionId: summary.id,
 					workspaceId: summary.workspaceId,
 					sessionTitle: title,
@@ -12900,7 +12988,6 @@ var SearchIndexCore = class {
 				if (existing !== void 0) this.syncReplaced = true;
 			}
 		} else if (existing !== void 0) await db.del(titleKey);
-		if (db.get(SESSION_META_PREFIX + summary.id) === void 0) await db.set(SESSION_META_PREFIX + summary.id, { kind: "sessionMeta" });
 		return {
 			truncated,
 			failed,
@@ -13089,6 +13176,7 @@ var SearchIndexCore = class {
 			const stepOrdinal = e.stepUuid !== void 0 ? stepState.byUuid[e.stepUuid] : void 0;
 			const doc = {
 				kind: "message",
+				sessionIdentity: summary.sessionIdentity,
 				sessionId: summary.id,
 				workspaceId: summary.workspaceId,
 				sessionTitle: summary.title ?? "",
@@ -13208,14 +13296,69 @@ var SearchIndexCore = class {
 		const boundary = page.kind === "keyset" ? page.boundary : void 0;
 		const matched = matchDocs(q, candidates, boundary, budget);
 		incomplete ??= matched.incomplete;
-		const { pageRows, hasMore } = paginateRows(q, page, matched.rows);
+		const index = this.readIndexView(serveDb, freshnessStale);
+		const sources = /* @__PURE__ */ new Map();
+		for (const row of matched.rows) {
+			const id = row.value.sessionId;
+			if (sources.has(id)) continue;
+			if (Date.now() > budget.deadlineAt) {
+				incomplete ??= "deadline";
+				break;
+			}
+			const meta = serveDb.get(SESSION_META_PREFIX + id);
+			sources.set(id, meta?.kind === "sessionMeta" ? meta : void 0);
+		}
+		const identities = /* @__PURE__ */ new Map();
+		const visible = [];
+		let deadlineTimer;
+		const deadline = new Promise((resolve) => {
+			deadlineTimer = setTimeout(() => resolve(null), Math.max(0, budget.deadlineAt - Date.now()));
+			deadlineTimer.unref?.();
+		});
+		try {
+			for (const row of matched.rows) {
+				if (Date.now() > budget.deadlineAt) {
+					incomplete ??= "deadline";
+					break;
+				}
+				const meta = sources.get(row.value.sessionId);
+				if (meta?.dir === void 0 || row.value.sessionIdentity === void 0 || meta.identity !== row.value.sessionIdentity) {
+					freshnessStale = true;
+					continue;
+				}
+				if (!identities.has(meta.dir)) try {
+					const identity = await queryDirectoryIdentity(meta.dir, budget.deadlineAt, deadline);
+					if (identity === null || Date.now() > budget.deadlineAt) {
+						incomplete ??= "deadline";
+						break;
+					}
+					identities.set(meta.dir, identity);
+				} catch (error) {
+					throw new GlobalSearchError("index_unavailable", `cannot verify search source: ${errorMessage(error)}`);
+				}
+				if (identities.get(meta.dir) === row.value.sessionIdentity) visible.push({
+					...row,
+					value: {
+						...row.value,
+						sessionTitle: meta.title ?? ""
+					}
+				});
+				else freshnessStale = true;
+			}
+		} finally {
+			if (deadlineTimer !== void 0) clearTimeout(deadlineTimer);
+		}
+		const { pageRows, hasMore } = paginateRows(q, page, visible);
 		return {
 			kind: "page",
 			rows: pageRows,
 			hasMore,
 			incomplete,
 			generation,
-			index: this.readIndexView(serveDb, freshnessStale)
+			index: {
+				...index,
+				freshnessStale: freshnessStale || this.db !== serveDb
+			}
 		};
 	}
 	async reindex() {
@@ -13274,7 +13417,7 @@ var SearchIndexCore = class {
 			generation: this.generation,
 			readOnly: this.db?.readOnly === true,
 			lockToken: this.lockToken,
-			degraded: this.lastRefreshError?.message ?? maintenance,
+			degraded: this.lastRefreshError?.message ?? maintenance ?? (stats?.kind === "stats" ? stats.degraded : void 0),
 			lifecycle: this.lifecycleState()
 		};
 	}
@@ -13287,7 +13430,7 @@ var SearchIndexCore = class {
 			documents: stats?.kind === "stats" ? stats.documents : 0,
 			readOnly: handle?.readOnly === true,
 			freshnessStale: true,
-			degraded: this.lastRefreshError?.message,
+			degraded: this.lastRefreshError?.message ?? (stats?.kind === "stats" ? stats.degraded : void 0),
 			lockToken: this.lockToken
 		};
 	}
@@ -13301,7 +13444,7 @@ var SearchIndexCore = class {
 			documents,
 			readOnly: db.readOnly,
 			freshnessStale,
-			degraded: this.lastRefreshError?.message,
+			degraded: this.lastRefreshError?.message ?? (stats?.kind === "stats" ? stats.degraded : void 0),
 			lockToken: this.lockToken
 		};
 	}