@cosmology/ast
Version:
Cosmos TypeScript AST generation
36 lines (35 loc) • 1.5 kB
JavaScript
import * as t from '@babel/types';
import { toAminoJsonMethod } from '../to-amino-json';
import { fromAminoJsonMethod } from '../from-amino-json';
import { getTypeUrl, getAminoTypeName } from '../utils';
export const createAminoConverterItem = ({ root, context, proto }) => {
const typeUrl = getTypeUrl(root, proto);
const aminoType = getAminoTypeName(context, root, proto);
const toAmino = !context.options.aminoEncoding.useLegacyInlineEncoding ?
t.memberExpression(t.identifier(proto.name), t.identifier('toAmino')) : toAminoJsonMethod({
context,
proto
});
const fromAmino = !context.options.aminoEncoding.useLegacyInlineEncoding ?
t.memberExpression(t.identifier(proto.name), t.identifier('fromAmino')) : fromAminoJsonMethod({
context,
proto
});
return t.objectProperty(t.stringLiteral(typeUrl), t.objectExpression([
t.objectProperty(t.identifier('aminoType'), t.stringLiteral(aminoType)),
t.objectProperty(t.identifier('toAmino'), toAmino),
t.objectProperty(t.identifier('fromAmino'), fromAmino)
]));
};
export const createAminoConverter = ({ name, root, context, protos }) => {
const items = protos.map(proto => {
return createAminoConverterItem({
context,
root,
proto
});
});
return t.exportNamedDeclaration(t.variableDeclaration('const', [
t.variableDeclarator(t.identifier(name), t.objectExpression(items))
]));
};