@constructive-io/graphql-codegen 2.23.3 → 2.24.1

package/cli/codegen/custom-queries.js

@@ -1,385 +1,285 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generateCustomQueryHook = generateCustomQueryHook;
 exports.generateAllCustomQueryHooks = generateAllCustomQueryHooks;
-const ts_ast_1 = require("./ts-ast");
+const t = __importStar(require("@babel/types"));
+const babel_ast_1 = require("./babel-ast");
 const schema_gql_ast_1 = require("./schema-gql-ast");
 const type_resolver_1 = require("./type-resolver");
 const utils_1 = require("./utils");
-/**
- * Generate a custom query hook file
- */
+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 } = options;
-    const project = (0, ts_ast_1.createProject)();
+    const { operation, typeRegistry, maxDepth = 2, skipQueryField = true, 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);
-    // Create type tracker to collect referenced types (with table type awareness)
     const tracker = (0, type_resolver_1.createTypeTracker)({ tableTypeNames });
-    // Generate GraphQL document
     const queryDocument = (0, schema_gql_ast_1.buildCustomQueryString)({
         operation,
         typeRegistry,
         maxDepth,
         skipQueryField,
     });
-    const sourceFile = (0, ts_ast_1.createSourceFile)(project, fileName);
-    // Add file header
-    const headerText = reactQueryEnabled
-        ? `Custom query hook for ${operation.name}`
-        : `Custom query functions for ${operation.name}`;
-    sourceFile.insertText(0, (0, ts_ast_1.createFileHeader)(headerText) + '\n\n');
-    // Generate variables interface if there are arguments (with tracking)
-    let variablesProps = [];
-    if (operation.args.length > 0) {
-        variablesProps = generateVariablesProperties(operation.args, tracker);
-    }
-    // Generate result interface (with tracking)
+    const statements = [];
+    const variablesProps = operation.args.length > 0
+        ? generateVariablesProperties(operation.args, tracker)
+        : [];
     const resultType = (0, type_resolver_1.typeRefToTsType)(operation.returnType, tracker);
-    const resultProps = [
-        { name: operation.name, type: resultType },
-    ];
-    // Get importable types from tracker (separated by source)
-    const schemaTypes = tracker.getImportableTypes(); // From schema-types.ts
-    const tableTypes = tracker.getTableTypes(); // From types.ts
-    // Add imports - conditionally include React Query imports
-    const imports = [];
+    const schemaTypes = tracker.getImportableTypes();
+    const tableTypes = tracker.getTableTypes();
     if (reactQueryEnabled) {
-        imports.push((0, ts_ast_1.createImport)({
-            moduleSpecifier: '@tanstack/react-query',
-            namedImports: ['useQuery'],
-            typeOnlyNamedImports: ['UseQueryOptions', 'QueryClient'],
-        }));
+        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);
     }
-    imports.push((0, ts_ast_1.createImport)({
-        moduleSpecifier: '../client',
-        namedImports: ['execute'],
-        typeOnlyNamedImports: ['ExecuteOptions'],
-    }));
-    // Add types.ts import for table entity types
+    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) {
-        imports.push((0, ts_ast_1.createImport)({
-            moduleSpecifier: '../types',
-            typeOnlyNamedImports: tableTypes,
-        }));
+        const typesImport = t.importDeclaration(tableTypes.map((tt) => t.importSpecifier(t.identifier(tt), t.identifier(tt))), t.stringLiteral('../types'));
+        typesImport.importKind = 'type';
+        statements.push(typesImport);
     }
-    // Add schema-types import for Input/Payload/Enum types
     if (schemaTypes.length > 0) {
-        imports.push((0, ts_ast_1.createImport)({
-            moduleSpecifier: '../schema-types',
-            typeOnlyNamedImports: schemaTypes,
-        }));
+        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);
     }
