From df7454af4ea565ce2ccb2b95930db3a63226f7cd Mon Sep 17 00:00:00 2001
From: Justin Edmund <justin@jedmund.com>
Date: Mon, 13 Jan 2025 06:36:01 -0800
Subject: [PATCH] Add migration to post-deploy script (#151)

---
 lib/granblue/post_deployment_manager.rb | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/lib/granblue/post_deployment_manager.rb b/lib/granblue/post_deployment_manager.rb
index 928d618..5150772 100644
--- a/lib/granblue/post_deployment_manager.rb
+++ b/lib/granblue/post_deployment_manager.rb
@@ -20,6 +20,7 @@ class PostDeploymentManager
   end
 
   def run
+    migrate_database
     import_new_data
     display_import_summary
     download_images
@@ -29,6 +30,24 @@ class PostDeploymentManager
 
   private
 
+  def migrate_database
+    log_header 'Running database migrations...', '-'
+    puts "\n"
+    if @test_mode
+      log_step "TEST MODE: Would run pending migrations..."
+    else
+      ActiveRecord::Migration.verbose = @verbose
+      version = ActiveRecord::Migrator.current_version
+      ActiveRecord::Tasks::DatabaseTasks.migrate
+      new_version = ActiveRecord::Migrator.current_version
+      if version == new_version
+        log_step "No pending migrations."
+      else
+        log_step "Migrated from version #{version} to #{new_version}"
+      end
+    end
+  end
+
   def import_new_data
     log_header 'Importing new data...'
     puts "\n"