@dedot/codegen 0.0.1-alpha.34 → 0.0.1-alpha.35

package/typink/generator/QueryGen.js

@@ -0,0 +1,68 @@
+import { normalizeContractTypeDef, normalizeLabel, } from '@dedot/contracts';
+import { stringCamelCase } from '@dedot/utils';
+import { beautifySourceCode, commentBlock, compileTemplate } from '../../utils.js';
+export class QueryGen {
+    contractMetadata;
+    typesGen;
+    constructor(contractMetadata, typeGen) {
+        this.contractMetadata = contractMetadata;
+        this.typesGen = typeGen;
+    }
+    generate(useSubPaths = false) {
+        this.typesGen.clearCache();
+        this.typesGen.typeImports.addKnownType('GenericSubstrateApi');
+        this.typesGen.typeImports.addContractType('GenericContractQuery', 'GenericContractQueryCall', 'ContractCallOptions', 'GenericContractCallResult', 'ContractCallResult');
+        const { messages } = this.contractMetadata.spec;
+        const queryCallsOut = this.doGenerate(messages, 'ContractCallOptions');
+        const importTypes = this.typesGen.typeImports.toImports({ useSubPaths });
+        const template = compileTemplate('typink/templates/query.hbs');
+        return beautifySourceCode(template({ importTypes, queryCallsOut }));
+    }
+    doGenerate(messages, optionsTypeName) {
+        let callsOut = '';
+        messages.forEach((messageDef) => {
+            const { label, docs, selector, args } = messageDef;
+            callsOut += `${commentBlock(docs, '\n', args.map((arg) => `@param {${this.typesGen.generateType(arg.type.type, 1)}} ${stringCamelCase(arg.label)}`), optionsTypeName ? `@param {${optionsTypeName}} options` : '', '\n', `@selector ${selector}`)}`;
+            callsOut += `${normalizeLabel(label)}: ${this.generateMethodDef(messageDef)};\n\n`;
+        });
+        return callsOut;
+    }
+    generateMethodDef(def) {
+        const { args, returnType } = def;
+        this.importType(returnType.type);
+        args.forEach(({ type: { type } }) => this.importType(type));
+        const paramsOut = this.generateParamsOut(args);
+        const typeOut = this.typesGen.generateType(returnType.type, 0, true);
+        return `GenericContractQueryCall<ChainApi, (${paramsOut && `${paramsOut},`} options: ContractCallOptions) => Promise<GenericContractCallResult<${typeOut}, ContractCallResult<ChainApi>>>>`;
+    }
+    generateParamsOut(args) {
+        return args
+            .map(({ type: { type }, label }) => `${stringCamelCase(label)}: ${this.typesGen.generateType(type, 1)}`)
+            .join(', ');
+    }
+    importType(typeId) {
+        const contractType = this.contractMetadata.types[typeId];
+        const typeDef = normalizeContractTypeDef(this.contractMetadata.types[typeId].type.def);
+        if (this.typesGen.includedTypes[contractType.id]) {
+            this.typesGen.addTypeImport(this.typesGen.includedTypes[contractType.id].name);
+            return;
+        }
+        const { type, value } = typeDef;
+        switch (type) {
+            case 'Compact':
+            case 'Primitive':
+            case 'BitSequence':
+                return;
+            case 'Struct':
+                return value.fields.forEach(({ typeId }) => this.importType(typeId));
+            case 'Enum':
+                return value.members.forEach(({ fields }) => fields.map(({ typeId }) => this.importType(typeId)).flat());
+            case 'Tuple':
+                return value.fields.forEach((typeId) => this.importType(typeId));
+            case 'SizedVec':
+                return this.importType(value.typeParam);
+            case 'Sequence':
+                return this.importType(value.typeParam);
+        }
+    }
+}

package/typink/generator/TxGen.d.ts

@@ -0,0 +1,6 @@
+import { ContractMessage } from '@dedot/contracts';
+import { QueryGen } from './QueryGen.js';
+export declare class TxGen extends QueryGen {
+    generate(useSubPaths?: boolean): Promise<string>;
+    generateMethodDef(def: ContractMessage): string;
+}

package/typink/generator/TxGen.js

