@constructive-io/graphql-codegen 2.23.3 → 2.24.1

package/cli/codegen/mutations.js

@@ -1,481 +1,483 @@
 "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.generateCreateMutationHook = generateCreateMutationHook;
 exports.generateUpdateMutationHook = generateUpdateMutationHook;
 exports.generateDeleteMutationHook = generateDeleteMutationHook;
 exports.generateAllMutationHooks = generateAllMutationHooks;
-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");
-/**
- * Check if a field is auto-generated and should be excluded from create inputs
- * Uses primary key from constraints and common timestamp patterns
- */
 function isAutoGeneratedField(fieldName, pkFieldNames) {
     const name = fieldName.toLowerCase();
-    // Exclude primary key fields (from constraints)
     if (pkFieldNames.has(fieldName))
         return true;
-    // Exclude common timestamp patterns (case-insensitive)
-    // These are typically auto-set by database triggers or defaults
     const timestampPatterns = [
         'createdat', 'created_at', 'createddate', 'created_date',
         'updatedat', 'updated_at', 'updateddate', 'updated_date',
-        'deletedat', 'deleted_at', // soft delete timestamps
+        'deletedat', 'deleted_at',
     ];
     return timestampPatterns.includes(name);
 }
-// ============================================================================
-// Create mutation hook generator
-// ============================================================================
-/**
- * Generate create mutation hook file content using AST
- * When reactQueryEnabled is false, returns null since mutations require React Query
- */
 function generateCreateMutationHook(table, options = {}) {
-    const { reactQueryEnabled = true, enumsFromSchemaTypes = [] } = options;
-    // Mutations require React Query - skip generation when disabled
+    const { reactQueryEnabled = true, enumsFromSchemaTypes = [], useCentralizedKeys = true, hasRelationships = false, tableTypeNames = new Set(), } = options;
     if (!reactQueryEnabled) {
         return null;
     }
     const enumSet = new Set(enumsFromSchemaTypes);
-    const project = (0, ts_ast_1.createProject)();
     const { typeName, singularName } = (0, utils_1.getTableNames)(table);
     const hookName = (0, utils_1.getCreateMutationHookName)(table);
+    const keysName = `${(0, utils_1.lcFirst)(typeName)}Keys`;
+    const mutationKeysName = `${(0, utils_1.lcFirst)(typeName)}MutationKeys`;
+    const scopeTypeName = `${typeName}Scope`;
     const mutationName = (0, utils_1.getCreateMutationName)(table);
     const scalarFields = (0, utils_1.getScalarFields)(table);
-    // Get primary key field names dynamically from table constraints
-    const pkFieldNames = new Set((0, utils_1.getPrimaryKeyInfo)(table).map(pk => pk.name));
-    // Collect which enums are used by this table's fields
+    const pkFieldNames = new Set((0, utils_1.getPrimaryKeyInfo)(table).map((pk) => pk.name));
     const usedEnums = new Set();
+    const usedTableTypes = new Set();
     for (const field of scalarFields) {
         const cleanType = field.type.gqlType.replace(/!/g, '');
         if (enumSet.has(cleanType)) {
             usedEnums.add(cleanType);
         }
+        else if (tableTypeNames.has(cleanType) && cleanType !== typeName) {
+            // Track table types used in scalar fields (excluding the main type which is already imported)
+            usedTableTypes.add(cleanType);
+        }
     }
-    // Generate GraphQL document via AST
     const mutationAST = (0, gql_ast_1.buildCreateMutationAST)({ table });
     const mutationDocument = (0, gql_ast_1.printGraphQL)(mutationAST);
-    const sourceFile = (0, ts_ast_1.createSourceFile)(project, (0, utils_1.getCreateMutationFileName)(table));
-    // Add file header
-    sourceFile.insertText(0, (0, ts_ast_1.createFileHeader)(`Create mutation hook for ${typeName}`) + '\n\n');
-    // Build import declarations
-    const imports = [
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '@tanstack/react-query',
-            namedImports: ['useMutation', 'useQueryClient'],
-            typeOnlyNamedImports: ['UseMutationOptions'],
-        }),
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../client',
-            namedImports: ['execute'],
-        }),
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../types',
-            typeOnlyNamedImports: [typeName],
-        }),
-    ];
-    // Add import for enum types from schema-types if any are used
+    const statements = [];
+    const reactQueryImport = t.importDeclaration([
+        t.importSpecifier(t.identifier('useMutation'), t.identifier('useMutation')),
+        t.importSpecifier(t.identifier('useQueryClient'), t.identifier('useQueryClient')),
+    ], t.stringLiteral('@tanstack/react-query'));
+    statements.push(reactQueryImport);
+    const reactQueryTypeImport = t.importDeclaration([t.importSpecifier(t.identifier('UseMutationOptions'), t.identifier('UseMutationOptions'))], 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);
+    // Import the main type and any other table types used in scalar fields
+    const allTypesToImport = [typeName, ...Array.from(usedTableTypes)].sort();
+    const typesImport = t.importDeclaration(allTypesToImport.map((t_) => t.importSpecifier(t.identifier(t_), t.identifier(t_))), t.stringLiteral('../types'));
+    typesImport.importKind = 'type';
+    statements.push(typesImport);
     if (usedEnums.size > 0) {
-        imports.push((0, ts_ast_1.createImport)({
-            moduleSpecifier: '../schema-types',
-            typeOnlyNamedImports: Array.from(usedEnums).sort(),
-        }));
+        const enumImport = t.importDeclaration(Array.from(usedEnums).sort().map((e) => t.importSpecifier(t.identifier(e), t.identifier(e))), t.stringLiteral('../schema-types'));
+        enumImport.importKind = 'type';
+        statements.push(enumImport);
+    }
+    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 mutationKeyImport = t.importDeclaration([t.importSpecifier(t.identifier(mutationKeysName), t.identifier(mutationKeysName))], t.stringLiteral('../mutation-keys'));
+        statements.push(mutationKeyImport);
     }
