@cosmology/ast/module/encoding/proto/register/utils.js

Cosmos TypeScript AST generation

import * as t from "@babel/types";
import { getAminoTypeName, getTypeUrl } from "@cosmology/utils";
export const createRegisterObject = (context, name, proto) => {
    if (name === "Any") {
        return;
    }
    context.addUtil("GlobalDecoderRegistry");
    const typeUrl = getTypeUrl(context.ref.proto, proto);
    if (!typeUrl)
        return;
    return t.expressionStatement(t.callExpression(t.memberExpression(t.identifier("GlobalDecoderRegistry"), t.identifier("register")), [
        t.memberExpression(t.identifier(name), t.identifier("typeUrl")),
        t.identifier(name),
    ]));
};
export const createRegisterAminoProtoMapping = (context, name, proto) => {
    if (name === "Any") {
        return;
    }
    context.addUtil("GlobalDecoderRegistry");
    const str = getAminoTypeName(context, context.ref.proto, proto);
    if (!str || str.startsWith("/"))
        return;
    const typeUrl = getTypeUrl(context.ref.proto, proto);
    if (!typeUrl)
        return;
    return t.expressionStatement(t.callExpression(t.memberExpression(t.identifier("GlobalDecoderRegistry"), t.identifier("registerAminoProtoMapping")), [
        t.memberExpression(t.identifier(name), t.identifier("aminoType")),
        t.memberExpression(t.identifier(name), t.identifier("typeUrl")),
    ]));
};
/**
 * create ast: if(!GlobalDecoderRegistry.registerExistingTypeUrl(SignerInfo.typeUrl)){
 *    return;
 *  }
 * @param context
 * @param name
 * @returns
 */
export const createIfGlobalDecoderRegistryRegisterExistingTypeUrl = (context, name, proto) => {
    if (name === "Any") {
        return;
    }
    const typeUrl = getTypeUrl(context.ref.proto, proto);
    if (!typeUrl)
        return;
    context.addUtil("GlobalDecoderRegistry");
    return t.ifStatement(t.unaryExpression("!", t.callExpression(t.memberExpression(t.identifier("GlobalDecoderRegistry"), t.identifier("registerExistingTypeUrl")), [
        t.memberExpression(t.identifier(name), t.identifier("typeUrl")),
    ])), t.blockStatement([t.returnStatement()]));
};