don't require vers header

BrettMayson · Nov 9, 2024 · 44b1100 · 44b1100
1 parent 55e9f0c
commit 44b1100
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/libs/pbo/src/error.rs b/libs/pbo/src/error.rs
@@ -7,9 +7,6 @@ pub enum Error {
     /// [`std::io::Error`]
     Io(#[from] std::io::Error),
 
-    #[error("Missing Vers header")]
-    /// Missing Vers header
-    NoVersHeader,
     #[error("HEMTT does not support the encountered PBO Mime type: {0}")]
     /// HEMTT does not support the encountered PBO Mime type
     UnsupportedMime(u32),

diff --git a/libs/pbo/src/read.rs b/libs/pbo/src/read.rs
@@ -15,7 +15,7 @@ use crate::{
 /// An existing PBO file that can be read from
 pub struct ReadablePbo<I: Seek + Read> {
     properties: IndexMap<String, String>,
-    vers_header: Header,
+    vers_header: Option<Header>,
     headers: Vec<Header>,
     checksum: Checksum,
     input: I,
@@ -54,7 +54,7 @@ impl<I: Seek + Read> ReadablePbo<I> {
             }
         }
 
-        let vers_header = vers_header.ok_or(Error::NoVersHeader)?;
+        let vers_header = vers_header;
 
         for header in &headers {
             input.seek(SeekFrom::Current(i64::from(header.size())))?;
@@ -177,7 +177,9 @@ impl<I: Seek + Read> ReadablePbo<I> {
     /// if a file does not exist, but a header for it does
     pub fn gen_checksum(&mut self) -> Result<Checksum, Error> {
         let mut headers: Cursor<Vec<u8>> = Cursor::new(Vec::new());
-        self.vers_header.write_pbo(&mut headers)?;
+        if let Some(vers_header) = &self.vers_header {
+            vers_header.write_pbo(&mut headers)?;
+        }
 
         if let Some(prefix) = self.properties.get("prefix") {
             headers.write_cstring(b"prefix")?;