PkmnLib_rs/src/dynamic_data/script_handling/volatile_scripts.rs

use crate::dynamic_data::script_handling::script::{Script, ScriptContainer};
use crate::dynamic_data::script_handling::script_set::ScriptSet;
use crate::{PkmnResult, StringKey};
use std::sync::Arc;

pub trait VolatileScripts<'a> {
    fn volatile_scripts(&self) -> &Arc<ScriptSet>;
    fn load_volatile_script(&self, key: &StringKey) -> PkmnResult<Option<Arc<dyn Script>>>;

    fn has_volatile_script(&self, key: &StringKey) -> bool {
        self.volatile_scripts().has(key)
    }

    fn get_volatile_script(&self, key: &StringKey) -> Option<ScriptContainer> {
        self.volatile_scripts().get(key)
    }

    fn add_volatile_script(&mut self, key: &StringKey) -> PkmnResult<Option<ScriptContainer>> {
        self.volatile_scripts()
            .stack_or_add(key, &|| self.load_volatile_script(key))
    }

    fn remove_volatile_script(&mut self, key: &StringKey) {
        self.volatile_scripts().remove(key)
    }
}