From 4a41b1a85cfc4b7bc99e75b1bafe7eb56e889e9a Mon Sep 17 00:00:00 2001
From: Piotr Witek <739075+piotrwitek@users.noreply.github.com>
Date: Wed, 19 Jun 2024 17:31:03 +0200
Subject: [PATCH] Fix position listing in Portfolio page (#3958)

* Fixed New swapped position is not listed in Portfolio page

* fixed type
---
 .../positions/handlers/aave-like/index.ts          |  9 ++++++---
 .../aave-like/mapDpmProtocolNameToUIName.ts        | 14 ++++++++++++++
 2 files changed, 20 insertions(+), 3 deletions(-)
 create mode 100644 handlers/portfolio/positions/handlers/aave-like/mapDpmProtocolNameToUIName.ts

diff --git a/handlers/portfolio/positions/handlers/aave-like/index.ts b/handlers/portfolio/positions/handlers/aave-like/index.ts
index 75d5f516a4..b924d0bc95 100644
--- a/handlers/portfolio/positions/handlers/aave-like/index.ts
+++ b/handlers/portfolio/positions/handlers/aave-like/index.ts
@@ -36,6 +36,7 @@ import { isAaveLikeLendingProtocol, LendingProtocol } from 'lendingProtocols'
 import { getAaveWstEthYield } from 'lendingProtocols/aave-v3/calculations/wstEthYield'
 
 import { getRawPositionDetails } from './getRawPositionDetails'
+import { mapDpmProtocolNameToUIName } from './mapDpmProtocolNameToUIName'
 
 const getAaveLikeBorrowPosition: GetAaveLikePositionHandlerType = async ({
   dpm,
@@ -421,9 +422,11 @@ export const aaveLikePositionsHandler: PortfolioPositionsHandler = async ({
   positionsCount,
   debug,
 }) => {
-  const aaveLikeDpmList = dpmList.filter(({ protocol }) =>
-    [aaveLikeProtocolNames.aavev3, aaveLikeProtocolNames.sparkv3].includes(protocol),
-  )
+  const aaveLikeDpmList = dpmList
+    .map(mapDpmProtocolNameToUIName)
+    .filter(({ protocol }) =>
+      [aaveLikeProtocolNames.aavev3, aaveLikeProtocolNames.sparkv3].includes(protocol),
+    )
   if (positionsCount) {
     return {
       positions: aaveLikeDpmList.map(({ vaultId }) => ({ positionId: vaultId })),
diff --git a/handlers/portfolio/positions/handlers/aave-like/mapDpmProtocolNameToUIName.ts b/handlers/portfolio/positions/handlers/aave-like/mapDpmProtocolNameToUIName.ts
new file mode 100644
index 0000000000..d50ba91774
--- /dev/null
+++ b/handlers/portfolio/positions/handlers/aave-like/mapDpmProtocolNameToUIName.ts
@@ -0,0 +1,14 @@
+import { ProtocolName } from '@summer_fi/summerfi-sdk-common'
+import { aaveLikeProtocolNames } from 'handlers/portfolio/positions/handlers/aave-like/helpers'
+import type { DpmSubgraphData } from 'handlers/portfolio/positions/helpers/getAllDpmsForWallet'
+
+// map SDK protocol name to the one hardcoded in the UI
+export const mapDpmProtocolNameToUIName = (dpm: DpmSubgraphData): DpmSubgraphData => {
+  if (dpm.protocol === ProtocolName.AaveV3) {
+    return {
+      ...dpm,
+      protocol: aaveLikeProtocolNames.aavev3,
+    }
+  }
+  return dpm
+}