steel_core/command/execution/

source.rs

use std::{collections::BTreeSet, ptr, sync::Arc};

use glam::DVec3;
use steel_registry::{
    vanilla_game_rules::{
        LOG_ADMIN_COMMANDS, MAX_COMMAND_FORKS, MAX_COMMAND_SEQUENCE_LENGTH, SEND_COMMAND_FEEDBACK,
    },
    world_clock::WorldClockRef,
};
use steel_utils::translations;
use text_components::{Modifier, TextComponent, format::Color};

use crate::{
    command::{
        brigadier::CommandSyntaxError,
        registration::{entity_selector_advanced_permission_expr, entity_selector_permission_expr},
        sender::CommandSender,
    },
    entity::{Entity, EntityAnchor, SharedEntity},
    permission::{
        PermissionContext, PermissionExpr, PermissionKey, PermissionMetadataExpression,
        PermissionRuleExpression, PermissionSet, PermissionState,
    },
    player::{DomainResidenceToken, KnownPlayer, Player},
    scoreboard::Scoreboard,
    server::Server,
    world::World,
};

use super::{CommandExecutionContext, GameProfileArgument};

type CommandResultCallbackFn = dyn Fn(bool, i32) + Send + Sync;

/// A callback invoked after a terminal command returns or fails.
#[derive(Clone, Default)]
pub(crate) struct CommandResultCallback {
    callback: Option<Arc<CommandResultCallbackFn>>,
}

impl CommandResultCallback {
    pub(crate) fn new(callback: impl Fn(bool, i32) + Send + Sync + 'static) -> Self {
        Self {
            callback: Some(Arc::new(callback)),
        }
    }

    pub(crate) const fn empty() -> Self {
        Self { callback: None }
    }

    pub(crate) fn chain(first: Self, second: Self) -> Self {
        match (first.callback, second.callback) {
            (None, None) => Self::empty(),
            (Some(callback), None) | (None, Some(callback)) => Self {
                callback: Some(callback),
            },
            (Some(first), Some(second)) => Self::new(move |success, result| {
                first(success, result);
                second(success, result);
            }),
        }
    }

    pub(crate) fn on_result(&self, success: bool, result: i32) {
        if let Some(callback) = &self.callback {
            callback(success, result);
        }
    }
}

/// Read-only source data exposed to command argument parsers and suggestion providers.
pub(crate) trait CommandArgumentSource: Send + Sync {
    fn default_world_clock(&self) -> Option<WorldClockRef> {
        None
    }

    fn domain_exists(&self, _domain: &str) -> bool {
        false
    }

    fn domain_names(&self) -> Vec<&str> {
        Vec::new()
    }

    fn command_world_names(&self) -> Vec<String> {
        Vec::new()
    }

    fn permission_context_world_names(&self) -> Vec<String> {
        Vec::new()
    }

    fn command_storage_keys(&self) -> Vec<String> {
        Vec::new()
    }

    fn permission_rule_suggestions(&self) -> Vec<String> {
        Vec::new()
    }

    fn permission_metadata_suggestions(&self) -> Vec<String> {
        Vec::new()
    }

    fn user_permission_rule_suggestions(&self, _targets: &GameProfileArgument) -> Vec<String> {
        Vec::new()
    }

    fn user_permission_metadata_suggestions(&self, _targets: &GameProfileArgument) -> Vec<String> {
        Vec::new()
    }

    fn group_permission_rule_suggestions(&self, _group: &str) -> Vec<String> {
        Vec::new()
    }

    fn group_permission_metadata_suggestions(&self, _group: &str) -> Vec<String> {
        Vec::new()
    }

    fn permission_group_names(&self) -> Vec<String> {
        Vec::new()
    }

    fn non_operator_profile_names(&self) -> Vec<String> {
        Vec::new()
    }

    fn all_profile_names(&self) -> Vec<String> {
        Vec::new()
    }

    fn operator_profile_names(&self) -> Vec<String> {
        Vec::new()
    }

    fn selector_player_names(&self) -> Vec<String> {
        Vec::new()
    }

    fn selector_team_names(&self) -> Vec<String> {
        Vec::new()
    }

