use crate::errors::ValidationError;
use crate::traits::ValueObject;
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub enum WeightUnit {
Mg,
G,
Kg,
T,
Oz,
Lb,
}
impl std::fmt::Display for WeightUnit {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
WeightUnit::Mg => write!(f, "mg"),
WeightUnit::G => write!(f, "g"),
WeightUnit::Kg => write!(f, "kg"),
WeightUnit::T => write!(f, "t"),
WeightUnit::Oz => write!(f, "oz"),
WeightUnit::Lb => write!(f, "lb"),
}
}
}
#[derive(Debug, Clone, PartialEq)]
pub struct WeightInput {
pub value: f64,
pub unit: WeightUnit,
}
#[derive(Debug, Clone, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct Weight {
value: f64,
unit: WeightUnit,
#[cfg_attr(feature = "serde", serde(skip))]
canonical: String,
}
impl ValueObject for Weight {
type Input = WeightInput;
type Output = str;
type Error = ValidationError;
fn new(input: Self::Input) -> Result<Self, Self::Error> {
if !input.value.is_finite() || input.value < 0.0 {
return Err(ValidationError::invalid("Weight", &input.value.to_string()));
}
let canonical = format!("{} {}", input.value, input.unit);
Ok(Self {
value: input.value,
unit: input.unit,
canonical,
})
}
fn value(&self) -> &Self::Output {
&self.canonical
}
fn into_inner(self) -> Self::Input {
WeightInput {
value: self.value,
unit: self.unit,
}
}
}
impl Weight {
pub fn amount(&self) -> f64 {
self.value
}
pub fn unit(&self) -> &WeightUnit {
&self.unit
}
}
impl std::fmt::Display for Weight {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", self.canonical)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn accepts_valid() {
let w = Weight::new(WeightInput {
value: 75.0,
unit: WeightUnit::Kg,
})
.unwrap();
assert_eq!(w.value(), "75 kg");
assert_eq!(w.amount(), 75.0);
}
#[test]
fn accepts_zero() {
assert!(
Weight::new(WeightInput {
value: 0.0,
unit: WeightUnit::G
})
.is_ok()
);
}
#[test]
fn rejects_negative() {
assert!(
Weight::new(WeightInput {
value: -1.0,
unit: WeightUnit::Kg
})
.is_err()
);
}
#[test]
fn rejects_nan() {
assert!(
Weight::new(WeightInput {
value: f64::NAN,
unit: WeightUnit::Kg
})
.is_err()
);
}
}