azalea_core/
difficulty.rs

1use std::{
2    fmt::{self, Debug},
3    io::{self, Cursor, Write},
4};
5
6use azalea_buf::{AzBuf, BufReadError};
7
8#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
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 AzBuf for Difficulty {
65    fn azalea_read(buf: &mut Cursor<&[u8]>) -> Result<Self, BufReadError> {
66        Ok(Difficulty::by_id(u8::azalea_read(buf)?))
67    }
68    fn azalea_write(&self, buf: &mut impl Write) -> io::Result<()> {
69        u8::azalea_write(&self.id(), buf)
70    }
71}
72
73#[cfg(test)]
74mod tests {
75    use super::*;
76
77    #[test]
78    fn test_difficulty_from_name() {
79        assert_eq!(
80            Difficulty::Peaceful,
81            Difficulty::from_name("peaceful").unwrap()
82        );
83        assert_eq!(Difficulty::Easy, Difficulty::from_name("easy").unwrap());
84        assert_eq!(Difficulty::Normal, Difficulty::from_name("normal").unwrap());
85        assert_eq!(Difficulty::Hard, Difficulty::from_name("hard").unwrap());
86        assert!(Difficulty::from_name("invalid").is_err());
87    }
88
89    #[test]
90    fn test_difficulty_id() {
91        assert_eq!(0, Difficulty::Peaceful.id());
92        assert_eq!(1, Difficulty::Easy.id());
93        assert_eq!(2, Difficulty::Normal.id());
94        assert_eq!(3, Difficulty::Hard.id());
95    }
96
97    #[test]
98    fn test_difficulty_name() {
99        assert_eq!("peaceful", Difficulty::Peaceful.name());
100        assert_eq!("easy", Difficulty::Easy.name());
101        assert_eq!("normal", Difficulty::Normal.name());
102        assert_eq!("hard", Difficulty::Hard.name());
103    }
104}
azalea_core/difficulty.rs

azalea_core/
difficulty.rs