    fn scoreboard_objective_names(&self) -> Vec<String> {
        Vec::new()
    }

    fn allows_entity_selectors(&self) -> bool {
        false
    }

    fn allows_advanced_entity_selectors(&self) -> bool {
        false
    }
}

/// Source behavior required by the Steel command scheduler.
pub(crate) trait ExecutionCommandSource:
    CommandArgumentSource + Sized + Send + Sync + 'static
{
    /// Returns whether delayed work may still execute for this exact source.
    fn execution_is_current(&self) -> bool {
        true
    }

    fn with_callback(&self, callback: CommandResultCallback) -> Self;

    fn callback(&self) -> CommandResultCallback;

    fn handle_error(&self, error: &CommandSyntaxError, forked: bool);
}

/// Permission lookup required while constructing and traversing Steel command trees.
pub(crate) trait CommandPermissionSource: ExecutionCommandSource {
    fn permission_state(&self, permission: &PermissionExpr) -> Option<PermissionState>;

    fn has_permission(&self, permission: &PermissionExpr) -> bool {
        self.permission_state(permission) == Some(PermissionState::Allow)
    }
}

/// Authorization state captured when command execution starts.
///
/// Vanilla keeps the initiating permission state when `/execute` changes the
/// execution entity, position, or world. Keeping this separate from the
/// mutable execution fields prevents source transforms from changing which
/// contextual Steel permissions apply. Player permissions come from the
/// persistence-published server state so delayed client refreshes cannot extend
/// revoked command access.
#[derive(Clone, Debug, PartialEq, Eq)]
struct CommandAuthorizationContext {
    permission_context: PermissionContext,
    player_permissions: Option<PermissionSet>,
    world_scope: CommandWorldScope,
}

#[derive(Clone, Debug, PartialEq, Eq)]
enum CommandWorldScope {
    Global,
    Domain(Box<str>),
}

impl CommandAuthorizationContext {
    fn for_player(world: steel_utils::Identifier, permissions: PermissionSet) -> Self {
        let domain = world.namespace.to_string().into_boxed_str();
        Self {
            permission_context: PermissionContext::for_world(world),
            player_permissions: Some(permissions),
            world_scope: CommandWorldScope::Domain(domain),
        }
    }

    fn unrestricted(world: steel_utils::Identifier) -> Self {
        Self {
            permission_context: PermissionContext::for_world(world),
            player_permissions: None,
            world_scope: CommandWorldScope::Global,
        }
    }

    const fn permission_context(&self) -> &PermissionContext {
        &self.permission_context
    }

    fn permission_state(&self, permission: &PermissionExpr) -> Option<PermissionState> {
        let Some(permissions) = &self.player_permissions else {
            return Some(PermissionState::Allow);
        };
        permissions.resolve_in(permission, self.permission_context())
    }

    fn allows_execution_domain(&self, domain: &str) -> bool {
        match &self.world_scope {
            CommandWorldScope::Global => true,
            CommandWorldScope::Domain(execution_domain) => execution_domain.as_ref() == domain,
        }
    }
}

/// Immutable Minecraft command execution source.
#[derive(Clone)]
pub(crate) struct CommandSource {
    sender: CommandSender,
    player: Option<Arc<Player>>,
    entity: Option<SharedEntity>,
    world: Arc<World>,
    server: Arc<Server>,
    position: DVec3,
    rotation: (f32, f32),
    anchor: EntityAnchor,
    authorization: CommandAuthorizationContext,
    sender_residence: Option<DomainResidenceToken>,
    effective_player_residence: Option<DomainResidenceToken>,
    callback: CommandResultCallback,
    silent: bool,
}

