@constructive-io/graphql-codegen 2.23.2 → 2.24.0

package/cli/codegen/queries.js

@@ -1,36 +1,84 @@
 "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.generateListQueryHook = generateListQueryHook;
 exports.generateSingleQueryHook = generateSingleQueryHook;
 exports.generateAllQueryHooks = generateAllQueryHooks;
-const ts_ast_1 = require("./ts-ast");
+const t = __importStar(require("@babel/types"));
+const babel_ast_1 = require("./babel-ast");
 const gql_ast_1 = require("./gql-ast");
 const utils_1 = require("./utils");
-// ============================================================================
-// List query hook generator
-// ============================================================================
-/**
- * Generate list query hook file content using AST
- */
+function createUnionType(values) {
+    return t.tsUnionType(values.map((v) => t.tsLiteralType(t.stringLiteral(v))));
+}
+function createFilterInterfaceDeclaration(name, fieldFilters, isExported = true) {
+    const properties = [];
+    for (const filter of fieldFilters) {
+        const prop = t.tsPropertySignature(t.identifier(filter.fieldName), t.tsTypeAnnotation(t.tsTypeReference(t.identifier(filter.filterType))));
+        prop.optional = true;
+        properties.push(prop);
+    }
+    const andProp = t.tsPropertySignature(t.identifier('and'), t.tsTypeAnnotation(t.tsArrayType(t.tsTypeReference(t.identifier(name)))));
+    andProp.optional = true;
+    properties.push(andProp);
+    const orProp = t.tsPropertySignature(t.identifier('or'), t.tsTypeAnnotation(t.tsArrayType(t.tsTypeReference(t.identifier(name)))));
+    orProp.optional = true;
+    properties.push(orProp);
+    const notProp = t.tsPropertySignature(t.identifier('not'), t.tsTypeAnnotation(t.tsTypeReference(t.identifier(name))));
+    notProp.optional = true;
+    properties.push(notProp);
+    const body = t.tsInterfaceBody(properties);
+    const interfaceDecl = t.tsInterfaceDeclaration(t.identifier(name), null, null, body);
+    if (isExported) {
+        return t.exportNamedDeclaration(interfaceDecl);
+    }
+    return interfaceDecl;
+}
 function generateListQueryHook(table, options = {}) {
-    const { reactQueryEnabled = true } = options;
-    const project = (0, ts_ast_1.createProject)();
+    const { reactQueryEnabled = true, useCentralizedKeys = true, hasRelationships = false, } = options;
     const { typeName, pluralName } = (0, utils_1.getTableNames)(table);
     const hookName = (0, utils_1.getListQueryHookName)(table);
     const queryName = (0, utils_1.getAllRowsQueryName)(table);
     const filterTypeName = (0, utils_1.getFilterTypeName)(table);
     const orderByTypeName = (0, utils_1.getOrderByTypeName)(table);
     const scalarFields = (0, utils_1.getScalarFields)(table);
-    // Generate GraphQL document via AST
+    const keysName = `${(0, utils_1.lcFirst)(typeName)}Keys`;
+    const scopeTypeName = `${typeName}Scope`;
     const queryAST = (0, gql_ast_1.buildListQueryAST)({ table });
     const queryDocument = (0, gql_ast_1.printGraphQL)(queryAST);
-    const sourceFile = (0, ts_ast_1.createSourceFile)(project, (0, utils_1.getListQueryFileName)(table));
-    // Add file header as leading comment
-    const headerText = reactQueryEnabled
-        ? `List query hook for ${typeName}`
-        : `List query functions for ${typeName}`;
-    sourceFile.insertText(0, (0, ts_ast_1.createFileHeader)(headerText) + '\n\n');
-    // Collect all filter types used by this table's fields
+    const statements = [];
     const filterTypesUsed = new Set();
     for (const field of scalarFields) {
         const filterType = (0, utils_1.getScalarFilterType)(field.type.gqlType, field.type.isArray);
@@ -38,47 +86,56 @@ function generateListQueryHook(table, options = {}) {
             filterTypesUsed.add(filterType);
         }
     }
-    // Add imports - conditionally include React Query imports
-    const imports = [];
     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'],
-    }), (0, ts_ast_1.createImport)({
-        moduleSpecifier: '../types',
-        typeOnlyNamedImports: [typeName, ...Array.from(filterTypesUsed)],
-    }));
-    sourceFile.addImportDeclarations(imports);
-    // Re-export entity type
-    sourceFile.addStatements(`\n// Re-export entity type for convenience\nexport type { ${typeName} };\n`);
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// GraphQL Document');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Add query document constant
-    sourceFile.addVariableStatement((0, ts_ast_1.createConst)(`${queryName}QueryDocument`, '`\n' + queryDocument + '`'));
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Types');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Generate filter interface
+    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);
+    const typesImport = t.importDeclaration([
+        t.importSpecifier(t.identifier(typeName), t.identifier(typeName)),
+        ...Array.from(filterTypesUsed).map((ft) => t.importSpecifier(t.identifier(ft), t.identifier(ft))),
+    ], t.stringLiteral('../types'));
+    typesImport.importKind = 'type';
+    statements.push(typesImport);
+    if (useCentralizedKeys) {
+        const queryKeyImport = t.importDeclaration([t.importSpecifier(t.identifier(keysName), t.identifier(keysName))], t.stringLiteral('../query-keys'));
+        statements.push(queryKeyImport);
+        if (hasRelationships) {
+            const scopeTypeImport = t.importDeclaration([
+                t.importSpecifier(t.identifier(scopeTypeName), t.identifier(scopeTypeName)),
+            ], t.stringLiteral('../query-keys'));
+            scopeTypeImport.importKind = 'type';
+            statements.push(scopeTypeImport);
+        }
+    }
+    const reExportDecl = t.exportNamedDeclaration(null, [t.exportSpecifier(t.identifier(typeName), t.identifier(typeName))], t.stringLiteral('../types'));
+    reExportDecl.exportKind = 'type';
+    statements.push(reExportDecl);
+    const queryDocConst = t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier(`${queryName}QueryDocument`), t.templateLiteral([
+            t.templateElement({ raw: '\n' + queryDocument, cooked: '\n' + queryDocument }, true),
+        ], [])),
+    ]);
+    statements.push(t.exportNamedDeclaration(queryDocConst));
     const fieldFilters = scalarFields
         .map((field) => {
         const filterType = (0, utils_1.getScalarFilterType)(field.type.gqlType, field.type.isArray);
         return filterType ? { fieldName: field.name, filterType } : null;
     })
         .filter((f) => f !== null);
