sneak past type issues

src/runner/cframe.rs

@@ -76,11 +76,5 @@ impl mlua::UserData for CFrame{
 		);
 	}
 }
-impl<'lua> mlua::FromLua<'lua> for CFrame{
+
-	fn from_lua(value:mlua::prelude::LuaValue<'lua>,_lua:&'lua mlua::prelude::Lua)->mlua::prelude::LuaResult<Self>{
+type_from_lua_userdata!(CFrame);
-		match value{
-			mlua::Value::UserData(ud)=>ud.take()?,
-			other=>Err(mlua::Error::runtime(format!("Expected CFrame got {:?}",other))),
-		}
-	}
-}

src/runner/color3.rs

@@ -40,11 +40,4 @@ impl mlua::UserData for Color3{
 	fn add_methods<'lua,M:mlua::UserDataMethods<'lua,Self>>(methods:&mut M){
 	}
 }
-impl<'lua> mlua::FromLua<'lua> for Color3{
+type_from_lua_userdata!(Color3);
-	fn from_lua(value:mlua::prelude::LuaValue<'lua>,_lua:&'lua mlua::prelude::Lua)->mlua::prelude::LuaResult<Self>{
-		match value{
-			mlua::Value::UserData(ud)=>ud.take()?,
-			other=>Err(mlua::Error::runtime(format!("Expected Color3 got {:?}",other))),
-		}
-	}
-}

src/runner/enum.rs

@@ -39,14 +39,7 @@ impl mlua::UserData for EnumItem{
 		});
 	}
 }
-impl<'lua> mlua::FromLua<'lua> for EnumItem{
+type_from_lua_userdata!(EnumItem);
-	fn from_lua(value:mlua::prelude::LuaValue<'lua>,_lua:&'lua mlua::prelude::Lua)->mlua::prelude::LuaResult<Self>{
-		match value{
-			mlua::Value::UserData(ud)=>ud.take()?,
-			other=>Err(mlua::Error::runtime(format!("Expected Enum got {:?}",other))),
-		}
-	}
-}
 
 impl mlua::UserData for EnumItems{
 	fn add_fields<'lua,F:mlua::UserDataFields<'lua,Self>>(_fields:&mut F){
@@ -61,14 +54,7 @@ impl mlua::UserData for EnumItems{
 		});
 	}
 }
-impl<'lua> mlua::FromLua<'lua> for EnumItems{
+type_from_lua_userdata!(EnumItems);
-	fn from_lua(value:mlua::prelude::LuaValue<'lua>,_lua:&'lua mlua::prelude::Lua)->mlua::prelude::LuaResult<Self>{
-		match value{
-			mlua::Value::UserData(ud)=>ud.take()?,
-			other=>Err(mlua::Error::runtime(format!("Expected Enum got {:?}",other))),
-		}
-	}
-}
 
 impl mlua::UserData for Enum{
 	fn add_fields<'lua,F:mlua::UserDataFields<'lua,Self>>(_fields:&mut F){
@@ -76,11 +62,4 @@ impl mlua::UserData for Enum{
 	fn add_methods<'lua,M:mlua::UserDataMethods<'lua,Self>>(_methods:&mut M){
 	}
 }
-impl<'lua> mlua::FromLua<'lua> for Enum{
+type_from_lua_userdata!(Enum);
-	fn from_lua(value:mlua::prelude::LuaValue<'lua>,_lua:&'lua mlua::prelude::Lua)->mlua::prelude::LuaResult<Self>{
-		match value{
-			mlua::Value::UserData(ud)=>ud.take()?,
-			other=>Err(mlua::Error::runtime(format!("Expected Enum got {:?}",other))),
-		}
-	}
-}

src/runner/instance.rs

@@ -29,6 +29,16 @@ fn get_full_name(dom:&rbx_dom_weak::WeakDom,instance:&rbx_dom_weak::Instance)->S
 	full_name
 }
 
+//workaround until I have an enum of classes
+struct Dereferent(Ref);
+impl mlua::UserData for Dereferent{}
+type_from_lua_userdata!(Dereferent);
+impl Referent for Dereferent{
+	fn referent(&self)->Ref{
+		self.0
+	}
+}
+
 trait Referent{
 	fn referent(&self)->Ref;
 	fn get<'a>(&self,dom:&'a WeakDom)->mlua::Result<&'a rbx_dom_weak::Instance>{
@@ -54,20 +64,16 @@ macro_rules! class{
 				self.referent
 			}
 		}