-    sourceFile.addImportDeclarations(imports);
-    // Add query document constant
-    sourceFile.addVariableStatement((0, ts_ast_1.createConst)(documentConstName, '`\n' + queryDocument + '`', {
-        docs: ['GraphQL query document'],
-    }));
-    // Add variables interface
-    if (operation.args.length > 0) {
-        sourceFile.addInterface((0, ts_ast_1.createInterface)(variablesTypeName, variablesProps));
+    if (useCentralizedKeys) {
+        const queryKeyImport = t.importDeclaration([t.importSpecifier(t.identifier('customQueryKeys'), t.identifier('customQueryKeys'))], t.stringLiteral('../query-keys'));
+        statements.push(queryKeyImport);
     }
-    // Add result interface
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(resultTypeName, resultProps));
-    // Query key factory
+    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) {
-        sourceFile.addVariableStatement((0, ts_ast_1.createConst)(queryKeyName, `(variables?: ${variablesTypeName}) =>
-  ['${operation.name}', variables] as const`, { docs: ['Query key factory for caching'] }));
+        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));
+    }
+    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));
+    if (useCentralizedKeys) {
+        const queryKeyConst = t.variableDeclaration('const', [
+            t.variableDeclarator(t.identifier(queryKeyName), t.memberExpression(t.identifier('customQueryKeys'), t.identifier(operation.name))),
+        ]);
+        const queryKeyExport = t.exportNamedDeclaration(queryKeyConst);
+        (0, babel_ast_1.addJSDocComment)(queryKeyExport, ['Query key factory - re-exported from query-keys.ts']);
+        statements.push(queryKeyExport);
+    }
+    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 {
-        sourceFile.addVariableStatement((0, ts_ast_1.createConst)(queryKeyName, `() => ['${operation.name}'] as const`, {
-            docs: ['Query key factory for caching'],
-        }));
+        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);
     }
-    // Generate hook function (only if React Query is enabled)
     if (reactQueryEnabled) {
-        const hookParams = generateHookParameters(operation, variablesTypeName, resultTypeName);
-        const hookBody = generateHookBody(operation, documentConstName, queryKeyName, variablesTypeName, resultTypeName);
-        const hookDoc = generateHookDoc(operation, hookName);
-        sourceFile.addFunction({
-            name: hookName,
-            isExported: true,
-            parameters: hookParams,
-            statements: hookBody,
-            docs: [{ description: hookDoc }],
-        });
+        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)))));
+            }
+        }
+        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))]))));
+        }
+        useQueryOptions.push(t.spreadElement(t.identifier('options')));
+        hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useQuery'), [t.objectExpression(useQueryOptions)])));
+        const hookParams = [];
+        if (hasArgs) {
+            hookParams.push((0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(variablesTypeName)), !hasRequiredArgs));
+        }
+        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,
+            '',
+            '@example',
+            '```tsx',
+            `const { data, isLoading } = ${exampleCall};`,
+            '',
+            `if (data?.${operation.name}) {`,
+            `  console.log(data.${operation.name});`,
+            '}',
+            '```',
+        ]);
+        statements.push(hookExport);
     }
-    // Add standalone functions section
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Standalone Functions (non-React)');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Generate standalone fetch function
     const fetchFnName = `fetch${(0, utils_1.ucFirst)(operation.name)}Query`;
