Fix update previous ids for custom networks

zeriontech · May 22, 2024 · e052599 · e052599
1 parent caf3c48
commit e052599
Showing 1 changed file with 20 additions and 5 deletions.
diff --git a/src/modules/ethereum/chains/ChainConfigStore.ts b/src/modules/ethereum/chains/ChainConfigStore.ts
@@ -30,6 +30,16 @@ function updateChainOrigin(origin: string, prevOrigin: string | null) {
   return origin;
 }
 
+function updatePreviousIds(
+  id: string,
+  previousId: string,
+  previousIds: string[] | null
+) {
+  return previousId !== id && !previousIds?.includes(previousId)
+    ? [...(previousIds || []), previousId]
+    : previousIds;
+}
+
 class ChainConfigStore extends PersistentStore<ChainConfig> {
   static initialState: ChainConfig = {
     version: 2,
@@ -69,10 +79,7 @@ class ChainConfigStore extends PersistentStore<ChainConfig> {
     );
     const existingEntry = existingItems.get(prevId);
     const existingPreviousIds = existingEntry?.previousIds || null;
-    const previousIds =
-      prevId !== id && !existingPreviousIds?.includes(prevId)
-        ? [...(existingPreviousIds || []), prevId]
-        : existingPreviousIds;
+    const previousIds = updatePreviousIds(id, prevId, existingPreviousIds);
     const now = Date.now();
     const newEntry: EthereumChainConfig = {
       origin: updateChainOrigin(origin, existingEntry?.origin || null),
@@ -122,7 +129,15 @@ class ChainConfigStore extends PersistentStore<ChainConfig> {
               `Unable to fetch network info by chainId: ${config.value.chainId}`
             );
           }
-          updatedEthereumChainConfigs.push({ ...config, id: network.id });
+          updatedEthereumChainConfigs.push({
+            ...config,
+            id: network.id,
+            previousIds: updatePreviousIds(
+              network.id,
+              config.id,
+              config.previousIds
+            ),
+          });
         } catch {
           updatedEthereumChainConfigs.push(config);
         }