@metamask-previews/network-controller 23.1.0-preview-62c1e797 → 23.1.0-preview-49487623

package/dist/NetworkController.cjs

@@ -106,9 +106,32 @@ 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 [defaultNetworks] - An array of Hex Chain IDs representing the default networks to be included.
  * @returns The default value for `networkConfigurationsByChainId`.
  */
-function getDefaultNetworkConfigurationsByChainId() {
+function getDefaultNetworkConfigurationsByChainId(defaultNetworks) {
+    const infuraNetworks = getDefaultInfuraNetworkConfigurationsByChainId();
+    const customNetworks = getDefaultCustomNetworkConfigurationsByChainId();
+    // If no default networks are provided, use the default Infura networks
+    if (!defaultNetworks || defaultNetworks.length === 0) {
+        return infuraNetworks;
+    }
+    return defaultNetworks.reduce((obj, chainId) => {
+        if ((0, utils_1.hasProperty)(infuraNetworks, chainId)) {
+            obj[chainId] = infuraNetworks[chainId];
+        }
+        else if ((0, utils_1.hasProperty)(customNetworks, chainId)) {
+            obj[chainId] = customNetworks[chainId];
+        }
+        return obj;
+    }, {});
+}
+/**
+ * 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 +156,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 [defaultNetworks] - An array of Hex Chain IDs representing the default networks to be included.
  * @returns The default NetworkController state.
  */
-function getDefaultNetworkControllerState() {
+function getDefaultNetworkControllerState(defaultNetworks) {
     const networksMetadata = {};
-    const networkConfigurationsByChainId = getDefaultNetworkConfigurationsByChainId();
+    const networkConfigurationsByChainId = getDefaultNetworkConfigurationsByChainId(defaultNetworks);
     return {
         selectedNetworkClientId: controller_utils_1.InfuraNetworkType.mainnet,
         networksMetadata,
@@ -293,8 +343,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, defaultNetworks, } = options;
+        const initialState = {
+            ...getDefaultNetworkControllerState(defaultNetworks),
+            ...state,
+        };
         validateNetworkControllerState(initialState);
         if (!infuraProjectId || typeof infuraProjectId !== 'string') {
             throw new Error('Invalid Infura project ID');