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<'input> {
11    pub identifier: Identifier<'input>,
12    pub parameters: Option<GenericParameters<'input>>,
13}
14
15#[derive(Debug, Clone, Copy, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
16pub enum MemberReferenceSelector<'input> {
17    Wildcard(Span),
18    Identifier(Identifier<'input>),
19    StartsWith(Identifier<'input>, Span),
20    EndsWith(Span, Identifier<'input>),
21}
22
23#[derive(Debug, Clone, Copy, Eq, PartialEq, Hash, Serialize, PartialOrd, Ord)]
24pub struct MemberReferenceType<'input> {
25    pub class: Identifier<'input>,
26    pub double_colon: Span,
27    pub member: MemberReferenceSelector<'input>,
28}
29
30impl HasSpan for ReferenceType<'_> {
31    fn span(&self) -> Span {
32        match &self.parameters {
33            Some(parameters) => self.identifier.span.join(parameters.span()),
34            None => self.identifier.span,
35        }
36    }
37}
38
39impl HasSpan for MemberReferenceSelector<'_> {
40    fn span(&self) -> Span {
41        match self {
42            MemberReferenceSelector::Wildcard(span) => *span,
43            MemberReferenceSelector::Identifier(identifier) => identifier.span,
44            MemberReferenceSelector::StartsWith(identifier, span) => identifier.span.join(*span),
45            MemberReferenceSelector::EndsWith(span, identifier) => span.join(identifier.span),
46        }
47    }
48}
49
50impl HasSpan for MemberReferenceType<'_> {
51    fn span(&self) -> Span {
52        self.class.span.join(self.member.span())
53    }
54}
55
56impl std::fmt::Display for ReferenceType<'_> {
57    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
58        if let Some(parameters) = &self.parameters {
59            write!(f, "{}{}", self.identifier, parameters)
60        } else {
61            write!(f, "{}", self.identifier)
62        }
63    }
64}
65
66impl std::fmt::Display for MemberReferenceSelector<'_> {
67    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
68        match self {
69            MemberReferenceSelector::Wildcard(_) => write!(f, "*"),
70            MemberReferenceSelector::Identifier(identifier) => write!(f, "{identifier}"),
71            MemberReferenceSelector::StartsWith(identifier, _) => write!(f, "{identifier}*"),
72            MemberReferenceSelector::EndsWith(_, identifier) => write!(f, "*{identifier}"),
73        }
74    }
75}
76
77impl std::fmt::Display for MemberReferenceType<'_> {
78    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
79        write!(f, "{}::{}", self.class, self.member)
80    }
81}
mago_type_syntax/ast/reference.rs

mago_type_syntax/ast/
reference.rs
