diff --git a/src/main.rs b/src/main.rs index d30ee8e..07ac9f4 100644 --- a/src/main.rs +++ b/src/main.rs @@ -22,9 +22,23 @@ fn spawn_snake(commands: &mut Commands, materials: Res) { .with(SnakeHead); } -fn snake_movement(mut head_positions: Query<(&SnakeHead, &mut Transform)>) { - for (_head, mut transform) in head_positions.iter_mut() { - transform.translation.y += 2.; +fn snake_movement( + keyboard_input: Res>, + mut head_positions: Query<&mut Transform, With>, +) { + 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.; + } } }