impl CommandSource {
    pub(crate) fn new(sender: CommandSender, server: Arc<Server>) -> Self {
        let player = sender.get_player().map(Arc::clone);
        let world = player.as_ref().map_or_else(
            || Arc::clone(server.overworld()),
            |player| player.get_world(),
        );
        let entity = player
            .as_ref()
            .map(|player| Arc::clone(player) as SharedEntity);
        let position = entity.as_ref().map_or_else(
            || {
                let level_data = world.level_data.read();
                let spawn = &level_data.data().spawn;
                DVec3::new(f64::from(spawn.x), f64::from(spawn.y), f64::from(spawn.z))
            },
            |entity| entity.position(),
        );
        let rotation = entity
            .as_ref()
            .map_or((0.0, 0.0), |entity| entity.rotation());
        let sender_residence = sender
            .get_player()
            .map(|player| player.domain_residence_token());
        let effective_player_residence = entity
            .as_ref()
            .and_then(|entity| entity.as_player())
            .map(Player::domain_residence_token);
        let authorization = match &sender {
            CommandSender::Player(player) => CommandAuthorizationContext::for_player(
                world.key.clone(),
                server.command_permission_snapshot(player.gameprofile.id),
            ),
            CommandSender::Console | CommandSender::Rcon => {
                CommandAuthorizationContext::unrestricted(world.key.clone())
            }
        };

        Self {
            sender,
            player,
            entity,
            world,
            server,
            position,
            rotation,
            anchor: EntityAnchor::default(),
            authorization,
            sender_residence,
            effective_player_residence,
            callback: CommandResultCallback::empty(),
            silent: false,
        }
    }

    #[expect(
        dead_code,
        reason = "source-aware runtime extensions need access to the original sender"
    )]
    pub(crate) const fn sender(&self) -> &CommandSender {
        &self.sender
    }

    pub(crate) const fn player(&self) -> Option<&Arc<Player>> {
        self.player.as_ref()
    }

    pub(crate) const fn entity(&self) -> Option<&SharedEntity> {
        self.entity.as_ref()
    }

    pub(crate) const fn world(&self) -> &Arc<World> {
        &self.world
    }

    pub(crate) const fn server(&self) -> &Arc<Server> {
        &self.server
    }

    /// Returns whether `/execute` may project this source into `domain`.
    ///
    /// Player-originated command chains remain confined to their initial
    /// domain. Console and Rcon sources are unrestricted.
    pub(crate) fn allows_execution_domain(&self, domain: &str) -> bool {
        self.authorization.allows_execution_domain(domain)
    }

    pub(crate) const fn position(&self) -> DVec3 {
        self.position
    }

    pub(crate) const fn rotation(&self) -> (f32, f32) {
        self.rotation
    }

    pub(crate) const fn anchor(&self) -> EntityAnchor {
        self.anchor
    }

    pub(crate) fn with_entity(&self, entity: SharedEntity) -> Self {
        let mut source = self.clone();
        source.player = entity.as_player().and_then(|entity_player| {
            self.server
                .get_players()
                .into_iter()
                .find(|player| ptr::eq(player.as_ref(), entity_player))
        });
        source.effective_player_residence = entity.as_player().map(Player::domain_residence_token);
        source.entity = Some(entity);
        source
    }

    pub(crate) fn with_world(&self, world: Arc<World>) -> Result<Self, CommandSyntaxError> {
        if !self.allows_execution_domain(world.domain()) {
            return Err(CommandSyntaxError::dynamic(
                "Players cannot execute commands across Steel domains",
            ));
        }
        let mut source = self.clone();
        if self.world.key != world.key {
            let scale =
                self.world.dimension_type.coordinate_scale / world.dimension_type.coordinate_scale;
            source.position.x *= scale;
            source.position.z *= scale;
        }
        source.world = world;
        Ok(source)
    }

    pub(crate) fn with_position(&self, position: DVec3) -> Self {
        let mut source = self.clone();
        source.position = position;
        source
    }

    pub(crate) fn with_rotation(&self, rotation: (f32, f32)) -> Self {
        let mut source = self.clone();
        source.rotation = normalize_rotation(rotation);
        source
    }

    pub(crate) fn with_anchor(&self, anchor: EntityAnchor) -> Self {
        let mut source = self.clone();
        source.anchor = anchor;
        source
    }

    pub(crate) fn facing_position(&self, target: DVec3) -> Self {
        let delta = target - self.anchor_position();
        let horizontal = delta.x.hypot(delta.z);
        let pitch = -delta.y.atan2(horizontal).to_degrees() as f32;
        let yaw = delta.z.atan2(delta.x).to_degrees() as f32 - 90.0;
        self.with_rotation((yaw, pitch))
    }

    pub(crate) fn anchor_position(&self) -> DVec3 {
        if self.anchor == EntityAnchor::Eyes
            && let Some(entity) = &self.entity
        {
            return DVec3::new(
                self.position.x,
                self.position.y + entity.get_eye_height(),
                self.position.z,
            );
        }
        self.position
    }

    #[expect(
        dead_code,
        reason = "silent command-source derivation is retained for future vanilla command paths"
    )]
    pub(crate) fn with_suppressed_output(&self) -> Self {
        let mut source = self.clone();
        source.silent = true;
        source
    }

    #[expect(
        dead_code,
        reason = "custom executors need to inspect silent command-source state"
    )]
    pub(crate) const fn is_silent(&self) -> bool {
        self.silent
    }

    pub(crate) fn send_success(&self, message: &TextComponent, broadcast_to_admins: bool) {
        if self.silent {
            return;
        }

        let accepts_success = self
            .sender
            .get_player()
            .is_none_or(|player| player.get_world().get_game_rule(&SEND_COMMAND_FEEDBACK));
        if accepts_success {
            self.sender.send_message(message);
        }
        if broadcast_to_admins {
            self.broadcast_to_admins(message);
        }
    }

    pub(crate) fn send_failure(&self, message: TextComponent) {
        if !self.silent {
            self.sender.send_message(&message.color(Color::Red));
        }
    }

    fn sequence_limit(&self) -> usize {
        let value = self.world.get_game_rule(&MAX_COMMAND_SEQUENCE_LENGTH);
        value.max(1) as usize
    }

    fn fork_limit(&self) -> usize {
        let value = self.world.get_game_rule(&MAX_COMMAND_FORKS);
        value.max(0) as usize
    }

    fn broadcast_to_admins(&self, message: &TextComponent) {
        let sender_name = admin_broadcast_source_name(self.entity.as_deref(), &self.sender);
        let broadcast = translations::CHAT_TYPE_ADMIN
            .message([sender_name, message.clone()])
            .component()
            .color(Color::Gray)
            .italic(true);

        if self.world.get_game_rule(&SEND_COMMAND_FEEDBACK) {
            let sender_uuid = self.sender.get_player().map(|player| player.gameprofile.id);
            for player in self.server.get_players() {
                if Some(player.gameprofile.id) != sender_uuid
                    && self.server.is_operator(player.gameprofile.id)
                {
                    player.send_message(&broadcast);
                }
            }
        }

        if !matches!(self.sender, CommandSender::Console)
            && self.world.get_game_rule(&LOG_ADMIN_COMMANDS)
        {
            CommandSender::Console.send_message(&broadcast);
        }
    }
}

