1 files changed, 50 insertions, 0 deletions
diff --git a/exporter/src/bin/yaml.rs b/exporter/src/bin/yaml.rs
new file mode 100644
index 0000000..4ef8933
--- /dev/null
+++ b/exporter/src/bin/yaml.rs
@@ -0,0 +1,50 @@
+use serde_yml::Value;
+use std::{
+    env::args,
+    fs::File,
+    io::{BufReader, stdout},
+};
+use unity_tools::{classes::HValue, serialized_file::SerializedFile, 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);
+
+    let node = fs.find_main_file().unwrap().to_owned();
+    let mut cab = fs.read(&node)?;
+
+    let mut file = SerializedFile::read(&mut cab)?;
+    for ob in file.objects.clone() {
+        if let Some(f) = &filter {
+            if file.get_object_type_tree(&ob)?.type_string != *f && ob.path_id.to_string() != *f {
+                continue;
+            }
+        }
+        let value = file.read_object(ob)?;
+        let hvalue = HValue::from_value(value)?;
+
+        let mut hvalue = serde_yml::to_value(hvalue)?;
+        trim_large_arrays(&mut hvalue);
+
+        serde_yml::to_writer(stdout(), &hvalue).unwrap();
+        println!()
+    }
+
+    Ok(())
+}
+
+fn trim_large_arrays(v: &mut Value) {
+    match v {
+        Value::Sequence(values) => {
+            values.iter_mut().for_each(trim_large_arrays);
+            while values.len() > 32 {
+                values.pop();
+            }
+        }
+        Value::Mapping(mapping) => mapping.map.values_mut().for_each(trim_large_arrays),
+        Value::Tagged(tagged_value) => trim_large_arrays(&mut tagged_value.value),
+        _ => (),
+    }
+}