-    // Add imports
-    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 mutation document constant
-    sourceFile.addVariableStatement((0, ts_ast_1.createConst)(`${mutationName}MutationDocument`, '`\n' + mutationDocument + '`'));
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Types');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Generate CreateInput type - exclude auto-generated fields
-    // Note: Not exported to avoid conflicts with schema-types
+    const reExportDecl = t.exportNamedDeclaration(null, [t.exportSpecifier(t.identifier(typeName), t.identifier(typeName))], t.stringLiteral('../types'));
+    reExportDecl.exportKind = 'type';
+    statements.push(reExportDecl);
+    const mutationDocConst = t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier(`${mutationName}MutationDocument`), t.templateLiteral([t.templateElement({ raw: '\n' + mutationDocument, cooked: '\n' + mutationDocument }, true)], [])),
+    ]);
+    statements.push(t.exportNamedDeclaration(mutationDocConst));
     const inputFields = scalarFields
         .filter((f) => !isAutoGeneratedField(f.name, pkFieldNames))
-        .map((f) => ({
-        name: f.name,
-        type: `${(0, utils_1.fieldTypeToTs)(f.type)} | null`,
-        optional: true,
-    }));
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${typeName}CreateInput`, inputFields, {
-        docs: [`Input type for creating a ${typeName}`],
-        isExported: false,
-    }));
-    // Variables interface
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`, [
-        {
-            name: 'input',
-            type: `{
-    ${(0, utils_1.lcFirst)(typeName)}: ${typeName}CreateInput;
-  }`,
-        },
-    ]));
-    // Result interface
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${(0, utils_1.ucFirst)(mutationName)}MutationResult`, [
-        {
-            name: mutationName,
-            type: `{
-    ${singularName}: ${typeName};
-  }`,
-        },
-    ]));
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Hook');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Hook function
-    sourceFile.addFunction({
-        name: hookName,
-        isExported: true,
-        parameters: [
-            {
-                name: 'options',
-                type: `Omit<UseMutationOptions<${(0, utils_1.ucFirst)(mutationName)}MutationResult, Error, ${(0, utils_1.ucFirst)(mutationName)}MutationVariables>, 'mutationFn'>`,
-                hasQuestionToken: true,
-            },
-        ],
-        statements: `const queryClient = useQueryClient();
-
-  return useMutation({
-    mutationFn: (variables: ${(0, utils_1.ucFirst)(mutationName)}MutationVariables) =>
-      execute<${(0, utils_1.ucFirst)(mutationName)}MutationResult, ${(0, utils_1.ucFirst)(mutationName)}MutationVariables>(
-        ${mutationName}MutationDocument,
-        variables
-      ),
-    onSuccess: () => {
-      // Invalidate list queries to refetch
-      queryClient.invalidateQueries({ queryKey: ['${typeName.toLowerCase()}', 'list'] });
-    },
-    ...options,
-  });`,
-        docs: [
-            {
-                description: `Mutation hook for creating a ${typeName}
-
-@example
-\`\`\`tsx
-const { mutate, isPending } = ${hookName}();
-
-mutate({
-  input: {
-    ${(0, utils_1.lcFirst)(typeName)}: {
-      // ... fields
-    },
-  },
-});
-\`\`\``,
-            },
-        ],
+        .map((f) => {
+        const prop = t.tsPropertySignature(t.identifier(f.name), t.tsTypeAnnotation(t.tsUnionType([
+            t.tsTypeReference(t.identifier((0, utils_1.fieldTypeToTs)(f.type))),
+            t.tsNullKeyword(),
+        ])));
+        prop.optional = true;
+        return prop;
     });
