Merge pull request #354 from the-hideout/handbook-kv

split info into handbook kv
the-hideout · Jan 20, 2025 · 6184479 · 6184479
2 parents 3035215 + 36687fc
commit 6184479
Show file tree

Hide file tree

Showing 8 changed files with 166 additions and 137 deletions.
diff --git a/datasources/handbook.mjs b/datasources/handbook.mjs
@@ -0,0 +1,100 @@
+import WorkerKV from '../utils/worker-kv.mjs';
+
+class HandbookAPI extends WorkerKV {
+    constructor(dataSource) {
+        super('handbook_data', dataSource);
+        //this.gameModes.push('pve');
+    }
+
+    async getCategory(context, info, id) {
+        const { cache } = await this.getCache(context, info);
+        return cache.ItemCategory[id] || cache.HandbookCategory[id];
+    }
+
+    async getTopCategory(context, info, id) {
+        const cat = await this.getCategory(context, info, id);
+        if (cat && cat.parent_id) return this.getTopCategory(context, info, cat.parent_id);
+        return cat;
+    }
+
+    async getCategories(context, info) {
+        const { cache } = await this.getCache(context, info);
+        if (!cache) {
+            return Promise.reject(new Error('Item cache is empty'));
+        }
+        const categories = [];
+        for (const id in cache.ItemCategory) {
+            categories.push(cache.ItemCategory[id]);
+        }
+        return categories;
+    }
+
+    async getCategoriesEnum(context, info) {
+        const cats = await this.getCategories(context, info);
+        const map = {};
+        for (const id in cats) {
+            map[cats[id].enumName] = cats[id];
+        }
+        return map;
+    }
+
+    async getHandbookCategory(context, info, id) {
+        const { cache } = await this.getCache(context, info);
+        return cache.HandbookCategory[id];
+    }
+
+    async getHandbookCategories(context, info) {
+        const { cache } = await this.getCache(context, info);
+        if (!cache) {
+            return Promise.reject(new Error('Item cache is empty'));
+        }
+        return Object.values(cache.HandbookCategory);
+    }
+
+    async getArmorMaterials(context, info) {
+        const { cache } = await this.getCache(context, info);
+        return Object.values(cache.ArmorMaterial).sort();
+    }
+
+    async getArmorMaterial(context, info, matKey) {
+        const { cache } = await this.getCache(context, info);
+        return cache.ArmorMaterial[matKey];
+    }
+
+    async getMasterings(context, info) {
+        const { cache } = await this.getCache(context, info);
+        return cache.Mastering;
+    }
+
+    async getMastering(context, info, mastId) {
+        const { cache } = await this.getCache(context, info);
+        return cache.Mastering.find(m => m.id === mastId);
+    }
+
+    async getSkills(context, info) {
+        const { cache } = await this.getCache(context, info);
+        return cache.Skill;
+    }
+
+    async getSkill(context, info, skillId) {
+        const { cache } = await this.getCache(context, info);
+        return cache.Skill.find(s => s.id === skillId);
+    }
+
+    async getPlayerLevels(context, info) {
+        const { cache } = await this.getCache(context, info);
+        return cache.PlayerLevel;
+    }
+
+    async getAllItemProperties(context, info) {
+        const { cache } = await this.getCache(context, info);
+        return cache.ItemProperties;
+    }
+
+    async getItemProperties(context, info, itemId) {
+        const { cache } = await this.getCache(context, info);
+        return cache.ItemProperties[itemId];
+    }
+}
+
+export default HandbookAPI;
diff --git a/datasources/index.mjs b/datasources/index.mjs
@@ -1,5 +1,6 @@
 import BartersAPI from './barters.mjs';
 import CraftsAPI from './crafts.mjs';
+import HandbookAPI from './handbook.mjs';
 import HideoutAPI from './hideout.mjs';
 import HistoricalPricesAPI from './historical-prices.mjs';
 import ArchivedPricesAPI from './archived-prices.mjs';
