mago_type_syntax/ast/
reference.rs

1use serde::Serialize;
2
3use mago_span::HasSpan;
4use mago_span::Span;
5
6use crate::ast::generics::GenericParameters;
7use crate::ast::identifier::Identifier;
8
9#[derive(Debug, Clone, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
10pub struct ReferenceType<'arena> {
11    pub identifier: Identifier<'arena>,
12    pub parameters: Option<GenericParameters<'arena>>,
13}
14
15#[derive(Debug, Clone, Copy, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
16pub enum MemberReferenceSelector<'arena> {
17    Wildcard(Span),
18    Identifier(Identifier<'arena>),
19    StartsWith(Identifier<'arena>, Span),
20    EndsWith(Span, Identifier<'arena>),
21}
22
23#[derive(Debug, Clone, Copy, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
24pub struct MemberReferenceType<'arena> {
25    pub class: Identifier<'arena>,
26    pub double_colon: Span,
27    pub member: MemberReferenceSelector<'arena>,
28}
29
30#[derive(Debug, Clone, Copy, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
31pub struct GlobalWildcardType<'arena> {
32    pub selector: GlobalWildcardSelector<'arena>,
33}
34
35#[derive(Debug, Clone, Copy, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
36pub enum GlobalWildcardSelector<'arena> {
37    StartsWith(Identifier<'arena>, Span),
38    EndsWith(Span, Identifier<'arena>),
39}
40
41impl HasSpan for ReferenceType<'_> {
42    fn span(&self) -> Span {
43        match &self.parameters {
44            Some(parameters) => self.identifier.span.join(parameters.span()),
45            None => self.identifier.span,
46        }
47    }
48}
49
50impl HasSpan for MemberReferenceSelector<'_> {
51    fn span(&self) -> Span {
52        match self {
53            MemberReferenceSelector::Wildcard(span) => *span,
54            MemberReferenceSelector::Identifier(identifier) => identifier.span,
55            MemberReferenceSelector::StartsWith(identifier, span) => identifier.span.join(*span),
56            MemberReferenceSelector::EndsWith(span, identifier) => span.join(identifier.span),
57        }
58    }
59}
60
61impl HasSpan for MemberReferenceType<'_> {
62    fn span(&self) -> Span {
63        self.class.span.join(self.member.span())
64    }
65}
66
67impl HasSpan for GlobalWildcardSelector<'_> {
68    fn span(&self) -> Span {
69        match self {
70            GlobalWildcardSelector::StartsWith(identifier, span) => identifier.span.join(*span),
71            GlobalWildcardSelector::EndsWith(span, identifier) => span.join(identifier.span),
72        }
73    }
74}
75
76impl HasSpan for GlobalWildcardType<'_> {
77    fn span(&self) -> Span {
78        self.selector.span()
79    }
80}
81
82impl std::fmt::Display for ReferenceType<'_> {
83    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
84        if let Some(parameters) = &self.parameters {
85            write!(f, "{}{}", self.identifier, parameters)
86        } else {
87            write!(f, "{}", self.identifier)
88        }
89    }
90}
91
92impl std::fmt::Display for MemberReferenceSelector<'_> {
93    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
94        match self {
95            MemberReferenceSelector::Wildcard(_) => write!(f, "*"),
96            MemberReferenceSelector::Identifier(identifier) => write!(f, "{identifier}"),
97            MemberReferenceSelector::StartsWith(identifier, _) => write!(f, "{identifier}*"),
98            MemberReferenceSelector::EndsWith(_, identifier) => write!(f, "*{identifier}"),
99        }
100    }
101}
102
103impl std::fmt::Display for MemberReferenceType<'_> {
104    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
105        write!(f, "{}::{}", self.class, self.member)
106    }
107}
108
109impl std::fmt::Display for GlobalWildcardSelector<'_> {
110    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
111        match self {
112            GlobalWildcardSelector::StartsWith(identifier, _) => write!(f, "{identifier}*"),
113            GlobalWildcardSelector::EndsWith(_, identifier) => write!(f, "*{identifier}"),
114        }
115    }
116}
117
118impl std::fmt::Display for GlobalWildcardType<'_> {
119    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
120        write!(f, "{}", self.selector)
121    }
122}
mago_type_syntax/ast/reference.rs

mago_type_syntax/ast/
reference.rs
