doukutsu-rs/src/inventory.rs

use std::cmp::Ordering;

use crate::engine_constants::EngineConstants;
use crate::shared_game_state::SharedGameState;
use crate::weapon::{Weapon, WeaponLevel, WeaponType};
use crate::player::{Player, TargetPlayer};
use crate::weapon::bullet::BulletManager;

#[derive(Clone, Copy)]
/// (id, amount)
pub struct Item(pub u16, pub u16);

#[derive(Clone)]
pub struct Inventory {
    pub current_item: u16,
    pub current_weapon: u16,
    items: Vec<Item>,
    weapons: Vec<Weapon>,
}

#[derive(Clone, Copy)]
pub enum TakeExperienceResult {
    None,
    LevelDown,
}

impl Inventory {
    #[allow(clippy::new_without_default)]
    pub fn new() -> Inventory {
        Inventory {
            current_item: 0,
            current_weapon: 0,
            items: Vec::with_capacity(16),
            weapons: Vec::with_capacity(16),
        }
    }

    pub fn add_item(&mut self, item_id: u16) {
        if !self.has_item(item_id) {
            self.items.push(Item(item_id, 1));
        } else if let Some(item) = self.get_item(item_id) {
            item.1 += 1;
        }
    }

    pub fn add_item_amount(&mut self, item_id: u16, amount: u16) {
        if !self.has_item(item_id) {
            self.items.push(Item(item_id, amount));
        } else if let Some(item) = self.get_item(item_id) {
            item.1 += amount;
        }
    }

    pub fn consume_item(&mut self, item_id: u16) {
        if let Some(item) = self.get_item(item_id) {
            if item.1 > 1 {
                item.1 -= 1;
                return;
            }
        }

        self.remove_item(item_id);
    }

    pub fn remove_item(&mut self, item_id: u16) {
        self.items.retain(|item| item.0 != item_id);
    }

    pub fn get_item(&mut self, item_id: u16) -> Option<&mut Item> {
        self.items.iter_mut().by_ref().find(|item| item.0 == item_id)
    }

    pub fn get_item_idx(&self, idx: usize) -> Option<&Item> {
        self.items.get(idx)
    }

    pub fn has_item(&self, item_id: u16) -> bool {
        self.items.iter().any(|item| item.0 == item_id)
    }

    pub fn has_item_amount(&self, item_id: u16, operator: Ordering, amount: u16) -> bool {
        let result = self.items.iter().any(|item| item.0 == item_id && item.1.cmp(&amount) == operator);

        if (operator == Ordering::Equal && amount == 0 && !result) || (operator == Ordering::Less && !self.has_item(item_id)) {
            return true;
        }

        result
    }

    pub fn add_weapon_data(&mut self, weapon_id: WeaponType, ammo: u16, max_ammo: u16, experience: u16, level: WeaponLevel) {
        let weapon = Weapon::new(
            weapon_id,
            level,
            experience,
            ammo,
            max_ammo,
        );

        if !self.has_weapon(weapon_id) {
            self.weapons.push(weapon);
        } else {
            let w = self.get_weapon_by_type_mut(weapon_id);
            if let Some(w) = w {
                *w = weapon
            }
        }
    }

    pub fn add_weapon(&mut self, weapon_id: WeaponType, ammo: u16) {
        let w = self.get_weapon_by_type_mut(weapon_id);
        if let Some(w) = w {
            w.ammo += ammo;
            w.max_ammo += ammo;
        } else {
            self.weapons.push(Weapon::new(
                weapon_id,
                WeaponLevel::Level1,
                0,
                ammo,
                ammo,
            ));
        }
    }

    pub fn trade_weapon(&mut self, old: Option<WeaponType>, new: WeaponType, max_ammo: u16) {
        if let Some(wtype) = old {
            if let Some(weapon) = self.get_weapon_by_type_mut(wtype) {
                let ammo = if max_ammo == 0 { weapon.max_ammo } else { max_ammo };
                *weapon = Weapon::new(new, WeaponLevel::Level1, 0, ammo, ammo);
            } else {
                self.add_weapon(new, max_ammo);
            }
        } else {
            self.add_weapon(new, max_ammo);
        }
    }

    pub fn remove_weapon(&mut self, wtype: WeaponType) {
        self.weapons.retain(|weapon| weapon.wtype != wtype);
    }

    pub fn get_weapon(&self, idx: usize) -> Option<&Weapon> {
        self.weapons.get(idx)
    }

    pub fn get_weapon_by_type_mut(&mut self, wtype: WeaponType) -> Option<&mut Weapon> {
        self.weapons.iter_mut().find(|weapon| weapon.wtype == wtype)
    }

    pub fn get_current_weapon(&self) -> Option<&Weapon> {
        self.weapons.get(self.current_weapon as usize)
    }

    pub fn get_current_weapon_mut(&mut self) -> Option<&mut Weapon> {
        self.weapons.get_mut(self.current_weapon as usize)
    }

