refactor: directly wrap structs

This commit is contained in:
Ryan 2025-03-12 18:28:30 -04:00
parent 44f7b9a00f
commit 85729401e5
Signed by: ErrorNoInternet
GPG Key ID: 2486BFB7B1E6A4A3
8 changed files with 50 additions and 80 deletions

View File

@ -241,12 +241,7 @@ async fn lua_init(client: Client, state: &State, globals: &Table) -> Result<()>
.map_err(Error::external) .map_err(Error::external)
})?, })?,
)?; )?;
globals.set( globals.set("client", client::Client(Some(client)))?;
"client",
client::Client {
inner: Some(client),
},
)?;
call_listeners(state, "init", ()).await; call_listeners(state, "init", ()).await;
Ok(()) Ok(())
} }

View File

@ -9,13 +9,11 @@ use log::error;
use mlua::{Lua, Result, Table, UserDataRef}; use mlua::{Lua, Result, Table, UserDataRef};
pub fn container(_lua: &Lua, client: &Client) -> Result<Option<ContainerRef>> { pub fn container(_lua: &Lua, client: &Client) -> Result<Option<ContainerRef>> {
Ok(client Ok(client.get_open_container().map(ContainerRef))
.get_open_container()
.map(|c| ContainerRef { inner: c }))
} }
pub fn held_item(_lua: &Lua, client: &Client) -> Result<ItemStack> { pub fn held_item(_lua: &Lua, client: &Client) -> Result<ItemStack> {
Ok(ItemStack::from(client.component::<Inventory>().held_item())) Ok(ItemStack(client.component::<Inventory>().held_item()))
} }
pub fn held_slot(_lua: &Lua, client: &Client) -> Result<u8> { pub fn held_slot(_lua: &Lua, client: &Client) -> Result<u8> {
@ -26,7 +24,7 @@ pub fn held_slot(_lua: &Lua, client: &Client) -> Result<u8> {
pub fn menu(lua: &Lua, client: &Client) -> Result<Table> { pub fn menu(lua: &Lua, client: &Client) -> Result<Table> {
fn from_slot_list<const N: usize>(s: SlotList<N>) -> Vec<ItemStack> { fn from_slot_list<const N: usize>(s: SlotList<N>) -> Vec<ItemStack> {
s.iter() s.iter()
.map(|i| ItemStack::from(i.to_owned())) .map(|i| ItemStack(i.to_owned()))
.collect::<Vec<_>>() .collect::<Vec<_>>()
} }
@ -40,11 +38,11 @@ pub fn menu(lua: &Lua, client: &Client) -> Result<Table> {
offhand, offhand,
}) => { }) => {
table.set("type", 0)?; table.set("type", 0)?;
table.set("craft_result", ItemStack::from(craft_result))?; table.set("craft_result", ItemStack(craft_result))?;
table.set("craft", from_slot_list(craft))?; table.set("craft", from_slot_list(craft))?;
table.set("armor", from_slot_list(armor))?; table.set("armor", from_slot_list(armor))?;
table.set("inventory", from_slot_list(inventory))?; table.set("inventory", from_slot_list(inventory))?;
table.set("offhand", ItemStack::from(offhand))?; table.set("offhand", ItemStack(offhand))?;
} }
Menu::Generic9x3 { contents, player } => { Menu::Generic9x3 { contents, player } => {
table.set("type", 3)?; table.set("type", 3)?;
@ -62,7 +60,7 @@ pub fn menu(lua: &Lua, client: &Client) -> Result<Table> {
player, player,
} => { } => {
table.set("type", 13)?; table.set("type", 13)?;
table.set("result", ItemStack::from(result))?; table.set("result", ItemStack(result))?;
table.set("grid", from_slot_list(grid))?; table.set("grid", from_slot_list(grid))?;
table.set("player", from_slot_list(player))?; table.set("player", from_slot_list(player))?;
} }
@ -78,7 +76,7 @@ pub fn menu(lua: &Lua, client: &Client) -> Result<Table> {
} => { } => {
table.set("type", 20)?; table.set("type", 20)?;
table.set("payments", from_slot_list(payments))?; table.set("payments", from_slot_list(payments))?;
table.set("result", ItemStack::from(result))?; table.set("result", ItemStack(result))?;
table.set("player", from_slot_list(player))?; table.set("player", from_slot_list(player))?;
} }
Menu::ShulkerBox { contents, player } => { Menu::ShulkerBox { contents, player } => {
@ -105,11 +103,11 @@ pub async fn open_container_at(
position.z as i32, position.z as i32,
)) ))
.await .await
.map(|c| Container { inner: c })) .map(Container))
} }
pub fn open_inventory(_lua: &Lua, client: &mut Client, _: ()) -> Result<Option<Container>> { pub fn open_inventory(_lua: &Lua, client: &mut Client, _: ()) -> Result<Option<Container>> {
Ok(client.open_inventory().map(|c| Container { inner: c })) Ok(client.open_inventory().map(Container))
} }
pub fn set_held_slot(_lua: &Lua, client: &Client, slot: u8) -> Result<()> { pub fn set_held_slot(_lua: &Lua, client: &Client, slot: u8) -> Result<()> {

View File

@ -14,25 +14,19 @@ use azalea::{Client as AzaleaClient, world::MinecraftEntityId};
use mlua::{Lua, Result, UserData, UserDataFields, UserDataMethods}; use mlua::{Lua, Result, UserData, UserDataFields, UserDataMethods};
use std::ops::{Deref, DerefMut}; use std::ops::{Deref, DerefMut};
pub struct Client { pub struct Client(pub Option<AzaleaClient>);
pub inner: Option<AzaleaClient>,
}
impl Deref for Client { impl Deref for Client {
type Target = AzaleaClient; type Target = AzaleaClient;
fn deref(&self) -> &Self::Target { fn deref(&self) -> &Self::Target {
self.inner self.0.as_ref().expect("should have received init event")
.as_ref()
.expect("should have received init event")
} }
} }
impl DerefMut for Client { impl DerefMut for Client {
fn deref_mut(&mut self) -> &mut Self::Target { fn deref_mut(&mut self) -> &mut Self::Target {
self.inner self.0.as_mut().expect("should have received init event")
.as_mut()
.expect("should have received init event")
} }
} }

View File

@ -1,24 +1,19 @@
use azalea::inventory::components::{CustomName, Damage, Food, MaxDamage}; use azalea::inventory::{
self,
components::{CustomName, Damage, Food, MaxDamage},
};
use mlua::{UserData, UserDataFields, UserDataMethods}; use mlua::{UserData, UserDataFields, UserDataMethods};
pub struct ItemStack { pub struct ItemStack(pub inventory::ItemStack);
pub inner: azalea::inventory::ItemStack,
}
impl From<azalea::inventory::ItemStack> for ItemStack {
fn from(inner: azalea::inventory::ItemStack) -> Self {
Self { inner }
}
}
impl UserData for ItemStack { impl UserData for ItemStack {
fn add_fields<F: UserDataFields<Self>>(f: &mut F) { fn add_fields<F: UserDataFields<Self>>(f: &mut F) {
f.add_field_method_get("is_empty", |_, this| Ok(this.inner.is_empty())); f.add_field_method_get("is_empty", |_, this| Ok(this.0.is_empty()));
f.add_field_method_get("is_present", |_, this| Ok(this.inner.is_present())); f.add_field_method_get("is_present", |_, this| Ok(this.0.is_present()));
f.add_field_method_get("count", |_, this| Ok(this.inner.count())); f.add_field_method_get("count", |_, this| Ok(this.0.count()));
f.add_field_method_get("kind", |_, this| Ok(this.inner.kind().to_string())); f.add_field_method_get("kind", |_, this| Ok(this.0.kind().to_string()));
f.add_field_method_get("custom_name", |_, this| { f.add_field_method_get("custom_name", |_, this| {
Ok(this.inner.as_present().map(|data| { Ok(this.0.as_present().map(|data| {
data.components data.components
.get::<CustomName>() .get::<CustomName>()
.map(|c| c.name.to_string()) .map(|c| c.name.to_string())
@ -26,13 +21,13 @@ impl UserData for ItemStack {
}); });
f.add_field_method_get("damage", |_, this| { f.add_field_method_get("damage", |_, this| {
Ok(this Ok(this
.inner .0
.as_present() .as_present()
.map(|data| data.components.get::<Damage>().map(|d| d.amount))) .map(|data| data.components.get::<Damage>().map(|d| d.amount)))
}); });
f.add_field_method_get("max_damage", |_, this| { f.add_field_method_get("max_damage", |_, this| {
Ok(this Ok(this
.inner .0
.as_present() .as_present()
.map(|data| data.components.get::<MaxDamage>().map(|d| d.amount))) .map(|data| data.components.get::<MaxDamage>().map(|d| d.amount)))
}); });
@ -40,7 +35,7 @@ impl UserData for ItemStack {
f.add_field_method_get("food", |lua, this| { f.add_field_method_get("food", |lua, this| {
Ok( Ok(
if let Some(food) = this if let Some(food) = this
.inner .0
.as_present() .as_present()
.and_then(|data| data.components.get::<Food>()) .and_then(|data| data.components.get::<Food>())
{ {
@ -59,10 +54,10 @@ impl UserData for ItemStack {
fn add_methods<M: UserDataMethods<Self>>(m: &mut M) { fn add_methods<M: UserDataMethods<Self>>(m: &mut M) {
m.add_method_mut("split", |_, this, count: u32| { m.add_method_mut("split", |_, this, count: u32| {
Ok(ItemStack::from(this.inner.split(count))) Ok(ItemStack(this.0.split(count)))
}); });
m.add_method_mut("update_empty", |_, this, (): ()| { m.add_method_mut("update_empty", |_, this, (): ()| {
this.inner.update_empty(); this.0.update_empty();
Ok(()) Ok(())
}); });
} }

View File

@ -6,22 +6,20 @@ use click::operation_from_table;
use item_stack::ItemStack; use item_stack::ItemStack;
use mlua::{Table, UserData, UserDataFields, UserDataMethods}; use mlua::{Table, UserData, UserDataFields, UserDataMethods};
pub struct Container { pub struct Container(pub ContainerHandle);
pub inner: ContainerHandle,
}
impl UserData for Container { impl UserData for Container {
fn add_fields<F: UserDataFields<Self>>(f: &mut F) { fn add_fields<F: UserDataFields<Self>>(f: &mut F) {
f.add_field_method_get("id", |_, this| Ok(this.inner.id())); f.add_field_method_get("id", |_, this| Ok(this.0.id()));
f.add_field_method_get("menu", |_, this| { f.add_field_method_get("menu", |_, this| {
Ok(this.inner.menu().map(|m| format!("{m:?}"))) Ok(this.0.menu().map(|m| format!("{m:?}")))
}); });
f.add_field_method_get("contents", |_, this| { f.add_field_method_get("contents", |_, this| {
Ok(this.inner.contents().map(|v| { Ok(this.0.contents().map(|v| {
v.iter() v.iter()
.map(|i| ItemStack::from(i.to_owned())) .map(|i| ItemStack(i.to_owned()))
.collect::<Vec<_>>() .collect::<Vec<_>>()
})) }))
}); });
@ -31,7 +29,7 @@ impl UserData for Container {
m.add_method( m.add_method(
"click", "click",
|_, this, (operation, operation_type): (Table, Option<u8>)| { |_, this, (operation, operation_type): (Table, Option<u8>)| {
this.inner this.0
.click(operation_from_table(operation, operation_type)?); .click(operation_from_table(operation, operation_type)?);
Ok(()) Ok(())
}, },
@ -39,22 +37,20 @@ impl UserData for Container {
} }
} }
pub struct ContainerRef { pub struct ContainerRef(pub ContainerHandleRef);
pub inner: ContainerHandleRef,
}
impl UserData for ContainerRef { impl UserData for ContainerRef {
fn add_fields<F: UserDataFields<Self>>(f: &mut F) { fn add_fields<F: UserDataFields<Self>>(f: &mut F) {
f.add_field_method_get("id", |_, this| Ok(this.inner.id())); f.add_field_method_get("id", |_, this| Ok(this.0.id()));
f.add_field_method_get("menu", |_, this| { f.add_field_method_get("menu", |_, this| {
Ok(this.inner.menu().map(|m| format!("{m:?}"))) Ok(this.0.menu().map(|m| format!("{m:?}")))
}); });
f.add_field_method_get("contents", |_, this| { f.add_field_method_get("contents", |_, this| {
Ok(this.inner.contents().map(|v| { Ok(this.0.contents().map(|v| {
v.iter() v.iter()
.map(|i| ItemStack::from(i.to_owned())) .map(|i| ItemStack(i.to_owned()))
.collect::<Vec<_>>() .collect::<Vec<_>>()
})) }))
}); });
@ -62,14 +58,14 @@ impl UserData for ContainerRef {
fn add_methods<M: UserDataMethods<Self>>(m: &mut M) { fn add_methods<M: UserDataMethods<Self>>(m: &mut M) {
m.add_method("close", |_, this, (): ()| { m.add_method("close", |_, this, (): ()| {
this.inner.close(); this.0.close();
Ok(()) Ok(())
}); });
m.add_method( m.add_method(
"click", "click",
|_, this, (operation, operation_type): (Table, Option<u8>)| { |_, this, (operation, operation_type): (Table, Option<u8>)| {
this.inner this.0
.click(operation_from_table(operation, operation_type)?); .click(operation_from_table(operation, operation_type)?);
Ok(()) Ok(())
}, },

View File

@ -4,7 +4,7 @@ macro_rules! crypt {
macro_rules! crypt_with { macro_rules! crypt_with {
($algo:ident) => {{ ($algo:ident) => {{
let encoding = $options.get("encoding").unwrap_or_default(); let encoding = $options.get("encoding").unwrap_or_default();
let key = &$options.get::<UserDataRef<AesKey>>("key")?.inner; let key = &$options.get::<UserDataRef<AesKey>>("key")?.0;
match encoding { match encoding {
1 => $algo::<Base64Encoding>::$op($text, &key), 1 => $algo::<Base64Encoding>::$op($text, &key),
2 => $algo::<Base64rEncoding>::$op($text, &key), 2 => $algo::<Base64rEncoding>::$op($text, &key),

View File

@ -1,12 +1,10 @@
use mlua::UserData; use mlua::UserData;
pub struct AesKey { pub struct AesKey(pub ncr::AesKey);
pub inner: ncr::AesKey,
}
impl UserData for AesKey { impl UserData for AesKey {
fn add_fields<F: mlua::UserDataFields<Self>>(f: &mut F) { fn add_fields<F: mlua::UserDataFields<Self>>(f: &mut F) {
f.add_field_method_get("base64", |_, this| Ok(this.inner.encode_base64())); f.add_field_method_get("base64", |_, this| Ok(this.0.encode_base64()));
f.add_field_method_get("bytes", |_, this| Ok(this.inner.as_ref().to_vec())); f.add_field_method_get("bytes", |_, this| Ok(this.0.as_ref().to_vec()));
} }
} }

View File

@ -14,29 +14,23 @@ pub fn register_globals(lua: &Lua, globals: &Table) -> Result<()> {
globals.set( globals.set(
"ncr_aes_key_from_passphrase", "ncr_aes_key_from_passphrase",
lua.create_function(|_, passphrase: Vec<u8>| { lua.create_function(|_, passphrase: Vec<u8>| {
Ok(AesKey { Ok(AesKey(ncr::AesKey::gen_from_passphrase(&passphrase)))
inner: ncr::AesKey::gen_from_passphrase(&passphrase),
})
})?, })?,
)?; )?;
globals.set( globals.set(
"ncr_aes_key_from_base64", "ncr_aes_key_from_base64",
lua.create_function(|_, base64: String| { lua.create_function(|_, base64: String| {
Ok(AesKey { Ok(AesKey(
inner: ncr::AesKey::decode_base64(&base64) ncr::AesKey::decode_base64(&base64)
.map_err(|error| Error::external(error.to_string()))?, .map_err(|error| Error::external(error.to_string()))?,
}) ))
})?, })?,
)?; )?;
globals.set( globals.set(
"ncr_generate_random_aes_key", "ncr_generate_random_aes_key",
lua.create_function(|_, (): ()| { lua.create_function(|_, (): ()| Ok(AesKey(ncr::AesKey::gen_random_key())))?,
Ok(AesKey {
inner: ncr::AesKey::gen_random_key(),
})
})?,
)?; )?;
globals.set( globals.set(