@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
JavaScript
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,
},
];
}
}