@constructive-io/graphql-codegen 2.23.2 → 2.24.0

package/cli/codegen/orm/custom-ops-generator.js

@@ -1,8 +1,42 @@
 "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.generateCustomQueryOpsFile = generateCustomQueryOpsFile;
 exports.generateCustomMutationOpsFile = generateCustomMutationOpsFile;
-const ts_ast_1 = require("../ts-ast");
+const t = __importStar(require("@babel/types"));
+const babel_ast_1 = require("../babel-ast");
 const utils_1 = require("../utils");
 const type_resolver_1 = require("../type-resolver");
 const scalars_1 = require("../scalars");
@@ -65,274 +99,197 @@ function getSelectTypeName(returnType) {
     }
     return null;
 }
+function createImportDeclaration(moduleSpecifier, namedImports, typeOnly = false) {
+    const specifiers = namedImports.map((name) => t.importSpecifier(t.identifier(name), t.identifier(name)));
+    const decl = t.importDeclaration(specifiers, t.stringLiteral(moduleSpecifier));
+    decl.importKind = typeOnly ? 'type' : 'value';
+    return decl;
+}
+function createVariablesInterface(op) {
+    if (op.args.length === 0)
+        return null;
+    const varTypeName = `${(0, utils_1.ucFirst)(op.name)}Variables`;
+    const props = op.args.map((arg) => {
+        const optional = !(0, type_resolver_1.isTypeRequired)(arg.type);
+        const prop = t.tsPropertySignature(t.identifier(arg.name), t.tsTypeAnnotation(parseTypeAnnotation((0, type_resolver_1.typeRefToTsType)(arg.type))));
+        prop.optional = optional;
+        return prop;
+    });
+    const interfaceDecl = t.tsInterfaceDeclaration(t.identifier(varTypeName), null, null, t.tsInterfaceBody(props));
+    return t.exportNamedDeclaration(interfaceDecl);
+}
+function parseTypeAnnotation(typeStr) {
+    if (typeStr === 'string')
+        return t.tsStringKeyword();
+    if (typeStr === 'number')
+        return t.tsNumberKeyword();
+    if (typeStr === 'boolean')
+        return t.tsBooleanKeyword();
+    if (typeStr === 'null')
+        return t.tsNullKeyword();
+    if (typeStr === 'undefined')
+        return t.tsUndefinedKeyword();
+    if (typeStr === 'unknown')
+        return t.tsUnknownKeyword();
+    if (typeStr.includes(' | ')) {
+        const parts = typeStr.split(' | ').map((p) => parseTypeAnnotation(p.trim()));
+        return t.tsUnionType(parts);
+    }
+    if (typeStr.endsWith('[]')) {
+        return t.tsArrayType(parseTypeAnnotation(typeStr.slice(0, -2)));
+    }
+    return t.tsTypeReference(t.identifier(typeStr));
+}
+function buildOperationMethod(op, operationType) {
+    const hasArgs = op.args.length > 0;
+    const varTypeName = `${(0, utils_1.ucFirst)(op.name)}Variables`;
+    const varDefs = op.args.map((arg) => ({
+        name: arg.name,
+        type: formatGraphQLType(arg.type),
+    }));
+    const selectTypeName = getSelectTypeName(op.returnType);
+    const payloadTypeName = (0, type_resolver_1.getTypeBaseName)(op.returnType);
+    // Build the arrow function parameters
+    const params = [];
+    if (hasArgs) {
+        const argsParam = t.identifier('args');
+        argsParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeReference(t.identifier(varTypeName)));
+        params.push(argsParam);
+    }
+    const optionsParam = t.identifier('options');
+    optionsParam.optional = true;
+    if (selectTypeName) {
+        optionsParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeLiteral([
+            (() => {
+                const prop = t.tsPropertySignature(t.identifier('select'), t.tsTypeAnnotation(t.tsTypeReference(t.identifier('S'))));
+                prop.optional = true;
+                return prop;
+            })(),
+        ]));
+    }
+    else {
+        optionsParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeLiteral([
+            (() => {
+                const prop = t.tsPropertySignature(t.identifier('select'), t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Record'), t.tsTypeParameterInstantiation([t.tsStringKeyword(), t.tsUnknownKeyword()]))));
+                prop.optional = true;
+                return prop;
+            })(),
+        ]));
+    }
+    params.push(optionsParam);
+    // Build the QueryBuilder call
+    const queryBuilderArgs = t.objectExpression([
+        t.objectProperty(t.identifier('client'), t.identifier('client'), false, true),
+        t.objectProperty(t.identifier('operation'), t.stringLiteral(operationType)),
+        t.objectProperty(t.identifier('operationName'), t.stringLiteral((0, utils_1.ucFirst)(op.name))),
+        t.objectProperty(t.identifier('fieldName'), t.stringLiteral(op.name)),
+        t.spreadElement(t.callExpression(t.identifier('buildCustomDocument'), [
+            t.stringLiteral(operationType),
+            t.stringLiteral((0, utils_1.ucFirst)(op.name)),
+            t.stringLiteral(op.name),
+            t.optionalMemberExpression(t.identifier('options'), t.identifier('select'), false, true),
+            hasArgs ? t.identifier('args') : t.identifier('undefined'),
+            t.arrayExpression(varDefs.map((v) => t.objectExpression([
+                t.objectProperty(t.identifier('name'), t.stringLiteral(v.name)),
+                t.objectProperty(t.identifier('type'), t.stringLiteral(v.type)),
+            ]))),
+        ])),
+    ]);
+    const newExpr = t.newExpression(t.identifier('QueryBuilder'), [queryBuilderArgs]);
+    // Add type parameter if we have a select type
+    if (selectTypeName && payloadTypeName) {
+        newExpr.typeParameters = t.tsTypeParameterInstantiation([
+            t.tsTypeLiteral([
+                t.tsPropertySignature(t.identifier(op.name), t.tsTypeAnnotation(t.tsTypeReference(t.identifier('InferSelectResult'), t.tsTypeParameterInstantiation([
+                    t.tsTypeReference(t.identifier(payloadTypeName)),
+                    t.tsTypeReference(t.identifier('S')),
+                ])))),
+            ]),
+        ]);
+    }
+    const arrowFunc = t.arrowFunctionExpression(params, newExpr);
+    // Add type parameters to arrow function if we have a select type
+    if (selectTypeName) {
+        const typeParam = t.tsTypeParameter(t.tsTypeReference(t.identifier(selectTypeName)), null, 'S');
+        typeParam.const = true;
+        arrowFunc.typeParameters = t.tsTypeParameterDeclaration([typeParam]);
+    }
+    return t.objectProperty(t.identifier(op.name), arrowFunc);
+}
 /**
  * Generate the query/index.ts file for custom query operations
  */
 function generateCustomQueryOpsFile(operations) {
-    const project = (0, ts_ast_1.createProject)();
-    const sourceFile = (0, ts_ast_1.createSourceFile)(project, 'index.ts');
+    const statements = [];
     // Collect all input type names and payload type names
     const inputTypeNames = collectInputTypeNamesFromOps(operations);
     const payloadTypeNames = collectPayloadTypeNamesFromOps(operations);
-    // Generate Select type names for payloads
     const selectTypeNames = payloadTypeNames.map((p) => `${p}Select`);
-    // Combine all type imports
-    const allTypeImports = [
-        ...new Set([...inputTypeNames, ...payloadTypeNames, ...selectTypeNames]),
-    ];
-    // Add file header
-    sourceFile.insertText(0, (0, ts_ast_1.createFileHeader)('Custom query operations') + '\n\n');
+    const allTypeImports = [...new Set([...inputTypeNames, ...payloadTypeNames, ...selectTypeNames])];
     // Add imports
-    sourceFile.addImportDeclarations([
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../client',
-            namedImports: ['OrmClient'],
-        }),
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../query-builder',
-            namedImports: ['QueryBuilder', 'buildCustomDocument'],
-        }),
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../select-types',
-            typeOnlyNamedImports: ['InferSelectResult'],
-        }),
-    ]);
-    // Import types from input-types if we have any
+    statements.push(createImportDeclaration('../client', ['OrmClient']));
+    statements.push(createImportDeclaration('../query-builder', ['QueryBuilder', 'buildCustomDocument']));
+    statements.push(createImportDeclaration('../select-types', ['InferSelectResult'], true));
     if (allTypeImports.length > 0) {
-        sourceFile.addImportDeclarations([
-            (0, ts_ast_1.createImport)({
-                moduleSpecifier: '../input-types',
-                typeOnlyNamedImports: allTypeImports,
-            }),
-        ]);
+        statements.push(createImportDeclaration('../input-types', allTypeImports, true));
     }
