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/unityfs.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src/unityfs.rs')

diff --git a/src/unityfs.rs b/src/unityfs.rs
index 0f3cab6..a23114d 100644
--- a/src/unityfs.rs
+++ b/src/unityfs.rs
@@ -142,6 +142,15 @@ impl<T: Read + Seek> UnityFS<T> {
     pub fn nodes(&self) -> &[NodeInfo] {
         &self.nodes
     }
+
+    pub fn find_main_file(&self) -> Option<&NodeInfo> {
+        self.nodes().iter().find(|n| {
+            !n.name.ends_with(".resource")
+                && !n.name.ends_with(".resS")
+                && !n.name.ends_with(".sharedAssets")
+        })
+    }
+
     pub fn read<'a>(&'a mut self, node: &NodeInfo) -> std::io::Result<NodeReader<'a, T>> {
         self.reader.seek(SeekFrom::Start(node.offset))?;
         Ok(NodeReader {
-- 
cgit v1.2.3-70-g09d2