fix: partially exclude fot from fee

Uniswap · Jul 18, 2024 · e9aa20f · e9aa20f
1 parent 44a4c39
commit e9aa20f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/lib/graphql/graphql-queries.ts b/lib/graphql/graphql-queries.ts
@@ -11,6 +11,9 @@ query Token($chain: Chain!, $address: String!) {
             feeData {
                 buyFeeBps
                 sellFeeBps
+                feeTakenOnTransfer
+                externalTransferFailed
+                sellReverted
             }
         }
     }

diff --git a/lib/graphql/graphql-schemas.ts b/lib/graphql/graphql-schemas.ts
@@ -21,5 +21,8 @@ export interface TokenInfo {
   feeData?: {
     buyFeeBps?: string
     sellFeeBps?: string
+    feeTakenOnTransfer?: boolean
+    externalTransferFailed?: boolean
+    sellReverted?: boolean
   }
 }
diff --git a/lib/graphql/graphql-token-fee-fetcher.ts b/lib/graphql/graphql-token-fee-fetcher.ts
@@ -46,9 +46,12 @@ export class GraphQLTokenFeeFetcher implements ITokenFeeFetcher {
           if (token.feeData?.buyFeeBps || token.feeData?.sellFeeBps) {
             const buyFeeBps = token.feeData.buyFeeBps ? BigNumber.from(token.feeData.buyFeeBps) : undefined
             const sellFeeBps = token.feeData.sellFeeBps ? BigNumber.from(token.feeData.sellFeeBps) : undefined
-            tokenFeeMap[token.address] = { buyFeeBps, sellFeeBps }
+            const feeTakenOnTransfer = token.feeData.feeTakenOnTransfer
+            const externalTransferFailed = token.feeData.externalTransferFailed
+            const sellReverted = token.feeData.sellReverted
+            tokenFeeMap[token.address] = { buyFeeBps, sellFeeBps, feeTakenOnTransfer, externalTransferFailed, sellReverted }
           } else {
-            tokenFeeMap[token.address] = { buyFeeBps: undefined, sellFeeBps: undefined }
+            tokenFeeMap[token.address] = { buyFeeBps: undefined, sellFeeBps: undefined, feeTakenOnTransfer: false, externalTransferFailed: false, sellReverted: false }
           }
         })
         metric.putMetric('GraphQLTokenFeeFetcherFetchFeesSuccess', 1, MetricLoggerUnit.Count)