@cosmology/ast
Version:
Cosmos TypeScript AST generation
126 lines (125 loc) • 5.59 kB
JavaScript
import * as t from "@babel/types";
import { objectPattern } from "../../../utils";
import { restoreExtension } from "@cosmology/utils";
export const rpcFuncArguments = () => {
return [
objectPattern([
t.objectProperty(t.identifier("rpc"), t.identifier("rpc"), false, true),
], t.tsTypeAnnotation(t.tsTypeLiteral([
t.tsPropertySignature(t.identifier("rpc"), t.tsTypeAnnotation(t.tsTypeReference(t.identifier("Rpc")))),
]))),
];
};
export const rpcClassArguments = () => {
return [
t.objectExpression([
t.objectProperty(t.identifier("rpc"), t.identifier("rpc"), false, true),
]),
];
};
export const rpcNewAwaitImport = (path, className) => {
return t.newExpression(t.memberExpression(t.awaitExpression(t.callExpression(t.import(), [t.stringLiteral(path)])), t.identifier(className), false), [t.identifier("rpc")]);
};
export const rpcNewTmAwaitImport = (path, className, clientName = "client", isNew = false) => {
const calleeExpr = t.memberExpression(t.awaitExpression(t.callExpression(t.import(), [t.stringLiteral(path)])), t.identifier(className), false);
const argsExpr = [t.identifier(clientName)];
return isNew
? t.newExpression(calleeExpr, argsExpr)
: t.callExpression(calleeExpr, argsExpr);
};
export const rpcRecursiveObjectProps = (names, leaf) => {
const [name, ...rest] = names;
let baseComponent;
if (names.length === 1) {
baseComponent = leaf ? leaf : t.identifier(name);
}
else {
baseComponent = rpcRecursiveObjectProps(rest, leaf);
}
return t.objectExpression([
t.objectProperty(t.identifier(name), baseComponent),
]);
};
export const rpcNestedImportObject = (obj, className, options) => {
if (typeof obj === "string") {
const path = restoreExtension(obj, options?.restoreImportExtension);
return rpcNewAwaitImport(path, className);
}
const keys = Object.keys(obj);
return t.objectExpression(keys.map((name) => {
return t.objectProperty(t.identifier(name), rpcNestedImportObject(obj[name], className, options));
}));
};
export const rpcTmNestedImportObject = (obj, className, clientName, options) => {
if (typeof obj === "string") {
const path = restoreExtension(obj, options?.restoreImportExtension);
return rpcNewTmAwaitImport(path, className, clientName, options?.isNew);
}
const keys = Object.keys(obj);
return t.objectExpression(keys.map((name) => {
return t.objectProperty(t.identifier(name), rpcTmNestedImportObject(obj[name], className, clientName, options));
}));
};
export const createScopedRpcFactory = (obj, identifier, className, options) => {
return t.exportNamedDeclaration(t.variableDeclaration("const", [
t.variableDeclarator(t.identifier(identifier), t.arrowFunctionExpression(rpcFuncArguments(),
//
rpcNestedImportObject(obj, className, options), true)),
]));
};
export const createScopedRpcTmFactory = (context, obj, identifier) => {
const newClientType = context.pluginValue("rpcClients.useConnectComet");
const extensions = context.pluginValue("rpcClients.extensions");
let functionParams;
const returnStatement = t.returnStatement(rpcTmNestedImportObject(obj, extensions ? "createRpcQueryExtension" : "createClientImpl", extensions ? "client" : "rpc", {
restoreImportExtension: context.options.restoreImportExtension,
}));
let functionStatements;
let awaitClientCreation;
if (extensions) {
// TODO: remove tendermint34client and options if not needed
if (newClientType) {
// use connectComet for dynamic client
context.addUtil("connectComet");
awaitClientCreation = t.callExpression(t.identifier("connectComet"), [t.identifier("rpcEndpoint")]);
}
else {
// use tendermint34 client
context.addUtil("Tendermint34Client");
awaitClientCreation = t.callExpression(t.memberExpression(t.identifier("Tendermint34Client"), t.identifier("connect")), [t.identifier("rpcEndpoint")]);
}
context.addUtil("HttpEndpoint");
context.addUtil("QueryClient");
functionParams = [
objectPattern([
t.objectProperty(t.identifier("rpcEndpoint"), t.identifier("rpcEndpoint"), false, true),
], t.tsTypeAnnotation(t.tsTypeLiteral([
t.tsPropertySignature(t.identifier("rpcEndpoint"), t.tsTypeAnnotation(t.tsUnionType([
t.tsStringKeyword(),
t.tsTypeReference(t.identifier("HttpEndpoint")),
]))),
]))),
];
functionStatements = [
t.variableDeclaration("const", [
t.variableDeclarator(t.identifier("tmClient"), t.awaitExpression(awaitClientCreation)),
]),
/////
t.variableDeclaration("const", [
t.variableDeclarator(t.identifier("client"), t.newExpression(t.identifier("QueryClient"), [
t.identifier("tmClient"),
])),
]),
returnStatement,
];
}
else {
functionParams = rpcFuncArguments();
functionStatements = [returnStatement];
}
return t.exportNamedDeclaration(t.variableDeclaration("const", [
t.variableDeclarator(
// createRPCQueryClient
t.identifier(identifier), t.arrowFunctionExpression(functionParams, t.blockStatement(functionStatements), true)),
]));
};