main file finder helper

1 files changed, 8 insertions, 12 deletions

diff --git a/src/bin/probe.rs b/src/bin/probe.rs
index 85b2a7a..6f46c0a 100644
--- a/src/bin/probe.rs
+++ b/src/bin/probe.rs
@@ -6,19 +6,15 @@ fn main() -> Result<()> {
     let file = BufReader::new(File::open(args().nth(1).unwrap())?);
     let mut fs = UnityFS::open(file)?;
 
-    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 ch = read_serialized_file_header(&mut cab)?;
+    let node = fs.find_main_file().unwrap().to_owned();
+    let mut cab = fs.read(&node)?;
+    let ch = read_serialized_file_header(&mut cab)?;
 
-        if fs.unity_version.is_ascii() && ch.generator_version.is_ascii() && ch.format < 100 {
-            println!(
-                "{}\t{}\t{}\t{}",
-                fs.file_version, fs.unity_version, ch.format, ch.generator_version
-            );
-        }
+    if fs.unity_version.is_ascii() && ch.generator_version.is_ascii() && ch.format < 100 {
+        println!(
+            "{}\t{}\t{}\t{}",
+            fs.file_version, fs.unity_version, ch.format, ch.generator_version
+        );
     }
 
     Ok(())