1
use serde::{Deserialize, Serialize};
2

            
3
use crate::{
4
    commands::{Command, CommandRequest, RequestParserError, empty_command_response},
5
    request_tokenizer::RequestTokenizer,
6
    types::{Priority, WindowRange},
7
};
8

            
9
pub struct Prio;
10

            
11
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
12
pub struct PrioRequest {
13
    pub prio: Priority,
14
    pub window: WindowRange,
15
}
16

            
17
impl PrioRequest {
18
    pub fn new(prio: Priority, window: WindowRange) -> Self {
19
        Self { prio, window }
20
    }
21
}
22

            
23
impl CommandRequest for PrioRequest {
24
    const COMMAND: &'static str = "prio";
25
    const MIN_ARGS: u32 = 2;
26
    const MAX_ARGS: Option<u32> = Some(2);
27

            
28
    fn into_request_enum(self) -> crate::Request {
29
        crate::Request::Prio(self.prio, self.window)
30
    }
31

            
32
    fn from_request_enum(request: crate::Request) -> Option<Self> {
33
        match request {
34
            crate::Request::Prio(prio, window) => Some(PrioRequest { prio, window }),
35
            _ => None,
36
        }
37
    }
38

            
39
    fn serialize(&self) -> String {
40
        format!("{} {} {}\n", Self::COMMAND, self.prio, self.window)
41
    }
42

            
43
    fn parse(mut parts: RequestTokenizer<'_>) -> Result<Self, RequestParserError> {
44
        let prio = parts.next().ok_or(Self::missing_arguments_error(0))?;
45
        let prio = prio
46
            .parse()
47
            .map_err(|_| RequestParserError::SubtypeParserError {
48
                argument_index: 0,
49
                expected_type: "Priority",
50
                raw_input: prio.to_string(),
51
            })?;
52

            
53
        let window = parts.next().ok_or(Self::missing_arguments_error(1))?;
54
        let window = window
55
            .parse()
56
            .map_err(|_| RequestParserError::SubtypeParserError {
57
                argument_index: 1,
58
                expected_type: "WindowRange",
59
                raw_input: window.to_string(),
60
            })?;
61

            
62
        Self::throw_if_too_many_arguments(parts)?;
63

            
64
        Ok(PrioRequest { prio, window })
65
    }
66
}
67

            
68
empty_command_response!(Prio);
69

            
70
impl Command for Prio {
71
    type Request = PrioRequest;
72
    type Response = PrioResponse;
73
}