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 PowerUnit {
W,
KW,
MW,
Hp,
}
impl std::fmt::Display for PowerUnit {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
PowerUnit::W => write!(f, "W"),
PowerUnit::KW => write!(f, "kW"),
PowerUnit::MW => write!(f, "MW"),
PowerUnit::Hp => write!(f, "hp"),
}
}
}
#[derive(Debug, Clone, PartialEq)]
pub struct PowerInput {
pub value: f64,
pub unit: PowerUnit,
}
#[derive(Debug, Clone, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct Power {
value: f64,
unit: PowerUnit,
#[cfg_attr(feature = "serde", serde(skip))]
canonical: String,
}
impl ValueObject for Power {
type Input = PowerInput;
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("Power", &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 {
PowerInput {
value: self.value,
unit: self.unit,
}
}
}
impl Power {
pub fn amount(&self) -> f64 {
self.value
}
pub fn unit(&self) -> &PowerUnit {
&self.unit
}
}
impl std::fmt::Display for Power {
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 p = Power::new(PowerInput {
value: 3.7,
unit: PowerUnit::KW,
})
.unwrap();
assert_eq!(p.value(), "3.7 kW");
}
#[test]
fn accepts_zero() {
assert!(
Power::new(PowerInput {
value: 0.0,
unit: PowerUnit::W
})
.is_ok()
);
}
#[test]
fn rejects_negative() {
assert!(
Power::new(PowerInput {
value: -1.0,
unit: PowerUnit::W
})
.is_err()
);
}
#[test]
fn rejects_nan() {
assert!(
Power::new(PowerInput {
value: f64::NAN,
unit: PowerUnit::W
})
.is_err()
);
}
}