@kubb/studio 5.11.2 → 5.11.4

package/README.md

@@ -85,9 +85,9 @@ running in a developer's project and a container running a fixed plugin set.
 the agent is a sandbox and carries its slug and Studio's version. From then on the agent pings every
 30 seconds, and Studio schedules one job at a time on the socket.
 
-With `allowRead`, the connect payload also includes a kept copy of the current `output.path` files.
-Studio can browse these files through `readFiles` before running a generation. Sandbox agents have
-no disk output, and output paths outside the project root are excluded.
+With `allowRead`, the connect payload includes the current `output.path` file list. Studio reads
+file contents from disk through `readFiles` when a user opens a file. Sandbox agents have no disk
+output, and output paths outside the project root are excluded.
 
 Studio ends a socket on purpose with one of these close codes:
 

package/dist/index.cjs

@@ -23,7 +23,6 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 //#endregion
 const require_protocol = require("./protocol.cjs");
 let node_crypto = require("node:crypto");
-let node_fs_promises = require("node:fs/promises");
 let node_os = require("node:os");
 let node_path = require("node:path");
 node_path = __toESM(node_path, 1);
@@ -32,6 +31,7 @@ node_process = __toESM(node_process, 1);
 let node_timers_promises = require("node:timers/promises");
 let node_url = require("node:url");
 let node_fs = require("node:fs");
+let node_fs_promises = require("node:fs/promises");
 let _kubb_core = require("@kubb/core");
 let capnweb = require("capnweb");
 let magicast = require("magicast");
