@metamask-previews/network-controller 23.1.0-preview-6a548b7c → 23.1.0-preview-9f91d514

package/dist/NetworkController.cjs

@@ -106,9 +106,27 @@ const controllerName = 'NetworkController';
  * Constructs a value for the state property `networkConfigurationsByChainId`
  * which will be used if it has not been provided to the constructor.
  *
+ * @param [additionalDefaultNetworks] - An array of Hex Chain IDs representing the additional networks to be included as default.
  * @returns The default value for `networkConfigurationsByChainId`.
  */
-function getDefaultNetworkConfigurationsByChainId() {
+function getDefaultNetworkConfigurationsByChainId(additionalDefaultNetworks = []) {
+    const infuraNetworks = getDefaultInfuraNetworkConfigurationsByChainId();
+    const customNetworks = getDefaultCustomNetworkConfigurationsByChainId();
+    return additionalDefaultNetworks.reduce((obj, chainId) => {
+        if ((0, utils_1.hasProperty)(customNetworks, chainId)) {
+            obj[chainId] = customNetworks[chainId];
+        }
+        return obj;
+    }, 
+    // Always include the infura networks in the default networks
+    infuraNetworks);
+}
+/**
+ * Constructs a `networkConfigurationsByChainId` object for all default Infura networks.
+ *
+ * @returns The `networkConfigurationsByChainId` object of all Infura networks.
+ */
+function getDefaultInfuraNetworkConfigurationsByChainId() {
     return Object.values(controller_utils_1.InfuraNetworkType).reduce((obj, infuraNetworkType) => {
         const chainId = controller_utils_1.ChainId[infuraNetworkType];
         const rpcEndpointUrl = 
@@ -133,15 +151,42 @@ function getDefaultNetworkConfigurationsByChainId() {
         return { ...obj, [chainId]: networkConfiguration };
     }, {});
 }
+/**
+ * Constructs a `networkConfigurationsByChainId` object for all default custom networks.
+ *
+ * @returns The `networkConfigurationsByChainId` object of all custom networks.
+ */
+function getDefaultCustomNetworkConfigurationsByChainId() {
+    const { ticker, rpcPrefs } = controller_utils_1.BUILT_IN_NETWORKS[controller_utils_1.BuiltInNetworkName.MegaETHTestnet];
+    return {
+        [controller_utils_1.ChainId[controller_utils_1.BuiltInNetworkName.MegaETHTestnet]]: {
+            blockExplorerUrls: [rpcPrefs.blockExplorerUrl],
+            chainId: controller_utils_1.ChainId[controller_utils_1.BuiltInNetworkName.MegaETHTestnet],
+            defaultRpcEndpointIndex: 0,
+            defaultBlockExplorerUrlIndex: 0,
+            name: controller_utils_1.NetworkNickname[controller_utils_1.BuiltInNetworkName.MegaETHTestnet],
+            nativeCurrency: ticker,
+            rpcEndpoints: [
+                {
+                    failoverUrls: [],
+                    networkClientId: controller_utils_1.BuiltInNetworkName.MegaETHTestnet,
+                    type: RpcEndpointType.Custom,
+                    url: controller_utils_1.BUILT_IN_CUSTOM_NETWORKS_RPC.MEGAETH_TESTNET,
+                },
+            ],
+        },
+    };
+}
 /**
  * Constructs properties for the NetworkController state whose values will be
  * used if not provided to the constructor.
  *
+ * @param [additionalDefaultNetworks] - An array of Hex Chain IDs representing the additional networks to be included as default.
  * @returns The default NetworkController state.
  */
-function getDefaultNetworkControllerState() {
+function getDefaultNetworkControllerState(additionalDefaultNetworks) {
     const networksMetadata = {};
-    const networkConfigurationsByChainId = getDefaultNetworkConfigurationsByChainId();
+    const networkConfigurationsByChainId = getDefaultNetworkConfigurationsByChainId(additionalDefaultNetworks);
     return {
         selectedNetworkClientId: controller_utils_1.InfuraNetworkType.mainnet,
         networksMetadata,
@@ -293,8 +338,11 @@ class NetworkController extends base_controller_1.BaseController {
      * @param options - The options; see {@link NetworkControllerOptions}.
      */
     constructor(options) {
-        const { messenger, state, infuraProjectId, log, getRpcServiceOptions } = options;
-        const initialState = { ...getDefaultNetworkControllerState(), ...state };
+        const { messenger, state, infuraProjectId, log, getRpcServiceOptions, additionalDefaultNetworks, } = options;
+        const initialState = {
+            ...getDefaultNetworkControllerState(additionalDefaultNetworks),
+            ...state,
+        };
         validateNetworkControllerState(initialState);
         if (!infuraProjectId || typeof infuraProjectId !== 'string') {
             throw new Error('Invalid Infura project ID');
@@ -368,10 +416,8 @@ class NetworkController extends base_controller_1.BaseController {
         // ESLint is mistaken here; `name` is a string.
         // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
         `${this.name}:getNetworkConfigurationByNetworkClientId`, this.getNetworkConfigurationByNetworkClientId.bind(this));
-        this.messagingSystem.registerActionHandler(
-        // TODO: Either fix this lint violation or explain why it's necessary to ignore.
-        // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
-        `${this.name}:getSelectedNetworkClient`, this.getSelectedNetworkClient.bind(this));
+        this.messagingSystem.registerActionHandler(`${this.name}:getSelectedNetworkClient`, this.getSelectedNetworkClient.bind(this));
+        this.messagingSystem.registerActionHandler(`${this.name}:getSelectedChainId`, this.getSelectedChainId.bind(this));
         this.messagingSystem.registerActionHandler(
         // ESLint is mistaken here; `name` is a string.
         // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
@@ -410,6 +456,15 @@ class NetworkController extends base_controller_1.BaseController {
         }
         return undefined;
     }
+    /**
+     * Accesses the chain ID from the selected network client.
+     *
+     * @returns The chain ID of the selected network client in hex format or undefined if there is no network client.
+     */
+    getSelectedChainId() {
+        const networkConfiguration = this.getNetworkConfigurationByNetworkClientId(this.state.selectedNetworkClientId);
+        return networkConfiguration?.chainId;
+    }
     /**
      * Internally, the Infura and custom network clients are categorized by type
      * so that when accessing either kind of network client, TypeScript knows