@kubb/plugin-faker 5.1.0-canary.20261008T204013 → 5.1.0

package/dist/index.cjs

@@ -574,14 +574,14 @@ const SCALAR_TYPES = /* @__PURE__ */ new Set([
 	"enum"
 ]);
 const declarationPrinter = (0, _kubb_plugin_ts.functionPrinter)({ mode: "declaration" });
-function Faker({ node, description, name, typeName, printer, canOverride }) {
+function Faker({ node, description, name, typeName, printer, canOverride, typeMode = "inferred" }) {
 	const fakerText = printer.print(node) ?? "undefined";
 	const isArray = node.type === "array";
 	const isObject = OBJECT_TYPES.has(node.type);
 	const isTuple = node.type === "tuple";
 	const isScalar = SCALAR_TYPES.has(node.type);
 	const isUnion = node.type === "union";
-	const useGenericOverride = canOverride && isObject;
+	const useObjectOverride = canOverride && isObject;
 	const fakerTextWithOverride = (() => {
 		if (canOverride && node.type === "tuple") return `data && data.length === ${node.items?.length ?? 0} && !data.includes(undefined) ? data : ${fakerText}`;
 		if (canOverride && isArray) return `[\n  ...${fakerText},\n  ...(data || []).filter((item) => item !== undefined),\n]`;
@@ -589,7 +589,7 @@ function Faker({ node, description, name, typeName, printer, canOverride }) {
 		return fakerText;
 	})();
 	const { dataType, returnType: resolvedReturnType } = resolveFakerTypeUsage(node, typeName, canOverride);
-	if (!useGenericOverride) {
+	if (!useObjectOverride) {
 		const dataParamName = /\bdata\b/.test(fakerTextWithOverride) ? "data" : "_data";
 		const params = (0, _kubb_plugin_ts.createFunctionParameters)({ params: [(0, _kubb_plugin_ts.createFunctionParameter)({
 			name: dataParamName,
@@ -613,7 +613,9 @@ function Faker({ node, description, name, typeName, printer, canOverride }) {
 			})
 		});
 	}
-	const functionSignature = `${description ? `/**\n   * @description ${jsStringEscape(description)}\n   */\n  ` : ""}export function ${name}<TData extends Partial<${typeName}> = object>(data?: TData)`;
+	const jsdoc = description ? `/**\n   * @description ${jsStringEscape(description)}\n   */\n  ` : "";
+	const functionSignature = typeMode === "schema" ? `${jsdoc}export function ${name}(data?: Partial<${typeName}>): ${typeName}` : `${jsdoc}export function ${name}<TData extends Partial<${typeName}> = object>(data?: TData)`;
+	const returnType = typeMode === "schema" ? typeName : "Omit<typeof defaultFakeData, keyof NoInfer<TData>> & NoInfer<TData>";
 	const { cyclicSchemas, schemaName } = printer.options;
 	const functionBody = node.type === "object" && !!cyclicSchemas && (node.properties ?? []).some((p) => (0, kubb_kit.containsCircularRef)(p.schema, {
 		circularSchemas: cyclicSchemas,
@@ -625,13 +627,13 @@ function Faker({ node, description, name, typeName, printer, canOverride }) {
       Object.defineProperty(defaultFakeData, key, { value, configurable: true, writable: true, enumerable: true })
     }
   }
-  return defaultFakeData as Omit<typeof defaultFakeData, keyof NoInfer<TData>> & NoInfer<TData>
+  return defaultFakeData as ${returnType}
 }` : `{
   const defaultFakeData = ${fakerText}
   return {
     ...defaultFakeData,
     ...(data || {}),
-  } as Omit<typeof defaultFakeData, keyof NoInfer<TData>> & NoInfer<TData>
+  } as ${returnType}
 }`;
 	return /* @__PURE__ */ (0, kubb_jsx_jsx_runtime.jsxs)(kubb_jsx.File.Source, {
 		name,
@@ -1150,7 +1152,7 @@ const fakerGenerator = (0, kubb_kit.defineGenerator)({
 	renderer: kubb_jsx.jsxRenderer,
 	schema(node, ctx) {
 		const { config, resolver, root } = ctx;
-		const { output, group, dateParser, regexGenerator, seed, locale, printer } = ctx.options;
+		const { output, group, dateParser, regexGenerator, seed, locale, printer, typeMode } = ctx.options;
 		const pluginTs = ctx.driver.getPlugin(_kubb_plugin_ts.pluginTsName);
 		if (!node.name || !pluginTs) return;
 		const tsResolver = ctx.driver.getResolver(_kubb_plugin_ts.pluginTsName);
@@ -1264,14 +1266,15 @@ const fakerGenerator = (0, kubb_kit.defineGenerator)({
 					description: node.description,
 					node,
 					printer: printerInstance,
-					canOverride
+					canOverride,
+					typeMode
 				})
 			]
 		});
 	},
 	operation(node, ctx) {
 		const { config, resolver, root } = ctx;
-		const { output, group, dateParser, regexGenerator, seed, locale, printer } = ctx.options;
+		const { output, group, dateParser, regexGenerator, seed, locale, printer, typeMode } = ctx.options;
 		const pluginTs = ctx.driver.getPlugin(_kubb_plugin_ts.pluginTsName);
 		if (!pluginTs) return;
 		const tsResolver = ctx.driver.getResolver(_kubb_plugin_ts.pluginTsName);
@@ -1422,7 +1425,8 @@ const fakerGenerator = (0, kubb_kit.defineGenerator)({
 						...printerInstance,
 						print: () => rewrittenFakerText
 					},
-					canOverride
+					canOverride,
+					typeMode
 				})
 			] });
 		}
@@ -1549,7 +1553,7 @@ const pluginFaker = (0, kubb_kit.definePlugin)((options) => {
 	const { output = {
 		path: "mocks",
 		barrel: { type: "named" }
-	}, seed, locale = "en", group, exclude = [], include, override = [], dateParser = "faker", regexGenerator = "faker", printer, resolver: userResolver, macros: userMacros } = options;
+	}, seed, locale = "en", group, exclude = [], include, override = [], typeMode = "inferred", dateParser = "faker", regexGenerator = "faker", printer, resolver: userResolver, macros: userMacros } = options;
 	const groupConfig = createGroupConfig(group);
 	return {
 		name: pluginFakerName,
@@ -1563,6 +1567,7 @@ const pluginFaker = (0, kubb_kit.definePlugin)((options) => {
 				exclude,
 				include,
 				override,
+				typeMode,
 				group: groupConfig,
 				dateParser,
 				regexGenerator,