use addTokensReceived method

DefiLlama · Jan 11, 2025 · 6d0631d · 6d0631d
1 parent c4e3664
commit 6d0631d
Showing 1 changed file with 6 additions and 34 deletions.
diff --git a/fees/goplus.ts b/fees/goplus.ts
@@ -1,45 +1,17 @@
 import { FetchOptions, SimpleAdapter } from "../adapters/types";
 import { CHAIN } from "../helpers/chains";
-import BigNumber from "bignumber.js";
+import { addTokensReceived } from "../helpers/token";
 
-const USDT_MINT = "0x55d398326f99059ff775485246999027b3197955";
 const GOPLUS_FOUNDATION = "0x34ebddd30ccbd3f1e385b41bdadb30412323e34f";
 const GOPLUS_REVENUE_POOL = "0x648d7f4ad39186949e37e9223a152435ab97706c";
 
-const BALANCE_ABI = 'erc20:balanceOf';
-const CALLS = [
-      {
-          target: USDT_MINT,
-          params: [GOPLUS_FOUNDATION]
-      },
-      {
-          target: USDT_MINT,
-          params: [GOPLUS_REVENUE_POOL]
-      },
-  ];
-
 const fetch = async (options: FetchOptions) => {
   const dailyFees = options.createBalances();
-  const totalFees = options.createBalances();
-  const dailyProtocolRevenue = options.createBalances();
-  const totalProtocolRevenue = options.createBalances();
-  const [foundationBalanceStart, revenueBalanceStart] = await options.fromApi.multiCall({
-    abi: BALANCE_ABI,
-    calls: CALLS
-  });
-  const [foundationBalanceEnd, revenueBalanceEnd] = await options.toApi.multiCall({
-    abi: BALANCE_ABI,
-    calls: CALLS
-  });
-  const dailyFoundationReceived = BigNumber(foundationBalanceEnd).minus(BigNumber(foundationBalanceStart));
-  const dailyRevenueReceived = BigNumber(revenueBalanceEnd).minus(BigNumber(revenueBalanceStart));
-  const dailyTotal = dailyFoundationReceived.plus(dailyRevenueReceived).toFixed(0);
-  const totalFee = BigNumber(foundationBalanceEnd).plus(BigNumber(revenueBalanceEnd)).toFixed(0);
-  dailyFees.add(USDT_MINT, dailyTotal);
-  totalFees.add(USDT_MINT, totalFee);
-  dailyProtocolRevenue.add(USDT_MINT, dailyTotal);
-  totalProtocolRevenue.add(USDT_MINT, totalFee);
-  return { dailyFees, totalFees, dailyRevenue: dailyProtocolRevenue, dailyProtocolRevenue, totalProtocolRevenue };
+
+  await addTokensReceived({ balances: dailyFees, target: GOPLUS_FOUNDATION, options, })
+  await addTokensReceived({ balances: dailyFees, target: GOPLUS_REVENUE_POOL, options, })
+
+  return { dailyFees, dailyRevenue: dailyFees, dailyProtocolRevenue: dailyFees };
 };
 
 const adapter: SimpleAdapter = {