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

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

            
9
pub struct Add;
10

            
11
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
12
pub struct AddRequest {
13
    uri: Uri,
14
    position: Option<SongPosition>,
15
}
16

            
17
impl AddRequest {
18
    pub fn new(uri: Uri, position: Option<SongPosition>) -> Self {
19
        Self { uri, position }
20
    }
21
}
22

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

            
28
    fn serialize(&self) -> String {
29
        match self.position {
30
            Some(position) => format!("{} {} {}\n", Self::COMMAND, self.uri, position),
31
            None => format!("{} {}\n", Self::COMMAND, self.uri),
32
        }
33
    }
34

            
35
    fn parse(mut parts: RequestTokenizer<'_>) -> Result<Self, RequestParserError> {
36
        let uri = match parts.next() {
37
            Some(s) => s,
38
            None => return Err(Self::missing_arguments_error(0)),
39
        };
40

            
41
        let position = match parts.next() {
42
            Some(s) => Some(s.parse::<SongPosition>().map_err(|_| {
43
                RequestParserError::SubtypeParserError {
44
                    argument_index: 1,
45
                    expected_type: "SongPosition",
46
                    raw_input: s.to_owned(),
47
                }
48
            })?),
49
            None => None,
50
        };
51

            
52
        Self::throw_if_too_many_arguments(parts)?;
53

            
54
        Ok(AddRequest {
55
            uri: uri.to_string(),
56
            position,
57
        })
58
    }
59
}
60

            
61
empty_command_response!(Add);
62

            
63
impl Command for Add {
64
    type Request = AddRequest;
65
    type Response = AddResponse;
66
}