-    const fetchParams = generateFetchParameters(operation, variablesTypeName);
-    const fetchBody = generateFetchBody(operation, documentConstName, variablesTypeName, resultTypeName);
-    const fetchDoc = generateFetchDoc(operation, fetchFnName);
-    sourceFile.addFunction({
-        name: fetchFnName,
-        isExported: true,
-        isAsync: true,
-        parameters: fetchParams,
-        returnType: `Promise<${resultTypeName}>`,
-        statements: fetchBody,
-        docs: [{ description: fetchDoc }],
-    });
-    // Generate prefetch function (only if React Query is enabled)
-    if (reactQueryEnabled) {
-        const prefetchFnName = `prefetch${(0, utils_1.ucFirst)(operation.name)}Query`;
-        const prefetchParams = generatePrefetchParameters(operation, variablesTypeName);
-        const prefetchBody = generatePrefetchBody(operation, documentConstName, queryKeyName, variablesTypeName, resultTypeName);
-        const prefetchDoc = generatePrefetchDoc(operation, prefetchFnName);
-        sourceFile.addFunction({
-            name: prefetchFnName,
-            isExported: true,
-            isAsync: true,
-            parameters: prefetchParams,
-            returnType: 'Promise<void>',
-            statements: prefetchBody,
-            docs: [{ description: prefetchDoc }],
-        });
-    }
-    return {
-        fileName,
-        content: (0, ts_ast_1.getFormattedOutput)(sourceFile),
-        operationName: operation.name,
-    };
-}
-// ============================================================================
-// Helper functions
-// ============================================================================
-/**
- * Generate interface properties from CleanArguments
- */
-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,
-    }));
-}
-/**
- * Generate hook function parameters
- */
-function generateHookParameters(operation, variablesTypeName, resultTypeName) {
-    const params = [];
-    // Add variables parameter if there are required args
-    const hasRequiredArgs = operation.args.some((arg) => (0, type_resolver_1.isTypeRequired)(arg.type));
-    if (operation.args.length > 0) {
-        params.push({
-            name: 'variables',
-            type: variablesTypeName,
-            hasQuestionToken: !hasRequiredArgs,
-        });
-    }
-    // Add options parameter
-    params.push({
-        name: 'options',
-        type: `Omit<UseQueryOptions<${resultTypeName}, Error>, 'queryKey' | 'queryFn'>`,
-        hasQuestionToken: true,
-    });
-    return params;
-}
-/**
- * Generate hook function body
- */
-function generateHookBody(operation, documentConstName, queryKeyName, variablesTypeName, resultTypeName) {
     const hasArgs = operation.args.length > 0;
     const hasRequiredArgs = operation.args.some((arg) => (0, type_resolver_1.isTypeRequired)(arg.type));
+    const fetchBodyStatements = [];
     if (hasArgs) {
-        // With variables
-        const enabledCondition = hasRequiredArgs
-            ? `enabled: !!variables && (options?.enabled !== false),`
-            : '';
-        return `return useQuery({
-    queryKey: ${queryKeyName}(variables),
-    queryFn: () => execute<${resultTypeName}, ${variablesTypeName}>(
-      ${documentConstName},
-      variables
-    ),
-    ${enabledCondition}
-    ...options,
-  });`;
+        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)),
+        ])));
     }
     else {
-        // No variables
-        return `return useQuery({
-    queryKey: ${queryKeyName}(),
-    queryFn: () => execute<${resultTypeName}>(${documentConstName}),
-    ...options,
-  });`;
+        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))])));
     }
-}
-/**
- * Generate hook JSDoc documentation
- */
-function generateHookDoc(operation, hookName) {
-    const description = operation.description
-        ? operation.description
-        : `Query hook for ${operation.name}`;
-    const hasArgs = operation.args.length > 0;
-    let example;
+    const fetchParams = [];
     if (hasArgs) {
-        const argNames = operation.args.map((a) => a.name).join(', ');
-        example = `
-@example
-\`\`\`tsx
-const { data, isLoading } = ${hookName}({ ${argNames} });
-
-if (data?.${operation.name}) {
-  console.log(data.${operation.name});
-}
-\`\`\``;
-    }
-    else {
-        example = `
-@example
-\`\`\`tsx
-const { data, isLoading } = ${hookName}();
-
-if (data?.${operation.name}) {
-  console.log(data.${operation.name});
-}
-\`\`\``;
+        fetchParams.push((0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(variablesTypeName)), !hasRequiredArgs));
     }
