Pumpkin/pumpkin-inventory/src/lib.rs

use crate::container_click::MouseClick;
use crate::player::PlayerInventory;
use pumpkin_data::screen::WindowType;
use pumpkin_world::item::ItemStack;

pub mod container_click;
mod crafting;
pub mod drag_handler;
mod error;
mod open_container;
pub mod player;
pub mod window_property;

pub use error::InventoryError;
pub use open_container::*;

pub struct ContainerStruct<const SLOTS: usize>([Option<ItemStack>; SLOTS]);

// `Container` needs to be `Sync + Send` to be able to be in the async server.
pub trait Container: Sync + Send {
    fn window_type(&self) -> &'static WindowType;

    fn window_name(&self) -> &'static str;

    fn handle_item_change(
        &mut self,
        carried_item: &mut Option<ItemStack>,
        slot: usize,
        mouse_click: MouseClick,
        taking_crafted: bool,
    ) -> Result<(), InventoryError> {
        let all_slots = self.all_slots();
        if slot > all_slots.len() {
            Err(InventoryError::InvalidSlot)?
        }
        if taking_crafted {
            match (all_slots[slot].as_mut(), carried_item.as_mut()) {
                (Some(s1), Some(s2)) => {
                    if s1.item.id == s2.item.id {
                        handle_item_change(all_slots[slot], carried_item, mouse_click);
                    }
                }
                (Some(_), None) => handle_item_change(all_slots[slot], carried_item, mouse_click),
                (None, None) | (None, Some(_)) => (),
            }
            return Ok(());
        }
        handle_item_change(carried_item, all_slots[slot], mouse_click);

        Ok(())
    }

    fn all_slots(&mut self) -> Box<[&mut Option<ItemStack>]>;

    fn all_slots_ref(&self) -> Box<[Option<&ItemStack>]>;

    fn clear_all_slots(&mut self) {
        let all_slots = self.all_slots();
        for stack in all_slots {
            *stack = None;
        }
    }

    fn all_combinable_slots(&self) -> Box<[Option<&ItemStack>]> {
        self.all_slots_ref()
    }

    fn all_combinable_slots_mut(&mut self) -> Box<[&mut Option<ItemStack>]> {
        self.all_slots()
    }

    fn internal_pumpkin_id(&self) -> u64 {
        0
    }

    fn craft(&mut self) -> bool {
        false
    }

    fn crafting_output_slot(&self) -> Option<usize> {
        None
    }

    fn slot_in_crafting_input_slots(&self, _slot: &usize) -> bool {
        false
    }

    fn crafted_item_slot(&self) -> Option<&ItemStack> {
        *self.all_slots_ref().get(self.crafting_output_slot()?)?
    }

    fn recipe_used(&mut self) {}
}

pub struct EmptyContainer;

impl Container for EmptyContainer {
    fn window_type(&self) -> &'static WindowType {
        unreachable!(
            "You should never be able to get here because this type is always wrapped in an `Option`."
        );
    }

    fn window_name(&self) -> &'static str {
        unreachable!(
            "You should never be able to get here because this type is always wrapped in an `Option`."
        );
    }

    fn all_slots(&mut self) -> Box<[&mut Option<ItemStack>]> {
        unreachable!(
            "You should never be able to get here because this type is always wrapped in an `Option`."
        );
    }

    fn all_slots_ref(&self) -> Box<[Option<&ItemStack>]> {
        unreachable!(
            "You should never be able to get here because this type is always wrapped in an `Option`."
        );
    }
}

pub fn handle_item_take(
    carried_item: &mut Option<ItemStack>,
    item_slot: &mut Option<ItemStack>,
    mouse_click: MouseClick,
) {
    let Some(item_stack) = item_slot else {
        return;
    };
    let mut new_item = item_stack.clone();

    match mouse_click {
        MouseClick::Left => {
            *item_slot = None;
        }
        MouseClick::Right => {
            let half = item_stack.item_count / 2;
            new_item.item_count = half;
            item_stack.item_count -= half;
            if item_stack.item_count == 0 {
                *item_slot = None;
            }
        }
    }
    *carried_item = Some(new_item);
}
pub fn handle_item_change(
    carried_slot: &mut Option<ItemStack>,
    current_slot: &mut Option<ItemStack>,
    mouse_click: MouseClick,
) {
    match (current_slot.as_mut(), carried_slot.as_mut()) {
        // Swap or combine current and carried
        (Some(current), Some(carried)) => {
            if current.item.id == carried.item.id {
                combine_stacks(carried_slot, current, mouse_click);
            } else if mouse_click == MouseClick::Left {
                std::mem::swap(carried_slot, current_slot);
            }
        }
        // Put held stack into empty slot
        (None, Some(carried_item_stack)) => match mouse_click {
            MouseClick::Left => {
                std::mem::swap(carried_slot, current_slot);
            }
            MouseClick::Right => {
                let new_stack = ItemStack {
                    item: carried_item_stack.item.clone(),
                    item_count: 1,
                };
                *current_slot = Some(new_stack);
                carried_item_stack.item_count -= 1;
                if carried_item_stack.item_count == 0 {
                    *carried_slot = None;
                }
            }
        },
        // Take stack into carried
        (Some(_current), None) => handle_item_take(carried_slot, current_slot, mouse_click),
        (None, None) => (),
    }
}

