Allow custom gemfile vars in Customize

thoughtbot · Feb 29, 2024 · e5d2cec · e5d2cec
1 parent 1495c7b
commit e5d2cec
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/lib/appraisal/appraisal.rb b/lib/appraisal/appraisal.rb
@@ -63,7 +63,7 @@ def git_source(*args, &block)
 
     def write_gemfile
       File.open(gemfile_path, "w") do |file|
-        signature = Customize.heading || "This file was generated by Appraisal"
+        signature = Customize.heading(self) || "This file was generated by Appraisal"
         file.puts([comment_lines(signature), quoted_gemfile].join("\n\n"))
       end
     end

diff --git a/lib/appraisal/customize.rb b/lib/appraisal/customize.rb
@@ -5,12 +5,22 @@ def initialize(heading: nil, single_quotes: false)
       @@single_quotes = single_quotes
     end
 
-    def self.heading
-      @@heading ||= nil
+    def self.heading(gemfile)
+      customize(@@heading, gemfile)
     end
 
     def self.single_quotes
       @@single_quotes ||= false
     end
+
+    private
+
+    def self.customize(heading, gemfile)
+      return nil unless heading
+
+      heading.gsub('%{gemfile}', gemfile.send('gemfile_name')).
+        gsub('%{gemfile_path}', gemfile.gemfile_path).
+        gsub('%{relative_gemfile_path}', gemfile.relative_gemfile_path)
+    end
   end
 end