azalea_core/
difficulty.rs

1use std::{
2    fmt::{self, Debug},
3    io::{self, Cursor, Write},
4};
5
6use azalea_buf::{AzaleaRead, AzaleaWrite, BufReadError};
7
8#[derive(Hash, Clone, Copy, Debug, PartialEq, Eq)]
9pub enum Difficulty {
10    Peaceful = 0,
11    Easy = 1,
12    Normal = 2,
13    Hard = 3,
14}
15
16pub struct InvalidDifficultyError(pub String);
17
18impl Debug for InvalidDifficultyError {
19    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
20        write!(f, "Invalid difficulty: {}", self.0)
21    }
22}
23
24impl Difficulty {
25    pub fn name(&self) -> &'static str {
26        match self {
27            Difficulty::Peaceful => "peaceful",
28            Difficulty::Easy => "easy",
29            Difficulty::Normal => "normal",
30            Difficulty::Hard => "hard",
31        }
32    }
33
34    pub fn from_name(name: &str) -> Result<Difficulty, InvalidDifficultyError> {
35        Ok(match name {
36            "peaceful" => Difficulty::Peaceful,
37            "easy" => Difficulty::Easy,
38            "normal" => Difficulty::Normal,
39            "hard" => Difficulty::Hard,
40            _ => return Err(InvalidDifficultyError(name.to_owned())),
41        })
42    }
43
44    pub fn by_id(id: u8) -> Difficulty {
45        match id % 4 {
46            0 => Difficulty::Peaceful,
47            1 => Difficulty::Easy,
48            2 => Difficulty::Normal,
49            3 => Difficulty::Hard,
50            _ => unreachable!(),
51        }
52    }
53
54    pub fn id(&self) -> u8 {
55        match self {
56            Difficulty::Peaceful => 0,
57            Difficulty::Easy => 1,
58            Difficulty::Normal => 2,
59            Difficulty::Hard => 3,
60        }
61    }
62}
63
64impl AzaleaRead for Difficulty {
65    fn azalea_read(buf: &mut Cursor<&[u8]>) -> Result<Self, BufReadError> {
66        Ok(Difficulty::by_id(u8::azalea_read(buf)?))
67    }
68}
69
70impl AzaleaWrite for Difficulty {
71    fn azalea_write(&self, buf: &mut impl Write) -> io::Result<()> {
72        u8::azalea_write(&self.id(), buf)
73    }
74}
75
76#[cfg(test)]
77mod tests {
78    use super::*;
79
80    #[test]
81    fn test_difficulty_from_name() {
82        assert_eq!(
83            Difficulty::Peaceful,
84            Difficulty::from_name("peaceful").unwrap()
85        );
86        assert_eq!(Difficulty::Easy, Difficulty::from_name("easy").unwrap());
87        assert_eq!(Difficulty::Normal, Difficulty::from_name("normal").unwrap());
88        assert_eq!(Difficulty::Hard, Difficulty::from_name("hard").unwrap());
89        assert!(Difficulty::from_name("invalid").is_err());
90    }
91
92    #[test]
93    fn test_difficulty_id() {
94        assert_eq!(0, Difficulty::Peaceful.id());
95        assert_eq!(1, Difficulty::Easy.id());
96        assert_eq!(2, Difficulty::Normal.id());
97        assert_eq!(3, Difficulty::Hard.id());
98    }
99
100    #[test]
101    fn test_difficulty_name() {
102        assert_eq!("peaceful", Difficulty::Peaceful.name());
103        assert_eq!("easy", Difficulty::Easy.name());
104        assert_eq!("normal", Difficulty::Normal.name());
105        assert_eq!("hard", Difficulty::Hard.name());
106    }
107}
azalea_core/difficulty.rs

azalea_core/
difficulty.rs
