more abstraction around unityfs to read multiple files from a single reader

1 files changed, 27 insertions, 8 deletions

diff --git a/src/classes/pptr.rs b/src/classes/pptr.rs
index 30f37ad..9b54cbb 100644
--- a/src/classes/pptr.rs
+++ b/src/classes/pptr.rs
@@ -54,17 +54,36 @@ impl<T: FromValue> PPtr<T> {
     pub fn is_null(&self) -> bool {
         self.path_id == 0 && self.file_id == 0
     }
-    pub fn load(&self, file: &mut SerializedFile<impl Read + Seek>) -> Result<T> {
+    pub fn load(
+        &self,
+        file: &mut SerializedFile<impl Read + Seek>,
+        shared_assets: Option<&mut SerializedFile<impl Read + Seek>>,
+    ) -> Result<T> {
         debug!(
             "loading PPtr<{}> file_id={} path_id={}",
             self.class, self.file_id, self.path_id
         );
-        let ob = file
-            .objects
-            .iter()
-            .find(|o| o.path_id == self.path_id)
-            .ok_or(anyhow!("object with path id {} not found", self.path_id))?
-            .to_owned();
-        file.read_object(ob)?.parse()
+        match self.file_id {
+            0 => {
+                let ob = file
+                    .objects
+                    .iter()
+                    .find(|o| o.path_id == self.path_id)
+                    .ok_or(anyhow!("object with path id {} not found", self.path_id))?
+                    .to_owned();
+                file.read_object(ob)?.parse()
+            }
+            1 => {
+                let file = shared_assets.unwrap();
+                let ob = file
+                    .objects
+                    .iter()
+                    .find(|o| o.path_id == self.path_id)
+                    .ok_or(anyhow!("object with path id {} not found", self.path_id))?
+                    .to_owned();
+                file.read_object(ob)?.parse()
+            }
+            _ => unimplemented!(),
+        }
     }
 }