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 AreaUnit {
Mm2,
Cm2,
M2,
Km2,
In2,
Ft2,
Ha,
}
#[cfg(feature = "serde")]
impl From<Area> for String {
fn from(v: Area) -> String {
v.canonical
}
}
impl TryFrom<String> for Area {
type Error = ValidationError;
fn try_from(s: String) -> Result<Self, Self::Error> {
Self::try_from(s.as_str())
}
}
impl std::fmt::Display for AreaUnit {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
AreaUnit::Mm2 => write!(f, "mm²"),
AreaUnit::Cm2 => write!(f, "cm²"),
AreaUnit::M2 => write!(f, "m²"),
AreaUnit::Km2 => write!(f, "km²"),
AreaUnit::In2 => write!(f, "in²"),
AreaUnit::Ft2 => write!(f, "ft²"),
AreaUnit::Ha => write!(f, "ha"),
}
}
}
#[derive(Debug, Clone, PartialEq)]
pub struct AreaInput {
pub value: f64,
pub unit: AreaUnit,
}
#[derive(Debug, Clone, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "serde", serde(try_from = "String", into = "String"))]
pub struct Area {
value: f64,
unit: AreaUnit,
canonical: String,
}
impl ValueObject for Area {
type Input = AreaInput;
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("Area", &input.value.to_string()));
}
let canonical = format!("{} {}", input.value, input.unit);
Ok(Self {
value: input.value,
unit: input.unit,
canonical,
})
}
fn into_inner(self) -> Self::Input {
AreaInput {
value: self.value,
unit: self.unit,
}
}
}
impl Area {
pub fn value(&self) -> &str {
&self.canonical
}
pub fn amount(&self) -> f64 {
self.value
}
pub fn unit(&self) -> &AreaUnit {
&self.unit
}
}
impl TryFrom<&str> for Area {
type Error = ValidationError;
fn try_from(value: &str) -> Result<Self, Self::Error> {
let err = || ValidationError::invalid("Area", value);
let (val_str, unit_str) = value.trim().split_once(' ').ok_or_else(err)?;
let val: f64 = val_str.trim().parse().map_err(|_| err())?;
let unit = match unit_str.trim() {
"mm²" => AreaUnit::Mm2,
"cm²" => AreaUnit::Cm2,
"m²" => AreaUnit::M2,
"km²" => AreaUnit::Km2,
"in²" => AreaUnit::In2,
"ft²" => AreaUnit::Ft2,
"ha" => AreaUnit::Ha,
_ => return Err(err()),
};
Self::new(AreaInput { value: val, unit })
}
}
impl std::fmt::Display for Area {
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 a = Area::new(AreaInput {
value: 50.0,
unit: AreaUnit::M2,
})
.unwrap();
assert_eq!(a.value(), "50 m²");
}
#[test]
fn accepts_zero() {
assert!(
Area::new(AreaInput {
value: 0.0,
unit: AreaUnit::M2
})
.is_ok()
);
}
#[test]
fn rejects_negative() {
assert!(
Area::new(AreaInput {
value: -1.0,
unit: AreaUnit::M2
})
.is_err()
);
}
#[test]
fn rejects_nan() {
assert!(
Area::new(AreaInput {
value: f64::NAN,
unit: AreaUnit::M2
})
.is_err()
);
}
#[test]
fn try_from_parses_valid() {
let a = Area::try_from("1.5 m²").unwrap();
assert_eq!(a.value(), "1.5 m²");
}
#[test]
fn try_from_rejects_no_space() {
assert!(Area::try_from("1.5").is_err());
}
#[test]
fn try_from_rejects_unknown_unit() {
assert!(Area::try_from("1.5 acres").is_err());
}
#[cfg(feature = "serde")]
#[test]
fn serde_roundtrip() {
let v = Area::try_from("1.5 m²").unwrap();
let json = serde_json::to_string(&v).unwrap();
let back: Area = serde_json::from_str(&json).unwrap();
assert_eq!(v.value(), back.value());
}
#[cfg(feature = "serde")]
#[test]
fn serde_serializes_as_canonical_string() {
let v = Area::try_from("1.5 m²").unwrap();
let json = serde_json::to_string(&v).unwrap();
assert!(json.contains("1.5"));
}
}