PkmnLib_rs/src/dynamic_data/models/battle_party.rs

use crate::dynamic_data::models::pokemon::Pokemon;
use crate::dynamic_data::models::pokemon_party::PokemonParty;
use std::sync::Arc;

#[derive(Debug)]
pub struct BattleParty<'own, 'library> {
    party: Arc<PokemonParty<'own, 'library>>,
    responsible_indices: Vec<(u8, u8)>,
}

impl<'own, 'library> BattleParty<'own, 'library> {
    pub fn new(party: Arc<PokemonParty<'own, 'library>>, responsible_indices: Vec<(u8, u8)>) -> Self {
        Self {
            party,
            responsible_indices,
        }
    }
    pub fn is_responsible_for_index(&self, side: u8, index: u8) -> bool {
        for responsible_index in &self.responsible_indices {
            if responsible_index.0 == side && responsible_index.1 == index {
                return true;
            }
        }
        false
    }

    pub fn has_pokemon_not_in_field(&self) -> bool {
        for pokemon in self.party.pokemon().iter().flatten() {
            if pokemon.is_usable() && !pokemon.is_on_battlefield() {
                return true;
            }
        }
        false
    }

    pub fn get_pokemon(&self, index: usize) -> &Option<Arc<Pokemon<'own, 'library>>> {
        self.party.at(index)
    }
}