impl ExecutionCommandSource for CommandSource {
    fn execution_is_current(&self) -> bool {
        let sender_is_current = self.sender.get_player().is_none_or(|player| {
            self.sender_residence.is_some_and(|residence| {
                self.server.command_world_for_player(player).is_some()
                    && player.is_domain_residence_current(residence)
            })
        });
        if !sender_is_current {
            return false;
        }

        self.entity
            .as_ref()
            .and_then(|entity| entity.as_player())
            .is_none_or(|player| {
                self.effective_player_residence.is_some_and(|residence| {
                    self.server.command_world_for_player(player).is_some()
                        && player.is_domain_residence_current(residence)
                })
            })
    }

    fn with_callback(&self, callback: CommandResultCallback) -> Self {
        let mut source = self.clone();
        source.callback = callback;
        source
    }

    fn callback(&self) -> CommandResultCallback {
        self.callback.clone()
    }

    fn handle_error(&self, error: &CommandSyntaxError, forked: bool) {
        if forked || self.silent {
            return;
        }
        self.send_failure(error.message_component());
        if let Some(context) = error.context_component() {
            self.sender.send_message(&context);
        }
    }
}

impl CommandArgumentSource for CommandSource {
    fn default_world_clock(&self) -> Option<WorldClockRef> {
        self.world.dimension_type.default_clock
    }

    fn domain_exists(&self, domain: &str) -> bool {
        self.server.worlds.has_domain(domain)
    }

    fn domain_names(&self) -> Vec<&str> {
        self.server.worlds.domain_names().collect()
    }

