ErrorNoWatcher/src/lua/client/movement.rs

use super::{Client, Direction, Vec3};
use azalea::{
    BlockPos, BotClientExt, SprintDirection, WalkDirection,
    interact::HitResultComponent,
    pathfinder::{
        ExecutingPath, GotoEvent, Pathfinder, PathfinderClientExt,
        goals::{BlockPosGoal, Goal, RadiusGoal, ReachBlockPosGoal, XZGoal, YGoal},
    },
};
use mlua::{FromLua, Lua, Result, Table, UserDataRef, Value};

pub fn direction(_lua: &Lua, client: &Client) -> Result<Direction> {
    let d = client.direction();
    Ok(Direction { x: d.0, y: d.1 })
}

pub fn eye_position(_lua: &Lua, client: &Client) -> Result<Vec3> {
    let p = client.eye_position();
    Ok(Vec3 {
        x: p.x,
        y: p.y,
        z: p.z,
    })
}

pub async fn goto(
    lua: Lua,
    client: UserDataRef<Client>,
    (data, metadata): (Value, Option<Table>),
) -> Result<()> {
    fn g(client: &Client, without_mining: bool, goal: impl Goal + Send + Sync + 'static) {
        if without_mining {
            client.goto_without_mining(goal);
        } else {
            client.goto(goal);
        }
    }

    let error = mlua::Error::FromLuaConversionError {
        from: data.type_name(),
        to: "Table".to_string(),
        message: None,
    };
    let (goal_type, without_mining) = metadata
        .map(|t| {
            (
                t.get("type").unwrap_or_default(),
                t.get("without_mining").unwrap_or_default(),
            )
        })
        .unwrap_or_default();

    #[allow(clippy::cast_possible_truncation)]
    match goal_type {
        1 => {
            let t = data.as_table().ok_or(error)?;
            let p = Vec3::from_lua(t.get("position")?, &lua)?;
            g(
                &client,
                without_mining,
                RadiusGoal {
                    pos: azalea::Vec3::new(p.x, p.y, p.z),
                    radius: t.get("radius")?,
                },
            );
        }
        2 => {
            let p = Vec3::from_lua(data, &lua)?;
            g(
                &client,
                without_mining,
                ReachBlockPosGoal {
                    pos: BlockPos::new(p.x as i32, p.y as i32, p.z as i32),
                    chunk_storage: client.world().read().chunks.clone(),
                },
            );
        }
        3 => {
            let t = data.as_table().ok_or(error)?;
            g(
                &client,
                without_mining,
                XZGoal {
                    x: t.get("x")?,
                    z: t.get("z")?,
                },
            );
        }
        4 => g(
            &client,
            without_mining,
            YGoal {
                y: data.as_integer().ok_or(error)?,
            },
        ),
        _ => {
            let p = Vec3::from_lua(data, &lua)?;
            g(
                &client,
                without_mining,
                BlockPosGoal(BlockPos::new(p.x as i32, p.y as i32, p.z as i32)),
            );
        }
    }

    while client.get_tick_broadcaster().recv().await.is_ok() {
        if client.ecs.lock().get::<GotoEvent>(client.entity).is_none() {
            break;
        }
    }

    Ok(())
}

pub fn jump(_lua: &Lua, client: &mut Client, _: ()) -> Result<()> {
    client.jump();
    Ok(())
}

pub fn looking_at(lua: &Lua, client: &Client) -> Result<Option<Table>> {
    let hr = client.component::<HitResultComponent>();
    Ok(if hr.miss {
        None
    } else {
        let result = lua.create_table()?;
        result.set(
            "position",
            Vec3 {
                x: f64::from(hr.block_pos.x),
                y: f64::from(hr.block_pos.y),
                z: f64::from(hr.block_pos.z),
            },
        )?;
        result.set("inside", hr.inside)?;
        result.set("world_border", hr.world_border)?;
        Some(result)
    })
}

pub fn look_at(_lua: &Lua, client: &mut Client, position: Vec3) -> Result<()> {
    client.look_at(azalea::Vec3::new(position.x, position.y, position.z));
    Ok(())
}

pub fn pathfinder(lua: &Lua, client: &Client) -> Result<Table> {
    let pathfinder = lua.create_table()?;
    pathfinder.set(
        "is_calculating",
        client.component::<Pathfinder>().is_calculating,
    )?;
    pathfinder.set(
        "is_executing",
        if let Some(p) = client.get_component::<ExecutingPath>() {
            pathfinder.set(
                "last_reached_node",
                Vec3 {
                    x: f64::from(p.last_reached_node.x),
                    y: f64::from(p.last_reached_node.y),
                    z: f64::from(p.last_reached_node.z),
                },
            )?;
            pathfinder.set(
                "last_node_reach_elapsed",
                p.last_node_reached_at.elapsed().as_millis(),
            )?;
            pathfinder.set("is_path_partial", p.is_path_partial)?;
            true
        } else {
            false
        },
    )?;
    Ok(pathfinder)
}

pub fn position(_lua: &Lua, client: &Client) -> Result<Vec3> {
    let p = client.position();
    Ok(Vec3 {
        x: p.x,
        y: p.y,
        z: p.z,
    })
}

pub fn set_direction(_lua: &Lua, client: &mut Client, direction: (f32, f32)) -> Result<()> {
    client.set_direction(direction.0, direction.1);
    Ok(())
}

pub fn set_jumping(_lua: &Lua, client: &mut Client, jumping: bool) -> Result<()> {
    client.set_jumping(jumping);
    Ok(())
}

pub fn sprint(_lua: &Lua, client: &mut Client, direction: u8) -> Result<()> {
    client.sprint(match direction {
        5 => SprintDirection::ForwardRight,
        6 => SprintDirection::ForwardLeft,
        _ => SprintDirection::Forward,
    });
    Ok(())
}

pub fn stop_pathfinding(_lua: &Lua, client: &Client, _: ()) -> Result<()> {
    client.stop_pathfinding();
    Ok(())
}

pub fn walk(_lua: &Lua, client: &mut Client, direction: u8) -> Result<()> {
    client.walk(match direction {
        1 => WalkDirection::Forward,
        2 => WalkDirection::Backward,
        3 => WalkDirection::Left,
        4 => WalkDirection::Right,
        5 => WalkDirection::ForwardRight,
        6 => WalkDirection::ForwardLeft,
        7 => WalkDirection::BackwardRight,
        8 => WalkDirection::BackwardLeft,
        _ => WalkDirection::None,
    });
    Ok(())
}