+    const createInputInterface = t.tsInterfaceDeclaration(t.identifier(`${typeName}CreateInput`), null, null, t.tsInterfaceBody(inputFields));
+    (0, babel_ast_1.addJSDocComment)(createInputInterface, [`Input type for creating a ${typeName}`]);
+    statements.push(createInputInterface);
+    const variablesInterfaceBody = t.tsInterfaceBody([
+        t.tsPropertySignature(t.identifier('input'), t.tsTypeAnnotation(t.tsTypeLiteral([
+            t.tsPropertySignature(t.identifier((0, utils_1.lcFirst)(typeName)), t.tsTypeAnnotation(t.tsTypeReference(t.identifier(`${typeName}CreateInput`)))),
+        ]))),
+    ]);
+    const variablesInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`), null, null, variablesInterfaceBody);
+    statements.push(t.exportNamedDeclaration(variablesInterface));
+    const resultInterfaceBody = t.tsInterfaceBody([
+        t.tsPropertySignature(t.identifier(mutationName), t.tsTypeAnnotation(t.tsTypeLiteral([
+            t.tsPropertySignature(t.identifier(singularName), t.tsTypeAnnotation(t.tsTypeReference(t.identifier(typeName)))),
+        ]))),
+    ]);
+    const resultInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationResult`), null, null, resultInterfaceBody);
+    statements.push(t.exportNamedDeclaration(resultInterface));
+    const hookBodyStatements = [];
+    hookBodyStatements.push(t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier('queryClient'), t.callExpression(t.identifier('useQueryClient'), [])),
+    ]));
+    const mutationOptions = [];
+    if (useCentralizedKeys) {
+        mutationOptions.push(t.objectProperty(t.identifier('mutationKey'), t.callExpression(t.memberExpression(t.identifier(mutationKeysName), t.identifier('create')), [])));
+    }
+    mutationOptions.push(t.objectProperty(t.identifier('mutationFn'), t.arrowFunctionExpression([(0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`)))], (0, babel_ast_1.createTypedCallExpression)(t.identifier('execute'), [t.identifier(`${mutationName}MutationDocument`), t.identifier('variables')], [
+        t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationResult`)),
+        t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`)),
+    ]))));
+    const invalidateQueryKey = useCentralizedKeys
+        ? t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('lists')), [])
+        : t.arrayExpression([t.stringLiteral(typeName.toLowerCase()), t.stringLiteral('list')]);
+    mutationOptions.push(t.objectProperty(t.identifier('onSuccess'), t.arrowFunctionExpression([], t.blockStatement([
+        t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('queryClient'), t.identifier('invalidateQueries')), [t.objectExpression([t.objectProperty(t.identifier('queryKey'), invalidateQueryKey)])])),
+    ]))));
+    mutationOptions.push(t.spreadElement(t.identifier('options')));
+    hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useMutation'), [t.objectExpression(mutationOptions)])));
+    const optionsTypeStr = `Omit<UseMutationOptions<${(0, utils_1.ucFirst)(mutationName)}MutationResult, Error, ${(0, utils_1.ucFirst)(mutationName)}MutationVariables>, 'mutationFn'>`;
+    const optionsParam = t.identifier('options');
+    optionsParam.optional = true;
+    optionsParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeReference(t.identifier(optionsTypeStr)));
+    const hookFunc = t.functionDeclaration(t.identifier(hookName), [optionsParam], t.blockStatement(hookBodyStatements));
+    const hookExport = t.exportNamedDeclaration(hookFunc);
+    (0, babel_ast_1.addJSDocComment)(hookExport, [
+        `Mutation hook for creating a ${typeName}`,
+        '',
+        '@example',
+        '```tsx',
+        `const { mutate, isPending } = ${hookName}();`,
+        '',
+        'mutate({',
+        '  input: {',
+        `    ${(0, utils_1.lcFirst)(typeName)}: {`,
+        '      // ... fields',
+        '    },',
+        '  },',
+        '});',
+        '```',
+    ]);
+    statements.push(hookExport);
+    const code = (0, babel_ast_1.generateCode)(statements);
+    const content = (0, utils_1.getGeneratedFileHeader)(`Create mutation hook for ${typeName}`) + '\n\n' + code;
     return {
         fileName: (0, utils_1.getCreateMutationFileName)(table),
-        content: (0, ts_ast_1.getFormattedOutput)(sourceFile),
+        content,
     };
 }