    fn command_world_names(&self) -> Vec<String> {
        let domain = self.world.domain();
        let mut names = Vec::new();
        for key in self.server.worlds.keys() {
            names.push(key.to_string());
            if key.namespace.as_ref() == domain {
                names.push(key.path.to_string());
            }
        }
        names
    }

    fn permission_context_world_names(&self) -> Vec<String> {
        self.server.worlds.keys().map(ToString::to_string).collect()
    }

    fn command_storage_keys(&self) -> Vec<String> {
        self.server
            .command_storage
            .get(self.world.domain())
            .map_or_else(Vec::new, |storage| {
                storage
                    .keys()
                    .into_iter()
                    .map(|key| key.to_string())
                    .collect()
            })
    }

    fn permission_rule_suggestions(&self) -> Vec<String> {
        self.server.permission_rule_suggestions()
    }

    fn permission_metadata_suggestions(&self) -> Vec<String> {
        self.server.permission_metadata_suggestions()
    }

    fn user_permission_rule_suggestions(&self, targets: &GameProfileArgument) -> Vec<String> {
        let mut suggestions = BTreeSet::new();
        for uuid in profile_argument_uuids(self, targets) {
            let Some(state) = self.server.player_permission_state(uuid) else {
                continue;
            };
            suggestions.extend(
                state
                    .overrides()
                    .entries()
                    .iter()
                    .filter(|entry| can_manage_permission(self, entry.key()))
                    .map(|entry| {
                        PermissionRuleExpression::new(entry.key().clone(), entry.context().clone())
                            .to_string()
                    }),
            );
        }
        suggestions.into_iter().collect()
    }

    fn user_permission_metadata_suggestions(&self, targets: &GameProfileArgument) -> Vec<String> {
        if !has_permission_key(self, "steel.permission.metadata") {
            return Vec::new();
        }
        let mut suggestions = BTreeSet::new();
        for uuid in profile_argument_uuids(self, targets) {
            let Some(state) = self.server.player_permission_state(uuid) else {
                continue;
            };
            suggestions.extend(state.metadata_overrides().entries().iter().map(|entry| {
                PermissionMetadataExpression::new(entry.key().clone(), entry.context().clone())
                    .to_string()
            }));
        }
        suggestions.into_iter().collect()
    }

    fn group_permission_rule_suggestions(&self, group: &str) -> Vec<String> {
        if !can_manage_group(self, group) {
            return Vec::new();
        }
        self.server
            .permission_groups
            .config_snapshot()
            .groups
            .get(group)
            .into_iter()
            .flat_map(|group| group.allow.iter().chain(&group.deny))
            .filter(|expression| {
                PermissionRuleExpression::parse(expression.as_str())
                    .is_ok_and(|expression| can_manage_permission(self, expression.key()))
            })
            .cloned()
            .collect::<BTreeSet<_>>()
            .into_iter()
            .collect()
    }

    fn group_permission_metadata_suggestions(&self, group: &str) -> Vec<String> {
        if !has_permission_key(self, "steel.permission.metadata") || !can_manage_group(self, group)
        {
            return Vec::new();
        }
        self.server
            .permission_groups
            .config_snapshot()
            .groups
            .get(group)
            .into_iter()
            .flat_map(|group| group.metadata.iter().map(|rule| rule.key.clone()))
            .collect::<BTreeSet<_>>()
            .into_iter()
            .collect()
    }

    fn permission_group_names(&self) -> Vec<String> {
        self.server.permission_groups.group_names()
    }

    fn non_operator_profile_names(&self) -> Vec<String> {
        profile_names(self.server(), false, false)
    }

    fn all_profile_names(&self) -> Vec<String> {
        all_profile_names(self.server())
    }

    fn operator_profile_names(&self) -> Vec<String> {
        profile_names(self.server(), true, true)
    }

    fn selector_player_names(&self) -> Vec<String> {
        let domain = self.world.domain();
        self.server
            .get_players()
            .into_iter()
            .filter(|player| {
                self.server
                    .command_world_for_player(player)
                    .is_some_and(|world| world.domain() == domain)
            })
            .map(|player| player.gameprofile.name.clone())
            .collect()
    }

