@paraswap/dex-lib 4.7.18-before-multi-route.0 → 4.7.18

package/build/dex/uniswap-v4/hooks/template.d.ts

@@ -0,0 +1,150 @@
+import { SwapSide } from '@paraswap/core';
+import { DeepReadonly } from 'ts-essentials';
+import { IDexHelper } from '../../../dex-helper';
+import { Address, Log, Logger } from '../../../types';
+import { Pool, PoolState, SubgraphPool } from '../types';
+/**
+ * Hook permission bits used by the Uniswap v4 hook address encoding.
+ * They match https://docs.uniswap.org/contracts/v4/concepts/hooks#permissions.
+ * Re-using Kyber's mapping so downstream teams have a familiar API.
+ */
+export declare enum HookPermission {
+    AFTER_REMOVE_LIQUIDITY_RETURNS_DELTA = 0,
+    AFTER_ADD_LIQUIDITY_RETURNS_DELTA = 1,
+    AFTER_SWAP_RETURNS_DELTA = 2,
+    BEFORE_SWAP_RETURNS_DELTA = 3,
+    AFTER_DONATE = 4,
+    BEFORE_DONATE = 5,
+    AFTER_SWAP = 6,
+    BEFORE_SWAP = 7,
+    AFTER_REMOVE_LIQUIDITY = 8,
+    BEFORE_REMOVE_LIQUIDITY = 9,
+    AFTER_ADD_LIQUIDITY = 10,
+    BEFORE_ADD_LIQUIDITY = 11,
+    AFTER_INITIALIZE = 12,
+    BEFORE_INITIALIZE = 13
+}
+export declare const hasHookPermission: (address: Address, permission: HookPermission) => boolean;
+export declare const canBeforeSwap: (address: Address) => boolean;
+export declare const canAfterSwap: (address: Address) => boolean;
+export type HookFactoryParams = {
+    dexHelper: IDexHelper;
+    dexKey: string;
+    logger: Logger;
+    network: number;
+    hookAddress: Address;
+};
+export type HookBootstrapParams = HookFactoryParams & {
+    pool: SubgraphPool;
+};
+export type HookStateContext = HookFactoryParams & {
+    pool: Pool;
+    poolState: DeepReadonly<PoolState>;
+    blockNumber: number;
+};
+export type HookBeforeSwapParams = HookStateContext & {
+    amountSpecified: bigint;
+    zeroForOne: boolean;
+    side: SwapSide;
+};
+export type HookBeforeSwapResult = {
+    /**
+     * Optional override for the amount specified that will be forwarded to pool math.
+     * Useful for hooks that deduct their own fee prior to the swap.
+     */
+    amountSpecified?: bigint;
+    /**
+     * Override for lpFee if the hook is dynamically adjusting the pool fee.
+     */
+    lpFeeOverride?: bigint;
+    /**
+     * Extra calldata that needs to be appended to router.swap parameters.
+     */
+    hookData?: string;
+    /**
+     * Optional state patch that will be merged in-memory before simulation continues.
+     */
+    statePatch?: Partial<PoolState>;
+    /**
+     * Free-form metadata (e.g. diagnostics, gas estimates) propagated to afterSwap.
+     */
+    metadata?: Record<string, unknown>;
+};
+export type HookAfterSwapParams = HookStateContext & {
+    amountIn: bigint;
+    amountOut: bigint;
+    zeroForOne: boolean;
+    side: SwapSide;
+    metadata?: Record<string, unknown>;
+};
+export type HookAfterSwapResult = {
+    deltaAmountIn?: bigint;
+    deltaAmountOut?: bigint;
+    statePatch?: Partial<PoolState>;
+};
+export type HookEventParams = HookFactoryParams & {
+    pool: Pool;
+    poolState: PoolState;
+    eventName: string;
+    eventArgs: any;
+    log: Log;
+    blockNumber: number;
+};
+export interface IUniswapV4Hook {
+    readonly name: string;
+    readonly hookAddress: Address;
+    readonly description?: string;
+    /**
+     * Optional hook-level initialization invoked once after the pool list is fetched.
+     */
+    bootstrap?(params: HookBootstrapParams): Promise<void>;
+    /**
+     * Allows hooks to filter pools beyond address equality (e.g. min TVL, custom fee).
+     */
+    supportsPool(pool: SubgraphPool): boolean;
+    /**
+     * Gives hooks a chance to enrich pool state (e.g. fetch additional storage slots).
+     */
+    extendPoolState?(context: HookStateContext): Promise<Partial<PoolState> | void>;
+    /**
+     * Invoked before SwapMath is executed to let hooks override inputs or provide hookData.
+     */
+    beforeSwap?(params: HookBeforeSwapParams): Promise<HookBeforeSwapResult | void>;
+    /**
+     * Invoked after pool math completes so hooks can settle their internal accounting.
+     */
+    afterSwap?(params: HookAfterSwapParams): Promise<HookAfterSwapResult | void>;
+    /**
+     * Hooks may request direct access to PoolManager events for bespoke state sync.
+     */
+    handlePoolEvent?(params: HookEventParams): Promise<Partial<PoolState> | void>;
+    /**
+     * Destroy internal caches.
+     */
+    destroy?(): void;
+}
+export declare abstract class BaseUniswapV4Hook implements IUniswapV4Hook {
+    readonly hookAddress: Address;
+    readonly name: string;
+    protected readonly params: HookFactoryParams;
+    readonly description?: string;
+    protected constructor(hookAddress: Address, name: string, params: HookFactoryParams);
+    bootstrap(_params: HookBootstrapParams): Promise<void>;
+    supportsPool(pool: SubgraphPool): boolean;
+    extendPoolState(_context: HookStateContext): Promise<Partial<PoolState> | void>;
+    beforeSwap(_params: HookBeforeSwapParams): Promise<HookBeforeSwapResult | void>;
+    afterSwap(_params: HookAfterSwapParams): Promise<HookAfterSwapResult | void>;
+    handlePoolEvent(_params: HookEventParams): Promise<Partial<PoolState> | void>;
+    destroy(): void;
+    protected get logger(): Logger;
+    protected get dexHelper(): IDexHelper;
+}
+export type HookFactory = (params: HookFactoryParams) => IUniswapV4Hook;
+export declare class UniswapV4HookRegistry {
+    private readonly factories;
+    registerFactory(address: Address, factory: HookFactory): this;
+    unregisterFactory(address: Address): boolean;
+    getFactory(address: Address): HookFactory | undefined;
+    createHook(address: Address, params: Omit<HookFactoryParams, 'hookAddress'>): IUniswapV4Hook | undefined;
+}
+export declare const uniswapV4HookRegistry: UniswapV4HookRegistry;