-// ============================================================================
-// Update mutation hook generator
-// ============================================================================
-/**
- * Generate update mutation hook file content using AST
- * When reactQueryEnabled is false, returns null since mutations require React Query
- */
 function generateUpdateMutationHook(table, options = {}) {
-    const { reactQueryEnabled = true, enumsFromSchemaTypes = [] } = options;
-    // Mutations require React Query - skip generation when disabled
+    const { reactQueryEnabled = true, enumsFromSchemaTypes = [], useCentralizedKeys = true, hasRelationships = false, tableTypeNames = new Set(), } = options;
     if (!reactQueryEnabled) {
         return null;
     }
-    // Check if update mutation exists
     if (table.query?.update === null) {
         return null;
     }
     const enumSet = new Set(enumsFromSchemaTypes);
-    const project = (0, ts_ast_1.createProject)();
     const { typeName, singularName } = (0, utils_1.getTableNames)(table);
     const hookName = (0, utils_1.getUpdateMutationHookName)(table);
     const mutationName = (0, utils_1.getUpdateMutationName)(table);
     const scalarFields = (0, utils_1.getScalarFields)(table);
-    // Get primary key info dynamically from table constraints
+    const keysName = `${(0, utils_1.lcFirst)(typeName)}Keys`;
+    const mutationKeysName = `${(0, utils_1.lcFirst)(typeName)}MutationKeys`;
+    const scopeTypeName = `${typeName}Scope`;
     const pkFields = (0, utils_1.getPrimaryKeyInfo)(table);
-    const pkField = pkFields[0]; // Use first PK field
-    const pkFieldNames = new Set(pkFields.map(pk => pk.name));
-    // Collect which enums are used by this table's fields
+    const pkField = pkFields[0];
+    const pkFieldNames = new Set(pkFields.map((pk) => pk.name));
     const usedEnums = new Set();
+    const usedTableTypes = new Set();
     for (const field of scalarFields) {
         const cleanType = field.type.gqlType.replace(/!/g, '');
         if (enumSet.has(cleanType)) {
             usedEnums.add(cleanType);
         }
+        else if (tableTypeNames.has(cleanType) && cleanType !== typeName) {
+            usedTableTypes.add(cleanType);
+        }
     }
-    // Generate GraphQL document via AST
     const mutationAST = (0, gql_ast_1.buildUpdateMutationAST)({ table });
     const mutationDocument = (0, gql_ast_1.printGraphQL)(mutationAST);
-    const sourceFile = (0, ts_ast_1.createSourceFile)(project, (0, utils_1.getUpdateMutationFileName)(table));
-    // Add file header
-    sourceFile.insertText(0, (0, ts_ast_1.createFileHeader)(`Update mutation hook for ${typeName}`) + '\n\n');
-    // Build import declarations
-    const imports = [
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '@tanstack/react-query',
-            namedImports: ['useMutation', 'useQueryClient'],
-            typeOnlyNamedImports: ['UseMutationOptions'],
-        }),
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../client',
-            namedImports: ['execute'],
-        }),
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../types',
-            typeOnlyNamedImports: [typeName],
-        }),
-    ];
-    // Add import for enum types from schema-types if any are used
+    const statements = [];
+    const reactQueryImport = t.importDeclaration([
+        t.importSpecifier(t.identifier('useMutation'), t.identifier('useMutation')),
+        t.importSpecifier(t.identifier('useQueryClient'), t.identifier('useQueryClient')),
+    ], t.stringLiteral('@tanstack/react-query'));
+    statements.push(reactQueryImport);
+    const reactQueryTypeImport = t.importDeclaration([t.importSpecifier(t.identifier('UseMutationOptions'), t.identifier('UseMutationOptions'))], 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);
+    // Import the main type and any other table types used in scalar fields
+    const allTypesToImportUpdate = [typeName, ...Array.from(usedTableTypes)].sort();
+    const typesImport = t.importDeclaration(allTypesToImportUpdate.map((t_) => t.importSpecifier(t.identifier(t_), t.identifier(t_))), t.stringLiteral('../types'));
+    typesImport.importKind = 'type';
+    statements.push(typesImport);
     if (usedEnums.size > 0) {
-        imports.push((0, ts_ast_1.createImport)({
-            moduleSpecifier: '../schema-types',
-            typeOnlyNamedImports: Array.from(usedEnums).sort(),
-        }));
+        const enumImport = t.importDeclaration(Array.from(usedEnums).sort().map((e) => t.importSpecifier(t.identifier(e), t.identifier(e))), t.stringLiteral('../schema-types'));
+        enumImport.importKind = 'type';
+        statements.push(enumImport);
     }
-    // Add imports
-    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 mutation document constant
-    sourceFile.addVariableStatement((0, ts_ast_1.createConst)(`${mutationName}MutationDocument`, '`\n' + mutationDocument + '`'));
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Types');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Generate Patch type - all fields optional, exclude primary key
-    // Note: Not exported to avoid conflicts with schema-types
+    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 mutationKeyImport = t.importDeclaration([t.importSpecifier(t.identifier(mutationKeysName), t.identifier(mutationKeysName))], t.stringLiteral('../mutation-keys'));
+        statements.push(mutationKeyImport);
+    }
+    const reExportDecl = t.exportNamedDeclaration(null, [t.exportSpecifier(t.identifier(typeName), t.identifier(typeName))], t.stringLiteral('../types'));
+    reExportDecl.exportKind = 'type';
+    statements.push(reExportDecl);
+    const mutationDocConst = t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier(`${mutationName}MutationDocument`), t.templateLiteral([t.templateElement({ raw: '\n' + mutationDocument, cooked: '\n' + mutationDocument }, true)], [])),
+    ]);
+    statements.push(t.exportNamedDeclaration(mutationDocConst));
     const patchFields = scalarFields
         .filter((f) => !pkFieldNames.has(f.name))
