From 58186bb96e38b92e426e75e08b0e0c95d05d319c Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Thu, 13 Mar 2025 19:50:50 +0100
Subject: main file finder helper

---
 src/bin/json.rs | 46 +++++++++++++++++++++-------------------------
 1 file changed, 21 insertions(+), 25 deletions(-)

(limited to 'src/bin/json.rs')

diff --git a/src/bin/json.rs b/src/bin/json.rs
index b969beb..d669148 100644
--- a/src/bin/json.rs
+++ b/src/bin/json.rs
@@ -14,33 +14,29 @@ fn main() -> anyhow::Result<()> {
     let filter = args().nth(2);
     let pretty = var("PRETTY").is_ok();
 
-    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)?;
-                if pretty {
-                    serde_json::to_writer_pretty(stdout(), &value.to_json()).unwrap();
-                } else {
-                    serde_json::to_writer(stdout(), &value.to_json()).unwrap();
+    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;
                 }
-                println!()
             }
+            let value = read_value(typetree, file.endianness, &mut cab)?;
+            if pretty {
+                serde_json::to_writer_pretty(stdout(), &value.to_json()).unwrap();
+            } else {
+                serde_json::to_writer(stdout(), &value.to_json()).unwrap();
+            }
+            println!()
         }
     }
 
-- 
cgit v1.2.3-70-g09d2