@kubb/core 5.0.0-alpha.29 → 5.0.0-alpha.30

package/dist/index.cjs

@@ -15,8 +15,8 @@ let _kubb_react_fabric_plugins = require("@kubb/react-fabric/plugins");
 let node_perf_hooks = require("node:perf_hooks");
 let fflate = require("fflate");
 let tinyexec = require("tinyexec");
-let node_process = require("node:process");
 let _kubb_react_fabric_jsx_runtime = require("@kubb/react-fabric/jsx-runtime");
+let node_process = require("node:process");
 let remeda = require("remeda");
 let empathic_package = require("empathic/package");
 empathic_package = require_chunk.__toESM(empathic_package);
@@ -901,6 +901,7 @@ function validateConcurrency(concurrency) {
 * - Each function receives the accumulated state from the previous call.
 * - Skips functions that return a falsy value (acts as a no-op for that step).
 * - Returns an array of all individual results.
+*  @deprecated
 */
 function hookSeq(promises) {
 	return promises.filter(Boolean).reduce((promise, func) => {
@@ -917,6 +918,7 @@ function hookSeq(promises) {
 *
 * - Stops as soon as `nullCheck` passes for a result (default: `!== null`).
 * - Subsequent functions are skipped once a match is found.
+*  @deprecated
 */
 function hookFirst(promises, nullCheck = (state) => state !== null) {
 	let promise = Promise.resolve(null);
@@ -931,6 +933,7 @@ function hookFirst(promises, nullCheck = (state) => state !== null) {
 *
 * - Limits simultaneous executions to `concurrency` (default: unlimited).
 * - Uses `Promise.allSettled` so individual failures do not cancel other tasks.
+*  @deprecated
 */
 function hookParallel(promises, concurrency = Number.POSITIVE_INFINITY) {
 	const limit = pLimit(concurrency);
@@ -967,7 +970,13 @@ var PluginDriver = class {
 	constructor(config, options) {
 		this.config = config;
 		this.options = options;
-		config.plugins.map((plugin) => Object.assign({ install() {} }, plugin)).sort((a, b) => {
+		config.plugins.map((plugin) => Object.assign({
+			buildStart() {},
+			buildEnd() {}
+		}, plugin)).filter((plugin) => {
+			if (typeof plugin.apply === "function") return plugin.apply(config);
+			return true;
+		}).sort((a, b) => {
 			if (b.pre?.includes(a.name)) return 1;
 			if (b.post?.includes(a.name)) return -1;
 			return 0;
@@ -983,9 +992,16 @@ var PluginDriver = class {
 		const baseContext = {
 			fabric: driver.options.fabric,
 			config: driver.config,
+			get root() {
+				return (0, node_path.resolve)(driver.config.root, driver.config.output.path);
+			},
+			getMode(output) {
+				return getMode((0, node_path.resolve)(driver.config.root, driver.config.output.path, output.path));
+			},
+			events: driver.options.events,
 			plugin,
 			getPlugin: driver.getPlugin.bind(driver),
-			events: driver.options.events,
+			requirePlugin: driver.requirePlugin.bind(driver),
 			driver,
 			addFile: async (...files) => {
 				await this.options.fabric.addFile(...files);
@@ -1005,6 +1021,15 @@ var PluginDriver = class {
 			get transformer() {
 				return plugin.transformer;
 			},
+			warn(message) {
+				driver.events.emit("warn", message);
+			},
+			error(error) {
+				driver.events.emit("error", typeof error === "string" ? new Error(error) : error);
+			},
+			info(message) {
+				driver.events.emit("info", message);
+			},
 			openInStudio(options) {
 				if (!driver.config.devtools || driver.#studioIsOpen) return;
 				if (typeof driver.config.devtools !== "object") throw new Error("Devtools must be an object");
@@ -1015,8 +1040,8 @@ var PluginDriver = class {
 			}
 		};
 		const mergedExtras = {};
-		for (const plugin of this.plugins.values()) if (typeof plugin.inject === "function") {
-			const result = plugin.inject.call(baseContext, baseContext);
+		for (const p of this.plugins.values()) if (typeof p.inject === "function") {
+			const result = p.inject.call(baseContext);
 			if (result !== null && typeof result === "object") Object.assign(mergedExtras, result);
 		}
 		return {
@@ -1234,6 +1259,11 @@ var PluginDriver = class {
 	getPlugin(pluginName) {
 		return this.plugins.get(pluginName);
 	}
+	requirePlugin(pluginName) {
+		const plugin = this.plugins.get(pluginName);
+		if (!plugin) throw new Error(`[kubb] Plugin "${pluginName}" is required but not found. Make sure it is included in your Kubb config.`);
+		return plugin;
+	}
 	/**
 	* Run an async plugin hook and return the result.
 	* @param hookName Name of the plugin hook. Must be either in `PluginHooks` or `OutputPluginValueHooks`.
@@ -1322,6 +1352,26 @@ var PluginDriver = class {
 	}
 };
 //#endregion
+//#region src/renderNode.tsx
+/**
+* Handles the return value of a plugin AST hook or generator method.
+*
+* - React element → rendered via an isolated react-fabric context, files merged into `fabric`
+* - `Array<FabricFile.File>` → upserted directly into `fabric`
+* - `void` / `null` / `undefined` → no-op (plugin handled it via `this.upsertFile`)
+*/
+async function applyHookResult(result, fabric) {
+	if (!result) return;
+	if (Array.isArray(result)) {
+		await fabric.upsertFile(...result);
+		return;
+	}
+	const fabricChild = (0, _kubb_react_fabric.createReactFabric)();
+	await fabricChild.render(/* @__PURE__ */ (0, _kubb_react_fabric_jsx_runtime.jsx)(_kubb_react_fabric.Fabric, { children: result }));
+	fabric.context.fileManager.upsert(...fabricChild.files);
+	fabricChild.unmount();
+}
+//#endregion
 //#region src/createStorage.ts
 /**
 * Creates a storage factory. Call the returned function with optional options to get the storage instance.
@@ -1419,7 +1469,7 @@ const fsStorage = createStorage(() => ({
 }));
 //#endregion
 //#region package.json
-var version = "5.0.0-alpha.29";
+var version = "5.0.0-alpha.30";
 //#endregion
 //#region src/utils/diagnostics.ts
 /**
@@ -1438,103 +1488,338 @@ function getDiagnosticInfo() {
 	};
 }
 //#endregion
-//#region src/build.ts
+//#region src/utils/TreeNode.ts
 /**
-* Initializes all Kubb infrastructure for a build without executing any plugins.
-*
-* - Validates the input path (when applicable).
-* - Applies config defaults (`root`, `output.*`, `devtools`).
-* - Creates the Fabric instance and wires storage, format, and lint hooks.
-* - Runs the adapter (if configured) to produce the universal `RootNode`.
+* Tree structure used to build per-directory barrel (`index.ts`) files from a
+* flat list of generated {@link FabricFile.File} entries.
 *
-* Pass the returned {@link SetupResult} directly to {@link safeBuild} or {@link build}
-* via the `overrides` argument to reuse the same infrastructure across multiple runs.
+* Each node represents either a directory or a file within the output tree.
+* Use {@link TreeNode.build} to construct a root node from a file list, then
+* traverse with {@link TreeNode.forEach}, {@link TreeNode.leaves}, or the
+* `*Deep` helpers.
 */
-async function setup(options) {
-	const { config: userConfig, events = new AsyncEventEmitter() } = options;
-	const sources = /* @__PURE__ */ new Map();
-	const diagnosticInfo = getDiagnosticInfo();
-	if (Array.isArray(userConfig.input)) await events.emit("warn", "This feature is still under development — use with caution");
-	await events.emit("debug", {
-		date: /* @__PURE__ */ new Date(),
-		logs: [
-			"Configuration:",
-			`  • Name: ${userConfig.name || "unnamed"}`,
-			`  • Root: ${userConfig.root || process.cwd()}`,
-			`  • Output: ${userConfig.output?.path || "not specified"}`,
-			`  • Plugins: ${userConfig.plugins?.length || 0}`,
-			"Output Settings:",
-			`  • Storage: ${userConfig.output?.storage ? `custom(${userConfig.output.storage.name})` : userConfig.output?.write === false ? "disabled" : "filesystem (default)"}`,
-			`  • Formatter: ${userConfig.output?.format || "none"}`,
-			`  • Linter: ${userConfig.output?.lint || "none"}`,
-			"Environment:",
-			Object.entries(diagnosticInfo).map(([key, value]) => `  • ${key}: ${value}`).join("\n")
-		]
-	});
-	try {
-		if (isInputPath(userConfig) && !new URLPath(userConfig.input.path).isURL) {
-			await exists(userConfig.input.path);
-			await events.emit("debug", {
-				date: /* @__PURE__ */ new Date(),
-				logs: [`✓ Input file validated: ${userConfig.input.path}`]
+var TreeNode = class TreeNode {
+	data;
+	parent;
+	children = [];
+	#cachedLeaves = void 0;
+	constructor(data, parent) {
+		this.data = data;
+		this.parent = parent;
+	}
+	addChild(data) {
+		const child = new TreeNode(data, this);
+		if (!this.children) this.children = [];
+		this.children.push(child);
+		return child;
+	}
+	/**
+	* Returns the root ancestor of this node, walking up via `parent` links.
+	*/
+	get root() {
+		if (!this.parent) return this;
+		return this.parent.root;
+	}
+	/**
+	* Returns all leaf descendants (nodes with no children) of this node.
+	*
+	* Results are cached after the first traversal.
+	*/
+	get leaves() {
+		if (!this.children || this.children.length === 0) return [this];
+		if (this.#cachedLeaves) return this.#cachedLeaves;
+		const leaves = [];
+		for (const child of this.children) leaves.push(...child.leaves);
+		this.#cachedLeaves = leaves;
+		return leaves;
+	}
+	/**
+	* Visits this node and every descendant in depth-first order.
+	*/
+	forEach(callback) {
+		if (typeof callback !== "function") throw new TypeError("forEach() callback must be a function");
+		callback(this);
+		for (const child of this.children) child.forEach(callback);
+		return this;
+	}
+	/**
+	* Finds the first leaf that satisfies `predicate`, or `undefined` when none match.
+	*/
+	findDeep(predicate) {
+		if (typeof predicate !== "function") throw new TypeError("find() predicate must be a function");
+		return this.leaves.find(predicate);
+	}
+	/**
+	* Calls `callback` for every leaf of this node.
+	*/
+	forEachDeep(callback) {
+		if (typeof callback !== "function") throw new TypeError("forEach() callback must be a function");
+		this.leaves.forEach(callback);
+	}
+	/**
+	* Returns all leaves that satisfy `callback`.
+	*/
+	filterDeep(callback) {
+		if (typeof callback !== "function") throw new TypeError("filter() callback must be a function");
+		return this.leaves.filter(callback);
+	}
+	/**
+	* Maps every leaf through `callback` and returns the resulting array.
+	*/
+	mapDeep(callback) {
+		if (typeof callback !== "function") throw new TypeError("map() callback must be a function");
+		return this.leaves.map(callback);
+	}
+	/**
+	* Builds a {@link TreeNode} tree from a flat list of files.
+	*
+	* - Filters to files under `root` (when provided) and skips `.json` files.
+	* - Returns `null` when no files match.
+	*/
+	static build(files, root) {
+		try {
+			const filteredTree = buildDirectoryTree(files, root);
+			if (!filteredTree) return null;
+			const treeNode = new TreeNode({
+				name: filteredTree.name,
+				path: filteredTree.path,
+				file: filteredTree.file,
+				type: getMode(filteredTree.path)
 			});
+			const recurse = (node, item) => {
+				const subNode = node.addChild({
+					name: item.name,
+					path: item.path,
+					file: item.file,
+					type: getMode(item.path)
+				});
+				if (item.children?.length) item.children?.forEach((child) => {
+					recurse(subNode, child);
+				});
+			};
+			filteredTree.children?.forEach((child) => {
+				recurse(treeNode, child);
+			});
+			return treeNode;
+		} catch (error) {
+			throw new Error("Something went wrong with creating barrel files with the TreeNode class", { cause: error });
 		}
-	} catch (caughtError) {
-		if (isInputPath(userConfig)) {
-			const error = caughtError;
-			throw new Error(`Cannot read file/URL defined in \`input.path\` or set with \`kubb generate PATH\` in the CLI of your Kubb config ${userConfig.input.path}`, { cause: error });
-		}
-	}
-	const definedConfig = {
-		root: userConfig.root || process.cwd(),
-		...userConfig,
-		output: {
-			write: true,
-			barrelType: "named",
-			extension: DEFAULT_EXTENSION,
-			defaultBanner: DEFAULT_BANNER,
-			...userConfig.output
-		},
-		devtools: userConfig.devtools ? {
-			studioUrl: DEFAULT_STUDIO_URL,
-			...typeof userConfig.devtools === "boolean" ? {} : userConfig.devtools
-		} : void 0,
-		plugins: userConfig.plugins
-	};
-	const storage = definedConfig.output.write === false ? null : definedConfig.output.storage ?? fsStorage();
-	if (definedConfig.output.clean) {
-		await events.emit("debug", {
-			date: /* @__PURE__ */ new Date(),
-			logs: ["Cleaning output directories", `  • Output: ${definedConfig.output.path}`]
-		});
-		await storage?.clear((0, node_path.resolve)(definedConfig.root, definedConfig.output.path));
 	}
-	const fabric = (0, _kubb_react_fabric.createFabric)();
-	fabric.use(_kubb_react_fabric_plugins.fsPlugin);
-	fabric.use(_kubb_react_fabric_parsers.typescriptParser);
-	fabric.context.on("files:processing:start", (files) => {
-		events.emit("files:processing:start", files);
-		events.emit("debug", {
-			date: /* @__PURE__ */ new Date(),
-			logs: [`Writing ${files.length} files...`]
-		});
+};
+const normalizePath = (p) => p.replaceAll("\\", "/");
+function buildDirectoryTree(files, rootFolder = "") {
+	const normalizedRootFolder = normalizePath(rootFolder);
+	const rootPrefix = normalizedRootFolder.endsWith("/") ? normalizedRootFolder : `${normalizedRootFolder}/`;
+	const filteredFiles = files.filter((file) => {
+		const normalizedFilePath = normalizePath(file.path);
+		return rootFolder ? normalizedFilePath.startsWith(rootPrefix) && !normalizedFilePath.endsWith(".json") : !normalizedFilePath.endsWith(".json");
 	});
-	fabric.context.on("file:processing:update", async (params) => {
-		const { file, source } = params;
-		await events.emit("file:processing:update", {
-			...params,
-			config: definedConfig,
-			source
+	if (filteredFiles.length === 0) return null;
+	const root = {
+		name: rootFolder || "",
+		path: rootFolder || "",
+		children: []
+	};
+	filteredFiles.forEach((file) => {
+		const parts = file.path.slice(rootFolder.length).split("/").filter(Boolean);
+		let currentLevel = root.children;
+		let currentPath = normalizePath(rootFolder);
+		parts.forEach((part, index) => {
+			currentPath = node_path.default.posix.join(currentPath, part);
+			let existingNode = currentLevel.find((node) => node.name === part);
+			if (!existingNode) {
+				if (index === parts.length - 1) existingNode = {
+					name: part,
+					file,
+					path: currentPath
+				};
+				else existingNode = {
+					name: part,
+					path: currentPath,
+					children: []
+				};
+				currentLevel.push(existingNode);
+			}
+			if (!existingNode.file) currentLevel = existingNode.children;
 		});
-		if (source) {
-			const key = (0, node_path.relative)((0, node_path.resolve)(definedConfig.root), file.path);
-			await storage?.setItem(key, source);
-			sources.set(file.path, source);
-		}
 	});
-	fabric.context.on("files:processing:end", async (files) => {
-		await events.emit("files:processing:end", files);
-		await events.emit("debug", {
+	return root;
+}
+//#endregion
+//#region src/utils/getBarrelFiles.ts
+/** biome-ignore-all lint/suspicious/useIterableCallbackReturn: not needed */
+function getBarrelFilesByRoot(root, files) {
+	const cachedFiles = /* @__PURE__ */ new Map();
+	TreeNode.build(files, root)?.forEach((treeNode) => {
+		if (!treeNode?.children || !treeNode.parent?.data.path) return;
+		const barrelFile = {
+			path: (0, node_path.join)(treeNode.parent?.data.path, "index.ts"),
+			baseName: "index.ts",
+			exports: [],
+			imports: [],
+			sources: []
+		};
+		const previousBarrelFile = cachedFiles.get(barrelFile.path);
+		treeNode.leaves.forEach((item) => {
+			if (!item.data.name) return;
+			(item.data.file?.sources || []).forEach((source) => {
+				if (!item.data.file?.path || !source.isIndexable || !source.name) return;
+				if (previousBarrelFile?.sources.some((item) => item.name === source.name && item.isTypeOnly === source.isTypeOnly)) return;
+				barrelFile.exports.push({
+					name: [source.name],
+					path: getRelativePath(treeNode.parent?.data.path, item.data.path),
+					isTypeOnly: source.isTypeOnly
+				});
+				barrelFile.sources.push({
+					name: source.name,
+					isTypeOnly: source.isTypeOnly,
+					value: "",
+					isExportable: false,
+					isIndexable: false
+				});
+			});
+		});
+		if (previousBarrelFile) {
+			previousBarrelFile.sources.push(...barrelFile.sources);
+			previousBarrelFile.exports?.push(...barrelFile.exports || []);
+		} else cachedFiles.set(barrelFile.path, barrelFile);
+	});
+	return [...cachedFiles.values()];
+}
+function trimExtName(text) {
+	const dotIndex = text.lastIndexOf(".");
+	if (dotIndex > 0 && !text.includes("/", dotIndex)) return text.slice(0, dotIndex);
+	return text;
+}
+/**
+* Generates `index.ts` barrel files for all directories under `root/output.path`.
+*
+* - Returns an empty array when `type` is falsy or `'propagate'`.
+* - Skips generation when the output path itself ends with `index` (already a barrel).
+* - When `type` is `'all'`, strips named exports so every re-export becomes a wildcard (`export * from`).
+* - Attaches `meta` to each barrel file for downstream plugin identification.
+*/
+async function getBarrelFiles(files, { type, meta = {}, root, output }) {
+	if (!type || type === "propagate") return [];
+	const pathToBuildFrom = (0, node_path.join)(root, output.path);
+	if (trimExtName(pathToBuildFrom).endsWith("index")) return [];
+	const barrelFiles = getBarrelFilesByRoot(pathToBuildFrom, files);
+	if (type === "all") return barrelFiles.map((file) => {
+		return {
+			...file,
+			exports: file.exports?.map((exportItem) => {
+				return {
+					...exportItem,
+					name: void 0
+				};
+			})
+		};
+	});
+	return barrelFiles.map((indexFile) => {
+		return {
+			...indexFile,
+			meta
+		};
+	});
+}
+//#endregion
+//#region src/build.ts
+/**
+* Initializes all Kubb infrastructure for a build without executing any plugins.
+*
+* - Validates the input path (when applicable).
+* - Applies config defaults (`root`, `output.*`, `devtools`).
+* - Creates the Fabric instance and wires storage, format, and lint hooks.
+* - Runs the adapter (if configured) to produce the universal `RootNode`.
+*
+* Pass the returned {@link SetupResult} directly to {@link safeBuild} or {@link build}
+* via the `overrides` argument to reuse the same infrastructure across multiple runs.
+*/
+async function setup(options) {
+	const { config: userConfig, events = new AsyncEventEmitter() } = options;
+	const sources = /* @__PURE__ */ new Map();
+	const diagnosticInfo = getDiagnosticInfo();
+	if (Array.isArray(userConfig.input)) await events.emit("warn", "This feature is still under development — use with caution");
+	await events.emit("debug", {
+		date: /* @__PURE__ */ new Date(),
+		logs: [
+			"Configuration:",
+			`  • Name: ${userConfig.name || "unnamed"}`,
+			`  • Root: ${userConfig.root || process.cwd()}`,
+			`  • Output: ${userConfig.output?.path || "not specified"}`,
+			`  • Plugins: ${userConfig.plugins?.length || 0}`,
+			"Output Settings:",
+			`  • Storage: ${userConfig.output?.storage ? `custom(${userConfig.output.storage.name})` : userConfig.output?.write === false ? "disabled" : "filesystem (default)"}`,
+			`  • Formatter: ${userConfig.output?.format || "none"}`,
+			`  • Linter: ${userConfig.output?.lint || "none"}`,
+			"Environment:",
+			Object.entries(diagnosticInfo).map(([key, value]) => `  • ${key}: ${value}`).join("\n")
+		]
+	});
+	try {
+		if (isInputPath(userConfig) && !new URLPath(userConfig.input.path).isURL) {
+			await exists(userConfig.input.path);
+			await events.emit("debug", {
+				date: /* @__PURE__ */ new Date(),
+				logs: [`✓ Input file validated: ${userConfig.input.path}`]
+			});
+		}
+	} catch (caughtError) {
+		if (isInputPath(userConfig)) {
+			const error = caughtError;
+			throw new Error(`Cannot read file/URL defined in \`input.path\` or set with \`kubb generate PATH\` in the CLI of your Kubb config ${userConfig.input.path}`, { cause: error });
+		}
+	}
+	const definedConfig = {
+		root: userConfig.root || process.cwd(),
+		...userConfig,
+		output: {
+			write: true,
+			barrelType: "named",
+			extension: DEFAULT_EXTENSION,
+			defaultBanner: DEFAULT_BANNER,
+			...userConfig.output
+		},
+		devtools: userConfig.devtools ? {
+			studioUrl: DEFAULT_STUDIO_URL,
+			...typeof userConfig.devtools === "boolean" ? {} : userConfig.devtools
+		} : void 0,
+		plugins: userConfig.plugins
+	};
+	const storage = definedConfig.output.write === false ? null : definedConfig.output.storage ?? fsStorage();
+	if (definedConfig.output.clean) {
+		await events.emit("debug", {
+			date: /* @__PURE__ */ new Date(),
+			logs: ["Cleaning output directories", `  • Output: ${definedConfig.output.path}`]
+		});
+		await storage?.clear((0, node_path.resolve)(definedConfig.root, definedConfig.output.path));
+	}
+	const fabric = (0, _kubb_react_fabric.createFabric)();
+	fabric.use(_kubb_react_fabric_plugins.fsPlugin);
+	fabric.use(_kubb_react_fabric_parsers.typescriptParser);
+	fabric.context.on("files:processing:start", (files) => {
+		events.emit("files:processing:start", files);
+		events.emit("debug", {
+			date: /* @__PURE__ */ new Date(),
+			logs: [`Writing ${files.length} files...`]
+		});
+	});
+	fabric.context.on("file:processing:update", async (params) => {
+		const { file, source } = params;
+		await events.emit("file:processing:update", {
+			...params,
+			config: definedConfig,
+			source
+		});
+		if (source) {
+			const key = (0, node_path.relative)((0, node_path.resolve)(definedConfig.root), file.path);
+			await storage?.setItem(key, source);
+			sources.set(file.path, source);
+		}
+	});
+	fabric.context.on("files:processing:end", async (files) => {
+		await events.emit("files:processing:end", files);
+		await events.emit("debug", {
 			date: /* @__PURE__ */ new Date(),
 			logs: [`✓ File write process completed for ${files.length} files`]
 		});
@@ -1600,6 +1885,51 @@ async function build(options, overrides) {
 	};
 }
 /**
+* Walks the AST and dispatches nodes to a plugin's direct AST hooks
+* (`schema`, `operation`, `operations`).
+*
+* - Each hook accepts a single handler **or an array** — all entries are called in sequence.
+* - Nodes that are excluded by `exclude`/`include` plugin options are skipped automatically.
+* - Return values are handled via `applyHookResult`: React elements are rendered,
+*   `FabricFile.File[]` are written via upsert, and `void` is a no-op (manual handling).
+* - Barrel files are generated automatically when `output.barrelType` is set.
+*/
+async function runPluginAstHooks(plugin, context) {
+	const { adapter, rootNode, resolver, fabric } = context;
+	const { exclude, include, override } = plugin.options;
+	if (!adapter || !rootNode) throw new Error(`[${plugin.name}] No adapter found. Add an OAS adapter (e.g. pluginOas()) before this plugin in your Kubb config.`);
+	const collectedOperations = [];
+	await (0, _kubb_ast.walk)(rootNode, {
+		depth: "shallow",
+		async schema(node) {
+			if (!plugin.schema) return;
+			const transformedNode = plugin.transformer ? (0, _kubb_ast.transform)(node, plugin.transformer) : node;
+			const options = resolver.resolveOptions(transformedNode, {
+				options: plugin.options,
+				exclude,
+				include,
+				override
+			});
+			if (options === null) return;
+			await applyHookResult(await plugin.schema.call(context, transformedNode, options), fabric);
+		},
+		async operation(node) {
+			const transformedNode = plugin.transformer ? (0, _kubb_ast.transform)(node, plugin.transformer) : node;
+			const options = resolver.resolveOptions(transformedNode, {
+				options: plugin.options,
+				exclude,
+				include,
+				override
+			});
+			if (options !== null) {
+				collectedOperations.push(transformedNode);
+				if (plugin.operation) await applyHookResult(await plugin.operation.call(context, transformedNode, options), fabric);
+			}
+		}
+	});
+	if (plugin.operations && collectedOperations.length > 0) await applyHookResult(await plugin.operations.call(context, collectedOperations, plugin.options), fabric);
+}
+/**
 * Runs a full Kubb build and captures errors instead of throwing.
 *
 * - Installs each plugin in order, recording failures in `failedPlugins`.
@@ -1618,15 +1948,26 @@ async function safeBuild(options, overrides) {
 		for (const plugin of driver.plugins.values()) {
 			const context = driver.getContext(plugin);
 			const hrStart = process.hrtime();
-			const installer = plugin.install.bind(context);
+			const { output } = plugin.options ?? {};
+			const root = (0, node_path.resolve)(config.root, config.output.path);
 			try {
 				const timestamp = /* @__PURE__ */ new Date();
 				await events.emit("plugin:start", plugin);
 				await events.emit("debug", {
 					date: timestamp,
-					logs: ["Installing plugin...", `  • Plugin Name: ${plugin.name}`]
+					logs: ["Starting plugin...", `  • Plugin Name: ${plugin.name}`]
 				});
-				await installer(context);
+				await plugin.buildStart.call(context);
+				if (plugin.schema || plugin.operation || plugin.operations) await runPluginAstHooks(plugin, context);
+				if (output) {
+					const barrelFiles = await getBarrelFiles(fabric.files, {
+						type: output.barrelType ?? "named",
+						root,
+						output,
+						meta: { pluginName: plugin.name }
+					});
+					await context.upsertFile(...barrelFiles);
+				}
 				const duration = getElapsedMs(hrStart);
 				pluginTimings.set(plugin.name, duration);
 				await events.emit("plugin:end", plugin, {
@@ -1635,7 +1976,7 @@ async function safeBuild(options, overrides) {
 				});
 				await events.emit("debug", {
 					date: /* @__PURE__ */ new Date(),
-					logs: [`✓ Plugin installed successfully (${formatMs(duration)})`]
+					logs: [`✓ Plugin started successfully (${formatMs(duration)})`]
 				});
 			} catch (caughtError) {
 				const error = caughtError;
@@ -1649,7 +1990,7 @@ async function safeBuild(options, overrides) {
 				await events.emit("debug", {
 					date: errorTimestamp,
 					logs: [
-						"✗ Plugin installation failed",
+						"✗ Plugin start failed",
 						`  • Plugin Name: ${plugin.name}`,
 						`  • Error: ${error.constructor.name} - ${error.message}`,
 						"  • Stack Trace:",
@@ -1703,6 +2044,10 @@ async function safeBuild(options, overrides) {
 		}
 		const files = [...fabric.files];
 		await fabric.write({ extension: config.output.extension });
+		for (const plugin of driver.plugins.values()) if (plugin.buildEnd) {
+			const context = driver.getContext(plugin);
+			await plugin.buildEnd.call(context);
+		}
 		return {
 			failedPlugins,
 			fabric,
@@ -1791,10 +2136,11 @@ function createAdapter(build) {
 * Creates a plugin factory. Call the returned function with optional options to get the plugin instance.
 *
 * @example
+* ```ts
 * export const myPlugin = createPlugin<MyPlugin>((options) => {
 *   return {
 *     name: 'my-plugin',
-*     options,
+*     get options() { return options },
 *     resolvePath(baseName) { ... },
 *     resolveName(name, type) { ... },
 *   }
@@ -1802,38 +2148,64 @@ function createAdapter(build) {
 *
 * // instantiate
 * const plugin = myPlugin({ output: { path: 'src/gen' } })
+* ```
 */
 function createPlugin(build) {
 	return (options) => build(options ?? {});
 }
 //#endregion
 //#region src/defineGenerator.ts
+/**
+* Defines a generator. Returns the object as-is with correct `this` typings.
+* No type discrimination (`type: 'react' | 'core'`) needed — `applyHookResult`
+* handles React elements and `File[]` uniformly.
+*/
 function defineGenerator(generator) {
-	if (generator.type === "react") return {
-		version: "2",
-		Operations() {
-			return null;
-		},
-		Operation() {
-			return null;
-		},
-		Schema() {
-			return null;
-		},
-		...generator
-	};
+	return generator;
+}
+/**
+* Merges an array of generators into a single generator.
+*
+* The merged generator's `schema`, `operation`, and `operations` methods run
+* the corresponding method from each input generator in sequence, applying each
+* result via `applyHookResult`. This eliminates the need to write the loop
+* manually in each plugin.
+*
+* @param generators - Array of generators to merge into a single generator.
+*
+* @example
+* ```ts
+* const merged = mergeGenerators(generators)
+*
+* return {
+*   name: pluginName,
+*   schema: merged.schema,
+*   operation: merged.operation,
+*   operations: merged.operations,
+* }
+* ```
+*/
+function mergeGenerators(generators) {
 	return {
-		version: "2",
-		async operations() {
-			return [];
-		},
-		async operation() {
-			return [];
+		name: generators.length > 0 ? generators.map((g) => g.name).join("+") : "merged",
+		async schema(node, options) {
+			for (const gen of generators) {
+				if (!gen.schema) continue;
+				await applyHookResult(await gen.schema.call(this, node, options), this.fabric);
+			}
 		},
-		async schema() {
-			return [];
+		async operation(node, options) {
+			for (const gen of generators) {
+				if (!gen.operation) continue;
+				await applyHookResult(await gen.operation.call(this, node, options), this.fabric);
+			}
 		},
-		...generator
+		async operations(nodes, options) {
+			for (const gen of generators) {
+				if (!gen.operations) continue;
+				await applyHookResult(await gen.operations.call(this, nodes, options), this.fabric);
+			}
+		}
 	};
 }
 //#endregion
@@ -2207,172 +2579,6 @@ function defineResolver(build) {
 	};
 }
 //#endregion
-//#region src/renderNode.tsx
-/**
-* Renders a React component for a list of operation nodes (V2 generators).
-*/
-async function renderOperations(nodes, options) {
-	const { config, fabric, plugin, Component, driver, adapter } = options;
-	if (!Component) return;
-	const fabricChild = (0, _kubb_react_fabric.createReactFabric)();
-	await fabricChild.render(/* @__PURE__ */ (0, _kubb_react_fabric_jsx_runtime.jsx)(_kubb_react_fabric.Fabric, { children: /* @__PURE__ */ (0, _kubb_react_fabric_jsx_runtime.jsx)(Component, {
-		config,
-		plugin,
-		driver,
-		adapter,
-		nodes,
-		options: options.options,
-		resolver: options.resolver
-	}) }));
-	fabric.context.fileManager.upsert(...fabricChild.files);
-	fabricChild.unmount();
-}
-/**
-* Renders a React component for a single operation node (V2 generators).
-*/
-async function renderOperation(node, options) {
-	const { config, fabric, plugin, Component, adapter, driver } = options;
-	if (!Component) return;
-	const fabricChild = (0, _kubb_react_fabric.createReactFabric)();
-	await fabricChild.render(/* @__PURE__ */ (0, _kubb_react_fabric_jsx_runtime.jsx)(_kubb_react_fabric.Fabric, { children: /* @__PURE__ */ (0, _kubb_react_fabric_jsx_runtime.jsx)(Component, {
-		config,
-		plugin,
-		driver,
-		adapter,
-		node,
-		options: options.options,
-		resolver: options.resolver
-	}) }));
-	fabric.context.fileManager.upsert(...fabricChild.files);
-	fabricChild.unmount();
-}
-/**
-* Renders a React component for a single schema node (V2 generators).
-*/
-async function renderSchema(node, options) {
-	const { config, fabric, plugin, Component, adapter, driver } = options;
-	if (!Component) return;
-	const fabricChild = (0, _kubb_react_fabric.createReactFabric)();
-	await fabricChild.render(/* @__PURE__ */ (0, _kubb_react_fabric_jsx_runtime.jsx)(_kubb_react_fabric.Fabric, { children: /* @__PURE__ */ (0, _kubb_react_fabric_jsx_runtime.jsx)(Component, {
-		config,
-		plugin,
-		driver,
-		adapter,
-		node,
-		options: options.options,
-		resolver: options.resolver
-	}) }));
-	fabric.context.fileManager.upsert(...fabricChild.files);
-	fabricChild.unmount();
-}
-/**
-* Dispatches a single schema node to all generators (react + core).
-* Resolves options per generator and skips excluded nodes.
-*/
-async function runGeneratorSchema(node, ctx) {
-	const { generators, plugin, resolver, exclude, include, override, fabric, adapter, config, driver } = ctx;
-	for (const generator of generators) {
-		const options = resolver.resolveOptions(node, {
-			options: plugin.options,
-			exclude,
-			include,
-			override
-		});
-		if (options === null) continue;
-		if (generator.type === "react" && generator.version === "2") await renderSchema(node, {
-			options,
-			resolver,
-			adapter,
-			config,
-			fabric,
-			Component: generator.Schema,
-			plugin,
-			driver
-		});
-		if (generator.type === "core" && generator.version === "2") {
-			const files = await generator.schema?.({
-				node,
-				options,
-				resolver,
-				adapter,
-				config,
-				plugin,
-				driver
-			}) ?? [];
-			await fabric.upsertFile(...files);
-		}
-	}
-}
-/**
-* Dispatches a single operation node to all generators (react + core).
-* Resolves options per generator and skips excluded nodes.
-*/
-async function runGeneratorOperation(node, ctx) {
-	const { generators, plugin, resolver, exclude, include, override, fabric, adapter, config, driver } = ctx;
-	for (const generator of generators) {
-		const options = resolver.resolveOptions(node, {
-			options: plugin.options,
-			exclude,
-			include,
-			override
-		});
-		if (options === null) continue;
-		if (generator.type === "react" && generator.version === "2") await renderOperation(node, {
-			options,
-			resolver,
-			adapter,
-			config,
-			fabric,
-			Component: generator.Operation,
-			plugin,
-			driver
-		});
-		if (generator.type === "core" && generator.version === "2") {
-			const files = await generator.operation?.({
-				node,
-				options,
-				resolver,
-				adapter,
-				config,
-				plugin,
-				driver
-			}) ?? [];
-			await fabric.upsertFile(...files);
-		}
-	}
-}
-/**
-* Batch-dispatches all collected operation nodes to every generator (react + core).
-* Uses `plugin.options` directly — no per-node option resolution.
-*/
-async function runGeneratorOperations(nodes, ctx) {
-	const { generators, plugin, resolver, fabric, adapter, config, driver } = ctx;
-	for (const generator of generators) {
-		if (generator.type === "react" && generator.version === "2") await renderOperations(nodes, {
-			options: plugin.options,
-			resolver,
-			adapter,
-			config,
-			fabric,
-			Component: generator.Operations,
-			plugin,
-			driver
-		});
-		if (generator.type === "core" && generator.version === "2") {
-			const files = await generator.operations?.({
-				nodes,
-				options: plugin.options,
-				resolver,
-				adapter,
-				config,
-				plugin,
-				driver
-			}) ?? [];
-			await fabric.upsertFile(...files);
-		}
-	}
-}
-//#endregion
 //#region src/storages/memoryStorage.ts
 /**
 * In-memory storage driver. Useful for testing and dry-run scenarios where
@@ -2536,241 +2742,6 @@ async function detectFormatter() {
 	return null;
 }
 //#endregion
-//#region src/utils/TreeNode.ts
-/**
-* Tree structure used to build per-directory barrel (`index.ts`) files from a
-* flat list of generated {@link FabricFile.File} entries.
-*
-* Each node represents either a directory or a file within the output tree.
-* Use {@link TreeNode.build} to construct a root node from a file list, then
-* traverse with {@link TreeNode.forEach}, {@link TreeNode.leaves}, or the
-* `*Deep` helpers.
-*/
-var TreeNode = class TreeNode {
-	data;
-	parent;
-	children = [];
-	#cachedLeaves = void 0;
-	constructor(data, parent) {
-		this.data = data;
-		this.parent = parent;
-	}
-	addChild(data) {
-		const child = new TreeNode(data, this);
-		if (!this.children) this.children = [];
-		this.children.push(child);
-		return child;
-	}
-	/**
-	* Returns the root ancestor of this node, walking up via `parent` links.
-	*/
-	get root() {
-		if (!this.parent) return this;
-		return this.parent.root;
-	}
-	/**
-	* Returns all leaf descendants (nodes with no children) of this node.
-	*
-	* Results are cached after the first traversal.
-	*/
-	get leaves() {
-		if (!this.children || this.children.length === 0) return [this];
-		if (this.#cachedLeaves) return this.#cachedLeaves;
-		const leaves = [];
-		for (const child of this.children) leaves.push(...child.leaves);
-		this.#cachedLeaves = leaves;
-		return leaves;
-	}
-	/**
-	* Visits this node and every descendant in depth-first order.
-	*/
-	forEach(callback) {
-		if (typeof callback !== "function") throw new TypeError("forEach() callback must be a function");
-		callback(this);
-		for (const child of this.children) child.forEach(callback);
-		return this;
-	}
-	/**
-	* Finds the first leaf that satisfies `predicate`, or `undefined` when none match.
-	*/
-	findDeep(predicate) {
-		if (typeof predicate !== "function") throw new TypeError("find() predicate must be a function");
-		return this.leaves.find(predicate);
-	}
-	/**
-	* Calls `callback` for every leaf of this node.
-	*/
-	forEachDeep(callback) {
-		if (typeof callback !== "function") throw new TypeError("forEach() callback must be a function");
-		this.leaves.forEach(callback);
-	}
-	/**
-	* Returns all leaves that satisfy `callback`.
-	*/
-	filterDeep(callback) {
-		if (typeof callback !== "function") throw new TypeError("filter() callback must be a function");
-		return this.leaves.filter(callback);
-	}
-	/**
-	* Maps every leaf through `callback` and returns the resulting array.
-	*/
-	mapDeep(callback) {
-		if (typeof callback !== "function") throw new TypeError("map() callback must be a function");
-		return this.leaves.map(callback);
-	}
-	/**
-	* Builds a {@link TreeNode} tree from a flat list of files.
-	*
-	* - Filters to files under `root` (when provided) and skips `.json` files.
-	* - Returns `null` when no files match.
-	*/
-	static build(files, root) {
-		try {
-			const filteredTree = buildDirectoryTree(files, root);
-			if (!filteredTree) return null;
-			const treeNode = new TreeNode({
-				name: filteredTree.name,
-				path: filteredTree.path,
-				file: filteredTree.file,
-				type: getMode(filteredTree.path)
-			});
-			const recurse = (node, item) => {
-				const subNode = node.addChild({
-					name: item.name,
-					path: item.path,
-					file: item.file,
-					type: getMode(item.path)
-				});
-				if (item.children?.length) item.children?.forEach((child) => {
-					recurse(subNode, child);
-				});
-			};
-			filteredTree.children?.forEach((child) => {
-				recurse(treeNode, child);
-			});
-			return treeNode;
-		} catch (error) {
-			throw new Error("Something went wrong with creating barrel files with the TreeNode class", { cause: error });
-		}
-	}
-};
-const normalizePath = (p) => p.replaceAll("\\", "/");
-function buildDirectoryTree(files, rootFolder = "") {
-	const normalizedRootFolder = normalizePath(rootFolder);
-	const rootPrefix = normalizedRootFolder.endsWith("/") ? normalizedRootFolder : `${normalizedRootFolder}/`;
-	const filteredFiles = files.filter((file) => {
-		const normalizedFilePath = normalizePath(file.path);
-		return rootFolder ? normalizedFilePath.startsWith(rootPrefix) && !normalizedFilePath.endsWith(".json") : !normalizedFilePath.endsWith(".json");
-	});
-	if (filteredFiles.length === 0) return null;
-	const root = {
-		name: rootFolder || "",
-		path: rootFolder || "",
-		children: []
-	};
-	filteredFiles.forEach((file) => {
-		const parts = file.path.slice(rootFolder.length).split("/").filter(Boolean);
-		let currentLevel = root.children;
-		let currentPath = normalizePath(rootFolder);
-		parts.forEach((part, index) => {
-			currentPath = node_path.default.posix.join(currentPath, part);
-			let existingNode = currentLevel.find((node) => node.name === part);
-			if (!existingNode) {
-				if (index === parts.length - 1) existingNode = {
-					name: part,
-					file,
-					path: currentPath
-				};
-				else existingNode = {
-					name: part,
-					path: currentPath,
-					children: []
-				};
-				currentLevel.push(existingNode);
-			}
-			if (!existingNode.file) currentLevel = existingNode.children;
-		});
-	});
-	return root;
-}
-//#endregion
-//#region src/utils/getBarrelFiles.ts
-/** biome-ignore-all lint/suspicious/useIterableCallbackReturn: not needed */
-function getBarrelFilesByRoot(root, files) {
-	const cachedFiles = /* @__PURE__ */ new Map();
-	TreeNode.build(files, root)?.forEach((treeNode) => {
-		if (!treeNode?.children || !treeNode.parent?.data.path) return;
-		const barrelFile = {
-			path: (0, node_path.join)(treeNode.parent?.data.path, "index.ts"),
-			baseName: "index.ts",
-			exports: [],
-			imports: [],
-			sources: []
-		};
-		const previousBarrelFile = cachedFiles.get(barrelFile.path);
-		treeNode.leaves.forEach((item) => {
-			if (!item.data.name) return;
-			(item.data.file?.sources || []).forEach((source) => {
-				if (!item.data.file?.path || !source.isIndexable || !source.name) return;
-				if (previousBarrelFile?.sources.some((item) => item.name === source.name && item.isTypeOnly === source.isTypeOnly)) return;
-				barrelFile.exports.push({
-					name: [source.name],
-					path: getRelativePath(treeNode.parent?.data.path, item.data.path),
-					isTypeOnly: source.isTypeOnly
-				});
-				barrelFile.sources.push({
-					name: source.name,
-					isTypeOnly: source.isTypeOnly,
-					value: "",
-					isExportable: false,
-					isIndexable: false
-				});
-			});
-		});
-		if (previousBarrelFile) {
-			previousBarrelFile.sources.push(...barrelFile.sources);
-			previousBarrelFile.exports?.push(...barrelFile.exports || []);
-		} else cachedFiles.set(barrelFile.path, barrelFile);
-	});
-	return [...cachedFiles.values()];
-}
-function trimExtName(text) {
-	const dotIndex = text.lastIndexOf(".");
-	if (dotIndex > 0 && !text.includes("/", dotIndex)) return text.slice(0, dotIndex);
-	return text;
-}
-/**
-* Generates `index.ts` barrel files for all directories under `root/output.path`.
-*
-* - Returns an empty array when `type` is falsy or `'propagate'`.
-* - Skips generation when the output path itself ends with `index` (already a barrel).
-* - When `type` is `'all'`, strips named exports so every re-export becomes a wildcard (`export * from`).
-* - Attaches `meta` to each barrel file for downstream plugin identification.
-*/
-async function getBarrelFiles(files, { type, meta = {}, root, output }) {
-	if (!type || type === "propagate") return [];
-	const pathToBuildFrom = (0, node_path.join)(root, output.path);
-	if (trimExtName(pathToBuildFrom).endsWith("index")) return [];
-	const barrelFiles = getBarrelFilesByRoot(pathToBuildFrom, files);
-	if (type === "all") return barrelFiles.map((file) => {
-		return {
-			...file,
-			exports: file.exports?.map((exportItem) => {
-				return {
-					...exportItem,
-					name: void 0
-				};
-			})
-		};
-	});
-	return barrelFiles.map((indexFile) => {
-		return {
-			...indexFile,
-			meta
-		};
-	});
-}
-//#endregion
 //#region src/utils/getConfigs.ts
 /**
 * Resolves a {@link ConfigInput} into a normalized array of {@link Config} objects.
@@ -2954,12 +2925,7 @@ exports.isInputPath = isInputPath;
 exports.linters = linters;
 exports.logLevel = logLevel;
 exports.memoryStorage = memoryStorage;
-exports.renderOperation = renderOperation;
-exports.renderOperations = renderOperations;
-exports.renderSchema = renderSchema;
-exports.runGeneratorOperation = runGeneratorOperation;
-exports.runGeneratorOperations = runGeneratorOperations;
-exports.runGeneratorSchema = runGeneratorSchema;
+exports.mergeGenerators = mergeGenerators;
 exports.safeBuild = safeBuild;
 exports.satisfiesDependency = satisfiesDependency;
 exports.setup = setup;