-        .map((f) => ({
-        name: f.name,
-        type: `${(0, utils_1.fieldTypeToTs)(f.type)} | null`,
-        optional: true,
-    }));
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${typeName}Patch`, patchFields, {
-        docs: [`Patch type for updating a ${typeName} - all fields optional`],
-        isExported: false,
-    }));
-    // Variables interface - use dynamic PK field name and type
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`, [
-        {
-            name: 'input',
-            type: `{
-    ${pkField.name}: ${pkField.tsType};
-    patch: ${typeName}Patch;
-  }`,
-        },
-    ]));
-    // Result interface
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${(0, utils_1.ucFirst)(mutationName)}MutationResult`, [
-        {
-            name: mutationName,
-            type: `{
-    ${singularName}: ${typeName};
-  }`,
-        },
-    ]));
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Hook');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Hook function
-    sourceFile.addFunction({
-        name: hookName,
-        isExported: true,
-        parameters: [
-            {
-                name: 'options',
-                type: `Omit<UseMutationOptions<${(0, utils_1.ucFirst)(mutationName)}MutationResult, Error, ${(0, utils_1.ucFirst)(mutationName)}MutationVariables>, 'mutationFn'>`,
-                hasQuestionToken: true,
-            },
-        ],
-        statements: `const queryClient = useQueryClient();
-
-  return useMutation({
-    mutationFn: (variables: ${(0, utils_1.ucFirst)(mutationName)}MutationVariables) =>
-      execute<${(0, utils_1.ucFirst)(mutationName)}MutationResult, ${(0, utils_1.ucFirst)(mutationName)}MutationVariables>(
-        ${mutationName}MutationDocument,
-        variables
-      ),
-    onSuccess: (_, variables) => {
-      // Invalidate specific item and list queries
-      queryClient.invalidateQueries({ queryKey: ['${typeName.toLowerCase()}', 'detail', variables.input.${pkField.name}] });
-      queryClient.invalidateQueries({ queryKey: ['${typeName.toLowerCase()}', 'list'] });
-    },
-    ...options,
-  });`,
-        docs: [
-            {
-                description: `Mutation hook for updating a ${typeName}
-
-@example
-\`\`\`tsx
-const { mutate, isPending } = ${hookName}();
-
-mutate({
-  input: {
-    ${pkField.name}: ${pkField.tsType === 'string' ? "'value-here'" : '123'},
-    patch: {
-      // ... fields to update
-    },
-  },
-});
-\`\`\``,
-            },
-        ],
+        .map((f) => {
+        const prop = t.tsPropertySignature(t.identifier(f.name), t.tsTypeAnnotation(t.tsUnionType([
+            t.tsTypeReference(t.identifier((0, utils_1.fieldTypeToTs)(f.type))),
+            t.tsNullKeyword(),
+        ])));
+        prop.optional = true;
+        return prop;
     });
