class RuboCop::Cop::Style::MapCompactWithConditionalBlock


array.reject { |e| some_condition? }
# good
array.select { |e| some_condition? }
# good
end.compact
e if some_condition?
array.map do |e|
# bad
end.compact
e
next if some_condition?
array.map do |e|
# bad
end.compact
end
next
else
e
if some_condition?
array.map do |e|
# bad
array.filter_map { |e| some_condition? ? e : next }
# bad
array.map { |e| some_condition? ? e : next }.compact
# bad
@example
This cop also handles ‘filter_map { … }`, similar to `map { … }.compact`.
Prefer `select` or `reject` over `map { … }.compact`.

def current(node)

def current(node)
  if node.method?(:compact)
    map_or_filter_map_method = node.children.first
    "#{map_or_filter_map_method.method_name} { ... }.compact"
  else
    'filter_map { ... }'
  end
end

def filter_map_range(node)

def filter_map_range(node)
  node.loc.selector.join(node.parent.source_range.end)
end

def inspect(node, block_argument_node, condition_node, return_value_node, range)

def inspect(node, block_argument_node, condition_node, return_value_node, range)
  return unless returns_block_argument?(block_argument_node, return_value_node)
  return if condition_node.parent.elsif?
  method = truthy_branch?(return_value_node) ? 'select' : 'reject'
  current = current(node)
  add_offense(range, message: format(MSG, current: current, method: method)) do |corrector|
    return if part_of_ignored_node?(node) || ignored_node?(node)
    corrector.replace(
      range, "#{method} { |#{block_argument_node.source}| #{condition_node.source} }"
    )
    ignore_node(node)
  end
end

def map_with_compact_range(node)

def map_with_compact_range(node)
  node.receiver.send_node.loc.selector.begin.join(node.source_range.end)
end

def on_send(node)

def on_send(node)
  map_candidate = node.children.first
  if (block_argument, condition, return_value = conditional_block(map_candidate))
    return unless node.method?(:compact) && node.arguments.empty?
    range = map_with_compact_range(node)
  elsif (block_argument, condition, return_value = conditional_block(node.parent))
    return unless node.method?(:filter_map)
    range = filter_map_range(node)
  else
    return
  end
  inspect(node, block_argument, condition, return_value, range)
end

def returns_block_argument?(block_argument_node, return_value_node)

def returns_block_argument?(block_argument_node, return_value_node)
  block_argument_node.name == return_value_node.children.first
end

def truthy_branch?(node)

def truthy_branch?(node)
  if node.parent.begin_type?
    truthy_branch_for_guard?(node)
  elsif node.parent.next_type?
    truthy_branch_for_if?(node.parent)
  else
    truthy_branch_for_if?(node)
  end
end

def truthy_branch_for_guard?(node)

def truthy_branch_for_guard?(node)
  if_node = node.left_sibling
  if if_node.if?
    if_node.if_branch.arguments.any?
  else
    if_node.if_branch.arguments.none?
  end
end

def truthy_branch_for_if?(node)

def truthy_branch_for_if?(node)
  if_node = node.parent
  if if_node.if? || if_node.ternary?
    if_node.if_branch == node
  elsif if_node.unless?
    if_node.else_branch == node
  end
end