    pub fn next_weapon(&mut self) {
        if (1 + self.current_weapon as usize) < self.weapons.len() {
            self.current_weapon += 1;
        } else {
            self.current_weapon = 0;
        }
    }

    pub fn prev_weapon(&mut self) {
        if self.current_weapon as usize > 0 {
            self.current_weapon -= 1;
        } else {
            self.current_weapon = self.weapons.len().saturating_sub(1) as u16;
        }
    }

    pub fn refill_all_ammo(&mut self) {
        for weapon in self.weapons.iter_mut() {
            weapon.ammo = weapon.max_ammo;
        }
    }

    pub fn reset_all_weapon_xp(&mut self) {
        for weapon in self.weapons.iter_mut() {
            weapon.level = WeaponLevel::Level1;
            weapon.experience = 0;
        }
    }

    pub fn reset_current_weapon_xp(&mut self) {
        if let Some(weapon) = self.get_current_weapon_mut() {
            weapon.reset_xp();
        }
    }

    pub fn take_xp(&mut self, exp: u16, state: &mut SharedGameState) -> TakeExperienceResult {
        let mut result = TakeExperienceResult::None;

        if let Some(weapon) = self.get_current_weapon_mut() {
            let lvl_table = state.constants.weapon.level_table[weapon.wtype as usize];
            let mut tmp_exp = weapon.experience as isize - exp as isize;

            if tmp_exp >= 0 {
                weapon.experience = tmp_exp as u16;
            } else {
                while tmp_exp < 0 {
                    if weapon.level == WeaponLevel::Level1 {
                        weapon.experience = 0;
                        tmp_exp = 0;
                    } else if weapon.experience < exp {
                        weapon.level = weapon.level.prev();
                        let max_level = lvl_table[weapon.level as usize - 1] as isize;
                        weapon.experience = (max_level + tmp_exp.max(-max_level)) as u16;
                        tmp_exp += max_level;

                        if weapon.wtype != WeaponType::Spur {
                            result = TakeExperienceResult::LevelDown;
                        }
                    }
                }
            }
        }

        result
    }

    pub fn add_xp(&mut self, exp: u16, player: &mut Player, state: &mut SharedGameState) {
        if let Some(weapon) = self.get_current_weapon_mut() {
            weapon.add_xp(exp, player, state);
        }
    }

    /// Get current experience state. Returns a (exp, max exp, max level/exp) tuple.
    pub fn get_current_max_exp(&self, constants: &EngineConstants) -> (u16, u16, bool) {
        if let Some(weapon) = self.weapons.get(self.current_weapon as usize) {
            weapon.get_max_exp(constants)
        } else {
            (0, 0, false)
        }
    }

    /// Get current ammunition state. Returns a (ammo, max ammo) tuple.
    pub fn get_current_ammo(&self) -> (u16, u16) {
        if let Some(weapon) = self.weapons.get(self.current_weapon as usize) {
            (weapon.ammo, weapon.max_ammo)
        } else {
            (0, 0)
        }
    }

    pub fn get_current_level(&self) -> WeaponLevel {
        if let Some(weapon) = self.weapons.get(self.current_weapon as usize) {
            weapon.level
        } else {
            WeaponLevel::None
        }
    }

    pub fn get_current_weapon_idx(&self) -> u16 {
        self.current_weapon
    }

    pub fn get_weapon_count(&self) -> usize {
        self.weapons.len()
    }

    pub fn has_weapon(&self, wtype: WeaponType) -> bool {
        self.weapons.iter().any(|weapon| weapon.wtype == wtype)
    }

    pub fn tick_weapons(&mut self, state: &mut SharedGameState, player: &mut Player, player_id: TargetPlayer, bullet_manager: &mut BulletManager) {
        if let Some(weapon) = self.get_current_weapon_mut() {
            weapon.tick(state, player, player_id, bullet_manager);
        }
    }
}

#[test]
fn inventory_test() {
    let mut inventory = Inventory::new();

    inventory.add_item(3);
    assert!(!inventory.has_item(2));
    assert!(inventory.has_item(3));

    assert!(inventory.has_item_amount(3, Ordering::Equal, 1));
    assert!(inventory.has_item_amount(3, Ordering::Less, 2));
    inventory.consume_item(3);

    assert!(inventory.has_item_amount(3, Ordering::Equal, 0));
    assert!(inventory.has_item_amount(3, Ordering::Less, 2));

    inventory.add_item(2);
    assert!(inventory.has_item(2));
    assert!(inventory.has_item_amount(2, Ordering::Equal, 1));
    assert!(!inventory.has_item_amount(2, Ordering::Less, 1));

    inventory.add_item(4);
    inventory.add_item(4);
    inventory.add_item(4);
    inventory.add_item(4);

    assert!(inventory.has_item(4));
    assert!(inventory.has_item_amount(4, Ordering::Greater, 3));
    assert!(inventory.has_item_amount(4, Ordering::Equal, 4));
    assert!(!inventory.has_item_amount(4, Ordering::Less, 2));
}