Move game logic plugin configuration to game-specific plugin

src/game/mod.rs

@@ -1,5 +1,13 @@
 mod objects;
+mod plugin;
 mod rng;
-pub mod runtime;
-pub mod seed;
-pub mod setup;
+mod runtime;
+mod seed;
+mod setup;
+pub mod state;
+
+#[allow(unused_imports)]
+pub mod prelude {
+    pub use super::plugin::{DataSource, GamePlugin};
+    pub use super::seed::Seed;
+}

src/game/objects.rs

@@ -1,7 +1,7 @@
 use avian2d::prelude::*;
 use bevy::prelude::*;
 
-use crate::AppState;
+use super::state::AppState;
 
 /// Basic implementation of a physics object
 #[derive(Component, Default)]

src/game/plugin.rs

@@ -0,0 +1,91 @@
+use std::net::SocketAddr;
+
+use avian2d::PhysicsPlugins;
+use bevy::{input::common_conditions::input_pressed, prelude::*};
+
+use crate::net::NetIOPlugin;
+
+use super::{
+    runtime::{move_camera, move_player, quit, zoom_camera},
+    seed::Seed,
+    setup::{check_for_seed, setup_balls, setup_from_seed, setup_player, setup_ui, setup_walls},
+    state::AppState,
+};
+
+#[derive(Debug, Clone, Copy)]
+pub enum DataSource {
+    Address(SocketAddr),
+    Seed(Seed),
+    None,
+}
+
+impl DataSource {
+    pub fn try_from_options(address: Option<SocketAddr>, seed: Option<Seed>) -> Option<Self> {
+        match (address, seed) {
+            (None, None) => Some(DataSource::None),
+            (None, Some(seed)) => Some(DataSource::Seed(seed)),
+            (Some(address), None) => Some(DataSource::Address(address)),
+            (Some(_), Some(_)) => None,
+        }
+    }
+
+    fn try_to_address(&self) -> Option<SocketAddr> {
+        match self {
+            DataSource::Address(address) => Some(*address),
+            _ => None,
+        }
+    }
+}
+
+pub struct GamePlugin {
+    source: DataSource,
+    port: u16,
+}
+
+impl GamePlugin {
+    pub fn new(source: DataSource, port: u16) -> Self {
+        Self { source, port }
+    }
+}
+
+impl Plugin for GamePlugin {
+    fn build(&self, app: &mut App) {
+        app.add_plugins((
+            NetIOPlugin::new(self.port, self.source.try_to_address()),
+            PhysicsPlugins::default().with_length_unit(50.0),
+        ))
+        .init_state::<AppState>()
+        .add_systems(Startup, setup_ui)
+        .add_systems(
+            OnEnter(AppState::InGame),
+            (
+                setup_from_seed,
+                (setup_player, setup_balls, setup_walls).after(setup_from_seed),
+            ),
+        )
+        .add_systems(
+            FixedUpdate,
+            check_for_seed.run_if(in_state(AppState::Loading)),
+        )
+        .add_systems(
+            Update,
+            (
+                ((move_player, move_camera).chain(), zoom_camera)
+                    .run_if(in_state(AppState::InGame)),
+                quit.run_if(input_pressed(KeyCode::KeyQ)),
+            ),
+        );
+
+        match self.source {
+            DataSource::Address(peer) => {
+                info!("Will retrieve seed from peer => {peer}");
+            }
+            DataSource::Seed(seed) => {
+                app.insert_resource(seed);
+            }
+            DataSource::None => {
+                app.insert_resource(Seed::random());
+            }
+        };
+    }
+}

src/game/seed.rs

@@ -7,7 +7,7 @@ use bevy::prelude::*;
 use rand::random;
 
 /// Value with which to initialize the PRNG
-#[derive(Resource, Clone, Copy)]
+#[derive(Resource, Debug, Clone, Copy)]
 pub struct Seed(u64);
 
 impl Seed {

src/game/setup.rs

@@ -11,7 +11,7 @@ use bevy::{
 use rand::{Rng as _, SeedableRng};
 use wyrand::WyRand;
 
-use crate::AppState;
+use super::state::AppState;
 
 use super::{
     objects::{Ball, Player, Radius, Wall},

src/game/state.rs

@@ -0,0 +1,9 @@
+use bevy::prelude::*;
+
+#[derive(States, Default, Debug, Clone, PartialEq, Eq, Hash)]
+#[states(scoped_entities)]
+pub enum AppState {
+    #[default]
+    Loading,
+    InGame,
+}