forked from rubocop/rubocop
/
statement_modifier.rb
73 lines (57 loc) · 2.21 KB
/
statement_modifier.rb
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
# frozen_string_literal: true
module RuboCop
module Cop
# Common functionality for modifier cops.
module StatementModifier
private
def single_line_as_modifier?(node)
return false if non_eligible_node?(node) ||
non_eligible_body?(node.body) ||
non_eligible_condition?(node.condition)
modifier_fits_on_single_line?(node)
end
def non_eligible_node?(node)
node.nonempty_line_count > 3 ||
!node.modifier_form? &&
processed_source.commented?(node.loc.end)
end
def non_eligible_body?(body)
body.nil? ||
body.empty_source? ||
body.begin_type? ||
processed_source.commented?(body.source_range)
end
def non_eligible_condition?(condition)
condition.each_node.any?(&:lvasgn_type?)
end
def modifier_fits_on_single_line?(node)
return true unless max_line_length
modifier_length = length_in_modifier_form(node, node.condition,
node.body.source_length)
modifier_length <= max_line_length
end
def length_in_modifier_form(node, cond, body_length)
keyword = node.loc.keyword
indentation = keyword.column * indentation_multiplier
kw_length = keyword.size
cond_length = cond.source_range.size
space = 1
indentation + body_length + space + kw_length + space + cond_length
end
def max_line_length
return unless config.for_cop('Layout/LineLength')['Enabled']
config.for_cop('Layout/LineLength')['Max']
end
def indentation_multiplier
indentation_style_config = config.for_cop('Layout/IndentationStyle')
return 1 if indentation_style_config['EnforcedStyle'] == 'spaces'
default_configuration = RuboCop::ConfigLoader.default_configuration
indentation_style_config['IndentationWidth'] ||
config.for_cop('Layout/IndentationWidth')['Width'] ||
default_configuration
.for_cop('Layout/IndentationStyle')['IndentationWidth'] ||
default_configuration.for_cop('Layout/IndentationWidth')['Width']
end
end
end
end