chore+feat(home-assistant): update to pyo3 0.27 and update extraction errors, switch out SmolStr for Arc<str>, tighten up light service calls and implement some for notify, start implementing units of measurement like for power

home-assistant/src/state/mod.rs

@@ -0,0 +1,64 @@
+use std::{convert::Infallible, str::FromStr};
+
+use pyo3::{exceptions::PyException, prelude::*};
+use snafu::{ResultExt, Snafu};
+
+pub mod error_state;
+pub mod unexpected_state;
+
+pub use error_state::ErrorState;
+pub use unexpected_state::UnexpectedState;
+
+#[derive(Debug, Clone, derive_more::Display)]
+pub enum HomeAssistantState<State> {
+    Ok(State),
+    Err(ErrorState),
+    UnexpectedErr(UnexpectedState),
+}
+
+impl<State: FromStr> FromStr for HomeAssistantState<State> {
+    type Err = Infallible;
+
+    fn from_str(s: &str) -> Result<Self, <Self as FromStr>::Err> {
+        if let Ok(ok) = State::from_str(s) {
+            return Ok(HomeAssistantState::Ok(ok));
+        }
+
+        if let Ok(error) = ErrorState::from_str(s) {
+            return Ok(HomeAssistantState::Err(error));
+        }
+
+        Ok(HomeAssistantState::UnexpectedErr(UnexpectedState(s.into())))
+    }
+}
+
+#[derive(Debug, Snafu)]
+pub enum ExtractHomeAssistantStateError {
+    /// couldn't extract the object as a string
+    ExtractStringError { source: PyErr },
+}
+
+impl From<ExtractHomeAssistantStateError> for PyErr {
+    fn from(error: ExtractHomeAssistantStateError) -> Self {
+        match &error {
+            ExtractHomeAssistantStateError::ExtractStringError { .. } => {
+                PyException::new_err(error.to_string())
+            }
+        }
+    }
+}
+
+// TODO: replace with a derive(PyFromStr) (analogous to serde_with::DeserializeFromStr) once I make one
+impl<'a, 'py, State: FromStr + FromPyObject<'a, 'py>> FromPyObject<'a, 'py>
+    for HomeAssistantState<State>
+{
+    type Error = ExtractHomeAssistantStateError;
+
+    fn extract(ob: Borrowed<'a, 'py, PyAny>) -> Result<Self, Self::Error> {
+        let s = ob.extract::<&str>().context(ExtractStringSnafu)?;
+
+        let Ok(state) = s.parse();
+
+        Ok(state)
+    }
+}