Added potion effects

crates/valence_generated/build/main.rs

@@ -5,9 +5,11 @@ mod chunk_view;
 mod item;
 mod packet_id;
 mod sound;
+mod status_effect;
 
 pub fn main() -> anyhow::Result<()> {
     write_generated_file(block::build()?, "block.rs")?;
+    write_generated_file(status_effect::build()?, "effect.rs")?;
     write_generated_file(item::build()?, "item.rs")?;
     write_generated_file(sound::build()?, "sound.rs")?;
     write_generated_file(packet_id::build()?, "packet_id.rs")?;

crates/valence_generated/build/status_effect.rs

@@ -0,0 +1,227 @@
+use heck::ToPascalCase;
+use proc_macro2::TokenStream;
+use quote::quote;
+use serde::Deserialize;
+use valence_build_utils::{ident, rerun_if_changed};
+
+#[derive(Deserialize, Debug)]
+pub enum StatusEffectCategory {
+    Beneficial,
+    Harmful,
+    Neutral,
+}
+
+#[derive(Deserialize, Debug)]
+pub struct StatusEffect {
+    id: u16,
+    name: String,
+    translation_key: String,
+    category: StatusEffectCategory,
+    color: u32,
+    instant: bool,
+}
+
+pub fn build() -> anyhow::Result<TokenStream> {
+    rerun_if_changed(["extracted/effects.json"]);
+
+    let effects =
+        serde_json::from_str::<Vec<StatusEffect>>(include_str!("../extracted/effects.json"))?;
+
+    let effect_count = effects.len();
+
+    let effect_from_raw_id_arms = effects
+        .iter()
+        .map(|effect| {
+            let id = &effect.id;
+            let name = ident(effect.name.to_pascal_case());
+
+            quote! {
+                #id => Some(Self::#name),
+            }
+        })
+        .collect::<TokenStream>();
+
+    let effect_to_raw_id_arms = effects
+        .iter()
+        .map(|effect| {
+            let id = &effect.id;
+            let name = ident(effect.name.to_pascal_case());
+
+            quote! {
+                Self::#name => #id,
+            }
+        })
+        .collect::<TokenStream>();
+
+    let effect_from_ident_arms = effects
+        .iter()
+        .map(|effect| {
+            // TODO: put the full resource identifier in the extracted JSON.
+            let path_name = &effect.name;
+            let ident_name = format!("minecraft:{}", &effect.name);
+
+            let name = ident(path_name.to_pascal_case());
+            quote! {
+                #ident_name => Some(Self::#name),
+            }
+        })
+        .collect::<TokenStream>();
+
+    let effect_to_ident_arms = effects
+        .iter()
+        .map(|effect| {
+            let str_name = &effect.name;
+            let name = ident(str_name.to_pascal_case());
+            quote! {
+                Self::#name => ident!(#str_name),
+            }
+        })
+        .collect::<TokenStream>();
+
+    let effect_to_translation_key_arms = effects
+        .iter()
+        .map(|effect| {
+            let str_name = &effect.translation_key;
+            let name = ident(effect.name.to_pascal_case());
+            quote! {
+                Self::#name => #str_name,
+            }
+        })
+        .collect::<TokenStream>();
+
+    let effect_to_category_arms = effects
+        .iter()
+        .map(|effect| {
+            let category = match &effect.category {
+                StatusEffectCategory::Beneficial => quote! { StatusEffectCategory::Beneficial },
+                StatusEffectCategory::Harmful => quote! { StatusEffectCategory::Harmful },
+                StatusEffectCategory::Neutral => quote! { StatusEffectCategory::Neutral },
+            };
+
+            let name = ident(effect.name.to_pascal_case());
+            quote! {
+                Self::#name => #category,
+            }
+        })
+        .collect::<TokenStream>();
+
+    let effect_to_color_arms = effects
+        .iter()
+        .map(|effect| {
+            let color = &effect.color;
+            let name = ident(effect.name.to_pascal_case());
+            quote! {
+                Self::#name => #color,
+            }
+        })
+        .collect::<TokenStream>();
+
+    let effect_to_instant_arms = effects
+        .iter()
+        .map(|effect| {
+            let instant = &effect.instant;
+            let name = ident(effect.name.to_pascal_case());
+            quote! {
+                Self::#name => #instant,
+            }
+        })
+        .collect::<TokenStream>();
+
+    let effect_variants = effects
+        .iter()
+        .map(|effect| ident(effect.name.to_pascal_case()))
+        .collect::<Vec<_>>();
+
+    Ok(quote! {
+        use valence_ident::{Ident, ident};
+
+        /// Represents a status effect category
+        #[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, Debug)]
+        pub enum StatusEffectCategory {
+            Beneficial,
+            Harmful,
+            Neutral,
+        }
+
+        /// Represents a status effect from the game
+        #[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, Debug)]
+        pub enum StatusEffect {
+            #(#effect_variants,)*
+        }
+
+        impl StatusEffect {
+            /// Constructs a effect from a raw item ID.
+            ///
+            /// If the given ID is invalid, `None` is returned.
+            pub const fn from_raw(id: u16) -> Option<Self> {
+                match id {
+                    #effect_from_raw_id_arms
+                    _ => None
+                }
+            }
+
+            /// Gets the raw effect ID from the effect
+            pub const fn to_raw(self) -> u16 {
+                match self {
+                    #effect_to_raw_id_arms
+                }
+            }
+
+            /// Construct a effect from its snake_case name.
+            ///
+            /// Returns `None` if the name is invalid.
+            pub fn from_ident(id: Ident<&str>) -> Option<Self> {
+                match id.as_str() {
+                    #effect_from_ident_arms
+                    _ => None
+                }
+            }
+
+            /// Gets the identifier of this effect.
+            pub const fn to_ident(self) -> Ident<&'static str> {
+                match self {
+                    #effect_to_ident_arms
+                }
+            }
+
+            /// Gets the name of this effect.
+            /// Same as [`StatusEffect::to_ident`], but doesn't take ownership.
+            pub const fn name(&self) -> Ident<&'static str> {
+                match self {
+                    #effect_to_ident_arms
+                }
+            }
+
+            /// Gets the translation key of this effect.
+            pub const fn translation_key(&self) -> &'static str {
+                match self {
+                    #effect_to_translation_key_arms
+                }
+            }
+
+            /// Gets the category of this effect.
+            pub const fn category(&self) -> StatusEffectCategory {
+                match self {
+                    #effect_to_category_arms
+                }
+            }
+
+            /// Gets the color of this effect.
+            pub const fn color(&self) -> u32 {
+                match self {
+                    #effect_to_color_arms
+                }
+            }
+
+            /// Gets whether this effect is instant.
+            pub const fn instant(&self) -> bool {
+                match self {
+                    #effect_to_instant_arms
+                }
+            }
+
+            /// An array of all effects.
+            pub const ALL: [Self; #effect_count] = [#(Self::#effect_variants,)*];
+        }
+    })
+}