package/build/dex/uniswap-v4/hooks/template.js

@@ -0,0 +1,104 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.uniswapV4HookRegistry = exports.UniswapV4HookRegistry = exports.BaseUniswapV4Hook = exports.canAfterSwap = exports.canBeforeSwap = exports.hasHookPermission = exports.HookPermission = void 0;
+/**
+ * Hook permission bits used by the Uniswap v4 hook address encoding.
+ * They match https://docs.uniswap.org/contracts/v4/concepts/hooks#permissions.
+ * Re-using Kyber's mapping so downstream teams have a familiar API.
+ */
+var HookPermission;
+(function (HookPermission) {
+    HookPermission[HookPermission["AFTER_REMOVE_LIQUIDITY_RETURNS_DELTA"] = 0] = "AFTER_REMOVE_LIQUIDITY_RETURNS_DELTA";
+    HookPermission[HookPermission["AFTER_ADD_LIQUIDITY_RETURNS_DELTA"] = 1] = "AFTER_ADD_LIQUIDITY_RETURNS_DELTA";
+    HookPermission[HookPermission["AFTER_SWAP_RETURNS_DELTA"] = 2] = "AFTER_SWAP_RETURNS_DELTA";
+    HookPermission[HookPermission["BEFORE_SWAP_RETURNS_DELTA"] = 3] = "BEFORE_SWAP_RETURNS_DELTA";
+    HookPermission[HookPermission["AFTER_DONATE"] = 4] = "AFTER_DONATE";
+    HookPermission[HookPermission["BEFORE_DONATE"] = 5] = "BEFORE_DONATE";
+    HookPermission[HookPermission["AFTER_SWAP"] = 6] = "AFTER_SWAP";
+    HookPermission[HookPermission["BEFORE_SWAP"] = 7] = "BEFORE_SWAP";
+    HookPermission[HookPermission["AFTER_REMOVE_LIQUIDITY"] = 8] = "AFTER_REMOVE_LIQUIDITY";
+    HookPermission[HookPermission["BEFORE_REMOVE_LIQUIDITY"] = 9] = "BEFORE_REMOVE_LIQUIDITY";
+    HookPermission[HookPermission["AFTER_ADD_LIQUIDITY"] = 10] = "AFTER_ADD_LIQUIDITY";
+    HookPermission[HookPermission["BEFORE_ADD_LIQUIDITY"] = 11] = "BEFORE_ADD_LIQUIDITY";
+    HookPermission[HookPermission["AFTER_INITIALIZE"] = 12] = "AFTER_INITIALIZE";
+    HookPermission[HookPermission["BEFORE_INITIALIZE"] = 13] = "BEFORE_INITIALIZE";
+})(HookPermission || (exports.HookPermission = HookPermission = {}));
+const ADDRESS_HEX_PREFIX = '0x';
+const normalizeAddress = (address) => address.toLowerCase().startsWith(ADDRESS_HEX_PREFIX)
+    ? address.toLowerCase()
+    : `${ADDRESS_HEX_PREFIX}${address.toLowerCase()}`;
+const decodePermissionBits = (address) => {
+    const normalized = normalizeAddress(address).slice(2).padStart(40, '0');
+    const lastTwoBytes = normalized.slice(-4);
+    return parseInt(lastTwoBytes, 16);
+};
+const hasHookPermission = (address, permission) => {
+    const bits = decodePermissionBits(address);
+    return (bits & (1 << permission)) !== 0;
+};
+exports.hasHookPermission = hasHookPermission;
+const canBeforeSwap = (address) => (0, exports.hasHookPermission)(address, HookPermission.BEFORE_SWAP);
+exports.canBeforeSwap = canBeforeSwap;
+const canAfterSwap = (address) => (0, exports.hasHookPermission)(address, HookPermission.AFTER_SWAP);
+exports.canAfterSwap = canAfterSwap;
+class BaseUniswapV4Hook {
+    hookAddress;
+    name;
+    params;
+    description;
+    constructor(hookAddress, name, params) {
+        this.hookAddress = hookAddress;
+        this.name = name;
+        this.params = params;
+    }
+    async bootstrap(_params) {
+        // No-op by default.
+    }
+    supportsPool(pool) {
+        return normalizeAddress(pool.hooks) === normalizeAddress(this.hookAddress);
+    }
+    async extendPoolState(_context) {
+        return undefined;
+    }
+    async beforeSwap(_params) {
+        return undefined;
+    }
+    async afterSwap(_params) {
+        return undefined;
+    }
+    async handlePoolEvent(_params) {
+        return undefined;
+    }
+    destroy() {
+        // No-op by default.
+    }
+    get logger() {
+        return this.params.logger;
+    }
+    get dexHelper() {
+        return this.params.dexHelper;
+    }
+}
+exports.BaseUniswapV4Hook = BaseUniswapV4Hook;
+class UniswapV4HookRegistry {
+    factories = new Map();
+    registerFactory(address, factory) {
+        this.factories.set(normalizeAddress(address), factory);
+        return this;
+    }
+    unregisterFactory(address) {
+        return this.factories.delete(normalizeAddress(address));
+    }
+    getFactory(address) {
+        return this.factories.get(normalizeAddress(address));
+    }
+    createHook(address, params) {
+        const factory = this.getFactory(address);
+        if (!factory)
+            return undefined;
+        return factory({ ...params, hookAddress: normalizeAddress(address) });
+    }
+}
+exports.UniswapV4HookRegistry = UniswapV4HookRegistry;
+exports.uniswapV4HookRegistry = new UniswapV4HookRegistry();
+//# sourceMappingURL=template.js.map

