UNPKG

@cosmwasm/ts-codegen

Version:

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

46 lines (45 loc) 1.88 kB
import * as w from '@cosmwasm/ts-codegen-ast'; import { getMessageProperties, RenderContext, } from '@cosmwasm/ts-codegen-ast'; import { pascal } from 'case'; import { findAndParseTypes, findExecuteMsg } from '../utils'; import { BuilderPluginBase } from './plugin-base'; export class MessageComposerPlugin extends BuilderPluginBase { initContext(contract, options) { return new RenderContext(contract, options, this.builder?.builderContext); } async doRender(name, context) { const { enabled } = this.options.messageComposer; if (!enabled) { return; } const { schemas } = context.contract; const localname = pascal(name) + '.message-composer.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}`)); // execute messages if (ExecuteMsg) { const children = getMessageProperties(ExecuteMsg); if (children.length > 0) { const TheClass = pascal(`${name}MsgComposer`); const Interface = pascal(`${name}Msg`); body.push(w.createMessageComposerInterface(context, Interface, ExecuteMsg)); body.push(w.createMessageComposerClass(context, TheClass, Interface, ExecuteMsg)); context.addProviderInfo(name, w.PROVIDER_TYPES.MESSAGE_COMPOSER_TYPE, TheClass, localname); } } if (Object.prototype.hasOwnProperty.call(typeHash, 'Coin')) { // @ts-ignore delete context.utils.Coin; } return [ { type: 'message-composer', localname, body, }, ]; } }