-		impl<'lua> mlua::FromLua<'lua> for $class{
+		type_from_lua_userdata!($class);
-			fn from_lua(value:mlua::Value<'lua>,_lua:&'lua mlua::Lua)->mlua::Result<Self>{
-				match value{
-					mlua::Value::UserData(ud)=>ud.take(),
-					other=>Err(mlua::Error::runtime(format!("Expected {} got {:?}",stringify!($class),other))),
-				}
-			}
-		}
 	};
 }
 macro_rules! class_composition{
 	($class:ident,($($superclass:ident),*))=>{
 		impl mlua::UserData for $class{
 			fn add_fields<'lua,F:mlua::UserDataFields<'lua,Self>>(fields:&mut F){
+				fields.add_field_method_get("Referent",|_,this|{
+					Ok(Dereferent(this.referent()))
+				});
 				$(
 					$superclass::composition_add_fields(fields);
 				)*
@@ -93,9 +99,10 @@ impl Instance{
 				Ok(Instance::new(instance.parent()))
 			})
 		});
-		fields.add_field_method_set("Parent",|lua,this,val:Self|{
+		fields.add_field_method_set("Parent",|lua,this,val:mlua::AnyUserData|{
+			let Dereferent(referent)=mlua::AnyUserDataExt::get(&val,"Referent")?;
 			dom(lua,|dom|{
-				dom.transfer_within(this.referent(),val.referent);
+				dom.transfer_within(this.referent(),referent);
 				Ok(())
 			})
 		});
@@ -150,10 +157,29 @@ impl Instance{
 				Ok(crate::context::class_is_a(instance.class.as_str(),classname.to_str()?))
 			})
 		);
-		methods.add_meta_function(mlua::MetaMethod::NewIndex,|lua,(this,index,value):(Self,mlua::String,mlua::Value)|
+		methods.add_meta_function(mlua::MetaMethod::Index,|lua,(this,index):(mlua::AnyUserData,mlua::String)|{
+			let index_str=index.to_str()?;
+			let dereferent:Dereferent=mlua::AnyUserDataExt::get(&this,"Referent")?;
 			dom(lua,|dom|{
-				//println!("__newindex t={this:?} i={index:?} v={value:?}");
+				let instance=dereferent.get(dom)?;
-				let instance=this.get_mut(dom)?;
+				//find a child with a matching name
+				let maybe_child=instance.children()
+				.iter()
+				.find(|&&r|
+					dom.get_by_ref(r)
+					.is_some_and(|instance|instance.name==index_str)
+				);
+				match maybe_child{
+					Some(&referent)=>Instance::new(referent).into_lua(lua),
+					None=>mlua::Value::Nil.into_lua(lua),
+				}
+			})
+		});
+		methods.add_meta_function(mlua::MetaMethod::NewIndex,|lua,(this,index,value):(mlua::AnyUserData,mlua::String,mlua::Value)|{
+			let dereferent:Dereferent=mlua::AnyUserDataExt::get(&this,"Referent")?;
+			dom(lua,|dom|{
+				let instance=dereferent.get_mut(dom)?;
+				//println!("__newindex t={} i={index:?} v={value:?}",instance.name);
 				let index_str=index.to_str()?;
 				let db=rbx_reflection_database::get();
 				let class=db.classes.get(instance.class.as_str()).ok_or(mlua::Error::runtime("Class missing"))?;
@@ -179,7 +205,7 @@ impl Instance{
 				}
 				Ok(())
 			})
-		);
+		});
 	}
 }
 

src/runner/macros.rs

@@ -0,0 +1,12 @@
+macro_rules! type_from_lua_userdata{
+	($asd:ident)=>{
+		impl<'lua> mlua::FromLua<'lua> for $asd{
+			fn from_lua(value:mlua::Value<'lua>,_lua:&'lua mlua::Lua)->Result<Self,mlua::Error>{
+				match value{
+					mlua::Value::UserData(ud)=>ud.take(),
+					other=>Err(mlua::Error::runtime(format!("Expected {} got {:?}",stringify!($asd),other))),
+				}
+			}
+		}
+	};
+}

src/runner/mod.rs

@@ -1,3 +1,5 @@
+#[macro_use]
+mod macros;
 mod runner;
 
 mod r#enum;

src/runner/runner.rs

@@ -16,8 +16,8 @@ pub enum Error{
 impl std::fmt::Display for Error{
 	fn fmt(&self,f:&mut std::fmt::Formatter<'_>)->std::fmt::Result{
 		match self{
-			Self::Lua{source,error:mlua::Error::RuntimeError(s)}=>write!(f,"lua error: {s}\nsource:{source}"),
+			Self::Lua{source,error}=>write!(f,"lua error: source:\n{source}\n{error}"),
-			Self::RustLua(mlua::Error::RuntimeError(s))=>write!(f,"rust-side lua error: {s}"),
+			Self::RustLua(error)=>write!(f,"rust-side lua error: {error}"),
 			other=>write!(f,"{other:?}"),
 		}
 	}

src/runner/vector3.rs

@@ -73,11 +73,4 @@ impl mlua::UserData for Vector3{
 	}
 }
 
-impl<'lua> mlua::FromLua<'lua> for Vector3{
+type_from_lua_userdata!(Vector3);
-	fn from_lua(value:mlua::prelude::LuaValue<'lua>,_lua:&'lua mlua::prelude::Lua)->mlua::prelude::LuaResult<Self>{
-		match value{
-			mlua::Value::UserData(ud)=>ud.take(),
-			other=>Err(mlua::Error::runtime(format!("Expected Vector3 got {:?}",other))),
-		}
-	}
-}