write "Ref" attribute based on internal roblox part id

Cargo.lock

@@ -609,6 +609,29 @@ dependencies = [
  "wasm-bindgen",
 ]
 
+[[package]]
+name = "lazy-regex"
+version = "3.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5d12be4595afdf58bd19e4a9f4e24187da2a66700786ff660a418e9059937a4c"
+dependencies = [
+ "lazy-regex-proc_macros",
+ "once_cell",
+ "regex",
+]
+
+[[package]]
+name = "lazy-regex-proc_macros"
+version = "3.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "44bcd58e6c97a7fcbaffcdc95728b393b8d98933bfadad49ed4097845b57ef0b"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "regex",
+ "syn 2.0.38",
+]
+
 [[package]]
 name = "lazy_static"
 version = "1.4.0"
@@ -688,11 +711,11 @@ dependencies = [
  "flate2",
  "image",
  "image_dds",
+ "lazy-regex",
  "rbx_binary",
  "rbx_dom_weak",
  "rbx_reflection_database",
  "rbx_xml",
- "regex",
 ]
 
 [[package]]

Cargo.toml

@@ -11,11 +11,11 @@ clap = { version = "4.4.2", features = ["derive"] }
 flate2 = "1.0.27"
 image = "0.24.7"
 image_dds = "0.1.1"
+lazy-regex = "3.1.0"
 rbx_binary = "0.7.1"
 rbx_dom_weak = "2.5.0"
 rbx_reflection_database = "0.2.7"
 rbx_xml = "0.13.1"
-regex = "1.9.5"
 
 #[profile.release]
 #lto = true

src/main.rs

@@ -17,6 +17,7 @@ enum Commands {
     ConvertTextures,
     DownloadMeshes(PathBufList),
     Extract(PathBufList),
+    WriteAttributes,
     Interactive,
     Replace,
     Scan,
@@ -56,6 +57,16 @@ fn recursive_collect_superclass(objects: &mut std::vec::Vec<rbx_dom_weak::types:
         }
     }
 }
+fn recursive_collect_regex(objects: &mut std::vec::Vec<rbx_dom_weak::types::Ref>,dom: &rbx_dom_weak::WeakDom, instance: &rbx_dom_weak::Instance, regex: &lazy_regex::Lazy<lazy_regex::Regex>){
+    for &referent in instance.children() {
+        if let Some(c) = dom.get_by_ref(referent) {
+            if regex.captures(c.name.as_str()).is_some(){
+                objects.push(c.referent());//copy ref
+            }
+            recursive_collect_regex(objects,dom,c,regex);
+        }
+    }
+}
 fn get_full_name(dom:&rbx_dom_weak::WeakDom,instance:&rbx_dom_weak::Instance) -> String{
     let mut full_name=instance.name.clone();
     let mut pref=instance.parent();
@@ -90,6 +101,11 @@ fn get_script_refs(dom:&rbx_dom_weak::WeakDom) -> Vec<rbx_dom_weak::types::Ref>{
     recursive_collect_superclass(&mut scripts, dom, dom.root(),"LuaSourceContainer");
     scripts
 }
+fn get_button_refs(dom:&rbx_dom_weak::WeakDom) -> Vec<rbx_dom_weak::types::Ref>{
+    let mut buttons = std::vec::Vec::new();
+    recursive_collect_regex(&mut buttons, dom, dom.root(),lazy_regex::regex!(r"Button(\d+)$"));
+    buttons
+}
 fn get_texture_refs(dom:&rbx_dom_weak::WeakDom) -> Vec<rbx_dom_weak::types::Ref>{
     let mut objects = std::vec::Vec::new();
     recursive_collect_superclass(&mut objects, dom, dom.root(),"Decal");
@@ -264,7 +280,7 @@ struct RobloxAssetIdParseErr;
 impl std::str::FromStr for RobloxAssetId {
     type Err=RobloxAssetIdParseErr;
     fn from_str(s: &str) -> Result<Self, Self::Err>{
-        let regman=regex::Regex::new(r"(\d+)$").unwrap();
+        let regman=lazy_regex::regex!(r"(\d+)$");
         if let Some(captures) = regman.captures(s) {
             if captures.len()==2{//captures[0] is all captures concatenated, and then each individual capture
                 if let Ok(id) = captures[0].parse::<u64>() {
@@ -961,6 +977,51 @@ fn unzip_all()->AResult<()>{
     Ok(())
 }
 
+fn write_attributes() -> AResult<()>{
+    for entry in std::fs::read_dir("maps/unprocessed")? {
+        let file_thing=entry?;
+        println!("processing map={:?}",file_thing.file_name());
+        let mut input = std::io::BufReader::new(std::fs::File::open(file_thing.path())?);
+        let mut dom = get_dom(&mut input)?;
+
+        let button_refs = get_button_refs(&dom);
+
+        for &button_ref in &button_refs {
+            if let Some(button)=dom.get_by_ref_mut(button_ref){
+                match button.properties.get_mut("Attributes"){
+                    Some(rbx_dom_weak::types::Variant::Attributes(attributes))=>{
+                        println!("Appending Ref={} to existing attributes for {}",button_ref,button.name);
+                        attributes.insert("Ref".to_string(),rbx_dom_weak::types::Variant::String(button_ref.to_string()));
+                    },
+                    None=>{
+                        println!("Creating new attributes with Ref={} for {}",button_ref,button.name);
+                        let mut attributes=rbx_dom_weak::types::Attributes::new();
+                        attributes.insert("Ref".to_string(),rbx_dom_weak::types::Variant::String(button_ref.to_string()));
+                        button.properties.insert("Attributes".to_string(),rbx_dom_weak::types::Variant::Attributes(attributes));
+                    }
+                    _=>unreachable!("Fetching attributes did not return attributes."),
+                }
+            }
+        }
+        let mut dest={
+            let mut dest=std::path::PathBuf::from("maps/attributes");
+            dest.push(file_thing.file_name());
+            let output = std::io::BufWriter::new(std::fs::File::create(dest)?);
+            //write workspace:GetChildren()[1]
+            let workspace_children=dom.root().children();
+            if workspace_children.len()!=1{
+                return Err(anyhow::Error::msg("there can only be one model"));
+            }
+            rbx_binary::to_writer(output, &dom, &[workspace_children[0]])?;
+            //move original to processed folder
+            std::path::PathBuf::from("maps/unaltered")
+        };
+        dest.push(file_thing.file_name());
+        std::fs::rename(file_thing.path(), dest)?;
+    }
+    Ok(())
+}
+
 fn main() -> AResult<()> {
     let cli = Cli::parse();
     match cli.command {
@@ -969,6 +1030,7 @@ fn main() -> AResult<()> {
         Commands::ConvertTextures=>convert_textures(),
         Commands::DownloadMeshes(pathlist)=>download_meshes(pathlist.paths),
         Commands::Extract(pathlist)=>extract(pathlist.paths),
+        Commands::WriteAttributes=>write_attributes(),
         Commands::Interactive=>interactive(),
         Commands::Replace=>replace(),
         Commands::Scan=>scan(),