-
-
Notifications
You must be signed in to change notification settings - Fork 3k
/
when_then.rb
37 lines (32 loc) · 884 Bytes
/
when_then.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
# frozen_string_literal: true
module RuboCop
module Cop
module Style
# This cop checks for `when;` uses in `case` expressions.
#
# @example
# # bad
# case foo
# when 1; 'baz'
# when 2; 'bar'
# end
#
# # good
# case foo
# when 1 then 'baz'
# when 2 then 'bar'
# end
class WhenThen < Base
extend AutoCorrector
MSG = 'Do not use `when %<expression>s;`. Use `when %<expression>s then` instead.'
def on_when(node)
return if node.multiline? || node.then? || !node.body
message = format(MSG, expression: node.conditions.map(&:source).join(', '))
add_offense(node.loc.begin, message: message) do |corrector|
corrector.replace(node.loc.begin, ' then')
end
end
end
end
end
end