Use Map to maintain delegator environment overrides

Author: jeffsmale90

packages/delegation-toolkit/src/delegatorEnvironment.ts

@@ -47,17 +47,10 @@ import { deployContract } from './write';
 type SupportedVersion = '1.0.0' | '1.1.0' | '1.2.0' | '1.3.0';
 export const PREFERRED_VERSION: SupportedVersion = '1.3.0';
 
-type DeployedEnvironments = Record<
-  SupportedVersion,
-  Record<number, DeleGatorEnvironment>
->;
+const contractOverrideMap: Map<string, DeleGatorEnvironment> = new Map();
 
-const contractOverrides: DeployedEnvironments = {
-  '1.0.0': {},
-  '1.1.0': {},
-  '1.2.0': {},
-  '1.3.0': {},
-};
+const getContractOverrideKey = (chainId: number, version: SupportedVersion) =>
+  `${version}:${chainId}`;
 
 /**
  * Overrides the deployed environment for a specific chain and version.
@@ -70,7 +63,10 @@ export function overrideDeployedEnvironment(
   version: SupportedVersion,
   environment: DeleGatorEnvironment,
 ) {
-  contractOverrides[version][chainId] = environment;
+  contractOverrideMap.set(
+    getContractOverrideKey(chainId, version),
+    environment,
+  );
 }
 
 /**
@@ -83,9 +79,10 @@ export function getDeleGatorEnvironment(
   chainId: number,
   version: SupportedVersion = PREFERRED_VERSION,
 ) {
-  const override = contractOverrides[version][chainId];
-  if (override) {
-    return override;
+  const overrideKey = getContractOverrideKey(chainId, version);
+
+  if (contractOverrideMap.has(overrideKey)) {
+    return contractOverrideMap.get(overrideKey);
   }
 
   const contracts = DELEGATOR_CONTRACTS[version]?.[chainId];