-    // Note: Not exported to avoid conflicts with schema-types
-    sourceFile.addInterface((0, ts_ast_1.createFilterInterface)(filterTypeName, fieldFilters, { isExported: false }));
-    // Generate OrderBy type
-    // Note: Not exported to avoid conflicts with schema-types
+    statements.push(createFilterInterfaceDeclaration(filterTypeName, fieldFilters, false));
     const orderByValues = [
         ...scalarFields.flatMap((f) => [
             `${(0, utils_1.toScreamingSnake)(f.name)}_ASC`,
@@ -88,373 +145,509 @@ function generateListQueryHook(table, options = {}) {
         'PRIMARY_KEY_ASC',
         'PRIMARY_KEY_DESC',
     ];
-    sourceFile.addTypeAlias((0, ts_ast_1.createTypeAlias)(orderByTypeName, (0, ts_ast_1.createUnionType)(orderByValues), { isExported: false }));
-    // Variables interface
-    const variablesProps = [
-        { name: 'first', type: 'number', optional: true },
-        { name: 'offset', type: 'number', optional: true },
-        { name: 'filter', type: filterTypeName, optional: true },
-        { name: 'orderBy', type: `${orderByTypeName}[]`, optional: true },
-    ];
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${(0, utils_1.ucFirst)(pluralName)}QueryVariables`, variablesProps));
-    // Result interface
-    const resultProps = [
-        {
-            name: queryName,
-            type: `{
-    totalCount: number;
-    nodes: ${typeName}[];
-    pageInfo: {
-      hasNextPage: boolean;
-      hasPreviousPage: boolean;
-      startCursor: string | null;
-      endCursor: string | null;
-    };
-  }`,
-        },
-    ];
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${(0, utils_1.ucFirst)(pluralName)}QueryResult`, resultProps));
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Query Key');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Query key factory
-    sourceFile.addVariableStatement((0, ts_ast_1.createConst)(`${queryName}QueryKey`, `(variables?: ${(0, utils_1.ucFirst)(pluralName)}QueryVariables) =>
-  ['${typeName.toLowerCase()}', 'list', variables] as const`));
-    // Add React Query hook section (only if enabled)
+    const orderByTypeAlias = t.tsTypeAliasDeclaration(t.identifier(orderByTypeName), null, createUnionType(orderByValues));
+    statements.push(orderByTypeAlias);
+    const variablesInterfaceBody = t.tsInterfaceBody([
+        (() => {
+            const p = t.tsPropertySignature(t.identifier('first'), t.tsTypeAnnotation(t.tsNumberKeyword()));
+            p.optional = true;
+            return p;
+        })(),
+        (() => {
+            const p = t.tsPropertySignature(t.identifier('offset'), t.tsTypeAnnotation(t.tsNumberKeyword()));
+            p.optional = true;
+            return p;
+        })(),
+        (() => {
+            const p = t.tsPropertySignature(t.identifier('filter'), t.tsTypeAnnotation(t.tsTypeReference(t.identifier(filterTypeName))));
+            p.optional = true;
+            return p;
+        })(),
+        (() => {
+            const p = t.tsPropertySignature(t.identifier('orderBy'), t.tsTypeAnnotation(t.tsArrayType(t.tsTypeReference(t.identifier(orderByTypeName)))));
+            p.optional = true;
+            return p;
+        })(),
+    ]);
+    const variablesInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(pluralName)}QueryVariables`), null, null, variablesInterfaceBody);
+    statements.push(t.exportNamedDeclaration(variablesInterface));
+    const pageInfoType = t.tsTypeLiteral([
+        t.tsPropertySignature(t.identifier('hasNextPage'), t.tsTypeAnnotation(t.tsBooleanKeyword())),
+        t.tsPropertySignature(t.identifier('hasPreviousPage'), t.tsTypeAnnotation(t.tsBooleanKeyword())),
+        t.tsPropertySignature(t.identifier('startCursor'), t.tsTypeAnnotation(t.tsUnionType([t.tsStringKeyword(), t.tsNullKeyword()]))),
+        t.tsPropertySignature(t.identifier('endCursor'), t.tsTypeAnnotation(t.tsUnionType([t.tsStringKeyword(), t.tsNullKeyword()]))),
+    ]);
+    const resultType = t.tsTypeLiteral([
+        t.tsPropertySignature(t.identifier('totalCount'), t.tsTypeAnnotation(t.tsNumberKeyword())),
+        t.tsPropertySignature(t.identifier('nodes'), t.tsTypeAnnotation(t.tsArrayType(t.tsTypeReference(t.identifier(typeName))))),
+        t.tsPropertySignature(t.identifier('pageInfo'), t.tsTypeAnnotation(pageInfoType)),
+    ]);
+    const resultInterfaceBody = t.tsInterfaceBody([
+        t.tsPropertySignature(t.identifier(queryName), t.tsTypeAnnotation(resultType)),
+    ]);
+    const resultInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(pluralName)}QueryResult`), null, null, resultInterfaceBody);
+    statements.push(t.exportNamedDeclaration(resultInterface));
+    if (useCentralizedKeys) {
+        const queryKeyConst = t.variableDeclaration('const', [
+            t.variableDeclarator(t.identifier(`${queryName}QueryKey`), t.memberExpression(t.identifier(keysName), t.identifier('list'))),
+        ]);
+        const queryKeyExport = t.exportNamedDeclaration(queryKeyConst);
+        (0, babel_ast_1.addJSDocComment)(queryKeyExport, [
+            'Query key factory - re-exported from query-keys.ts',
+        ]);
+        statements.push(queryKeyExport);
+    }
+    else {
+        const queryKeyArrow = t.arrowFunctionExpression([
+            (0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(pluralName)}QueryVariables`)), true),
+        ], t.tsAsExpression(t.arrayExpression([
+            t.stringLiteral(typeName.toLowerCase()),
+            t.stringLiteral('list'),
+            t.identifier('variables'),
+        ]), t.tsTypeReference(t.identifier('const'))));
+        const queryKeyConst = t.variableDeclaration('const', [
+            t.variableDeclarator(t.identifier(`${queryName}QueryKey`), queryKeyArrow),
+        ]);
+        statements.push(t.exportNamedDeclaration(queryKeyConst));
+    }
     if (reactQueryEnabled) {
-        sourceFile.addStatements('\n// ============================================================================');
-        sourceFile.addStatements('// Hook');
-        sourceFile.addStatements('// ============================================================================\n');
-        // Hook function
-        sourceFile.addFunction({
-            name: hookName,
-            isExported: true,
-            parameters: [
-                {
-                    name: 'variables',
-                    type: `${(0, utils_1.ucFirst)(pluralName)}QueryVariables`,
-                    hasQuestionToken: true,
-                },
-                {
-                    name: 'options',
-                    type: `Omit<UseQueryOptions<${(0, utils_1.ucFirst)(pluralName)}QueryResult, Error>, 'queryKey' | 'queryFn'>`,
-                    hasQuestionToken: true,
-                },
-            ],
-            statements: `return useQuery({
-    queryKey: ${queryName}QueryKey(variables),
-    queryFn: () => execute<${(0, utils_1.ucFirst)(pluralName)}QueryResult, ${(0, utils_1.ucFirst)(pluralName)}QueryVariables>(
-      ${queryName}QueryDocument,
-      variables
-    ),
-    ...options,
-  });`,
-            docs: [
-                {
-                    description: `Query hook for fetching ${typeName} list
-
-@example
-\`\`\`tsx
-const { data, isLoading } = ${hookName}({
-  first: 10,
-  filter: { name: { equalTo: "example" } },
-  orderBy: ['CREATED_AT_DESC'],
-});
-\`\`\``,
-                },
-            ],
-        });
+        const hookBodyStatements = [];
+        if (hasRelationships && useCentralizedKeys) {
+            hookBodyStatements.push(t.variableDeclaration('const', [
+                t.variableDeclarator(t.objectPattern([
+                    t.objectProperty(t.identifier('scope'), t.identifier('scope'), false, true),
+                    t.restElement(t.identifier('queryOptions')),
+                ]), t.logicalExpression('??', t.identifier('options'), t.objectExpression([]))),
+            ]));
+            hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useQuery'), [
+                t.objectExpression([
+                    t.objectProperty(t.identifier('queryKey'), t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('list')), [t.identifier('variables'), t.identifier('scope')])),
+                    t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], t.callExpression(t.identifier('execute'), [
+                        t.identifier(`${queryName}QueryDocument`),
+                        t.identifier('variables'),
+                    ]))),
+                    t.spreadElement(t.identifier('queryOptions')),
+                ]),
+            ])));
+        }
+        else if (useCentralizedKeys) {
+            hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useQuery'), [
+                t.objectExpression([
+                    t.objectProperty(t.identifier('queryKey'), t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('list')), [t.identifier('variables')])),
+                    t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], t.callExpression(t.identifier('execute'), [
+                        t.identifier(`${queryName}QueryDocument`),
+                        t.identifier('variables'),
+                    ]))),
+                    t.spreadElement(t.identifier('options')),
+                ]),
+            ])));
+        }
+        else {
+            hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useQuery'), [
+                t.objectExpression([
+                    t.objectProperty(t.identifier('queryKey'), t.callExpression(t.identifier(`${queryName}QueryKey`), [
+                        t.identifier('variables'),
+                    ])),
+                    t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], t.callExpression(t.identifier('execute'), [
+                        t.identifier(`${queryName}QueryDocument`),
+                        t.identifier('variables'),
+                    ]))),
+                    t.spreadElement(t.identifier('options')),
+                ]),
+            ])));
+        }
+        const hookParams = [
+            (0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(pluralName)}QueryVariables`)), true),
+        ];
+        let optionsTypeStr;
+        if (hasRelationships && useCentralizedKeys) {
+            optionsTypeStr = `Omit<UseQueryOptions<${(0, utils_1.ucFirst)(pluralName)}QueryResult, Error>, 'queryKey' | 'queryFn'> & { scope?: ${scopeTypeName} }`;
+        }
+        else {
+            optionsTypeStr = `Omit<UseQueryOptions<${(0, utils_1.ucFirst)(pluralName)}QueryResult, 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 docLines = [
+            `Query hook for fetching ${typeName} list`,
+            '',
+            '@example',
+            '```tsx',
+            `const { data, isLoading } = ${hookName}({`,
+            '  first: 10,',
+            '  filter: { name: { equalTo: "example" } },',
+            "  orderBy: ['CREATED_AT_DESC'],",
+            '});',
+            '```',
+        ];
+        if (hasRelationships && useCentralizedKeys) {
+            docLines.push('');
+            docLines.push('@example With scope for hierarchical cache invalidation');
+            docLines.push('```tsx');
+            docLines.push(`const { data } = ${hookName}(`);
+            docLines.push('  { first: 10 },');
+            docLines.push("  { scope: { parentId: 'parent-id' } }");
+            docLines.push(');');
+            docLines.push('```');
+        }
+        (0, babel_ast_1.addJSDocComment)(hookExport, docLines);
+        statements.push(hookExport);
     }
