@constructive-io/graphql-codegen 4.0.2 → 4.1.0

package/esm/core/codegen/cli/custom-command-generator.js

@@ -0,0 +1,119 @@
+import * as t from '@babel/types';
+import { toKebabCase } from 'komoji';
+import { generateCode } from '../babel-ast';
+import { getGeneratedFileHeader } from '../utils';
+import { buildQuestionsArray } from './arg-mapper';
+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 buildArgvType() {
+    return t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Partial'), t.tsTypeParameterInstantiation([
+        t.tsTypeReference(t.identifier('Record'), t.tsTypeParameterInstantiation([
+            t.tsStringKeyword(),
+            t.tsUnknownKeyword(),
+        ])),
+    ])));
+}
+function buildJsonLog(expr) {
+    return t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('console'), t.identifier('log')), [
+        t.callExpression(t.memberExpression(t.identifier('JSON'), t.identifier('stringify')), [expr, t.nullLiteral(), t.numericLiteral(2)]),
+    ]));
+}
+function buildErrorCatch(errorMessage) {
+    return t.catchClause(t.identifier('error'), t.blockStatement([
+        t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('console'), t.identifier('error')), [t.stringLiteral(errorMessage)])),
+        t.ifStatement(t.binaryExpression('instanceof', t.identifier('error'), t.identifier('Error')), t.blockStatement([
+            t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('console'), t.identifier('error')), [
+                t.memberExpression(t.identifier('error'), t.identifier('message')),
+            ])),
+        ])),
+        t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('process'), t.identifier('exit')), [t.numericLiteral(1)])),
+    ]));
+}
+function buildOrmCustomCall(opKind, opName, argsExpr) {
+    return t.callExpression(t.memberExpression(t.callExpression(t.memberExpression(t.memberExpression(t.identifier('client'), t.identifier(opKind)), t.identifier(opName)), [argsExpr]), t.identifier('execute')), []);
+}
+export function generateCustomCommand(op, options) {
+    const commandName = toKebabCase(op.name);
+    const opKind = op.kind === 'mutation' ? 'mutation' : 'query';
+    const statements = [];
+    const executorPath = options?.executorImportPath ?? '../executor';
+    const imports = ['getClient'];
+    if (options?.saveToken) {
+        imports.push('getStore');
+    }
+    statements.push(createImportDeclaration('inquirerer', ['CLIOptions', 'Inquirerer']));
+    statements.push(createImportDeclaration(executorPath, imports));
+    const questionsArray = op.args.length > 0
+        ? buildQuestionsArray(op.args)
+        : t.arrayExpression([]);
+    const argvParam = t.identifier('argv');
+    argvParam.typeAnnotation = buildArgvType();
+    const prompterParam = t.identifier('prompter');
+    prompterParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Inquirerer')));
+    const optionsParam = t.identifier('_options');
+    optionsParam.typeAnnotation = t.tsTypeAnnotation(t.tsTypeReference(t.identifier('CLIOptions')));
+    const bodyStatements = [
+        t.ifStatement(t.logicalExpression('||', t.memberExpression(t.identifier('argv'), t.identifier('help')), t.memberExpression(t.identifier('argv'), t.identifier('h'))), t.blockStatement([
+            t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('console'), t.identifier('log')), [
+                t.stringLiteral(`${commandName} - ${op.description || op.name}\n\nUsage: ${commandName} [OPTIONS]\n`),
+            ])),
+            t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('process'), t.identifier('exit')), [t.numericLiteral(0)])),
+        ])),
+    ];
+    if (op.args.length > 0) {
+        bodyStatements.push(t.variableDeclaration('const', [
+            t.variableDeclarator(t.identifier('answers'), t.awaitExpression(t.callExpression(t.memberExpression(t.identifier('prompter'), t.identifier('prompt')), [t.identifier('argv'), questionsArray]))),
+        ]));
+    }
+    const getClientArgs = options?.targetName
+        ? [t.stringLiteral(options.targetName)]
+        : [];
+    bodyStatements.push(t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier('client'), t.callExpression(t.identifier('getClient'), getClientArgs)),
+    ]));
+    const argsExpr = op.args.length > 0
+        ? t.identifier('answers')
+        : t.objectExpression([]);
+    bodyStatements.push(t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier('result'), t.awaitExpression(buildOrmCustomCall(opKind, op.name, argsExpr))),
+    ]));
+    if (options?.saveToken) {
+        bodyStatements.push(t.ifStatement(t.logicalExpression('&&', t.memberExpression(t.identifier('argv'), t.identifier('saveToken')), t.identifier('result')), t.blockStatement([
+            t.variableDeclaration('const', [
+                t.variableDeclarator(t.identifier('tokenValue'), t.logicalExpression('||', t.memberExpression(t.identifier('result'), t.identifier('token'), false), t.logicalExpression('||', t.memberExpression(t.identifier('result'), t.identifier('jwtToken'), false), t.memberExpression(t.identifier('result'), t.identifier('accessToken'), false)))),
+            ]),
+            t.ifStatement(t.identifier('tokenValue'), t.blockStatement([
+                t.variableDeclaration('const', [
+                    t.variableDeclarator(t.identifier('s'), t.callExpression(t.identifier('getStore'), [])),
+                ]),
+                t.variableDeclaration('const', [
+                    t.variableDeclarator(t.identifier('ctx'), t.callExpression(t.memberExpression(t.identifier('s'), t.identifier('getCurrentContext')), [])),
+                ]),
+                t.ifStatement(t.identifier('ctx'), t.blockStatement([
+                    t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('s'), t.identifier('setCredentials')), [
+                        t.memberExpression(t.identifier('ctx'), t.identifier('name')),
+                        t.objectExpression([
+                            t.objectProperty(t.identifier('token'), t.identifier('tokenValue')),
+                        ]),
+                    ])),
+                    t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('console'), t.identifier('log')), [t.stringLiteral('Token saved to current context.')])),
+                ])),
+            ])),
+        ])));
+    }
+    bodyStatements.push(buildJsonLog(t.identifier('result')));
+    const tryBlock = t.tryStatement(t.blockStatement(bodyStatements), buildErrorCatch(`Failed: ${op.name}`));
+    statements.push(t.exportDefaultDeclaration(t.arrowFunctionExpression([argvParam, prompterParam, optionsParam], t.blockStatement([tryBlock]), true)));
+    const header = getGeneratedFileHeader(`CLI command for ${op.kind} ${op.name}`);
+    const code = generateCode(statements);
+    return {
+        fileName: options?.targetName
+            ? `commands/${options.targetName}/${commandName}.ts`
+            : `commands/${commandName}.ts`,
+        content: header + '\n' + code,
+    };
+}

