auto_look.rs

use azalea::{
    app::{App, Plugin},
    bot::LookAtEvent,
    ecs::prelude::*,
    entity::{Position, dimensions::EntityDimensions, metadata::Player},
    nearest_entity::EntityFinder,
    pathfinder::Pathfinder,
    physics::PhysicsSystems,
    prelude::*,
};

use crate::prelude::{GameTicks, LocalSettings};

/// Automatically look at the closest player in range
pub struct AutoLookPlugin;

impl Plugin for AutoLookPlugin {
    fn build(&self, app: &mut App) {
        app.add_systems(GameTick, Self::handle_auto_look.before(PhysicsSystems));
    }
}

impl AutoLookPlugin {
    pub fn handle_auto_look(
        mut query: Query<(Entity, &Pathfinder, &GameTicks, &LocalSettings)>,
        entities: EntityFinder<With<Player>>,
        targets: Query<(&Position, Option<&EntityDimensions>)>,
        mut look_at_events: MessageWriter<LookAtEvent>,
    ) {
        for (entity, pathfinder, game_ticks, local_settings) in &mut query {
            if !local_settings.auto_look.enabled {
                continue;
            }

            if let Some(_goal) = &pathfinder.goal {
                continue;
            }

            if game_ticks.0 % local_settings.auto_look.delay_ticks != 0 {
                continue;
            }

            let Some(target) = entities.nearest_to_entity(entity, f64::MAX) else {
                continue;
            };

            let Ok((target_pos, target_entity_dimensions)) = targets.get(target) else {
                continue;
            };

            let mut position = **target_pos;
            if let Some(entity_dimensions) = target_entity_dimensions {
                position.y += f64::from(entity_dimensions.eye_height);
            }

            look_at_events.write(LookAtEvent { entity, position });
        }
    }
}