chore: handle comments

Consensys · Dec 9, 2024 · 0674b12 · 0674b12
1 parent e224437
commit 0674b12
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 25 deletions.
diff --git a/packages/starknet-snap/src/config.ts b/packages/starknet-snap/src/config.ts
@@ -22,6 +22,9 @@ export type SnapConfig = {
   defaultNetwork: Network;
   availableNetworks: Network[];
   preloadTokens: Erc20Token[];
+  rpcEndpoint: {
+    [key: string]: string;
+  };
   explorer: {
     [key: string]: string;
   };
@@ -49,6 +52,15 @@ export const Config: SnapConfig = {
     STARKNET_SEPOLIA_TESTNET_NETWORK,
   ],
 
+  rpcEndpoint: {
+    [constants.StarknetChainId.SN_MAIN]:
+      // eslint-disable-next-line no-restricted-globals
+      `https://starknet-mainnet.infura.io/v3/${process.env.DIN_API_KEY ?? ''}`,
+    [constants.StarknetChainId.SN_SEPOLIA]:
+      // eslint-disable-next-line no-restricted-globals
+      `https://starknet-sepolia.infura.io/v3/${process.env.DIN_API_KEY ?? ''}`,
+  },
+
   explorer: {
     [constants.StarknetChainId.SN_MAIN]:
       // eslint-disable-next-line no-template-curly-in-string

diff --git a/packages/starknet-snap/src/utils/din.ts b/packages/starknet-snap/src/utils/din.ts
diff --git a/packages/starknet-snap/src/utils/din.test.ts → ...rknet-snap/src/utils/rpc-provider.test.ts b/packages/starknet-snap/src/utils/din.test.ts → ...rknet-snap/src/utils/rpc-provider.test.ts
@@ -1,6 +1,6 @@
 import { constants } from 'starknet';
 
-import { getRPCUrl } from './din';
+import { getRPCUrl } from './rpc-provider';
 
 describe('getRPCUrl', () => {
   it('returns Mainnet RPC URL if chain id is Mainnet', () => {

diff --git a/packages/starknet-snap/src/utils/rpc-provider.ts b/packages/starknet-snap/src/utils/rpc-provider.ts
@@ -0,0 +1,17 @@
+import { constants } from 'starknet';
+
+import { Config } from '../config';
+
+/**
+ *
+ * @param chainId
+ */
+export function getRPCUrl(chainId: string) {
+  switch (chainId) {
+    case constants.StarknetChainId.SN_MAIN:
+      return Config.rpcEndpoint[constants.StarknetChainId.SN_MAIN];
+    default:
+    case constants.StarknetChainId.SN_SEPOLIA:
+      return Config.rpcEndpoint[constants.StarknetChainId.SN_SEPOLIA];
+  }
+}
diff --git a/packages/starknet-snap/src/utils/starknetUtils.ts b/packages/starknet-snap/src/utils/starknetUtils.ts
@@ -69,11 +69,11 @@ import {
   ETHER_SEPOLIA_TESTNET,
   BlockIdentifierEnum,
 } from './constants';
-import { getRPCUrl } from './din';
 import { DeployRequiredError, UpgradeRequiredError } from './exceptions';
 import { hexToString } from './formatter-utils';
 import { getAddressKey } from './keyPair';
 import { logger } from './logger';
+import { getRPCUrl } from './rpc-provider';
 import { toJson } from './serializer';
 import {
   getAccount,