@@ -0,0 +1,21 @@
+import { beautifySourceCode, compileTemplate } from '../../utils.js';
+import { QueryGen } from './QueryGen.js';
+export class TxGen extends QueryGen {
+    generate(useSubPaths = false) {
+        this.typesGen.clearCache();
+        this.typesGen.typeImports.addKnownType('GenericSubstrateApi');
+        this.typesGen.typeImports.addContractType('GenericContractTx', 'GenericContractTxCall', 'ContractTxOptions', 'ContractSubmittableExtrinsic');
+        const { messages } = this.contractMetadata.spec;
+        const txMessages = messages.filter((one) => one.mutates);
+        const txCallsOut = this.doGenerate(txMessages, 'ContractTxOptions');
+        const importTypes = this.typesGen.typeImports.toImports({ useSubPaths });
+        const template = compileTemplate('typink/templates/tx.hbs');
+        return beautifySourceCode(template({ importTypes, txCallsOut }));
+    }
+    generateMethodDef(def) {
+        const { args } = def;
+        args.forEach(({ type: { type } }) => this.importType(type));
+        const paramsOut = this.generateParamsOut(args);
+        return `GenericContractTxCall<ChainApi, (${paramsOut && `${paramsOut},`} options: ContractTxOptions) => ContractSubmittableExtrinsic<ChainApi>>`;
+    }
+}

package/typink/generator/TypesGen.d.ts

@@ -0,0 +1,7 @@
+import { ContractMetadata } from '@dedot/contracts';
+import { BaseTypesGen } from '../../shared/index.js';
+export declare class TypesGen extends BaseTypesGen {
+    contractMetadata: ContractMetadata;
+    constructor(contractMetadata: ContractMetadata);
+    generate(useSubPaths?: boolean): Promise<string>;
+}

package/typink/generator/TypesGen.js

@@ -0,0 +1,24 @@
+import { extractContractTypes } from '@dedot/contracts';
+import { BaseTypesGen } from '../../shared/index.js';
+import { beautifySourceCode, compileTemplate } from '../../utils.js';
+const SKIP_TYPES = ['Result', 'Option'];
+export class TypesGen extends BaseTypesGen {
+    contractMetadata;
+    constructor(contractMetadata) {
+        super(extractContractTypes(contractMetadata));
+        this.contractMetadata = contractMetadata;
+        this.skipTypes = SKIP_TYPES;
+        this.includedTypes = this.includeTypes();
+    }
+    generate(useSubPaths = false) {
+        let defTypeOut = '';
+        Object.values(this.includedTypes)
+            .filter(({ skip, knownType }) => !(skip || knownType))
+            .forEach(({ nameOut, id }) => {
+            defTypeOut += `export type ${nameOut} = ${this.generateType(id, 0, true)};\n\n`;
+        });
+        const importTypes = this.typeImports.toImports({ excludeModules: ['./types'], useSubPaths });
+        const template = compileTemplate('typink/templates/types.hbs');
+        return beautifySourceCode(template({ importTypes, defTypeOut }));
+    }
+}

package/typink/generator/index.d.ts

@@ -0,0 +1,6 @@
+export * from './IndexGen.js';
+export * from './QueryGen.js';
+export * from './TypesGen.js';
+export * from './TxGen.js';
+export * from './ConstructorTxGen.js';
+export * from './ConstructorQueryGen.js';

package/typink/generator/index.js

@@ -0,0 +1,6 @@
+export * from './IndexGen.js';
+export * from './QueryGen.js';
+export * from './TypesGen.js';
+export * from './TxGen.js';
+export * from './ConstructorTxGen.js';
+export * from './ConstructorQueryGen.js';

package/typink/index.d.ts

@@ -0,0 +1,2 @@
+import { ContractMetadata } from '@dedot/contracts';
+export declare function generateContractTypes(metadata: ContractMetadata | string, contract?: string, outDir?: string, extension?: string, useSubPaths?: boolean): Promise<void>;

package/typink/index.js

@@ -0,0 +1,30 @@
+import { parseRawMetadata } from '@dedot/contracts';
+import fs from 'fs';
+import path from 'path';
+import { IndexGen, QueryGen, TxGen, TypesGen, ConstructorTxGen, ConstructorQueryGen } from './generator/index.js';
+export async function generateContractTypes(metadata, contract, outDir = '.', extension = 'd.ts', useSubPaths = false) {
+    let contractMetadata = typeof metadata === 'string' ? parseRawMetadata(metadata) : metadata;
+    contract = contract || contractMetadata.contract.name;
+    const dirPath = path.resolve(outDir, contract);
+    const typesFileName = path.join(dirPath, `types.${extension}`);
+    const queryTypesFileName = path.join(dirPath, `query.${extension}`);
+    const txTypesFileName = path.join(dirPath, `tx.${extension}`);
+    const constructorTxTypesFileName = path.join(dirPath, `constructor-tx.${extension}`);
+    const constructorQueryTypesFileName = path.join(dirPath, `constructor-query.${extension}`);
+    const indexTypesFileName = path.join(dirPath, `index.${extension}`);
+    if (!fs.existsSync(dirPath)) {
+        fs.mkdirSync(dirPath, { recursive: true });
+    }
+    const typesGen = new TypesGen(contractMetadata);
+    const queryGen = new QueryGen(contractMetadata, typesGen);
+    const txGen = new TxGen(contractMetadata, typesGen);
+    const constructorTxGen = new ConstructorTxGen(contractMetadata, typesGen);
+    const constructorQueryGen = new ConstructorQueryGen(contractMetadata, typesGen);
+    const indexGen = new IndexGen(contractMetadata);
+    fs.writeFileSync(typesFileName, await typesGen.generate(useSubPaths));
+    fs.writeFileSync(queryTypesFileName, await queryGen.generate(useSubPaths));
+    fs.writeFileSync(txTypesFileName, await txGen.generate(useSubPaths));
+    fs.writeFileSync(constructorQueryTypesFileName, await constructorQueryGen.generate(useSubPaths));
+    fs.writeFileSync(constructorTxTypesFileName, await constructorTxGen.generate(useSubPaths));
+    fs.writeFileSync(indexTypesFileName, await indexGen.generate(useSubPaths));
+}

