Split create address filter into two

poap-xyz · Apr 13, 2024 · 48e7e0c · 48e7e0c
1 parent a8c793b
commit 48e7e0c
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 26 deletions.
diff --git a/packages/poaps/src/PoapsClient.ts b/packages/poaps/src/PoapsClient.ts
@@ -22,6 +22,7 @@ import {
   createBetweenFilter,
   createEqFilter,
   createInFilter,
+  createNotNullAddressFilter,
   createOrderBy,
   nextCursor,
   PaginatedResult,
@@ -73,9 +74,9 @@ export class PoapsClient {
       offset,
       orderBy: createOrderBy<PoapsSortFields>(sortField, sortDir),
       where: {
-        ...createAddressFilter(
+        ...createAddressFilter('collector_address', collectorAddress),
+        ...createNotNullAddressFilter(
           'collector_address',
-          collectorAddress,
           filterZeroAddress,
           filterDeadAddress,
         ),

diff --git a/packages/utils/src/queries/filter.ts b/packages/utils/src/queries/filter.ts
@@ -42,40 +42,43 @@ export function createBoolFilter(
     : {};
 }
 
-// eslint-disable-next-line complexity
 export function createAddressFilter(
   key: string,
   value?: string,
-  filterZeroAddress?: boolean,
-  filterDeadAddress?: boolean,
-): FieldFilter<
-  Partial<EqFilter<string>> &
-    Partial<NeqFilter<string>> &
-    Partial<NinFilter<string>>
-> {
-  return filterZeroAddress || filterDeadAddress || value
+): FieldFilter<Partial<EqFilter<string>>> {
+  return value
     ? {
         [key]: {
-          ...(filterZeroAddress && filterDeadAddress
-            ? {
-                _nin: [
-                  '0x0000000000000000000000000000000000000000',
-                  '0x000000000000000000000000000000000000dead',
-                ],
-              }
-            : {}),
-          ...(filterZeroAddress && !filterDeadAddress
-            ? { _neq: '0x0000000000000000000000000000000000000000' }
-            : {}),
-          ...(!filterZeroAddress && filterDeadAddress
-            ? { _neq: '0x000000000000000000000000000000000000dead' }
-            : {}),
-          ...(value ? { _eq: value.toLowerCase() } : {}),
+          _eq: value.toLowerCase(),
         },
       }
     : {};
 }
 
+export function createNotNullAddressFilter(
+  key: string,
+  filterZeroAddress = true,
+  filterDeadAddress = true,
+): FieldFilter<Partial<NeqFilter<string>> & Partial<NinFilter<string>>> {
+  if (filterZeroAddress && filterDeadAddress) {
+    return {
+      [key]: {
+        _nin: [
+          '0x0000000000000000000000000000000000000000',
+          '0x000000000000000000000000000000000000dead',
+        ],
+      },
+    };
+  }
+  if (filterZeroAddress) {
+    return { [key]: { _neq: '0x0000000000000000000000000000000000000000' } };
+  }
+  if (filterDeadAddress) {
+    return { [key]: { _neq: '0x000000000000000000000000000000000000dead' } };
+  }
+  return {};
+}
+
 export function createInFilter(
   key: string,
   values?: Array<Value>,