move duplicated migration type setting code to function

Author: RhizoNymph

packages/doppler-v3-indexer/src/indexer/shared/entities/pool.ts

@@ -11,6 +11,8 @@ import { Address } from "viem";
 import { computeMarketCap } from "../oracle";
 import { getLockableV3PoolData } from "@app/utils/v3-utils/getV3PoolData";
 import { chainConfigs } from "@app/config";
+import { AssetData } from "@app/types";
+import { Network } from "@app/types/config-types";
 
 export const fetchExistingPool = async ({
   poolAddress,
@@ -87,21 +89,7 @@ export const insertPoolIfNotExists = async ({
     totalSupply: assetTotalSupply,
   });
 
-  let migrationType;
-  if (
-    assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v2.v2Migrator.toLowerCase() 
-    || assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v2.nimCustomV2Migrator.toLowerCase()
-  ) {
-    migrationType = "v2";
-  } else if (
-    assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v3.v3Migrator.toLowerCase()
-    || assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v3.nimCustomV3Migrator.toLowerCase()
-  ) {
-    migrationType = "v3";
-  } else if (assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v4.v4Migrator.toLowerCase()) {
-    migrationType = "v4";
-  }
-
+  let migrationType = getMigrationType(assetData, chain.name);
 
   return await db.insert(pool).values({
     ...poolData,
@@ -229,20 +217,7 @@ export const insertPoolIfNotExistsV4 = async ({
     graduationBalance: 0n,
   });
 
-  let migrationType;
-  if (
-    assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v2.v2Migrator.toLowerCase()
-    || assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v2.nimCustomV2Migrator.toLowerCase()
-  ) {
-    migrationType = "v2";
-  } else if (
-    assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v3.v3Migrator.toLowerCase()
-    || assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v3.nimCustomV3Migrator.toLowerCase()
-  ) {
-    migrationType = "v3";
-  } else if (assetData.liquidityMigrator.toLowerCase() === chainConfigs[chain.name].addresses.v4.v4Migrator.toLowerCase()) {
-    migrationType = "v4";
-  }
+  let migrationType = getMigrationType(assetData, chain.name);
 
   return await db.insert(pool).values({
     address,
@@ -363,3 +338,21 @@ export const insertLockableV3PoolIfNotExists = async ({
     integrator: assetData.integrator,
   });
 };
+
+function getMigrationType(assetData: AssetData, chainName: Network): string {
+  if (
+    assetData.liquidityMigrator.toLowerCase() === chainConfigs[chainName].addresses.v2.v2Migrator.toLowerCase()
+    || assetData.liquidityMigrator.toLowerCase() === chainConfigs[chainName].addresses.v2.nimCustomV2Migrator.toLowerCase()
+  ) {
+    return "v2";
+  } else if (
+    assetData.liquidityMigrator.toLowerCase() === chainConfigs[chainName].addresses.v3.v3Migrator.toLowerCase()
+    || assetData.liquidityMigrator.toLowerCase() === chainConfigs[chainName].addresses.v3.nimCustomV3Migrator.toLowerCase()
+  ) {
+    return "v3";
+  } else if (assetData.liquidityMigrator.toLowerCase() === chainConfigs[chainName].addresses.v4.v4Migrator.toLowerCase()) {
+    return "v4";
+  } else {
+    return "unknown";
+  }
+}