main file finder helper

1 files changed, 22 insertions, 25 deletions

diff --git a/src/bin/yaml.rs b/src/bin/yaml.rs
index e3b1df8..2899c4c 100644
--- a/src/bin/yaml.rs
+++ b/src/bin/yaml.rs
@@ -15,34 +15,31 @@ fn main() -> anyhow::Result<()> {
     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 node = fs.find_main_file().unwrap().to_owned();
+    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)?;
+            }
+            let value = read_value(typetree, file.endianness, &mut cab)?;
+            let hvalue = HValue::from_value(value)?;
 
-                let mut hvalue = serde_yml::to_value(hvalue)?;
-                reduce_large_arrays(&mut hvalue);
+            let mut hvalue = serde_yml::to_value(hvalue)?;
+            reduce_large_arrays(&mut hvalue);
 
-                serde_yml::to_writer(stdout(), &hvalue).unwrap();
-                println!()
-            }
+            serde_yml::to_writer(stdout(), &hvalue).unwrap();
+            println!()
         }
     }