refactor(SNAP): refactor starkNet_getCurrentNetwork api

packages/starknet-snap/src/index.ts

@@ -13,7 +13,6 @@ import { declareContract } from './declareContract';
 import { estimateAccDeployFee } from './estimateAccountDeployFee';
 import { estimateFees } from './estimateFees';
 import { extractPublicKey } from './extractPublicKey';
-import { getCurrentNetwork } from './getCurrentNetwork';
 import { getErc20TokenBalance } from './getErc20TokenBalance';
 import { getStarkName } from './getStarkName';
 import { getStoredErc20Tokens } from './getStoredErc20Tokens';
@@ -48,6 +47,7 @@ import {
   switchNetwork,
   getDeploymentData,
   watchAsset,
+  getCurrentNetwork,
 } from './rpcs';
 import { sendTransaction } from './sendTransaction';
 import { signDeployAccountTransaction } from './signDeployAccountTransaction';
@@ -241,7 +241,7 @@ export const onRpcRequest: OnRpcRequestHandler = async ({ request }) => {
         );
 
       case 'starkNet_getCurrentNetwork':
-        return await getCurrentNetwork(apiParams);
+        return await getCurrentNetwork.execute(null);
 
       case 'starkNet_getStoredNetworks':
         return await getStoredNetworks(apiParams);

packages/starknet-snap/src/rpcs/get-current-network.test.ts

@@ -0,0 +1,38 @@
+import { NetworkStateManager } from '../state/network-state-manager';
+import type { Network } from '../types/snapState';
+import { STARKNET_MAINNET_NETWORK } from '../utils/constants';
+import { getCurrentNetwork } from './get-current-network';
+
+jest.mock('../utils/logger');
+
+describe('getCurrentNetwork', () => {
+  const mockNetworkStateManager = ({
+    currentNetwork = STARKNET_MAINNET_NETWORK,
+  }: {
+    currentNetwork?: Network;
+  }) => {
+    const getCurrentNetworkSpy = jest.spyOn(
+      NetworkStateManager.prototype,
+      'getCurrentNetwork',
+    );
+
+    getCurrentNetworkSpy.mockResolvedValue(currentNetwork);
+
+    return { getCurrentNetworkSpy };
+  };
+
+  it('return the selected network', async () => {
+    const currentNetwork = STARKNET_MAINNET_NETWORK;
+    const { getCurrentNetworkSpy } = mockNetworkStateManager({
+      currentNetwork,
+    });
+
+    const result = await getCurrentNetwork.execute(null);
+
+    expect(getCurrentNetworkSpy).toHaveBeenCalled();
+    expect(result).toStrictEqual({
+      name: currentNetwork.name,
+      chainId: currentNetwork.chainId,
+    });
+  });
+});

packages/starknet-snap/src/rpcs/get-current-network.ts

@@ -0,0 +1,58 @@
+import type { constants } from 'starknet';
+import type { Infer } from 'superstruct';
+import { literal, string, object } from 'superstruct';
+
+import { NetworkStateManager } from '../state/network-state-manager';
+import { RpcController, ChainIdStruct } from '../utils';
+
+export const GetCurrentNetworkRequestStruct = literal(null);
+
+export const GetCurrentNetworkResponseStruct = object({
+  name: string(),
+  chainId: ChainIdStruct,
+});
+
+export type GetCurrentNetworkParams = Infer<
+  typeof GetCurrentNetworkRequestStruct
+>;
+
+export type GetCurrentNetworkResponse = Infer<
+  typeof GetCurrentNetworkResponseStruct
+>;
+
+/**
+ * The RPC handler to get the current network.
+ */
+export class GetCurrentNetworkRpc extends RpcController<
+  GetCurrentNetworkParams,
+  GetCurrentNetworkResponse
+> {
+  protected requestStruct = GetCurrentNetworkRequestStruct;
+
+  protected responseStruct = GetCurrentNetworkResponseStruct;
+
+  /**
+   * Execute the get the current network.
+   *
+   * @param _
+   * @returns A promise that resolve to the current network.
+   */
+  async execute(
+    _: GetCurrentNetworkParams,
+  ): Promise<GetCurrentNetworkResponse> {
+    return super.execute(_);
+  }
+
+  protected async handleRequest(
+    _: GetCurrentNetworkParams,
+  ): Promise<GetCurrentNetworkResponse> {
+    const networkStateMgr = new NetworkStateManager();
+    const network = await networkStateMgr.getCurrentNetwork();
+    return {
+      name: network.name,
+      chainId: network.chainId as unknown as constants.StarknetChainId,
+    };
+  }
+}
+
+export const getCurrentNetwork = new GetCurrentNetworkRpc();

packages/starknet-snap/src/rpcs/index.ts

@@ -8,3 +8,4 @@ export * from './verify-signature';
 export * from './switch-network';
 export * from './get-deployment-data';
 export * from './watch-asset';
+export * from './get-current-network';