126 lines
3.6 KiB
Rust
126 lines
3.6 KiB
Rust
use azalea::{
|
|
BlockPos,
|
|
blocks::{BlockState, BlockStates},
|
|
ecs::query::{With, Without},
|
|
entity::{
|
|
Dead, EntityKind, EntityUuid, LookDirection, Pose, Position as AzaleaPosition,
|
|
metadata::{CustomName, Owneruuid, Player},
|
|
},
|
|
world::MinecraftEntityId,
|
|
};
|
|
use mlua::{Function, Lua, Result, Table, UserDataRef};
|
|
|
|
use super::{Client, Direction, Vec3};
|
|
|
|
pub fn blocks(
|
|
_lua: &Lua,
|
|
client: &Client,
|
|
(nearest_to, block_states): (Vec3, Vec<u16>),
|
|
) -> Result<Vec<Vec3>> {
|
|
#[allow(clippy::cast_possible_truncation)]
|
|
Ok(client
|
|
.world()
|
|
.read()
|
|
.find_blocks(
|
|
BlockPos::new(
|
|
nearest_to.x as i32,
|
|
nearest_to.y as i32,
|
|
nearest_to.z as i32,
|
|
),
|
|
&BlockStates {
|
|
set: block_states
|
|
.into_iter()
|
|
.flat_map(BlockState::try_from)
|
|
.collect(),
|
|
},
|
|
)
|
|
.map(Vec3::from)
|
|
.collect())
|
|
}
|
|
|
|
pub async fn all_entities(lua: Lua, client: UserDataRef<Client>, (): ()) -> Result<Vec<Table>> {
|
|
let mut matched = Vec::with_capacity(256);
|
|
for (position, custom_name, kind, uuid, direction, id, owner_uuid, pose) in
|
|
get_entities!(client)
|
|
{
|
|
let table = lua.create_table()?;
|
|
table.set("position", position)?;
|
|
table.set("custom_name", custom_name)?;
|
|
table.set("kind", kind)?;
|
|
table.set("uuid", uuid)?;
|
|
table.set("direction", direction)?;
|
|
table.set("id", id)?;
|
|
table.set(
|
|
"owner_uuid",
|
|
owner_uuid.and_then(|v| *v).map(|v| v.to_string()),
|
|
)?;
|
|
table.set("pose", pose)?;
|
|
matched.push(table);
|
|
}
|
|
Ok(matched)
|
|
}
|
|
|
|
pub async fn entities(
|
|
lua: Lua,
|
|
client: UserDataRef<Client>,
|
|
filter_fn: Function,
|
|
) -> Result<Vec<Table>> {
|
|
let mut matched = Vec::new();
|
|
for (position, custom_name, kind, uuid, direction, id, owner_uuid, pose) in
|
|
get_entities!(client)
|
|
{
|
|
let table = lua.create_table()?;
|
|
table.set("position", position)?;
|
|
table.set("custom_name", custom_name)?;
|
|
table.set("kind", kind)?;
|
|
table.set("uuid", uuid)?;
|
|
table.set("direction", direction)?;
|
|
table.set("id", id)?;
|
|
table.set(
|
|
"owner_uuid",
|
|
owner_uuid.and_then(|v| *v).map(|v| v.to_string()),
|
|
)?;
|
|
table.set("pose", pose)?;
|
|
if filter_fn.call_async::<bool>(&table).await? {
|
|
matched.push(table);
|
|
}
|
|
}
|
|
Ok(matched)
|
|
}
|
|
|
|
pub async fn all_players(lua: Lua, client: UserDataRef<Client>, (): ()) -> Result<Vec<Table>> {
|
|
let mut matched = Vec::new();
|
|
for (id, uuid, kind, position, direction, pose) in get_players!(client) {
|
|
let table = lua.create_table()?;
|
|
table.set("id", id)?;
|
|
table.set("uuid", uuid)?;
|
|
table.set("kind", kind)?;
|
|
table.set("position", position)?;
|
|
table.set("direction", direction)?;
|
|
table.set("pose", pose)?;
|
|
matched.push(table);
|
|
}
|
|
Ok(matched)
|
|
}
|
|
|
|
pub async fn players(
|
|
lua: Lua,
|
|
client: UserDataRef<Client>,
|
|
filter_fn: Function,
|
|
) -> Result<Vec<Table>> {
|
|
let mut matched = Vec::new();
|
|
for (id, uuid, kind, position, direction, pose) in get_players!(client) {
|
|
let table = lua.create_table()?;
|
|
table.set("id", id)?;
|
|
table.set("uuid", uuid)?;
|
|
table.set("kind", kind)?;
|
|
table.set("position", position)?;
|
|
table.set("direction", direction)?;
|
|
table.set("pose", pose)?;
|
|
if filter_fn.call_async::<bool>(&table).await? {
|
|
matched.push(table);
|
|
}
|
|
}
|
|
Ok(matched)
|
|
}
|