Implement FromStr for GameMode

This may be useful for downstream applications and it fits into the
pattern of implementing it for Boss/Encounter.

1 files changed, 48 insertions, 0 deletions

diff --git a/src/gamedata.rs b/src/gamedata.rs
index a07ac05..bbe6e74 100644
--- a/src/gamedata.rs
+++ b/src/gamedata.rs
@@ -26,6 +26,28 @@ pub enum GameMode {
     WvW,
 }
 
+/// Error for when converting a string to a game mode fails.
+#[derive(Debug, Clone, Hash, PartialEq, Eq, Error)]
+#[error("Invalid encounter identifier: {0}")]
+pub struct ParseGameModeError(String);
+
+impl FromStr for GameMode {
+    type Err = ParseGameModeError;
+
+    fn from_str(s: &str) -> Result<Self, Self::Err> {
+        let lower = s.to_lowercase();
+        match &lower as &str {
+            "raid" => Ok(GameMode::Raid),
+            "fractal" => Ok(GameMode::Fractal),
+            "strike" => Ok(GameMode::Strike),
+            "golem" => Ok(GameMode::Golem),
+            "wvw" => Ok(GameMode::WvW),
+
+            _ => Err(ParseGameModeError(s.to_owned())),
+        }
+    }
+}
+
 /// Enum containing all encounters with their IDs.
 ///
 /// An encounter is a fight or event for which a log can exist. An encounter consists of no, one or
@@ -869,6 +891,32 @@ mod tests {
     use super::*;
 
     #[test]
+    fn test_gamemode_parsing_ok() {
+        use GameMode::*;
+        let tests: &[(&'static str, GameMode)] = &[
+            ("raid", Raid),
+            ("Raid", Raid),
+            ("fractal", Fractal),
+            ("Fractal", Fractal),
+            ("strike", Strike),
+            ("Strike", Strike),
+            ("golem", Golem),
+            ("Golem", Golem),
+            ("wvw", WvW),
+            ("WvW", WvW),
+        ];
+
+        for (input, expected) in tests {
+            assert_eq!(
+                input.parse(),
+                Ok(*expected),
+                "parsing input {:?} failed",
+                input
+            );
+        }
+    }
+
+    #[test]
     fn test_encounter_parsing_ok() {
         use Encounter::*;
         let tests: &[(&'static str, Encounter)] = &[