PkmnLib_rs/src/script_implementations/wasm/script_function_cache.rs

use std::sync::Arc;

use parking_lot::RwLock;
use paste::paste;

use crate::dynamic_data::{DynamicLibrary, TurnChoice};
use crate::script_implementations::wasm::extern_ref::{ExternRef, VecExternRef};
use crate::script_implementations::wasm::script_resolver::WebAssemblyEnvironmentData;
use crate::static_data::EffectParameter;
use crate::StringKey;
use wasmer::TypedFunction;

/// A macro to generate the script function cache a bit easier.
macro_rules! script_function_cache {
    (
        $(
            $name:ident($($par_type:ty),*$(,)?)
        )*
    ) => {
        #[derive(Default)]
        #[allow(unused_parens)]
        pub struct ScriptFunctionCache {
            $(
                $name: RwLock<Option<TypedFunction<(u32$(, $par_type)*), ()>>>,
            )*
        }

        impl ScriptFunctionCache {
            $(
                paste! {
                    #[cold]
                    #[allow(unused_parens)]
                    fn [<initialize_ $name>](&self, env: &Arc<WebAssemblyEnvironmentData>) {
                        let exported = env.exported_functions();
                        let f = exported.get::<StringKey>(&stringify!([< script_ $name >]).into());
                        if let Some(f) = f {
                            let func: TypedFunction<(u32 $(,$par_type)*), ()> = f.typed(&env.store_ref()).unwrap();
                            let _ = self.$name.write().insert(func);
                        }
                    }


                    #[allow(unused_parens)]
                    pub(crate) fn [<$name>](
                        &self,
                        env: &Arc<WebAssemblyEnvironmentData>,
                    ) -> Option<TypedFunction<(u32 $(,$par_type)*), ()>> {
                        {
                            let read_lock = self.$name.read();
                            if let Some(f) = read_lock.as_ref() {
                                return Some(f.clone());
                            }
                        }
                        self.[<initialize_ $name>](env);
                        self.$name.read().as_ref().cloned()
                    }
                }
            )*
        }
    }
}

script_function_cache! {
    stack()
    on_remove()
    on_initialize(ExternRef<DynamicLibrary>, VecExternRef<EffectParameter>)
    on_before_turn(ExternRef<TurnChoice>)
    change_speed(ExternRef<TurnChoice>, u32)
    change_priority(ExternRef<TurnChoice>, u32)
}