mago_type_syntax/ast/
array.rs

1use serde::Serialize;
2
3use mago_span::HasSpan;
4use mago_span::Span;
5
6use crate::ast::generics::GenericParameters;
7use crate::ast::keyword::Keyword;
8
9#[derive(Debug, Clone, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
10#[repr(C)]
11pub struct ArrayType<'input> {
12    pub keyword: Keyword<'input>,
13    pub parameters: Option<GenericParameters<'input>>,
14}
15
16#[derive(Debug, Clone, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
17#[repr(C)]
18pub struct NonEmptyArrayType<'input> {
19    pub keyword: Keyword<'input>,
20    pub parameters: Option<GenericParameters<'input>>,
21}
22
23#[derive(Debug, Clone, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
24#[repr(C)]
25pub struct AssociativeArrayType<'input> {
26    pub keyword: Keyword<'input>,
27    pub parameters: Option<GenericParameters<'input>>,
28}
29
30#[derive(Debug, Clone, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
31#[repr(C)]
32pub struct ListType<'input> {
33    pub keyword: Keyword<'input>,
34    pub parameters: Option<GenericParameters<'input>>,
35}
36
37#[derive(Debug, Clone, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
38#[repr(C)]
39pub struct NonEmptyListType<'input> {
40    pub keyword: Keyword<'input>,
41    pub parameters: Option<GenericParameters<'input>>,
42}
43
44impl HasSpan for ArrayType<'_> {
45    fn span(&self) -> Span {
46        match &self.parameters {
47            Some(parameters) => self.keyword.span.join(parameters.span()),
48            None => self.keyword.span,
49        }
50    }
51}
52
53impl HasSpan for NonEmptyArrayType<'_> {
54    fn span(&self) -> Span {
55        match &self.parameters {
56            Some(parameters) => self.keyword.span.join(parameters.span()),
57            None => self.keyword.span,
58        }
59    }
60}
61
62impl HasSpan for AssociativeArrayType<'_> {
63    fn span(&self) -> Span {
64        match &self.parameters {
65            Some(parameters) => self.keyword.span.join(parameters.span()),
66            None => self.keyword.span,
67        }
68    }
69}
70
71impl HasSpan for ListType<'_> {
72    fn span(&self) -> Span {
73        match &self.parameters {
74            Some(parameters) => self.keyword.span.join(parameters.span()),
75            None => self.keyword.span,
76        }
77    }
78}
79
80impl HasSpan for NonEmptyListType<'_> {
81    fn span(&self) -> Span {
82        match &self.parameters {
83            Some(parameters) => self.keyword.span.join(parameters.span()),
84            None => self.keyword.span,
85        }
86    }
87}
88
89impl std::fmt::Display for ArrayType<'_> {
90    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
91        if let Some(parameters) = &self.parameters {
92            write!(f, "{}{}", self.keyword, parameters)
93        } else {
94            write!(f, "{}", self.keyword)
95        }
96    }
97}
98
99impl std::fmt::Display for NonEmptyArrayType<'_> {
100    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
101        if let Some(parameters) = &self.parameters {
102            write!(f, "{}{}", self.keyword, parameters)
103        } else {
104            write!(f, "{}", self.keyword)
105        }
106    }
107}
108
109impl std::fmt::Display for AssociativeArrayType<'_> {
110    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
111        if let Some(parameters) = &self.parameters {
112            write!(f, "{}{}", self.keyword, parameters)
113        } else {
114            write!(f, "{}", self.keyword)
115        }
116    }
117}
118
119impl std::fmt::Display for ListType<'_> {
120    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
121        if let Some(parameters) = &self.parameters {
122            write!(f, "{}{}", self.keyword, parameters)
123        } else {
124            write!(f, "{}", self.keyword)
125        }
126    }
127}
128
129impl std::fmt::Display for NonEmptyListType<'_> {
130    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
131        if let Some(parameters) = &self.parameters {
132            write!(f, "{}{}", self.keyword, parameters)
133        } else {
134            write!(f, "{}", self.keyword)
135        }
136    }
137}