package/build/dex/uniswap-v4/hooks/template.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"template.js","sourceRoot":"","sources":["../../../../src/dex/uniswap-v4/hooks/template.ts"],"names":[],"mappings":";;;AAOA;;;;GAIG;AACH,IAAY,cAeX;AAfD,WAAY,cAAc;IACxB,mHAAwC,CAAA;IACxC,6GAAqC,CAAA;IACrC,2FAA4B,CAAA;IAC5B,6FAA6B,CAAA;IAC7B,mEAAgB,CAAA;IAChB,qEAAiB,CAAA;IACjB,+DAAc,CAAA;IACd,iEAAe,CAAA;IACf,uFAA0B,CAAA;IAC1B,yFAA2B,CAAA;IAC3B,kFAAwB,CAAA;IACxB,oFAAyB,CAAA;IACzB,4EAAqB,CAAA;IACrB,8EAAsB,CAAA;AACxB,CAAC,EAfW,cAAc,8BAAd,cAAc,QAezB;AAED,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAEhC,MAAM,gBAAgB,GAAG,CAAC,OAAgB,EAAU,EAAE,CACpD,OAAO,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,kBAAkB,CAAC;IAClD,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE;IACvB,CAAC,CAAC,GAAG,kBAAkB,GAAG,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;AAEtD,MAAM,oBAAoB,GAAG,CAAC,OAAgB,EAAU,EAAE;IACxD,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;IACxE,MAAM,YAAY,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1C,OAAO,QAAQ,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;AACpC,CAAC,CAAC;AAEK,MAAM,iBAAiB,GAAG,CAC/B,OAAgB,EAChB,UAA0B,EACjB,EAAE;IACX,MAAM,IAAI,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAC3C,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC;AAC1C,CAAC,CAAC;AANW,QAAA,iBAAiB,qBAM5B;AAEK,MAAM,aAAa,GAAG,CAAC,OAAgB,EAAW,EAAE,CACzD,IAAA,yBAAiB,EAAC,OAAO,EAAE,cAAc,CAAC,WAAW,CAAC,CAAC;AAD5C,QAAA,aAAa,iBAC+B;AAElD,MAAM,YAAY,GAAG,CAAC,OAAgB,EAAW,EAAE,CACxD,IAAA,yBAAiB,EAAC,OAAO,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;AAD3C,QAAA,YAAY,gBAC+B;AAsHxD,MAAsB,iBAAiB;IAInB;IACA;IACG;IALZ,WAAW,CAAU;IAE9B,YACkB,WAAoB,EACpB,IAAY,EACT,MAAyB;QAF5B,gBAAW,GAAX,WAAW,CAAS;QACpB,SAAI,GAAJ,IAAI,CAAQ;QACT,WAAM,GAAN,MAAM,CAAmB;IAC3C,CAAC;IAEJ,KAAK,CAAC,SAAS,CAAC,OAA4B;QAC1C,oBAAoB;IACtB,CAAC;IAED,YAAY,CAAC,IAAkB;QAC7B,OAAO,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,QAA0B;QAE1B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,UAAU,CACd,OAA6B;QAE7B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,SAAS,CACb,OAA4B;QAE5B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,OAAwB;QAExB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO;QACL,oBAAoB;IACtB,CAAC;IAED,IAAc,MAAM;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;IAC5B,CAAC;IAED,IAAc,SAAS;QACrB,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;IAC/B,CAAC;CACF;AApDD,8CAoDC;AAID,MAAa,qBAAqB;IACf,SAAS,GAAG,IAAI,GAAG,EAAuB,CAAC;IAE5D,eAAe,CAAC,OAAgB,EAAE,OAAoB;QACpD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,iBAAiB,CAAC,OAAgB;QAChC,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED,UAAU,CAAC,OAAgB;QACzB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,UAAU,CACR,OAAgB,EAChB,MAA8C;QAE9C,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAC/B,OAAO,OAAO,CAAC,EAAE,GAAG,MAAM,EAAE,WAAW,EAAE,gBAAgB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACxE,CAAC;CACF;AAxBD,sDAwBC;AAEY,QAAA,qBAAqB,GAAG,IAAI,qBAAqB,EAAE,CAAC"}

package/build/dex/uniswap-v4/hooks/types.d.ts

