class RuboCop::Cop::Style::WhenThen

def autocorrect(node)
  lambda do |corrector|
    corrector.replace(node.loc.begin, ' then')
  end
end

def on_when(node)
  return if node.multiline? || node.then? || !node.body
  add_offense(node, :begin)
end