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
use crate::nodes::InnerNode;
use crate::nodes::InspectVec;
use crate::source::Range;
use crate::Node;
/// Represents a class definition (using a `class` keyword, `Class.new` is just a method call)
#[derive(Debug, Clone, PartialEq)]
pub struct Class {
/// Name of the class, `String("Foo")` for `class Foo; end`
pub name: Node,
/// Superclass. Can be an expression in cases like `class A < (obj.foo + 1); end`
///
/// `None` if no explicit superclass given (i.e. `class Foo; end`)
pub superclass: Option<Node>,
/// Body of the method, `None` if there's no body.
pub body: Option<Node>,
/// Location of the `class` keyword.
///
/// ```text
/// class Foo; end
/// ~~~~~
/// ```
pub keyword_l: Range,
/// Location of the `<` operator
///
/// ```text
/// class A < B; end
/// ~
/// ```
///
/// `None` if there's no explicit superclass given.
pub operator_l: Option<Range>,
/// Location of the `end` keyword.
///
/// ```text
/// class Foo; end
/// ~~~
/// ```
pub end_l: Range,
/// Location of the full expression
///
/// ```text
/// class Foo; end
/// ~~~~~~~~~~~~~~
/// ```
pub expression_l: Range,
}
impl InnerNode for Class {
fn expression(&self) -> &Range {
&self.expression_l
}
fn inspected_children(&self, indent: usize) -> Vec<String> {
let mut result = InspectVec::new(indent);
result.push_node(&self.name);
result.push_maybe_node_or_nil(&self.superclass);
result.push_maybe_node_or_nil(&self.body);
result.strings()
}
fn str_type(&self) -> &'static str {
"class"
}
fn print_with_locs(&self) {
println!("{}", self.inspect(0));
self.expression_l.print("expression");
self.end_l.print("end");
if let Some(range) = &self.operator_l {
range.print("operator");
}
self.keyword_l.print("keyword");
if let Some(node) = &self.body {
node.inner_ref().print_with_locs();
}
if let Some(node) = &self.superclass {
node.inner_ref().print_with_locs();
}
self.name.inner_ref().print_with_locs();
}
}