Move snake with keys

merwan · Jul 8, 2024 · d3e9967 · d3e9967
1 parent 993ad5f
commit d3e9967
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -33,7 +33,21 @@ fn spawn_snake(mut commands: Commands) {
         .insert(SnakeHead);
 }
 
-fn snake_movement(mut query: Query<&mut Transform, With<SnakeHead>>) {
+fn snake_movement(
+    keyboard_input: Res<ButtonInput<KeyCode>>,
+    mut query: Query<&mut Transform, With<SnakeHead>>,
+) {
     let mut transform = query.single_mut();
-    transform.translation.y += 2.;
+    if keyboard_input.pressed(KeyCode::ArrowLeft) {
+        transform.translation.x -= 2.;
+    }
+    if keyboard_input.pressed(KeyCode::ArrowRight) {
+        transform.translation.x += 2.;
+    }
+    if keyboard_input.pressed(KeyCode::ArrowDown) {
+        transform.translation.y -= 2.;
+    }
+    if keyboard_input.pressed(KeyCode::ArrowUp) {
+        transform.translation.y += 2.;
+    }
 }