package/esm/core/codegen/cli/docs-generator.d.ts

@@ -0,0 +1,26 @@
+import type { CleanTable, CleanOperation } from '../../../types/schema';
+import type { GeneratedDocFile, McpTool } from '../docs-utils';
+export { resolveDocsConfig } from '../docs-utils';
+export type { GeneratedDocFile, McpTool } from '../docs-utils';
+export declare function generateReadme(tables: CleanTable[], customOperations: CleanOperation[], toolName: string): GeneratedDocFile;
+export declare function generateAgentsDocs(tables: CleanTable[], customOperations: CleanOperation[], toolName: string): GeneratedDocFile;
+export declare function getCliMcpTools(tables: CleanTable[], customOperations: CleanOperation[], toolName: string): McpTool[];
+export declare function generateSkills(tables: CleanTable[], customOperations: CleanOperation[], toolName: string): GeneratedDocFile[];
+export interface MultiTargetDocsInput {
+    toolName: string;
+    builtinNames: {
+        auth: string;
+        context: string;
+    };
+    targets: Array<{
+        name: string;
+        endpoint: string;
+        tables: CleanTable[];
+        customOperations: CleanOperation[];
+        isAuthTarget?: boolean;
+    }>;
+}
+export declare function generateMultiTargetReadme(input: MultiTargetDocsInput): GeneratedDocFile;
+export declare function generateMultiTargetAgentsDocs(input: MultiTargetDocsInput): GeneratedDocFile;
+export declare function getMultiTargetCliMcpTools(input: MultiTargetDocsInput): McpTool[];
+export declare function generateMultiTargetSkills(input: MultiTargetDocsInput): GeneratedDocFile[];