StrafesNET/roblox_emulator
3
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

try setting up methods on init

Browse Source
This commit is contained in:
Quaternions 2024-10-21 18:01:55 -07:00
parent 57affe7a95
commit c162bb34d7
2 changed files with 30 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/runner/runner.rs
View File

@ -33,6 +33,7 @@ fn init(lua:&mlua::Lua)->mlua::Result<()>{
#[cfg(feature="run-service")]
crate::scheduler::set_globals(lua,&globals)?;
super::script_signal::set_globals(lua,&globals)?;
super::r#enum::set_globals(lua,&globals)?;
super::color3::set_globals(lua,&globals)?;
super::vector3::set_globals(lua,&globals)?;

61
src/runner/script_signal.rs
View File

@ -1,5 +1,7 @@
use std::{cell::RefCell,rc::Rc};
use mlua::UserDataFields;
#[derive(Clone)]
struct FunctionList{
functions:Vec<mlua::Function>,
@ -41,10 +43,9 @@ pub struct ScriptSignal{
// Emulate the garbage roblox api.
// ScriptConnection should not exist.
// :Disconnect should be a method on ScriptSignal, and this would be avoided entirely.
callbacks:RcFunctionList,
connections:RcFunctionList,
once:RcFunctionList,
wait:Rc<RefCell<Vec<mlua::Thread>>>,
wait_function:std::cell::OnceCell<mlua::Function>,
}
pub struct ScriptConnection{
connection:RcFunctionList,
@ -53,14 +54,13 @@ pub struct ScriptConnection{
impl ScriptSignal{
pub fn new()->Self{
Self{
callbacks:RcFunctionList::new(),
connections:RcFunctionList::new(),
once:RcFunctionList::new(),
wait:Rc::new(RefCell::new(Vec::new())),
wait_function:std::cell::OnceCell::new(),
}
}
pub fn fire(&self,args:&mlua::MultiValue){
self.callbacks.fire(args);
self.connections.fire(args);
//Replace the FunctionList with an empty one and drop the borrow
let once=std::mem::replace(&mut *self.once.functions.borrow_mut(),FunctionList::new());
once.fire(args);
@ -73,9 +73,9 @@ impl ScriptSignal{
}
}
pub fn connect(&self,function:mlua::Function)->ScriptConnection{
self.callbacks.functions.borrow_mut().functions.push(function.clone());
self.connections.functions.borrow_mut().functions.push(function.clone());
ScriptConnection{
connection:self.callbacks.clone(),
connection:self.connections.clone(),
function,
}
}
@ -109,32 +109,29 @@ return function()
return coroutine_yield()
end";
pub fn set_globals(lua:&mlua::Lua,globals:&mlua::Table)->Result<(),mlua::Error>{
let coroutine_table=globals.get::<mlua::Table>("coroutine")?;
let wait_thread=lua.create_function(wait_thread)?;
//create wait function environment
let wait_env=lua.create_table()?;
wait_env.raw_set("coroutine",coroutine_table)?;
wait_env.raw_set("wait_thread",wait_thread)?;
//construct wait function from Lua code
let wait=lua.load(LUA_WAIT)
.set_name("wait")
.set_environment(wait_env)
.call::<mlua::Function>(())?;
lua.register_userdata_type::<ScriptSignal>(|reg|{
reg.add_field("Wait",wait);
})?;
Ok(())
}
impl mlua::UserData for ScriptSignal{
fn add_fields<F:mlua::UserDataFields<Self>>(fields:&mut F){
fields.add_field_method_get("Wait",|lua,this|{
Ok(match this.wait_function.get(){
Some(f)=>f.clone(),
None=>{
let coroutine_table=lua.globals().get::<mlua::Table>("coroutine")?;
let wait_thread=lua.create_function(wait_thread)?;
//create wait function environment
let wait_env=lua.create_table()?;
wait_env.raw_set("coroutine",coroutine_table)?;
wait_env.raw_set("wait_thread",wait_thread)?;
//construct wait function from Lua code
let wait=lua.load(LUA_WAIT)
.set_name("wait")
.set_environment(wait_env)
.call::<mlua::Function>(())?;
this.wait_function.set(wait.clone()).unwrap();
wait
}
})
});
}
fn add_methods<M:mlua::UserDataMethods<Self>>(methods:&mut M){
methods.add_method("Connect",|_lua,this,f:mlua::Function|
Ok(this.connect(f))