@kubb/studio 5.3.11 → 5.3.12

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import { t as __name } from "./rolldown-runtime-CRm0XQPb.js";
-import { AgentApi, AgentPermissions, ClientInfo, ConfigEdit, ConnectMessagePayload, GenerateInput, GenerateResult, GenerationEvent, GenerationEventPayloads, GenerationEventType, GenerationRun, PublishSnapshotInput, PublishSnapshotResult, RpcConnection, RpcConnector, StudioApi, generationEventTypes } from "./protocol.js";
+import { AgentApi, AgentPermissions, ClientInfo, ConfigEdit, ConnectMessagePayload, FileChange, GenerateInput, GenerateResult, GenerationEvent, GenerationEventPayloads, GenerationEventType, GenerationRun, PublishSnapshotInput, PublishSnapshotResult, RpcConnection, RpcConnector, StudioApi, generationEventTypes } from "./protocol.js";
 import { Config, Hookable, KubbHooks } from "@kubb/core";
 import { Storage } from "unstorage";
 //#region src/api.d.ts
@@ -554,5 +554,5 @@ export declare function pollForPairingToken({ studioUrl, session, signal }: Poll
  */
 export declare const connectWebSocketRpc: RpcConnector;
 //#endregion
-export { type AgentApi, type Client, type ClientOptions, type ConfigEdit, type ConnectMessagePayload, type ConnectionOptions, type GenerateInput, type GenerateResult, type GenerationEvent, type GenerationEventPayloads, type GenerationEventType, type GenerationRun, type PublishSnapshotInput, type PublishSnapshotResult, type RpcConnection, type RpcConnector, type StudioAgent, type StudioApi, type StudioConnectedContext, type StudioJob, type StudioJobStatus, type StudioSnapshot, generationEventTypes };
+export { type AgentApi, type Client, type ClientOptions, type ConfigEdit, type ConnectMessagePayload, type ConnectionOptions, type FileChange, type GenerateInput, type GenerateResult, type GenerationEvent, type GenerationEventPayloads, type GenerationEventType, type GenerationRun, type PublishSnapshotInput, type PublishSnapshotResult, type RpcConnection, type RpcConnector, type StudioAgent, type StudioApi, type StudioConnectedContext, type StudioJob, type StudioJobStatus, type StudioSnapshot, generationEventTypes };
 //# sourceMappingURL=index.d.ts.map

package/dist/index.js

@@ -676,7 +676,7 @@ async function createAgent({ studioUrl, token, name, machineToken }) {
 }
 //#endregion
 //#region package.json
-var version = "5.3.11";
+var version = "5.3.12";
 //#endregion
 //#region src/hooks.ts
 /**
@@ -1906,7 +1906,7 @@ function createGenerationStream(hooks, jobId, options = {}) {
 		const { peerDependencies, missingDependencies } = await resolvePeerDependencies(config.plugins.map(({ name }) => name));
 		const keys = await storage.readKeys();
 		const paths = new Set(keys.map((key) => relativeStoragePath(config.root, key)));
-		options.onGenerationEnd?.({
+		if ((status ?? "success") === "success") options.onGenerationEnd?.({
 			storage,
 			root: config.root,
 			paths,
@@ -2063,6 +2063,36 @@ const connectWebSocketRpc = async ({ url, token, local }) => {
 * How many files are read from storage at once when serving `readFiles` or packing a snapshot.
 */
 const FILE_READ_CONCURRENCY = 50;
+/**
+* Reads every file a run produced back out of its storage.
+*/
+async function readSnapshot(generation) {
+	const snapshot = /* @__PURE__ */ new Map();
+	await inParallel({
+		items: [...generation.paths],
+		limit: FILE_READ_CONCURRENCY,
+		run: async (path) => {
+			const content = await generation.storage.readItem(absoluteStoragePath(generation.root, path));
+			if (content !== null) snapshot.set(path, content);
+		}
+	});
+	return snapshot;
+}
+/**
+* How each path differs between two runs. Paths with identical content are left out.
+*/
+function diffSnapshots(previous, current) {
+	const changes = {};
+	for (const [path, content] of current) {
+		if (!previous.has(path)) {
+			changes[path] = "added";
+			continue;
+		}
+		if (previous.get(path) !== content) changes[path] = "changed";
+	}
+	for (const path of previous.keys()) if (!current.has(path)) changes[path] = "removed";
+	return changes;
+}
 var GenerationRunTarget = class extends RpcTarget {
 	generationStream;
 	generationResult;
@@ -2161,6 +2191,7 @@ var StudioSession = class {
 	#isGenerating = false;
 	#heartbeatTimer;
 	#lastGeneration;
+	#previousGeneration;
 	/**
 	* Resolves when Studio calls {@link StudioSession.connect}. `studio:ready` waits on this so the
 	* host does not queue jobs before the agent session is registered.
@@ -2385,6 +2416,7 @@ var StudioSession = class {
 				await this.#warn(`Ignored the spec from Studio; set ${remedy} to generate from it`);
 			}
 			const resolvedPlugins = plugins ?? config.plugins;
+			if (this.#lastGeneration) this.#previousGeneration = await readSnapshot(this.#lastGeneration);
 			this.#lastGeneration = void 0;
 			const detach = [setupHookListener(this.#hooks, root, controller.signal)];
 			try {
@@ -2413,12 +2445,20 @@ var StudioSession = class {
 				command,
 				info: `${resolvedPlugins.length} plugin${resolvedPlugins.length === 1 ? "" : "s"}, ${this.#canWrite ? "written to disk" : "in memory"}${inputOverride !== void 0 ? ", from a Studio spec" : ""}`
 			});
-			const files = [...this.#lastGeneration?.paths ?? []];
-			return {
+			const generation = this.#lastGeneration;
+			const files = [...generation?.paths ?? []];
+			const previous = this.#previousGeneration;
+			if (!generation || !previous) return {
 				status: "success",
 				files,
 				fileCount: files.length
 			};
+			return {
+				status: "success",
+				files,
+				fileCount: files.length,
+				changes: diffSnapshots(previous, await readSnapshot(generation))
+			};
 		} finally {
 			this.#isGenerating = false;
 		}
@@ -2535,6 +2575,16 @@ var StudioSession = class {
 		if (!Array.isArray(data.paths)) return this.#refuse("Ignored files: the message carried no paths", "The request carried no paths");
 		const { paths } = data;
 		if (paths.length > 50) return this.#refuse(`Ignored files: requested ${paths.length} paths, more than the 50 allowed per request`, `At most 50 paths may be requested at once`);
+		if (data.revision === "previous") {
+			const previous = this.#previousGeneration;
+			if (!previous) return this.#refuse("Ignored files: no previous generation to read from", "No previous generation to compare against");
+			const files = Object.fromEntries(paths.filter((path) => previous.has(path)).map((path) => [path, previous.get(path)]));
+			await this.#hooks.callHook("studio:command:end", {
+				command,
+				info: `read ${Object.keys(files).length}/${paths.length} requested file${paths.length === 1 ? "" : "s"} from the previous run`
+			});
+			return { files };
+		}
 		const generation = this.#lastGeneration;
 		if (!generation) return this.#refuse("Ignored files: no prior generation to read from", "No prior generation to read from, run a generation first");
 		const requested = paths.filter((path) => generation.paths.has(path));