wit_encoder/
enum_.rs

1use crate::{Docs, Ident};
2
3/// A variant without a payload
4#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Default)]
5#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
6#[cfg_attr(feature = "serde", serde(rename_all = "kebab-case"))]
7pub struct Enum {
8    pub(crate) cases: Vec<EnumCase>,
9}
10
11impl<C> FromIterator<C> for Enum
12where
13    C: Into<EnumCase>,
14{
15    fn from_iter<T: IntoIterator<Item = C>>(iter: T) -> Self {
16        Self {
17            cases: iter.into_iter().map(|c| c.into()).collect(),
18        }
19    }
20}
21
22impl Enum {
23    pub fn empty() -> Self {
24        Self::default()
25    }
26
27    pub fn case(&mut self, case: impl Into<EnumCase>) {
28        self.cases.push(case.into());
29    }
30
31    pub fn cases(&self) -> &[EnumCase] {
32        &self.cases
33    }
34
35    pub fn cases_mut(&mut self) -> &mut Vec<EnumCase> {
36        &mut self.cases
37    }
38}
39
40#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
41#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
42#[cfg_attr(feature = "serde", serde(rename_all = "kebab-case"))]
43pub struct EnumCase {
44    pub(crate) name: Ident,
45    pub(crate) docs: Option<Docs>,
46}
47
48impl<N> From<N> for EnumCase
49where
50    N: Into<Ident>,
51{
52    fn from(value: N) -> Self {
53        Self {
54            name: value.into(),
55            docs: None,
56        }
57    }
58}
59
60impl EnumCase {
61    pub fn new(name: impl Into<Ident>) -> Self {
62        Self {
63            name: name.into(),
64            docs: None,
65        }
66    }
67
68    pub fn name(&self) -> &Ident {
69        &self.name
70    }
71
72    pub fn set_name(&mut self, name: impl Into<Ident>) {
73        self.name = name.into();
74    }
75
76    pub fn docs(&self) -> &Option<Docs> {
77        &self.docs
78    }
79
80    pub fn set_docs(&mut self, docs: Option<impl Into<Docs>>) {
81        self.docs = docs.map(|d| d.into());
82    }
83}
wit_encoder/enum_.rs

wit_encoder/
enum_.rs
