fix wrong Consumable component implementation and add set_equipment test

2 files changed, 39 insertions, 0 deletions

diff --git a/azalea-inventory/src/item/consume_effect.rs b/azalea-inventory/src/item/consume_effect.rs
new file mode 100644
index 00000000..898f64ff
--- /dev/null
+++ b/azalea-inventory/src/item/consume_effect.rs
@@ -0,0 +1,37 @@
+use azalea_buf::AzBuf;
+use azalea_core::resource_location::ResourceLocation;
+use azalea_registry::{HolderSet, MobEffect, SoundEvent};
+
+use crate::components::MobEffectInstance;
+
+#[derive(Clone, PartialEq, AzBuf)]
+pub enum ConsumeEffect {
+    ApplyEffects {
+        effects: Vec<MobEffectInstance>,
+        probability: f32,
+    },
+    RemoveEffects(HolderSet<MobEffect, ResourceLocation>),
+    ClearAllEffects,
+    TeleportRandomly {
+        diameter: f32,
+    },
+    PlaySound {
+        sound: SoundEvent,
+    },
+}
+
+impl From<ConsumeEffect> for azalea_registry::ConsumeEffectKind {
+    fn from(effect: ConsumeEffect) -> Self {
+        match effect {
+            ConsumeEffect::ApplyEffects { .. } => azalea_registry::ConsumeEffectKind::ApplyEffects,
+            ConsumeEffect::RemoveEffects { .. } => {
+                azalea_registry::ConsumeEffectKind::RemoveEffects
+            }
+            ConsumeEffect::ClearAllEffects => azalea_registry::ConsumeEffectKind::ClearAllEffects,
+            ConsumeEffect::TeleportRandomly { .. } => {
+                azalea_registry::ConsumeEffectKind::TeleportRandomly
+            }
+            ConsumeEffect::PlaySound { .. } => azalea_registry::ConsumeEffectKind::PlaySound,
+        }
+    }
+}
diff --git a/azalea-inventory/src/item/mod.rs b/azalea-inventory/src/item/mod.rs
index 7f077c6a..bbff2d50 100644
--- a/azalea-inventory/src/item/mod.rs
+++ b/azalea-inventory/src/item/mod.rs
@@ -1,3 +1,5 @@
+pub mod consume_effect;
+
 pub trait MaxStackSizeExt {
     /// Get the maximum stack size for this item.
     ///