diff --git a/lib/rubocop/extension/generator/generator.rb b/lib/rubocop/extension/generator/generator.rb index c9c30d7..a7d89d7 100644 --- a/lib/rubocop/extension/generator/generator.rb +++ b/lib/rubocop/extension/generator/generator.rb @@ -85,6 +85,11 @@ def self.defaults! - lib/#{name}.rb YML + patch "lib/#{dirname}.rb", %r{^require.*/version.$}, <<~RUBY.chomp + \\0 + require 'yaml' + RUBY + patch "lib/#{dirname}.rb", /^ end\nend/, <<~RUBY PROJECT_ROOT = Pathname.new(__dir__).parent.parent.expand_path.freeze CONFIG_DEFAULT = PROJECT_ROOT.join('config', 'default.yml').freeze