Merge pull request #115 from ThalaLabs/pool-filter

Add pool filter in ThalaswapRouter Option argument. So that user can filter out some pools for the router
ThalaLabs · Aug 12, 2024 · 7961f25 · 7961f25
2 parents 9fe467c + 9e6b60c
commit 7961f25
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/.changeset/long-garlics-dance.md b/.changeset/long-garlics-dance.md
@@ -0,0 +1,5 @@
+---
+"@thalalabs/router-sdk": minor
+---
+
+Add pool filter in ThalaswapRouter Option argument. So that user can filter out some pools for the router.
diff --git a/packages/thalaswap-router/src/ThalaswapRouter.ts b/packages/thalaswap-router/src/ThalaswapRouter.ts
@@ -63,6 +63,7 @@ const scaleUp = (amount: number, decimals: number): number => {
 
 type Options = {
   maxAllowedSwapPercentage?: number;
+  poolFilter?: (pool: Pool) => boolean;
 };
 
 class ThalaswapRouter {
@@ -102,6 +103,11 @@ class ThalaswapRouter {
     const graph: Graph = {};
 
     for (const pool of pools) {
+      // Apply pool filter if provided
+      if (this.options.poolFilter && !this.options.poolFilter(pool)) {
+        continue;
+      }
+
       // Convert pool data to LiquidityPool type
       const assets = ["asset0", "asset1", "asset2", "asset3"]
         .filter((a) => pool[a as AssetIndex])