-    // Add section comment for standalone functions
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Standalone Functions (non-React)');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Fetch function (standalone, no React)
-    sourceFile.addFunction({
-        name: `fetch${(0, utils_1.ucFirst)(pluralName)}Query`,
-        isExported: true,
-        isAsync: true,
-        parameters: [
-            {
-                name: 'variables',
-                type: `${(0, utils_1.ucFirst)(pluralName)}QueryVariables`,
-                hasQuestionToken: true,
-            },
-            {
-                name: 'options',
-                type: 'ExecuteOptions',
-                hasQuestionToken: true,
-            },
-        ],
-        returnType: `Promise<${(0, utils_1.ucFirst)(pluralName)}QueryResult>`,
-        statements: `return execute<${(0, utils_1.ucFirst)(pluralName)}QueryResult, ${(0, utils_1.ucFirst)(pluralName)}QueryVariables>(
-    ${queryName}QueryDocument,
-    variables,
-    options
-  );`,
-        docs: [
-            {
-                description: `Fetch ${typeName} list without React hooks
-
-@example
-\`\`\`ts
-// Direct fetch
-const data = await fetch${(0, utils_1.ucFirst)(pluralName)}Query({ first: 10 });
-
-// With QueryClient
-const data = await queryClient.fetchQuery({
-  queryKey: ${queryName}QueryKey(variables),
-  queryFn: () => fetch${(0, utils_1.ucFirst)(pluralName)}Query(variables),
-});
-\`\`\``,
-            },
-        ],
-    });
-    // Prefetch function (for SSR/QueryClient) - only if React Query is enabled
+    const fetchFuncBody = t.blockStatement([
+        t.returnStatement(t.callExpression(t.identifier('execute'), [
+            t.identifier(`${queryName}QueryDocument`),
+            t.identifier('variables'),
+            t.identifier('options'),
+        ])),
+    ]);
+    const fetchFunc = t.functionDeclaration(t.identifier(`fetch${(0, utils_1.ucFirst)(pluralName)}Query`), [
+        (0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(pluralName)}QueryVariables`)), true),
+        (0, babel_ast_1.typedParam)('options', t.tsTypeReference(t.identifier('ExecuteOptions')), true),
+    ], fetchFuncBody);
+    fetchFunc.async = true;
+    fetchFunc.returnType = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Promise'), t.tsTypeParameterInstantiation([
+        t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(pluralName)}QueryResult`)),
+    ])));
+    const fetchExport = t.exportNamedDeclaration(fetchFunc);
+    (0, babel_ast_1.addJSDocComment)(fetchExport, [
+        `Fetch ${typeName} list without React hooks`,
+        '',
+        '@example',
+        '```ts',
+        '// Direct fetch',
+        `const data = await fetch${(0, utils_1.ucFirst)(pluralName)}Query({ first: 10 });`,
+        '',
+        '// With QueryClient',
+        'const data = await queryClient.fetchQuery({',
+        `  queryKey: ${queryName}QueryKey(variables),`,
+        `  queryFn: () => fetch${(0, utils_1.ucFirst)(pluralName)}Query(variables),`,
+        '});',
+        '```',
+    ]);
+    statements.push(fetchExport);
     if (reactQueryEnabled) {
-        sourceFile.addFunction({
-            name: `prefetch${(0, utils_1.ucFirst)(pluralName)}Query`,
-            isExported: true,
-            isAsync: true,
-            parameters: [
-                {
-                    name: 'queryClient',
-                    type: 'QueryClient',
-                },
-                {
-                    name: 'variables',
-                    type: `${(0, utils_1.ucFirst)(pluralName)}QueryVariables`,
-                    hasQuestionToken: true,
-                },
-                {
-                    name: 'options',
-                    type: 'ExecuteOptions',
-                    hasQuestionToken: true,
-                },
-            ],
-            returnType: 'Promise<void>',
-            statements: `await queryClient.prefetchQuery({
-    queryKey: ${queryName}QueryKey(variables),
-    queryFn: () => execute<${(0, utils_1.ucFirst)(pluralName)}QueryResult, ${(0, utils_1.ucFirst)(pluralName)}QueryVariables>(
-      ${queryName}QueryDocument,
-      variables,
-      options
-    ),
-  });`,
-            docs: [
-                {
-                    description: `Prefetch ${typeName} list for SSR or cache warming
-
-@example
-\`\`\`ts
-await prefetch${(0, utils_1.ucFirst)(pluralName)}Query(queryClient, { first: 10 });
-\`\`\``,
-                },
-            ],
-        });
+        const prefetchParams = [
+            (0, babel_ast_1.typedParam)('queryClient', t.tsTypeReference(t.identifier('QueryClient'))),
+            (0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(pluralName)}QueryVariables`)), true),
+        ];
+        if (hasRelationships && useCentralizedKeys) {
+            prefetchParams.push((0, babel_ast_1.typedParam)('scope', t.tsTypeReference(t.identifier(scopeTypeName)), true));
+        }
+        prefetchParams.push((0, babel_ast_1.typedParam)('options', t.tsTypeReference(t.identifier('ExecuteOptions')), true));
+        let prefetchQueryKeyExpr;
+        if (hasRelationships && useCentralizedKeys) {
+            prefetchQueryKeyExpr = t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('list')), [t.identifier('variables'), t.identifier('scope')]);
+        }
+        else if (useCentralizedKeys) {
+            prefetchQueryKeyExpr = t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('list')), [t.identifier('variables')]);
+        }
+        else {
+            prefetchQueryKeyExpr = t.callExpression(t.identifier(`${queryName}QueryKey`), [t.identifier('variables')]);
+        }
+        const prefetchFuncBody = t.blockStatement([
+            t.expressionStatement(t.awaitExpression(t.callExpression(t.memberExpression(t.identifier('queryClient'), t.identifier('prefetchQuery')), [
+                t.objectExpression([
+                    t.objectProperty(t.identifier('queryKey'), prefetchQueryKeyExpr),
+                    t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], t.callExpression(t.identifier('execute'), [
+                        t.identifier(`${queryName}QueryDocument`),
+                        t.identifier('variables'),
+                        t.identifier('options'),
+                    ]))),
+                ]),
+            ]))),
+        ]);
+        const prefetchFunc = t.functionDeclaration(t.identifier(`prefetch${(0, utils_1.ucFirst)(pluralName)}Query`), prefetchParams, prefetchFuncBody);
+        prefetchFunc.async = true;
+        prefetchFunc.returnType = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Promise'), t.tsTypeParameterInstantiation([t.tsVoidKeyword()])));
+        const prefetchExport = t.exportNamedDeclaration(prefetchFunc);
+        (0, babel_ast_1.addJSDocComment)(prefetchExport, [
+            `Prefetch ${typeName} list for SSR or cache warming`,
+            '',
+            '@example',
+            '```ts',
+            `await prefetch${(0, utils_1.ucFirst)(pluralName)}Query(queryClient, { first: 10 });`,
+            '```',
+        ]);
+        statements.push(prefetchExport);
     }
