From 2826a91581f3ccd776eb0f34ef6d8a1575fd7648 Mon Sep 17 00:00:00 2001
From: Sam Bostock <sam.bostock@shopify.com>
Date: Sat, 4 Feb 2023 19:36:11 -0500
Subject: [PATCH] Patch missing `require 'yaml'` in file using YAML

---
 lib/rubocop/extension/generator/generator.rb | 5 +++++
 1 file changed, 5 insertions(+)

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