Class: RuboCop::Cop::Performance::Count
- Defined in:
- lib/rubocop/cop/performance/count.rb
Overview
This cop is used to identify usages of count
on an
Enumerable
and change them to size
.
Constant Summary collapse
- MSG =
'Use `count` instead of `%s...%s`.'
- SELECTORS =
[:reject, :select]
- COUNTERS =
[:count, :length, :size]
Constants included from Util
Util::ASGN_NODES, Util::EQUALS_ASGN_NODES, Util::OPERATOR_METHODS, Util::PROC_NEW_NODE, Util::SHORTHAND_ASGN_NODES
Instance Attribute Summary
Attributes inherited from Cop
#config, #corrections, #offenses, #processed_source
Instance Method Summary collapse
Methods inherited from Cop
#add_offense, all, #config_to_allow_offenses, #config_to_allow_offenses=, #cop_config, #cop_name, cop_name, cop_type, #correct, #debug?, #display_cop_names?, #display_style_guide?, inherited, #initialize, #join_force?, lint?, match?, #message, non_rails, qualified_cop_name, rails?, #relevant_file?, #style_guide_url
Methods included from AutocorrectLogic
#autocorrect?, #autocorrect_enabled?, #autocorrect_requested?, #support_autocorrect?
Methods included from IgnoredNode
#ignore_node, #ignored_node?, #part_of_ignored_node?
Methods included from Util
begins_its_line?, block_length, command?, comment_line?, const_name, directions, first_part_of_call_chain, lambda?, lambda_or_proc?, line_range, move_pos, numeric_range_size, on_node, operator?, parentheses?, proc?, range_with_surrounding_comma, range_with_surrounding_space, source_range, strip_quotes, within_node?
Methods included from PathUtil
hidden?, issue_deprecation_warning, match_path?, relative_path
Constructor Details
This class inherits a constructor from RuboCop::Cop::Cop
Instance Method Details
#autocorrect(node) ⇒ Object
52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
# File 'lib/rubocop/cop/performance/count.rb', line 52 def autocorrect(node) expression, first_method, second_method, = parse(node) return if first_method == :reject || second_method == :reject selector = if SELECTORS.include?(first_method) expression.loc.selector else expression.parent.loc.selector end lambda do |corrector| range = Parser::Source::Range.new(node.loc.expression.source_buffer, node.loc.dot.begin_pos, node.loc.expression.end_pos) corrector.remove(range) corrector.replace(selector, 'count') end end |
#on_send(node) ⇒ Object
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
# File 'lib/rubocop/cop/performance/count.rb', line 29 def on_send(node) expression, first_method, second_method, third_method = parse(node) return unless COUNTERS.include?(third_method) begin_pos = if SELECTORS.include?(first_method) return if second_method.is_a?(Symbol) expression.loc.selector.begin_pos else return unless SELECTORS.include?(second_method) expression.parent.loc.selector.begin_pos end return if node.parent && node.parent.block_type? range = Parser::Source::Range.new(node.loc.expression.source_buffer, begin_pos, node.loc.expression.end_pos) add_offense(node, range, format(MSG, first_method || second_method, third_method)) end |