@kubb/plugin-react-query 5.1.0-canary.20260907T104725 → 5.1.0

package/dist/index.js

@@ -416,6 +416,9 @@ function resolveContractClient(options) {
 * so several dependents reading the same client plugin for one operation in a single pass
 * (react-query's query/mutation/infinite generators, vue-query, swr, the MCP handler, ...) share
 * one computed result instead of each re-deriving the name and path.
+*
+* `returnType` mirrors the client plugin's own `returnType` option (`'full'` when unset), so a
+* dependent's generated call body can match it instead of assuming the full `RequestResult` shape.
 */
 function resolveClientOperation(options) {
 	const { clientPlugin, driver, node, root, output, cache } = options;
@@ -432,7 +435,8 @@ function resolveClientOperation(options) {
 		return {
 			name: resolver.name(node.operationId),
 			path: file.path,
-			clientPath: path.resolve(root, ".kubb/client.ts")
+			clientPath: path.resolve(root, ".kubb/client.ts"),
+			returnType: plugin?.options?.returnType ?? "full"
 		};
 	});
 }
@@ -542,6 +546,24 @@ function buildClientCall(node, options) {
 	].filter((part) => part !== null).join(", ")} })`;
 }
 /**
+* Builds the query/mutation body that resolves a `buildClientCall` expression down to the bare
+* success body. `returnType` mirrors the registered client plugin's own `returnType` option: with
+* `'data'` the call already resolves to the bare body, so it is returned directly; with `'full'`
+* (the default) the call carries `unwrap()`, so the body calls it instead of destructuring `data`
+* off the full result by hand.
+*
+* @example
+* ```ts
+* buildCallResultBody(buildClientCall(node, { clientName: 'getPetById' }))
+* // return getPetById({ ...config, throwOnError: true }).unwrap()
+* ```
+*/
+function buildCallResultBody(call, options = {}) {
+	const { returnType = "full" } = options;
+	if (returnType === "data") return `return await ${call}`;
+	return `return ${call}.unwrap()`;
+}
+/**
 * Builds the `TData` / `TError` type expressions shared by every generated hook, joining the resolved
 * success responses into `TData` and wrapping the error responses in `ResponseErrorConfig<...>`.
 */
@@ -697,7 +719,7 @@ function buildQueryKeyParams(node, options) {
 //#region ../../internals/tanstack-query/src/components/InfiniteQueryOptions.tsx
 const declarationPrinter$7 = functionPrinter({ mode: "declaration" });
 const callPrinter$3 = functionPrinter({ mode: "call" });
-function InfiniteQueryOptions({ name, clientName, initialPageParam, cursorParam, nextParam, previousParam, node, tsResolver, queryParam, queryKeyName, queryKeyType = "typeof queryKey", memberTypeWrapper, unwrapName }) {
+function InfiniteQueryOptions({ name, clientName, initialPageParam, cursorParam, nextParam, previousParam, node, tsResolver, queryParam, queryKeyName, queryKeyType = "typeof queryKey", memberTypeWrapper, unwrapName, returnType = "full" }) {
 	const { TData: queryFnDataType, TError: errorType } = buildResponseTypes(node, tsResolver);
 	const { queryParamsTypeName, pageParamType } = resolvePageParamType(node, {
 		resolver: tsResolver,
@@ -716,11 +738,11 @@ function InfiniteQueryOptions({ name, clientName, initialPageParam, cursorParam,
 		memberTypeWrapper
 	});
 	const paramsSignature = declarationPrinter$7.print(paramsNode) ?? "";
-	const queryFnBody = `return ${buildClientCall(node, {
+	const queryFnBody = buildCallResultBody(buildClientCall(node, {
 		clientName,
 		signal: true,
 		unwrapName
-	})}.unwrap()`;
+	}), { returnType });
 	const hasNewParams = nextParam != null || previousParam != null;
 	const [getNextPageParamExpr, getPreviousPageParamExpr] = (() => {
 		if (hasNewParams) {
@@ -1249,7 +1271,7 @@ function buildMutationConfigParamsNode(node) {
 		default: "{}"
 	})] });
 }
-function MutationOptions({ name, clientName, node, tsResolver, mutationKeyName }) {
+function MutationOptions({ name, clientName, node, tsResolver, mutationKeyName, returnType = "full" }) {
 	const { TData, TError } = buildResponseTypes(node, tsResolver);
 	const configParamsNode = buildMutationConfigParamsNode(node);
 	const paramsSignature = declarationPrinter$3.print(configParamsNode) ?? "";
@@ -1258,10 +1280,10 @@ function MutationOptions({ name, clientName, node, tsResolver, mutationKeyName }
 	const groupedParamsNode = createFunctionParameters({ params: groupedParam ? [groupedParam] : [] });
 	const TRequest = hasMutationParams ? tsResolver.response.options(node) : "undefined";
 	const argBindingStr = hasMutationParams ? callPrinter$2.print(groupedParamsNode) ?? "" : "_";
-	const mutationFnBody = `return ${buildClientCall(node, {
+	const mutationFnBody = buildCallResultBody(buildClientCall(node, {
 		clientName,
 		signal: false
-	})}.unwrap()`;
+	}), { returnType });
 	return /* @__PURE__ */ jsx(File.Source, {
 		name,
 		isExportable: true,
@@ -1429,16 +1451,16 @@ function Query({ name, queryKeyTypeName, queryOptionsName, queryKeyName, node, t
 //#region src/components/QueryOptions.tsx
 const declarationPrinter = functionPrinter({ mode: "declaration" });
 const callPrinter = functionPrinter({ mode: "call" });
-function QueryOptions({ name, clientName, node, tsResolver, queryKeyName }) {
+function QueryOptions({ name, clientName, node, tsResolver, queryKeyName, returnType = "full" }) {
 	const { TData, TError } = buildResponseTypes(node, tsResolver);
 	const queryKeyParamsNode = buildQueryKeyParams(node, { resolver: tsResolver });
 	const queryKeyParamsCall = callPrinter.print(queryKeyParamsNode) ?? "";
 	const paramsNode = buildQueryOptionsParams(node, { resolver: tsResolver });
 	const paramsSignature = declarationPrinter.print(paramsNode) ?? "";
-	const queryFnBody = `return ${buildClientCall(node, {
+	const queryFnBody = buildCallResultBody(buildClientCall(node, {
 		clientName,
 		signal: true
-	})}.unwrap()`;
+	}), { returnType });
 	return /* @__PURE__ */ jsx(File.Source, {
 		name,
 		isExportable: true,
@@ -1606,7 +1628,8 @@ const infiniteQueryGenerator = defineGenerator({
 					nextParam: infiniteOptions.nextParam,
 					previousParam: infiniteOptions.previousParam,
 					initialPageParam: infiniteOptions.initialPageParam,
-					queryParam: infiniteOptions.queryParam
+					queryParam: infiniteOptions.queryParam,
+					returnType: contractOp.returnType
 				}),
 				/* @__PURE__ */ jsx(File.Import, {
 					name: ["useInfiniteQuery"],
@@ -1754,7 +1777,8 @@ const mutationGenerator = defineGenerator({
 					clientName: calledClientName,
 					mutationKeyName,
 					node,
-					tsResolver
+					tsResolver,
+					returnType: contractOp.returnType
 				}),
 				mutation && hooks && /* @__PURE__ */ jsxs(Fragment, { children: [
 					/* @__PURE__ */ jsx(File.Import, {
@@ -1905,7 +1929,8 @@ const queryGenerator = defineGenerator({
 					clientName: calledClientName,
 					queryKeyName,
 					node,
-					tsResolver
+					tsResolver,
+					returnType: contractOp.returnType
 				}),
 				query && hooks && /* @__PURE__ */ jsxs(Fragment, { children: [
 					/* @__PURE__ */ jsx(File.Import, {
@@ -2078,7 +2103,8 @@ const suspenseInfiniteQueryGenerator = defineGenerator({
 					nextParam: infiniteOptions.nextParam,
 					previousParam: infiniteOptions.previousParam,
 					initialPageParam: infiniteOptions.initialPageParam,
-					queryParam: infiniteOptions.queryParam
+					queryParam: infiniteOptions.queryParam,
+					returnType: contractOp.returnType
 				}),
 				/* @__PURE__ */ jsx(File.Import, {
 					name: ["useSuspenseInfiniteQuery"],
@@ -2233,7 +2259,8 @@ const suspenseQueryGenerator = defineGenerator({
 					clientName: calledClientName,
 					queryKeyName,
 					node,
-					tsResolver
+					tsResolver,
+					returnType: contractOp.returnType
 				}),
 				/* @__PURE__ */ jsx(File.Import, {
 					name: ["useSuspenseQuery"],