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
	}
}