+    const code = (0, babel_ast_1.generateCode)(statements);
+    const headerText = reactQueryEnabled
+        ? `List query hook for ${typeName}`
+        : `List query functions for ${typeName}`;
+    const content = (0, utils_1.getGeneratedFileHeader)(headerText) + '\n\n' + code;
     return {
         fileName: (0, utils_1.getListQueryFileName)(table),
-        content: (0, ts_ast_1.getFormattedOutput)(sourceFile),
+        content,
     };
 }
-// ============================================================================
-// Single item query hook generator
-// ============================================================================
-/**
- * Generate single item query hook file content using AST
- */
 function generateSingleQueryHook(table, options = {}) {
-    const { reactQueryEnabled = true } = options;
-    const project = (0, ts_ast_1.createProject)();
+    const { reactQueryEnabled = true, useCentralizedKeys = true, hasRelationships = false, } = options;
     const { typeName, singularName } = (0, utils_1.getTableNames)(table);
     const hookName = (0, utils_1.getSingleQueryHookName)(table);
     const queryName = (0, utils_1.getSingleRowQueryName)(table);
-    // Get primary key info dynamically from table constraints
+    const keysName = `${(0, utils_1.lcFirst)(typeName)}Keys`;
+    const scopeTypeName = `${typeName}Scope`;
     const pkFields = (0, utils_1.getPrimaryKeyInfo)(table);
-    // For simplicity, use first PK field (most common case)
-    // Composite PKs would need more complex handling
     const pkField = pkFields[0];
     const pkName = pkField.name;
     const pkTsType = pkField.tsType;
-    // Generate GraphQL document via AST
     const queryAST = (0, gql_ast_1.buildSingleQueryAST)({ table });
     const queryDocument = (0, gql_ast_1.printGraphQL)(queryAST);
-    const sourceFile = (0, ts_ast_1.createSourceFile)(project, (0, utils_1.getSingleQueryFileName)(table));
-    // Add file header
-    const headerText = reactQueryEnabled
-        ? `Single item query hook for ${typeName}`
-        : `Single item query functions for ${typeName}`;
-    sourceFile.insertText(0, (0, ts_ast_1.createFileHeader)(headerText) + '\n\n');
-    // Add imports - conditionally include React Query imports
-    const imports = [];
+    const statements = [];
     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);
