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';