@@ -53,6 +53,10 @@ export interface IBaseHook {
     afterAddLiquidity?(sender: string, key: PoolKey, params: ModifyLiquidityParams, delta0: BalanceDelta, delta1: BalanceDelta, hookData: string): [string, BalanceDelta];
     afterRemoveLiquidity?(sender: string, key: PoolKey, params: ModifyLiquidityParams, delta0: BalanceDelta, delta1: BalanceDelta, hookData: string): [string, BalanceDelta];
     beforeSwap?(sender: string, key: PoolKey, params: SwapParams, hookData: string): [string, BeforeSwapDelta, number];
+    /**
+     * The hook implementing this method should return the final
+     * modified amount exactly as swapExactAmountIn/Out would return
+     */
     afterSwap?(sender: string, key: PoolKey, params: SwapParams, delta: BalanceDelta, hookData: string): bigint;
     beforeDonate?(sender: string, key: PoolKey, amount0: string, amount1: string, hookData: string): string;
     afterDonate?(sender: string, key: PoolKey, amount0: string, amount1: string, hookData: string): string;

package/build/dex/uniswap-v4/subgraph.d.ts

@@ -3,10 +3,10 @@ import { IDexHelper } from '../../dex-helper';
 import { Logger } from 'log4js';
 import { Address } from '@paraswap/core';
 export declare function queryTicksForPool(dexHelper: IDexHelper, logger: Logger, dexKey: string, subgraphUrl: string, blockNumber: number, id: string, skip: number, limit: number, latestBlock?: boolean): Promise<SubgraphTick[]>;
-export declare function queryAvailablePoolsForToken(dexHelper: IDexHelper, logger: Logger, dexKey: string, subgraphUrl: string, tokenAddress: string, limit: number, staticPoolsList?: string[]): Promise<{
+export declare function queryAvailablePoolsForToken(dexHelper: IDexHelper, logger: Logger, dexKey: string, subgraphUrl: string, tokenAddress: string, limit: number, staticPoolsList?: string[], hooks?: string[]): Promise<{
     pools0: SubgraphConnectorPool[];
     pools1: SubgraphConnectorPool[];
 }>;
-export declare function queryAvailablePoolsForPairFromSubgraph(dexHelper: IDexHelper, subgraphUrl: string, srcToken: Address, destToken: Address): Promise<SubgraphPool[]>;
-export declare function queryOnePageForAllAvailablePoolsFromSubgraph(dexHelper: IDexHelper, logger: Logger, dexKey: string, subgraphUrl: string, blockNumber: number, skip: number, limit: number, latestBlock?: boolean): Promise<SubgraphPool[]>;
-export declare function queryPoolsFromSubgraph(dexHelper: IDexHelper, subgraphUrl: string, poolIds: string[]): Promise<SubgraphPool[] | null>;
+export declare function queryAvailablePoolsForPairFromSubgraph(dexHelper: IDexHelper, subgraphUrl: string, srcToken: Address, destToken: Address, hooks?: string[]): Promise<SubgraphPool[]>;
+export declare function queryOnePageForAllAvailablePoolsFromSubgraph(dexHelper: IDexHelper, logger: Logger, dexKey: string, subgraphUrl: string, blockNumber: number, skip: number, limit: number, hooks?: string[], latestBlock?: boolean): Promise<SubgraphPool[]>;
+export declare function queryPoolsFromSubgraph(dexHelper: IDexHelper, subgraphUrl: string, poolIds: string[], hooks?: string[]): Promise<SubgraphPool[]>;

package/build/dex/uniswap-v4/subgraph.js

@@ -37,15 +37,15 @@ async function queryTicksForPool(dexHelper, logger, dexKey, subgraphUrl, blockNu
     }
     return res.data.ticks || [];
 }
-async function queryAvailablePoolsForToken(dexHelper, logger, dexKey, subgraphUrl, tokenAddress, limit, staticPoolsList) {
+async function queryAvailablePoolsForToken(dexHelper, logger, dexKey, subgraphUrl, tokenAddress, limit, staticPoolsList, hooks = [constants_2.NULL_ADDRESS]) {
     const list = staticPoolsList
         ? staticPoolsList.map(t => `"${t}"`).join(',')
         : '';
-    const poolsQuery = `query ($token: Bytes!, $hooks: Bytes!, $minTVL: Int!, $count: Int) {
+    const poolsQuery = `query ($token: Bytes!, $hooks: [Bytes!], $minTVL: Int!, $count: Int) {
     pools0: pools(
       where: {
         token0: $token
-        hooks: $hooks
+        hooks_in: $hooks
         liquidity_gt: 0
         totalValueLockedUSD_gte: $minTVL
         ${list ? `id_in: [${list}]` : ''}
@@ -71,7 +71,7 @@ async function queryAvailablePoolsForToken(dexHelper, logger, dexKey, subgraphUr
     pools1: pools(
       where: {
         token1: $token
-        hooks: $hooks
+        hooks_in: $hooks
         liquidity_gt: 0
         totalValueLockedUSD_gte: $minTVL
         ${list ? `id_in: [${list}]` : ''}
@@ -101,20 +101,20 @@ async function queryAvailablePoolsForToken(dexHelper, logger, dexKey, subgraphUr
         variables: {
             token: tokenAddress,
             count: limit,
-            hooks: constants_2.NULL_ADDRESS,
+            hooks,
             minTVL: constants_1.POOL_MIN_TVL_USD,
         },
     }, { timeout: constants_1.SUBGRAPH_TIMEOUT });
     if (res.errors && res.errors.length) {
         throw new Error(res.errors[0].message);
     }
-    return { pools0: res.data.pools0, pools1: res.data.pools1 };
+    return res.data;
 }
-async function queryAvailablePoolsForPairFromSubgraph(dexHelper, subgraphUrl, srcToken, destToken) {
+async function queryAvailablePoolsForPairFromSubgraph(dexHelper, subgraphUrl, srcToken, destToken, hooks = [constants_2.NULL_ADDRESS]) {
     const ticksLimit = 300;
-    const poolsQuery = `query ($token0: Bytes!, $token1: Bytes!, $minTVL: Int!, $hooks: Bytes!) {
+    const poolsQuery = `query ($token0: Bytes!, $token1: Bytes!, $minTVL: Int!, $hooks: [Bytes!]) {
       pools(
-        where: { token0: $token0, token1: $token1, hooks: $hooks, liquidity_gt: 0, totalValueLockedUSD_gte: $minTVL },
+        where: { token0: $token0, token1: $token1, hooks_in: $hooks, liquidity_gt: 0, totalValueLockedUSD_gte: $minTVL },
         orderBy: totalValueLockedUSD
         orderDirection: desc
       ) {
@@ -146,7 +146,7 @@ async function queryAvailablePoolsForPairFromSubgraph(dexHelper, subgraphUrl, sr
             token0,
             token1,
             minTVL: constants_1.POOL_MIN_TVL_USD,
-            hooks: constants_2.NULL_ADDRESS,
+            hooks,
         },
     }, { timeout: constants_1.SUBGRAPH_TIMEOUT });
     if (res.errors && res.errors.length) {
@@ -154,10 +154,10 @@ async function queryAvailablePoolsForPairFromSubgraph(dexHelper, subgraphUrl, sr
     }
     return res.data.pools;
 }
-async function queryOnePageForAllAvailablePoolsFromSubgraph(dexHelper, logger, dexKey, subgraphUrl, blockNumber, skip, limit, latestBlock = false) {
-    const poolsQuery = `query ($skip: Int!, $minTVL: Int!, $hooks: Bytes!) {
+async function queryOnePageForAllAvailablePoolsFromSubgraph(dexHelper, logger, dexKey, subgraphUrl, blockNumber, skip, limit, hooks = [constants_2.NULL_ADDRESS], latestBlock = false) {
+    const poolsQuery = `query ($skip: Int!, $minTVL: Int!, $hooks: [Bytes!]) {
       pools(
-        where: { hooks: $hooks, liquidity_gt: 0, totalValueLockedUSD_gte: $minTVL },
+        where: { hooks_in: $hooks, liquidity_gt: 0, totalValueLockedUSD_gte: $minTVL },
         ${latestBlock ? '' : `block: { number: ${blockNumber} }`}
         orderBy: totalValueLockedUSD
         orderDirection: desc
@@ -180,15 +180,15 @@ async function queryOnePageForAllAvailablePoolsFromSubgraph(dexHelper, logger, d
     const res = await dexHelper.httpRequest.querySubgraph(subgraphUrl, {
         query: poolsQuery,
         variables: {
-            skip: skip,
-            hooks: constants_2.NULL_ADDRESS,
+            skip,
+            hooks,
             minTVL: constants_1.POOL_MIN_TVL_USD,
         },
     }, { timeout: constants_1.SUBGRAPH_TIMEOUT });
     if (res.errors && res.errors.length) {
         if (res.errors[0].message.includes('missing block')) {
             logger.info(`${dexKey}: subgraph fallback to the latest block...`);
-            return queryOnePageForAllAvailablePoolsFromSubgraph(dexHelper, logger, dexKey, subgraphUrl, blockNumber, skip, limit, true);
+            return queryOnePageForAllAvailablePoolsFromSubgraph(dexHelper, logger, dexKey, subgraphUrl, blockNumber, skip, limit, hooks, true);
         }
         else {
             throw new Error(res.errors[0].message);
@@ -196,9 +196,9 @@ async function queryOnePageForAllAvailablePoolsFromSubgraph(dexHelper, logger, d
     }
     return res.data.pools;
 }
-async function queryPoolsFromSubgraph(dexHelper, subgraphUrl, poolIds) {
-    const poolsQuery = `query ($minTVL: Int!, $hooks: Bytes!, $pools: [Bytes!]!) {
-      pools(where: {liquidity_gt: 0, totalValueLockedUSD_gte: $minTVL, id_in: $pools}) {
+async function queryPoolsFromSubgraph(dexHelper, subgraphUrl, poolIds, hooks = [constants_2.NULL_ADDRESS]) {
+    const poolsQuery = `query ($minTVL: Int!, $hooks: [Bytes!], $pools: [Bytes!]!) {
+      pools(where: {liquidity_gt: 0, totalValueLockedUSD_gte: $minTVL, id_in: $pools, hooks_in: $hooks}) {
         id
         fee: feeTier
         volumeUSD
@@ -215,11 +215,14 @@ async function queryPoolsFromSubgraph(dexHelper, subgraphUrl, poolIds) {
     const res = await dexHelper.httpRequest.querySubgraph(subgraphUrl, {
         query: poolsQuery,
         variables: {
-            hooks: constants_2.NULL_ADDRESS,
+            hooks,
             minTVL: constants_1.POOL_MIN_TVL_USD,
             pools: poolIds,
         },
     }, { timeout: constants_1.SUBGRAPH_TIMEOUT });
-    return res?.data?.pools ?? null;
+    if (res.errors && res.errors.length) {
+        throw new Error(res.errors[0].message);
+    }
+    return res?.data?.pools ?? [];
 }
 //# sourceMappingURL=subgraph.js.map

package/build/dex/uniswap-v4/subgraph.js.map

@@ -1 +1 @@
-{"version":3,"file":"subgraph.js","sourceRoot":"","sources":["../../../src/dex/uniswap-v4/subgraph.ts"],"names":[],"mappings":";;AAOA,8CA8DC;AAED,kEAgGC;AAED,wFA+DC;AAED,oGAsEC;AAED,wDAwCC;AAzVD,2CAAiE;AAIjE,+CAA+C;AAExC,KAAK,UAAU,iBAAiB,CACrC,SAAqB,EACrB,MAAc,EACd,MAAc,EACd,WAAmB,EACnB,WAAmB,EACnB,EAAU,EACV,IAAY,EACZ,KAAa,EACb,WAAW,GAAG,KAAK;IAEnB,MAAM,UAAU,GAAG;;;iBAGJ,KAAK;;;;UAIZ,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB,WAAW,IAAI;;;;;;IAM5D,CAAC;IAEH,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAMnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE;KAChC,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACpD,MAAM,CAAC,IAAI,CACT,GAAG,MAAM,wDAAwD,CAClE,CAAC;YACF,OAAO,iBAAiB,CACtB,SAAS,EACT,MAAM,EACN,MAAM,EACN,WAAW,EACX,WAAW,EACX,EAAE,EACF,IAAI,EACJ,KAAK,EACL,IAAI,CACL,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;AAC9B,CAAC;AAEM,KAAK,UAAU,2BAA2B,CAC/C,SAAqB,EACrB,MAAc,EACd,MAAc,EACd,WAAmB,EACnB,YAAoB,EACpB,KAAa,EACb,eAA0B;IAK1B,MAAM,IAAI,GAAG,eAAe;QAC1B,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;QAC9C,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,UAAU,GAAG;;;;;;;UAOX,IAAI,CAAC,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;UA0B9B,IAAI,CAAC,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE;;;;;;;;;;;;;;;;;;;;;CAqBvC,CAAC;IAEA,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAOnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE;YACT,KAAK,EAAE,YAAY;YACnB,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,wBAAY;YACnB,MAAM,EAAE,4BAAgB;SACzB;KACF,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;AAC9D,CAAC;AAEM,KAAK,UAAU,sCAAsC,CAC1D,SAAqB,EACrB,WAAmB,EACnB,QAAiB,EACjB,SAAkB;IAElB,MAAM,UAAU,GAAG,GAAG,CAAC;IAEvB,MAAM,UAAU,GAAG;;;;;;;;;;;;;;;;;uBAiBE,UAAU;;;;;;;MAO3B,CAAC;IAEL,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACpB,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC,GAAG,QAAQ,CAAC,SAAS,EAAE,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC;QACvB,CAAC,CAAC,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAE5B,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAMnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE;YACT,MAAM;YACN,MAAM;YACN,MAAM,EAAE,4BAAgB;YACxB,KAAK,EAAE,wBAAY;SACpB;KACF,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC;AACxB,CAAC;AAEM,KAAK,UAAU,4CAA4C,CAChE,SAAqB,EACrB,MAAc,EACd,MAAc,EACd,WAAmB,EACnB,WAAmB,EACnB,IAAY,EACZ,KAAa,EACb,WAAW,GAAG,KAAK;IAEnB,MAAM,UAAU,GAAG;;;UAGX,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB,WAAW,IAAI;;;;iBAI/C,KAAK;;;;;;;;;;;;;;MAchB,CAAC;IAEL,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAMnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE;YACT,IAAI,EAAE,IAAI;YACV,KAAK,EAAE,wBAAY;YACnB,MAAM,EAAE,4BAAgB;SACzB;KACF,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,4CAA4C,CAAC,CAAC;YACnE,OAAO,4CAA4C,CACjD,SAAS,EACT,MAAM,EACN,MAAM,EACN,WAAW,EACX,WAAW,EACX,IAAI,EACJ,KAAK,EACL,IAAI,CACL,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC;AACxB,CAAC;AAEM,KAAK,UAAU,sBAAsB,CAC1C,SAAqB,EACrB,WAAmB,EACnB,OAAiB;IAEjB,MAAM,UAAU,GAAG;;;;;;;;;;;;;;MAcf,CAAC;IAEL,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAMnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE;YACT,KAAK,EAAE,wBAAY;YACnB,MAAM,EAAE,4BAAgB;YACxB,KAAK,EAAE,OAAO;SACf;KACF,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,OAAO,GAAG,EAAE,IAAI,EAAE,KAAK,IAAI,IAAI,CAAC;AAClC,CAAC"}
+{"version":3,"file":"subgraph.js","sourceRoot":"","sources":["../../../src/dex/uniswap-v4/subgraph.ts"],"names":[],"mappings":";;AAOA,8CA8DC;AAED,kEAiGC;AAED,wFAgEC;AAED,oGAwEC;AAED,wDA6CC;AAlWD,2CAAiE;AAIjE,+CAA+C;AAExC,KAAK,UAAU,iBAAiB,CACrC,SAAqB,EACrB,MAAc,EACd,MAAc,EACd,WAAmB,EACnB,WAAmB,EACnB,EAAU,EACV,IAAY,EACZ,KAAa,EACb,WAAW,GAAG,KAAK;IAEnB,MAAM,UAAU,GAAG;;;iBAGJ,KAAK;;;;UAIZ,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB,WAAW,IAAI;;;;;;IAM5D,CAAC;IAEH,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAMnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE;KAChC,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACpD,MAAM,CAAC,IAAI,CACT,GAAG,MAAM,wDAAwD,CAClE,CAAC;YACF,OAAO,iBAAiB,CACtB,SAAS,EACT,MAAM,EACN,MAAM,EACN,WAAW,EACX,WAAW,EACX,EAAE,EACF,IAAI,EACJ,KAAK,EACL,IAAI,CACL,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;AAC9B,CAAC;AAEM,KAAK,UAAU,2BAA2B,CAC/C,SAAqB,EACrB,MAAc,EACd,MAAc,EACd,WAAmB,EACnB,YAAoB,EACpB,KAAa,EACb,eAA0B,EAC1B,QAAkB,CAAC,wBAAY,CAAC;IAKhC,MAAM,IAAI,GAAG,eAAe;QAC1B,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;QAC9C,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,UAAU,GAAG;;;;;;;UAOX,IAAI,CAAC,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;UA0B9B,IAAI,CAAC,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE;;;;;;;;;;;;;;;;;;;;;CAqBvC,CAAC;IAEA,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAOnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE;YACT,KAAK,EAAE,YAAY;YACnB,KAAK,EAAE,KAAK;YACZ,KAAK;YACL,MAAM,EAAE,4BAAgB;SACzB;KACF,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,GAAG,CAAC,IAAI,CAAC;AAClB,CAAC;AAEM,KAAK,UAAU,sCAAsC,CAC1D,SAAqB,EACrB,WAAmB,EACnB,QAAiB,EACjB,SAAkB,EAClB,QAAkB,CAAC,wBAAY,CAAC;IAEhC,MAAM,UAAU,GAAG,GAAG,CAAC;IAEvB,MAAM,UAAU,GAAG;;;;;;;;;;;;;;;;;uBAiBE,UAAU;;;;;;;MAO3B,CAAC;IAEL,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACpB,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC,GAAG,QAAQ,CAAC,SAAS,EAAE,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC;QACvB,CAAC,CAAC,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAE5B,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAMnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE;YACT,MAAM;YACN,MAAM;YACN,MAAM,EAAE,4BAAgB;YACxB,KAAK;SACN;KACF,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC;AACxB,CAAC;AAEM,KAAK,UAAU,4CAA4C,CAChE,SAAqB,EACrB,MAAc,EACd,MAAc,EACd,WAAmB,EACnB,WAAmB,EACnB,IAAY,EACZ,KAAa,EACb,QAAkB,CAAC,wBAAY,CAAC,EAChC,WAAW,GAAG,KAAK;IAEnB,MAAM,UAAU,GAAG;;;UAGX,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB,WAAW,IAAI;;;;iBAI/C,KAAK;;;;;;;;;;;;;;MAchB,CAAC;IAEL,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAMnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE;YACT,IAAI;YACJ,KAAK;YACL,MAAM,EAAE,4BAAgB;SACzB;KACF,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,4CAA4C,CAAC,CAAC;YACnE,OAAO,4CAA4C,CACjD,SAAS,EACT,MAAM,EACN,MAAM,EACN,WAAW,EACX,WAAW,EACX,IAAI,EACJ,KAAK,EACL,KAAK,EACL,IAAI,CACL,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC;AACxB,CAAC;AAEM,KAAK,UAAU,sBAAsB,CAC1C,SAAqB,EACrB,WAAmB,EACnB,OAAiB,EACjB,QAAkB,CAAC,wBAAY,CAAC;IAEhC,MAAM,UAAU,GAAG;;;;;;;;;;;;;;MAcf,CAAC;IAEL,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,aAAa,CAMnD,WAAW,EACX;QACE,KAAK,EAAE,UAAU;QACjB,SAAS,EAAE;YACT,KAAK;YACL,MAAM,EAAE,4BAAgB;YACxB,KAAK,EAAE,OAAO;SACf;KACF,EACD,EAAE,OAAO,EAAE,4BAAgB,EAAE,CAC9B,CAAC;IAEF,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,GAAG,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;AAChC,CAAC"}

package/build/dex/uniswap-v4/types.d.ts

@@ -1,4 +1,5 @@
 import { Address, NumberAsString } from '../../types';
+import { HookConstructor } from './hooks/types';
 export type ModifyLiquidityParams = {
     liquidityDelta: bigint;
     tickLower: bigint;
@@ -95,6 +96,7 @@ export type DexParams = {
     stateView: string;
     skipPoolsWithUnconventionalFees?: boolean;
     stateMulticall: string;
+    supportedHooks?: HookConstructor[];
 };
 export type PoolKey = {
     currency0: Address;

package/build/dex/uniswap-v4/uniswap-v4-pool-manager.d.ts

@@ -8,6 +8,7 @@ import { LogDescription } from '@ethersproject/abi/lib.esm';
 import { FactoryState } from '../uniswap-v3/types';
 import { UniswapV4Pool } from './uniswap-v4-pool';
 import { PoolState } from './types';
+import { IBaseHook } from './hooks/types';
 export declare class UniswapV4PoolManager extends StatefulEventSubscriber<PoolManagerState> {
     readonly dexHelper: IDexHelper;
     private readonly network;
@@ -24,7 +25,9 @@ export declare class UniswapV4PoolManager extends StatefulEventSubscriber<PoolMa
     stateMulticallIface: Interface;
     private wethAddress;
     private poolsCacheKey;
-    constructor(dexHelper: IDexHelper, parentName: string, network: number, config: DexParams, logger: Logger, mapKey?: string);
+    private supportedHookAddresses;
+    private hookInstancesByAddress;
+    constructor(dexHelper: IDexHelper, parentName: string, network: number, config: DexParams, supportedHooks: IBaseHook[], logger: Logger, mapKey?: string);
     initialize(blockNumber: number, options?: InitializeStateOptions<PoolManagerState>): Promise<void>;
     generateState(): FactoryState;
     protected processLog(_: DeepReadonly<FactoryState>, log: Readonly<Log>): Promise<FactoryState>;
@@ -33,6 +36,9 @@ export declare class UniswapV4PoolManager extends StatefulEventSubscriber<PoolMa
     private queryAllAvailablePools;
     handleInitializeEvent(event: LogDescription, log: Log): Promise<PoolManagerState>;
     private isPoolWithUnconventionalFees;
+    private registerHookPools;
+    private getHook;
     private getBitmapRange;
     generateMultiplePoolStates(pools: SubgraphConnectorPool[], blockNumber: number): Promise<(PoolState | null)[]>;
+    private isHookSupported;
 }

package/build/dex/uniswap-v4/uniswap-v4-pool-manager.js

@@ -31,7 +31,9 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
     stateMulticallIface;
     wethAddress;
     poolsCacheKey = 'pools_cache';
-    constructor(dexHelper, parentName, network, config, logger, mapKey = '') {
+    supportedHookAddresses;
+    hookInstancesByAddress;
+    constructor(dexHelper, parentName, network, config, supportedHooks, logger, mapKey = '') {
         super(parentName, `${parentName} PoolManager`, dexHelper, logger, false, mapKey);
         this.dexHelper = dexHelper;
         this.network = network;
@@ -44,6 +46,11 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
         this.wethAddress =
             this.dexHelper.config.data.wrappedNativeTokenAddress.toLowerCase();
         this.logDecoder = (log) => this.poolManagerIface.parseLog(log);
+        this.hookInstancesByAddress = Object.fromEntries(supportedHooks.map(hook => [hook.address.toLowerCase(), hook]));
+        this.supportedHookAddresses = [
+            constants_1.NULL_ADDRESS, // always include pools without hooks
+            ...Object.keys(this.hookInstancesByAddress),
+        ];
         // Add handlers
         this.handlers['Initialize'] = this.handleInitializeEvent.bind(this);
     }
@@ -74,7 +81,14 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
             this.eventPools[_poolId] = null;
             return null;
         }
-        eventPool = new uniswap_v4_pool_1.UniswapV4Pool(this.dexHelper, this.parentName, this.network, this.config, this.logger, this.mapKey, _poolId, subgraphPool.token0.address.toLowerCase(), subgraphPool.token1.address.toLowerCase(), subgraphPool.fee, subgraphPool.hooks, subgraphPool.tickSpacing);
+        const poolKey = {
+            currency0: subgraphPool.token0.address.toLowerCase(),
+            currency1: subgraphPool.token1.address.toLowerCase(),
+            fee: subgraphPool.fee,
+            tickSpacing: parseInt(subgraphPool.tickSpacing),
+            hooks: subgraphPool.hooks,
+        };
+        eventPool = new uniswap_v4_pool_1.UniswapV4Pool(this.dexHelper, this.parentName, this.network, this.config, this.logger, this.mapKey, _poolId, poolKey.currency0, poolKey.currency1, poolKey.fee, poolKey.hooks, poolKey.tickSpacing.toString(), this.getHook(_poolId, poolKey));
         await eventPool.initialize(blockNumber);
         this.eventPools[_poolId] = eventPool;
         return this.eventPools[_poolId];
@@ -118,7 +132,7 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
     async queryAllAvailablePools(blockNumber) {
         const staticPoolsList = config_1.UniswapV4PoolsList[this.network];
         if (staticPoolsList) {
-            return staticPoolsList;
+            return staticPoolsList.filter(pool => this.isHookSupported(pool.hooks.toLowerCase()));
         }
         const cachedPoolsRaw = await this.dexHelper.cache.getAndCacheLocally(this.parentName, this.network, this.poolsCacheKey, constants_2.POOL_CACHE_REFRESH_INTERVAL);
         let cachedPools = [];
@@ -127,6 +141,7 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
             const poolsTTL = await this.dexHelper.cache.ttl(this.parentName, this.network, this.poolsCacheKey);
             if (cachedPools.length &&
                 poolsTTL > constants_2.POOL_CACHE_STORE_INTERVAL - constants_2.POOL_CACHE_REFRESH_INTERVAL) {
+                this.registerHookPools(cachedPools);
                 return cachedPools;
             }
             this.logger.info(`Pools cache TTL is ${poolsTTL}, refreshing`);
@@ -134,13 +149,14 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
         let pools = [];
         try {
             const defaultPerPageLimit = 1000;
+            const hooksToQuery = this.supportedHookAddresses;
             let curPage = 0;
-            let currentSubgraphPools = await (0, subgraph_1.queryOnePageForAllAvailablePoolsFromSubgraph)(this.dexHelper, this.logger, this.parentName, this.config.subgraphURL, blockNumber, curPage * defaultPerPageLimit, defaultPerPageLimit);
+            let currentSubgraphPools = await (0, subgraph_1.queryOnePageForAllAvailablePoolsFromSubgraph)(this.dexHelper, this.logger, this.parentName, this.config.subgraphURL, blockNumber, curPage * defaultPerPageLimit, defaultPerPageLimit, hooksToQuery);
             pools = pools.concat(currentSubgraphPools);
             while (currentSubgraphPools.length === defaultPerPageLimit) {
                 curPage++;
                 currentSubgraphPools =
-                    await (0, subgraph_1.queryOnePageForAllAvailablePoolsFromSubgraph)(this.dexHelper, this.logger, this.parentName, this.config.subgraphURL, blockNumber, curPage * defaultPerPageLimit, defaultPerPageLimit);
+                    await (0, subgraph_1.queryOnePageForAllAvailablePoolsFromSubgraph)(this.dexHelper, this.logger, this.parentName, this.config.subgraphURL, blockNumber, curPage * defaultPerPageLimit, defaultPerPageLimit, hooksToQuery);
                 pools = pools.concat(currentSubgraphPools);
             }
             if (this.config.skipPoolsWithUnconventionalFees) {
@@ -155,6 +171,7 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
         // always refresh pools in cache, even when subgraph queries failed
         // so next time we can use previously cached pools
         this.dexHelper.cache.setexAndCacheLocally(this.parentName, this.network, this.poolsCacheKey, constants_2.POOL_CACHE_STORE_INTERVAL, JSON.stringify(pools));
+        this.registerHookPools(pools);
         return pools;
     }
     async handleInitializeEvent(event, log) {
@@ -164,9 +181,7 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
         const fee = event.args.fee;
         const tickSpacing = parseInt(event.args.tickSpacing);
         const hooks = event.args.hooks;
-        const sqrtPriceX96 = BigInt(event.args.sqrtPriceX96);
-        const tick = parseInt(event.args.tick);
-        if (hooks !== constants_1.NULL_ADDRESS) {
+        if (!this.isHookSupported(hooks.toLowerCase())) {
             this.logger.warn(`Pool ${id} has hooks ${hooks}, which is not supported yet. Skipping.`);
             return {};
         }
@@ -193,7 +208,14 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
             },
             tickSpacing: tickSpacing.toString(),
         });
-        const eventPool = new uniswap_v4_pool_1.UniswapV4Pool(this.dexHelper, this.parentName, this.network, this.config, this.logger, this.mapKey, id, currency0.toLowerCase(), currency1.toLowerCase(), fee, hooks, tickSpacing.toString());
+        const poolKey = {
+            currency0: currency0.toLowerCase(),
+            currency1: currency1.toLowerCase(),
+            fee: fee.toString(),
+            tickSpacing,
+            hooks,
+        };
+        const eventPool = new uniswap_v4_pool_1.UniswapV4Pool(this.dexHelper, this.parentName, this.network, this.config, this.logger, this.mapKey, id, poolKey.currency0, poolKey.currency1, poolKey.fee, poolKey.hooks, poolKey.tickSpacing.toString(), this.getHook(id, poolKey));
         await eventPool.initialize(log.blockNumber);
         this.eventPools[id] = eventPool;
         return {};
@@ -201,6 +223,26 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
     isPoolWithUnconventionalFees(fee) {
         return +fee % 100 !== 0;
     }
+    registerHookPools(pools) {
+        for (const pool of pools) {
+            const poolKey = {
+                currency0: pool.token0.address.toLowerCase(),
+                currency1: pool.token1.address.toLowerCase(),
+                fee: pool.fee,
+                tickSpacing: parseInt(pool.tickSpacing),
+                hooks: pool.hooks,
+            };
+            this.getHook(pool.id, poolKey);
+        }
+    }
+    getHook(poolId, poolKey) {
+        const hook = this.hookInstancesByAddress[poolKey.hooks.toLowerCase()];
+        if (!hook) {
+            return undefined;
+        }
+        hook.registerPool(poolId, poolKey);
+        return hook;
+    }
     getBitmapRange() {
         const networkId = this.dexHelper.config.data.network;
         const tickBitMapToUse = constants_3.TICK_BITMAP_TO_USE_BY_CHAIN[networkId] ?? constants_3.TICK_BITMAP_TO_USE;
@@ -282,6 +324,9 @@ class UniswapV4PoolManager extends stateful_event_subscriber_1.StatefulEventSubs
         });
         return poolStates;
     }
+    isHookSupported(hook) {
+        return this.supportedHookAddresses.includes(hook.toLowerCase());
+    }
 }
 exports.UniswapV4PoolManager = UniswapV4PoolManager;
 //# sourceMappingURL=uniswap-v4-pool-manager.js.map