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

use crate::*;
use crate::{api::ErrorApi, types::BoxedBytes};

pub struct BytesArgLoader<'a, SE>
where
    SE: ErrorApi,
{
    bytes: &'a [BoxedBytes],
    signal_error: SE,
}

impl<'a, SE> BytesArgLoader<'a, SE>
where
    SE: ErrorApi,
{
    pub fn new(bytes: &'a [BoxedBytes], signal_error: SE) -> Self {
        BytesArgLoader {
            bytes,
            signal_error,
        }
    }
}

impl<'a, SE> ErrorApi for BytesArgLoader<'a, SE>
where
    SE: ErrorApi,
{
    #[inline]
    fn signal_error(&self, message: &[u8]) -> ! {
        self.signal_error.signal_error(message)
    }
}

impl<'a, SE> DynArgInput<&'a [u8]> for BytesArgLoader<'a, SE>
where
    SE: ErrorApi,
{
    #[inline]
    fn has_next(&self) -> bool {
        !self.bytes.is_empty()
    }

    fn next_arg_input(&mut self) -> &'a [u8] {
        if self.bytes.is_empty() {
            self.signal_error(err_msg::ARG_WRONG_NUMBER);
        }
        let result = self.bytes[0].as_slice();
        self.bytes = &self.bytes[1..];
        result
    }
}