feat: CLI V4 support (Uniswap#700)

5ky9uy · Sep 17, 2024 · d8b2bb7 · d8b2bb7
1 parent ab246ae
commit d8b2bb7
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 2 deletions.
diff --git a/cli/base-command.ts b/cli/base-command.ts
@@ -17,6 +17,7 @@ import {
   CachingTokenListProvider,
   CachingTokenProviderWithFallback,
   CachingV3PoolProvider,
+  CachingV4PoolProvider,
   CHAIN_IDS_LIST,
   EIP1559GasPriceProvider,
   EthEstimateGasSimulator,
@@ -44,7 +45,8 @@ import {
   UniswapMulticallProvider,
   V2PoolProvider,
   V3PoolProvider,
-  V3RouteWithValidQuote
+  V3RouteWithValidQuote,
+  V4PoolProvider
 } from '../src';
 import {
   LegacyGasPriceProvider
@@ -286,6 +288,11 @@ export abstract class BaseCommand extends Command {
         new NodeCache({ stdTTL: 15, useClones: true })
       );
 
+      const v4PoolProvider = new CachingV4PoolProvider(
+        chainId,
+        new V4PoolProvider(chainId, multicall2Provider),
+        new NodeJSCache(new NodeCache({ stdTTL: 360, useClones: false }))
+      );
       const v3PoolProvider = new CachingV3PoolProvider(
         chainId,
         new V3PoolProvider(chainId, multicall2Provider),
@@ -312,6 +319,7 @@ export abstract class BaseCommand extends Command {
         process.env.TENDERLY_NODE_API_KEY!,
         v2PoolProvider,
         v3PoolProvider,
+        v4PoolProvider,
         provider,
         portionProvider,
         { [ChainId.ARBITRUM_ONE]: 1 },
@@ -325,6 +333,7 @@ export abstract class BaseCommand extends Command {
         provider,
         v2PoolProvider,
         v3PoolProvider,
+        v4PoolProvider,
         portionProvider
       );
 

diff --git a/cli/commands/quote.ts b/cli/commands/quote.ts
@@ -5,9 +5,17 @@ import { Currency, Percent, TradeType } from '@uniswap/sdk-core';
 import dotenv from 'dotenv';
 import _ from 'lodash';
 
-import { ID_TO_CHAIN_ID, MapWithLowerCaseKey, nativeOnChain, parseAmount, SwapRoute, SwapType, } from '../../src';
+import {
+  ID_TO_CHAIN_ID,
+  MapWithLowerCaseKey,
+  nativeOnChain,
+  parseAmount,
+  SwapRoute,
+  SwapType
+} from '../../src';
 import { NATIVE_NAMES_BY_ID, TO_PROTOCOL } from '../../src/util';
 import { BaseCommand } from '../base-command';
+import { UniversalRouterVersion } from '@uniswap/universal-router-sdk';
 
 dotenv.config();
 
@@ -138,6 +146,7 @@ export class Quote extends BaseCommand {
             recipient,
             slippageTolerance: new Percent(5, 100),
             simulate: simulate ? { fromAddress: recipient } : undefined,
+            version: UniversalRouterVersion.V2_0
           }
           : undefined,
         {

diff --git a/src/util/protocols.ts b/src/util/protocols.ts
@@ -2,6 +2,8 @@ import { Protocol } from '@uniswap/router-sdk';
 
 export const TO_PROTOCOL = (protocol: string): Protocol => {
   switch (protocol.toLowerCase()) {
+    case 'v4':
+      return Protocol.V4;
     case 'v3':
       return Protocol.V3;
     case 'v2':