Cursor grab addition

bytestring-net · Jun 30, 2024 · 9d5f2af · 9d5f2af
1 parent c601cec
commit 9d5f2af
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/crates/bevy_lunex/src/logic/cursor.rs b/crates/bevy_lunex/src/logic/cursor.rs
@@ -1,5 +1,5 @@
 use crate::*;
-use bevy::{utils::HashMap, window::PrimaryWindow};
+use bevy::{utils::HashMap, window::{CursorGrabMode, PrimaryWindow}};
 
 
 // #===================#
@@ -15,6 +15,8 @@ pub struct Cursor2d {
     cursor_atlas_map: HashMap<CursorIcon, (usize, Vec2)>,
     /// A toggle if this cursor should replace the native cursor
     native_cursor: bool,
+    /// If the cursor is allowed to leave window
+    pub confined: bool,
     /// A toggle if the cursor should be hidden
     pub hidden: bool,
 }
@@ -26,9 +28,15 @@ impl Cursor2d {
             cursor_request_priority: 0.0,
             cursor_atlas_map: HashMap::new(),
             native_cursor: true,
+            confined: false,
             hidden: false,
         }
     }
+    /// If the cursor is allowed to leave window
+    pub fn confined(mut self, confined: bool) -> Self {
+        self.confined = confined;
+        self
+    }
     /// A toggle if this cursor should be native
     pub fn native_cursor(mut self, enable: bool) -> Self {
         self.native_cursor = enable;
@@ -60,6 +68,12 @@ fn cursor_update(
             window.cursor.visible = if cursor.native_cursor { !cursor.hidden } else { false };
             if window.cursor.visible { window.cursor.icon = cursor.cursor_request; }
 
+            if cursor.confined {
+                window.cursor.grab_mode = CursorGrabMode::Locked;
+            } else {
+                window.cursor.grab_mode = CursorGrabMode::None;
+            }
+
             match window.cursor_position() {
                 Some(position) => {