+    }
+    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);
+    const typesImport = t.importDeclaration([t.importSpecifier(t.identifier(typeName), t.identifier(typeName))], t.stringLiteral('../types'));
+    typesImport.importKind = 'type';
+    statements.push(typesImport);
+    if (useCentralizedKeys) {
+        const queryKeyImport = t.importDeclaration([t.importSpecifier(t.identifier(keysName), t.identifier(keysName))], t.stringLiteral('../query-keys'));
+        statements.push(queryKeyImport);
+        if (hasRelationships) {
+            const scopeTypeImport = t.importDeclaration([
+                t.importSpecifier(t.identifier(scopeTypeName), t.identifier(scopeTypeName)),
+            ], t.stringLiteral('../query-keys'));
+            scopeTypeImport.importKind = 'type';
+            statements.push(scopeTypeImport);
+        }
+    }
+    const reExportDecl = t.exportNamedDeclaration(null, [t.exportSpecifier(t.identifier(typeName), t.identifier(typeName))], t.stringLiteral('../types'));
+    reExportDecl.exportKind = 'type';
+    statements.push(reExportDecl);
+    const queryDocConst = t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier(`${queryName}QueryDocument`), t.templateLiteral([
+            t.templateElement({ raw: '\n' + queryDocument, cooked: '\n' + queryDocument }, true),
+        ], [])),
+    ]);
+    statements.push(t.exportNamedDeclaration(queryDocConst));
+    const pkTypeAnnotation = pkTsType === 'string'
+        ? t.tsStringKeyword()
+        : pkTsType === 'number'
+            ? t.tsNumberKeyword()
+            : t.tsTypeReference(t.identifier(pkTsType));
+    const variablesInterfaceBody = t.tsInterfaceBody([
+        t.tsPropertySignature(t.identifier(pkName), t.tsTypeAnnotation(pkTypeAnnotation)),
+    ]);
+    const variablesInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(singularName)}QueryVariables`), null, null, variablesInterfaceBody);
+    statements.push(t.exportNamedDeclaration(variablesInterface));
+    const resultInterfaceBody = t.tsInterfaceBody([
+        t.tsPropertySignature(t.identifier(queryName), t.tsTypeAnnotation(t.tsUnionType([
+            t.tsTypeReference(t.identifier(typeName)),
+            t.tsNullKeyword(),
+        ]))),
+    ]);
+    const resultInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(singularName)}QueryResult`), null, null, resultInterfaceBody);
+    statements.push(t.exportNamedDeclaration(resultInterface));
+    if (useCentralizedKeys) {
+        const queryKeyConst = t.variableDeclaration('const', [
+            t.variableDeclarator(t.identifier(`${queryName}QueryKey`), t.memberExpression(t.identifier(keysName), t.identifier('detail'))),
+        ]);
+        const queryKeyExport = t.exportNamedDeclaration(queryKeyConst);
+        (0, babel_ast_1.addJSDocComment)(queryKeyExport, [
+            'Query key factory - re-exported from query-keys.ts',
+        ]);
+        statements.push(queryKeyExport);
+    }
+    else {
+        const queryKeyArrow = t.arrowFunctionExpression([(0, babel_ast_1.typedParam)(pkName, pkTypeAnnotation)], t.tsAsExpression(t.arrayExpression([
+            t.stringLiteral(typeName.toLowerCase()),
+            t.stringLiteral('detail'),
+            t.identifier(pkName),
+        ]), t.tsTypeReference(t.identifier('const'))));
+        const queryKeyConst = t.variableDeclaration('const', [
+            t.variableDeclarator(t.identifier(`${queryName}QueryKey`), queryKeyArrow),
+        ]);
+        statements.push(t.exportNamedDeclaration(queryKeyConst));
     }
