@net-protocol/score 0.1.2 → 0.1.3

package/dist/index.d.mts

@@ -164,20 +164,24 @@ declare function calculatePriceFromSqrtPriceX96(sqrtPriceX96: bigint, token0Deci
  * Discover the best Uniswap pools for multiple token pairs.
  * Makes two on-chain calls: pool discovery + pool info retrieval.
  */
-declare function discoverPools({ publicClient, pairs, }: {
+declare function discoverPools({ publicClient, pairs, chainId, }: {
     publicClient: PublicClient;
     pairs: {
         tokenAddress: string;
         baseTokenAddress?: string;
     }[];
+    /** Chain ID used to resolve the WETH address. Defaults to Base (8453). */
+    chainId?: number;
 }): Promise<PoolDiscoveryResult[]>;
 /**
  * Discover the best WETH pool for a single token.
  * Convenience wrapper around discoverPools.
  */
-declare function discoverTokenPool({ publicClient, tokenAddress, }: {
+declare function discoverTokenPool({ publicClient, tokenAddress, chainId, }: {
     publicClient: PublicClient;
     tokenAddress: string;
+    /** Chain ID used to resolve the WETH address. Defaults to Base (8453). */
+    chainId?: number;
 }): Promise<PoolDiscoveryResult | null>;
 
 declare const SCORE_CONTRACT: {
@@ -216,8 +220,11 @@ declare const MULTI_VERSION_UNISWAP_POOL_INFO_RETRIEVER: {
     readonly address: Address;
     readonly abi: Abi;
 };
-declare const WETH_ADDRESS: Address;
+/**
+ * Get the WETH address for a given chain.
+ */
+declare function getWethAddress(chainId: number): Address;
 declare const NULL_ADDRESS: Address;
 declare const UPVOTE_PRICE_ETH = 0.000025;
 
-export { ALL_STRATEGY_ADDRESSES, DYNAMIC_SPLIT_STRATEGY, DecodedStrategyMetadata, GetAppKeyScoresOptions, GetStrategyKeyScoresOptions, GetUpvotesForItemsOptions, GetUpvotesOptions, LEGACY_UPVOTE_V1_ADDRESS, LEGACY_UPVOTE_V2_ADDRESS, MULTI_VERSION_UNISWAP_BULK_POOL_FINDER, MULTI_VERSION_UNISWAP_POOL_INFO_RETRIEVER, NULL_ADDRESS, PURE_ALPHA_STRATEGY, PoolDiscoveryResult, PoolKey, SCORE_CONTRACT, SUPPORTED_SCORE_CHAINS, ScoreClient, ScoreClientOptions, UNIV234_POOLS_STRATEGY, UPVOTE_APP, UPVOTE_PRICE_ETH, UPVOTE_STORAGE_APP, WETH_ADDRESS, calculatePriceFromSqrtPriceX96, decodeStrategyMetadata, decodeUpvoteMessage, decodeUpvoteStorageBlob, discoverPools, discoverTokenPool, encodePoolKey, encodeUpvoteKey, extractStrategyAddress, isDynamicSplitStrategy, isPureAlphaStrategy, isStrategyMessage, isUniv234PoolsStrategy, isUserUpvoteMessage, selectStrategy, tokenAddressToUpvoteKeyString };
+export { ALL_STRATEGY_ADDRESSES, DYNAMIC_SPLIT_STRATEGY, DecodedStrategyMetadata, GetAppKeyScoresOptions, GetStrategyKeyScoresOptions, GetUpvotesForItemsOptions, GetUpvotesOptions, LEGACY_UPVOTE_V1_ADDRESS, LEGACY_UPVOTE_V2_ADDRESS, MULTI_VERSION_UNISWAP_BULK_POOL_FINDER, MULTI_VERSION_UNISWAP_POOL_INFO_RETRIEVER, NULL_ADDRESS, PURE_ALPHA_STRATEGY, PoolDiscoveryResult, PoolKey, SCORE_CONTRACT, SUPPORTED_SCORE_CHAINS, ScoreClient, ScoreClientOptions, UNIV234_POOLS_STRATEGY, UPVOTE_APP, UPVOTE_PRICE_ETH, UPVOTE_STORAGE_APP, calculatePriceFromSqrtPriceX96, decodeStrategyMetadata, decodeUpvoteMessage, decodeUpvoteStorageBlob, discoverPools, discoverTokenPool, encodePoolKey, encodeUpvoteKey, extractStrategyAddress, getWethAddress, isDynamicSplitStrategy, isPureAlphaStrategy, isStrategyMessage, isUniv234PoolsStrategy, isUserUpvoteMessage, selectStrategy, tokenAddressToUpvoteKeyString };

package/dist/index.d.ts

@@ -164,20 +164,24 @@ declare function calculatePriceFromSqrtPriceX96(sqrtPriceX96: bigint, token0Deci
  * Discover the best Uniswap pools for multiple token pairs.
  * Makes two on-chain calls: pool discovery + pool info retrieval.
  */
-declare function discoverPools({ publicClient, pairs, }: {
+declare function discoverPools({ publicClient, pairs, chainId, }: {
     publicClient: PublicClient;
     pairs: {
         tokenAddress: string;
         baseTokenAddress?: string;
     }[];
+    /** Chain ID used to resolve the WETH address. Defaults to Base (8453). */
+    chainId?: number;
 }): Promise<PoolDiscoveryResult[]>;
 /**
  * Discover the best WETH pool for a single token.
  * Convenience wrapper around discoverPools.
  */
-declare function discoverTokenPool({ publicClient, tokenAddress, }: {
+declare function discoverTokenPool({ publicClient, tokenAddress, chainId, }: {
     publicClient: PublicClient;
     tokenAddress: string;
+    /** Chain ID used to resolve the WETH address. Defaults to Base (8453). */
+    chainId?: number;
 }): Promise<PoolDiscoveryResult | null>;
 
 declare const SCORE_CONTRACT: {
@@ -216,8 +220,11 @@ declare const MULTI_VERSION_UNISWAP_POOL_INFO_RETRIEVER: {
     readonly address: Address;
     readonly abi: Abi;
 };
-declare const WETH_ADDRESS: Address;
+/**
+ * Get the WETH address for a given chain.
+ */
+declare function getWethAddress(chainId: number): Address;
 declare const NULL_ADDRESS: Address;
 declare const UPVOTE_PRICE_ETH = 0.000025;
 
-export { ALL_STRATEGY_ADDRESSES, DYNAMIC_SPLIT_STRATEGY, DecodedStrategyMetadata, GetAppKeyScoresOptions, GetStrategyKeyScoresOptions, GetUpvotesForItemsOptions, GetUpvotesOptions, LEGACY_UPVOTE_V1_ADDRESS, LEGACY_UPVOTE_V2_ADDRESS, MULTI_VERSION_UNISWAP_BULK_POOL_FINDER, MULTI_VERSION_UNISWAP_POOL_INFO_RETRIEVER, NULL_ADDRESS, PURE_ALPHA_STRATEGY, PoolDiscoveryResult, PoolKey, SCORE_CONTRACT, SUPPORTED_SCORE_CHAINS, ScoreClient, ScoreClientOptions, UNIV234_POOLS_STRATEGY, UPVOTE_APP, UPVOTE_PRICE_ETH, UPVOTE_STORAGE_APP, WETH_ADDRESS, calculatePriceFromSqrtPriceX96, decodeStrategyMetadata, decodeUpvoteMessage, decodeUpvoteStorageBlob, discoverPools, discoverTokenPool, encodePoolKey, encodeUpvoteKey, extractStrategyAddress, isDynamicSplitStrategy, isPureAlphaStrategy, isStrategyMessage, isUniv234PoolsStrategy, isUserUpvoteMessage, selectStrategy, tokenAddressToUpvoteKeyString };
+export { ALL_STRATEGY_ADDRESSES, DYNAMIC_SPLIT_STRATEGY, DecodedStrategyMetadata, GetAppKeyScoresOptions, GetStrategyKeyScoresOptions, GetUpvotesForItemsOptions, GetUpvotesOptions, LEGACY_UPVOTE_V1_ADDRESS, LEGACY_UPVOTE_V2_ADDRESS, MULTI_VERSION_UNISWAP_BULK_POOL_FINDER, MULTI_VERSION_UNISWAP_POOL_INFO_RETRIEVER, NULL_ADDRESS, PURE_ALPHA_STRATEGY, PoolDiscoveryResult, PoolKey, SCORE_CONTRACT, SUPPORTED_SCORE_CHAINS, ScoreClient, ScoreClientOptions, UNIV234_POOLS_STRATEGY, UPVOTE_APP, UPVOTE_PRICE_ETH, UPVOTE_STORAGE_APP, calculatePriceFromSqrtPriceX96, decodeStrategyMetadata, decodeUpvoteMessage, decodeUpvoteStorageBlob, discoverPools, discoverTokenPool, encodePoolKey, encodeUpvoteKey, extractStrategyAddress, getWethAddress, isDynamicSplitStrategy, isPureAlphaStrategy, isStrategyMessage, isUniv234PoolsStrategy, isUserUpvoteMessage, selectStrategy, tokenAddressToUpvoteKeyString };

package/dist/index.js

@@ -1975,7 +1975,19 @@ var MULTI_VERSION_UNISWAP_POOL_INFO_RETRIEVER = {
   address: "0x7A9EF0AC6F6a254cd570B05D62D094D3aa5067f1",
   abi: multi_version_uniswap_pool_info_retriever_default
 };
-var WETH_ADDRESS = "0x4200000000000000000000000000000000000006";
+var WETH_BY_CHAIN = {
+  8453: "0x4200000000000000000000000000000000000006",
+  // Base (L2 predeploy)
+  1: "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2"
+  // Ethereum mainnet
+};
+function getWethAddress(chainId) {
+  const addr = WETH_BY_CHAIN[chainId];
+  if (!addr) {
+    throw new Error(`Score: No WETH address for chain ${chainId}`);
+  }
+  return addr;
+}
 var NULL_ADDRESS = "0x0000000000000000000000000000000000000000";
 var UPVOTE_PRICE_ETH = 25e-6;
 var encodeUpvoteKey = (tokenAddress) => {
@@ -2417,10 +2429,10 @@ var V4_HOOKS = [
 ];
 var LIQUIDITY_THRESHOLD_TO_PREFER_V2_V3_POOLS = 0.1 * 1e18;
 var LIQUIDITY_THRESHOLD_TO_CONSIDER_V2_V3_POOLS = 0.01 * 1e18;
-function normalizeTokenPairs(pairs) {
+function normalizeTokenPairs(pairs, wethAddress) {
   return pairs.map((pair) => {
     const tokenA = pair.tokenAddress;
-    const tokenB = pair.baseTokenAddress || WETH_ADDRESS;
+    const tokenB = pair.baseTokenAddress || wethAddress;
     return {
       originalPair: pair,
       tokenA: tokenA.toLowerCase() < tokenB.toLowerCase() ? tokenA : tokenB,
@@ -2437,7 +2449,7 @@ function buildDiscoveryArgs(normalizedPairs) {
     v4Hooks: V4_HOOKS
   };
 }
-function parsePoolDiscoveries(poolResults, pairs) {
+function parsePoolDiscoveries(poolResults, pairs, wethAddress) {
   if (!poolResults) {
     return {
       v2PoolAddresses: [],
@@ -2474,7 +2486,7 @@ function parsePoolDiscoveries(poolResults, pairs) {
         v2PoolAddresses.push(poolAddressValue);
         v2PoolAddressToPair[poolAddress] = {
           tokenAddress: pair.tokenAddress,
-          baseTokenAddress: pair.baseTokenAddress || WETH_ADDRESS,
+          baseTokenAddress: pair.baseTokenAddress || wethAddress,
           fee: feeNum
         };
       }
@@ -2485,7 +2497,7 @@ function parsePoolDiscoveries(poolResults, pairs) {
         v3PoolAddresses.push(poolAddressValue);
         v3PoolAddressToPair[poolAddress] = {
           tokenAddress: pair.tokenAddress,
-          baseTokenAddress: pair.baseTokenAddress || WETH_ADDRESS,
+          baseTokenAddress: pair.baseTokenAddress || wethAddress,
           fee: feeNum
         };
       }
@@ -2496,7 +2508,7 @@ function parsePoolDiscoveries(poolResults, pairs) {
         v4PoolKeys.push(poolKey);
         v4PoolKeyToPair[poolKeyString] = {
           tokenAddress: pair.tokenAddress,
-          baseTokenAddress: pair.baseTokenAddress || WETH_ADDRESS,
+          baseTokenAddress: pair.baseTokenAddress || wethAddress,
           fee: feeNum
         };
       }
@@ -2623,19 +2635,19 @@ function constructPoolKey(info, pair, version, v4PoolKey) {
     return void 0;
   }
 }
-function getWethBalanceWei(pool) {
-  if (pool.token0?.toLowerCase() === WETH_ADDRESS.toLowerCase()) {
+function getWethBalanceWei(pool, wethAddress) {
+  if (pool.token0?.toLowerCase() === wethAddress.toLowerCase()) {
     return Number(pool.token0Balance);
-  } else if (pool.token1?.toLowerCase() === WETH_ADDRESS.toLowerCase()) {
+  } else if (pool.token1?.toLowerCase() === wethAddress.toLowerCase()) {
     return Number(pool.token1Balance);
   } else {
     return Number(pool.baseTokenBalance);
   }
 }
-function filterV2V3PoolsByLiquidity(pools, threshold) {
+function filterV2V3PoolsByLiquidity(pools, threshold, wethAddress) {
   return pools.filter((pool) => {
     if (!pool.poolAddress) return false;
-    const wethBalanceWei = getWethBalanceWei(pool);
+    const wethBalanceWei = getWethBalanceWei(pool, wethAddress);
     return wethBalanceWei >= threshold;
   });
 }
@@ -2658,7 +2670,7 @@ function selectBestV2V3PoolByFee(pools) {
     return a.fee < b.fee ? a : b;
   });
 }
-function selectBestPoolPerPair(allPools) {
+function selectBestPoolPerPair(allPools, wethAddress) {
   const poolsByPair = {};
   for (const pool of allPools) {
     const key = pool.tokenAddress.toLowerCase() + "_" + pool.baseTokenAddress.toLowerCase();
@@ -2673,7 +2685,8 @@ function selectBestPoolPerPair(allPools) {
     } else {
       const v2v3PoolsWithPreferredLiquidity = filterV2V3PoolsByLiquidity(
         group,
-        LIQUIDITY_THRESHOLD_TO_PREFER_V2_V3_POOLS
+        LIQUIDITY_THRESHOLD_TO_PREFER_V2_V3_POOLS,
+        wethAddress
       );
       if (v2v3PoolsWithPreferredLiquidity.length > 0) {
         best = selectBestV2V3Pool(v2v3PoolsWithPreferredLiquidity);
@@ -2684,7 +2697,8 @@ function selectBestPoolPerPair(allPools) {
         } else {
           const v2v3PoolsWithAcceptableLiquidity = filterV2V3PoolsByLiquidity(
             group,
-            LIQUIDITY_THRESHOLD_TO_CONSIDER_V2_V3_POOLS
+            LIQUIDITY_THRESHOLD_TO_CONSIDER_V2_V3_POOLS,
+            wethAddress
           );
           if (v2v3PoolsWithAcceptableLiquidity.length > 0) {
             best = selectBestV2V3PoolByFee(v2v3PoolsWithAcceptableLiquidity);
@@ -2700,10 +2714,12 @@ function selectBestPoolPerPair(allPools) {
 }
 async function discoverPools({
   publicClient,
-  pairs
+  pairs,
+  chainId = 8453
 }) {
   if (pairs.length === 0) return [];
-  const normalizedPairs = normalizeTokenPairs(pairs);
+  const wethAddress = getWethAddress(chainId);
+  const normalizedPairs = normalizeTokenPairs(pairs, wethAddress);
   const discoveryArgs = buildDiscoveryArgs(normalizedPairs);
   const poolResults = await actions.readContract(publicClient, {
     address: MULTI_VERSION_UNISWAP_BULK_POOL_FINDER.address,
@@ -2711,7 +2727,7 @@ async function discoverPools({
     functionName: "getPoolsMultiVersion",
     args: [discoveryArgs]
   });
-  const discoveries = parsePoolDiscoveries(poolResults, pairs);
+  const discoveries = parsePoolDiscoveries(poolResults, pairs, wethAddress);
   const totalPools = discoveries.v2PoolAddresses.length + discoveries.v3PoolAddresses.length + discoveries.v4PoolKeys.length;
   if (totalPools === 0) return [];
   const poolInfos = await actions.readContract(publicClient, {
@@ -2722,7 +2738,7 @@ async function discoverPools({
       discoveries.v2PoolAddresses,
       discoveries.v3PoolAddresses,
       discoveries.v4PoolKeys,
-      WETH_ADDRESS
+      wethAddress
     ]
   });
   if (!Array.isArray(poolInfos) || poolInfos.length === 0) return [];
@@ -2731,7 +2747,7 @@ async function discoverPools({
     if (!poolData || !poolData.pair) return null;
     return {
       tokenAddress: poolData.pair.tokenAddress,
-      baseTokenAddress: poolData.pair.baseTokenAddress || WETH_ADDRESS,
+      baseTokenAddress: poolData.pair.baseTokenAddress || wethAddress,
       poolAddress: poolData.version === 4 ? null : info.poolAddress,
       price: calculatePoolPrice(info, poolData.pair, poolData.version),
       baseTokenBalance: String(info.baseTokenBalance || 0),
@@ -2748,7 +2764,7 @@ async function discoverPools({
       )
     };
   }).filter((p) => p !== null);
-  const bestPools = selectBestPoolPerPair(allPools);
+  const bestPools = selectBestPoolPerPair(allPools, wethAddress);
   return bestPools.map(
     (pool) => ({
       tokenAddress: pool.tokenAddress,
@@ -2767,11 +2783,13 @@ async function discoverPools({
 }
 async function discoverTokenPool({
   publicClient,
-  tokenAddress
+  tokenAddress,
+  chainId = 8453
 }) {
   const results = await discoverPools({
     publicClient,
-    pairs: [{ tokenAddress }]
+    pairs: [{ tokenAddress }],
+    chainId
   });
   return results[0] ?? null;
 }
@@ -2791,7 +2809,6 @@ exports.UNIV234_POOLS_STRATEGY = UNIV234_POOLS_STRATEGY;
 exports.UPVOTE_APP = UPVOTE_APP;
 exports.UPVOTE_PRICE_ETH = UPVOTE_PRICE_ETH;
 exports.UPVOTE_STORAGE_APP = UPVOTE_STORAGE_APP;
-exports.WETH_ADDRESS = WETH_ADDRESS;
 exports.calculatePriceFromSqrtPriceX96 = calculatePriceFromSqrtPriceX96;
 exports.decodeStrategyMetadata = decodeStrategyMetadata;
 exports.decodeUpvoteMessage = decodeUpvoteMessage;
@@ -2807,6 +2824,7 @@ exports.getScoreKey = getScoreKey;
 exports.getStorageScoreKey = getStorageScoreKey;
 exports.getStorageUpvoteContext = getStorageUpvoteContext;
 exports.getTokenScoreKey = getTokenScoreKey;
+exports.getWethAddress = getWethAddress;
 exports.isDynamicSplitStrategy = isDynamicSplitStrategy;
 exports.isPureAlphaStrategy = isPureAlphaStrategy;
 exports.isStrategyMessage = isStrategyMessage;