From 9a7620dd2c1e60584288186806165e4b7cccf456 Mon Sep 17 00:00:00 2001
From: TrooperCrypto <trooper.crypto7@gmail.com>
Date: Sat, 2 Apr 2022 21:42:55 +0200
Subject: [PATCH] enable different fee tokens

---
 marketmaker.js | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/marketmaker.js b/marketmaker.js
index 28f6f63..2433d0a 100644
--- a/marketmaker.js
+++ b/marketmaker.js
@@ -16,6 +16,7 @@ const MARKETS = {};
 const CHAINLINK_PROVIDERS = {};
 const UNISWAP_V3_PROVIDERS = {};
 const PAST_ORDER_LIST = {};
+const FEE_TOKENS = [];
 
 let uniswap_error_counter = 0;
 let chainlink_error_counter = 0;
@@ -195,6 +196,18 @@ async function handleMessage(json) {
             const newBaseFee = MARKETS[marketId].baseFee;
             const newQuoteFee = MARKETS[marketId].quoteFee;
             console.log(`marketinfo ${marketId} - update baseFee ${oldBaseFee} -> ${newBaseFee}, quoteFee ${oldQuoteFee} -> ${newQuoteFee}`);
+            if(
+              marketInfo.baseAsset.enabledForFees &&
+              !FEE_TOKENS.includes(marketInfo.baseAsset.id)
+            ) {
+              FEE_TOKENS.push(marketInfo.baseAsset.id);
+            } 
+            if(
+              marketInfo.quoteAsset.enabledForFees &&
+              !FEE_TOKENS.includes(marketInfo.quoteAsset.id)
+            ) {
+              FEE_TOKENS.push(marketInfo.quoteAsset.id);
+            } 
             break
         default:
             break
@@ -415,11 +428,15 @@ async function broadcastFill(chainId, orderId, swapOffer, fillOrder, wallet) {
         zigzagws.send(JSON.stringify(orderCommitMsg));
         return;
     }
+    // select token to match user's fee token
+    const feeToken = (FEE_TOKENS.includes(swapOffer.tokenSell))
+      ? swapOffer.tokenSell
+      : 0
     const randInt = (Math.random()*1000).toFixed(0);
     console.time('syncswap' + randInt);
     const swap = await wallet['syncWallet'].syncSwap({
         orders: [swapOffer, fillOrder],
-        feeToken: "ETH",
+        feeToken: feeToken,
         nonce: fillOrder.nonce
     });
     const txHash = swap.txHash.split(":")[1];