Controlling the snake

src/main.rs

@@ -22,9 +22,23 @@ fn spawn_snake(commands: &mut Commands, materials: Res<Materials>) {
         .with(SnakeHead);
 }
 
-fn snake_movement(mut head_positions: Query<(&SnakeHead, &mut Transform)>) {
+fn snake_movement(
-    for (_head, mut transform) in head_positions.iter_mut() {
+    keyboard_input: Res<Input<KeyCode>>,
-        transform.translation.y += 2.;
+    mut head_positions: Query<&mut Transform, With<SnakeHead>>,
+) {
+    for mut transform in head_positions.iter_mut() {
+        if keyboard_input.pressed(KeyCode::Left) {
+            transform.translation.x -= 2.;
+        }
+        if keyboard_input.pressed(KeyCode::Right) {
+            transform.translation.x += 2.;
+        }
+        if keyboard_input.pressed(KeyCode::Down) {
+            transform.translation.y -= 2.;
+        }
+        if keyboard_input.pressed(KeyCode::Up) {
+            transform.translation.y += 2.;
+        }
     }
 }