Add migration version to install/update generator templates (#426)

bensheldon · Oct 18, 2021 · 71b52d6 · 71b52d6
1 parent 377cde2
commit 71b52d6
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 2 deletions.
diff --git a/lib/generators/good_job/install_generator.rb b/lib/generators/good_job/install_generator.rb
@@ -19,5 +19,11 @@ class InstallGenerator < Rails::Generators::Base
     def create_migration_file
       migration_template 'migrations/create_good_jobs.rb.erb', File.join(db_migrate_path, "create_good_jobs.rb")
     end
+
+    private
+
+    def migration_version
+      "[#{ActiveRecord::VERSION::STRING.to_f}]"
+    end
   end
 end
diff --git a/lib/generators/good_job/templates/install/migrations/create_good_jobs.rb.erb b/lib/generators/good_job/templates/install/migrations/create_good_jobs.rb.erb
@@ -1,5 +1,5 @@
 # frozen_string_literal: true
-class CreateGoodJobs < ActiveRecord::Migration[5.2]
+class CreateGoodJobs < ActiveRecord::Migration<%= migration_version %>
   def change
     enable_extension 'pgcrypto'
 

diff --git a/.../update/migrations/01_create_good_jobs.rb → ...ate/migrations/01_create_good_jobs.rb.erb b/.../update/migrations/01_create_good_jobs.rb → ...ate/migrations/01_create_good_jobs.rb.erb
@@ -1,5 +1,5 @@
 # frozen_string_literal: true
-class CreateGoodJobs < ActiveRecord::Migration[5.2]
+class CreateGoodJobs < ActiveRecord::Migration<%= migration_version %>
   def change
     enable_extension 'pgcrypto'
 

diff --git a/lib/generators/good_job/update_generator.rb b/lib/generators/good_job/update_generator.rb
@@ -24,5 +24,11 @@ def update_migration_files
         migration_template "migrations/#{template_file}", File.join(db_migrate_path, destination_file), skip: true
       end
     end
+
+    private
+
+    def migration_version
+      "[#{ActiveRecord::VERSION::STRING.to_f}]"
+    end
   end
 end