class RuboCop::Cop::Performance::ArraySemiInfiniteRangeSlice

def correction(receiver, range_node)
  method_call = if range_node.begin
                  "drop(#{range_node.begin.value})"
                elsif range_node.irange_type?
                  "take(#{range_node.end.value + 1})"
                else
                  "take(#{range_node.end.value})"
                end
  "#{receiver.source}.#{method_call}"
end