+    const patchInterface = t.tsInterfaceDeclaration(t.identifier(`${typeName}Patch`), null, null, t.tsInterfaceBody(patchFields));
+    (0, babel_ast_1.addJSDocComment)(patchInterface, [`Patch type for updating a ${typeName} - all fields optional`]);
+    statements.push(patchInterface);
+    const pkTypeAnnotation = pkField.tsType === 'string'
+        ? t.tsStringKeyword()
+        : pkField.tsType === 'number'
+            ? t.tsNumberKeyword()
+            : t.tsTypeReference(t.identifier(pkField.tsType));
+    const variablesInterfaceBody = t.tsInterfaceBody([
+        t.tsPropertySignature(t.identifier('input'), t.tsTypeAnnotation(t.tsTypeLiteral([
+            t.tsPropertySignature(t.identifier(pkField.name), t.tsTypeAnnotation(pkTypeAnnotation)),
+            t.tsPropertySignature(t.identifier('patch'), t.tsTypeAnnotation(t.tsTypeReference(t.identifier(`${typeName}Patch`)))),
+        ]))),
+    ]);
+    const variablesInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`), null, null, variablesInterfaceBody);
+    statements.push(t.exportNamedDeclaration(variablesInterface));
+    const resultInterfaceBody = t.tsInterfaceBody([
+        t.tsPropertySignature(t.identifier(mutationName), t.tsTypeAnnotation(t.tsTypeLiteral([
+            t.tsPropertySignature(t.identifier(singularName), t.tsTypeAnnotation(t.tsTypeReference(t.identifier(typeName)))),
+        ]))),
+    ]);
+    const resultInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationResult`), null, null, resultInterfaceBody);
+    statements.push(t.exportNamedDeclaration(resultInterface));
+    const hookBodyStatements = [];
+    hookBodyStatements.push(t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier('queryClient'), t.callExpression(t.identifier('useQueryClient'), [])),
+    ]));
+    const mutationOptions = [];
+    if (useCentralizedKeys) {
+        mutationOptions.push(t.objectProperty(t.identifier('mutationKey'), t.memberExpression(t.identifier(mutationKeysName), t.identifier('all'))));
+    }
+    mutationOptions.push(t.objectProperty(t.identifier('mutationFn'), t.arrowFunctionExpression([(0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`)))], (0, babel_ast_1.createTypedCallExpression)(t.identifier('execute'), [t.identifier(`${mutationName}MutationDocument`), t.identifier('variables')], [
+        t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationResult`)),
+        t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`)),
+    ]))));
+    const detailQueryKey = useCentralizedKeys
+        ? t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('detail')), [t.memberExpression(t.memberExpression(t.identifier('variables'), t.identifier('input')), t.identifier(pkField.name))])
+        : t.arrayExpression([
+            t.stringLiteral(typeName.toLowerCase()),
+            t.stringLiteral('detail'),
+            t.memberExpression(t.memberExpression(t.identifier('variables'), t.identifier('input')), t.identifier(pkField.name)),
+        ]);
+    const listQueryKey = useCentralizedKeys
+        ? t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('lists')), [])
+        : t.arrayExpression([t.stringLiteral(typeName.toLowerCase()), t.stringLiteral('list')]);
+    mutationOptions.push(t.objectProperty(t.identifier('onSuccess'), t.arrowFunctionExpression([t.identifier('_'), t.identifier('variables')], t.blockStatement([
+        t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('queryClient'), t.identifier('invalidateQueries')), [t.objectExpression([t.objectProperty(t.identifier('queryKey'), detailQueryKey)])])),
+        t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('queryClient'), t.identifier('invalidateQueries')), [t.objectExpression([t.objectProperty(t.identifier('queryKey'), listQueryKey)])])),
+    ]))));
+    mutationOptions.push(t.spreadElement(t.identifier('options')));
+    hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useMutation'), [t.objectExpression(mutationOptions)])));
+    const optionsTypeStr = `Omit<UseMutationOptions<${(0, utils_1.ucFirst)(mutationName)}MutationResult, Error, ${(0, utils_1.ucFirst)(mutationName)}MutationVariables>, 'mutationFn'>`;
+    const optionsParam = t.identifier('options');
+    optionsParam.optional = true;
+    optionsParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeReference(t.identifier(optionsTypeStr)));
+    const hookFunc = t.functionDeclaration(t.identifier(hookName), [optionsParam], t.blockStatement(hookBodyStatements));
+    const hookExport = t.exportNamedDeclaration(hookFunc);
+    (0, babel_ast_1.addJSDocComment)(hookExport, [
+        `Mutation hook for updating a ${typeName}`,
+        '',
+        '@example',
+        '```tsx',
+        `const { mutate, isPending } = ${hookName}();`,
+        '',
+        'mutate({',
+        '  input: {',
+        `    ${pkField.name}: ${pkField.tsType === 'string' ? "'value-here'" : '123'},`,
+        '    patch: {',
+        '      // ... fields to update',
+        '    },',
+        '  },',
+        '});',
+        '```',
+    ]);
+    statements.push(hookExport);
+    const code = (0, babel_ast_1.generateCode)(statements);
+    const content = (0, utils_1.getGeneratedFileHeader)(`Update mutation hook for ${typeName}`) + '\n\n' + code;
     return {
         fileName: (0, utils_1.getUpdateMutationFileName)(table),
-        content: (0, ts_ast_1.getFormattedOutput)(sourceFile),
+        content,
     };
 }
