crates/border-wars/src/lib.rs

@@ -2,11 +2,11 @@
 
 use bevy::prelude::*;
 
-pub mod menu;
+pub mod scenes;
 
-/// The state of the game.
+/// The current scene of the game.
 #[derive(Debug, Clone, Copy, Default, Eq, PartialEq, Hash, States)]
-pub enum GameState {
+pub enum CurrentScene {
     /// When we are in the main menu.
     #[default]
     Menu,

crates/border-wars/src/main.rs

@@ -1,13 +1,11 @@
 //! The main entry point of the game.
 
 use bevy::prelude::*;
-use border_wars::menu::MenuPlugin;
+use border_wars::scenes::ScenesPlugin;
-use border_wars::GameState;
 
 fn main() {
     App::new()
         .add_plugins(DefaultPlugins)
-        .add_state::<GameState>()
+        .add_plugins(ScenesPlugin)
-        .add_plugins(MenuPlugin)
         .run();
 }

crates/border-wars/src/scenes/menu.rs

@@ -1,26 +1,23 @@
 //! The main menu of the game.
 
 use bevy::prelude::*;
-use bevy_egui::{egui, EguiContexts, EguiPlugin};
+use bevy_egui::{egui, EguiContexts};
 
-use crate::GameState;
+use crate::CurrentScene;
 
 /// The plugin for the menu.
 pub struct MenuPlugin;
 
 impl Plugin for MenuPlugin {
     fn build(&self, app: &mut App) {
-        app.add_plugins(EguiPlugin).add_systems(
+        app.add_systems(Update, menu_ui.run_if(in_state(CurrentScene::Menu)));
-            Update,
-            menu_ui.run_if(state_exists_and_equals(GameState::Menu)),
-        );
     }
 }
 /// Display the UI of the menu to host a game or join one.
 fn menu_ui(
     mut ctx: EguiContexts,
     mut connection_string: Local<String>,
-    mut next_state: ResMut<NextState<GameState>>,
+    mut next_scene: ResMut<NextState<CurrentScene>>,
 ) {
     egui::CentralPanel::default().show(ctx.ctx_mut(), |ui| {
         ui.heading("Border Wars");
@@ -33,7 +30,7 @@ fn menu_ui(
             ui.text_edit_singleline(&mut *connection_string);
 
             if ui.button("Join").clicked() {
-                next_state.set(GameState::Game);
+                next_scene.set(CurrentScene::Game);
                 // TODO: connect to the game
             }
         });
@@ -41,7 +38,7 @@ fn menu_ui(
         ui.separator();
 
         if ui.button("Create new game").clicked() {
-            next_state.set(GameState::Lobby);
+            next_scene.set(CurrentScene::Lobby);
             // TODO: create a new game
         }
     });

crates/border-wars/src/scenes/mod.rs

@@ -0,0 +1,19 @@
+//! The file containing all scenes programs.
+
+use bevy::prelude::*;
+use bevy_egui::EguiPlugin;
+
+use crate::CurrentScene;
+
+pub mod menu;
+
+/// The plugin for all scenes.
+pub struct ScenesPlugin;
+
+impl Plugin for ScenesPlugin {
+    fn build(&self, app: &mut App) {
+        app.add_plugins(EguiPlugin)
+            .add_state::<CurrentScene>()
+            .add_plugins(menu::MenuPlugin);
+    }
+}