@@ -23,6 +24,7 @@ class DataSource {
         this.worker = {
             barter: new BartersAPI(this),
             craft: new CraftsAPI(this),
+            handbook: new HandbookAPI(this),
             hideout: new HideoutAPI(this),
             historicalPrice: new HistoricalPricesAPI(this),
             archivedPrice: new ArchivedPricesAPI(this),

diff --git a/datasources/items.mjs b/datasources/items.mjs
@@ -170,7 +170,7 @@ class ItemsAPI extends WorkerKV {
         if (!items) {
             items = Object.values(cache.Item);
         }
-        const categories = (await this.getCategories(context, info)).filter(cat => names.includes(cat.enumName));
+        const categories = (await context.data.worker.handbook.getCategories(context, info)).filter(cat => names.includes(cat.enumName));
         return items.filter((item) => {
             return item.categories.some(catId => categories.some(cat => cat.id === catId));
         });
@@ -181,7 +181,7 @@ class ItemsAPI extends WorkerKV {
         if (!items) {
             items = Object.values(cache.Item);
         }
-        const categories = (await this.getHandbookCategories(context, info)).filter(cat => names.includes(cat.enumName));
+        const categories = (await context.data.worker.handbook.getHandbookCategories(context, info)).filter(cat => names.includes(cat.enumName));
         return items.filter((item) => {
             return item.handbookCategories.some(catId => categories.some(cat => cat.id === catId));
         });
@@ -216,108 +216,24 @@ class ItemsAPI extends WorkerKV {
         });
     }
 
-    async getCategory(context, info, id) {
-        const { cache } = await this.getCache(context, info);
-        return cache.ItemCategory[id] || cache.HandbookCategory[id];
-    }
-
-    async getTopCategory(context, info, id) {
-        const cat = await this.getCategory(context, info, id);
-        if (cat && cat.parent_id) return this.getTopCategory(context, info, cat.parent_id);
-        return cat;
-    }
-
-    async getCategories(context, info) {
-        const { cache } = await this.getCache(context, info);
-        if (!cache) {
-            return Promise.reject(new Error('Item cache is empty'));
-        }
-        const categories = [];
-        for (const id in cache.ItemCategory) {
-            categories.push(cache.ItemCategory[id]);
-        }
-        return categories;
-    }
-
-    async getCategoriesEnum(context, info) {
-        const cats = await this.getCategories(context, info);
-        const map = {};
-        for (const id in cats) {
-            map[cats[id].enumName] = cats[id];
-        }
-        return map;
-    }
-
-    async getHandbookCategory(context, info, id) {
-        const { cache } = await this.getCache(context, info);
-        return cache.HandbookCategory[id];
-    }
-
-    async getHandbookCategories(context, info) {
-        const { cache } = await this.getCache(context, info);
-        if (!cache) {
-            return Promise.reject(new Error('Item cache is empty'));
-        }
-        return Object.values(cache.HandbookCategory);
-    }
-
     async getFleaMarket(context, info) {
         const { cache } = await this.getCache(context, info);
         return cache.FleaMarket;
     }
 
-    async getArmorMaterials(context, info) {
-        const { cache } = await this.getCache(context, info);
-        return Object.values(cache.ArmorMaterial).sort();
-    }
-
-    async getArmorMaterial(context, info, matKey) {
-        const { cache } = await this.getCache(context, info);
-        return cache.ArmorMaterial[matKey];
-    }
-
-    async getMasterings(context, info) {
-        const { cache } = await this.getCache(context, info);
-        return cache.Mastering;
-    }
-
-    async getMastering(context, info, mastId) {
-        const { cache } = await this.getCache(context, info);
-        return cache.Mastering.find(m => m.id === mastId);
-    }
-
-    async getSkills(context, info) {
-        const { cache } = await this.getCache(context, info);
-        return cache.Skill;
-    }
-
-    async getSkill(context, info, skillId) {
-        const { cache } = await this.getCache(context, info);
-        return cache.Skill.find(s => s.id === skillId);
-    }
-
     async getAmmoList(context, info) {
         const allAmmo = await this.getItemsByBsgCategoryId(context, info, '5485a8684bdc2da71d8b4567').then(ammoItems => {
             // ignore bb
             return ammoItems.filter(item => item.id !== '6241c316234b593b5676b637');
         });
+        const itemProperties = await context.data.worker.handbook.getAllItemProperties(context, info);
         return allAmmo.map(item => {
             return {
                 ...item,
-                ...item.properties
+                ...itemProperties[item.id],
             };
         });
     }
-
-    async getPlayerLevels(context, info) {
-        const { cache } = await this.getCache(context, info);
-        return cache.PlayerLevel;
-    }
-
-    async getTypes(context, info) {
-        const { cache } = await this.getCache(context, info);
-        return cache.ItemType;
-    }
 }
 
 export default ItemsAPI;
diff --git a/resolvers/hideoutResolver.mjs b/resolvers/hideoutResolver.mjs
@@ -50,7 +50,7 @@ export default {
             return context.data.worker.hideout.getLocale(data.name, context, info);
         },
         skill(data, args, context, info) {
-            return context.data.worker.item.getSkill(context, info, data.name);
+            return context.data.worker.handbook.getSkill(context, info, data.name);
         },
     },
     HideoutModule: {