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

mago_type_syntax/ast/
reference.rs
