feat: Update Polygon from MATIC to POL

app/scripts/migrations/121.2.test.ts

@@ -0,0 +1,274 @@
+import { cloneDeep } from 'lodash';
+import { migrate, version } from './121.2';
+
+const oldVersion = 121.1;
+
+describe(`migration #${version}`, () => {
+  it('updates the version metadata', async () => {
+    const oldStorage = {
+      meta: { version: oldVersion },
+      data: {},
+    };
+
+    const newStorage = await migrate(oldStorage);
+
+    expect(newStorage.meta).toStrictEqual({ version });
+  });
+
+  it('Does nothing if `networkConfigurations` or `providerConfig` are not in the network controller state', async () => {
+    const oldState = {
+      NetworkController: {
+        selectedNetworkClientId: 'mainnet',
+      },
+    };
+
+    const transformedState = await migrate({
+      meta: { version: oldVersion },
+      data: cloneDeep(oldState),
+    });
+
+    expect(transformedState.data).toStrictEqual(oldState);
+  });
+
+  it('Updates MATIC ticker to POL and updates imageURL in networkConfigurations', async () => {
+    const oldState = {
+      NetworkController: {
+        networkConfigurations: {
+          '0x89': {
+            chainId: '0x89',
+            ticker: 'MATIC',
+            rpcPrefs: {
+              imageUrl: './images/matic-token.svg',
+            },
+          },
+        },
+      },
+    };
+
+    const expectedState = {
+      NetworkController: {
+        networkConfigurations: {
+          '0x89': {
+            chainId: '0x89',
+            ticker: 'POL',
+            rpcPrefs: {
+              imageUrl: './images/pol-token.svg',
+            },
+          },
+        },
+      },
+    };
+
+    const transformedState = await migrate({
+      meta: { version: oldVersion },
+      data: cloneDeep(oldState),
+    });
+
+    expect(transformedState.data).toStrictEqual(expectedState);
+  });
+
+  it('Does not update ticker to POL if ticker is not MATIC, but still updates imageURL in networkConfigurations', async () => {
+    const oldState = {
+      NetworkController: {
+        networkConfigurations: {
+          '0x89': {
+            chainId: '0x89',
+            ticker: 'NOT_MATIC',
+            rpcPrefs: {
+              imageUrl: './images/matic-token.svg',
+            },
+          },
+        },
+      },
+    };
+
+    const expectedState = {
+      NetworkController: {
+        networkConfigurations: {
+          '0x89': {
+            chainId: '0x89',
+            ticker: 'NOT_MATIC',
+            rpcPrefs: {
+              imageUrl: './images/pol-token.svg',
+            },
+          },
+        },
+      },
+    };
+
+    const transformedState = await migrate({
+      meta: { version: oldVersion },
+      data: cloneDeep(oldState),
+    });
+
+    expect(transformedState.data).toStrictEqual(expectedState);
+  });
+
+  it('Does not update tickers for other network configurations, updates only ticker and imageURL for chain 0x89', async () => {
+    const oldState = {
+      NetworkController: {
+        networkConfigurations: {
+          '0x89': {
+            chainId: '0x89',
+            ticker: 'MATIC',
+            rpcPrefs: {
+              imageUrl: './images/matic-token.svg',
+            },
+          },
+          '0x1': {
+            chainId: '0x1',
+            ticker: 'ETH',
+            rpcPrefs: {
+              imageUrl: './images/eth-token.svg',
+            },
+          },
+        },
+      },
+    };
+
+    const expectedState = {
+      NetworkController: {
+        networkConfigurations: {
+          '0x89': {
+            chainId: '0x89',
+            ticker: 'POL',
+            rpcPrefs: {
+              imageUrl: './images/pol-token.svg',
+            },
+          },
+          '0x1': {
+            chainId: '0x1',
+            ticker: 'ETH',
+            rpcPrefs: {
+              imageUrl: './images/eth-token.svg',
+            },
+          },
+        },
+      },
+    };
+
+    const transformedState = await migrate({
+      meta: { version: oldVersion },
+      data: cloneDeep(oldState),
+    });
+
+    expect(transformedState.data).toStrictEqual(expectedState);
+  });
+
+  it('Does nothing if Polygon ChainId (0x89) is not in networkConfigurations', async () => {
+    const oldState = {
+      NetworkController: {
+        networkConfigurations: {
+          '0x1': {
+            chainId: '0x1',
+            ticker: 'ETH',
+            rpcPrefs: {
+              imageUrl: './images/eth-token.svg',
+            },
+          },
+          '0x2a': {
+            chainId: '0x2a',
+            ticker: 'KOVAN',
+            rpcPrefs: {
+              imageUrl: './images/kovan-token.svg',
+            },
+          },
+        },
+      },
+    };
+
+    const transformedState = await migrate({
+      meta: { version: oldVersion },
+      data: cloneDeep(oldState),
+    });
+
+    expect(transformedState.data).toStrictEqual(oldState);
+  });
+
+  it('Updates Polygon ChainId (0x89) in ProviderConfig if exists, and ticker is set to MATIC, and updates imageUrl', async () => {
+    const oldState = {
+      NetworkController: {
+        providerConfig: {
+          chainId: '0x89',
+          ticker: 'MATIC',
+          rpcPrefs: {
+            imageUrl: './images/matic-token.svg',
+          },
+        },
+      },
+    };
+
+    const expectedState = {
+      NetworkController: {
+        providerConfig: {
+          chainId: '0x89',
+          ticker: 'POL',
+          rpcPrefs: {
+            imageUrl: './images/pol-token.svg',
+          },
+        },
+      },
+    };
+
+    const transformedState = await migrate({
+      meta: { version: oldVersion },
+      data: cloneDeep(oldState),
+    });
+
+    expect(transformedState.data).toStrictEqual(expectedState);
+  });
+
+  it('Does nothing if Polygon ChainId (0x89) is not in providerConfig', async () => {
+    const oldState = {
+      NetworkController: {
+        providerConfig: {
+          chainId: '0x1',
+          ticker: 'ETH',
+          rpcPrefs: {
+            imageUrl: './images/eth-token.svg',
+          },
+        },
+      },
+    };
+
+    const transformedState = await migrate({
+      meta: { version: oldVersion },
+      data: cloneDeep(oldState),
+    });
+
+    expect(transformedState.data).toStrictEqual(oldState);
+  });
+
+  it('Does not update ticker if Polygon ChainId (0x89) is in providerConfig, but ticker is not MATIC, but still updates imageUrl', async () => {
+    const oldState = {
+      NetworkController: {
+        providerConfig: {
+          chainId: '0x89',
+          ticker: 'NOT_MATIC',
+          rpcPrefs: {
+            imageUrl: './images/matic-token.svg',
+          },
+        },
+      },
+    };
+
+    const expectedState = {
+      NetworkController: {
+        providerConfig: {
+          chainId: '0x89',
+          ticker: 'NOT_MATIC',
+          rpcPrefs: {
+            imageUrl: './images/pol-token.svg',
+          },
+        },
+      },
+    };
+
+    const transformedState = await migrate({
+      meta: { version: oldVersion },
+      data: cloneDeep(oldState),
+    });
+
+    expect(transformedState.data).toStrictEqual(expectedState);
+  });
+});

