use super::{ hotbar::{self, Slot as HotbarSlot}, img_ids, item_imgs::{ItemImgs, ItemKey}, }; use crate::ui::slot::{self, SlotKey, SumSlot}; use common::comp::{ item::{ tool::{AbilityMap, ToolKind}, ItemKind, }, slot::InvSlotId, Energy, Inventory, }; use conrod_core::{image, Color}; pub use common::comp::slot::{ArmorSlot, EquipSlot}; #[derive(Clone, Copy, Debug, PartialEq)] pub enum SlotKind { Inventory(InventorySlot), Equip(EquipSlot), Hotbar(HotbarSlot), Trade(TradeSlot), /* Spellbook(SpellbookSlot), TODO */ } pub type SlotManager = slot::SlotManager; #[derive(Clone, Copy, Debug, PartialEq)] pub struct InventorySlot(pub InvSlotId); impl SlotKey for InventorySlot { type ImageKey = ItemKey; fn image_key(&self, source: &Inventory) -> Option<(Self::ImageKey, Option)> { source.get(self.0).map(|i| (i.into(), None)) } fn amount(&self, source: &Inventory) -> Option { source .get(self.0) .map(|item| item.amount()) .filter(|amount| *amount > 1) } fn image_id(key: &Self::ImageKey, source: &ItemImgs) -> image::Id { source.img_id_or_not_found_img(key.clone()) } } impl SlotKey for EquipSlot { type ImageKey = ItemKey; fn image_key(&self, source: &Inventory) -> Option<(Self::ImageKey, Option)> { let item = source.equipped(*self); item.map(|i| (i.into(), None)) } fn amount(&self, _: &Inventory) -> Option { None } fn image_id(key: &Self::ImageKey, source: &ItemImgs) -> image::Id { source.img_id_or_not_found_img(key.clone()) } } #[derive(Clone, Copy, Debug, PartialEq)] pub struct TradeSlot { pub index: usize, pub quantity: u32, pub invslot: Option, } impl SlotKey for TradeSlot { type ImageKey = ItemKey; fn image_key(&self, source: &Inventory) -> Option<(Self::ImageKey, Option)> { self.invslot .and_then(|inv_id| InventorySlot(inv_id).image_key(source)) } fn amount(&self, source: &Inventory) -> Option { self.invslot .and_then(|inv_id| InventorySlot(inv_id).amount(source)) .map(|x| x.min(self.quantity)) } fn image_id(key: &Self::ImageKey, source: &ItemImgs) -> image::Id { source.img_id_or_not_found_img(key.clone()) } } #[derive(Clone, PartialEq)] pub enum HotbarImage { Item(ItemKey), FireAoe, SnakeArrow, SwordWhirlwind, HammerLeap, AxeLeapSlash, BowJumpBurst, } type HotbarSource<'a> = (&'a hotbar::State, &'a Inventory, &'a Energy, &'a AbilityMap); type HotbarImageSource<'a> = (&'a ItemImgs, &'a img_ids::Imgs); impl<'a> SlotKey, HotbarImageSource<'a>> for HotbarSlot { type ImageKey = HotbarImage; fn image_key( &self, (hotbar, inventory, energy, ability_map): &HotbarSource<'a>, ) -> Option<(Self::ImageKey, Option)> { hotbar.get(*self).and_then(|contents| match contents { hotbar::SlotContents::Inventory(idx) => inventory .get(idx) .map(|item| HotbarImage::Item(item.into())) .map(|i| (i, None)), hotbar::SlotContents::Ability3 => { let tool = match inventory.equipped(EquipSlot::Mainhand).map(|i| i.kind()) { Some(ItemKind::Tool(tool)) => Some(tool), _ => None, }; tool.and_then(|tool| { match tool.kind { ToolKind::Staff => Some(HotbarImage::FireAoe), ToolKind::Hammer => Some(HotbarImage::HammerLeap), ToolKind::Axe => Some(HotbarImage::AxeLeapSlash), ToolKind::Bow => Some(HotbarImage::BowJumpBurst), ToolKind::Debug => Some(HotbarImage::SnakeArrow), ToolKind::Sword => Some(HotbarImage::SwordWhirlwind), _ => None, } .map(|i| { ( i, if let Some(skill) = tool.get_abilities(ability_map).skills.get(0) { if energy.current() >= skill.get_energy_cost() { Some(Color::Rgba(1.0, 1.0, 1.0, 1.0)) } else { Some(Color::Rgba(0.3, 0.3, 0.3, 0.8)) } } else { Some(Color::Rgba(1.0, 1.0, 1.0, 1.0)) }, ) }) }) }, }) } fn amount(&self, (hotbar, inventory, _, _): &HotbarSource<'a>) -> Option { hotbar .get(*self) .and_then(|content| match content { hotbar::SlotContents::Inventory(idx) => inventory.get(idx), hotbar::SlotContents::Ability3 => None, }) .map(|item| item.amount()) .filter(|amount| *amount > 1) } fn image_id(key: &Self::ImageKey, (item_imgs, imgs): &HotbarImageSource<'a>) -> image::Id { match key { HotbarImage::Item(key) => item_imgs.img_id_or_not_found_img(key.clone()), HotbarImage::SnakeArrow => imgs.snake_arrow_0, HotbarImage::FireAoe => imgs.fire_aoe, HotbarImage::SwordWhirlwind => imgs.sword_whirlwind, HotbarImage::HammerLeap => imgs.hammerleap, HotbarImage::AxeLeapSlash => imgs.skill_axe_leap_slash, HotbarImage::BowJumpBurst => imgs.skill_bow_jump_burst, } } } impl From for SlotKind { fn from(inventory: InventorySlot) -> Self { Self::Inventory(inventory) } } impl From for SlotKind { fn from(equip: EquipSlot) -> Self { Self::Equip(equip) } } impl From for SlotKind { fn from(hotbar: HotbarSlot) -> Self { Self::Hotbar(hotbar) } } impl From for SlotKind { fn from(trade: TradeSlot) -> Self { Self::Trade(trade) } } impl SumSlot for SlotKind {}