feat(api): support filtering items by item class (#61)

wowserhq · Dec 3, 2023 · d456e2a · d456e2a
1 parent 4d246e0
commit d456e2a
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 3 deletions.
diff --git a/packages/spelunker-api/src/lib/core/Query.mjs b/packages/spelunker-api/src/lib/core/Query.mjs
@@ -20,6 +20,13 @@ class Query {
     return this;
   }
 
+  filter(filters) {
+    if (filters) {
+      this.entity.filter(this, filters);
+    }
+    return this;
+  }
+
   slice() {
     notImplemented(this, 'slice');
   }

diff --git a/packages/spelunker-api/src/lib/entities/Item.mjs b/packages/spelunker-api/src/lib/entities/Item.mjs
@@ -27,6 +27,12 @@ class Item extends DatabaseEntity {
     query.where('name', 'LIKE', `%${searchQuery}%`);
   }
 
+  static filter(query, filters) {
+    if (filters?.itemClassIds?.length > 0) {
+      query.andWhere('class', 'IN', filters.itemClassIds);
+    }
+  }
+
   get id() {
     return this.data.entry;
   }

diff --git a/packages/spelunker-api/src/lib/graph/root.mjs b/packages/spelunker-api/src/lib/graph/root.mjs
@@ -29,7 +29,7 @@ export default {
   factions: ({ searchQuery }) => Faction.query.search(searchQuery),
   faction: ({ id }) => Faction.find(id),
 
-  items: ({ searchQuery }) => Item.query.search(searchQuery),
+  items: ({ searchQuery, ...filters }) => Item.query.search(searchQuery).filter(filters),
   item: ({ id }) => Item.find(id),
 
   itemSets: ({ searchQuery }) => ItemSet.query.search(searchQuery),

diff --git a/packages/spelunker-api/src/lib/graph/schema/QueryType.mjs b/packages/spelunker-api/src/lib/graph/schema/QueryType.mjs
@@ -1,5 +1,6 @@
 import {
   GraphQLInt,
+  GraphQLList,
   GraphQLNonNull,
   GraphQLObjectType,
   GraphQLString,
@@ -29,10 +30,11 @@ const finderFor = (type, idType = GraphQLInt) => ({
   },
 });
 
-const searchableCollectionFor = (type) => (
+const searchableCollectionFor = (type, filters = {}) => (
   CollectionType.definitionFor(type, {
     args: {
       searchQuery: { type: GraphQLString },
+      ...filters,
     },
   })
 );
@@ -55,7 +57,9 @@ export default new GraphQLObjectType({
     factions: searchableCollectionFor(FactionType),
     faction: finderFor(FactionType),
 
-    items: searchableCollectionFor(ItemType),
+    items: searchableCollectionFor(ItemType, {
+      itemClassIds: { type: new GraphQLList(GraphQLInt) },
+    }),
     item: finderFor(ItemType),
 
     itemSets: searchableCollectionFor(ItemSetType),