pub fn combine_stacks(
    carried_slot: &mut Option<ItemStack>,
    slot: &mut ItemStack,
    mouse_click: MouseClick,
) {
    let Some(carried_item) = carried_slot else {
        return;
    };

    debug_assert!(carried_item.item.id == slot.item.id);
    let max_size = carried_item.item.components.max_stack_size;

    let carried_change = match mouse_click {
        MouseClick::Left => carried_item.item_count,
        MouseClick::Right => 1,
    };

    // TODO: Check for item stack max size here
    if slot.item_count + carried_change <= max_size {
        slot.item_count += carried_change;
        carried_item.item_count -= carried_change;
        if carried_item.item_count == 0 {
            *carried_slot = None;
        }
    } else {
        let left_over = slot.item_count + carried_change - max_size;
        slot.item_count = max_size;
        carried_item.item_count = left_over;
    }
}

pub struct OptionallyCombinedContainer<'a, 'b> {
    container: Option<&'a mut Box<dyn Container>>,
    inventory: &'b mut PlayerInventory,
}
impl<'a, 'b> OptionallyCombinedContainer<'a, 'b> {
    pub fn new(
        player_inventory: &'b mut PlayerInventory,
        container: Option<&'a mut Box<dyn Container>>,
    ) -> Self {
        Self {
            inventory: player_inventory,
            container,
        }
    }
    /// Returns `None` if the slot is in the player's inventory. Returns `Some(Option<&ItemStack>)` if it's inside of the container.
    pub fn get_slot_excluding_inventory(&self, slot: usize) -> Option<Option<&ItemStack>> {
        self.container.as_ref()?.all_slots_ref().get(slot).copied()
    }
}

impl<'a> Container for OptionallyCombinedContainer<'a, 'a> {
    fn window_type(&self) -> &'static WindowType {
        if let Some(container) = &self.container {
            container.window_type()
        } else {
            &WindowType::Generic9x1
        }
    }

    fn window_name(&self) -> &'static str {
        self.container
            .as_ref()
            .map(|container| container.window_name())
            .unwrap_or(self.inventory.window_name())
    }

    fn all_slots(&mut self) -> Box<[&mut Option<ItemStack>]> {
        match &mut self.container {
            Some(container) => {
                let mut slots = container.all_slots().into_vec();
                slots.extend(self.inventory.all_combinable_slots_mut());
                slots.into_boxed_slice()
            }
            None => self.inventory.all_slots(),
        }
    }

    fn all_slots_ref(&self) -> Box<[Option<&ItemStack>]> {
        match &self.container {
            Some(container) => {
                let mut slots = container.all_slots_ref().into_vec();
                slots.extend(self.inventory.all_combinable_slots());
                slots.into_boxed_slice()
            }
            None => self.inventory.all_slots_ref(),
        }
    }

    fn craft(&mut self) -> bool {
        match &mut self.container {
            Some(container) => container.craft(),
            None => self.inventory.craft(),
        }
    }

    fn crafting_output_slot(&self) -> Option<usize> {
        match &self.container {
            Some(container) => container.crafting_output_slot(),
            None => self.inventory.crafting_output_slot(),
        }
    }

    fn slot_in_crafting_input_slots(&self, slot: &usize) -> bool {
        match &self.container {
            Some(container) => {
                // We don't have to worry about length due to inventory crafting slots being inaccessible
                // while inside container interfaces.
                container.slot_in_crafting_input_slots(slot)
            }
            None => self.inventory.slot_in_crafting_input_slots(slot),
        }
    }

    fn recipe_used(&mut self) {
        match &mut self.container {
            Some(container) => container.recipe_used(),
            None => self.inventory.recipe_used(),
        }
    }
}