-    // Generate variable definitions type for each operation
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Variable Types');
-    sourceFile.addStatements('// ============================================================================\n');
+    // Generate variable interfaces
     for (const op of operations) {
-        if (op.args.length > 0) {
-            const varTypeName = `${(0, utils_1.ucFirst)(op.name)}Variables`;
-            const props = op.args.map((arg) => {
-                const optional = !(0, type_resolver_1.isTypeRequired)(arg.type);
-                return `${arg.name}${optional ? '?' : ''}: ${(0, type_resolver_1.typeRefToTsType)(arg.type)};`;
-            });
-            sourceFile.addStatements(`export interface ${varTypeName} {\n  ${props.join('\n  ')}\n}\n`);
-        }
+        const varInterface = createVariablesInterface(op);
+        if (varInterface)
+            statements.push(varInterface);
     }
     // Generate factory function
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Query Operations Factory');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Build the operations object
-    const operationMethods = operations.map((op) => {
-        const hasArgs = op.args.length > 0;
-        const varTypeName = `${(0, utils_1.ucFirst)(op.name)}Variables`;
-        const varDefs = op.args.map((arg) => ({
-            name: arg.name,
-            type: formatGraphQLType(arg.type),
-        }));
-        const varDefsJson = JSON.stringify(varDefs);
-        // Get Select type for return type
-        const selectTypeName = getSelectTypeName(op.returnType);
-        const payloadTypeName = (0, type_resolver_1.getTypeBaseName)(op.returnType);
-        // Use typed select if available, otherwise fall back to Record<string, unknown>
-        const selectType = selectTypeName ?? 'Record<string, unknown>';
-        const returnTypePart = selectTypeName && payloadTypeName
-            ? `{ ${op.name}: InferSelectResult<${payloadTypeName}, S> }`
-            : 'unknown';
-        if (hasArgs) {
-            if (selectTypeName) {
-                return `${op.name}: <const S extends ${selectType}>(args: ${varTypeName}, options?: { select?: S }) =>
-      new QueryBuilder<${returnTypePart}>({
-        client,
-        operation: 'query',
-        operationName: '${(0, utils_1.ucFirst)(op.name)}',
-        fieldName: '${op.name}',
-        ...buildCustomDocument('query', '${(0, utils_1.ucFirst)(op.name)}', '${op.name}', options?.select, args, ${varDefsJson}),
-      })`;
-            }
-            else {
-                return `${op.name}: (args: ${varTypeName}, options?: { select?: Record<string, unknown> }) =>
-      new QueryBuilder({
-        client,
-        operation: 'query',
-        operationName: '${(0, utils_1.ucFirst)(op.name)}',
-        fieldName: '${op.name}',
-        ...buildCustomDocument('query', '${(0, utils_1.ucFirst)(op.name)}', '${op.name}', options?.select, args, ${varDefsJson}),
-      })`;
-            }
-        }
-        else {
-            // No args - still provide typed select
-            if (selectTypeName) {
-                return `${op.name}: <const S extends ${selectType}>(options?: { select?: S }) =>
-      new QueryBuilder<${returnTypePart}>({
-        client,
-        operation: 'query',
-        operationName: '${(0, utils_1.ucFirst)(op.name)}',
-        fieldName: '${op.name}',
-        ...buildCustomDocument('query', '${(0, utils_1.ucFirst)(op.name)}', '${op.name}', options?.select, undefined, []),
-      })`;
-            }
-            else {
-                return `${op.name}: (options?: { select?: Record<string, unknown> }) =>
-      new QueryBuilder({
-        client,
-        operation: 'query',
-        operationName: '${(0, utils_1.ucFirst)(op.name)}',
-        fieldName: '${op.name}',
-        ...buildCustomDocument('query', '${(0, utils_1.ucFirst)(op.name)}', '${op.name}', options?.select, undefined, []),
-      })`;
-            }
-        }
-    });
-    sourceFile.addFunction({
-        name: 'createQueryOperations',
-        isExported: true,
-        parameters: [{ name: 'client', type: 'OrmClient' }],
-        statements: `return {
-    ${operationMethods.join(',\n    ')},
-  };`,
-    });
+    const operationProperties = operations.map((op) => buildOperationMethod(op, 'query'));
+    const returnObj = t.objectExpression(operationProperties);
+    const returnStmt = t.returnStatement(returnObj);
+    const clientParam = t.identifier('client');
+    clientParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('OrmClient')));
+    const factoryFunc = t.functionDeclaration(t.identifier('createQueryOperations'), [clientParam], t.blockStatement([returnStmt]));
+    statements.push(t.exportNamedDeclaration(factoryFunc));
+    const header = (0, utils_1.getGeneratedFileHeader)('Custom query operations');
+    const code = (0, babel_ast_1.generateCode)(statements);
     return {
         fileName: 'query/index.ts',
-        content: (0, ts_ast_1.getFormattedOutput)(sourceFile),
+        content: header + '\n' + code,
     };
 }
 /**
  * Generate the mutation/index.ts file for custom mutation operations
  */
 function generateCustomMutationOpsFile(operations) {
-    const project = (0, ts_ast_1.createProject)();
-    const sourceFile = (0, ts_ast_1.createSourceFile)(project, 'index.ts');
+    const statements = [];
     // Collect all input type names and payload type names
     const inputTypeNames = collectInputTypeNamesFromOps(operations);
     const payloadTypeNames = collectPayloadTypeNamesFromOps(operations);
-    // Generate Select type names for payloads
     const selectTypeNames = payloadTypeNames.map((p) => `${p}Select`);
-    // Combine all type imports
-    const allTypeImports = [
-        ...new Set([...inputTypeNames, ...payloadTypeNames, ...selectTypeNames]),
-    ];
-    // Add file header
-    sourceFile.insertText(0, (0, ts_ast_1.createFileHeader)('Custom mutation operations') + '\n\n');
+    const allTypeImports = [...new Set([...inputTypeNames, ...payloadTypeNames, ...selectTypeNames])];
     // Add imports
-    sourceFile.addImportDeclarations([
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../client',
-            namedImports: ['OrmClient'],
-        }),
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../query-builder',
-            namedImports: ['QueryBuilder', 'buildCustomDocument'],
-        }),
-        (0, ts_ast_1.createImport)({
-            moduleSpecifier: '../select-types',
-            typeOnlyNamedImports: ['InferSelectResult'],
-        }),
-    ]);
-    // Import types from input-types if we have any
+    statements.push(createImportDeclaration('../client', ['OrmClient']));
+    statements.push(createImportDeclaration('../query-builder', ['QueryBuilder', 'buildCustomDocument']));
+    statements.push(createImportDeclaration('../select-types', ['InferSelectResult'], true));
     if (allTypeImports.length > 0) {
-        sourceFile.addImportDeclarations([
-            (0, ts_ast_1.createImport)({
-                moduleSpecifier: '../input-types',
-                typeOnlyNamedImports: allTypeImports,
-            }),
-        ]);
+        statements.push(createImportDeclaration('../input-types', allTypeImports, true));
     }
