2 files changed, 22 insertions, 0 deletions

diff --git a/src/constants.rs b/src/constants.rs
index 2b1a8bf..30ba4f1 100644
--- a/src/constants.rs
+++ b/src/constants.rs
@@ -12,6 +12,10 @@ pub const HA_NUMBER_MODE_AUTO: &str = "auto";
 pub const HA_NUMBER_MODE_BOX: &str = "box";
 pub const HA_NUMBER_MODE_SLIDER: &str = "slider";
 
+pub const HA_STATE_CLASS_MEASUREMENT: &str = "measurement";
+pub const HA_STATE_CLASS_TOTAL: &str = "total";
+pub const HA_STATE_CLASS_TOTAL_INCREASING: &str = "total_increasing";
+
 pub const HA_DEVICE_CLASS_SENSOR_APPARENT_POWER: &str = "apparent_power";
 pub const HA_DEVICE_CLASS_SENSOR_AQI: &str = "aqi";
 pub const HA_DEVICE_CLASS_SENSOR_ATMOSPHERIC_PRESSURE: &str = "atmospheric_pressure";
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,