package/typink/templates/constructor-query.hbs

@@ -0,0 +1,7 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export interface ConstructorQuery<ChainApi extends GenericSubstrateApi> extends GenericConstructorQuery<ChainApi> {
+    {{{ constructorsOut }}}
+}

package/typink/templates/constructor-tx.hbs

@@ -0,0 +1,7 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export interface ConstructorTx<ChainApi extends GenericSubstrateApi> extends GenericConstructorTx<ChainApi> {
+    {{{ constructorsOut }}}
+}

package/typink/templates/index.hbs

@@ -0,0 +1,14 @@
+// Generated by dedot cli
+
+{{{importTypes}}}
+import { ContractQuery } from './query';
+import { ContractTx } from './tx';
+import { ConstructorQuery } from './constructor-query';
+import { ConstructorTx } from './constructor-tx';
+
+export interface {{{contractName}}}<Rv extends RpcVersion = RpcV2, ChainApi extends VersionedGenericSubstrateApi = SubstrateApi> extends GenericContractApi<Rv, ChainApi> {
+    query: ContractQuery<ChainApi[Rv]>;
+    tx: ContractTx<ChainApi[Rv]>;
+    constructorQuery: ConstructorQuery<ChainApi[Rv]>;
+    constructorTx: ConstructorTx<ChainApi[Rv]>;
+}

package/typink/templates/query.hbs

@@ -0,0 +1,7 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export interface ContractQuery<ChainApi extends GenericSubstrateApi> extends GenericContractQuery<ChainApi> {
+    {{{ queryCallsOut }}}
+}

package/typink/templates/tx.hbs

@@ -0,0 +1,7 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export interface ContractTx<ChainApi extends GenericSubstrateApi> extends GenericContractTx<ChainApi> {
+    {{{ txCallsOut }}}
+}

package/typink/templates/types.hbs

@@ -0,0 +1,5 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+{{{ defTypeOut }}}

package/{generator/utils.d.ts → utils.d.ts}

@@ -2,7 +2,7 @@ export declare const WRAPPER_TYPE_REGEX: RegExp;
 export declare const TUPLE_TYPE_REGEX: RegExp;
 export declare const commentBlock: (...docs: (string | string[])[]) => string;
 export declare const beautifySourceCode: (source: string) => Promise<string>;
-export declare const compileTemplate: (templateFileName: string) => HandlebarsTemplateDelegate<any>;
+export declare const compileTemplate: (templateFile: string) => HandlebarsTemplateDelegate<any>;
 /**
  * Check if a word is TypeScript/JavaScript reserved
  * @param word

package/{generator/utils.js → utils.js}

@@ -7,7 +7,7 @@ export const WRAPPER_TYPE_REGEX = /^(\w+)<(.*)>$/;
 export const TUPLE_TYPE_REGEX = /^\[(.*)]$/;
 export const commentBlock = (...docs) => {
     const flatLines = docs.flat();
-    if (flatLines.length === 0) {
+    if (flatLines.length === 0 || !flatLines.some((o) => !!o)) {
         return '';
     }
     else {
@@ -19,11 +19,11 @@ ${flatLines.map((line) => `* ${line.replaceAll(/\s+/g, ' ').trim()}`).join('\n')
     }
 };
 export const beautifySourceCode = async (source) => {
-    const prettierOptions = await prettier.resolveConfig(path.resolve(currentDirname(), '../../../../.prettierrc.js'));
+    const prettierOptions = await prettier.resolveConfig(path.resolve(currentDirname(), '../../../.prettierrc.cjs'));
     return prettier.format(source, { parser: 'babel-ts', ...prettierOptions });
 };
-export const compileTemplate = (templateFileName) => {
-    const templateFilePath = path.resolve(currentDirname(), `../templates/${templateFileName}`);
+export const compileTemplate = (templateFile) => {
+    const templateFilePath = path.resolve(currentDirname(), templateFile);
     return handlebars.compile(fs.readFileSync(templateFilePath, 'utf8'));
 };
 // TODO add more reserved words