1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

use no_std_io::io::{Read, Write};

use crate::{writer::Writer, DekuError, DekuReader, DekuWriter};

impl<'a, T: DekuReader<'a, Ctx>, Ctx: Copy> DekuReader<'a, Ctx> for Option<T> {
    fn from_reader_with_ctx<R: Read>(
        reader: &mut crate::reader::Reader<R>,
        inner_ctx: Ctx,
    ) -> Result<Self, DekuError> {
        let val = <T>::from_reader_with_ctx(reader, inner_ctx)?;
        Ok(Some(val))
    }
}

impl<T: DekuWriter<Ctx>, Ctx: Copy> DekuWriter<Ctx> for Option<T> {
    fn to_writer<W: Write>(&self, writer: &mut Writer<W>, inner_ctx: Ctx) -> Result<(), DekuError> {
        self.as_ref()
            .map_or(Ok(()), |v| v.to_writer(writer, inner_ctx))
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use no_std_io::io::Cursor;

    use crate::reader::Reader;

    #[test]
    fn test_option_read() {
        use crate::ctx::*;
        let input = &[1u8, 2, 3, 4];
        let mut cursor = Cursor::new(input);
        let mut reader = Reader::new(&mut cursor);
        let v = Option::<u32>::from_reader_with_ctx(&mut reader, Endian::Little).unwrap();
        assert_eq!(v, Some(0x04030201))
    }

    #[test]
    fn test_option_write() {
        let mut writer = Writer::new(vec![]);
        Some(true).to_writer(&mut writer, ()).unwrap();
        assert_eq!(vec![1], writer.inner);
    }
}