@@ -141,6 +141,27 @@ async function read(path) {
 	return (0, node_fs_promises.readFile)(path, { encoding: "utf8" });
 }
 /**
+* Writes UTF-8 content verbatim, including empty files and surrounding whitespace.
+* Uses Bun.write under Bun and creates missing parent directories under either runtime.
+*
+* @example
+* await writeRaw('./cache/output.ts', '  // keep these bytes\n\n')
+*/
+async function writeRaw(path, content) {
+	const resolved = (0, node_path.resolve)(path);
+	if (runtime.isBun) {
+		await Bun.write(resolved, content);
+		return;
+	}
+	try {
+		await (0, node_fs_promises.writeFile)(resolved, content, { encoding: "utf-8" });
+	} catch (error) {
+		if (error.code !== "ENOENT") throw error;
+		await (0, node_fs_promises.mkdir)((0, node_path.dirname)(resolved), { recursive: true });
+		await (0, node_fs_promises.writeFile)(resolved, content, { encoding: "utf-8" });
+	}
+}
+/**
 * Resolves to `true` when `path` is `parent` itself or nested inside it. Both sides are resolved
 * to absolute paths first, so relative and `..`-containing inputs compare correctly.
 *
@@ -199,7 +220,7 @@ function getElapsedMs(hrStart) {
 }
 //#endregion
 //#region package.json
-var version = "5.11.2";
+var version = "5.11.4";
 //#endregion
 //#region src/resolveConfig.ts
 /**
@@ -1395,21 +1416,20 @@ const hashOf = (content) => (0, node_crypto.createHash)("sha1").update(content).
 /**
 * 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);
+const totalBytes = (entries) => entries.reduce((sum, { output, inputBytes }) => sum + output.bytes + (inputBytes ?? 0), 0);
 function positiveNumber(value) {
 	const parsed = Number(value);
 	return value && Number.isFinite(parsed) && parsed > 0 ? parsed : void 0;
 }
 /**
 * How many generations an agent keeps and how large they may get, read from
-* `KUBB_AGENT_MAX_GENERATIONS`, `KUBB_AGENT_MAX_GENERATIONS_MB` and `KUBB_AGENT_MAX_SNAPSHOT_MB`.
+* `KUBB_AGENT_MAX_GENERATIONS` and `KUBB_AGENT_MAX_GENERATIONS_MB`.
 * An unset or invalid value keeps the default.
 */
 function resolveGenerationLimits() {
 	return {
 		maxCount: Math.max(1, Math.floor(positiveNumber(node_process.default.env.KUBB_AGENT_MAX_GENERATIONS) ?? 8)),
-		maxMb: positiveNumber(node_process.default.env.KUBB_AGENT_MAX_GENERATIONS_MB) ?? 100,
-		maxSnapshotMb: positiveNumber(node_process.default.env.KUBB_AGENT_MAX_SNAPSHOT_MB) ?? 50
+		maxMb: positiveNumber(node_process.default.env.KUBB_AGENT_MAX_GENERATIONS_MB) ?? 100
 	};
 }
 /**
@@ -1456,7 +1476,7 @@ async function generate({ config, hooks, signal, allowExec }) {
 	};
 }
 /**
-* What the output directory holds on disk before a run. `undefined` when `outputPath` is not a real
+* What the output directory holds when the editor connects. `undefined` when `outputPath` is not a real
 * subdirectory of `root` (listing the root would take in every source file) or holds more than `maxFiles`.
 */
 async function listDisk({ root, outputPath, maxFiles }) {
@@ -1476,8 +1496,8 @@ async function listDisk({ root, outputPath, maxFiles }) {
 /**
 * Keeps recent generations by job id in `storage`, with an index next to them so a store on disk
 * survives a restart. A generation is only ever looked up by its job id, so on a pooled sandbox one
-* tenant never reaches another's output. The oldest go past `maxCount` or `maxMb`, but the newest
-* always stays.
+* tenant never reaches another's output. Initial disk output and generated jobs each get their
+* own `maxCount` and `maxMb` budget. Each budget always keeps its newest entry.
 */
 function createGenerationStore({ storage, maxCount, maxMb, ttlMs, now = Date.now }) {
 	let index;
@@ -1496,7 +1516,7 @@ function createGenerationStore({ storage, maxCount, maxMb, ttlMs, now = Date.now
 	/**
 	* Copies `files` into the store as one set of `jobId`. Above `maxSetMb` only the hashes are kept.
 	*/
-	async function keep({ jobId, source, files, maxSetMb, normalize = source === require_protocol.FileSource.OUTPUT }) {
+	async function keep({ jobId, files, maxSetMb }) {
 		const maxSetBytes = maxSetMb * MB;
 		const hashes = {};
 		let bytes = 0;
@@ -1507,14 +1527,13 @@ function createGenerationStore({ storage, maxCount, maxMb, ttlMs, now = Date.now
 				if (path.split("/").includes("..")) return;
 				const content = await files.storage.readItem((0, node_path.resolve)(files.root, path));
 				if (content === null) return;
-				if (normalize && !content.trim()) return;
-				hashes[path] = hashOf(normalize ? `${content.trim()}\n` : content);
+				hashes[path] = hashOf(content);
 				bytes += Buffer.byteLength(content);
-				if (bytes <= maxSetBytes) await storage.writeItem(`${dirOf(jobId)}${source}/${path}`, content);
+				if (bytes <= maxSetBytes) await storage.writeItem(`${dirOf(jobId)}output/${path}`, content);
 			}
 		});
 		if (bytes > maxSetBytes) {
-			await storage.empty(`${dirOf(jobId)}${source}/`);
+			await storage.empty(`${dirOf(jobId)}output/`);
 			return {
 				paths: [],
 				hashes,
@@ -1530,10 +1549,50 @@ function createGenerationStore({ storage, maxCount, maxMb, ttlMs, now = Date.now
 	async function drop(jobId) {
 		await storage.empty(dirOf(jobId));
 	}
+	async function add(generation) {
+		const current = await load();
+		for (const expired of current.filter((entry) => !isLive(entry))) await drop(expired.jobId);
+		const live = current.filter((entry) => entry.jobId !== generation.jobId && isLive(entry));
+		const sharesBudget = (entry) => Boolean(entry.initial) === Boolean(generation.initial);
+		const entries = live.filter(sharesBudget);
+		entries.push({
+			...generation,
+			keptAt: now()
+		});
+		while (entries.length > 1 && (entries.length > maxCount || totalBytes(entries) > maxMb * MB)) await drop(entries.shift().jobId);
+		index = [...live.filter((entry) => !sharesBudget(entry)), ...entries];
+		await storage.writeItem(INDEX_KEY, JSON.stringify(index));
+	}
 	return {
 		keep,
 		drop,
 		/**
+		* Captures initial disk output, reusing a retained ID when its hashes and readable paths match.
+		* A temporary candidate is discarded on reuse, so reconnects do not consume retention slots.
+		*/
+		async keepInitial(files) {
+			const jobId = (0, node_crypto.randomUUID)();
+			const output = await keep({
+				jobId,
+				files,
+				maxSetMb: maxMb
+			});
+			const hashes = Object.entries(output.hashes);
+			const previous = (await load()).findLast((entry) => entry.initial && isLive(entry) && entry.output.paths.length === output.paths.length && Object.keys(entry.output.hashes).length === hashes.length && hashes.every(([path, hash]) => entry.output.hashes[path] === hash));
+			if (previous) {
+				await drop(jobId);
+				await add(previous);
+				return previous;
+			}
+			const generation = {
+				jobId,
+				output,
+				initial: true
+			};
+			await add(generation);
+			return generation;
+		},
+		/**
 		* 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.
 		*/
@@ -1561,30 +1620,19 @@ function createGenerationStore({ storage, maxCount, maxMb, ttlMs, now = Date.now
 			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();
-			for (const expired of current.filter((entry) => !isLive(entry))) await drop(expired.jobId);
-			const entries = current.filter((entry) => entry.jobId !== generation.jobId && isLive(entry));
-			entries.push({
-				...generation,
-				keptAt: now()
-			});
-			while (entries.length > 1 && (entries.length > maxCount || totalBytes(entries) > maxMb * MB)) await drop(entries.shift().jobId);
-			index = entries;
-			await storage.writeItem(INDEX_KEY, JSON.stringify(entries));
-		},
+		add,
 		/**
 		* 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, cursor = 0, limit }) {
-			const kept = new Set(generation[source]?.paths);
+		async read({ generation, paths, cursor = 0, limit }) {
+			const kept = new Set(generation.output.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}`);
+					const content = await storage.readItem(`${dirOf(generation.jobId)}output/${path}`);
 					if (content === null) return;
 					if (limit === void 0) {
 						files[path] = content;
@@ -1859,6 +1907,25 @@ function isValidPage({ cursor, limit }) {
 	if (cursor !== void 0 && (!Number.isSafeInteger(cursor) || cursor < 0)) return false;
 	return Number.isSafeInteger(limit) && limit >= 1 && limit <= 2e3;
 }
+/**
+* Maps generated paths into a temporary directory while keeping the project root for input resolution.
+*/
+function stagingStorage({ root, directory }) {
+	const storage = (0, _kubb_core.fsStorage)();
+	const target = (key) => {
+		const relative = node_path.default.relative(root, node_path.default.resolve(root, key));
+		if (relative === ".." || relative.startsWith(`..${node_path.default.sep}`) || node_path.default.isAbsolute(relative)) throw new Error("Output path escapes the project root");
+		return node_path.default.join(directory, relative);
+	};
+	return {
+		name: "fs",
+		readItem: (key) => storage.readItem(target(key)),
+		writeItem: (key, value) => storage.writeItem(target(key), value),
+		removeItem: (key) => storage.removeItem(target(key)),
+		readKeys: (base) => storage.readKeys(base ? target(base) : directory),
+		empty: (base) => base ? storage.empty(target(base)) : Promise.resolve()
+	};
+}
 var GenerationRunTarget = class extends capnweb.RpcTarget {
 	#run;
 	constructor(run) {
@@ -1883,6 +1950,7 @@ var Agent = class extends capnweb.RpcTarget {
 	#session;
 	#limits = resolveGenerationLimits();
 	#store;
+	#sandboxCacheRoot = node_path.default.join((0, node_os.tmpdir)(), `kubb-studio-${(0, node_crypto.randomUUID)()}`);
 	#diffs = /* @__PURE__ */ new Map();
 	constructor({ options, session }) {
 		super();
@@ -1904,13 +1972,19 @@ var Agent = class extends capnweb.RpcTarget {
 	}
 	get #generations() {
 		this.#store ??= createGenerationStore({
-			storage: this.#sandbox ? (0, _kubb_core.memoryStorage)() : (0, _kubb_core.cacheStorage)({ root: this.#options.root }),
+			storage: (0, _kubb_core.cacheStorage)({ root: this.#sandbox ? this.#sandboxCacheRoot : this.#options.root }),
 			maxCount: this.#limits.maxCount,
 			maxMb: this.#limits.maxMb,
 			ttlMs: this.#sandbox ? SANDBOX_GENERATION_TTL_MS : void 0
 		});
 		return this.#store;
 	}
+	async [Symbol.asyncDispose]() {
+		if (this.#sandbox) await (0, node_fs_promises.rm)((0, _kubb_core.resolveCacheDir)(this.#sandboxCacheRoot), {
+			recursive: true,
+			force: true
+		});
+	}
 	#warn({ message, permission }) {
 		return emit(this.#options.hooks, "studio:warn", {
 			message,
@@ -2022,18 +2096,7 @@ var Agent = class extends capnweb.RpcTarget {
 			maxFiles: DISK_SNAPSHOT_MAX_FILES
 		});
 		if (!files) return void 0;
-		const jobId = (0, node_crypto.randomUUID)();
-		const output = await this.#generations.keep({
-			jobId,
-			source: require_protocol.FileSource.OUTPUT,
-			files,
-			maxSetMb: this.#limits.maxSnapshotMb,
-			normalize: false
-		});
-		await this.#generations.add({
-			jobId,
-			output
-		});
+		const { jobId, output } = await this.#generations.keepInitial(files);
 		return {
 			jobId,
 			files: output.paths,
@@ -2057,7 +2120,9 @@ var Agent = class extends capnweb.RpcTarget {
 			}
 		});
 	}
-	/** Runs one job and keeps its output, and its input with `allowRead`, for later reads. */
+	/**
+	* Keeps a job's output and permitted parsed input, then cleans up its temporary generation directories.
+	*/
 	async #runGeneration(input, controller) {
 		this.#assertConnected();
 		if (this.#session.closed) throw new _kubb_core.Diagnostics.Error({
@@ -2065,7 +2130,7 @@ var Agent = class extends capnweb.RpcTarget {
 			severity: "error",
 			message: "Connection is closed"
 		});
-		if (this.#session.active) return this.#refuse({ message: "A generation is already in progress, please wait for it to finish" });
+		if (this.#session.active) return this.#refuse({ message: require_protocol.AgentMessage.GENERATION_IN_PROGRESS });
 		const settled = Promise.withResolvers();
 		this.#session.active = {
 			jobId: input.jobId,
@@ -2073,11 +2138,14 @@ var Agent = class extends capnweb.RpcTarget {
 			settled: settled.promise
 		};
 		let root = this.#options.root;
+		let outputRoot;
 		try {
 			if (this.#sandbox) root = await (0, node_fs_promises.mkdtemp)(node_path.default.join((0, node_os.tmpdir)(), "kubb-job-"));
+			if (!this.#sandbox && !this.#permissions.allowWrite) outputRoot = await (0, node_fs_promises.mkdtemp)(node_path.default.join((0, node_os.tmpdir)(), "kubb-output-"));
 			return await this.#generate({
 				input,
 				root,
+				outputRoot,
 				signal: controller.signal
 			});
 		} finally {
@@ -2088,15 +2156,21 @@ var Agent = class extends capnweb.RpcTarget {
 				recursive: true,
 				force: true
 			})]).catch(() => {});
+			if (outputRoot) await (0, node_fs_promises.rm)(outputRoot, {
+				recursive: true,
+				force: true
+			});
 			this.#session.active = void 0;
 			settled.resolve();
 		}
 	}
-	/** Merges Studio's config over the disk config, refuses output paths outside the output folder, then generates in `root`. */
-	async #generate({ input, root, signal }) {
+	/**
+	* Merges Studio's config over the project config and confines generation to the configured output folder.
+	*/
+	async #generate({ input, root, outputRoot, signal }) {
 		const { jobId, config: patch } = input;
 		const studioInput = this.#sandbox ? patch?.input : void 0;
-		if (this.#sandbox && (typeof studioInput !== "string" || !studioInput.trim() || (0, _kubb_core.getInputKind)(studioInput) !== "inline")) return this.#refuse({ message: "A sandbox agent generates only from the OpenAPI document content Studio sends, not from a file path or URL" });
+		if (this.#sandbox && (typeof studioInput !== "string" || !studioInput.trim() || (0, _kubb_core.getInputKind)(studioInput) !== "inline")) return this.#refuse({ message: require_protocol.AgentMessage.SANDBOX_INPUT_ONLY });
 		const [config, extensions] = await Promise.all([this.#options.loadConfig(), this.#installedExtensions()]);
 		const [plugins, adapter, parsers] = await Promise.all([
 			mergePlugins({
@@ -2124,37 +2198,28 @@ var Agent = class extends capnweb.RpcTarget {
 		});
 		const activePlugins = plugins ?? config.plugins;
 		if (patch?.input && !this.#sandbox) await emit(this.#options.hooks, "kubb:warn", { message: "Ignored the spec from Studio: generating from a Studio spec is only available to a sandbox agent" });
-		const diskFiles = !this.#sandbox && this.#permissions.allowRead ? await listDisk({
-			root,
-			outputPath: config.output.path,
-			maxFiles: DISK_SNAPSHOT_MAX_FILES
-		}) : void 0;
-		const disk = diskFiles ? await this.#generations.keep({
-			jobId,
-			source: require_protocol.FileSource.DISK,
-			files: diskFiles,
-			maxSetMb: this.#limits.maxSnapshotMb
-		}) : void 0;
 		const files = await generate({
 			config: {
 				...config,
 				root,
 				input: studioInput ?? config.input,
-				storage: this.#permissions.allowWrite ? (0, _kubb_core.fsStorage)() : (0, _kubb_core.memoryStorage)(),
+				storage: outputRoot ? stagingStorage({
+					root,
+					directory: outputRoot
+				}) : (0, _kubb_core.fsStorage)(),
 				plugins: activePlugins,
 				adapter,
 				parsers
 			},
 			hooks: this.#options.hooks,
 			signal,
-			allowExec: this.#permissions.allowExec
+			allowExec: this.#permissions.allowWrite && this.#permissions.allowExec
 		}).catch(async (error) => {
 			await this.#generations.drop(jobId);
 			throw error;
 		});
 		const output = await this.#generations.keep({
 			jobId,
-			source: require_protocol.FileSource.OUTPUT,
 			files,
 			maxSetMb: this.#limits.maxMb
 		});
@@ -2167,16 +2232,14 @@ var Agent = class extends capnweb.RpcTarget {
 		await this.#generations.add({
 			jobId,
 			output,
-			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" : ""}`);
+		await this.#logCommand("generate", `${activePlugins.length} plugin${activePlugins.length === 1 ? "" : "s"}, ${this.#permissions.allowWrite ? "written to disk" : "staged on disk"}${studioInput !== void 0 ? ", from a Studio spec" : ""}`);
 		return {
 			status: "success",
 			files: [...files.paths],
 			fileCount: files.paths.size,
-			hashes: output.hashes,
-			disk: disk ? { hashes: disk.hashes } : void 0
+			hashes: output.hashes
 		};
 	}
 	async saveConfig(input) {
@@ -2229,7 +2292,7 @@ var Agent = class extends capnweb.RpcTarget {
 			message: require_protocol.AgentMessage.READ_AST_NOT_ALLOWED,
 			permission: require_protocol.AgentPermission.READ_AST
 		});
-		if (typeof input.jobId !== "string" || !input.jobId) return this.#refuse({ message: "The request must name a job" });
+		if (typeof input.jobId !== "string" || !input.jobId) return this.#refuse({ message: require_protocol.AgentMessage.JOB_REQUIRED });
 		const generation = await this.#generations.get(input.jobId);
 		if (!generation) return this.#refuse({ message: require_protocol.AgentMessage.GENERATION_GONE });
 		const kept = generation.inputBytes ? await this.#generations.readInput(input.jobId) : void 0;
@@ -2246,17 +2309,14 @@ var Agent = class extends capnweb.RpcTarget {
 			message: require_protocol.AgentMessage.READ_NOT_ALLOWED,
 			permission: require_protocol.AgentPermission.READ
 		});
-		if (!Array.isArray(input.paths) || typeof input.jobId !== "string" || !input.jobId) return this.#refuse({ message: "The request must name a job and carry paths" });
+		if (!Array.isArray(input.paths) || typeof input.jobId !== "string" || !input.jobId) return this.#refuse({ message: require_protocol.AgentMessage.FILE_REQUEST_INVALID });
 		const { paths } = input;
-		if (paths.length > 50) return this.#refuse({ message: `At most 50 paths may be requested at once` });
+		if (paths.length > 50) return this.#refuse({ message: require_protocol.AgentMessage.TOO_MANY_FILES });
+		if (!isValidPage(input)) return this.#refuse({ message: require_protocol.AgentMessage.INVALID_PAGE });
 		const generation = await this.#generations.get(input.jobId);
 		if (!generation) return this.#refuse({ message: require_protocol.AgentMessage.GENERATION_GONE });
-		if (!isValidPage(input)) return this.#refuse({ message: `The cursor must be a whole line from 0 and come with a limit of 1 to ${require_protocol.MAX_PAGE_LINES} lines` });
-		const source = input.source === require_protocol.FileSource.DISK ? require_protocol.FileSource.DISK : require_protocol.FileSource.OUTPUT;
-		if (!generation[source]) return this.#refuse({ message: "This agent kept no snapshot of the files on disk for that job" });
 		const result = await this.#generations.read({
 			generation,
-			source,
 			paths,
 			cursor: input.cursor,
 			limit: input.limit
@@ -2274,26 +2334,22 @@ var Agent = class extends capnweb.RpcTarget {
 			permission: require_protocol.AgentPermission.READ
 		});
 		const { jobId, path, base, cursor = 0, limit } = input;
-		if (typeof jobId !== "string" || !jobId || typeof path !== "string" || !path || typeof base?.jobId !== "string" || !base.jobId) return this.#refuse({ message: "The request must name a job, a base, and a path" });
+		if (typeof jobId !== "string" || !jobId || typeof path !== "string" || !path || typeof base?.jobId !== "string" || !base.jobId) return this.#refuse({ message: require_protocol.AgentMessage.DIFF_REQUEST_INVALID });
 		if (!isValidPage({
 			cursor,
 			limit
-		})) return this.#refuse({ message: `The cursor must be a whole line from 0 and come with a limit of 1 to ${require_protocol.MAX_PAGE_LINES} lines` });
+		})) return this.#refuse({ message: require_protocol.AgentMessage.INVALID_PAGE });
 		const [generation, baseGeneration] = await Promise.all([this.#generations.get(jobId), this.#generations.get(base.jobId)]);
 		if (!generation || !baseGeneration) return this.#refuse({ message: require_protocol.AgentMessage.GENERATION_GONE });
-		const baseSource = base.source === require_protocol.FileSource.DISK ? require_protocol.FileSource.DISK : require_protocol.FileSource.OUTPUT;
-		if (!baseGeneration[baseSource]) return this.#refuse({ message: "This agent kept no snapshot of the files on disk for that job" });
 		const [after, before] = await Promise.all([this.#generations.read({
 			generation,
-			source: require_protocol.FileSource.OUTPUT,
 			paths: [path]
 		}), this.#generations.read({
 			generation: baseGeneration,
-			source: baseSource,
 			paths: [path]
 		})]);
 		const afterContent = after.files[path];
-		if (afterContent === void 0 && path in generation.output.hashes) return this.#refuse({ message: "This agent kept only the hashes of that job" });
+		if (afterContent === void 0 && path in generation.output.hashes) return this.#refuse({ message: require_protocol.AgentMessage.HASHES_ONLY });
 		const beforeContent = before.files[path];
 		if (beforeContent === void 0) return {
 			hunks: [],
@@ -2310,7 +2366,6 @@ var Agent = class extends capnweb.RpcTarget {
 		const key = JSON.stringify([
 			jobId,
 			base.jobId,
-			baseSource,
 			path
 		]);
 		let diff = this.#diffs.get(key);
@@ -2367,14 +2422,15 @@ async function runSession({ options, openRpc, url, instanceId }) {
 		connected: Promise.withResolvers(),
 		closed: false
 	};
+	const agent = new Agent({
+		options,
+		session
+	});
 	const rpc = openRpc({
 		url,
 		token,
 		instanceId,
-		agent: new Agent({
-			options,
-			session
-		})
+		agent
 	});
 	signal?.addEventListener("abort", rpc.close, { once: true });
 	let timer;
@@ -2409,6 +2465,7 @@ async function runSession({ options, openRpc, url, instanceId }) {
 		await emit(hooks, "studio:warn", { message: "Canceled the running generation: the connection to Kubb Studio closed" });
 	}
 	await active?.settled;
+	await agent[Symbol.asyncDispose]();
 	return {
 		...close,
 		connected: session.info !== void 0
@@ -2718,8 +2775,7 @@ async function createSnapshotPackage({ files, packageInfo }) {
 	try {
 		await (0, node_fs_promises.mkdir)(dist);
 		await Promise.all([...sources].map(async ([path, { content }]) => {
-			await (0, node_fs_promises.mkdir)((0, node_path.dirname)((0, node_path.join)(root, path)), { recursive: true });
-			await (0, node_fs_promises.writeFile)((0, node_path.join)(root, path), content);
+			await writeRaw((0, node_path.join)(root, path), content);
 		}));
 		const entry = [...sources.keys()].filter((path) => /\.(?:[cm]?[jt]sx?)$/.test(path)).map((path) => (0, node_path.join)(root, path));
 		if (entry.length) {
@@ -2759,7 +2815,7 @@ async function createSnapshotPackage({ files, packageInfo }) {
 		const chunks = [...new Map([
 			["package.json", JSON.stringify(manifest, null, 2)],
 			...[...sources].map(([path, { content }]) => [path, content]),
-			...await Promise.all(built.map(async (key) => [`dist/${key}`, await (0, node_fs_promises.readFile)((0, node_path.join)(dist, key), "utf8")]))
+			...await Promise.all(built.map(async (key) => [`dist/${key}`, await read((0, node_path.join)(dist, key))]))
 		])].flatMap(([path, content]) => {
 			const bytes = Buffer.from(content);
 			return [
@@ -2790,7 +2846,10 @@ async function createSnapshot({ studioUrl = defaultStudioUrl, apiKey, config, ho
 		path: "/api/snapshots"
 	});
 	const generated = await generate({
-		config,
+		config: {
+			...config,
+			storage: (0, _kubb_core.fsStorage)()
+		},
 		hooks,
 		signal,
 		allowExec: true