@cosmwasm/ts-codegen/esm/plugins/message-builder.js

@cosmwasm/ts-codegen converts your CosmWasm smart contracts into dev-friendly TypeScript classes so you can focus on shipping code.

import * as w from '@cosmwasm/ts-codegen-ast';
import { getMessageProperties, RenderContext, } from '@cosmwasm/ts-codegen-ast';
import { pascal } from 'case';
import { findAndParseTypes, findExecuteMsg, findQueryMsg } from '../utils';
import { BuilderPluginBase } from './plugin-base';
export class MessageBuilderPlugin extends BuilderPluginBase {
    initContext(contract, options) {
        return new RenderContext(contract, options, this.builder?.builderContext);
    }
    async doRender(name, context) {
        const { enabled } = this.options.messageBuilder;
        if (!enabled) {
            return;
        }
        const { schemas } = context.contract;
        const localname = pascal(name) + '.message-builder.ts';
        const TypesFile = pascal(name) + '.types';
        const ExecuteMsg = findExecuteMsg(schemas);
        const typeHash = await findAndParseTypes(schemas);
        const body = [];
        body.push(w.importStmt(Object.keys(typeHash), `./${TypesFile}`));
        body.push(w.importStmt(['CamelCasedProperties'], 'type-fest'));
        // execute messages
        if (ExecuteMsg) {
            const children = getMessageProperties(ExecuteMsg);
            if (children.length > 0) {
                const className = pascal(`${name}ExecuteMsgBuilder`);
                body.push(w.createMessageBuilderClass(context, className, ExecuteMsg));
            }
        }
        const QueryMsg = findQueryMsg(schemas);
        // query messages
        if (QueryMsg) {
            const children = getMessageProperties(QueryMsg);
            if (children.length > 0) {
                const className = pascal(`${name}QueryMsgBuilder`);
                body.push(w.createMessageBuilderClass(context, className, QueryMsg));
            }
        }
        if (Object.prototype.hasOwnProperty.call(typeHash, 'Coin')) {
            // @ts-ignore
            delete context.utils.Coin;
        }
        return [
            {
                type: 'message-builder',
                localname,
                body,
            },
        ];
    }
}