class RuboCop::Cop::Layout::EmptyLineAfterGuardClause

end
end
return if need_return?
do_something
if something?
def foo
# also good
end
bar
return if something_different?
return if something?
def foo
# good
end
bar
return if need_return?
def foo
# good
end
bar
return if need_return?
def foo
# bad
@example
This cop enforces empty line after guard clause

def autocorrect(node) 

def autocorrect(node)
  lambda do |corrector|
    node_range = if node.respond_to?(:heredoc?) && node.heredoc?
                   range_by_whole_lines(node.loc.heredoc_body)
                 else
                   range_by_whole_lines(node.source_range)
                 end
    corrector.insert_after(node_range, "\n")
  end
end

def contains_guard_clause?(node) 

def contains_guard_clause?(node)
  node.if_branch&.guard_clause?
end

def correct_style?(node) 

def correct_style?(node)
  !contains_guard_clause?(node) ||
    next_line_rescue_or_ensure?(node) ||
    next_sibling_parent_empty_or_else?(node) ||
    next_sibling_empty_or_guard_clause?(node)
end

def heredoc_line(node, heredoc_node) 

def heredoc_line(node, heredoc_node)
  heredoc_body = heredoc_node.loc.heredoc_body
  num_of_heredoc_lines =
    heredoc_body.last_line - heredoc_body.first_line
  node.last_line + num_of_heredoc_lines + END_OF_HEREDOC_LINE
end

def last_argument(node) 

def last_argument(node)
  node.if_branch.children.last
end

def last_argument_is_heredoc?(node) 

def last_argument_is_heredoc?(node)
  last_children = node.if_branch
  return false unless last_children&.send_type?
  last_argument = last_argument(node)
  last_argument.respond_to?(:heredoc?) && last_argument.heredoc?
end

def next_line_empty?(line) 

def next_line_empty?(line)
  processed_source[line].blank?
end

def next_line_rescue_or_ensure?(node) 

def next_line_rescue_or_ensure?(node)
  parent = node.parent
  parent.nil? || parent.rescue_type? || parent.ensure_type?
end

def next_sibling_empty_or_guard_clause?(node) 

def next_sibling_empty_or_guard_clause?(node)
  next_sibling = node.parent.children[node.sibling_index + 1]
  return true if next_sibling.nil?
  next_sibling.if_type? && contains_guard_clause?(next_sibling)
end

def next_sibling_parent_empty_or_else?(node) 

def next_sibling_parent_empty_or_else?(node)
  next_sibling = node.parent.children[node.sibling_index + 1]
  return true if next_sibling.nil?
  parent = next_sibling.parent
  parent&.if_type? && parent&.else?
end

def offense_location(node) 

def offense_location(node)
  if node.loc.respond_to?(:end) && node.loc.end
    :end
  else
    :expression
  end
end

def on_if(node) 

def on_if(node)
  return if correct_style?(node)
  if node.modifier_form? && last_argument_is_heredoc?(node)
    heredoc_node = last_argument(node)
    return if next_line_empty?(heredoc_line(node, heredoc_node))
    add_offense(heredoc_node, location: :heredoc_end)
  else
    return if next_line_empty?(node.last_line)
    add_offense(node, location: offense_location(node))
  end
end