class RuboCop::Cop::Layout::EmptyLinesAroundAccessModifier

def allowed_only_before_style?(node)
  if node.special_modifier?
    return false if next_line_empty?(node.last_line)
  end
  previous_line_empty?(node.first_line)
end

def autocorrect(node)
  lambda do |corrector|
    line = range_by_whole_lines(node.source_range)
    unless previous_line_empty?(node.first_line)
      corrector.insert_before(line, "\n")
    end
    correct_next_line_if_denied_style(corrector, node, line)
  end
end

def block_start?(line)
  return false unless @block_line
  line == @block_line + 1
end

def body_end?(line)
  return false unless @class_or_module_def_last_line
  line == @class_or_module_def_last_line - 1
end

def class_def?(line)
  return false unless @class_or_module_def_first_line
  line == @class_or_module_def_first_line + 1
end

def correct_next_line_if_denied_style(corrector, node, line)
  case style
  when :around
    unless next_line_empty?(node.last_line)
      corrector.insert_after(line, "\n")
    end
  when :only_before
    if next_line_empty?(node.last_line)
      range = next_empty_line_range(node)
      corrector.remove(range)
    end
  end
end

def empty_lines_around?(node)
  previous_line_empty?(node.first_line) &&
    next_line_empty?(node.last_line)
end

def initialize(config = nil, options = nil)
  super
  @block_line = nil
end

def message(node)
  case style
  when :around
    message_for_around_style(node)
  when :only_before
    message_for_only_before_style(node)
  end
end

def message_for_around_style(node)
  send_line = node.first_line
  if block_start?(send_line) ||
     class_def?(send_line)
    format(MSG_AFTER, modifier: node.loc.selector.source)
  else
    format(MSG_BEFORE_AND_AFTER, modifier: node.loc.selector.source)
  end
end

def message_for_only_before_style(node)
  modifier = node.loc.selector.source
  if next_line_empty?(node.last_line)
    format(MSG_AFTER_FOR_ONLY_BEFORE, modifier: modifier)
  else
    format(MSG_BEFORE_FOR_ONLY_BEFORE, modifier: modifier)
  end
end

def next_empty_line_range(node)
  source_range(processed_source.buffer, node.last_line + 1, 0)
end

def next_line_empty?(last_send_line)
  next_line = processed_source[last_send_line]
  body_end?(last_send_line) || next_line.blank?
end

def on_block(node)
  @block_line = node.source_range.first_line
end

def on_class(node)
  @class_or_module_def_first_line = if node.parent_class
                                      node.parent_class.first_line
                                    else
                                      node.source_range.first_line
                                    end
  @class_or_module_def_last_line = node.source_range.last_line
end

def on_module(node)
  @class_or_module_def_first_line = node.source_range.first_line
  @class_or_module_def_last_line = node.source_range.last_line
end

def on_sclass(node)
  @class_or_module_def_first_line =
    node.identifier.source_range.first_line
  @class_or_module_def_last_line =
    node.source_range.last_line
end

def on_send(node)
  return unless node.bare_access_modifier?
  case style
  when :around
    return if empty_lines_around?(node)
  when :only_before
    return if allowed_only_before_style?(node)
  end
  add_offense(node)
end

def previous_line_empty?(send_line)
  previous_line = previous_line_ignoring_comments(processed_source,
                                                  send_line)
  block_start?(send_line) ||
    class_def?(send_line) ||
    previous_line.blank?
end

def previous_line_ignoring_comments(processed_source, send_line)
  processed_source[0..send_line - 2].reverse.find do |line|
    !comment_line?(line)
  end
end