@cosmology/ast/src/encoding/proto/register/utils.ts

Cosmos TypeScript AST generation

import * as t from "@babel/types";
import { ProtoType } from "@cosmology/types";
import { ProtoParseContext } from "../../context";
import { getAminoTypeName, getTypeUrl } from "@cosmology/utils";

export const createRegisterObject = (
    context: ProtoParseContext,
    name: string,
    proto: ProtoType
) => {
    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: ProtoParseContext,
    name: string,
    proto: ProtoType
) => {
    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: ProtoParseContext,
    name: string,
    proto: ProtoType
) => {
    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()])
    );
};