@cosmology/ast
Version:
Cosmos TypeScript AST generation
55 lines (54 loc) • 2.47 kB
JavaScript
import * as t from '@babel/types';
import { grpcFuncArguments } from './grpc-web';
import { restoreExtension } from '@cosmology/utils';
// endpoint = endpoint.replace(/\/*$/g, "");
export const grpcGatewayClientScaffold = () => {
return [
t.expressionStatement(t.assignmentExpression("=", t.identifier('endpoint'), t.callExpression(t.memberExpression(t.identifier('endpoint'), t.identifier('replace')), [
t.regExpLiteral("\\/*$", "g"),
t.stringLiteral("")
])))
];
};
const grpcGatewayNewAwaitImport = (path, className, options) => {
return t.newExpression(t.memberExpression(t.awaitExpression(t.callExpression(t.import(), [
t.stringLiteral(restoreExtension(path, options?.restoreImportExtension))
])), t.identifier(className), false), [
t.identifier('endpoint')
]);
};
const grpcGatewayNestedImportObject = (obj, className, options) => {
//make className dynamic based on object
if (typeof obj === 'string') {
const serviceType = obj.split(".").pop();
switch (serviceType) {
case "Query":
className = 'QueryClientImpl';
// console.log("This is a Query RPC.");
break;
case "Service":
className = 'ServiceClientImpl';
// console.log("This is a Service RPC.");
break;
case "msg":
className = 'MsgClientImpl';
// console.log("This is a Message RPC.");
break;
default:
console.log("grpc service error!! This should not happend. Undefined service type");
}
return grpcGatewayNewAwaitImport(obj, className, options);
}
const keys = Object.keys(obj);
return t.objectExpression(keys.map(name => {
return t.objectProperty(t.identifier(name), grpcGatewayNestedImportObject(obj[name], className, options));
}));
};
export const createScopedGrpcGatewayFactory = (context, obj, identifier) => {
return t.exportNamedDeclaration(t.variableDeclaration('const', [
t.variableDeclarator(
// createGrpcGatewayQueryClient
t.identifier(identifier), t.arrowFunctionExpression(grpcFuncArguments(), t.blockStatement(grpcGatewayClientScaffold().concat(t.returnStatement(grpcGatewayNestedImportObject(obj, 'Query', //doesn't really matter here, remove if not prove useful
context.options)))), true))
]));
};