Move game logic plugin configuration to game-specific plugin

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());
+            }
+        };
+    }
+}