-    imports.push((0, ts_ast_1.createImport)({
-        moduleSpecifier: '../client',
-        namedImports: ['execute'],
-        typeOnlyNamedImports: ['ExecuteOptions'],
-    }), (0, ts_ast_1.createImport)({
-        moduleSpecifier: '../types',
-        typeOnlyNamedImports: [typeName],
-    }));
-    sourceFile.addImportDeclarations(imports);
-    // Re-export entity type
-    sourceFile.addStatements(`\n// Re-export entity type for convenience\nexport type { ${typeName} };\n`);
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// GraphQL Document');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Add query document constant
-    sourceFile.addVariableStatement((0, ts_ast_1.createConst)(`${queryName}QueryDocument`, '`\n' + queryDocument + '`'));
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Types');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Variables interface - use dynamic PK field name and type
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${(0, utils_1.ucFirst)(singularName)}QueryVariables`, [
-        { name: pkName, type: pkTsType },
-    ]));
-    // Result interface
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${(0, utils_1.ucFirst)(singularName)}QueryResult`, [
-        { name: queryName, type: `${typeName} | null` },
-    ]));
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Query Key');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Query key factory - use dynamic PK field name and type
-    sourceFile.addVariableStatement((0, ts_ast_1.createConst)(`${queryName}QueryKey`, `(${pkName}: ${pkTsType}) =>
-  ['${typeName.toLowerCase()}', 'detail', ${pkName}] as const`));
-    // Add React Query hook section (only if enabled)
     if (reactQueryEnabled) {
-        sourceFile.addStatements('\n// ============================================================================');
-        sourceFile.addStatements('// Hook');
-        sourceFile.addStatements('// ============================================================================\n');
-        // Hook function - use dynamic PK field name and type
-        sourceFile.addFunction({
-            name: hookName,
-            isExported: true,
-            parameters: [
-                { name: pkName, type: pkTsType },
-                {
-                    name: 'options',
-                    type: `Omit<UseQueryOptions<${(0, utils_1.ucFirst)(singularName)}QueryResult, Error>, 'queryKey' | 'queryFn'>`,
-                    hasQuestionToken: true,
-                },
-            ],
-            statements: `return useQuery({
-    queryKey: ${queryName}QueryKey(${pkName}),
-    queryFn: () => execute<${(0, utils_1.ucFirst)(singularName)}QueryResult, ${(0, utils_1.ucFirst)(singularName)}QueryVariables>(
-      ${queryName}QueryDocument,
-      { ${pkName} }
-    ),
-    enabled: !!${pkName} && (options?.enabled !== false),
-    ...options,
-  });`,
-            docs: [
-                {
-                    description: `Query hook for fetching a single ${typeName} by primary key
-
-@example
-\`\`\`tsx
-const { data, isLoading } = ${hookName}(${pkTsType === 'string' ? "'value-here'" : '123'});
-
-if (data?.${queryName}) {
-  console.log(data.${queryName}.${pkName});
-}
-\`\`\``,
-                },
-            ],
-        });
+        const hookBodyStatements = [];
+        if (hasRelationships && useCentralizedKeys) {
+            hookBodyStatements.push(t.variableDeclaration('const', [
+                t.variableDeclarator(t.objectPattern([
+                    t.objectProperty(t.identifier('scope'), t.identifier('scope'), false, true),
+                    t.restElement(t.identifier('queryOptions')),
+                ]), t.logicalExpression('??', t.identifier('options'), t.objectExpression([]))),
+            ]));
+            hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useQuery'), [
+                t.objectExpression([
+                    t.objectProperty(t.identifier('queryKey'), t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('detail')), [
+                        t.memberExpression(t.identifier('variables'), t.identifier(pkName)),
+                        t.identifier('scope'),
+                    ])),
+                    t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], t.callExpression(t.identifier('execute'), [
+                        t.identifier(`${queryName}QueryDocument`),
+                        t.identifier('variables'),
+                    ]))),
+                    t.spreadElement(t.identifier('queryOptions')),
+                ]),
+            ])));
+        }
+        else if (useCentralizedKeys) {
+            hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useQuery'), [
+                t.objectExpression([
+                    t.objectProperty(t.identifier('queryKey'), t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('detail')), [
+                        t.memberExpression(t.identifier('variables'), t.identifier(pkName)),
+                    ])),
+                    t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], t.callExpression(t.identifier('execute'), [
+                        t.identifier(`${queryName}QueryDocument`),
+                        t.identifier('variables'),
+                    ]))),
+                    t.spreadElement(t.identifier('options')),
+                ]),
+            ])));
+        }
+        else {
+            hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useQuery'), [
+                t.objectExpression([
+                    t.objectProperty(t.identifier('queryKey'), t.callExpression(t.identifier(`${queryName}QueryKey`), [
+                        t.memberExpression(t.identifier('variables'), t.identifier(pkName)),
+                    ])),
+                    t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], t.callExpression(t.identifier('execute'), [
+                        t.identifier(`${queryName}QueryDocument`),
+                        t.identifier('variables'),
+                    ]))),
+                    t.spreadElement(t.identifier('options')),
+                ]),
+            ])));
+        }
+        const hookParams = [
+            (0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(singularName)}QueryVariables`))),
+        ];
+        let optionsTypeStr;
+        if (hasRelationships && useCentralizedKeys) {
+            optionsTypeStr = `Omit<UseQueryOptions<${(0, utils_1.ucFirst)(singularName)}QueryResult, Error>, 'queryKey' | 'queryFn'> & { scope?: ${scopeTypeName} }`;
+        }
+        else {
+            optionsTypeStr = `Omit<UseQueryOptions<${(0, utils_1.ucFirst)(singularName)}QueryResult, 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 docLines = [
+            `Query hook for fetching a single ${typeName}`,
+            '',
+            '@example',
+            '```tsx',
+            `const { data, isLoading } = ${hookName}({ ${pkName}: 'some-id' });`,
+            '```',
+        ];
+        if (hasRelationships && useCentralizedKeys) {
+            docLines.push('');
+            docLines.push('@example With scope for hierarchical cache invalidation');
+            docLines.push('```tsx');
+            docLines.push(`const { data } = ${hookName}(`);
+            docLines.push(`  { ${pkName}: 'some-id' },`);
+            docLines.push("  { scope: { parentId: 'parent-id' } }");
+            docLines.push(');');
+            docLines.push('```');
+        }
+        (0, babel_ast_1.addJSDocComment)(hookExport, docLines);
+        statements.push(hookExport);
     }
