Merge pull request #519 from beethovenxfi/feature/add-user-address-to…

…-pools-query Add user address filter to pools query
beethovenxfi · Nov 15, 2023 · 37ff70c · 37ff70c
2 parents 7539083 + 1e0cecb
commit 37ff70c
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 8 deletions.
diff --git a/modules/pool/lib/pool-gql-loader.service.ts b/modules/pool/lib/pool-gql-loader.service.ts
@@ -132,6 +132,7 @@ export class PoolGqlLoaderService {
     private mapQueryArgsToPoolQuery(args: QueryPoolGetPoolsArgs): Prisma.PrismaPoolFindManyArgs {
         let orderBy: Prisma.PrismaPoolOrderByWithRelationInput = {};
         const orderDirection = args.orderDirection || undefined;
+        const userAddress = args.userAddress;
 
         switch (args.orderBy) {
             case 'totalLiquidity':
@@ -207,6 +208,35 @@ export class PoolGqlLoaderService {
             });
         }
 
+        const userArgs: Prisma.PrismaPoolWhereInput = userAddress
+            ? {
+                  OR: [
+                      {
+                          userWalletBalances: {
+                              some: {
+                                  userAddress: {
+                                      equals: userAddress,
+                                      mode: 'insensitive' as const,
+                                  },
+                                  balanceNum: { gt: 0 },
+                              },
+                          },
+                      },
+                      {
+                          userStakedBalances: {
+                              some: {
+                                  userAddress: {
+                                      equals: userAddress,
+                                      mode: 'insensitive' as const,
+                                  },
+                                  balanceNum: { gt: 0 },
+                              },
+                          },
+                      },
+                  ],
+              }
+            : {};
+
         const filterArgs: Prisma.PrismaPoolWhereInput = {
             dynamicData: {
                 totalSharesNum: {
@@ -232,11 +262,15 @@ export class PoolGqlLoaderService {
                 mode: 'insensitive',
             },
             categories: {
-                every: {
-                    category: {
-                        notIn: ['BLACK_LISTED', ...(where?.categoryNotIn || [])],
-                    },
-                },
+                ...(where?.categoryNotIn
+                    ? {
+                          every: {
+                              category: {
+                                  notIn: where.categoryNotIn,
+                              },
+                          },
+                      }
+                    : {}),
                 ...(where?.categoryIn
                     ? {
                           some: {
@@ -272,18 +306,22 @@ export class PoolGqlLoaderService {
         if (!textSearch) {
             return {
                 ...baseQuery,
-                where: filterArgs,
+                where: {
+                    ...filterArgs,
+                    ...userArgs,
+                },
             };
         }
 
         return {
             ...baseQuery,
             where: {
                 OR: [
-                    { name: textSearch, ...filterArgs },
-                    { symbol: textSearch, ...filterArgs },
+                    { name: textSearch, ...filterArgs, ...userArgs },
+                    { symbol: textSearch, ...filterArgs, ...userArgs },
                     {
                         ...filterArgs,
+                        ...userArgs,
                         allTokens: {
                             some: {
                                 OR: [

diff --git a/modules/pool/pool.gql b/modules/pool/pool.gql
@@ -7,6 +7,7 @@ extend type Query {
         orderDirection: GqlPoolOrderDirection
         where: GqlPoolFilter
         textSearch: String
+        userAddress: String
     ): [GqlPoolMinimal!]!
     poolGetPoolsCount(
         first: Int
@@ -15,6 +16,7 @@ extend type Query {
         orderDirection: GqlPoolOrderDirection
         where: GqlPoolFilter
         textSearch: String
+        userAddress: String
     ): Int!
     poolGetSwaps(first: Int, skip: Int, where: GqlPoolSwapFilter): [GqlPoolSwap!]!
     poolGetBatchSwaps(first: Int, skip: Int, where: GqlPoolSwapFilter): [GqlPoolBatchSwap!]!