@kubb/studio 5.4.2 → 5.4.3

package/dist/index.js

@@ -1,5 +1,5 @@
 import { t as __name } from "./rolldown-runtime-CRm0XQPb.js";
-import { AGENT_INSTANCE_HEADER, AgentCloseCode, GENERATION_GONE_MESSAGE, generationEventTypes } from "./protocol.js";
+import { AGENT_INSTANCE_HEADER, AgentCloseCode, GENERATION_GONE_MESSAGE, MAX_PAGE_LINES, generationEventTypes } from "./protocol.js";
 import { createRequire } from "node:module";
 import { createHash, hash, randomBytes, randomUUID } from "node:crypto";
 import { setTimeout as setTimeout$1 } from "node:timers/promises";
@@ -695,7 +695,7 @@ function resolveAgentCapacity(env = process.env) {
 }
 //#endregion
 //#region package.json
-var version = "5.4.2";
+var version = "5.4.3";
 //#endregion
 //#region src/operations/resolveConfig.ts
 /**
@@ -1696,20 +1696,35 @@ function createGenerationStore({ storage, maxCount, maxMb, ttlMs, now = Date.now
 			await storage.writeItem(INDEX_KEY, JSON.stringify(entries));
 		},
 		/**
-		* Reads the requested paths the set holds, skipping any it does not.
+		* Reads the requested paths the set holds, skipping any it does not, one page of lines when `limit` is given.
 		*/
-		async read({ generation, source, paths }) {
+		async read({ generation, source, paths, cursor = 0, limit }) {
 			const kept = new Set(generation[source]?.paths);
 			const files = {};
+			const pages = {};
 			await inParallel({
 				items: paths.filter((path) => kept.has(path)),
 				limit: READ_CONCURRENCY,
 				run: async (path) => {
 					const content = await storage.readItem(`${dirOf(generation.jobId)}${source}/${path}`);
-					if (content !== null) files[path] = content;
+					if (content === null) return;
+					if (limit === void 0) {
+						files[path] = content;
+						return;
+					}
+					const lines = content.split(/\r\n|\r|\n/);
+					const end = Math.min(cursor + limit, lines.length);
+					files[path] = lines.slice(cursor, end).join("\n");
+					pages[path] = {
+						nextCursor: end < lines.length ? end : null,
+						totalLines: lines.length
+					};
 				}
 			});
-			return files;
+			return limit === void 0 ? { files } : {
+				files,
+				pages
+			};
 		}
 	};
 }
@@ -2375,6 +2390,11 @@ function planEnd(close) {
 		retry: true
 	};
 }
+function isValidPage({ cursor, limit }) {
+	if (limit === void 0) return cursor === void 0;
+	if (cursor !== void 0 && (!Number.isSafeInteger(cursor) || cursor < 0)) return false;
+	return Number.isSafeInteger(limit) && limit >= 1 && limit <= 2e3;
+}
 /**
 * One agent-to-Studio RPC transport: opening it, keeping it alive, and serving remote methods.
 * `runConnection` owns the lifetime and retry of each attempt.
@@ -2790,7 +2810,7 @@ var StudioSession = class {
 		const missing = generation.missingDependencies.filter((dependency) => !bundled.has(dependency));
 		if (missing.length) return this.#refuse(`Ignored snapshot: missing dependencies: ${missing.join(", ")}`, `Missing dependencies: ${missing.join(", ")}`);
 		try {
-			const files = await this.#generations.read({
+			const { files } = await this.#generations.read({
 				generation,
 				source: "output",
 				paths: generation.output.paths
@@ -2835,18 +2855,21 @@ var StudioSession = class {
 		if (typeof data.jobId !== "string" || !data.jobId) return this.#refuse("Ignored files: the message named no job", "The request named no generation job");
 		const generation = await this.#generations.get(data.jobId);
 		if (!generation) return this.#refuse(`Ignored files: job ${data.jobId} is not kept on this agent`, GENERATION_GONE_MESSAGE);
+		if (!isValidPage(data)) return this.#refuse("Ignored files: the message carried an invalid page", `The cursor must be a whole line from 0 and come with a limit of 1 to ${MAX_PAGE_LINES} lines`);
 		const source = data.source === "disk" ? "disk" : "output";
 		if (!generation[source]) return this.#refuse("Ignored files: that job has no snapshot of the files on disk", "This agent kept no snapshot of the files on disk for that job");
-		const files = await this.#generations.read({
+		const result = await this.#generations.read({
 			generation,
 			source,
-			paths
+			paths,
+			cursor: data.cursor,
+			limit: data.limit
 		});
 		await this.#hooks.callHook("studio:command:end", {
 			command,
-			info: `read ${Object.keys(files).length}/${paths.length} requested file${paths.length === 1 ? "" : "s"}`
+			info: `read ${Object.keys(result.files).length}/${paths.length} requested file${paths.length === 1 ? "" : "s"}`
 		});
-		return { files };
+		return result;
 	}
 };
 //#endregion