fix: remove curve label suffix in a non invasive manner

Zapper-fi · Jun 27, 2022 · c3be480 · c3be480
1 parent 7bf3ed5
commit c3be480
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/apps/curve/helpers/curve.pool.token-helper.ts b/src/apps/curve/helpers/curve.pool.token-helper.ts
@@ -59,6 +59,15 @@ type CurvePoolTokenHelperParams<T = CurveToken, V = Erc20> = {
   }) => Promise<number>;
 };
 
+const labelAllowList = ['3pool', '2pool', 'tricrypto2', 'tricrypto'];
+function rebindLabel(opt: { expectedLabel?: string; fallbackLabel: string }) {
+  if (opt.expectedLabel && labelAllowList.includes(opt.expectedLabel.toLowerCase())) {
+    return opt.expectedLabel;
+  }
+
+  return opt.fallbackLabel;
+}
+
 const isMetaPool = (token: Token) =>
   token.type === ContractType.APP_TOKEN &&
   token.appId === CURVE_DEFINITION.id &&
@@ -142,7 +151,10 @@ export class CurvePoolTokenHelper {
 
         // Display Properties
         const underlyingLabels = tokens.map(v => (isMetaPool(v) ? getLabelFromToken(v) : v.symbol)); // Flatten metapool label
-        const label = definition.label ?? underlyingLabels.join(' / ');
+        const label = rebindLabel({
+          expectedLabel: definition.label,
+          fallbackLabel: underlyingLabels.join(' / '),
+        });
         const secondaryLabel = reservePercentages.map(p => `${Math.floor(p * 100)}%`).join(' / ');
         const images = underlyingTokens.map(t => getImagesFromToken(t)).flat();