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