Add InputEvent::direction, left_direction and right_direction

Author: fwcd

lighthouse-protocol/src/input/input_event.rs

@@ -1,6 +1,8 @@
 use serde::{Deserialize, Serialize};
 
-use super::{EventSource, GamepadEvent, KeyEvent, MouseEvent};
+use crate::Direction;
+
+use super::{EventSource, GamepadControlEvent, GamepadEvent, KeyEvent, MouseEvent};
 
 /// A user input event, as generated by the new frontend (LUNA).
 #[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
@@ -20,6 +22,38 @@ impl InputEvent {
             InputEvent::Gamepad(GamepadEvent { source, .. }) => source,
         }
     }
+
+    /// Parses the input event as an arbitrary direction.
+    pub fn direction(&self) -> Option<Direction> {
+        self.left_direction().or_else(|| self.right_direction())
+    }
+
+    /// The direction if the input event represents a WASD key, D-pad or left stick.
+    /// Commonly used e.g. for movement in games.
+    pub fn left_direction(&self) -> Option<Direction> {
+        match self {
+            InputEvent::Key(key) => key.wasd_direction(),
+            InputEvent::Gamepad(gamepad) => match &gamepad.control {
+                GamepadControlEvent::Button(button) => button.d_pad_direction(),
+                GamepadControlEvent::Axis2D(axis2d) if axis2d.index == 0 => axis2d.direction(),
+                _ => None,
+            },
+            _ => None,
+        }
+    }
+
+    /// The direction if the input event represents an arrow key or right stick.
+    /// Commonly used e.g. for camera control in games.
+    pub fn right_direction(&self) -> Option<Direction> {
+        match self {
+            InputEvent::Key(key) => key.arrow_direction(),
+            InputEvent::Gamepad(gamepad) => match &gamepad.control {
+                GamepadControlEvent::Axis2D(axis2d) if axis2d.index == 1 => axis2d.direction(),
+                _ => None,
+            },
+            _ => None,
+        }
+    }
 }
 
 #[cfg(test)]