    fn selector_team_names(&self) -> Vec<String> {
        self.server
            .scoreboards
            .get(self.world.domain())
            .map_or_else(Vec::new, Scoreboard::team_names)
    }

    fn scoreboard_objective_names(&self) -> Vec<String> {
        self.server
            .scoreboards
            .get(self.world.domain())
            .map_or_else(Vec::new, Scoreboard::objective_names)
    }

    fn allows_entity_selectors(&self) -> bool {
        let Ok(permission) = entity_selector_permission_expr() else {
            tracing::error!("built-in entity selector permission key is invalid");
            return false;
        };
        CommandPermissionSource::has_permission(self, &permission)
    }

    fn allows_advanced_entity_selectors(&self) -> bool {
        let Ok(permission) = entity_selector_advanced_permission_expr() else {
            tracing::error!("built-in advanced entity selector permission key is invalid");
            return false;
        };
        CommandPermissionSource::has_permission(self, &permission)
    }
}

fn profile_argument_uuids(
    source: &CommandSource,
    argument: &GameProfileArgument,
) -> BTreeSet<uuid::Uuid> {
    match argument {
        GameProfileArgument::Selector(selector) => {
            selector.find_online_profile_players(source).map_or_else(
                |_| BTreeSet::new(),
                |players| {
                    players
                        .into_iter()
                        .map(|player| player.gameprofile.id)
                        .collect()
                },
            )
        }
        GameProfileArgument::Direct(value) => {
            let known = source.server.known_players();
            let uuid = uuid::Uuid::parse_str(value)
                .ok()
                .or_else(|| known.by_name(value).map(KnownPlayer::uuid))
                .or_else(|| {
                    source
                        .server
                        .get_players()
                        .into_iter()
                        .find(|player| player.gameprofile.name.eq_ignore_ascii_case(value))
                        .map(|player| player.gameprofile.id)
                });
            uuid.into_iter().collect()
        }
    }
}

fn can_manage_permission(source: &CommandSource, permission: &PermissionKey) -> bool {
    has_permission_key(
        source,
        &format!("steel.permission.manage.{}", permission.as_str()),
    )
}

fn can_manage_group(source: &CommandSource, group: &str) -> bool {
    has_permission_key(source, &format!("steel.permission.group.{group}"))
}

fn has_permission_key(source: &CommandSource, value: &str) -> bool {
    PermissionKey::parse(value)
        .is_ok_and(|key| CommandPermissionSource::has_permission(source, &PermissionExpr::key(key)))
}

fn profile_names(server: &Server, operator: bool, include_known: bool) -> Vec<String> {
    let players = server.get_players();
    let mut names = Vec::new();
    for player in &players {
        if operator == server.is_operator(player.gameprofile.id) {
            names.push(player.gameprofile.name.clone());
        }
    }

    if !include_known {
        return names;
    }

    for known in server.known_players().entries() {
        if names
            .iter()
            .any(|name| name.eq_ignore_ascii_case(known.last_known_name()))
            || players.iter().any(|player| {
                player
                    .gameprofile
                    .name
                    .eq_ignore_ascii_case(known.last_known_name())
            })
        {
            continue;
        }
        let is_operator = server.is_operator(known.uuid());
        if operator == is_operator {
            names.push(known.last_known_name().to_owned());
        }
    }
    names
}

fn all_profile_names(server: &Server) -> Vec<String> {
    let players = server.get_players();
    let mut names = players
        .iter()
        .map(|player| player.gameprofile.name.clone())
        .collect::<Vec<_>>();
    for known in server.known_players().entries() {
        if names
            .iter()
            .any(|name| name.eq_ignore_ascii_case(known.last_known_name()))
        {
            continue;
        }
        names.push(known.last_known_name().to_owned());
    }
    names
}

impl CommandPermissionSource for CommandSource {
    fn permission_state(&self, permission: &PermissionExpr) -> Option<PermissionState> {
        self.authorization.permission_state(permission)
    }
}

impl CommandExecutionContext<CommandSource> {
    pub(crate) fn for_source(source: &CommandSource) -> Self {
        Self::new(source.sequence_limit(), source.fork_limit())
    }
}

