1use std::io::Cursor;
5
6use crate::ByteBuffer;
7use crate::ByteSpan;
8use binrw::BinRead;
9use binrw::BinWrite;
10use binrw::binrw;
11
12#[binrw]
13#[derive(Debug)]
14#[brw(little)]
15#[brw(magic = b"UWB1")]
16pub struct Uwb {
17 pub file_size: u32,
19 #[br(temp)]
21 #[bw(calc = uwcs.len() as u32)]
22 uwc_count: u32,
23 #[br(count = uwc_count)]
24 pub uwcs: Vec<Uwc>,
25}
26
27#[binrw]
28#[derive(Debug)]
29#[brw(little)]
30#[brw(magic = b"UWC1")]
31pub struct Uwc {
32 pub file_size: u32,
34 #[br(count = file_size - 8)]
36 pub unk1: Vec<u8>,
37}
38
39impl Uwb {
40 pub fn from_existing(buffer: ByteSpan) -> Option<Self> {
42 let mut cursor = Cursor::new(buffer);
43 Uwb::read(&mut cursor).ok()
44 }
45
46 pub fn write_to_buffer(&self) -> Option<ByteBuffer> {
47 let mut buffer = ByteBuffer::new();
48
49 {
50 let mut cursor = Cursor::new(&mut buffer);
51 self.write_le(&mut cursor).ok()?;
52 }
53
54 Some(buffer)
55 }
56}
57
58#[cfg(test)]
59mod tests {
60 use std::fs::read;
61 use std::path::PathBuf;
62
63 use super::*;
64
65 #[test]
66 fn test_invalid() {
67 let mut d = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
68 d.push("resources/tests");
69 d.push("random");
70
71 Uwb::from_existing(&read(d).unwrap());
73 }
74}