-    // Generate variable definitions type for each operation
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Variable Types');
-    sourceFile.addStatements('// ============================================================================\n');
+    // Generate variable interfaces
     for (const op of operations) {
-        if (op.args.length > 0) {
-            const varTypeName = `${(0, utils_1.ucFirst)(op.name)}Variables`;
-            const props = op.args.map((arg) => {
-                const optional = !(0, type_resolver_1.isTypeRequired)(arg.type);
-                return `${arg.name}${optional ? '?' : ''}: ${(0, type_resolver_1.typeRefToTsType)(arg.type)};`;
-            });
-            sourceFile.addStatements(`export interface ${varTypeName} {\n  ${props.join('\n  ')}\n}\n`);
-        }
+        const varInterface = createVariablesInterface(op);
+        if (varInterface)
+            statements.push(varInterface);
     }
     // Generate factory function
-    sourceFile.addStatements('\n// ============================================================================');
-    sourceFile.addStatements('// Mutation Operations Factory');
-    sourceFile.addStatements('// ============================================================================\n');
-    // Build the operations object
-    const operationMethods = operations.map((op) => {
-        const hasArgs = op.args.length > 0;
-        const varTypeName = `${(0, utils_1.ucFirst)(op.name)}Variables`;
-        const varDefs = op.args.map((arg) => ({
-            name: arg.name,
-            type: formatGraphQLType(arg.type),
-        }));
-        const varDefsJson = JSON.stringify(varDefs);
-        // Get Select type for return type
-        const selectTypeName = getSelectTypeName(op.returnType);
-        const payloadTypeName = (0, type_resolver_1.getTypeBaseName)(op.returnType);
-        // Use typed select if available, otherwise fall back to Record<string, unknown>
-        const selectType = selectTypeName ?? 'Record<string, unknown>';
-        const returnTypePart = selectTypeName && payloadTypeName
-            ? `{ ${op.name}: InferSelectResult<${payloadTypeName}, S> }`
-            : 'unknown';
-        if (hasArgs) {
-            if (selectTypeName) {
-                return `${op.name}: <const S extends ${selectType}>(args: ${varTypeName}, options?: { select?: S }) =>
-      new QueryBuilder<${returnTypePart}>({
-        client,
-        operation: 'mutation',
-        operationName: '${(0, utils_1.ucFirst)(op.name)}',
-        fieldName: '${op.name}',
-        ...buildCustomDocument('mutation', '${(0, utils_1.ucFirst)(op.name)}', '${op.name}', options?.select, args, ${varDefsJson}),
-      })`;
-            }
-            else {
-                return `${op.name}: (args: ${varTypeName}, options?: { select?: Record<string, unknown> }) =>
-      new QueryBuilder({
-        client,
-        operation: 'mutation',
-        operationName: '${(0, utils_1.ucFirst)(op.name)}',
-        fieldName: '${op.name}',
-        ...buildCustomDocument('mutation', '${(0, utils_1.ucFirst)(op.name)}', '${op.name}', options?.select, args, ${varDefsJson}),
-      })`;
-            }
-        }
-        else {
-            // No args - still provide typed select
-            if (selectTypeName) {
-                return `${op.name}: <const S extends ${selectType}>(options?: { select?: S }) =>
-      new QueryBuilder<${returnTypePart}>({
-        client,
-        operation: 'mutation',
-        operationName: '${(0, utils_1.ucFirst)(op.name)}',
-        fieldName: '${op.name}',
-        ...buildCustomDocument('mutation', '${(0, utils_1.ucFirst)(op.name)}', '${op.name}', options?.select, undefined, []),
-      })`;
-            }
-            else {
-                return `${op.name}: (options?: { select?: Record<string, unknown> }) =>
-      new QueryBuilder({
-        client,
-        operation: 'mutation',
-        operationName: '${(0, utils_1.ucFirst)(op.name)}',
-        fieldName: '${op.name}',
-        ...buildCustomDocument('mutation', '${(0, utils_1.ucFirst)(op.name)}', '${op.name}', options?.select, undefined, []),
-      })`;
-            }
-        }
-    });
-    sourceFile.addFunction({
-        name: 'createMutationOperations',
-        isExported: true,
-        parameters: [{ name: 'client', type: 'OrmClient' }],
-        statements: `return {
-    ${operationMethods.join(',\n    ')},
-  };`,
-    });
+    const operationProperties = operations.map((op) => buildOperationMethod(op, 'mutation'));
+    const returnObj = t.objectExpression(operationProperties);
+    const returnStmt = t.returnStatement(returnObj);
+    const clientParam = t.identifier('client');
+    clientParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('OrmClient')));
+    const factoryFunc = t.functionDeclaration(t.identifier('createMutationOperations'), [clientParam], t.blockStatement([returnStmt]));
+    statements.push(t.exportNamedDeclaration(factoryFunc));
+    const header = (0, utils_1.getGeneratedFileHeader)('Custom mutation operations');
+    const code = (0, babel_ast_1.generateCode)(statements);
     return {
         fileName: 'mutation/index.ts',
-        content: (0, ts_ast_1.getFormattedOutput)(sourceFile),
+        content: header + '\n' + code,
     };
 }
 /**

package/cli/codegen/orm/input-types-generator.d.ts

@@ -1,3 +1,15 @@
+/**
+ * Input types generator for ORM client (Babel AST-based)
+ *
+ * Generates TypeScript interfaces for:
+ * 1. Scalar filter types (StringFilter, IntFilter, UUIDFilter, etc.)
+ * 2. Entity interfaces (User, Order, etc.)
+ * 3. Table filter types (UserFilter, OrderFilter, etc.)
+ * 4. OrderBy enums (UsersOrderBy, OrdersOrderBy, etc.)
+ * 5. Input types (LoginInput, CreateUserInput, etc.)
+ *
+ * Uses Babel AST for robust code generation.
+ */
 import type { TypeRegistry, CleanArgument, CleanTable } from '../../../types/schema';
 export interface GeneratedInputTypesFile {
     fileName: string;
@@ -16,6 +28,6 @@ export declare function collectPayloadTypeNames(operations: Array<{
     returnType: CleanArgument['type'];
 }>): Set<string>;
 /**
- * Generate comprehensive input-types.ts file using ts-morph AST
+ * Generate comprehensive input-types.ts file using Babel AST
  */
 export declare function generateInputTypesFile(typeRegistry: TypeRegistry, usedInputTypes: Set<string>, tables?: CleanTable[], usedPayloadTypes?: Set<string>): GeneratedInputTypesFile;