@kubb/studio 5.9.4 → 5.9.5

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import { t as __name } from "./rolldown-runtime-CRm0XQPb.js";
-import { L as SnapshotUploadMetadata, a as AgentConnectInfo, c as ConfigEdit, j as PublishedSnapshot, l as ConfigEditOutcome, o as AgentPermissions, w as InstalledExtensions } from "./protocol-gXoAXvT3.js";
+import { M as PublishedSnapshot, V as SnapshotUploadMetadata, a as AgentConnectInfo, c as ConfigEdit, l as ConfigEditOutcome, o as AgentPermissions, w as InstalledExtensions } from "./protocol-ixLkhvRJ.js";
 import { Config, Hookable, KubbHooks } from "@kubb/core";
 //#region src/connect.d.ts
 /**
@@ -46,7 +46,7 @@ declare global {
        * A finished command and what it did, such as `applied 2/3 edits to kubb.config.ts`.
        */
       'studio:command': [ctx: {
-        command: 'generate' | 'saveConfig' | 'readFiles';
+        command: 'generate' | 'saveConfig' | 'readFiles' | 'readInput';
         info: string;
       }];
       /**

package/dist/index.js

@@ -1,5 +1,5 @@
 import "./rolldown-runtime-CRm0XQPb.js";
-import { AGENT_INSTANCE_HEADER, AGENT_SOCKET_PATH, AgentCloseCode, GENERATION_GONE_MESSAGE, MAX_PAGE_LINES, MAX_SNAPSHOT_BYTES, generationEventTypes } from "./protocol.js";
+import { AGENT_INSTANCE_HEADER, AGENT_SOCKET_PATH, AgentCloseCode, GENERATION_GONE_MESSAGE, MAX_PAGE_LINES, MAX_SNAPSHOT_BYTES, NO_INPUT_MESSAGE, READ_NOT_ALLOWED_MESSAGE, generationEventTypes } from "./protocol.js";
 import { createRequire } from "node:module";
 import { createHash, hash, randomUUID } from "node:crypto";
 import { mkdir, mkdtemp, readFile, rm, writeFile } from "node:fs/promises";
@@ -13,6 +13,7 @@ import { Diagnostics, Hookable, cacheStorage, createKubb, fsStorage, getInputKin
 import { RpcTarget, newWebSocketRpcSession } from "capnweb";
 import { detectCodeFormat, parseExpression, parseModule } from "magicast";
 import { stripVTControlCharacters } from "node:util";
+import { ast } from "@kubb/ast";
 import { ofetch } from "ofetch";
 import { gzipSync } from "node:zlib";
 //#region ../../internals/utils/src/casing.ts
@@ -152,7 +153,7 @@ function getElapsedMs(hrStart) {
 }
 //#endregion
 //#region package.json
-var version = "5.9.4";
+var version = "5.9.5";
 //#endregion
 //#region src/resolveConfig.ts
 /**
@@ -1083,7 +1084,10 @@ function relativeStoragePath({ root, filePath }) {
 * The content fingerprint Studio compares snapshots by. Agent runs and CI uploads must agree on it.
 */
 const hashOf = (content) => createHash("sha1").update(content).digest("hex").slice(0, 16);
-const totalBytes = (entries) => entries.reduce((sum, { output, disk }) => sum + output.bytes + (disk?.bytes ?? 0), 0);
+/**
+* The bytes `entries` keep on disk: output, written files, and kept input.
+*/
+const totalBytes = (entries) => entries.reduce((sum, { output, disk, inputBytes }) => sum + output.bytes + (disk?.bytes ?? 0) + (inputBytes ?? 0), 0);
 function positiveNumber(value) {
 	const parsed = Number(value);
 	return value && Number.isFinite(parsed) && parsed > 0 ? parsed : void 0;
@@ -1126,7 +1130,11 @@ async function generate({ config, hooks, signal, allowExec }) {
 		hooks,
 		signal
 	});
