more inspection tools

1 files changed, 44 insertions, 0 deletions

diff --git a/src/bin/yaml.rs b/src/bin/yaml.rs
new file mode 100644
index 0000000..e541c62
--- /dev/null
+++ b/src/bin/yaml.rs
@@ -0,0 +1,44 @@
+use std::{
+    env::args,
+    fs::File,
+    io::{BufReader, Seek, SeekFrom, stdout},
+};
+use unity_tools::{
+    classes::HValue, object::read_value, serialized_file::read_serialized_file, unityfs::UnityFS,
+};
+
+fn main() -> anyhow::Result<()> {
+    env_logger::init_from_env("LOG");
+    let file = BufReader::new(File::open(args().nth(1).unwrap())?);
+    let mut fs = UnityFS::open(file)?;
+    let filter = args().nth(2);
+
+    for node in fs.nodes().to_vec() {
+        if node.name.ends_with(".resource") || node.name.ends_with(".resS") {
+            continue;
+        }
+        let mut cab = fs.read(&node)?;
+        let file = read_serialized_file(&mut cab)?;
+        for ob in file.objects {
+            cab.seek(SeekFrom::Start(ob.data_offset))?;
+            let typetree = if ob.type_id < 0 {
+                unimplemented!()
+            } else {
+                &file.types[ob.type_id as usize]
+            };
+            if let Some(typetree) = &typetree.type_tree {
+                if let Some(f) = &filter {
+                    if typetree.type_string != *f && ob.path_id.to_string() != *f {
+                        continue;
+                    }
+                }
+                let value = read_value(typetree, file.endianness, &mut cab)?;
+                let hvalue = HValue::from_value(value)?;
+                serde_yml::to_writer(stdout(), &hvalue).unwrap();
+                println!()
+            }
+        }
+    }
+
+    Ok(())
+}