Decouple gun from player

src/projectile.rs

@@ -1,12 +1,6 @@
 use bevy::prelude::*;
 
-use crate::{
+use crate::{GameState, collision::Collided, damagable::Damagable, ships::Factions};
-    GameState,
-    collision::{Collided, Collider},
-    damagable::Damagable,
-    ships::Factions,
-    velocity::Velocity,
-};
 
 #[derive(Component)]
 pub struct Projectile {
@@ -16,7 +10,10 @@ pub struct Projectile {
 }
 
 #[derive(Resource)]
-pub struct ProjectileSprite(Handle<Image>);
+pub struct ProjectileSprite {
+    pub player_projectile: Handle<Image>,
+    pub enemy_projectile: Handle<Image>,
+}
 
 pub struct ProjectilePlugin;
 
@@ -32,16 +29,20 @@ impl Projectile {
 
 impl Plugin for ProjectilePlugin {
     fn build(&self, app: &mut App) {
-        app.add_systems(OnEnter(GameState::Game), projectile_startup)
+        app.add_systems(Startup, projectile_startup)
             .add_systems(
                 FixedPreUpdate,
                 projectile_despawn_countdown.run_if(in_state(GameState::Game)),
-            );
+            )
+            .add_observer(observe_collision);
     }
 }
 
 pub fn projectile_startup(mut commands: Commands, asset_server: Res<AssetServer>) {
-    commands.insert_resource(ProjectileSprite(asset_server.load("projectile.png")));
+    commands.insert_resource(ProjectileSprite {
+        player_projectile: asset_server.load("projectile.png"),
+        enemy_projectile: asset_server.load("enemy_projectile.png"),
+    });
 }
 
 pub fn projectile_despawn_countdown(
@@ -57,24 +58,6 @@ pub fn projectile_despawn_countdown(
     }
 }
 
-pub fn spawn_projectile(
-    commands: &mut Commands,
-    sprite: Res<ProjectileSprite>,
-    transform: Transform,
-    damage: u32,
-    faction: Factions,
-) {
-    commands
-        .spawn((
-            Projectile::new(damage, faction),
-            Sprite::from(sprite.0.clone()),
-            Collider::new(8.),
-            transform,
-            Velocity::moving(1024.0, 0.0),
-        ))
-        .observe(observe_collision);
-}
-
 pub fn observe_collision(
     collision: On<Collided>,
     mut commands: Commands,

src/ships/gun.rs

@@ -0,0 +1,59 @@
+use bevy::prelude::*;
+use rand::random_range;
+
+use crate::{
+    collision::Collider,
+    projectile::{Projectile, ProjectileSprite},
+    ships::Factions,
+    velocity::Velocity,
+};
+
+#[derive(Component)]
+pub struct Gun {
+    pub shoot: bool,
+    shoot_timer: Timer,
+    damage: u32,
+    spread_radius: f32,
+}
+
+impl Gun {
+    pub fn new(damage: u32, spread: f32, firerate: f32) -> Self {
+        Self {
+            shoot: false,
+            shoot_timer: Timer::from_seconds(firerate, TimerMode::Once),
+            damage,
+            spread_radius: spread,
+        }
+    }
+}
+
+pub fn gun_shooting_system(
+    mut commands: Commands,
+    time: Res<Time>,
+    projectile_sprite: Res<ProjectileSprite>,
+    query: Query<(&mut Gun, &Transform, &Factions)>,
+) {
+    for (mut gun, transform, faction) in query {
+        gun.shoot_timer.tick(time.delta());
+
+        if gun.shoot_timer.is_finished() == false || gun.shoot == false {
+            return;
+        }
+
+        let mut bullet_transform = transform.clone();
+        bullet_transform.rotate_z(random_range(-gun.spread_radius..gun.spread_radius));
+
+        commands.spawn((
+            Projectile::new(gun.damage, faction.clone()),
+            bullet_transform,
+            Velocity::moving(1024., 0.),
+            Collider::new(8.),
+            Sprite::from(match faction {
+                Factions::EnemyFaction => projectile_sprite.enemy_projectile.clone(),
+                Factions::PlayerFaction => projectile_sprite.player_projectile.clone(),
+            }),
+        ));
+
+        gun.shoot_timer.reset();
+    }
+}

src/ships/mod.rs

@@ -3,9 +3,10 @@ use bevy::prelude::*;
 use crate::GameState;
 
 pub mod enemy;
+pub mod gun;
 pub mod player;
 
-#[derive(Component)]
+#[derive(Component, Copy, Clone)]
 pub enum Factions {
     PlayerFaction,
     EnemyFaction,
@@ -36,7 +37,8 @@ impl Plugin for ShipsPlugin {
                 FixedUpdate,
                 (
                     player::player_movement_system.run_if(in_state(GameState::Game)),
-                    player::player_shooting_system,
+                    player::player_shooting_system.run_if(in_state(GameState::Game)),
+                    gun::gun_shooting_system.run_if(in_state(GameState::Game)),
                 ),
             );
     }

src/ships/player.rs

@@ -1,8 +1,8 @@
 use crate::{
     collision::Collider,
     damagable::Damagable,
-    projectile::{ProjectileSprite, spawn_projectile},
+    projectile::ProjectileSprite,
-    ships::Factions,
+    ships::{Factions, gun::Gun},
     velocity::Velocity,
 };
 use bevy::prelude::*;
@@ -16,12 +16,6 @@ pub struct PlayerMovement {
     stop_epsilon: f32,
 }
 
-#[derive(Component)]
-pub struct PlayerGun {
-    shoot_timer: Timer,
-    damage: u32,
-}
-
 pub fn spawn_player(commands: &mut Commands, sprite: Handle<Image>, at: Vec2) {
     commands.spawn((
         PlayerMovement {
@@ -29,10 +23,7 @@ pub fn spawn_player(commands: &mut Commands, sprite: Handle<Image>, at: Vec2) {
             deceleration: 300.,
             stop_epsilon: 50.,
         },
-        PlayerGun {
+        Gun::new(1, f32::to_radians(15.), 0.2),
-            shoot_timer: Timer::from_seconds(0.2, TimerMode::Once),
-            damage: 1,
-        },
         Collider::new(8.),
         Sprite::from(sprite),
         Transform::from_xyz(at.x, at.y, 0.),
@@ -100,30 +91,9 @@ pub(super) fn player_movement_system(
 }
 
 pub fn player_shooting_system(
-    mut commands: Commands,
-    time: Res<Time>,
     keyboard_input: Res<ButtonInput<KeyCode>>,
-    projectile_sprite: Res<ProjectileSprite>,
+    query: Single<&mut Gun, With<PlayerMovement>>,
-    query: Single<(&mut PlayerGun, &Transform)>,
 ) {
-    let (mut gun, transform) = query.into_inner();
+    let mut gun = query.into_inner();
-    gun.shoot_timer.tick(time.delta());
+    gun.shoot = keyboard_input.pressed(KeyCode::Space);
-
-    if gun.shoot_timer.is_finished() == false || keyboard_input.pressed(KeyCode::Space) == false {
-        return;
-    }
-
-    let five_degrees = f32::to_radians(5.);
-    let mut bullet_transform = transform.clone();
-    bullet_transform.rotate_z(random_range(-five_degrees..five_degrees));
-
-    spawn_projectile(
-        &mut commands,
-        projectile_sprite,
-        bullet_transform,
-        gun.damage,
-        Factions::PlayerFaction,
-    );
-
-    gun.shoot_timer.reset();
 }