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
use crate::nodes::InnerNode;
use crate::nodes::InspectVec;
use crate::source::Range;
use crate::Node;
/// Represents method definition using `def` keyword (not on a singleton, see `Defs` node).
#[derive(Debug, Clone, PartialEq)]
pub struct Def {
/// Name of the method, `String("foo")` for `def foo; end`
pub name: String,
/// Arguments of a method, `None` if there's no arguments.
///
/// All information about parentheses around arguments is stored in this node.
pub args: Option<Node>,
/// Body of a method, `None` if there's no body.
pub body: Option<Node>,
/// Location of the `def` keyword.
///
/// ```text
/// def foo; end
/// ~~~
/// ```
pub keyword_l: Range,
/// Location of the method name.
///
/// ```text
/// def foo; end
/// ~~~
/// ```
pub name_l: Range,
/// Location of the `end` keyword.
///
/// ```text
/// def foo; end
/// ~~~
/// ```
///
/// `None` for endless method definition
pub end_l: Option<Range>,
/// Location of the `=` operator for endless method definition
///
/// ```text
/// def m() = 1
/// ~
/// ```
///
/// `None` for regular method definition
pub assignment_l: Option<Range>,
/// Location of the full expression
///
/// ```text
/// def m(a); foo; end
/// ~~~~~~~~~~~~~~~~~~
/// ```
pub expression_l: Range,
}
impl InnerNode for Def {
fn expression(&self) -> &Range {
&self.expression_l
}
fn inspected_children(&self, indent: usize) -> Vec<String> {
let mut result = InspectVec::new(indent);
result.push_str(&self.name);
result.push_maybe_node_or_nil(&self.args);
result.push_maybe_node_or_nil(&self.body);
result.strings()
}
fn str_type(&self) -> &'static str {
"def"
}
fn print_with_locs(&self) {
println!("{}", self.inspect(0));
self.expression_l.print("expression");
if let Some(range) = &self.assignment_l {
range.print("assignment");
}
if let Some(range) = &self.end_l {
range.print("end");
}
self.name_l.print("name");
self.keyword_l.print("keyword");
if let Some(node) = &self.body {
node.inner_ref().print_with_locs();
}
if let Some(node) = &self.args {
node.inner_ref().print_with_locs();
}
}
}