Skip to main content

mago_type_syntax/ast/
iterable.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)]
10pub struct IterableType<'input> {
11    pub keyword: Keyword<'input>,
12    pub parameters: Option<GenericParameters<'input>>,
13}
14
15impl HasSpan for IterableType<'_> {
16    fn span(&self) -> Span {
17        match &self.parameters {
18            Some(parameters) => self.keyword.span.join(parameters.span()),
19            None => self.keyword.span,
20        }
21    }
22}
23
24impl std::fmt::Display for IterableType<'_> {
25    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
26        if let Some(parameters) = &self.parameters {
27            write!(f, "{}{}", self.keyword, parameters)
28        } else {
29            write!(f, "{}", self.keyword)
30        }
31    }
32}