@dedot/codegen 0.0.1-next.41a5fa17.16 → 0.0.1-next.49d80ab2.3

package/{generator → chaintypes/generator}/TxGen.js

@@ -1,11 +1,11 @@
-import { ApiGen } from '../generator/index.js';
 import { stringCamelCase, stringPascalCase } from '@dedot/utils';
-import { beautifySourceCode, commentBlock, compileTemplate, isReservedWord } from './utils.js';
+import { beautifySourceCode, commentBlock, compileTemplate, isReservedWord } from '../../utils.js';
+import { ApiGen } from '../generator/index.js';
 export class TxGen extends ApiGen {
-    generate() {
+    generate(useSubPaths = false) {
         const { pallets, types } = this.metadata;
         this.typesGen.clearCache();
-        this.typesGen.typeImports.addKnownType('GenericChainTx', 'GenericTxCall', 'ISubmittableExtrinsic', 'ISubmittableResult', 'IRuntimeTxCall');
+        this.typesGen.typeImports.addKnownType('GenericChainTx', 'GenericTxCall', 'ISubmittableExtrinsic', 'ISubmittableResult', 'IRuntimeTxCall', 'RpcVersion', 'RpcV2', 'ISubmittableExtrinsicLegacy', 'TransactionStatusLegacy', 'TransactionStatusV2');
         const { callTypeId, addressTypeId, signatureTypeId } = this.metadata.extrinsic;
         const callTypeIn = this.typesGen.generateType(callTypeId, 1);
         const addressTypeIn = this.typesGen.generateType(addressTypeId, 1);
@@ -17,11 +17,11 @@ export class TxGen extends ApiGen {
         for (let pallet of pallets) {
             if (pallet.calls === undefined)
                 continue;
-            const { type } = types[pallet.calls];
-            if (type.tag !== 'Enum')
+            const { typeDef } = types[pallet.calls];
+            if (typeDef.type !== 'Enum')
                 continue;
-            const isFlatEnum = type.value.members.every((m) => m.fields.length === 0);
-            const typedTxs = type.value.members
+            const isFlatEnum = typeDef.value.members.every((m) => m.fields.length === 0);
+            const typedTxs = typeDef.value.members
                 .map((one) => ({
                 functionName: stringCamelCase(one.name),
                 params: one.fields.map((f) => ({
@@ -41,22 +41,24 @@ export class TxGen extends ApiGen {
             txDefsOut += commentBlock(`Pallet \`${pallet.name}\`'s transaction calls`);
             txDefsOut += `${stringCamelCase(pallet.name)}: {
           ${typedTxs
-                .map(({ functionName, params, docs, callInput }) => `${commentBlock(docs, '\n', params.map((p) => `@param {${p.type}} ${p.normalizedName} ${p.docs}`))}${functionName}: GenericTxCall<(${params.map((p) => `${p.normalizedName}: ${p.type}`).join(', ')}) => ChainSubmittableExtrinsic<${callInput}>>`)
+                .map(({ functionName, params, docs, callInput }) => `${commentBlock(docs, '\n', params.map((p) => `@param {${p.type}} ${p.normalizedName} ${p.docs}`))}${functionName}: GenericTxCall<Rv, (${params.map((p) => `${p.normalizedName}: ${p.type}`).join(', ')}) => ChainSubmittableExtrinsic<Rv, ${callInput}>>`)
                 .join(',\n')}
             
-          ${commentBlock('Generic pallet tx call')}[callName: string]: GenericTxCall<TxCall>,
+          ${commentBlock('Generic pallet tx call')}[callName: string]: GenericTxCall<Rv, TxCall<Rv>>,
         },`;
         }
-        const importTypes = this.typesGen.typeImports.toImports();
+        const importTypes = this.typesGen.typeImports.toImports({ useSubPaths });
         // TODO make explicit separate type for Extra
         const defTypes = `
-    export type ChainSubmittableExtrinsic<T extends IRuntimeTxCall = ${callTypeIn}> = 
+    export type ChainSubmittableExtrinsic<Rv extends RpcVersion, T extends IRuntimeTxCall = ${callTypeIn}> = 
         Extrinsic<${addressTypeIn}, T, ${signatureTypeIn}, any[]> &
-        ISubmittableExtrinsic<ISubmittableResult<FrameSystemEventRecord>>
+        (Rv extends RpcV2
+          ? ISubmittableExtrinsic<ISubmittableResult<FrameSystemEventRecord, TransactionStatusV2>>
+          : ISubmittableExtrinsicLegacy<ISubmittableResult<FrameSystemEventRecord, TransactionStatusLegacy>>)
         
-    export type TxCall = (...args: any[]) => ChainSubmittableExtrinsic;    
+    export type TxCall<Rv extends RpcVersion> = (...args: any[]) => ChainSubmittableExtrinsic<Rv>;    
 `;
-        const template = compileTemplate('tx.hbs');
+        const template = compileTemplate('chaintypes/templates/tx.hbs');
         return beautifySourceCode(template({ importTypes, defTypes, txDefsOut }));
     }
     #normalizeParamName(name) {

package/chaintypes/generator/TypesGen.d.ts

@@ -0,0 +1,11 @@
+import { MetadataLatest, PortableRegistry, TypeId } from '@dedot/codecs';
+import { EnumOptions } from '@dedot/shape';
+import { BaseTypesGen } from '../../shared/index.js';
+export declare class TypesGen extends BaseTypesGen {
+    metadata: MetadataLatest;
+    registry: PortableRegistry;
+    constructor(metadata: MetadataLatest);
+    generate(useSubPaths?: boolean): Promise<string>;
+    shouldGenerateTypeIn(id: TypeId): boolean;
+    getEnumOptions(typeId: TypeId): EnumOptions;
+}

package/chaintypes/generator/TypesGen.js

@@ -0,0 +1,55 @@
+import { PortableRegistry } from '@dedot/codecs';
+import { BaseTypesGen } from '../../shared/index.js';
+import { beautifySourceCode, commentBlock, compileTemplate } from '../../utils.js';
+// Skip generate types for these
+// as we do have native types for them
+const SKIP_TYPES = [
+    'BoundedBTreeMap',
+    'BoundedBTreeSet',
+    'BoundedVec',
+    'Box',
+    'BTreeMap',
+    'BTreeSet',
+    'Cow',
+    'Option',
+    'Range',
+    'RangeInclusive',
+    'Result',
+    'WeakBoundedVec',
+    'WrapperKeepOpaque',
+    'WrapperOpaque',
+];
+export class TypesGen extends BaseTypesGen {
+    metadata;
+    registry;
+    constructor(metadata) {
+        super(metadata.types);
+        this.metadata = metadata;
+        this.skipTypes = SKIP_TYPES;
+        this.registry = new PortableRegistry(this.metadata);
+        this.includedTypes = this.includeTypes();
+    }
+    generate(useSubPaths = false) {
+        this.clearCache();
+        let defTypeOut = '';
+        Object.values(this.includedTypes)
+            .filter(({ skip, knownType }) => !(skip || knownType))
+            .forEach(({ name, nameOut, id, docs }) => {
+            defTypeOut += `${commentBlock(docs)}export type ${nameOut} = ${this.generateType(id, 0, true)};\n\n`;
+            if (this.shouldGenerateTypeIn(id)) {
+                defTypeOut += `export type ${name} = ${this.generateType(id)};\n\n`;
+            }
+        });
+        const importTypes = this.typeImports.toImports({ excludeModules: ['./types'], useSubPaths });
+        const template = compileTemplate('chaintypes/templates/types.hbs');
+        return beautifySourceCode(template({ importTypes, defTypeOut }));
+    }
+    shouldGenerateTypeIn(id) {
+        const { callTypeId } = this.metadata.extrinsic;
+        const palletCallTypeIds = this.registry.getFieldTypeIdsFromEnum(callTypeId);
+        return callTypeId === id || palletCallTypeIds.includes(id);
+    }
+    getEnumOptions(typeId) {
+        return this.registry.getEnumOptions(typeId);
+    }
+}

package/{generator → chaintypes/generator}/index.d.ts

@@ -2,7 +2,7 @@ export * from './TypesGen.js';
 export * from './ApiGen.js';
 export * from './ConstsGen.js';
 export * from './QueryGen.js';
-export * from './RpcGen.js';
+export * from './JsonRpcGen.js';
 export * from './IndexGen.js';
 export * from './ErrorsGen.js';
 export * from './EventsGen.js';

package/{generator → chaintypes/generator}/index.js

@@ -2,7 +2,7 @@ export * from './TypesGen.js';
 export * from './ApiGen.js';
 export * from './ConstsGen.js';
 export * from './QueryGen.js';
-export * from './RpcGen.js';
+export * from './JsonRpcGen.js';
 export * from './IndexGen.js';
 export * from './ErrorsGen.js';
 export * from './EventsGen.js';

package/chaintypes/index.d.ts

@@ -0,0 +1,3 @@
+import { MetadataLatest } from '@dedot/codecs';
+export declare function generateTypesFromEndpoint(chain: string, endpoint: string, outDir?: string, extension?: string, useSubPaths?: boolean): Promise<void>;
+export declare function generateTypes(chain: string, metadata: MetadataLatest, rpcMethods: string[], runtimeApis: Record<string, number>, outDir?: string, extension?: string, useSubPaths?: boolean): Promise<void>;

package/chaintypes/index.js

@@ -0,0 +1,49 @@
+import { LegacyClient } from '@dedot/api';
+import { WsProvider } from '@dedot/providers';
+import { stringCamelCase } from '@dedot/utils';
+import * as fs from 'fs';
+import * as path from 'path';
+import { ConstsGen, ErrorsGen, EventsGen, IndexGen, JsonRpcGen, QueryGen, RuntimeApisGen, TxGen, TypesGen, } from './generator/index.js';
+export async function generateTypesFromEndpoint(chain, endpoint, outDir, extension = 'd.ts', useSubPaths = false) {
+    const api = await LegacyClient.new(new WsProvider(endpoint));
+    const { methods } = await api.rpc.rpc_methods();
+    const apis = api.runtimeVersion.apis || {};
+    if (!chain) {
+        chain = stringCamelCase(api.runtimeVersion.specName || 'local');
+    }
+    await generateTypes(chain, api.metadata.latest, methods, apis, outDir, extension, useSubPaths);
+    await api.disconnect();
+}
+export async function generateTypes(chain, metadata, rpcMethods, runtimeApis, outDir = '.', extension = 'd.ts', useSubPaths = false) {
+    const dirPath = path.resolve(outDir, chain);
+    const defTypesFileName = path.join(dirPath, `types.${extension}`);
+    const constsTypesFileName = path.join(dirPath, `consts.${extension}`);
+    const queryTypesFileName = path.join(dirPath, `query.${extension}`);
+    const jsonRpcFileName = path.join(dirPath, `json-rpc.${extension}`);
+    const indexFileName = path.join(dirPath, `index.${extension}`);
+    const errorsFileName = path.join(dirPath, `errors.${extension}`);
+    const eventsFileName = path.join(dirPath, `events.${extension}`);
+    const runtimeApisFileName = path.join(dirPath, `runtime.${extension}`);
+    const txFileName = path.join(dirPath, `tx.${extension}`);
+    if (!fs.existsSync(dirPath)) {
+        fs.mkdirSync(dirPath, { recursive: true });
+    }
+    const typesGen = new TypesGen(metadata);
+    const constsGen = new ConstsGen(typesGen);
+    const queryGen = new QueryGen(typesGen);
+    const jsonRpcGen = new JsonRpcGen(typesGen, rpcMethods);
+    const indexGen = new IndexGen(chain);
+    const errorsGen = new ErrorsGen(typesGen);
+    const eventsGen = new EventsGen(typesGen);
+    const runtimeApisGen = new RuntimeApisGen(typesGen, runtimeApis);
+    const txGen = new TxGen(typesGen);
+    fs.writeFileSync(defTypesFileName, await typesGen.generate(useSubPaths));
+    fs.writeFileSync(errorsFileName, await errorsGen.generate(useSubPaths));
+    fs.writeFileSync(eventsFileName, await eventsGen.generate(useSubPaths));
+    fs.writeFileSync(jsonRpcFileName, await jsonRpcGen.generate(useSubPaths));
+    fs.writeFileSync(queryTypesFileName, await queryGen.generate(useSubPaths));
+    fs.writeFileSync(constsTypesFileName, await constsGen.generate(useSubPaths));
+    fs.writeFileSync(txFileName, await txGen.generate(useSubPaths));
+    fs.writeFileSync(indexFileName, await indexGen.generate(useSubPaths));
+    fs.writeFileSync(runtimeApisFileName, await runtimeApisGen.generate(useSubPaths));
+}

package/chaintypes/templates/consts.hbs

@@ -0,0 +1,7 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export interface ChainConsts<Rv extends RpcVersion> extends GenericChainConsts<Rv> {
+  {{{ defTypeOut }}}
+}

package/chaintypes/templates/errors.hbs

@@ -0,0 +1,7 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export interface ChainErrors<Rv extends RpcVersion> extends GenericChainErrors<Rv> {
+  {{{ defTypeOut }}}
+}

package/chaintypes/templates/events.hbs

@@ -0,0 +1,7 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export interface ChainEvents<Rv extends RpcVersion> extends GenericChainEvents<Rv> {
+  {{{ defTypeOut }}}
+}

package/chaintypes/templates/index.hbs

@@ -0,0 +1,27 @@
+// Generated by dedot cli
+
+{{{importTypes}}}
+import { ChainConsts } from './consts';
+import { ChainStorage } from './query';
+import { ChainJsonRpcApis } from './json-rpc';
+import { ChainErrors } from './errors';
+import { ChainEvents } from './events';
+import { RuntimeApis } from './runtime';
+import { ChainTx } from './tx';
+
+export * from './types';
+
+export interface Versioned{{{interfaceName}}}Api<Rv extends RpcVersion> extends GenericSubstrateApi<Rv> {
+  rpc: ChainJsonRpcApis<Rv>;
+  consts: ChainConsts<Rv>;
+  query: ChainStorage<Rv>;
+  errors: ChainErrors<Rv>;
+  events: ChainEvents<Rv>;
+  call: RuntimeApis<Rv>;
+  tx: ChainTx<Rv>;
+}
+
+export interface {{{interfaceName}}}Api {
+  legacy: Versioned{{{interfaceName}}}Api<RpcLegacy>;
+  v2: Versioned{{{interfaceName}}}Api<RpcV2>;
+}

package/chaintypes/templates/json-rpc.hbs

@@ -0,0 +1,5 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export type ChainJsonRpcApis<Rv extends RpcVersion> = Pick<JsonRpcApis, {{{jsonRpcMethods}}}> & GenericJsonRpcApis<Rv>;

package/chaintypes/templates/query.hbs

@@ -0,0 +1,7 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export interface ChainStorage<Rv extends RpcVersion> extends GenericChainStorage<Rv> {
+  {{{ defTypeOut }}}
+}

package/chaintypes/templates/runtime.hbs

@@ -0,0 +1,8 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+export interface RuntimeApis<Rv extends RpcVersion> extends GenericRuntimeApis<Rv> {
+{{{ runtimeCallsOut }}}
+}
+

package/chaintypes/templates/tx.hbs

@@ -0,0 +1,9 @@
+// Generated by dedot cli
+
+{{{ importTypes }}}
+
+{{{ defTypes }}}
+
+export interface ChainTx<Rv extends RpcVersion> extends GenericChainTx<Rv, TxCall<Rv>> {
+  {{{ txDefsOut }}}
+}

package/{cjs → chaintypes}/templates/types.hbs

@@ -1,4 +1,4 @@
-// Generated by @dedot/codegen
+// Generated by dedot cli
 
 {{{ importTypes }}}
 

package/cjs/{generator → chaintypes/generator}/ConstsGen.js

@@ -2,13 +2,13 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ConstsGen = void 0;
 const utils_1 = require("@dedot/utils");
-const index_js_1 = require("../generator/index.js");
-const utils_js_1 = require("./utils.js");
-class ConstsGen extends index_js_1.ApiGen {
-    generate() {
+const utils_js_1 = require("../../utils.js");
+const ApiGen_js_1 = require("./ApiGen.js");
+class ConstsGen extends ApiGen_js_1.ApiGen {
+    generate(useSubPaths = false) {
         const { pallets } = this.metadata;
         this.typesGen.clearCache();
-        this.typesGen.typeImports.addKnownType('GenericChainConsts');
+        this.typesGen.typeImports.addKnownType('GenericChainConsts', 'RpcVersion');
         let defTypeOut = '';
         for (let pallet of pallets) {
             const typedConstants = pallet.constants.map((one) => ({
@@ -23,8 +23,8 @@ class ConstsGen extends index_js_1.ApiGen {
         ${(0, utils_js_1.commentBlock)('Generic pallet constant')}[name: string]: any,
       },`;
         }
-        const importTypes = this.typesGen.typeImports.toImports();
-        const template = (0, utils_js_1.compileTemplate)('consts.hbs');
+        const importTypes = this.typesGen.typeImports.toImports({ useSubPaths });
+        const template = (0, utils_js_1.compileTemplate)('chaintypes/templates/consts.hbs');
         return (0, utils_js_1.beautifySourceCode)(template({ importTypes, defTypeOut }));
     }
 }

package/cjs/{generator → chaintypes/generator}/ErrorsGen.js

@@ -1,14 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ErrorsGen = void 0;
-const ApiGen_js_1 = require("./ApiGen.js");
-const utils_js_1 = require("./utils.js");
 const utils_1 = require("@dedot/utils");
+const utils_js_1 = require("../../utils.js");
+const ApiGen_js_1 = require("./ApiGen.js");
 class ErrorsGen extends ApiGen_js_1.ApiGen {
-    generate() {
+    generate(useSubPaths = false) {
         const { pallets } = this.metadata;
         this.typesGen.clearCache();
-        this.typesGen.typeImports.addKnownType('GenericChainErrors', 'GenericPalletError');
+        this.typesGen.typeImports.addKnownType('GenericChainErrors', 'GenericPalletError', 'RpcVersion');
         let defTypeOut = '';
         for (let pallet of pallets) {
             const errorTypeId = pallet.error;
@@ -19,21 +19,21 @@ class ErrorsGen extends ApiGen_js_1.ApiGen {
             defTypeOut += (0, utils_js_1.commentBlock)(`Pallet \`${pallet.name}\`'s errors`);
             defTypeOut += `${(0, utils_1.stringCamelCase)(pallet.name)}: {
         ${errorDefs
-                .map(({ name, docs }) => `${(0, utils_js_1.commentBlock)(docs)}${(0, utils_1.stringPascalCase)(name)}: GenericPalletError`)
+                .map(({ name, docs }) => `${(0, utils_js_1.commentBlock)(docs)}${(0, utils_1.stringPascalCase)(name)}: GenericPalletError<Rv>`)
                 .join(',\n')}
           
-        ${(0, utils_js_1.commentBlock)('Generic pallet error')}[error: string]: GenericPalletError,
+        ${(0, utils_js_1.commentBlock)('Generic pallet error')}[error: string]: GenericPalletError<Rv>,
       },`;
         }
-        const importTypes = this.typesGen.typeImports.toImports();
-        const template = (0, utils_js_1.compileTemplate)('errors.hbs');
+        const importTypes = this.typesGen.typeImports.toImports({ useSubPaths });
+        const template = (0, utils_js_1.compileTemplate)('chaintypes/templates/errors.hbs');
         return (0, utils_js_1.beautifySourceCode)(template({ importTypes, defTypeOut }));
     }
     #getErrorDefs(errorTypeId) {
         const def = this.metadata.types[errorTypeId];
         (0, utils_1.assert)(def, `Error def not found for id ${errorTypeId}`);
-        const { tag, value } = def.type;
-        (0, utils_1.assert)(tag === 'Enum', `Invalid pallet error type!`);
+        const { type, value } = def.typeDef;
+        (0, utils_1.assert)(type === 'Enum', 'Invalid pallet error type!');
         return value.members;
     }
 }

package/cjs/{generator → chaintypes/generator}/EventsGen.js

@@ -1,14 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.EventsGen = void 0;
-const ApiGen_js_1 = require("./ApiGen.js");
-const utils_js_1 = require("./utils.js");
 const utils_1 = require("@dedot/utils");
+const utils_js_1 = require("../../utils.js");
+const ApiGen_js_1 = require("./ApiGen.js");
 class EventsGen extends ApiGen_js_1.ApiGen {
-    generate() {
+    generate(useSubPaths = false) {
         const { pallets } = this.metadata;
         this.typesGen.clearCache();
-        this.typesGen.typeImports.addKnownType('GenericChainEvents', 'GenericPalletEvent');
+        this.typesGen.typeImports.addKnownType('GenericChainEvents', 'GenericPalletEvent', 'RpcVersion');
         let defTypeOut = '';
         for (let pallet of pallets) {
             const eventTypeId = pallet.event;
@@ -31,21 +31,21 @@ class EventsGen extends ApiGen_js_1.ApiGen {
                 }
                 return { ...def, genericParts };
             })
-                .map(({ name, docs, fields, genericParts }) => `${(0, utils_js_1.commentBlock)(docs)}${(0, utils_1.stringPascalCase)(name)}: GenericPalletEvent<${genericParts.join(', ')}>`)
+                .map(({ name, docs, fields, genericParts }) => `${(0, utils_js_1.commentBlock)(docs)}${(0, utils_1.stringPascalCase)(name)}: GenericPalletEvent<Rv, ${genericParts.join(', ')}>`)
                 .join(',\n')}
           
-        ${(0, utils_js_1.commentBlock)('Generic pallet event')}[prop: string]: GenericPalletEvent,
+        ${(0, utils_js_1.commentBlock)('Generic pallet event')}[prop: string]: GenericPalletEvent<Rv>,
       },`;
         }
-        const importTypes = this.typesGen.typeImports.toImports();
-        const template = (0, utils_js_1.compileTemplate)('events.hbs');
+        const importTypes = this.typesGen.typeImports.toImports({ useSubPaths });
+        const template = (0, utils_js_1.compileTemplate)('chaintypes/templates/events.hbs');
         return (0, utils_js_1.beautifySourceCode)(template({ importTypes, defTypeOut }));
     }
     #getEventDefs(typeId) {
         const def = this.metadata.types[typeId];
         (0, utils_1.assert)(def, `Event def not found for id ${typeId}`);
-        const { tag, value } = def.type;
-        (0, utils_1.assert)(tag === 'Enum', 'Invalid pallet event type!');
+        const { type, value } = def.typeDef;
+        (0, utils_1.assert)(type === 'Enum', 'Invalid pallet event type!');
         return value.members;
     }
     #generateMemberDefType(fields) {

package/cjs/chaintypes/generator/IndexGen.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IndexGen = void 0;
+const utils_1 = require("@dedot/utils");
+const index_js_1 = require("../../shared/index.js");
+const utils_js_1 = require("../../utils.js");
+class IndexGen {
+    chain;
+    constructor(chain) {
+        this.chain = chain;
+    }
+    async generate(useSubPaths = false) {
+        const interfaceName = (0, utils_1.stringPascalCase)(this.chain);
+        const typeImports = new index_js_1.TypeImports();
+        typeImports.addKnownType('GenericSubstrateApi', 'RpcLegacy', 'RpcV2', 'RpcVersion');
+        const importTypes = typeImports.toImports({ useSubPaths });
+        const template = (0, utils_js_1.compileTemplate)('chaintypes/templates/index.hbs');
+        return (0, utils_js_1.beautifySourceCode)(template({ importTypes, interfaceName }));
+    }
+}
+exports.IndexGen = IndexGen;

package/cjs/chaintypes/generator/JsonRpcGen.js

@@ -0,0 +1,59 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.JsonRpcGen = void 0;
+const api_1 = require("@dedot/api");
+const utils_js_1 = require("../../utils.js");
+const ApiGen_js_1 = require("./ApiGen.js");
+const HIDDEN_RPCS = [
+    // Ref: https://github.com/paritytech/polkadot-sdk/blob/43415ef58c143b985e09015cd000dbd65f6d3997/substrate/client/rpc-servers/src/lib.rs#L152C9-L158
+    'rpc_methods',
+];
+const ALIAS_RPCS = [
+    // TODO include these into the specs
+    'account_nextIndex',
+    'system_dryRunAt',
+    'state_callAt',
+    'state_getKeysPagedAt',
+    'state_getStorageAt',
+    'state_getStorageHashAt',
+    'state_getStorageSizeAt',
+    'chain_getRuntimeVersion',
+    'chain_getHead',
+    'chain_getFinalisedHead',
+    'childstate_getKeysPagedAt',
+    // Subscription pairs
+    'chain_subscribeRuntimeVersion',
+    'chain_unsubscribeRuntimeVersion',
+    'chain_unsubscribeNewHead',
+    'chain_subscribeNewHead',
+    'subscribe_newHead',
+    'unsubscribe_newHead',
+    'chain_subscribeFinalisedHeads',
+    'chain_unsubscribeFinalisedHeads',
+];
+class JsonRpcGen extends ApiGen_js_1.ApiGen {
+    typesGen;
+    rpcMethods;
+    constructor(typesGen, rpcMethods) {
+        super(typesGen);
+        this.typesGen = typesGen;
+        this.rpcMethods = rpcMethods;
+        HIDDEN_RPCS.filter((one) => !rpcMethods.includes(one)).forEach((one) => rpcMethods.push(one));
+        rpcMethods.sort();
+    }
+    generate(useSubPaths = false) {
+        this.typesGen.clearCache();
+        this.typesGen.typeImports.addKnownType('GenericJsonRpcApis', 'RpcVersion');
+        this.typesGen.typeImports.addKnownJsonRpcType('JsonRpcApis');
+        const toExclude = Object.values(api_1.subscriptionsInfo).flat();
+        const importTypes = this.typesGen.typeImports.toImports({ useSubPaths });
+        const template = (0, utils_js_1.compileTemplate)('chaintypes/templates/json-rpc.hbs');
+        const jsonRpcMethods = this.rpcMethods
+            .filter((one) => !ALIAS_RPCS.includes(one)) // exclude alias rpcs
+            .filter((one) => !toExclude.includes(one)) // exclude unsubscribe methods
+            .map((one) => `'${one}'`)
+            .join(' | ');
+        return (0, utils_js_1.beautifySourceCode)(template({ importTypes, jsonRpcMethods }));
+    }
+}
+exports.JsonRpcGen = JsonRpcGen;

package/cjs/{generator → chaintypes/generator}/QueryGen.js

@@ -2,40 +2,48 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.QueryGen = void 0;
 const utils_1 = require("@dedot/utils");
-const index_js_1 = require("../generator/index.js");
-const utils_js_1 = require("./utils.js");
-class QueryGen extends index_js_1.ApiGen {
-    generate() {
+const utils_js_1 = require("../../utils.js");
+const ApiGen_js_1 = require("./ApiGen.js");
+class QueryGen extends ApiGen_js_1.ApiGen {
+    generate(useSubPaths = false) {
         const { pallets } = this.metadata;
         this.typesGen.clearCache();
-        this.typesGen.typeImports.addKnownType('GenericChainStorage', 'GenericStorageQuery', 'Callback');
+        this.typesGen.typeImports.addKnownType('GenericChainStorage', 'GenericStorageQuery', 'Callback', 'RpcVersion');
         let defTypeOut = '';
         for (let pallet of pallets) {
             const storage = pallet.storage;
             if (!storage)
                 continue;
             const queries = storage.entries.map((one) => this.#generateEntry(one));
-            const queryDefs = queries.map(({ name, valueType, keyType, docs }) => `${(0, utils_js_1.commentBlock)(docs)}${name}: GenericStorageQuery<(${keyType}) => ${valueType}>`);
+            const queryDefs = queries.map(({ name, valueType, keyType, docs, keyTypeOut }) => {
+                if (keyTypeOut) {
+                    return `${(0, utils_js_1.commentBlock)(docs)}${name}: GenericStorageQuery<Rv, (${keyType}) => ${valueType}, ${keyTypeOut}>`;
+                }
+                else {
+                    return `${(0, utils_js_1.commentBlock)(docs)}${name}: GenericStorageQuery<Rv, (${keyType}) => ${valueType}>`;
+                }
+            });
             defTypeOut += (0, utils_js_1.commentBlock)(`Pallet \`${pallet.name}\`'s storage queries`);
             defTypeOut += `${(0, utils_1.stringCamelCase)(pallet.name)}: {
         ${queryDefs.join(',\n')}
         
-        ${(0, utils_js_1.commentBlock)('Generic pallet storage query')}[storage: string]: GenericStorageQuery;
+        ${(0, utils_js_1.commentBlock)('Generic pallet storage query')}[storage: string]: GenericStorageQuery<Rv>;
       },`;
         }
-        const importTypes = this.typesGen.typeImports.toImports();
-        const template = (0, utils_js_1.compileTemplate)('query.hbs');
+        const importTypes = this.typesGen.typeImports.toImports({ useSubPaths });
+        const template = (0, utils_js_1.compileTemplate)('chaintypes/templates/query.hbs');
         return (0, utils_js_1.beautifySourceCode)(template({ importTypes, defTypeOut }));
     }
     #generateEntry(entry) {
-        const { name, type, docs, modifier } = entry;
-        let valueType, keyType;
-        if (type.tag === 'Plain') {
-            valueType = this.typesGen.generateType(type.value.valueTypeId, 1, true);
+        const { name, storageType, docs, modifier } = entry;
+        let valueType, keyTypeOut, keyTypeIn;
+        if (storageType.type === 'Plain') {
+            valueType = this.typesGen.generateType(storageType.value.valueTypeId, 1, true);
         }
-        else if (type.tag === 'Map') {
-            valueType = this.typesGen.generateType(type.value.valueTypeId, 1, true);
-            keyType = this.typesGen.generateType(type.value.keyTypeId, 1);
+        else if (storageType.type === 'Map') {
+            valueType = this.typesGen.generateType(storageType.value.valueTypeId, 1, true);
+            keyTypeOut = this.typesGen.generateType(storageType.value.keyTypeId, 1, true);
+            keyTypeIn = this.typesGen.generateType(storageType.value.keyTypeId, 1);
         }
         else {
             throw Error('Invalid entry type!');
@@ -43,14 +51,15 @@ class QueryGen extends index_js_1.ApiGen {
         const isOptional = modifier === 'Optional';
         valueType = `${valueType}${isOptional ? ' | undefined' : ''}`;
         docs.push('\n');
-        if (keyType) {
-            docs.push(`@param {${keyType}} arg`);
+        if (keyTypeIn) {
+            docs.push(`@param {${keyTypeIn}} arg`);
         }
         docs.push(`@param {Callback<${valueType}> =} callback`);
         return {
             name: (0, utils_1.normalizeName)(name),
             valueType,
-            keyType: keyType ? `arg: ${keyType}` : '',
+            keyType: keyTypeIn ? `arg: ${keyTypeIn}` : '',
+            keyTypeOut,
             docs,
         };
     }