-	const result = await kubb.generate();
+	let input = null;
+	const unhook = kubb.hooks.hook("kubb:build:end", (ctx) => {
+		if (ctx.config === kubb.config) input = kubb.driver.inputNode;
+	});
+	const result = await kubb.generate().finally(unhook);
 	if (!result.success) throw generationFailure(result.diagnostics);
 	const { storage, root } = kubb.config;
 	return {
@@ -1135,7 +1143,8 @@ async function generate({ config, hooks, signal, allowExec }) {
 		paths: new Set(result.files.map((file) => relativeStoragePath({
 			root,
 			filePath: file.path
-		})))
+		}))),
+		input
 	};
 }
 /**
@@ -1216,6 +1225,33 @@ function createGenerationStore({ storage, maxCount, maxMb, ttlMs, now = Date.now
 	return {
 		keep,
 		drop,
+		/**
+		* Keeps the `InputNode` of `jobId` and returns its size in bytes, or 0 when it keeps none. A ref schema loses
+		* the copy of its target, which `input.schemas` holds: with it, a large spec outgrows what `JSON.stringify` can return.
+		*/
+		async keepInput({ jobId, input, maxSetMb }) {
+			if (!input) return 0;
+			let json;
+			try {
+				json = JSON.stringify(ast.transform(input, { schema: (node) => node.type === "ref" && node.schema ? {
+					...node,
+					schema: void 0
+				} : void 0 }));
+			} catch {
+				return 0;
+			}
+			const bytes = Buffer.byteLength(json);
+			if (bytes > maxSetMb * MB) return 0;
+			await storage.writeItem(`${dirOf(jobId)}input.json`, json);
+			return bytes;
+		},
+		/**
+		* The input `keepInput` kept for `jobId`, or `undefined` when there is none.
+		*/
+		async readInput(jobId) {
+			const stored = await storage.readItem(`${dirOf(jobId)}input.json`);
+			return stored === null ? void 0 : JSON.parse(stored);
+		},
 		get: async (jobId) => (await load()).find((generation) => generation.jobId === jobId && isLive(generation)),
 		async add(generation) {
 			const current = await load();
@@ -1666,6 +1702,7 @@ var Agent = class extends RpcTarget {
 			}
 		});
 	}
+	/** Runs one job and keeps its output, and its input with `allowRead`, for later reads. */
 	async #runGeneration(input, controller) {
 		this.#assertConnected();
 		if (this.#session.closed) throw new Error("Connection is closed");
@@ -1753,11 +1790,17 @@ var Agent = class extends RpcTarget {
 			files,
 			maxSetMb: this.#limits.maxMb
 		});
+		const inputBytes = await this.#generations.keepInput({
+			jobId,
+			input: this.#permissions.allowRead ? files.input : null,
+			maxSetMb: this.#limits.maxMb
+		});
 		if (!output.paths.length && Object.keys(output.hashes).length) await this.#warn({ message: "Kept only the hashes of this generation: its output is too large to keep" });
 		await this.#generations.add({
 			jobId,
 			output,
-			disk
+			disk,
+			inputBytes
 		});
 		await this.#logCommand("generate", `${activePlugins.length} plugin${activePlugins.length === 1 ? "" : "s"}, ${this.#permissions.allowWrite ? "written to disk" : "in memory"}${studioInput !== void 0 ? ", from a Studio spec" : ""}`);
 		return {
@@ -1809,10 +1852,30 @@ var Agent = class extends RpcTarget {
 			return refuse(getErrorMessage(error));
 		}
 	}
+	/**
+	* Returns the Kubb AST a kept job parsed its spec into. Refused without `allowRead`.
+	*/
+	async readInput(input) {
+		this.#assertConnected();
+		if (!this.#permissions.allowRead) return this.#refuse({
+			message: READ_NOT_ALLOWED_MESSAGE,
+			permission: "allowRead"
+		});
+		if (typeof input.jobId !== "string" || !input.jobId) return this.#refuse({ message: "The request must name a job" });
+		const generation = await this.#generations.get(input.jobId);
+		if (!generation) return this.#refuse({ message: GENERATION_GONE_MESSAGE });
+		const kept = generation.inputBytes ? await this.#generations.readInput(input.jobId) : void 0;
+		if (!kept) return this.#refuse({ message: NO_INPUT_MESSAGE });
+		await this.#logCommand("readInput", "read the API input");
+		return { input: kept };
+	}
+	/**
+	* Returns the requested files of a kept job, one page of lines at a time. Refused without `allowRead`.
+	*/
 	async readFiles(input) {
 		this.#assertConnected();
 		if (!this.#permissions.allowRead) return this.#refuse({
-			message: "The agent was not granted permission to read generated files",
+			message: READ_NOT_ALLOWED_MESSAGE,
 			permission: "allowRead"
 		});
 		if (!Array.isArray(input.paths) || typeof input.jobId !== "string" || !input.jobId) return this.#refuse({ message: "The request must name a job and carry paths" });