-// ============================================================================
-// Delete mutation hook generator
-// ============================================================================
-/**
- * Generate delete mutation hook file content using AST
- * When reactQueryEnabled is false, returns null since mutations require React Query
- */
 function generateDeleteMutationHook(table, options = {}) {
-    const { reactQueryEnabled = true } = options;
-    // Mutations require React Query - skip generation when disabled
+    const { reactQueryEnabled = true, useCentralizedKeys = true, hasRelationships = false, } = options;
     if (!reactQueryEnabled) {
         return null;
     }
-    // Check if delete mutation exists
     if (table.query?.delete === null) {
         return null;
     }
-    const project = (0, ts_ast_1.createProject)();
     const { typeName } = (0, utils_1.getTableNames)(table);
     const hookName = (0, utils_1.getDeleteMutationHookName)(table);
     const mutationName = (0, utils_1.getDeleteMutationName)(table);
-    // Get primary key info dynamically from table constraints
+    const keysName = `${(0, utils_1.lcFirst)(typeName)}Keys`;
+    const mutationKeysName = `${(0, utils_1.lcFirst)(typeName)}MutationKeys`;
+    const scopeTypeName = `${typeName}Scope`;
     const pkFields = (0, utils_1.getPrimaryKeyInfo)(table);
-    const pkField = pkFields[0]; // Use first PK field
-    // Generate GraphQL document via AST
+    const pkField = pkFields[0];
     const mutationAST = (0, gql_ast_1.buildDeleteMutationAST)({ table });
     const mutationDocument = (0, gql_ast_1.printGraphQL)(mutationAST);
-    const sourceFile = (0, ts_ast_1.createSourceFile)(project, (0, utils_1.getDeleteMutationFileName)(table));
-    // Add file header
-    sourceFile.insertText(0, (0, ts_ast_1.createFileHeader)(`Delete mutation hook for ${typeName}`) + '\n\n');
-    // Add imports
-    sourceFile.addImportDeclarations([
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '@tanstack/react-query',
-            namedImports: ['useMutation', 'useQueryClient'],
-            typeOnlyNamedImports: ['UseMutationOptions'],
-        }),
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../client',
-            namedImports: ['execute'],
-        }),
+    const statements = [];
+    const reactQueryImport = t.importDeclaration([
+        t.importSpecifier(t.identifier('useMutation'), t.identifier('useMutation')),
+        t.importSpecifier(t.identifier('useQueryClient'), t.identifier('useQueryClient')),
+    ], t.stringLiteral('@tanstack/react-query'));
+    statements.push(reactQueryImport);
+    const reactQueryTypeImport = t.importDeclaration([t.importSpecifier(t.identifier('UseMutationOptions'), t.identifier('UseMutationOptions'))], 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);
+    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 mutationKeyImport = t.importDeclaration([t.importSpecifier(t.identifier(mutationKeysName), t.identifier(mutationKeysName))], t.stringLiteral('../mutation-keys'));
+        statements.push(mutationKeyImport);
+    }
+    const mutationDocConst = t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier(`${mutationName}MutationDocument`), t.templateLiteral([t.templateElement({ raw: '\n' + mutationDocument, cooked: '\n' + mutationDocument }, true)], [])),
     ]);
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// GraphQL Document');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Add mutation document constant
-    sourceFile.addVariableStatement((0, ts_ast_1.createConst)(`${mutationName}MutationDocument`, '`\n' + mutationDocument + '`'));
-    // 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)(mutationName)}MutationVariables`, [
-        {
-            name: 'input',
-            type: `{
-    ${pkField.name}: ${pkField.tsType};
-  }`,
-        },
-    ]));
-    // Result interface
-    sourceFile.addInterface((0, ts_ast_1.createInterface)(`${(0, utils_1.ucFirst)(mutationName)}MutationResult`, [
-        {
-            name: mutationName,
-            type: `{
-    clientMutationId: string | null;
-    deleted${(0, utils_1.ucFirst)(pkField.name)}: ${pkField.tsType} | null;
-  }`,
-        },
+    statements.push(t.exportNamedDeclaration(mutationDocConst));
+    const pkTypeAnnotation = pkField.tsType === 'string'
+        ? t.tsStringKeyword()
+        : pkField.tsType === 'number'
+            ? t.tsNumberKeyword()
+            : t.tsTypeReference(t.identifier(pkField.tsType));
+    const variablesInterfaceBody = t.tsInterfaceBody([
+        t.tsPropertySignature(t.identifier('input'), t.tsTypeAnnotation(t.tsTypeLiteral([
+            t.tsPropertySignature(t.identifier(pkField.name), t.tsTypeAnnotation(pkTypeAnnotation)),
+        ]))),
+    ]);
+    const variablesInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`), null, null, variablesInterfaceBody);
+    statements.push(t.exportNamedDeclaration(variablesInterface));
+    const deletedPkProp = t.tsPropertySignature(t.identifier(`deleted${(0, utils_1.ucFirst)(pkField.name)}`), t.tsTypeAnnotation(t.tsUnionType([pkTypeAnnotation, t.tsNullKeyword()])));
+    const clientMutationIdProp = t.tsPropertySignature(t.identifier('clientMutationId'), t.tsTypeAnnotation(t.tsUnionType([t.tsStringKeyword(), t.tsNullKeyword()])));
+    const resultInterfaceBody = t.tsInterfaceBody([
+        t.tsPropertySignature(t.identifier(mutationName), t.tsTypeAnnotation(t.tsTypeLiteral([clientMutationIdProp, deletedPkProp]))),
+    ]);
+    const resultInterface = t.tsInterfaceDeclaration(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationResult`), null, null, resultInterfaceBody);
+    statements.push(t.exportNamedDeclaration(resultInterface));
+    const hookBodyStatements = [];
+    hookBodyStatements.push(t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier('queryClient'), t.callExpression(t.identifier('useQueryClient'), [])),
     ]));
