@deepseek-ai/dsh-api-session-controller 0.2.0-rc.2 → 0.2.1-alpha.2

package/lib/index.js

@@ -17,6 +17,8 @@ import { SessionTitleInvalidError } from "@deepseek-ai/dsh-session-title";
 import { canonicalClientTimeZone } from "@deepseek-ai/dsh-util-time";
 import { assertNever } from "@deepseek-ai/dsh-util-values";
 import { Deque } from "@deepseek-ai/dsh-deque";
+import { performance } from "node:perf_hooks";
+import { scheduler } from "node:timers/promises";
 import { z as z$1 } from "zod";
 import { isUserInvocable } from "@deepseek-ai/dsh-skill";
 import { FsError } from "@deepseek-ai/dsh-fs";
@@ -1831,9 +1833,14 @@ function truncateUnicodeCodePoints(value, maximum) {
 /** Owns list projection registration, bounded cold summaries, and authorized search. */
 var ApiSessionList = class {
 	ctx;
-	/** @param ctx - Host context carrying Session, query, persistence, and projection services. */
-	constructor(ctx) {
+	workSliceMs;
+	/**
+	* @param ctx - Host context carrying Session, query, persistence, and projection services.
+	* @param workSliceMs - Resolved positive integral list-work budget in milliseconds.
+	*/
+	constructor(ctx, workSliceMs) {
 		this.ctx = ctx;
+		this.workSliceMs = workSliceMs;
 		ctx.sessionProjections.register({
 			key: "sessionListMetadata",
 			stateSchema: sessionListMetadataSchema,
@@ -1874,6 +1881,7 @@ var ApiSessionList = class {
 			sessionId: session.id,
 			updatedAt: updatedAt(session.header, metadata),
 			agentAvailable: this.ctx.agents.get(session.id)?.session === session,
+			formatStatus: "current",
 			running: this.ctx.agents.get(session.id)?.status === "running",
 			blank: metadata?.blank ?? session.seq === 0,
 			...listFields(session.header),
@@ -1882,7 +1890,7 @@ var ApiSessionList = class {
 	}
 	/**
 	* Read every visible attached and persisted Session without activating an Agent.
-	* @param signal - optional cancellation for persistence reads.
+	* @param signal - optional cancellation for persistence reads and summary generation.
 	* @returns visible Session summaries ordered by activity.
 	*/
 	async list(signal) {
@@ -1891,26 +1899,39 @@ var ApiSessionList = class {
 		signal?.throwIfAborted();
 		const items = [];
 		const cold = [];
+		let yieldDeadline = performance.now() + this.workSliceMs;
 		for (const record of records) {
+			signal?.throwIfAborted();
 			const live = this.ctx.sessions.get(record.header.id);
-			if (live !== void 0) {
-				items.push(this.summaryFor(live));
-				continue;
+			if (live !== void 0) items.push(this.summaryFor(live));
+			else if (record.header.cwd !== void 0) cold.push(record);
+			if (performance.now() >= yieldDeadline) {
+				await scheduler.yield();
+				signal?.throwIfAborted();
+				yieldDeadline = performance.now() + this.workSliceMs;
+			}
+		}
+		for (const record of cold) {
+			signal?.throwIfAborted();
+			items.push(this.summarizeCold(record));
+			if (performance.now() >= yieldDeadline) {
+				await scheduler.yield();
+				signal?.throwIfAborted();
+				yieldDeadline = performance.now() + this.workSliceMs;
 			}
-			if (record.header.cwd === void 0) continue;
-			cold.push(record.header);
 		}
-		for (const header of cold) items.push(this.summarizeCold(header));
+		signal?.throwIfAborted();
 		items.sort((left, right) => right.updatedAt - left.updatedAt);
 		return items;
 	}
-	summarizeCold(header) {
+	summarizeCold({ header, formatStatus }) {
 		const projections = this.projectionsFor(header, void 0);
 		const metadata = projections?.values.sessionListMetadata;
 		return {
 			sessionId: header.id,
 			updatedAt: updatedAt(header, metadata),
 			agentAvailable: false,
+			...formatStatus === void 0 ? {} : { formatStatus },
 			running: false,
 			blank: metadata?.blank ?? false,
 			...listFields(header),
@@ -2242,7 +2263,8 @@ let SessionSkillCatalog = (() => {
 		static inject = [
 			"agents",
 			"sessionQuery",
-			"typert"
+			"typert",
+			"workingDirectory"
 		];
 		/** @param ctx - Host context carrying Session reads and optional skill/preset services. */
 		constructor(ctx) {
@@ -2275,8 +2297,10 @@ let SessionSkillCatalog = (() => {
 					try {
 						const observation = __addDisposableResource$1(env_2, await this.ctx.sessionQuery.observeSession(sessionId), false);
 						if (observation.projections === void 0) throw new Error("skill catalog requires a projected Session observation");
-						cwd = observation.header.cwd;
+						cwd = observation.projections.values.workingDirectory ?? observation.header.cwd ?? this.ctx.workingDirectory.defaultDirectory;
 						agentPreset = observation.projections.values.agentPreset ?? void 0;
+						const live = this.ctx.agents.get(sessionId);
+						if (live !== void 0) cwd = await this.ctx.workingDirectory.ensure(live, signal);
 					} catch (e_1) {
 						env_2.error = e_1;
 						env_2.hasError = true;
@@ -2287,7 +2311,6 @@ let SessionSkillCatalog = (() => {
 					if (error instanceof SessionQueryError && error.code === "SESSION_QUERY_SESSION_NOT_FOUND") throw new RemoteError("session/not-found", `session "${sessionId}" not found`, { sessionId });
 					throw new RemoteError("gateway/internal", `session "${sessionId}" could not be inspected: ${String(error)}`, {});
 				}
-				if (cwd === void 0) throw new RemoteError("gateway/internal", `session "${sessionId}" has no project cwd`, {});
 				const live = this.ctx.agents.get(sessionId);
 				const presets = this.ctx.get("agentPresets");
 				const skillRegistry = (live === void 0 ? void 0 : presets?.serviceFor(live, "skills")) ?? this.ctx.get("skills");
@@ -2826,7 +2849,10 @@ let SessionController = (() => {
 			"typert",
 			"workspaceRegistry"
 		];
-		static Config = z.object({ nativeOpen: z.boolean() });
+		static Config = z.object({
+			nativeOpen: z.boolean(),
+			listWorkSliceMs: z.natural().min(1).default(16)
+		});
 		agents = __runInitializers(this, _instanceExtraInitializers);
 		commands;
 		controlState;
@@ -2840,11 +2866,12 @@ let SessionController = (() => {
 		promotions = /* @__PURE__ */ new Set();
 		/**
 		* @param ctx - Host context containing the Session capability assembly.
-		* @param config - native-opener deployment policy.
+		* @param config - native-opener and list-scheduling deployment policy.
 		* @param internals - host integrations replaceable by direct unit tests.
 		*/
 		constructor(ctx, config, internals = {}) {
 			super(ctx, "sessionController", { namespace: "session" });
+			const resolved = SessionController.Config(config);
 			installModelSelectionProjection(ctx);
 			this.agents = new ApiSessionAgentController(ctx);
 			this.commands = new SessionCommandController(ctx, this.agents, process.cwd());
@@ -2860,7 +2887,7 @@ let SessionController = (() => {
 			this.history = new SessionHistoryController(ctx, (observation) => {
 				this.promote(observation);
 			});
-			this.listState = new ApiSessionList(ctx);
+			this.listState = new ApiSessionList(ctx, resolved.listWorkSliceMs);
 			this.fileApplications = internals.fileApplications ?? nativeFileApplications;
 			this.openFileApplication = internals.openFileApplication ?? openNativeFileApplication;
 			this.openPath = internals.openPath ?? openNativeAssociatedPath;
@@ -2948,7 +2975,7 @@ let SessionController = (() => {
 		/**
 		* Read all visible Session rows without resuming an Agent.
 		* @param _request - reserved empty list request.
-		* @param signal - cancellation for persistence reads.
+		* @param signal - cancellation for persistence reads and summary generation.
 		* @returns visible Session summaries ordered by activity.
 		*/
 		async list(_request, signal) {
@@ -3081,10 +3108,13 @@ let SessionController = (() => {
 		* Fork one cold-readable exact event prefix into a new Session. An omitted
 		* boundary selects the latest completed-turn prefix; an open cut receives
 		* synthetic fork closers.
-		* @param request - source Session and optional exact inclusive event boundary.
+		* @param request - source Session, optional exact inclusive event boundary, and migration preflight choice.
 		* @returns the new Session identity.
 		*/
-		fork(request) {
+		async fork(request) {
+			if (request.allowMigration === false && this.ctx.sessions.get(request.sessionId) === void 0) {
+				if ((await this.ctx.get("sessionPersistence")?.stat(request.sessionId))?.formatStatus === "migration-required" && this.ctx.sessions.get(request.sessionId) === void 0) throw new RemoteError("session/migration-required", "Open the source Session to complete migration before forking", { sessionId: request.sessionId });
+			}
 			return this.commands.fork(request);
 		}
 		/**
@@ -3141,10 +3171,10 @@ let SessionController = (() => {
 			return this.history.follow(request, signal);
 		}
 		/**
-		* Read all registered projections without activating an Agent.
+		* Read exact projections, returning cached hints only when migration is required.
 		* @param request - Session whose current values are required.
-		* @param signal - cancellation for the Session observation.
-		* @returns complete baseline, or null when the Session does not exist.
+		* @param signal - cancellation for the Session read.
+		* @returns a sequenced baseline, cached hints when migration is deferred, or null when absent.
 		*/
 		async projections(request, signal) {
 			const { sessionId } = request;
@@ -3156,9 +3186,22 @@ let SessionController = (() => {
 					hasError: false
 				};
 				try {
+					signal.throwIfAborted();
+					const live = this.liveProjections(sessionId);
+					if (live !== void 0) return live;
+					const stored = await this.ctx.get("sessionPersistence")?.stat(sessionId, { signal });
+					signal.throwIfAborted();
+					const attached = this.liveProjections(sessionId);
+					if (attached !== void 0) return attached;
+					if (stored === void 0) return null;
+					if (stored.formatStatus === "migration-required") return {
+						kind: "migration-required",
+						values: { ...(this.ctx.get("sessionProjectionCache")?.cachedSnapshot(stored.header))?.values }
+					};
 					const projections = __addDisposableResource(env_2, await this.ctx.sessionQuery.observeSession(sessionId, { signal }), false).projections;
 					if (projections === void 0) throw new RemoteError("session/projections-unavailable", "Session projections are unavailable", {});
 					return {
+						kind: "sequenced",
 						asOfSeq: projections.asOfSeq,
 						values: projections.values
 					};
@@ -3169,12 +3212,22 @@ let SessionController = (() => {
 					__disposeResources(env_2);
 				}
 			} catch (error) {
-				if (error instanceof SessionQueryError && error.code === "SESSION_QUERY_SESSION_NOT_FOUND") return null;
+				if (error instanceof SessionQueryError && error.code === "SESSION_QUERY_SESSION_NOT_FOUND") return this.liveProjections(sessionId) ?? null;
 				if (signal.aborted || error instanceof SessionQueryError && error.code === "SESSION_QUERY_ABORTED") throw new RemoteError("gateway/cancelled", "Session projection read was cancelled", {}, { cause: error });
 				if (error instanceof RemoteError) throw error;
 				throw new RemoteError("gateway/internal", "Session projection read failed", {}, { cause: error });
 			}
 		}
+		liveProjections(sessionId) {
+			const session = this.ctx.sessions.get(sessionId);
+			if (session === void 0) return void 0;
+			const snapshot = this.ctx.sessionProjections.snapshot(session);
+			return {
+				kind: "sequenced",
+				asOfSeq: snapshot.asOfSeq,
+				values: snapshot.values
+			};
+		}
 		/**
 		* Stream a complete live-control baseline followed by replacement frames.
 		* @param signal - cancellation owned by the Remote stream carrier.