forked from StrafesNET/map-tool
74 lines
2.5 KiB
Rust
74 lines
2.5 KiB
Rust
fn class_is_a(class: &str, superclass: &str) -> bool {
|
|
if class==superclass {
|
|
return true
|
|
}
|
|
let class_descriptor=rbx_reflection_database::get().classes.get(class);
|
|
if let Some(descriptor) = &class_descriptor {
|
|
if let Some(class_super) = &descriptor.superclass {
|
|
return class_is_a(&class_super, superclass)
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
fn recursive_collect_scripts(scripts: &mut std::vec::Vec<rbx_dom_weak::types::Ref>,dom: &rbx_dom_weak::WeakDom, instance: &rbx_dom_weak::Instance){
|
|
for &referent in instance.children() {
|
|
if let Some(c) = dom.get_by_ref(referent) {
|
|
if class_is_a(c.class.as_str(), "LuaSourceContainer") {
|
|
scripts.push(c.referent());//copy ref
|
|
}
|
|
recursive_collect_scripts(scripts,dom,c);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
// Using buffered I/O is recommended with rbx_binary
|
|
let input = std::io::BufReader::new(std::fs::File::open("map.rbxm")?);
|
|
|
|
let dom = rbx_binary::from_reader(input)?;
|
|
|
|
//Construct allowed scripts
|
|
let mut allowed = std::collections::HashSet::<String>::new();
|
|
for entry in std::fs::read_dir("allowed")? {
|
|
allowed.insert(std::fs::read_to_string(entry?.path())?);
|
|
}
|
|
|
|
let mut scripts = std::vec::Vec::<rbx_dom_weak::types::Ref>::new();
|
|
recursive_collect_scripts(&mut scripts, &dom, dom.root());
|
|
|
|
//check scribb
|
|
let mut any_failed=false;
|
|
for (i,&referent) in scripts.iter().enumerate() {
|
|
if let Some(script) = dom.get_by_ref(referent) {
|
|
if let Some(rbx_dom_weak::types::Variant::String(s)) = script.properties.get("Source") {
|
|
if allowed.contains(s) {
|
|
println!("pass");
|
|
}else{
|
|
println!("fail");
|
|
any_failed=true;
|
|
std::fs::write(format!("blocked/{}.lua",i),s)?;
|
|
}
|
|
}else{
|
|
println!("failed to get source");
|
|
any_failed=true;
|
|
}
|
|
}else{
|
|
println!("failed to deref script");
|
|
any_failed=true;
|
|
}
|
|
}
|
|
if any_failed {
|
|
println!("One or more scripts are not allowed.");
|
|
return Ok(())//everything is not ok but idk how to return an error LMAO
|
|
}
|
|
println!("All scripts passed!");
|
|
// std::process::Command::new("rbxcompiler")
|
|
// .arg("--compile=false")
|
|
// .arg("--group=6980477")
|
|
// .arg("--asset=5692139100")
|
|
// .arg("--input=map.rbxm")
|
|
// .spawn()?;
|
|
Ok(())
|
|
}
|