-    // Add section comment
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Hook');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Hook function
-    sourceFile.addFunction({
-        name: hookName,
-        isExported: true,
-        parameters: [
-            {
-                name: 'options',
-                type: `Omit<UseMutationOptions<${(0, utils_1.ucFirst)(mutationName)}MutationResult, Error, ${(0, utils_1.ucFirst)(mutationName)}MutationVariables>, 'mutationFn'>`,
-                hasQuestionToken: true,
-            },
-        ],
-        statements: `const queryClient = useQueryClient();
-
-  return useMutation({
-    mutationFn: (variables: ${(0, utils_1.ucFirst)(mutationName)}MutationVariables) =>
-      execute<${(0, utils_1.ucFirst)(mutationName)}MutationResult, ${(0, utils_1.ucFirst)(mutationName)}MutationVariables>(
-        ${mutationName}MutationDocument,
-        variables
-      ),
-    onSuccess: (_, variables) => {
-      // Remove from cache and invalidate list
-      queryClient.removeQueries({ queryKey: ['${typeName.toLowerCase()}', 'detail', variables.input.${pkField.name}] });
-      queryClient.invalidateQueries({ queryKey: ['${typeName.toLowerCase()}', 'list'] });
-    },
-    ...options,
-  });`,
-        docs: [
-            {
-                description: `Mutation hook for deleting a ${typeName}
-
-@example
-\`\`\`tsx
-const { mutate, isPending } = ${hookName}();
-
-mutate({
-  input: {
-    ${pkField.name}: ${pkField.tsType === 'string' ? "'value-to-delete'" : '123'},
-  },
-});
-\`\`\``,
-            },
-        ],
-    });
+    const mutationOptions = [];
+    if (useCentralizedKeys) {
+        mutationOptions.push(t.objectProperty(t.identifier('mutationKey'), t.memberExpression(t.identifier(mutationKeysName), t.identifier('all'))));
+    }
+    mutationOptions.push(t.objectProperty(t.identifier('mutationFn'), t.arrowFunctionExpression([(0, babel_ast_1.typedParam)('variables', t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`)))], (0, babel_ast_1.createTypedCallExpression)(t.identifier('execute'), [t.identifier(`${mutationName}MutationDocument`), t.identifier('variables')], [
+        t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationResult`)),
+        t.tsTypeReference(t.identifier(`${(0, utils_1.ucFirst)(mutationName)}MutationVariables`)),
+    ]))));
+    const detailQueryKey = useCentralizedKeys
+        ? t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('detail')), [t.memberExpression(t.memberExpression(t.identifier('variables'), t.identifier('input')), t.identifier(pkField.name))])
+        : t.arrayExpression([
+            t.stringLiteral(typeName.toLowerCase()),
+            t.stringLiteral('detail'),
+            t.memberExpression(t.memberExpression(t.identifier('variables'), t.identifier('input')), t.identifier(pkField.name)),
+        ]);
+    const listQueryKey = useCentralizedKeys
+        ? t.callExpression(t.memberExpression(t.identifier(keysName), t.identifier('lists')), [])
+        : t.arrayExpression([t.stringLiteral(typeName.toLowerCase()), t.stringLiteral('list')]);
+    mutationOptions.push(t.objectProperty(t.identifier('onSuccess'), t.arrowFunctionExpression([t.identifier('_'), t.identifier('variables')], t.blockStatement([
+        t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('queryClient'), t.identifier('removeQueries')), [t.objectExpression([t.objectProperty(t.identifier('queryKey'), detailQueryKey)])])),
+        t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('queryClient'), t.identifier('invalidateQueries')), [t.objectExpression([t.objectProperty(t.identifier('queryKey'), listQueryKey)])])),
+    ]))));
+    mutationOptions.push(t.spreadElement(t.identifier('options')));
+    hookBodyStatements.push(t.returnStatement(t.callExpression(t.identifier('useMutation'), [t.objectExpression(mutationOptions)])));
+    const optionsTypeStr = `Omit<UseMutationOptions<${(0, utils_1.ucFirst)(mutationName)}MutationResult, Error, ${(0, utils_1.ucFirst)(mutationName)}MutationVariables>, 'mutationFn'>`;
+    const optionsParam = t.identifier('options');
+    optionsParam.optional = true;
+    optionsParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeReference(t.identifier(optionsTypeStr)));
+    const hookFunc = t.functionDeclaration(t.identifier(hookName), [optionsParam], t.blockStatement(hookBodyStatements));
+    const hookExport = t.exportNamedDeclaration(hookFunc);
+    (0, babel_ast_1.addJSDocComment)(hookExport, [
+        `Mutation hook for deleting a ${typeName}`,
+        '',
+        '@example',
+        '```tsx',
+        `const { mutate, isPending } = ${hookName}();`,
+        '',
+        'mutate({',
+        '  input: {',
+        `    ${pkField.name}: ${pkField.tsType === 'string' ? "'value-to-delete'" : '123'},`,
+        '  },',
+        '});',
+        '```',
+    ]);
+    statements.push(hookExport);
+    const code = (0, babel_ast_1.generateCode)(statements);
+    const content = (0, utils_1.getGeneratedFileHeader)(`Delete mutation hook for ${typeName}`) + '\n\n' + code;
     return {
         fileName: (0, utils_1.getDeleteMutationFileName)(table),
-        content: (0, ts_ast_1.getFormattedOutput)(sourceFile),
+        content,
     };
 }
-// ============================================================================
-// Batch generator
-// ============================================================================
-/**
- * Generate all mutation hook files for all tables
- * When reactQueryEnabled is false, returns empty array since mutations require React Query
- */
 function generateAllMutationHooks(tables, options = {}) {
     const files = [];
     for (const table of tables) {