refactor volume module

1 files changed, 27 insertions, 0 deletions

diff --git a/src/volume/unit.rs b/src/volume/unit.rs
new file mode 100644
index 0000000..1b542d3
--- /dev/null
+++ b/src/volume/unit.rs
@@ -0,0 +1,27 @@
+use std::fmt::{self, Display, Formatter};
+
+#[derive(Copy, Clone, Debug, Default, Eq, PartialEq)]
+pub enum Unit {
+    CubicMetre,
+    #[default]
+    Litre,
+    CubicCentimetre,
+}
+
+impl Unit {
+    pub const ALL: [Self; 3] = [Self::CubicMetre, Self::Litre, Self::CubicCentimetre];
+}
+
+impl Display for Unit {
+    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
+        write!(
+            f,
+            "{}",
+            match self {
+                Self::CubicMetre => "m³",
+                Self::Litre => "L",
+                Self::CubicCentimetre => "cc",
+            }
+        )
+    }
+}