@kittycorn-labs/universal-router-sdk 1.2.0 → 1.3.0

package/dist/universal-router-sdk.esm.js

@@ -10,6 +10,7 @@ import { Trade, Pair, Route as Route$2 } from '@uniswap/v2-sdk';
 import { Protocol, MixedRouteTrade, partitionMixedRouteByProtocol, getOutputOfPools, MixedRoute, MixedRouteSDK, encodeMixedRouteToPath, Trade as Trade$3 } from '@kittycorn-labs/router-sdk';
 import 'jsbi';
 import 'bignumber.js';
+import { getSupportUnderlyingByTokenize, getSupportTokenizeByUnderlying } from '@kittycorn-labs/smart-order-router';
 
 function _arrayLikeToArray(r, a) {
   (null == a || a > r.length) && (a = r.length);
@@ -839,8 +840,85 @@ function addV3Swap(planner, _ref2, tradeType, options, payerIsUser, routerMustCu
     planner.addCommand(CommandType.V3_SWAP_EXACT_OUT, [routerMustCustody ? ROUTER_AS_RECIPIENT : options.recipient, trade.minimumAmountOut(options.slippageTolerance).quotient.toString(), trade.maximumAmountIn(options.slippageTolerance).quotient.toString(), path, payerIsUser]);
   }
 }
-function addV4Swap(planner, _ref3, tradeType, options, payerIsUser, routerMustCustody) {
+function getReducePools(pools, tokenIn, tokenOut) {
+  if (pools.length < 3) {
+    return pools;
+  }
+  var chainId = pools[0].chainId;
+  var token0 = getSupportUnderlyingByTokenize(chainId, tokenIn); // assume tokenIn is tokenize
+  var token1 = getSupportUnderlyingByTokenize(chainId, tokenOut); // assume tokenOut is tokenize
+  var isSupport0 = token0 !== undefined;
+  var isSupport1 = token1 !== undefined;
+  var underlyingIn = isSupport0 ? token0 : tokenIn;
+  var underlyingOut = isSupport1 ? token1 : tokenOut;
+  // Find reduce pool for
+  // - first pool is underlying and tokenize for tokenIn
+  // - last pool is underlying and previous tokenize for tokenOut
+  if (!isSupport0) {
+    var tokenizeIn = getSupportTokenizeByUnderlying(chainId, underlyingIn);
+    if (tokenizeIn !== undefined) {
+      var reduceFirst = pools[0].token0.equals(underlyingIn) && pools[0].token1.equals(tokenizeIn) || pools[0].token0.equals(tokenizeIn) && pools[0].token1.equals(underlyingIn);
+      if (reduceFirst) {
+        pools.shift();
+      }
+    }
+  }
+  if (!isSupport1) {
+    var tokenizeOut = getSupportTokenizeByUnderlying(chainId, underlyingOut);
+    if (tokenizeOut !== undefined) {
+      var lastIndex = pools.length - 1;
+      var reduceLast = pools[lastIndex].token0.equals(underlyingOut) && pools[lastIndex].token1.equals(tokenizeOut) || pools[lastIndex].token0.equals(tokenizeOut) && pools[lastIndex].token1.equals(underlyingOut);
+      if (reduceLast) {
+        pools.pop();
+      }
+    }
+  }
+  return pools;
+}
+function buildV4Planner(trade, tradeType, options, payerIsUser, routerMustCustody) {
   var _options$recipient;
+  var chainId = trade.route.chainId;
+  var currencyIn = trade.route.pathInput;
+  var currencyOut = trade.route.pathOutput;
+  var slippageToleranceOnSwap = routerMustCustody && tradeType == TradeType.EXACT_INPUT ? undefined : options.slippageTolerance;
+  // Make tokens path from pools
+  var path = [];
+  var nextToken = currencyIn;
+  trade.route.pools.forEach(function (pool) {
+    if (nextToken.equals(pool.token0)) {
+      path.push(pool.token0);
+      nextToken = pool.token1;
+    } else {
+      path.push(pool.token1);
+      nextToken = pool.token0;
+    }
+  });
+  var v4Planner = new V4Planner();
+  // Add trade swap type with encode pathKeys
+  v4Planner.addTrade(trade, slippageToleranceOnSwap);
+  // Pay currency in
+  v4Planner.addSettle(currencyIn, payerIsUser);
+  var token0 = getSupportUnderlyingByTokenize(chainId, currencyIn);
+  var isSupport0 = token0 !== undefined;
+  // Build action planner for path
+  for (var i = 1; i < path.length; i++) {
+    var token1 = getSupportUnderlyingByTokenize(chainId, path[i]);
+    var isSupport1 = token1 !== undefined;
+    if (isSupport0 && path[i].equals(token0)) {
+      v4Planner.addTake(path[i - 1], ROUTER_AS_RECIPIENT);
+      v4Planner.addSettle(token0, false);
+    } else if (isSupport1 && path[i - 1].equals(token1)) {
+      v4Planner.addTake(token1, ROUTER_AS_RECIPIENT);
+      v4Planner.addSettle(path[i], false);
+    }
+    isSupport0 = isSupport1;
+    token0 = token1;
+  }
+  // Take currency out
+  v4Planner.addTake(currencyOut, routerMustCustody ? ROUTER_AS_RECIPIENT : (_options$recipient = options.recipient) != null ? _options$recipient : SENDER_AS_RECIPIENT);
+  return v4Planner;
+}
+function addV4Swap(planner, _ref3, tradeType, options, payerIsUser, routerMustCustody) {
   var inputAmount = _ref3.inputAmount,
     outputAmount = _ref3.outputAmount,
     route = _ref3.route;
@@ -848,6 +926,8 @@ function addV4Swap(planner, _ref3, tradeType, options, payerIsUser, routerMustCu
   var pools = route.pools.map(function (p) {
     return p;
   });
+  // reduce pools for tokenize
+  pools = getReducePools(pools, inputAmount.currency, outputAmount.currency);
   var v4Route = new Route(pools, inputAmount.currency, outputAmount.currency);
   var trade = Trade$2.createUncheckedTrade({
     route: v4Route,
@@ -855,11 +935,8 @@ function addV4Swap(planner, _ref3, tradeType, options, payerIsUser, routerMustCu
     outputAmount: outputAmount,
     tradeType: tradeType
   });
-  var slippageToleranceOnSwap = routerMustCustody && tradeType == TradeType.EXACT_INPUT ? undefined : options.slippageTolerance;
-  var v4Planner = new V4Planner();
-  v4Planner.addTrade(trade, slippageToleranceOnSwap);
-  v4Planner.addSettle(trade.route.pathInput, payerIsUser);
-  v4Planner.addTake(trade.route.pathOutput, routerMustCustody ? ROUTER_AS_RECIPIENT : (_options$recipient = options.recipient) != null ? _options$recipient : SENDER_AS_RECIPIENT);
+  // build planner
+  var v4Planner = buildV4Planner(trade, tradeType, options, payerIsUser, routerMustCustody);
   planner.addCommand(CommandType.V4_SWAP, [v4Planner.finalize()]);
 }
 // encode a mixed route swap, i.e. including both v2 and v3 pools