storage

nodejs/controller/storage/organizer.js

@@ -0,0 +1,103 @@
+'use strict';
+
+const Vec3 = require('vec3');
+const conf = require('../../conf');
+
+class Organizer {
+	constructor() {
+		this.categories = conf.storage?.categories || {
+			minerals: ['diamond', 'netherite_ingot', 'gold_ingot', 'iron_ingot'],
+			food: ['bread', 'cooked_porkchop', 'steak'],
+			tools: ['diamond_sword', 'diamond_pickaxe', 'netherite_pickaxe'],
+			armor: ['diamond_chestplate', 'netherite_helmet'],
+			blocks: ['stone', 'dirt', 'cobblestone'],
+			redstone: ['redstone', 'repeater', 'piston'],
+			misc: []
+		};
+	}
+
+	categorizeItem(itemName) {
+		// Fast path: check each category
+		for (const [category, items] of Object.entries(this.categories)) {
+			if (items.includes(itemName)) {
+				return category;
+			}
+		}
+		return 'misc';
+	}
+
+	async findShulkerForItem(database, itemId, categoryName) {
+		// Find shulker with matching item that has space
+		const shulker = await database.findShulkerForItem(itemId, categoryName);
+		return shulker;
+	}
+
+	async findEmptyShulkerSlot(database, categoryName) {
+		// Find an empty shulker in the appropriate category and row (prefer row 4 for empty storage)
+		const chests = await database.getChests();
+		
+		// Filter chests by category and row 4 (top row for empty/new shulkers)
+		const categoryChests = chests.filter(c => 
+			c.category === categoryName && c.row === 4
+		).sort((a, b) => a.column - b.column); // Left to right
+
+		for (const chest of categoryChests) {
+			const shulkers = await database.getShulkersByChest(chest.id);
+			
+			// Find first shulker that's empty (slotCount = 0) or has space
+			for (const shulker of shulkers) {
+				if (!shulker.item_focus) {
+					// Empty shulker available
+					return {
+						chest_id: chest.id,
+						chestPosition: new Vec3(chest.pos_x, chest.pos_y, chest.pos_z),
+						chestSlot: shulker.slot,
+						shulker_id: shulker.id
+					};
+				}
+			}
+		}
+
+		// If no empty shulker, look for first available slot in row 4
+		// ... this would need to scan actual chest for empty slots
+		return null;
+	}
+
+	async sortItemIntoStorage(bot, database, item, categoryName) {
+		// Find existing shulker with same item and space
+		const existingShulker = await this.findShulkerForItem(database, item.id, categoryName);
+
+		if (existingShulker) {
+			// Space available, add to existing shulker
+			console.log(`Organizer: Found shulker ${existingShulker.id} for ${item.name}`);
+			return existingShulker;
+		} else {
+			// Need new shulker
+			console.log(`Organizer: Creating new shulker for ${item.name} (${categoryName})`);
+			
+			const shulkerSlot = await this.findEmptyShulkerSlot(database, categoryName);
+			if (!shulkerSlot) {
+				console.log(`Organizer: No available shulker slot for ${item.name}`);
+				return null;
+			}
+
+			// Create/prepare new shulker
+			await database.upsertShulker(
+				shulkerSlot.chest_id,
+				shulkerSlot.chestSlot,
+				'shulker_box',
+				categoryName,
+				item.name // item_focus
+			);
+
+			console.log(`Organizer: Created shulker at chest ${shulkerSlot.chest_id}, slot ${shulkerSlot.chestSlot}`);
+			return {
+				chest_id: shulkerSlot.chest_id,
+				slot: shulkerSlot.chestSlot,
+				new: true
+			};
+		}
+	}
+}
+
+module.exports = Organizer;