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

use crate::api::{EndpointArgumentApi, ErrorApi};
use crate::err_msg;
use crate::{ArgDecodeInput, DynArgInput};

pub struct EndpointDynArgLoader<AA>
where
	AA: EndpointArgumentApi + 'static,
{
	api: AA,
	current_index: i32,
	num_arguments: i32,
}

impl<AA> EndpointDynArgLoader<AA>
where
	AA: EndpointArgumentApi + 'static,
{
	pub fn new(api: AA) -> Self {
		let num_arguments = api.get_num_arguments();
		EndpointDynArgLoader {
			api,
			current_index: 0,
			num_arguments,
		}
	}
}

impl<AA> ErrorApi for EndpointDynArgLoader<AA>
where
	AA: EndpointArgumentApi + ErrorApi + 'static,
{
	#[inline]
	fn signal_error(&self, message: &[u8]) -> ! {
		self.api.signal_error(message)
	}
}

impl<AA> DynArgInput<ArgDecodeInput<AA>> for EndpointDynArgLoader<AA>
where
	AA: EndpointArgumentApi + Clone + 'static,
{
	fn has_next(&self) -> bool {
		self.current_index < self.num_arguments
	}

	fn next_arg_input(&mut self) -> ArgDecodeInput<AA> {
		if self.current_index >= self.num_arguments {
			self.signal_error(err_msg::ARG_WRONG_NUMBER)
		} else {
			let arg_input = ArgDecodeInput::new(self.api.clone(), self.current_index);
			self.current_index += 1;
			arg_input
		}
	}
}