Class: RuboCop::Cop::Generator

Inherits:
Object
  • Object
show all
Defined in:
lib/rubocop/cop/generator.rb,
lib/rubocop/cop/generator/require_file_injector.rb

Overview

Source and spec generator for new cops

This generator will take a cop name and generate a source file and spec file when given a valid qualified cop name.

Defined Under Namespace

Classes: RequireFileInjector

Constant Summary collapse

SOURCE_TEMPLATE = 
<<-RUBY.strip_indent
  # frozen_string_literal: true

  # TODO: when finished, run `rake generate_cops_documentation` to update the docs
  module RuboCop
    module Cop
      module %<department>s
        # TODO: Write cop description and example of bad / good code.
        #
        # @example
        #   # bad
        #   bad_method()
        #
        #   # bad
        #   bad_method(args)
        #
        #   # good
        #   good_method()
        #
        #   # good
        #   good_method(args)
        class %<cop_name>s < Cop
          # TODO: Implement the cop into here.
          #
          # In many cases, you can use a node matcher for matching node pattern.
          # See. https://github.com/bbatsov/rubocop/blob/master/lib/rubocop/node_pattern.rb
          #
          # For example
          MSG = 'Use `#good_method` instead of `#bad_method`.'.freeze

          def_node_matcher :bad_method?, <<-PATTERN
            (send nil? :bad_method ...)
          PATTERN

          def on_send(node)
            return unless bad_method?(node)
            add_offense(node)
          end
        end
      end
    end
  end
RUBY
SPEC_TEMPLATE = 
<<-SPEC.strip_indent
  # frozen_string_literal: true

  RSpec.describe RuboCop::Cop::%<department>s::%<cop_name>s do
    subject(:cop) { described_class.new(config) }

    let(:config) { RuboCop::Config.new }

    # TODO: Write test code
    #
    # For example
    it 'registers an offense when using `#bad_method`' do
      expect_offense(<<-RUBY.strip_indent)
        bad_method
        ^^^^^^^^^^ Use `#good_method` instead of `#bad_method`.
      RUBY
    end

    it 'does not register an offense when using `#good_method`' do
      expect_no_offenses(<<-RUBY.strip_indent)
        good_method
      RUBY
    end
  end
SPEC

Instance Method Summary collapse

Constructor Details

#initialize(name, output: $stdout) ⇒ Generator

Returns a new instance of Generator.

Raises:

  • (ArgumentError) 


80
81
82
83
84
85
86
 
# File 'lib/rubocop/cop/generator.rb', line 80

def initialize(name, output: $stdout)
  @badge = Badge.parse(name)
  @output = output
  return if badge.qualified?

  raise ArgumentError, 'Specify a cop name with Department/Name style'
end

Instance Method Details

#inject_config(config_file_path: 'config/enabled.yml') ⇒ Object 



103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
 
# File 'lib/rubocop/cop/generator.rb', line 103

def inject_config(config_file_path: 'config/enabled.yml')
  config = File.readlines(config_file_path)
  content = <<-YAML.strip_indent
    #{badge}:
      Description: 'TODO: Write a description of the cop.'
      Enabled: true

  YAML
  target_line = config.find.with_index(1) do |line, index|
    next if line =~ /^[\s#]/
    break index - 1 if badge.to_s < line
  end
  config.insert(target_line, content)
  File.write(config_file_path, config.join)
  output.puts <<-MESSAGE.strip_indent
    [modify] A configuration for the cop is added into #{config_file_path}.
             If you want to disable the cop by default, move the added config to config/disabled.yml
  MESSAGE
end

#inject_require(root_file_path: 'lib/rubocop.rb') ⇒ Object 



96
97
98
99
100
101
 
# File 'lib/rubocop/cop/generator.rb', line 96

def inject_require(root_file_path: 'lib/rubocop.rb')
  RequireFileInjector.new(
    source_path: source_path,
    root_file_path: root_file_path
  ).inject
end

#todo ⇒ Object 



123
124
125
126
127
128
129
130
131
 
# File 'lib/rubocop/cop/generator.rb', line 123

def todo
  <<-TODO.strip_indent
    Do 3 steps:
      1. Add an entry to the "New features" section in CHANGELOG.md,
         e.g. "Add new `#{badge}` cop. ([@your_id][])"
      2. Modify the description of #{badge} in config/enabled.yml
      3. Implement your new cop in the generated file!
  TODO
end

#write_source ⇒ Object 



88
89
90
 
# File 'lib/rubocop/cop/generator.rb', line 88

def write_source
  write_unless_file_exists(source_path, generated_source)
end

#write_spec ⇒ Object 



92
93
94
 
# File 'lib/rubocop/cop/generator.rb', line 92

def write_spec
  write_unless_file_exists(spec_path, generated_spec)
end