-
Notifications
You must be signed in to change notification settings - Fork 48
/
modifiers.rs
109 lines (91 loc) · 3.17 KB
/
modifiers.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
use super::*;
use crate::visitor::ast_visitor::*;
use eyre::Result;
use serde::{Deserialize, Serialize};
use std::fmt::Display;
#[derive(Clone, Debug, Deserialize, Eq, Serialize, PartialEq, Hash)]
#[serde(rename_all = "camelCase")]
pub struct ModifierDefinition {
pub body: Block,
pub overrides: Option<OverrideSpecifier>,
pub documentation: Option<Documentation>,
pub name: String,
pub name_location: Option<String>,
pub parameters: ParameterList,
pub r#virtual: Option<bool>,
pub visibility: Visibility,
pub src: String,
pub id: NodeID,
}
impl Node for ModifierDefinition {
fn accept(&self, visitor: &mut impl ASTConstVisitor) -> Result<()> {
if visitor.visit_modifier_definition(self)? {
// TODO: should we implement a string based visitor?
// self.name.accept(visitor)?;
self.body.accept(visitor)?;
self.parameters.accept(visitor)?;
}
visitor.end_visit_modifier_definition(self)
}
}
impl Display for ModifierDefinition {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.write_str("modifier")?;
if !self.name.is_empty() {
f.write_fmt(format_args!(" {}", self.name))?;
}
f.write_fmt(format_args!("{}", self.parameters))?;
if self.visibility != Visibility::Internal {
f.write_fmt(format_args!("{} {}", self.parameters, self.visibility))?;
}
if let Some(true) = self.r#virtual {
f.write_fmt(format_args!(" virtual"))?;
}
if let Some(overrides) = self.overrides.as_ref() {
f.write_fmt(format_args!(" {overrides}"))?;
}
f.write_fmt(format_args!(" {}", self.body))
}
}
#[derive(Clone, Debug, Deserialize, Serialize, PartialEq, PartialOrd, Eq, Ord, Hash)]
#[serde(rename_all = "camelCase")]
pub enum ModifierInvocationKind {
ModifierInvocation,
BaseConstructorSpecifier,
}
#[derive(Clone, Debug, Deserialize, Eq, Serialize, PartialEq, Hash)]
#[serde(rename_all = "camelCase")]
pub struct ModifierInvocation {
pub arguments: Option<Vec<Expression>>,
pub modifier_name: IdentifierPath,
pub src: String,
pub id: NodeID,
pub kind: Option<ModifierInvocationKind>,
}
impl Node for ModifierInvocation {
fn accept(&self, visitor: &mut impl ASTConstVisitor) -> Result<()> {
if visitor.visit_modifier_invocation(self)? {
self.modifier_name.accept(visitor)?;
if self.arguments.is_some() {
list_accept(self.arguments.as_ref().unwrap(), visitor)?;
}
}
visitor.end_visit_modifier_invocation(self)
}
}
impl Display for ModifierInvocation {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.write_fmt(format_args!("{}", self.modifier_name))?;
if let Some(arguments) = self.arguments.as_ref() {
f.write_str("(")?;
for (i, argument) in arguments.iter().enumerate() {
if i > 0 {
f.write_str(", ")?;
}
f.write_fmt(format_args!("{argument}"))?;
}
f.write_str(")")?;
}
Ok(())
}
}