gesha_rust_types/
enum_variant.rs

1use crate::{DataType, EnumVariantName};
2use std::fmt::{Display, Formatter};
3
4/// rf. https://doc.rust-lang.org/reference/items/enumerations.html
5#[derive(Clone, Debug, Hash, Eq, PartialEq)]
6pub struct EnumVariant {
7    pub name: EnumVariantName,
8    pub attributes: Vec<EnumVariantAttribute>,
9    pub case: EnumCase,
10    _hide_default_constructor: bool,
11}
12
13impl EnumVariant {
14    pub fn unit(
15        name: EnumVariantName,
16        constant: EnumConstant,
17        attributes: Vec<EnumVariantAttribute>,
18    ) -> Self {
19        EnumVariant {
20            name,
21            attributes,
22            case: EnumCase::Unit(constant),
23            _hide_default_constructor: true,
24        }
25    }
26    pub fn tuple(
27        name: EnumVariantName,
28        types: Vec<DataType>,
29        attributes: Vec<EnumVariantAttribute>,
30    ) -> Self {
31        EnumVariant {
32            name,
33            attributes,
34            case: EnumCase::Tuple(types),
35            _hide_default_constructor: true,
36        }
37    }
38}
39
40#[derive(Clone, Debug, Hash, Eq, PartialEq)]
41pub enum EnumCase {
42    Unit(EnumConstant),
43    Tuple(Vec<DataType>),
44}
45
46#[derive(Clone, Debug, Hash, Eq, PartialEq)]
47pub enum EnumConstant {
48    Bool(bool),
49    I32(i32),
50    I64(i64),
51    Null,
52    Str(String),
53}
54
55impl EnumConstant {
56    pub fn type_name(&self) -> &str {
57        match self {
58            EnumConstant::Bool(_) => "bool",
59            EnumConstant::I32(_) => "i32",
60            EnumConstant::I64(_) => "i64",
61            EnumConstant::Null => "null",
62            EnumConstant::Str(_) => "str",
63        }
64    }
65}
66
67impl Display for EnumConstant {
68    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
69        match self {
70            EnumConstant::Bool(x) => Display::fmt(x, f),
71            EnumConstant::I32(x) => Display::fmt(x, f),
72            EnumConstant::I64(x) => Display::fmt(x, f),
73            EnumConstant::Null => Display::fmt("null", f),
74            EnumConstant::Str(x) => Display::fmt(&format!(r#""{x}""#), f),
75        }
76    }
77}
78
79#[derive(Clone, Debug, Hash, Eq, PartialEq)]
80pub struct EnumVariantAttribute(String);
81
82impl EnumVariantAttribute {
83    pub fn new<A: Into<String>>(a: A) -> Self {
84        Self(a.into())
85    }
86}
87
88impl Display for EnumVariantAttribute {
89    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
90        Display::fmt(&self.0, f)
91    }
92}
gesha_rust_types/enum_variant.rs

gesha_rust_types/
enum_variant.rs
