extend alternative fees to LPs

penumbra-zone · Jan 15, 2025 · b886b58 · b886b58
1 parent 97842ea
commit b886b58
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 0 deletions.
diff --git a/packages/services/src/view-service/fees.ts b/packages/services/src/view-service/fees.ts
@@ -89,6 +89,40 @@ export const extractAltFee = async (
     }
   }
 
+  const positionOpen = request.positionOpens
+    .map(assetIn => assetIn.position?.phi?.pair?.asset1)
+    .find(Boolean);
+  if (positionOpen) {
+    const assetId = await getAssetFromGasPriceTable(request, indexedDb, swapAsset);
+    if (assetId) {
+      return assetId;
+    }
+  }
+
+  const positionClose = request.positionCloses.map(a => a.positionId).find(Boolean);
+  if (positionClose) {
+    const closePosition = await indexedDb.getPosition(positionClose);
+    if (closePosition?.phi?.pair?.asset1) {
+      const inputAssetId = closePosition.phi.pair.asset1;
+      const assetId = await getAssetFromGasPriceTable(request, indexedDb, inputAssetId);
+      if (assetId) {
+        return assetId;
+      }
+    }
+  }
+
+  const positonWithdraw = request.positionWithdraws.map(a => a.positionId).find(Boolean);
+  if (positonWithdraw) {
+    const withdrawPosition = await indexedDb.getPosition(positonWithdraw);
+    if (withdrawPosition?.phi?.pair?.asset1) {
+      const inputAssetId = withdrawPosition.phi.pair.asset1;
+      const assetId = await getAssetFromGasPriceTable(request, indexedDb, inputAssetId);
+      if (assetId) {
+        return assetId;
+      }
+    }
+  }
+
   if (request.undelegations.length > 0) {
     const assetId = await getAssetFromGasPriceTable(request, indexedDb);
     if (assetId) {

diff --git a/packages/storage/src/indexed-db/index.ts b/packages/storage/src/indexed-db/index.ts
@@ -993,4 +993,15 @@ export class IndexedDb implements IndexedDbInterface {
         new Amount({ hi: 0n, lo: 0n }),
       );
   }
+
+  async getPosition(positionId: PositionId): Promise<Position | undefined> {
+    assertPositionId(positionId);
+    const position = await this.db.get('POSITIONS', uint8ArrayToBase64(positionId.inner));
+
+    if (!position) {
+      return undefined;
+    }
+
+    return Position.fromJson(position.position);
+  }
 }
diff --git a/packages/types/src/indexed-db.ts b/packages/types/src/indexed-db.ts
@@ -153,6 +153,8 @@ export interface IndexedDbInterface {
   hasTokenBalance(addressIndex: number, assetId: AssetId): Promise<boolean>;
 
   totalNoteBalance(accountIndex: number, assetId: AssetId): Promise<Amount>;
+
+  getPosition(positionId: PositionId): Promise<Position | undefined>;
 }
 
 export interface PenumbraDb extends DBSchema {