-    return description + '\n' + example;
-}
-// ============================================================================
-// Standalone function generators
-// ============================================================================
-/**
- * Generate fetch function parameters
- */
-function generateFetchParameters(operation, variablesTypeName) {
-    const params = [];
-    if (operation.args.length > 0) {
-        const hasRequiredArgs = operation.args.some((arg) => (0, type_resolver_1.isTypeRequired)(arg.type));
-        params.push({
-            name: 'variables',
-            type: variablesTypeName,
-            hasQuestionToken: !hasRequiredArgs,
-        });
-    }
-    params.push({
-        name: 'options',
-        type: 'ExecuteOptions',
-        hasQuestionToken: true,
-    });
-    return params;
-}
-/**
- * Generate fetch function body
- */
-function generateFetchBody(operation, documentConstName, variablesTypeName, resultTypeName) {
-    if (operation.args.length > 0) {
-        return `return execute<${resultTypeName}, ${variablesTypeName}>(
-    ${documentConstName},
-    variables,
-    options
-  );`;
-    }
-    else {
-        return `return execute<${resultTypeName}>(${documentConstName}, undefined, options);`;
-    }
-}
-/**
- * Generate fetch function documentation
- */
-function generateFetchDoc(operation, fnName) {
-    const description = `Fetch ${operation.name} without React hooks`;
-    if (operation.args.length > 0) {
-        const argNames = operation.args.map((a) => a.name).join(', ');
-        return `${description}
-
-@example
-\`\`\`ts
-const data = await ${fnName}({ ${argNames} });
-\`\`\``;
-    }
-    else {
-        return `${description}
-
-@example
-\`\`\`ts
-const data = await ${fnName}();
-\`\`\``;
-    }
-}
-/**
- * Generate prefetch function parameters
- */
-function generatePrefetchParameters(operation, variablesTypeName) {
-    const params = [
-        { name: 'queryClient', type: 'QueryClient' },
-    ];
-    if (operation.args.length > 0) {
-        const hasRequiredArgs = operation.args.some((arg) => (0, type_resolver_1.isTypeRequired)(arg.type));
-        params.push({
-            name: 'variables',
-            type: variablesTypeName,
-            hasQuestionToken: !hasRequiredArgs,
-        });
-    }
-    params.push({
-        name: 'options',
-        type: 'ExecuteOptions',
-        hasQuestionToken: true,
-    });
-    return params;
-}
-/**
- * Generate prefetch function body
- */
-function generatePrefetchBody(operation, documentConstName, queryKeyName, variablesTypeName, resultTypeName) {
-    if (operation.args.length > 0) {
-        return `await queryClient.prefetchQuery({
-    queryKey: ${queryKeyName}(variables),
-    queryFn: () => execute<${resultTypeName}, ${variablesTypeName}>(
-      ${documentConstName},
-      variables,
-      options
-    ),
-  });`;
-    }
-    else {
-        return `await queryClient.prefetchQuery({
-    queryKey: ${queryKeyName}(),
-    queryFn: () => execute<${resultTypeName}>(${documentConstName}, undefined, options),
-  });`;
-    }
-}
-/**
- * Generate prefetch function documentation
- */
-function generatePrefetchDoc(operation, fnName) {
-    const description = `Prefetch ${operation.name} for SSR or cache warming`;
-    if (operation.args.length > 0) {
-        const argNames = operation.args.map((a) => a.name).join(', ');
-        return `${description}
-
-@example
-\`\`\`ts
-await ${fnName}(queryClient, { ${argNames} });
-\`\`\``;
-    }
-    else {
-        return `${description}
-
-@example
-\`\`\`ts
-await ${fnName}(queryClient);
-\`\`\``;
+    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);
+    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)),
+            ]))));
+        }
+        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));
+        }
+        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;
+    return {
+        fileName,
+        content,
+        operationName: operation.name,
+    };
 }
-/**
- * Generate all custom query hook files
- */
 function generateAllCustomQueryHooks(options) {
-    const { operations, typeRegistry, maxDepth = 2, skipQueryField = true, reactQueryEnabled = true, tableTypeNames } = options;
+    const { operations, typeRegistry, maxDepth = 2, skipQueryField = true, reactQueryEnabled = true, tableTypeNames, useCentralizedKeys = true, } = options;
     return operations
         .filter((op) => op.kind === 'query')
         .map((operation) => generateCustomQueryHook({
@@ -389,5 +289,6 @@ function generateAllCustomQueryHooks(options) {
         skipQueryField,
         reactQueryEnabled,
         tableTypeNames,
+        useCentralizedKeys,
     }));
 }