Skip to main content

mago_type_syntax/cst/
iterable.rs

1use mago_span::HasSpan;
2use mago_span::Span;
3
4use crate::cst::generics::GenericParameters;
5use crate::cst::keyword::Keyword;
6
7#[derive(Debug, Clone, Eq, PartialEq, Hash, PartialOrd, Ord)]
8#[cfg_attr(feature = "serde", derive(serde::Serialize))]
9pub struct IterableType<'arena> {
10    pub keyword: Keyword<'arena>,
11    pub parameters: Option<GenericParameters<'arena>>,
12}
13
14impl HasSpan for IterableType<'_> {
15    fn span(&self) -> Span {
16        match &self.parameters {
17            Some(parameters) => self.keyword.span.join(parameters.span()),
18            None => self.keyword.span,
19        }
20    }
21}
22
23impl std::fmt::Display for IterableType<'_> {
24    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
25        if let Some(parameters) = &self.parameters {
26            write!(f, "{}{}", self.keyword, parameters)
27        } else {
28            write!(f, "{}", self.keyword)
29        }
30    }
31}