added state class constants/enum

1 files changed, 18 insertions, 0 deletions

diff --git a/src/entity_sensor.rs b/src/entity_sensor.rs
index d70e80e..4b84c66 100644
--- a/src/entity_sensor.rs
+++ b/src/entity_sensor.rs
@@ -2,6 +2,24 @@ use crate::{
     Entity, EntityCommonConfig, EntityConfig, NumericSensorState, TemperatureUnit, constants,
 };
 
+#[derive(Debug, Default, Clone, Copy, PartialEq, Eq)]
+pub enum StateClass {
+    #[default]
+    Measurement,
+    Total,
+    TotalIncreasing,
+}
+
+impl StateClass {
+    pub fn as_str(&self) -> &'static str {
+        match self {
+            StateClass::Measurement => constants::HA_STATE_CLASS_MEASUREMENT,
+            StateClass::Total => constants::HA_STATE_CLASS_TOTAL,
+            StateClass::TotalIncreasing => constants::HA_STATE_CLASS_TOTAL_INCREASING,
+        }
+    }
+}
+
 #[derive(Debug, Default)]
 pub struct TemperatureSensorConfig {
     pub common: EntityCommonConfig,