From d16680c8ece405c11682f2d0e4fa1d2a31ee5669 Mon Sep 17 00:00:00 2001
From: Justin Edmund <justin@jedmund.com>
Date: Sun, 12 Jan 2025 19:03:39 -0800
Subject: [PATCH] Rebuild search indices in post deploy

---
 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 c24086e..52e0443 100644
--- a/lib/granblue/post_deployment_manager.rb
+++ b/lib/granblue/post_deployment_manager.rb
@@ -19,6 +19,8 @@ class PostDeploymentManager
     import_new_data
     display_import_summary
     download_images
+    rebuild_search_indices
+    display_completion_message
   end
 
   private
@@ -44,6 +46,15 @@ class PostDeploymentManager
     end
   end
 
+  def rebuild_search_indices
+    log_step "\nRebuilding search indices..."
+
+    [Character, Summon, Weapon, Job].each do |model|
+      log_verbose "Rebuilding search index for #{model.name}..."
+      PgSearch::Multisearch.rebuild(model)
+    end
+  end
+
   def display_import_summary
     log_step "\nImport Summary:"
     display_record_summary("New", @new_records)
@@ -104,6 +115,14 @@ class PostDeploymentManager
     puts e.backtrace.take(5) if @verbose
   end
 
+  def display_completion_message
+    if @test_mode
+      log_step "\n✓ Test run completed successfully!"
+    else
+      log_step "\n✓ Post-deployment tasks completed successfully!"
+    end
+  end
+
   def all_records_empty?
     @new_records.values.all?(&:empty?) && @updated_records.values.all?(&:empty?)
   end