@constructive-io/graphql-codegen 3.2.1 → 3.3.1

package/core/codegen/custom-queries.js

@@ -35,243 +35,534 @@ var __importStar = (this && this.__importStar) || (function () {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generateCustomQueryHook = generateCustomQueryHook;
 exports.generateAllCustomQueryHooks = generateAllCustomQueryHooks;
+/**
+ * Custom query hook generators for non-table operations (Babel AST-based)
+ *
+ * Generates hooks for operations discovered via schema introspection
+ * that are NOT table CRUD operations (e.g., currentUser, nodeById, etc.)
+ *
+ * Delegates to ORM custom query operations:
+ *   getClient().query.operationName(args, { select }).unwrap()
+ *
+ * Output structure:
+ * queries/
+ *   useCurrentUserQuery.ts
+ *   useNodeQuery.ts
+ *   ...
+ */
 const t = __importStar(require("@babel/types"));
 const babel_ast_1 = require("./babel-ast");
-const schema_gql_ast_1 = require("./schema-gql-ast");
+const hooks_ast_1 = require("./hooks-ast");
+const select_helpers_1 = require("./select-helpers");
 const type_resolver_1 = require("./type-resolver");
 const utils_1 = require("./utils");
-function generateVariablesProperties(args, tracker) {
-    return args.map((arg) => ({
-        name: arg.name,
-        type: (0, type_resolver_1.typeRefToTsType)(arg.type, tracker),
-        optional: !(0, type_resolver_1.isTypeRequired)(arg.type),
-        docs: arg.description ? [arg.description] : undefined,
-    }));
-}
 function generateCustomQueryHook(options) {
-    const { operation, typeRegistry, maxDepth = 2, skipQueryField = true, reactQueryEnabled = true, tableTypeNames, useCentralizedKeys = true, } = options;
+    const { operation, typeRegistry, reactQueryEnabled = true, tableTypeNames, useCentralizedKeys = true, } = options;
     const hookName = (0, type_resolver_1.getOperationHookName)(operation.name, 'query');
     const fileName = (0, type_resolver_1.getOperationFileName)(operation.name, 'query');
-    const variablesTypeName = (0, type_resolver_1.getOperationVariablesTypeName)(operation.name, 'query');
-    const resultTypeName = (0, type_resolver_1.getOperationResultTypeName)(operation.name, 'query');
-    const documentConstName = (0, type_resolver_1.getDocumentConstName)(operation.name, 'query');
     const queryKeyName = (0, type_resolver_1.getQueryKeyName)(operation.name);
+    const varTypeName = `${(0, utils_1.ucFirst)(operation.name)}Variables`;
     const tracker = (0, type_resolver_1.createTypeTracker)({ tableTypeNames });
-    const queryDocument = (0, schema_gql_ast_1.buildCustomQueryString)({
-        operation,
-        typeRegistry,
-        maxDepth,
-        skipQueryField,
-    });
-    const statements = [];
-    const variablesProps = operation.args.length > 0
-        ? generateVariablesProperties(operation.args, tracker)
-        : [];
+    const hasArgs = operation.args.length > 0;
+    const hasRequiredArgs = operation.args.some((arg) => (0, type_resolver_1.isTypeRequired)(arg.type));
     const resultType = (0, type_resolver_1.typeRefToTsType)(operation.returnType, tracker);
-    const schemaTypes = tracker.getImportableTypes();
-    const tableTypes = tracker.getTableTypes();
+    for (const arg of operation.args) {
+        (0, type_resolver_1.typeRefToTsType)(arg.type, tracker);
+    }
+    const selectTypeName = (0, select_helpers_1.getSelectTypeName)(operation.returnType);
+    const payloadTypeName = (0, type_resolver_1.getTypeBaseName)(operation.returnType);
+    const hasSelect = !!selectTypeName && !!payloadTypeName;
+    const statements = [];
+    // Imports
     if (reactQueryEnabled) {
-        const reactQueryImport = t.importDeclaration([t.importSpecifier(t.identifier('useQuery'), t.identifier('useQuery'))], t.stringLiteral('@tanstack/react-query'));
-        statements.push(reactQueryImport);
-        const reactQueryTypeImport = t.importDeclaration([
-            t.importSpecifier(t.identifier('UseQueryOptions'), t.identifier('UseQueryOptions')),
-            t.importSpecifier(t.identifier('QueryClient'), t.identifier('QueryClient')),
-        ], t.stringLiteral('@tanstack/react-query'));
-        reactQueryTypeImport.importKind = 'type';
-        statements.push(reactQueryTypeImport);
+        statements.push((0, hooks_ast_1.createImportDeclaration)('@tanstack/react-query', ['useQuery']));
+        statements.push((0, hooks_ast_1.createImportDeclaration)('@tanstack/react-query', ['UseQueryOptions', 'UseQueryResult', 'QueryClient'], true));
     }
-    const clientImport = t.importDeclaration([t.importSpecifier(t.identifier('execute'), t.identifier('execute'))], t.stringLiteral('../client'));
-    statements.push(clientImport);
-    const clientTypeImport = t.importDeclaration([t.importSpecifier(t.identifier('ExecuteOptions'), t.identifier('ExecuteOptions'))], t.stringLiteral('../client'));
-    clientTypeImport.importKind = 'type';
-    statements.push(clientTypeImport);
-    if (tableTypes.length > 0) {
-        const typesImport = t.importDeclaration(tableTypes.map((tt) => t.importSpecifier(t.identifier(tt), t.identifier(tt))), t.stringLiteral('../types'));
-        typesImport.importKind = 'type';
-        statements.push(typesImport);
+    statements.push((0, hooks_ast_1.createImportDeclaration)('../client', ['getClient']));
+    statements.push((0, hooks_ast_1.createImportDeclaration)('../selection', ['buildSelectionArgs']));
+    statements.push((0, hooks_ast_1.createImportDeclaration)('../selection', ['SelectionConfig'], true));
+    if (useCentralizedKeys) {
+        statements.push((0, hooks_ast_1.createImportDeclaration)('../query-keys', ['customQueryKeys']));
     }
-    if (schemaTypes.length > 0) {
-        const schemaTypesImport = t.importDeclaration(schemaTypes.map((st) => t.importSpecifier(t.identifier(st), t.identifier(st))), t.stringLiteral('../schema-types'));
-        schemaTypesImport.importKind = 'type';
-        statements.push(schemaTypesImport);
+    if (hasArgs) {
+        statements.push((0, hooks_ast_1.createImportDeclaration)('../../orm/query', [varTypeName], true));
     }
-    if (useCentralizedKeys) {
-        const queryKeyImport = t.importDeclaration([t.importSpecifier(t.identifier('customQueryKeys'), t.identifier('customQueryKeys'))], t.stringLiteral('../query-keys'));
-        statements.push(queryKeyImport);
+    const inputTypeImports = [];
+    if (hasSelect) {
+        inputTypeImports.push(selectTypeName);
+        inputTypeImports.push(payloadTypeName);
+    }
+    else {
+        const baseName = (0, type_resolver_1.getTypeBaseName)(operation.returnType);
+        if (baseName && !tracker.referencedTypes.has('__skip__')) {
+            for (const refType of tracker.referencedTypes) {
+                if (!inputTypeImports.includes(refType)) {
+                    inputTypeImports.push(refType);
+                }
+            }
+        }
+    }
+    if (inputTypeImports.length > 0) {
+        statements.push((0, hooks_ast_1.createImportDeclaration)('../../orm/input-types', inputTypeImports, true));
+    }
+    if (hasSelect) {
+        statements.push((0, hooks_ast_1.createImportDeclaration)('../../orm/select-types', ['InferSelectResult', 'StrictSelect'], true));
     }
-    const queryDocConst = t.variableDeclaration('const', [
-        t.variableDeclarator(t.identifier(documentConstName), t.templateLiteral([t.templateElement({ raw: '\n' + queryDocument, cooked: '\n' + queryDocument }, true)], [])),
-    ]);
-    const queryDocExport = t.exportNamedDeclaration(queryDocConst);
-    (0, babel_ast_1.addJSDocComment)(queryDocExport, ['GraphQL query document']);
-    statements.push(queryDocExport);
-    if (operation.args.length > 0) {
-        const variablesInterfaceProps = variablesProps.map((vp) => {
-            const prop = t.tsPropertySignature(t.identifier(vp.name), t.tsTypeAnnotation(t.tsTypeReference(t.identifier(vp.type))));
-            prop.optional = vp.optional;
-            return prop;
-        });
-        const variablesInterface = t.tsInterfaceDeclaration(t.identifier(variablesTypeName), null, null, t.tsInterfaceBody(variablesInterfaceProps));
-        statements.push(t.exportNamedDeclaration(variablesInterface));
+    // Re-exports
+    if (hasArgs) {
+        statements.push((0, hooks_ast_1.createTypeReExport)([varTypeName], '../../orm/query'));
+    }
+    if (hasSelect) {
+        statements.push((0, hooks_ast_1.createTypeReExport)([selectTypeName], '../../orm/input-types'));
     }
-    const resultInterfaceBody = t.tsInterfaceBody([
-        t.tsPropertySignature(t.identifier(operation.name), t.tsTypeAnnotation(t.tsTypeReference(t.identifier(resultType)))),
-    ]);
-    const resultInterface = t.tsInterfaceDeclaration(t.identifier(resultTypeName), null, null, resultInterfaceBody);
-    statements.push(t.exportNamedDeclaration(resultInterface));
+    // Query key
     if (useCentralizedKeys) {
-        const queryKeyConst = t.variableDeclaration('const', [
+        const keyDecl = t.exportNamedDeclaration(t.variableDeclaration('const', [
             t.variableDeclarator(t.identifier(queryKeyName), t.memberExpression(t.identifier('customQueryKeys'), t.identifier(operation.name))),
+        ]));
+        (0, hooks_ast_1.addJSDocComment)(keyDecl, [
+            'Query key factory - re-exported from query-keys.ts',
         ]);
-        const queryKeyExport = t.exportNamedDeclaration(queryKeyConst);
-        (0, babel_ast_1.addJSDocComment)(queryKeyExport, ['Query key factory - re-exported from query-keys.ts']);
-        statements.push(queryKeyExport);
+        statements.push(keyDecl);
     }
-    else if (operation.args.length > 0) {
-        const queryKeyArrow = t.arrowFunctionExpression([(0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(variablesTypeName)), true)], t.tsAsExpression(t.arrayExpression([t.stringLiteral(operation.name), t.identifier('variables')]), t.tsTypeReference(t.identifier('const'))));
-        const queryKeyConst = t.variableDeclaration('const', [
-            t.variableDeclarator(t.identifier(queryKeyName), queryKeyArrow),
-        ]);
-        const queryKeyExport = t.exportNamedDeclaration(queryKeyConst);
-        (0, babel_ast_1.addJSDocComment)(queryKeyExport, ['Query key factory for caching']);
-        statements.push(queryKeyExport);
+    else if (hasArgs) {
+        const keyFn = t.arrowFunctionExpression([(0, hooks_ast_1.createFunctionParam)('variables', (0, hooks_ast_1.typeRef)(varTypeName), true)], (0, babel_ast_1.asConst)(t.arrayExpression([
+            t.stringLiteral(operation.name),
+            t.identifier('variables'),
+        ])));
+        const keyDecl = t.exportNamedDeclaration(t.variableDeclaration('const', [
+            t.variableDeclarator(t.identifier(queryKeyName), keyFn),
+        ]));
+        (0, hooks_ast_1.addJSDocComment)(keyDecl, ['Query key factory for caching']);
+        statements.push(keyDecl);
     }
     else {
-        const queryKeyArrow = t.arrowFunctionExpression([], t.tsAsExpression(t.arrayExpression([t.stringLiteral(operation.name)]), t.tsTypeReference(t.identifier('const'))));
-        const queryKeyConst = t.variableDeclaration('const', [
-            t.variableDeclarator(t.identifier(queryKeyName), queryKeyArrow),
-        ]);
-        const queryKeyExport = t.exportNamedDeclaration(queryKeyConst);
-        (0, babel_ast_1.addJSDocComment)(queryKeyExport, ['Query key factory for caching']);
-        statements.push(queryKeyExport);
+        const keyFn = t.arrowFunctionExpression([], (0, babel_ast_1.asConst)(t.arrayExpression([t.stringLiteral(operation.name)])));
+        const keyDecl = t.exportNamedDeclaration(t.variableDeclaration('const', [
+            t.variableDeclarator(t.identifier(queryKeyName), keyFn),
+        ]));
+        (0, hooks_ast_1.addJSDocComment)(keyDecl, ['Query key factory for caching']);
+        statements.push(keyDecl);
     }
+    // Helper to build query key call expression
+    const buildQueryKeyCall = (withVars) => {
+        if (useCentralizedKeys) {
+            return withVars
+                ? (0, hooks_ast_1.callExpr)(t.identifier(queryKeyName), [t.identifier('variables')])
+                : (0, hooks_ast_1.callExpr)(t.identifier(queryKeyName), []);
+        }
+        return withVars
+            ? (0, hooks_ast_1.callExpr)(t.identifier(queryKeyName), [t.identifier('variables')])
+            : (0, hooks_ast_1.callExpr)(t.identifier(queryKeyName), []);
+    };
+    // Helper to build the fields+StrictSelect intersection type
+    const fieldsSelectionType = (s) => t.tsParenthesizedType(t.tsIntersectionType([
+        t.tsTypeLiteral([
+            t.tsPropertySignature(t.identifier('fields'), t.tsTypeAnnotation(s)),
+        ]),
+        (0, hooks_ast_1.typeRef)('StrictSelect', [s, (0, hooks_ast_1.typeRef)(selectTypeName)]),
+    ]));
+    const selectedResultType = (sel) => (0, hooks_ast_1.customSelectResultTypeLiteral)(operation.name, operation.returnType, payloadTypeName, sel);
+    // Hook
     if (reactQueryEnabled) {
-        const hasArgs = operation.args.length > 0;
-        const hasRequiredArgs = operation.args.some((arg) => (0, type_resolver_1.isTypeRequired)(arg.type));
-        const hookBodyStatements = [];
-        const useQueryOptions = [];
-        if (hasArgs) {
-            useQueryOptions.push(t.objectProperty(t.identifier('queryKey'), t.callExpression(t.identifier(queryKeyName), [t.identifier('variables')])));
-            useQueryOptions.push(t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], (0, babel_ast_1.createTypedCallExpression)(t.identifier('execute'), [t.identifier(documentConstName), t.identifier('variables')], [
-                t.tsTypeReference(t.identifier(resultTypeName)),
-                t.tsTypeReference(t.identifier(variablesTypeName)),
-            ]))));
-            if (hasRequiredArgs) {
-                useQueryOptions.push(t.objectProperty(t.identifier('enabled'), t.logicalExpression('&&', t.unaryExpression('!', t.unaryExpression('!', t.identifier('variables'))), t.binaryExpression('!==', t.optionalMemberExpression(t.identifier('options'), t.identifier('enabled'), false, true), t.booleanLiteral(false)))));
+        const description = operation.description || `Query hook for ${operation.name}`;
+        const argNames = operation.args.map((a) => a.name).join(', ');
+        const exampleCall = hasArgs
+            ? hasSelect
+                ? `${hookName}({ variables: { ${argNames} }, selection: { fields: { id: true } } })`
+                : `${hookName}({ variables: { ${argNames} } })`
+            : hasSelect
+                ? `${hookName}({ selection: { fields: { id: true } } })`
+                : `${hookName}()`;
+        if (hasSelect) {
+            // Overload 1: with selection.fields
+            const o1Props = [];
+            if (hasArgs) {
+                const varType = hasRequiredArgs
+                    ? (0, hooks_ast_1.typeRef)(varTypeName)
+                    : t.tsUnionType([(0, hooks_ast_1.typeRef)(varTypeName), t.tsUndefinedKeyword()]);
+                o1Props.push(t.tsPropertySignature(t.identifier('variables'), t.tsTypeAnnotation(varType)));
             }
+            o1Props.push(t.tsPropertySignature(t.identifier('selection'), t.tsTypeAnnotation(fieldsSelectionType((0, hooks_ast_1.sRef)()))));
+            const o1ParamType = t.tsIntersectionType([
+                t.tsTypeLiteral(o1Props),
+                (0, hooks_ast_1.omitType)((0, hooks_ast_1.typeRef)('UseQueryOptions', [
+                    selectedResultType((0, hooks_ast_1.sRef)()),
+                    (0, hooks_ast_1.typeRef)('Error'),
+                    (0, hooks_ast_1.typeRef)('TData'),
+                ]), ['queryKey', 'queryFn']),
+            ]);
+            const o1 = (0, hooks_ast_1.exportDeclareFunction)(hookName, (0, hooks_ast_1.createSAndTDataTypeParams)(selectTypeName, selectedResultType((0, hooks_ast_1.sRef)())), [(0, hooks_ast_1.createFunctionParam)('params', o1ParamType)], (0, hooks_ast_1.typeRef)('UseQueryResult', [(0, hooks_ast_1.typeRef)('TData')]));
+            (0, hooks_ast_1.addJSDocComment)(o1, [
+                description,
+                '',
+                '@example',
+                '```tsx',
+                `const { data, isLoading } = ${exampleCall};`,
+                '',
+                `if (data?.${operation.name}) {`,
+                `  console.log(data.${operation.name});`,
+                '}',
+                '```',
+            ]);
+            statements.push(o1);
+            // Implementation
+            const implProps = [];
+            if (hasArgs) {
+                const varProp = t.tsPropertySignature(t.identifier('variables'), t.tsTypeAnnotation((0, hooks_ast_1.typeRef)(varTypeName)));
+                if (!hasRequiredArgs)
+                    varProp.optional = true;
+                implProps.push(varProp);
+            }
+            const implSelProp = t.tsPropertySignature(t.identifier('selection'), t.tsTypeAnnotation((0, hooks_ast_1.selectionConfigType)((0, hooks_ast_1.typeRef)(selectTypeName))));
+            implProps.push(implSelProp);
+            const implParamType = t.tsIntersectionType([
+                t.tsTypeLiteral(implProps),
+                (0, hooks_ast_1.useQueryOptionsImplType)(),
+            ]);
+            const implParam = (0, hooks_ast_1.createFunctionParam)('params', implParamType, false);
+            const body = [];
+            if (hasArgs) {
+                body.push((0, hooks_ast_1.constDecl)('variables', hasRequiredArgs
+                    ? t.memberExpression(t.identifier('params'), t.identifier('variables'))
+                    : t.logicalExpression('??', t.memberExpression(t.identifier('params'), t.identifier('variables')), t.objectExpression([]))));
+            }
+            body.push((0, hooks_ast_1.buildSelectionArgsCall)(selectTypeName));
+            if (hasArgs) {
+                const destructPattern = t.objectPattern([
+                    t.objectProperty(t.identifier('variables'), t.identifier('_variables'), false, false),
+                    t.objectProperty(t.identifier('selection'), t.identifier('_selection'), false, false),
+                    t.restElement(t.identifier('queryOptions')),
+                ]);
+                body.push(t.variableDeclaration('const', [
+                    t.variableDeclarator(destructPattern, t.logicalExpression('??', t.identifier('params'), t.objectExpression([]))),
+                ]));
+                body.push((0, hooks_ast_1.voidStatement)('_variables'));
+                body.push((0, hooks_ast_1.voidStatement)('_selection'));
+            }
+            else {
+                body.push((0, hooks_ast_1.destructureParamsWithSelection)('queryOptions'));
+                body.push((0, hooks_ast_1.voidStatement)('_selection'));
+            }
+            const selectArgExpr = t.objectExpression([
+                (0, hooks_ast_1.objectProp)('select', t.memberExpression(t.identifier('args'), t.identifier('select'))),
+            ]);
+            const queryFnArgs = hasArgs
+                ? [
+                    hasRequiredArgs
+                        ? t.tsNonNullExpression(t.identifier('variables'))
+                        : t.identifier('variables'),
+                    selectArgExpr,
+                ]
+                : [selectArgExpr];
+            const queryFnExpr = t.arrowFunctionExpression([], (0, hooks_ast_1.getClientCustomCallUnwrap)('query', operation.name, queryFnArgs));
+            const extraProps = [];
+            const enabledExpr = hasRequiredArgs
+                ? t.logicalExpression('&&', t.unaryExpression('!', t.unaryExpression('!', t.identifier('variables'))), t.binaryExpression('!==', t.optionalMemberExpression(t.identifier('params'), t.identifier('enabled'), false, true), t.booleanLiteral(false)))
+                : undefined;
+            extraProps.push((0, hooks_ast_1.spreadObj)(t.identifier('queryOptions')));
+            body.push((0, hooks_ast_1.returnUseQuery)(buildQueryKeyCall(hasArgs), queryFnExpr, extraProps, enabledExpr));
+            statements.push((0, hooks_ast_1.exportFunction)(hookName, null, [implParam], body));
         }
         else {
-            useQueryOptions.push(t.objectProperty(t.identifier('queryKey'), t.callExpression(t.identifier(queryKeyName), [])));
-            useQueryOptions.push(t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], (0, babel_ast_1.createTypedCallExpression)(t.identifier('execute'), [t.identifier(documentConstName)], [t.tsTypeReference(t.identifier(resultTypeName))]))));
+            // Without select: simple hook (scalar return type)
+            const resultTypeLiteral = t.tsTypeLiteral([
+                t.tsPropertySignature(t.identifier(operation.name), t.tsTypeAnnotation((0, hooks_ast_1.typeRef)(resultType))),
+            ]);
+            const optType = (0, hooks_ast_1.omitType)((0, hooks_ast_1.typeRef)('UseQueryOptions', [
+                resultTypeLiteral,
+                (0, hooks_ast_1.typeRef)('Error'),
+                (0, hooks_ast_1.typeRef)('TData'),
+            ]), ['queryKey', 'queryFn']);
+            let paramType;
+            if (hasArgs) {
+                const varProp = t.tsPropertySignature(t.identifier('variables'), t.tsTypeAnnotation((0, hooks_ast_1.typeRef)(varTypeName)));
+                if (!hasRequiredArgs)
+                    varProp.optional = true;
+                paramType = t.tsIntersectionType([t.tsTypeLiteral([varProp]), optType]);
+            }
+            else {
+                paramType = optType;
+            }
+            const implParam = (0, hooks_ast_1.createFunctionParam)('params', paramType, !hasRequiredArgs);
+            const hookDecl = (0, hooks_ast_1.exportDeclareFunction)(hookName, (0, hooks_ast_1.createTDataTypeParam)(resultTypeLiteral), [implParam], (0, hooks_ast_1.typeRef)('UseQueryResult', [(0, hooks_ast_1.typeRef)('TData')]));
+            (0, hooks_ast_1.addJSDocComment)(hookDecl, [
+                description,
+                '',
+                '@example',
+                '```tsx',
+                `const { data, isLoading } = ${exampleCall};`,
+                '',
+                `if (data?.${operation.name}) {`,
+                `  console.log(data.${operation.name});`,
+                '}',
+                '```',
+            ]);
+            const body = [];
+            if (hasArgs) {
+                body.push((0, hooks_ast_1.constDecl)('variables', hasRequiredArgs
+                    ? t.optionalMemberExpression(t.identifier('params'), t.identifier('variables'), false, true)
+                    : t.logicalExpression('??', t.optionalMemberExpression(t.identifier('params'), t.identifier('variables'), false, true), t.objectExpression([]))));
+                const destructPattern = t.objectPattern([
+                    t.objectProperty(t.identifier('variables'), t.identifier('_variables'), false, false),
+                    t.restElement(t.identifier('queryOptions')),
+                ]);
+                body.push(t.variableDeclaration('const', [
+                    t.variableDeclarator(destructPattern, t.logicalExpression('??', t.identifier('params'), t.objectExpression([]))),
+                ]));
+                body.push((0, hooks_ast_1.voidStatement)('_variables'));
+            }
+            else {
+                body.push((0, hooks_ast_1.constDecl)('queryOptions', t.logicalExpression('??', t.identifier('params'), t.objectExpression([]))));
+            }
+            const queryFnArgs = hasArgs
+                ? [
+                    hasRequiredArgs
+                        ? t.tsNonNullExpression(t.identifier('variables'))
+                        : t.identifier('variables'),
+                ]
+                : [];
+            const queryFnExpr = t.arrowFunctionExpression([], (0, hooks_ast_1.getClientCustomCallUnwrap)('query', operation.name, queryFnArgs));
+            const enabledExpr = hasRequiredArgs
+                ? t.logicalExpression('&&', t.unaryExpression('!', t.unaryExpression('!', t.identifier('variables'))), t.binaryExpression('!==', t.optionalMemberExpression(t.identifier('params'), t.identifier('enabled'), false, true), t.booleanLiteral(false)))
+                : undefined;
+            body.push((0, hooks_ast_1.returnUseQuery)(buildQueryKeyCall(hasArgs), queryFnExpr, [(0, hooks_ast_1.spreadObj)(t.identifier('queryOptions'))], enabledExpr));
+            // We need the implementation version (not declare), with return type
+            statements.push(hookDecl);
+            statements.push((0, hooks_ast_1.exportFunction)(hookName, (0, hooks_ast_1.createTDataTypeParam)(resultTypeLiteral), [implParam], body, (0, hooks_ast_1.typeRef)('UseQueryResult', [(0, hooks_ast_1.typeRef)('TData')])));
         }
-        useQueryOptions.push(t.spreadElement(t.identifier('options')));
-        hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useQuery'), [t.objectExpression(useQueryOptions)])));
-        const hookParams = [];
+    }
+    // Fetch function (non-hook)
+    const fetchFnName = `fetch${(0, utils_1.ucFirst)(operation.name)}Query`;
+    const fetchArgNames = operation.args.map((a) => a.name).join(', ');
+    const fetchExampleCall = hasArgs
+        ? hasSelect
+            ? `${fetchFnName}({ variables: { ${fetchArgNames} }, selection: { fields: { id: true } } })`
+            : `${fetchFnName}({ variables: { ${fetchArgNames} } })`
+        : hasSelect
+            ? `${fetchFnName}({ selection: { fields: { id: true } } })`
+            : `${fetchFnName}()`;
+    if (hasSelect) {
+        // Overload 1: with fields
+        const f1Props = [];
         if (hasArgs) {
-            hookParams.push((0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(variablesTypeName)), !hasRequiredArgs));
+            const varType = hasRequiredArgs
+                ? (0, hooks_ast_1.typeRef)(varTypeName)
+                : t.tsUnionType([(0, hooks_ast_1.typeRef)(varTypeName), t.tsUndefinedKeyword()]);
+            f1Props.push(t.tsPropertySignature(t.identifier('variables'), t.tsTypeAnnotation(varType)));
         }
-        const optionsTypeStr = `Omit<UseQueryOptions<${resultTypeName}, Error>, 'queryKey' | 'queryFn'>`;
-        const optionsParam = t.identifier('options');
-        optionsParam.optional = true;
-        optionsParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeReference(t.identifier(optionsTypeStr)));
-        hookParams.push(optionsParam);
-        const hookFunc = t.functionDeclaration(t.identifier(hookName), hookParams, t.blockStatement(hookBodyStatements));
-        const hookExport = t.exportNamedDeclaration(hookFunc);
-        const description = operation.description || `Query hook for ${operation.name}`;
-        const argNames = operation.args.map((a) => a.name).join(', ');
-        const exampleCall = hasArgs ? `${hookName}({ ${argNames} })` : `${hookName}()`;
-        (0, babel_ast_1.addJSDocComment)(hookExport, [
-            description,
+        f1Props.push(t.tsPropertySignature(t.identifier('selection'), t.tsTypeAnnotation(fieldsSelectionType((0, hooks_ast_1.sRef)()))));
+        const f1Decl = (0, hooks_ast_1.exportAsyncDeclareFunction)(fetchFnName, (0, hooks_ast_1.createSTypeParam)(selectTypeName), [(0, hooks_ast_1.createFunctionParam)('params', t.tsTypeLiteral(f1Props))], (0, hooks_ast_1.typeRef)('Promise', [selectedResultType((0, hooks_ast_1.sRef)())]));
+        (0, hooks_ast_1.addJSDocComment)(f1Decl, [
+            `Fetch ${operation.name} without React hooks`,
             '',
             '@example',
-            '```tsx',
-            `const { data, isLoading } = ${exampleCall};`,
-            '',
-            `if (data?.${operation.name}) {`,
-            `  console.log(data.${operation.name});`,
-            '}',
+            '```ts',
+            `const data = await ${fetchExampleCall};`,
             '```',
         ]);
-        statements.push(hookExport);
-    }
-    const fetchFnName = `fetch${(0, utils_1.ucFirst)(operation.name)}Query`;
-    const hasArgs = operation.args.length > 0;
-    const hasRequiredArgs = operation.args.some((arg) => (0, type_resolver_1.isTypeRequired)(arg.type));
-    const fetchBodyStatements = [];
-    if (hasArgs) {
-        fetchBodyStatements.push(t.returnStatement((0, babel_ast_1.createTypedCallExpression)(t.identifier('execute'), [t.identifier(documentConstName), t.identifier('variables'), t.identifier('options')], [
-            t.tsTypeReference(t.identifier(resultTypeName)),
-            t.tsTypeReference(t.identifier(variablesTypeName)),
-        ])));
+        statements.push(f1Decl);
+        // Implementation
+        const fImplProps = [];
+        if (hasArgs) {
+            const varProp = t.tsPropertySignature(t.identifier('variables'), t.tsTypeAnnotation((0, hooks_ast_1.typeRef)(varTypeName)));
+            if (!hasRequiredArgs)
+                varProp.optional = true;
+            fImplProps.push(varProp);
+        }
+        const fImplSelProp = t.tsPropertySignature(t.identifier('selection'), t.tsTypeAnnotation((0, hooks_ast_1.selectionConfigType)((0, hooks_ast_1.typeRef)(selectTypeName))));
+        fImplProps.push(fImplSelProp);
+        const fBody = [];
+        if (hasArgs) {
+            fBody.push((0, hooks_ast_1.constDecl)('variables', hasRequiredArgs
+                ? t.memberExpression(t.identifier('params'), t.identifier('variables'))
+                : t.logicalExpression('??', t.memberExpression(t.identifier('params'), t.identifier('variables')), t.objectExpression([]))));
+        }
+        fBody.push((0, hooks_ast_1.buildSelectionArgsCall)(selectTypeName));
+        const selectArgExpr = t.objectExpression([
+            (0, hooks_ast_1.objectProp)('select', t.memberExpression(t.identifier('args'), t.identifier('select'))),
+        ]);
+        const fCallArgs = hasArgs
+            ? [
+                hasRequiredArgs
+                    ? t.tsNonNullExpression(t.identifier('variables'))
+                    : t.identifier('variables'),
+                selectArgExpr,
+            ]
+            : [selectArgExpr];
+        fBody.push(t.returnStatement((0, hooks_ast_1.getClientCustomCallUnwrap)('query', operation.name, fCallArgs)));
+        statements.push((0, hooks_ast_1.exportAsyncFunction)(fetchFnName, null, [
+            (0, hooks_ast_1.createFunctionParam)('params', t.tsTypeLiteral(fImplProps), false),
+        ], fBody));
     }
     else {
-        fetchBodyStatements.push(t.returnStatement((0, babel_ast_1.createTypedCallExpression)(t.identifier('execute'), [t.identifier(documentConstName), t.identifier('undefined'), t.identifier('options')], [t.tsTypeReference(t.identifier(resultTypeName))])));
-    }
-    const fetchParams = [];
-    if (hasArgs) {
-        fetchParams.push((0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(variablesTypeName)), !hasRequiredArgs));
+        const fBody = [];
+        if (hasArgs) {
+            const fProps = [];
+            const varProp = t.tsPropertySignature(t.identifier('variables'), t.tsTypeAnnotation((0, hooks_ast_1.typeRef)(varTypeName)));
+            if (!hasRequiredArgs)
+                varProp.optional = true;
+            fProps.push(varProp);
+            fBody.push((0, hooks_ast_1.constDecl)('variables', hasRequiredArgs
+                ? t.optionalMemberExpression(t.identifier('params'), t.identifier('variables'), false, true)
+                : t.logicalExpression('??', t.optionalMemberExpression(t.identifier('params'), t.identifier('variables'), false, true), t.objectExpression([]))));
+            const fCallArgs = hasRequiredArgs
+                ? [t.tsNonNullExpression(t.identifier('variables'))]
+                : [t.identifier('variables')];
+            fBody.push(t.returnStatement((0, hooks_ast_1.getClientCustomCallUnwrap)('query', operation.name, fCallArgs)));
+            const fDecl = (0, hooks_ast_1.exportAsyncFunction)(fetchFnName, null, [
+                (0, hooks_ast_1.createFunctionParam)('params', t.tsTypeLiteral(fProps), !hasRequiredArgs),
+            ], fBody);
+            (0, hooks_ast_1.addJSDocComment)(fDecl, [
+                `Fetch ${operation.name} without React hooks`,
+                '',
+                '@example',
+                '```ts',
+                `const data = await ${fetchExampleCall};`,
+                '```',
+            ]);
+            statements.push(fDecl);
+        }
+        else {
+            fBody.push(t.returnStatement((0, hooks_ast_1.getClientCustomCallUnwrap)('query', operation.name, [])));
+            const fDecl = (0, hooks_ast_1.exportAsyncFunction)(fetchFnName, null, [], fBody);
+            (0, hooks_ast_1.addJSDocComment)(fDecl, [
+                `Fetch ${operation.name} without React hooks`,
+                '',
+                '@example',
+                '```ts',
+                `const data = await ${fetchExampleCall};`,
+                '```',
+            ]);
+            statements.push(fDecl);
+        }
     }
-    fetchParams.push((0, babel_ast_1.typedParam)('options', t.tsTypeReference(t.identifier('ExecuteOptions')), true));
-    const fetchFunc = t.functionDeclaration(t.identifier(fetchFnName), fetchParams, t.blockStatement(fetchBodyStatements));
-    fetchFunc.async = true;
-    fetchFunc.returnType = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Promise'), t.tsTypeParameterInstantiation([t.tsTypeReference(t.identifier(resultTypeName))])));
-    const fetchExport = t.exportNamedDeclaration(fetchFunc);
-    const argNames = operation.args.map((a) => a.name).join(', ');
-    const fetchExampleCall = hasArgs ? `${fetchFnName}({ ${argNames} })` : `${fetchFnName}()`;
-    (0, babel_ast_1.addJSDocComment)(fetchExport, [
-        `Fetch ${operation.name} without React hooks`,
-        '',
-        '@example',
-        '```ts',
-        `const data = await ${fetchExampleCall};`,
-        '```',
-    ]);
-    statements.push(fetchExport);
+    // Prefetch function
     if (reactQueryEnabled) {
         const prefetchFnName = `prefetch${(0, utils_1.ucFirst)(operation.name)}Query`;
-        const prefetchBodyStatements = [];
-        const prefetchQueryOptions = [];
-        if (hasArgs) {
-            prefetchQueryOptions.push(t.objectProperty(t.identifier('queryKey'), t.callExpression(t.identifier(queryKeyName), [t.identifier('variables')])));
-            prefetchQueryOptions.push(t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], (0, babel_ast_1.createTypedCallExpression)(t.identifier('execute'), [t.identifier(documentConstName), t.identifier('variables'), t.identifier('options')], [
-                t.tsTypeReference(t.identifier(resultTypeName)),
-                t.tsTypeReference(t.identifier(variablesTypeName)),
-            ]))));
+        const prefetchArgNames = operation.args.map((a) => a.name).join(', ');
+        const prefetchExampleCall = hasArgs
+            ? hasSelect
+                ? `${prefetchFnName}(queryClient, { variables: { ${prefetchArgNames} }, selection: { fields: { id: true } } })`
+                : `${prefetchFnName}(queryClient, { variables: { ${prefetchArgNames} } })`
+            : hasSelect
+                ? `${prefetchFnName}(queryClient, { selection: { fields: { id: true } } })`
+                : `${prefetchFnName}(queryClient)`;
+        if (hasSelect) {
+            // Overload 1: with fields
+            const p1Props = [];
+            if (hasArgs) {
+                const varType = hasRequiredArgs
+                    ? (0, hooks_ast_1.typeRef)(varTypeName)
+                    : t.tsUnionType([(0, hooks_ast_1.typeRef)(varTypeName), t.tsUndefinedKeyword()]);
+                p1Props.push(t.tsPropertySignature(t.identifier('variables'), t.tsTypeAnnotation(varType)));
+            }
+            p1Props.push(t.tsPropertySignature(t.identifier('selection'), t.tsTypeAnnotation(fieldsSelectionType((0, hooks_ast_1.sRef)()))));
+            const p1Decl = (0, hooks_ast_1.exportAsyncDeclareFunction)(prefetchFnName, (0, hooks_ast_1.createSTypeParam)(selectTypeName), [
+                (0, hooks_ast_1.createFunctionParam)('queryClient', (0, hooks_ast_1.typeRef)('QueryClient')),
+                (0, hooks_ast_1.createFunctionParam)('params', t.tsTypeLiteral(p1Props)),
+            ], (0, hooks_ast_1.typeRef)('Promise', [t.tsVoidKeyword()]));
+            (0, hooks_ast_1.addJSDocComment)(p1Decl, [
+                `Prefetch ${operation.name} for SSR or cache warming`,
+                '',
+                '@example',
+                '```ts',
+                `await ${prefetchExampleCall};`,
+                '```',
+            ]);
+            statements.push(p1Decl);
+            // Implementation
+            const pImplProps = [];
+            if (hasArgs) {
+                const varProp = t.tsPropertySignature(t.identifier('variables'), t.tsTypeAnnotation((0, hooks_ast_1.typeRef)(varTypeName)));
+                if (!hasRequiredArgs)
+                    varProp.optional = true;
+                pImplProps.push(varProp);
+            }
+            const pImplSelProp = t.tsPropertySignature(t.identifier('selection'), t.tsTypeAnnotation((0, hooks_ast_1.selectionConfigType)((0, hooks_ast_1.typeRef)(selectTypeName))));
+            pImplProps.push(pImplSelProp);
+            const pBody = [];
+            if (hasArgs) {
+                pBody.push((0, hooks_ast_1.constDecl)('variables', hasRequiredArgs
+                    ? t.memberExpression(t.identifier('params'), t.identifier('variables'))
+                    : t.logicalExpression('??', t.memberExpression(t.identifier('params'), t.identifier('variables')), t.objectExpression([]))));
+            }
+            pBody.push((0, hooks_ast_1.buildSelectionArgsCall)(selectTypeName));
+            const selectArgExpr = t.objectExpression([
+                (0, hooks_ast_1.objectProp)('select', t.memberExpression(t.identifier('args'), t.identifier('select'))),
+            ]);
+            const pCallArgs = hasArgs
+                ? [
+                    hasRequiredArgs
+                        ? t.tsNonNullExpression(t.identifier('variables'))
+                        : t.identifier('variables'),
+                    selectArgExpr,
+                ]
+                : [selectArgExpr];
+            const prefetchQueryCall = (0, hooks_ast_1.callExpr)(t.memberExpression(t.identifier('queryClient'), t.identifier('prefetchQuery')), [
+                t.objectExpression([
+                    (0, hooks_ast_1.objectProp)('queryKey', buildQueryKeyCall(hasArgs)),
+                    (0, hooks_ast_1.objectProp)('queryFn', t.arrowFunctionExpression([], (0, hooks_ast_1.getClientCustomCallUnwrap)('query', operation.name, pCallArgs))),
+                ]),
+            ]);
+            pBody.push(t.expressionStatement(t.awaitExpression(prefetchQueryCall)));
+            statements.push((0, hooks_ast_1.exportAsyncFunction)(prefetchFnName, null, [
+                (0, hooks_ast_1.createFunctionParam)('queryClient', (0, hooks_ast_1.typeRef)('QueryClient')),
+                (0, hooks_ast_1.createFunctionParam)('params', t.tsTypeLiteral(pImplProps), false),
+            ], pBody, (0, hooks_ast_1.typeRef)('Promise', [t.tsVoidKeyword()])));
         }
         else {
-            prefetchQueryOptions.push(t.objectProperty(t.identifier('queryKey'), t.callExpression(t.identifier(queryKeyName), [])));
-            prefetchQueryOptions.push(t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], (0, babel_ast_1.createTypedCallExpression)(t.identifier('execute'), [t.identifier(documentConstName), t.identifier('undefined'), t.identifier('options')], [t.tsTypeReference(t.identifier(resultTypeName))]))));
-        }
-        prefetchBodyStatements.push(t.expressionStatement(t.awaitExpression(t.callExpression(t.memberExpression(t.identifier('queryClient'), t.identifier('prefetchQuery')), [t.objectExpression(prefetchQueryOptions)]))));
-        const prefetchParams = [
-            (0, babel_ast_1.typedParam)('queryClient', t.tsTypeReference(t.identifier('QueryClient'))),
-        ];
-        if (hasArgs) {
-            prefetchParams.push((0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(variablesTypeName)), !hasRequiredArgs));
+            // Without select
+            const pBody = [];
+            const pParams = [
+                (0, hooks_ast_1.createFunctionParam)('queryClient', (0, hooks_ast_1.typeRef)('QueryClient')),
+            ];
+            if (hasArgs) {
+                const pProps = [];
+                const varProp = t.tsPropertySignature(t.identifier('variables'), t.tsTypeAnnotation((0, hooks_ast_1.typeRef)(varTypeName)));
+                if (!hasRequiredArgs)
+                    varProp.optional = true;
+                pProps.push(varProp);
+                pParams.push((0, hooks_ast_1.createFunctionParam)('params', t.tsTypeLiteral(pProps), !hasRequiredArgs));
+                pBody.push((0, hooks_ast_1.constDecl)('variables', hasRequiredArgs
+                    ? t.optionalMemberExpression(t.identifier('params'), t.identifier('variables'), false, true)
+                    : t.logicalExpression('??', t.optionalMemberExpression(t.identifier('params'), t.identifier('variables'), false, true), t.objectExpression([]))));
+                const pCallArgs = hasRequiredArgs
+                    ? [t.tsNonNullExpression(t.identifier('variables'))]
+                    : [t.identifier('variables')];
+                const prefetchQueryCall = (0, hooks_ast_1.callExpr)(t.memberExpression(t.identifier('queryClient'), t.identifier('prefetchQuery')), [
+                    t.objectExpression([
+                        (0, hooks_ast_1.objectProp)('queryKey', buildQueryKeyCall(true)),
+                        (0, hooks_ast_1.objectProp)('queryFn', t.arrowFunctionExpression([], (0, hooks_ast_1.getClientCustomCallUnwrap)('query', operation.name, pCallArgs))),
+                    ]),
+                ]);
+                pBody.push(t.expressionStatement(t.awaitExpression(prefetchQueryCall)));
+            }
+            else {
+                const prefetchQueryCall = (0, hooks_ast_1.callExpr)(t.memberExpression(t.identifier('queryClient'), t.identifier('prefetchQuery')), [
+                    t.objectExpression([
+                        (0, hooks_ast_1.objectProp)('queryKey', buildQueryKeyCall(false)),
+                        (0, hooks_ast_1.objectProp)('queryFn', t.arrowFunctionExpression([], (0, hooks_ast_1.getClientCustomCallUnwrap)('query', operation.name, []))),
+                    ]),
+                ]);
+                pBody.push(t.expressionStatement(t.awaitExpression(prefetchQueryCall)));
+            }
+            const pDecl = (0, hooks_ast_1.exportAsyncFunction)(prefetchFnName, null, pParams, pBody, (0, hooks_ast_1.typeRef)('Promise', [t.tsVoidKeyword()]));
+            (0, hooks_ast_1.addJSDocComment)(pDecl, [
+                `Prefetch ${operation.name} for SSR or cache warming`,
+                '',
+                '@example',
+                '```ts',
+                `await ${prefetchExampleCall};`,
+                '```',
+            ]);
+            statements.push(pDecl);
         }
