From 6702430fd2de510975772f0cd1dd350641a27eb9 Mon Sep 17 00:00:00 2001
From: Simon Warta <simon@warta.it>
Date: Wed, 7 Sep 2022 15:16:50 +0200
Subject: [PATCH] CReate initial Module type to collect symbols by module

---
 packages/stargate/src/modules/index.ts | 34 ++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/packages/stargate/src/modules/index.ts b/packages/stargate/src/modules/index.ts
index e96cf863ee..4ad8f4f1e9 100644
--- a/packages/stargate/src/modules/index.ts
+++ b/packages/stargate/src/modules/index.ts
@@ -1,3 +1,13 @@
+import { GeneratedType } from "@cosmjs/proto-signing";
+
+import { AminoConverters } from "../aminotypes";
+import { createStakingAminoConverters } from "./staking/aminomessages";
+import { stakingTypes } from "./staking/messages";
+import { setupStakingExtension } from "./staking/queries";
+import { setupTxExtension } from "./tx/queries";
+import { createVestingAminoConverters } from "./vesting/aminomessages";
+import { vestingTypes } from "./vesting/messages";
+
 export { AuthExtension, setupAuthExtension } from "./auth/queries";
 export { createAuthzAminoConverters } from "./authz/aminomessages";
 export { authzTypes } from "./authz/messages";
@@ -96,3 +106,27 @@ export {
   isAminoMsgCreateVestingAccount,
 } from "./vesting/aminomessages";
 export { vestingTypes } from "./vesting/messages";
+
+interface Module<CreateAminoConverters extends ((...args: any[]) => AminoConverters) | null, SetupQuery> {
+  readonly messageTypes: ReadonlyArray<[string, GeneratedType]> | null;
+  readonly createAminoConverters: CreateAminoConverters;
+  readonly setupQueryExtension: SetupQuery;
+}
+
+export const vesting: Module<typeof createVestingAminoConverters, null> = {
+  messageTypes: vestingTypes,
+  createAminoConverters: createVestingAminoConverters,
+  setupQueryExtension: null,
+};
+
+export const staking: Module<typeof createStakingAminoConverters, typeof setupStakingExtension> = {
+  messageTypes: stakingTypes,
+  createAminoConverters: createStakingAminoConverters,
+  setupQueryExtension: setupStakingExtension,
+};
+
+export const tx: Module<null, typeof setupTxExtension> = {
+  messageTypes: null,
+  createAminoConverters: null,
+  setupQueryExtension: setupTxExtension,
+};