@cosmwasm/ts-codegen
Version:
@cosmwasm/ts-codegen converts your CosmWasm smart contracts into dev-friendly TypeScript classes so you can focus on shipping code.
80 lines (79 loc) • 3.36 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.MessageBuilderPlugin = void 0;
const w = __importStar(require("@cosmwasm/ts-codegen-ast"));
const ts_codegen_ast_1 = require("@cosmwasm/ts-codegen-ast");
const case_1 = require("case");
const utils_1 = require("../utils");
const plugin_base_1 = require("./plugin-base");
class MessageBuilderPlugin extends plugin_base_1.BuilderPluginBase {
initContext(contract, options) {
return new ts_codegen_ast_1.RenderContext(contract, options, this.builder?.builderContext);
}
async doRender(name, context) {
const { enabled } = this.options.messageBuilder;
if (!enabled) {
return;
}
const { schemas } = context.contract;
const localname = (0, case_1.pascal)(name) + '.message-builder.ts';
const TypesFile = (0, case_1.pascal)(name) + '.types';
const ExecuteMsg = (0, utils_1.findExecuteMsg)(schemas);
const typeHash = await (0, utils_1.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 = (0, ts_codegen_ast_1.getMessageProperties)(ExecuteMsg);
if (children.length > 0) {
const className = (0, case_1.pascal)(`${name}ExecuteMsgBuilder`);
body.push(w.createMessageBuilderClass(context, className, ExecuteMsg));
}
}
const QueryMsg = (0, utils_1.findQueryMsg)(schemas);
// query messages
if (QueryMsg) {
const children = (0, ts_codegen_ast_1.getMessageProperties)(QueryMsg);
if (children.length > 0) {
const className = (0, case_1.pascal)(`${name}QueryMsgBuilder`);
body.push(w.createMessageBuilderClass(context, className, QueryMsg));
}
}
if (typeHash.hasOwnProperty('Coin')) {
// @ts-ignore
delete context.utils.Coin;
}
return [
{
type: 'message-builder',
localname,
body
}
];
}
}
exports.MessageBuilderPlugin = MessageBuilderPlugin;