-        prefetchParams.push((0, babel_ast_1.typedParam)('options', t.tsTypeReference(t.identifier('ExecuteOptions')), true));
-        const prefetchFunc = t.functionDeclaration(t.identifier(prefetchFnName), prefetchParams, t.blockStatement(prefetchBodyStatements));
-        prefetchFunc.async = true;
-        prefetchFunc.returnType = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Promise'), t.tsTypeParameterInstantiation([t.tsVoidKeyword()])));
-        const prefetchExport = t.exportNamedDeclaration(prefetchFunc);
-        const prefetchExampleCall = hasArgs
-            ? `${prefetchFnName}(queryClient, { ${argNames} })`
-            : `${prefetchFnName}(queryClient)`;
-        (0, babel_ast_1.addJSDocComment)(prefetchExport, [
-            `Prefetch ${operation.name} for SSR or cache warming`,
-            '',
-            '@example',
-            '```ts',
-            `await ${prefetchExampleCall};`,
-            '```',
-        ]);
-        statements.push(prefetchExport);
     }
-    const code = (0, babel_ast_1.generateCode)(statements);
     const headerText = reactQueryEnabled
         ? `Custom query hook for ${operation.name}`
         : `Custom query functions for ${operation.name}`;
-    const content = (0, utils_1.getGeneratedFileHeader)(headerText) + '\n\n' + code;
+    const content = (0, hooks_ast_1.generateHookFileCode)(headerText, statements);
     return {
         fileName,
         content,
@@ -279,13 +570,12 @@ function generateCustomQueryHook(options) {
     };
 }
 function generateAllCustomQueryHooks(options) {
-    const { operations, typeRegistry, maxDepth = 2, skipQueryField = true, reactQueryEnabled = true, tableTypeNames, useCentralizedKeys = true, } = options;
+    const { operations, typeRegistry, skipQueryField = true, reactQueryEnabled = true, tableTypeNames, useCentralizedKeys = true, } = options;
     return operations
         .filter((op) => op.kind === 'query')
         .map((operation) => generateCustomQueryHook({
         operation,
         typeRegistry,
-        maxDepth,
         skipQueryField,
         reactQueryEnabled,
         tableTypeNames,