-    // Add section comment for standalone functions
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Standalone Functions (non-React)');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Fetch function (standalone, no React) - use dynamic PK
-    sourceFile.addFunction({
-        name: `fetch${(0, utils_1.ucFirst)(singularName)}Query`,
-        isExported: true,
-        isAsync: true,
-        parameters: [
-            { name: pkName, type: pkTsType },
-            {
-                name: 'options',
-                type: 'ExecuteOptions',
-                hasQuestionToken: true,
-            },
-        ],
-        returnType: `Promise<${(0, utils_1.ucFirst)(singularName)}QueryResult>`,
-        statements: `return execute<${(0, utils_1.ucFirst)(singularName)}QueryResult, ${(0, utils_1.ucFirst)(singularName)}QueryVariables>(
-    ${queryName}QueryDocument,
-    { ${pkName} },
-    options
-  );`,
-        docs: [
-            {
-                description: `Fetch a single ${typeName} by primary key without React hooks
-
-@example
-\`\`\`ts
-const data = await fetch${(0, utils_1.ucFirst)(singularName)}Query(${pkTsType === 'string' ? "'value-here'" : '123'});
-\`\`\``,
-            },
-        ],
-    });
-    // Prefetch function (for SSR/QueryClient) - only if React Query is enabled, use dynamic PK
+    const fetchFuncBody = t.blockStatement([
+        t.returnStatement(t.callExpression(t.identifier('execute'), [
+            t.identifier(`${queryName}QueryDocument`),
+            t.identifier('variables'),
+            t.identifier('options'),
+        ])),
+    ]);
+    const fetchFunc = t.functionDeclaration(t.identifier(`fetch${(0, utils_1.ucFirst)(singularName)}Query`), [
+        (0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(singularName)}QueryVariables`))),
+        (0, babel_ast_1.typedParam)('options', t.tsTypeReference(t.identifier('ExecuteOptions')), true),
+    ], fetchFuncBody);
+    fetchFunc.async = true;
+    fetchFunc.returnType = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Promise'), t.tsTypeParameterInstantiation([
+        t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(singularName)}QueryResult`)),
+    ])));
+    const fetchExport = t.exportNamedDeclaration(fetchFunc);
+    (0, babel_ast_1.addJSDocComment)(fetchExport, [
+        `Fetch a single ${typeName} without React hooks`,
+        '',
+        '@example',
+        '```ts',
+        `const data = await fetch${(0, utils_1.ucFirst)(singularName)}Query({ ${pkName}: 'some-id' });`,
+        '```',
+    ]);
+    statements.push(fetchExport);
     if (reactQueryEnabled) {
-        sourceFile.addFunction({
-            name: `prefetch${(0, utils_1.ucFirst)(singularName)}Query`,
-            isExported: true,
-            isAsync: true,
-            parameters: [
-                { name: 'queryClient', type: 'QueryClient' },
-                { name: pkName, type: pkTsType },
-                {
-                    name: 'options',
-                    type: 'ExecuteOptions',
-                    hasQuestionToken: true,
-                },
-            ],
-            returnType: 'Promise<void>',
-            statements: `await queryClient.prefetchQuery({
-    queryKey: ${queryName}QueryKey(${pkName}),
-    queryFn: () => execute<${(0, utils_1.ucFirst)(singularName)}QueryResult, ${(0, utils_1.ucFirst)(singularName)}QueryVariables>(
-      ${queryName}QueryDocument,
-      { ${pkName} },
-      options
-    ),
-  });`,
-            docs: [
-                {
-                    description: `Prefetch a single ${typeName} for SSR or cache warming
-
-@example
-\`\`\`ts
-await prefetch${(0, utils_1.ucFirst)(singularName)}Query(queryClient, ${pkTsType === 'string' ? "'value-here'" : '123'});
-\`\`\``,
-                },
-            ],
-        });
+        const prefetchParams = [
+            (0, babel_ast_1.typedParam)('queryClient', t.tsTypeReference(t.identifier('QueryClient'))),
+            (0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(singularName)}QueryVariables`))),
+        ];
+        if (hasRelationships && useCentralizedKeys) {
+            prefetchParams.push((0, babel_ast_1.typedParam)('scope', t.tsTypeReference(t.identifier(scopeTypeName)), true));
+        }
+        prefetchParams.push((0, babel_ast_1.typedParam)('options', t.tsTypeReference(t.identifier('ExecuteOptions')), true));
+        let prefetchQueryKeyExpr;
+        if (hasRelationships && useCentralizedKeys) {
+            prefetchQueryKeyExpr = t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('detail')), [
+                t.memberExpression(t.identifier('variables'), t.identifier(pkName)),
+                t.identifier('scope'),
+            ]);
+        }
+        else if (useCentralizedKeys) {
+            prefetchQueryKeyExpr = t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('detail')), [t.memberExpression(t.identifier('variables'), t.identifier(pkName))]);
+        }
+        else {
+            prefetchQueryKeyExpr = t.callExpression(t.identifier(`${queryName}QueryKey`), [t.memberExpression(t.identifier('variables'), t.identifier(pkName))]);
+        }
+        const prefetchFuncBody = t.blockStatement([
+            t.expressionStatement(t.awaitExpression(t.callExpression(t.memberExpression(t.identifier('queryClient'), t.identifier('prefetchQuery')), [
+                t.objectExpression([
+                    t.objectProperty(t.identifier('queryKey'), prefetchQueryKeyExpr),
+                    t.objectProperty(t.identifier('queryFn'), t.arrowFunctionExpression([], t.callExpression(t.identifier('execute'), [
+                        t.identifier(`${queryName}QueryDocument`),
+                        t.identifier('variables'),
+                        t.identifier('options'),
+                    ]))),
+                ]),
+            ]))),
+        ]);
+        const prefetchFunc = t.functionDeclaration(t.identifier(`prefetch${(0, utils_1.ucFirst)(singularName)}Query`), prefetchParams, prefetchFuncBody);
+        prefetchFunc.async = true;
+        prefetchFunc.returnType = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Promise'), t.tsTypeParameterInstantiation([t.tsVoidKeyword()])));
+        const prefetchExport = t.exportNamedDeclaration(prefetchFunc);
+        (0, babel_ast_1.addJSDocComment)(prefetchExport, [
+            `Prefetch a single ${typeName} for SSR or cache warming`,
+            '',
+            '@example',
+            '```ts',
+            `await prefetch${(0, utils_1.ucFirst)(singularName)}Query(queryClient, { ${pkName}: 'some-id' });`,
+            '```',
+        ]);
+        statements.push(prefetchExport);
     }
+    const code = (0, babel_ast_1.generateCode)(statements);
+    const headerText = reactQueryEnabled
+        ? `Single item query hook for ${typeName}`
+        : `Single item query functions for ${typeName}`;
+    const content = (0, utils_1.getGeneratedFileHeader)(headerText) + '\n\n' + code;
     return {
         fileName: (0, utils_1.getSingleQueryFileName)(table),
-        content: (0, ts_ast_1.getFormattedOutput)(sourceFile),
+        content,
     };
 }
-// ============================================================================
-// Batch generator
-// ============================================================================
-/**
- * Generate all query hook files for all tables
- */
 function generateAllQueryHooks(tables, options = {}) {
     const files = [];
     for (const table of tables) {