Get current state as option, no more panicking at the end of a machine

cbosoft · Aug 5, 2024 · 7d6d385 · 7d6d385
1 parent d441dd0
commit 7d6d385
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/state_machine.rs b/src/state_machine.rs
@@ -42,8 +42,11 @@ impl StateMachine {
         }
     }
 
-    pub fn current_state(&self) -> String {
-        self.current_state.as_ref().unwrap().name().clone()
+    pub fn current_state(&self) -> Option<String> {
+        match self.current_state.as_ref() {
+            Some(s) => Some(s.name().clone()),
+            None => None
+        }
     }
 
     pub fn run<I: Serialize, O: DeserializeOwned>(&mut self, i: I) -> SML_Result<Option<O>> {