Target UI multiplayer (#95)

crates/hedz_reloaded/src/aim/mod.rs

@@ -1,13 +1,15 @@
-mod marker;
-mod target_ui;
-
 use crate::{
-    GameState, control::Inputs, head::ActiveHead, heads_database::HeadsDatabase,
-    hitpoints::Hitpoints, physics_layers::GameLayer, player::Player, tb_entities::EnemySpawn,
+    GameState,
+    control::Inputs,
+    head::ActiveHead,
+    heads_database::HeadsDatabase,
+    hitpoints::Hitpoints,
+    physics_layers::GameLayer,
+    player::{LocalPlayer, Player},
+    tb_entities::EnemySpawn,
 };
 use avian3d::prelude::*;
 use bevy::prelude::*;
-use marker::MarkerEvent;
 use serde::{Deserialize, Serialize};
 use std::f32::consts::PI;
 
@@ -21,7 +23,12 @@ pub struct AimTarget(pub Option<Entity>);
 pub struct AimState {
     pub range: f32,
     pub max_angle: f32,
-    pub spawn_marker: bool,
+}
+
+#[derive(Event)]
+pub enum MarkerEvent {
+    Spawn(Entity),
+    Despawn,
 }
 
 impl Default for AimState {
@@ -29,7 +36,6 @@ impl Default for AimState {
         Self {
             range: 80.,
             max_angle: PI / 8.,
-            spawn_marker: true,
         }
     }
 }
@@ -38,20 +44,12 @@ pub fn plugin(app: &mut App) {
     app.register_type::<AimState>();
     app.register_type::<AimTarget>();
 
-    app.add_plugins(target_ui::plugin);
-    app.add_plugins(marker::plugin);
+    app.register_required_components::<ActiveHead, AimState>();
 
     app.add_systems(
         Update,
         (update_player_aim, update_npc_aim, head_change).run_if(in_state(GameState::Playing)),
     );
-    app.add_systems(Update, add_aim);
-}
-
-fn add_aim(mut commands: Commands, query: Query<Entity, Added<ActiveHead>>) {
-    for e in query.iter() {
-        commands.entity(e).insert(AimState::default());
-    }
 }
 
 fn head_change(
@@ -70,12 +68,19 @@ fn update_player_aim(
     mut commands: Commands,
     potential_targets: Query<(Entity, &Transform), With<Hitpoints>>,
     mut player_aim: Query<
-        (Entity, &AimState, &mut AimTarget, &GlobalTransform, &Inputs),
+        (
+            Entity,
+            &AimState,
+            &mut AimTarget,
+            &GlobalTransform,
+            &Inputs,
+            Has<LocalPlayer>,
+        ),
         With<Player>,
     >,
     spatial_query: SpatialQuery,
 ) {
-    for (player, state, mut aim_target, global_tf, inputs) in player_aim.iter_mut() {
+    for (player, state, mut aim_target, global_tf, inputs, is_local) in player_aim.iter_mut() {
         let (player_pos, player_forward) = (global_tf.translation(), inputs.look_dir);
 
         let mut new_target = None;
@@ -114,13 +119,14 @@ fn update_player_aim(
         }
 
         if new_target != aim_target.0 {
-            if state.spawn_marker {
+            if is_local {
                 if let Some(target) = new_target {
                     commands.trigger(MarkerEvent::Spawn(target));
                 } else {
                     commands.trigger(MarkerEvent::Despawn);
                 }
             }
+
             aim_target.0 = new_target;
         }
     }