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
|
# frozen_string_literal: true
module RuboCop
module AST
# A node extension for `when` nodes. This will be used in place of a plain
# node when the builder constructs the AST, making its methods available
# to all `when` nodes within RuboCop.
class WhenNode < Node
# Returns an array of all the conditions in the `when` branch.
#
# @return [Array<Node>] an array of condition nodes
def conditions
node_parts[0...-1]
end
# @deprecated Use `conditions.each`
def each_condition(&block)
return conditions.to_enum(__method__) unless block_given?
conditions.each(&block)
self
end
# Returns the index of the `when` branch within the `case` statement.
#
# @return [Integer] the index of the `when` branch
def branch_index
parent.when_branches.index(self)
end
# Checks whether the `when` node has a `then` keyword.
#
# @return [Boolean] whether the `when` node has a `then` keyword
def then?
loc.begin&.is?('then')
end
# Returns the body of the `when` node.
#
# @return [Node, nil] the body of the `when` node
def body
node_parts[-1]
end
end
end
end
|
