azalea_core/
difficulty.rsuse std::{
fmt::{Debug, Error, Formatter},
io::{Cursor, Write},
};
use azalea_buf::{BufReadError, McBufReadable, McBufWritable};
#[derive(Hash, Clone, Copy, Debug, PartialEq, Eq)]
pub enum Difficulty {
PEACEFUL = 0,
EASY = 1,
NORMAL = 2,
HARD = 3,
}
pub enum Err {
InvalidDifficulty(String),
}
impl Debug for Err {
fn fmt(&self, f: &mut Formatter) -> Result<(), Error> {
match self {
Err::InvalidDifficulty(s) => write!(f, "Invalid difficulty: {s}"),
}
}
}
impl Difficulty {
pub fn name(&self) -> &'static str {
match self {
Difficulty::PEACEFUL => "peaceful",
Difficulty::EASY => "easy",
Difficulty::NORMAL => "normal",
Difficulty::HARD => "hard",
}
}
pub fn from_name(name: &str) -> Result<Difficulty, Err> {
match name {
"peaceful" => Ok(Difficulty::PEACEFUL),
"easy" => Ok(Difficulty::EASY),
"normal" => Ok(Difficulty::NORMAL),
"hard" => Ok(Difficulty::HARD),
_ => Err(Err::InvalidDifficulty(name.to_string())),
}
}
pub fn by_id(id: u8) -> Difficulty {
match id % 4 {
0 => Difficulty::PEACEFUL,
1 => Difficulty::EASY,
2 => Difficulty::NORMAL,
3 => Difficulty::HARD,
_ => panic!("Unknown difficulty id: {id}"),
}
}
pub fn id(&self) -> u8 {
match self {
Difficulty::PEACEFUL => 0,
Difficulty::EASY => 1,
Difficulty::NORMAL => 2,
Difficulty::HARD => 3,
}
}
}
impl McBufReadable for Difficulty {
fn read_from(buf: &mut Cursor<&[u8]>) -> Result<Self, BufReadError> {
Ok(Difficulty::by_id(u8::read_from(buf)?))
}
}
impl McBufWritable for Difficulty {
fn write_into(&self, buf: &mut impl Write) -> Result<(), std::io::Error> {
u8::write_into(&self.id(), buf)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_difficulty_from_name() {
assert_eq!(
Difficulty::PEACEFUL,
Difficulty::from_name("peaceful").unwrap()
);
assert_eq!(Difficulty::EASY, Difficulty::from_name("easy").unwrap());
assert_eq!(Difficulty::NORMAL, Difficulty::from_name("normal").unwrap());
assert_eq!(Difficulty::HARD, Difficulty::from_name("hard").unwrap());
assert!(Difficulty::from_name("invalid").is_err());
}
#[test]
fn test_difficulty_id() {
assert_eq!(0, Difficulty::PEACEFUL.id());
assert_eq!(1, Difficulty::EASY.id());
assert_eq!(2, Difficulty::NORMAL.id());
assert_eq!(3, Difficulty::HARD.id());
}
#[test]
fn test_difficulty_name() {
assert_eq!("peaceful", Difficulty::PEACEFUL.name());
assert_eq!("easy", Difficulty::EASY.name());
assert_eq!("normal", Difficulty::NORMAL.name());
assert_eq!("hard", Difficulty::HARD.name());
}
}