diff --git a/src/lib.rs b/src/lib.rs index 0c717499..2b37b9a6 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -1 +1,2 @@ pub mod framework; +pub mod load_roblox; diff --git a/src/load_roblox.rs b/src/load_roblox.rs new file mode 100644 index 00000000..2648c519 --- /dev/null +++ b/src/load_roblox.rs @@ -0,0 +1,28 @@ +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 +} + +pub fn get_objects(buf_thing: std::io::BufReader<&[u8]>, superclass: &str) -> Result, Box> { + // Using buffered I/O is recommended with rbx_binary + let dom = rbx_binary::from_reader(buf_thing)?; + + let mut objects = std::vec::Vec::::new(); + //move matching instances into objects + let (_,mut instances) = dom.into_raw(); + for (_,instance) in instances.drain() { + if class_is_a(instance.class.as_str(), superclass) { + objects.push(instance); + } + } + + return Ok(objects) +}