aboutsummaryrefslogtreecommitdiff
path: root/azalea-inventory/src/item/consume_effect.rs
blob: 45b32704f37148dd95b9d3723496cc1539d702f7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
use azalea_buf::AzBuf;
use azalea_core::codec_utils::is_default;
use azalea_registry::{
    HolderSet,
    builtin::{ConsumeEffectKind, MobEffect, SoundEvent},
    identifier::Identifier,
};
use serde::Serialize;

use crate::components::MobEffectInstance;

#[derive(AzBuf, Clone, Debug, PartialEq, Serialize)]
#[serde(rename_all = "snake_case", tag = "type")]
pub enum ConsumeEffect {
    ApplyEffects {
        #[serde(skip_serializing_if = "is_default")]
        effects: Vec<MobEffectInstance>,
        #[serde(skip_serializing_if = "is_default")]
        probability: f32,
    },
    RemoveEffects {
        #[serde(skip_serializing_if = "is_default")]
        effects: HolderSet<MobEffect, Identifier>,
    },
    ClearAllEffects,
    TeleportRandomly {
        #[serde(skip_serializing_if = "is_default")]
        diameter: f32,
    },
    PlaySound {
        sound: SoundEvent,
    },
}

impl From<ConsumeEffect> for ConsumeEffectKind {
    fn from(effect: ConsumeEffect) -> Self {
        match effect {
            ConsumeEffect::ApplyEffects { .. } => ConsumeEffectKind::ApplyEffects,
            ConsumeEffect::RemoveEffects { .. } => ConsumeEffectKind::RemoveEffects,
            ConsumeEffect::ClearAllEffects => ConsumeEffectKind::ClearAllEffects,
            ConsumeEffect::TeleportRandomly { .. } => ConsumeEffectKind::TeleportRandomly,
            ConsumeEffect::PlaySound { .. } => ConsumeEffectKind::PlaySound,
        }
    }
}