fn admin_broadcast_source_name(
    entity: Option<&dyn Entity>,
    sender: &CommandSender,
) -> TextComponent {
    entity.map_or_else(
        || TextComponent::plain(sender.to_string()),
        |entity| TextComponent::plain(entity.plain_text_name()),
    )
}

fn normalize_rotation((mut yaw, mut pitch): (f32, f32)) -> (f32, f32) {
    yaw = yaw.rem_euclid(360.0);
    if yaw >= 180.0 {
        yaw -= 360.0;
    }
    pitch = pitch.rem_euclid(360.0);
    if pitch >= 180.0 {
        pitch -= 360.0;
    }
    (yaw, pitch)
}

#[cfg(test)]
mod tests {
    use std::sync::Weak;

    use glam::DVec3;
    use steel_registry::{entity_type::EntityTypeRef, vanilla_entities};
    use steel_utils::Identifier;
    use text_components::TextComponent;

    use crate::command::sender::CommandSender;
    use crate::entity::{Entity, EntityBase};
    use crate::permission::{
        PermissionContext, PermissionEntry, PermissionExpr, PermissionKey, PermissionSet,
        PermissionState,
    };

    use super::{CommandAuthorizationContext, admin_broadcast_source_name, normalize_rotation};

    struct NamedTestEntity {
        base: EntityBase,
    }

    crate::entity::impl_test_downcast_type!(NamedTestEntity);

    impl Entity for NamedTestEntity {
        fn base(&self) -> &EntityBase {
            &self.base
        }

        fn entity_type(&self) -> EntityTypeRef {
            &vanilla_entities::ITEM
        }

        fn plain_text_name(&self) -> String {
            "Executor".to_owned()
        }
    }

    fn permission_key(value: &str) -> PermissionKey {
        match PermissionKey::parse(value) {
            Ok(key) => key,
            Err(error) => panic!("test permission key should parse: {error}"),
        }
    }

    #[test]
    fn rotation_normalization_matches_command_source_stack() {
        assert_eq!(normalize_rotation((540.0, -540.0)), (-180.0, -180.0));
        assert_eq!(normalize_rotation((-181.0, 181.0)), (179.0, -179.0));
    }

    #[test]
    fn admin_broadcast_uses_current_execution_entity_name() {
        let entity = NamedTestEntity {
            base: EntityBase::new(
                1,
                DVec3::ZERO,
                vanilla_entities::ITEM.dimensions,
                Weak::new(),
            ),
        };

        assert_eq!(
            admin_broadcast_source_name(Some(&entity), &CommandSender::Console),
            TextComponent::plain("Executor")
        );
        assert_eq!(
            admin_broadcast_source_name(None, &CommandSender::Console),
            TextComponent::plain("Server")
        );
    }

    #[test]
    fn authorization_context_captures_initial_world_scope() {
        let world = Identifier::new("lobby", "spawn");
        let authorization =
            CommandAuthorizationContext::for_player(world.clone(), PermissionSet::new());

        assert_eq!(
            authorization.permission_context(),
            &PermissionContext::for_world(world)
        );
        assert!(authorization.allows_execution_domain("lobby"));
        assert!(!authorization.allows_execution_domain("survival"));
    }

    #[test]
    fn authorization_context_uses_published_snapshot_instead_of_stale_player_permissions() {
        let world = Identifier::new("lobby", "spawn");
        let permission = permission_key("minecraft.command.stop");
        let stale_player_permissions =
            PermissionSet::from_entries([PermissionEntry::allow(permission.clone())]);
        assert!(stale_player_permissions.allows_key(&permission));

        let authorization = CommandAuthorizationContext::for_player(world, PermissionSet::new());

        assert_eq!(
            authorization.permission_state(&PermissionExpr::key(permission)),
            None
        );
    }

    #[test]
    fn unrestricted_authorization_allows_console_permissions() {
        let authorization =
            CommandAuthorizationContext::unrestricted(Identifier::new("default", "overworld"));
        let permission = PermissionExpr::key(permission_key("minecraft.command.stop"));

        assert_eq!(
            authorization.permission_state(&permission),
            Some(PermissionState::Allow)
        );
        assert!(authorization.allows_execution_domain("default"));
        assert!(authorization.allows_execution_domain("survival"));
    }
}