Sha256: 4154bc56022ca6c3d956a03d29d74cf06be7121d4e05f29838c179ebde6de596

Contents?: true

Size: 1.87 KB

Versions: 20

Compression:

Stored size: 1.87 KB

Contents

# frozen_string_literal: true

module RuboCop
  module Cop
    module Performance
      # This cop identifies places where `reverse.first(n)` and `reverse.first`
      # can be replaced by `last(n).reverse` and `last`.
      #
      # @example
      #
      #   # bad
      #   array.reverse.first(5)
      #   array.reverse.first
      #
      #   # good
      #   array.last(5).reverse
      #   array.last
      #
      class ReverseFirst < Base
        include RangeHelp
        extend AutoCorrector

        MSG = 'Use `%<good_method>s` instead of `%<bad_method>s`.'
        RESTRICT_ON_SEND = %i[first].freeze

        def_node_matcher :reverse_first_candidate?, <<~PATTERN
          (send $(send _ :reverse) :first (int _)?)
        PATTERN

        def on_send(node)
          reverse_first_candidate?(node) do |receiver|
            range = correction_range(receiver, node)
            message = build_message(node)

            add_offense(range, message: message) do |corrector|
              replacement = build_good_method(node)

              corrector.replace(range, replacement)
            end
          end
        end

        private

        def correction_range(receiver, node)
          range_between(receiver.loc.selector.begin_pos, node.loc.expression.end_pos)
        end

        def build_message(node)
          good_method = build_good_method(node)
          bad_method = build_bad_method(node)
          format(MSG, good_method: good_method, bad_method: bad_method)
        end

        def build_good_method(node)
          if node.arguments?
            "last(#{node.arguments.first.source}).reverse"
          else
            'last'
          end
        end

        def build_bad_method(node)
          if node.arguments?
            "reverse.first(#{node.arguments.first.source})"
          else
            'reverse.first'
          end
        end
      end
    end
  end
end

Version data entries

20 entries across 20 versions & 3 rubygems

Version Path
scrapbook-0.3.2 vendor/ruby/2.7.0/gems/rubocop-performance-1.13.3/lib/rubocop/cop/performance/reverse_first.rb
scrapbook-0.3.1 vendor/ruby/2.7.0/gems/rubocop-performance-1.13.3/lib/rubocop/cop/performance/reverse_first.rb
op_connect-0.1.2 vendor/bundle/ruby/3.1.0/gems/rubocop-performance-1.13.3/lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.13.3 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.13.2 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.13.1 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.13.0 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.12.0 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.11.5 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.11.4 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.11.3 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.11.2 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.11.1 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.11.0 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.10.2 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.10.1 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.10.0 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.9.2 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.9.1 lib/rubocop/cop/performance/reverse_first.rb
rubocop-performance-1.9.0 lib/rubocop/cop/performance/reverse_first.rb