@cosmwasm/ts-codegen
Version:
@cosmwasm/ts-codegen converts your CosmWasm smart contracts into dev-friendly TypeScript classes so you can focus on shipping code.
59 lines (58 loc) • 2.58 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, findQueryMsg } from '../utils';
import { BuilderPluginBase } from './plugin-base';
export class ClientPlugin extends BuilderPluginBase {
initContext(contract, options) {
return new RenderContext(contract, options, this.builder?.builderContext);
}
async doRender(name, context) {
const { enabled } = this.options.client;
if (!enabled) {
return;
}
const { schemas } = context.contract;
const localname = pascal(name) + '.client.ts';
const TypesFile = pascal(name) + '.types';
const QueryMsg = findQueryMsg(schemas);
const ExecuteMsg = findExecuteMsg(schemas);
const typeHash = await findAndParseTypes(schemas);
let Client = null;
let Instance = null;
let QueryClient = null;
let ReadOnlyInstance = null;
const body = [];
body.push(w.importStmt(Object.keys(typeHash), `./${TypesFile}`));
// query messages
if (QueryMsg) {
QueryClient = pascal(`${name}QueryClient`);
ReadOnlyInstance = pascal(`${name}ReadOnlyInterface`);
body.push(w.createQueryInterface(context, ReadOnlyInstance, QueryMsg));
body.push(w.createQueryClass(context, QueryClient, ReadOnlyInstance, QueryMsg));
context.addProviderInfo(name, w.PROVIDER_TYPES.QUERY_CLIENT_TYPE, QueryClient, localname);
}
// execute messages
if (ExecuteMsg) {
const children = getMessageProperties(ExecuteMsg);
if (children.length > 0) {
Client = pascal(`${name}Client`);
Instance = pascal(`${name}Interface`);
body.push(w.createExecuteInterface(context, Instance, this.options.client.execExtendsQuery ? ReadOnlyInstance : null, ExecuteMsg));
body.push(w.createExecuteClass(context, Client, Instance, this.options.client.execExtendsQuery ? QueryClient : null, ExecuteMsg));
context.addProviderInfo(name, w.PROVIDER_TYPES.SIGNING_CLIENT_TYPE, Client, localname);
}
}
if (Object.prototype.hasOwnProperty.call(typeHash, 'Coin')) {
// @ts-ignore
delete context.utils.Coin;
}
return [
{
type: 'client',
localname,
body,
},
];
}
}