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

package/dist/index.js

@@ -3,15 +3,15 @@ import { EventEmitter } from "node:events";
 import { readFileSync } from "node:fs";
 import { access, mkdir, readFile, readdir, rm, writeFile } from "node:fs/promises";
 import path, { basename, dirname, extname, join, posix, relative, resolve } from "node:path";
-import { composeTransformers, composeTransformers as composeTransformers$1, definePrinter, isOperationNode, isSchemaNode } from "@kubb/ast";
+import { composeTransformers, composeTransformers as composeTransformers$1, definePrinter, isOperationNode, isSchemaNode, transform, walk } from "@kubb/ast";
 import { Fabric, createFabric, createReactFabric } from "@kubb/react-fabric";
 import { typescriptParser } from "@kubb/react-fabric/parsers";
 import { fsPlugin } from "@kubb/react-fabric/plugins";
 import { performance } from "node:perf_hooks";
 import { deflateSync } from "fflate";
 import { x } from "tinyexec";
-import { version } from "node:process";
 import { jsx } from "@kubb/react-fabric/jsx-runtime";
+import { version } from "node:process";
 import { sortBy } from "remeda";
 import * as pkg from "empathic/package";
 import { coerce, satisfies } from "semver";
@@ -895,6 +895,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) => {
@@ -911,6 +912,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);
@@ -925,6 +927,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);
@@ -961,7 +964,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;
@@ -977,9 +986,16 @@ var PluginDriver = class {
 		const baseContext = {
 			fabric: driver.options.fabric,
 			config: driver.config,
+			get root() {
+				return resolve(driver.config.root, driver.config.output.path);
+			},
+			getMode(output) {
+				return getMode(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);
@@ -999,6 +1015,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");
@@ -1009,8 +1034,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 {
@@ -1228,6 +1253,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`.
@@ -1316,6 +1346,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 = createReactFabric();
+	await fabricChild.render(/* @__PURE__ */ jsx(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.
@@ -1413,7 +1463,7 @@ const fsStorage = createStorage(() => ({
 }));
 //#endregion
 //#region package.json
-var version$1 = "5.0.0-alpha.29";
+var version$1 = "5.0.0-alpha.30";
 //#endregion
 //#region src/utils/diagnostics.ts
 /**
@@ -1432,103 +1482,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(resolve(definedConfig.root, definedConfig.output.path));
 	}
-	const fabric = createFabric();
-	fabric.use(fsPlugin);
-	fabric.use(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 = path.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 = relative(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: 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 = 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(resolve(definedConfig.root, definedConfig.output.path));
+	}
+	const fabric = createFabric();
+	fabric.use(fsPlugin);
+	fabric.use(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 = relative(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`]
 		});
@@ -1594,6 +1879,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 walk(rootNode, {
+		depth: "shallow",
+		async schema(node) {
+			if (!plugin.schema) return;
+			const transformedNode = plugin.transformer ? 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 ? 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`.
@@ -1612,15 +1942,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 = 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, {
@@ -1629,7 +1970,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;
@@ -1643,7 +1984,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:",
@@ -1697,6 +2038,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,
@@ -1785,10 +2130,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) { ... },
 *   }
@@ -1796,38 +2142,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
@@ -2201,172 +2573,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 = createReactFabric();
-	await fabricChild.render(/* @__PURE__ */ jsx(Fabric, { children: /* @__PURE__ */ 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 = createReactFabric();
-	await fabricChild.render(/* @__PURE__ */ jsx(Fabric, { children: /* @__PURE__ */ 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 = createReactFabric();
-	await fabricChild.render(/* @__PURE__ */ jsx(Fabric, { children: /* @__PURE__ */ 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
@@ -2530,241 +2736,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 = path.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: 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 = 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.
@@ -2904,6 +2875,6 @@ function satisfiesDependency(dependency, version, cwd) {
 	return satisfies(semVer, version);
 }
 //#endregion
-export { AsyncEventEmitter, FunctionParams, PluginDriver, URLPath, build, build as default, buildDefaultBanner, composeTransformers, createAdapter, createPlugin, createStorage, defaultResolveBanner, defaultResolveFile, defaultResolveFooter, defaultResolveOptions, defaultResolvePath, defineConfig, defineGenerator, defineLogger, definePresets, definePrinter, defineResolver, detectFormatter, detectLinter, formatters, fsStorage, getBarrelFiles, getConfigs, getMode, getPreset, isInputPath, linters, logLevel, memoryStorage, renderOperation, renderOperations, renderSchema, runGeneratorOperation, runGeneratorOperations, runGeneratorSchema, safeBuild, satisfiesDependency, setup };
+export { AsyncEventEmitter, FunctionParams, PluginDriver, URLPath, build, build as default, buildDefaultBanner, composeTransformers, createAdapter, createPlugin, createStorage, defaultResolveBanner, defaultResolveFile, defaultResolveFooter, defaultResolveOptions, defaultResolvePath, defineConfig, defineGenerator, defineLogger, definePresets, definePrinter, defineResolver, detectFormatter, detectLinter, formatters, fsStorage, getBarrelFiles, getConfigs, getMode, getPreset, isInputPath, linters, logLevel, memoryStorage, mergeGenerators, safeBuild, satisfiesDependency, setup };
 
 //# sourceMappingURL=index.js.map