add getDefaultTokenBalances

PolkaGate · Dec 30, 2023 · 13b70b5 · 13b70b5
1 parent 2c3b43d
commit 13b70b5
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 20 deletions.
diff --git a/packages/snap/src/index.ts b/packages/snap/src/index.ts
@@ -4,9 +4,8 @@ import { getAddress, signJSON, signRaw } from './rpc';
 import { getMetadataList, setMetadata } from './rpc/metadata';
 import { getKeyPair } from './util/getKeyPair';
 import { DEFAULT_CHAIN_NAME } from './defaults';
-import { getBalances } from './util/getBalance';
-import { getGenesisHash } from './chains';
 import { accountDemo } from './ui/accountDemo';
+import { getDefaultTokenBalances } from './util/getDefaultTokenBalances';
 
 export const onRpcRequest: OnRpcRequestHandler = async ({
   origin,
@@ -43,8 +42,7 @@ export const onRpcRequest: OnRpcRequestHandler = async ({
  */
 export const onHomePage: OnHomePageHandler = async () => {
   const { address } = await getKeyPair(DEFAULT_CHAIN_NAME);
-  const genesisHash = getGenesisHash('westend'); // For testing purposes
-  const balances = await getBalances(genesisHash, address);
+  const balances = await getDefaultTokenBalances(address);
 
   return {
     content: accountDemo(address, balances),

diff --git a/packages/snap/src/rpc/getAddress.ts b/packages/snap/src/rpc/getAddress.ts
@@ -1,8 +1,7 @@
 import { DEFAULT_CHAIN_NAME } from '../defaults';
 import { getKeyPair } from '../util/getKeyPair';
-import { getBalances } from '../util/getBalance';
-import { getGenesisHash } from '../chains';
 import { accountDemo } from '../ui/accountDemo';
+import { getDefaultTokenBalances } from '../util/getDefaultTokenBalances';
 
 export const getAddress = async (chainName?: string): Promise<string> => {
   const account = await getKeyPair(chainName || DEFAULT_CHAIN_NAME);
@@ -24,20 +23,7 @@ export const getAddress = async (chainName?: string): Promise<string> => {
  * @param address - The any chain address.
  */
 async function showAccount(address: string) {
-  const westendGenesisHash = getGenesisHash('westend'); // These will be changed when dropdown component will be available
-  const westendBalances = await getBalances(westendGenesisHash, address);
-
-  const polkadotGenesisHash = getGenesisHash('polkadot'); // These will be changed when dropdown component will be available
-  const polkadotBalances = await getBalances(polkadotGenesisHash, address);
-
-  const kusamaGenesisHash = getGenesisHash('kusama'); // These will be changed when dropdown component will be available
-  const kusamaBalances = await getBalances(kusamaGenesisHash, address);
-
-  const balances = {
-    westendBalances,
-    polkadotBalances,
-    kusamaBalances,
-  };
+  const balances = await getDefaultTokenBalances(address);
 
   /** to show the address to user */
   snap.request({

diff --git a/packages/snap/src/util/getDefaultTokenBalances.ts b/packages/snap/src/util/getDefaultTokenBalances.ts
@@ -0,0 +1,24 @@
+import { getGenesisHash } from '../chains';
+import { getBalances } from './getBalance';
+
+/**
+ * To get balances on the tree default chains, Polkadot, Kusama, and Westend.
+ *
+ * @param address - The any chain address.
+ */
+export async function getDefaultTokenBalances(address: string) {
+  const westendGenesisHash = getGenesisHash('westend'); // These will be changed when dropdown component will be available
+  const westendBalances = await getBalances(westendGenesisHash, address);
+
+  const polkadotGenesisHash = getGenesisHash('polkadot'); // These will be changed when dropdown component will be available
+  const polkadotBalances = await getBalances(polkadotGenesisHash, address);
+
+  const kusamaGenesisHash = getGenesisHash('kusama'); // These will be changed when dropdown component will be available
+  const kusamaBalances = await getBalances(kusamaGenesisHash, address);
+
+  return {
+    westendBalances,
+    polkadotBalances,
+    kusamaBalances,
+  };
+}