aldrin_core/introspection/
variant.rs

1use super::LexicalId;
2use crate::error::{DeserializeError, SerializeError};
3use crate::value_deserializer::{Deserialize, Deserializer};
4use crate::value_serializer::{Serialize, Serializer};
5use num_enum::{IntoPrimitive, TryFromPrimitive};
6
7#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
8pub struct Variant {
9    id: u32,
10    name: String,
11    variant_type: Option<LexicalId>,
12}
13
14impl Variant {
15    pub(super) fn new(id: u32, name: impl Into<String>, variant_type: Option<LexicalId>) -> Self {
16        Self {
17            id,
18            name: name.into(),
19            variant_type,
20        }
21    }
22
23    pub fn id(&self) -> u32 {
24        self.id
25    }
26
27    pub fn name(&self) -> &str {
28        &self.name
29    }
30
31    pub fn variant_type(&self) -> Option<LexicalId> {
32        self.variant_type
33    }
34}
35
36#[derive(IntoPrimitive, TryFromPrimitive)]
37#[repr(u32)]
38enum VariantField {
39    Id = 0,
40    Name = 1,
41    VariantType = 2,
42}
43
44impl Serialize for Variant {
45    fn serialize(&self, serializer: Serializer) -> Result<(), SerializeError> {
46        let mut serializer = serializer.serialize_struct(3)?;
47
48        serializer.serialize_field(VariantField::Id, &self.id)?;
49        serializer.serialize_field(VariantField::Name, &self.name)?;
50        serializer.serialize_field(VariantField::VariantType, &self.variant_type)?;
51
52        serializer.finish()
53    }
54}
55
56impl Deserialize for Variant {
57    fn deserialize(deserializer: Deserializer) -> Result<Self, DeserializeError> {
58        let mut deserializer = deserializer.deserialize_struct()?;
59
60        let mut id = None;
61        let mut name = None;
62        let mut variant_type = None;
63
64        while deserializer.has_more_fields() {
65            let deserializer = deserializer.deserialize_field()?;
66
67            match deserializer.try_id()? {
68                VariantField::Id => id = deserializer.deserialize().map(Some)?,
69                VariantField::Name => name = deserializer.deserialize().map(Some)?,
70                VariantField::VariantType => variant_type = deserializer.deserialize()?,
71            }
72        }
73
74        deserializer.finish(Self {
75            id: id.ok_or(DeserializeError::InvalidSerialization)?,
76            name: name.ok_or(DeserializeError::InvalidSerialization)?,
77            variant_type,
78        })
79    }
80}
aldrin_core/introspection/variant.rs

aldrin_core/introspection/
variant.rs