app/scripts/migrations/121.2.ts

@@ -0,0 +1,93 @@
+import { hasProperty, isObject } from '@metamask/utils';
+import { cloneDeep } from 'lodash';
+import { CHAIN_IDS } from '../../../shared/constants/network';
+
+export const version = 121.2;
+
+type VersionedData = {
+  meta: { version: number };
+  data: Record<string, unknown>;
+};
+
+/**
+ * Migrates MATIC ticker in Network Configuration to POL ticker as per the direction in https://polygon.technology/blog/save-the-date-matic-pol-migration-coming-september-4th-everything-you-need-to-know
+ *
+ * @param originalVersionedData - Versioned MetaMask extension state, exactly what we persist to dist.
+ * @param originalVersionedData.meta - State metadata.
+ * @param originalVersionedData.meta.version - The current state version.
+ * @param originalVersionedData.data - The persisted MetaMask state, keyed by controller.
+ * @returns Updated versioned MetaMask extension state.
+ */
+export async function migrate(
+  originalVersionedData: VersionedData,
+): Promise<VersionedData> {
+  const versionedData = cloneDeep(originalVersionedData);
+  versionedData.meta.version = version;
+  transformState(versionedData.data);
+  return versionedData;
+}
+
+function transformState(state: Record<string, unknown>): void {
+  const networkControllerState = state.NetworkController;
+  if (
+    hasProperty(state, 'NetworkController') &&
+    isObject(networkControllerState) &&
+    hasProperty(networkControllerState, 'networkConfigurations') &&
+    isObject(networkControllerState.networkConfigurations)
+  ) {
+    for (const networkConfiguration of Object.values(
+      networkControllerState.networkConfigurations,
+    )) {
+      if (
+        isObject(networkConfiguration) &&
+        networkConfiguration.chainId === CHAIN_IDS.POLYGON
+      ) {
+        // update image path regardless of ticker
+        if (
+          hasProperty(networkConfiguration, 'rpcPrefs') &&
+          isObject(networkConfiguration.rpcPrefs) &&
+          hasProperty(networkConfiguration.rpcPrefs, 'imageUrl') &&
+          networkConfiguration.rpcPrefs.imageUrl === './images/matic-token.svg'
+        ) {
+          networkConfiguration.rpcPrefs.imageUrl = './images/pol-token.svg';
+        }
+        // update ticker only if MATIC
+        if (
+          hasProperty(networkConfiguration, 'ticker') &&
+          networkConfiguration.ticker === 'MATIC'
+        ) {
+          networkConfiguration.ticker = 'POL';
+        }
+      }
+    }
+  }
+
+  // handle legacy NetworkController versions (with providerConfig)
+  if (
+    hasProperty(state, 'NetworkController') &&
+    isObject(networkControllerState) &&
+    hasProperty(networkControllerState, 'providerConfig') &&
+    isObject(networkControllerState.providerConfig) &&
+    hasProperty(networkControllerState.providerConfig, 'chainId') &&
+    networkControllerState.providerConfig.chainId === CHAIN_IDS.POLYGON
+  ) {
+    // update image path regardless of ticker
+    if (
+      hasProperty(networkControllerState.providerConfig, 'rpcPrefs') &&
+      isObject(networkControllerState.providerConfig.rpcPrefs) &&
+      hasProperty(networkControllerState.providerConfig.rpcPrefs, 'imageUrl') &&
+      networkControllerState.providerConfig.rpcPrefs.imageUrl ===
+        './images/matic-token.svg'
+    ) {
+      networkControllerState.providerConfig.rpcPrefs.imageUrl =
+        './images/pol-token.svg';
+    }
+    // update ticker only if MATIC
+    if (
+      hasProperty(networkControllerState.providerConfig, 'ticker') &&
+      networkControllerState.providerConfig.ticker === 'MATIC'
+    ) {
+      networkControllerState.providerConfig.ticker = 'POL';
+    }
+  }
+}

app/scripts/migrations/index.js

@@ -139,6 +139,7 @@ const migrations = [
   require('./120.6'),
   require('./121'),
   require('./121.1'),
+  require('./121.2'),
   require('./122'),
   require('./123'),
   require('./124'),