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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

use core::marker::PhantomData;
use crate::core::{ enc, dec };


pub struct RawValue<'de>(&'de [u8]);

struct RawValueReader<'r, 'de, R>
where R: dec::Read<'de>
{
    reader: &'r mut R,
    readn: usize,
    _phantom: PhantomData<&'de [u8]>
}

impl<'r, 'de, R> RawValueReader<'r, 'de, R>
where R: dec::Read<'de>
{
    #[inline]
    fn new(reader: &'r mut R) -> RawValueReader<'r, 'de, R> {
        RawValueReader {
            reader,
            readn: 0,
            _phantom: PhantomData
        }
    }
}

impl<'r, 'de, R> dec::Read<'de> for RawValueReader<'r, 'de, R>
where R: dec::Read<'de>
{
    type Error = R::Error;

    #[inline]
    fn fill<'short>(&'short mut self, want: usize) -> Result<dec::Reference<'de, 'short>, Self::Error> {
        let want = match self.readn.checked_add(want) {
            Some(n) => n,
            None => return Ok(dec::Reference::Long(&[]))
        };

        let buf = match self.reader.fill(want)? {
            dec::Reference::Long(buf)
                if buf.len() >= self.readn => dec::Reference::Long(&buf[self.readn..]),
            dec::Reference::Long(_) => dec::Reference::Long(&[]),
            dec::Reference::Short(buf) => dec::Reference::Short(buf)
        };

        Ok(buf)
    }

    #[inline]
    fn advance(&mut self, n: usize) {
        self.readn += n;
    }

    #[inline]
    fn step_in(&mut self) -> bool {
        self.reader.step_in()
    }

    #[inline]
    fn step_out(&mut self) {
        self.reader.step_out()
    }
}

impl<'de> dec::Decode<'de> for RawValue<'de> {
    #[inline]
    fn decode<R: dec::Read<'de>>(reader: &mut R) -> Result<Self, dec::Error<R::Error>> {
        let name = &"raw-value";

        let mut reader = RawValueReader::new(reader);
        let _ignore = dec::IgnoredAny::decode(&mut reader)?;

        let buf = match reader.reader.fill(reader.readn).map_err(dec::Error::Read)? {
            dec::Reference::Long(buf)
                if buf.len() >= reader.readn => &buf[..reader.readn],
            dec::Reference::Long(buf) => return Err(dec::Error::require_length(name, Some(buf.len()))),
            dec::Reference::Short(_) => return Err(dec::Error::require_borrowed(name))
        };

        reader.reader.advance(reader.readn);

        Ok(RawValue(buf))
    }
}

impl<'de> enc::Encode for RawValue<'de> {
    #[inline]
    fn encode<W: enc::Write>(&self, writer: &mut W) -> Result<(), enc::Error<W::Error>> {
        writer.push(self.0).map_err(enc::Error::Write)
    }
}

#[test]
#[cfg(feature = "use_std")]
fn test_raw_value() {
    use crate::core::enc::Encode;
    use crate::core::dec::Decode;
    use crate::core::utils::{ BufWriter, SliceReader };
    use crate::core::types;

    let buf = {
        let mut buf = BufWriter::new(Vec::new());

        types::Map(&[
            ("bar", types::Map(&[
                ("value", 0x99u32)
            ][..]))
        ][..]).encode(&mut buf).unwrap();

        buf
    };

    let mut reader = SliceReader::new(buf.buffer());
    let map = <types::Map<Vec<(&str, RawValue<'_>)>>>::decode(&mut reader).unwrap();

    assert_eq!(map.0.len(), 1);
    assert_eq!(map.0[0].0, "bar");

    let bar_raw_value = &map.0[0].1;

    let buf2 = {
        let mut buf = BufWriter::new(Vec::new());

        types::Map(&[
            ("bar", bar_raw_value)
        ][..]).encode(&mut buf).unwrap();

        buf
    };

    assert_eq!(buf.buffer(), buf2.buffer());

    type Bar<'a> = types::Map<Vec<(&'a str, u32)>>;

    let mut reader = SliceReader::new(buf2.buffer());
    let map2 = <types::Map<Vec<(&str, Bar)>>>::decode(&mut reader).unwrap();

    assert_eq!(map2.0.len(), 1);
    assert_eq!(map2.0[0].0, "bar");

    let bar = &map2.0[0].1;

    assert_eq!(bar.0.len(), 1);
    assert_eq!(bar.0[0].0, "value");
    assert_eq!(bar.0[0].1, 0x99);
}