1 files changed, 68 insertions, 0 deletions
diff --git a/priv/repo/migrations/20211218181647_combine_activities_and_objects.exs b/priv/repo/migrations/20211218181647_combine_activities_and_objects.exs
index 7fabf72f3..5de82c790 100644
--- a/priv/repo/migrations/20211218181647_combine_activities_and_objects.exs
+++ b/priv/repo/migrations/20211218181647_combine_activities_and_objects.exs
@@ -1,10 +1,18 @@
 defmodule Pleroma.Repo.Migrations.CombineActivitiesAndObjects do
   use Ecto.Migration
+  require Integer
+
+  alias Pleroma.Clippy
+  alias Pleroma.Repo
+
+  import Ecto.Query
 
   @function_name "update_status_visibility_counter_cache"
   @trigger_name "status_visibility_counter_cache_trigger"
 
   def up do
+    clippy = start_clippy_heartbeats()
+
     # Lock both tables to avoid a running server meddling with our transaction
     execute("LOCK TABLE activities")
     execute("LOCK TABLE objects")
@@ -86,6 +94,10 @@ defmodule Pleroma.Repo.Migrations.CombineActivitiesAndObjects do
 
     execute("drop function if exists thread_visibility(actor varchar, activity_id varchar)")
     execute(update_thread_visibility())
+
+    flush()
+
+    stop_clippy_heartbeats(clippy)
   end
 
   def down do
@@ -156,4 +168,60 @@ defmodule Pleroma.Repo.Migrations.CombineActivitiesAndObjects do
     $$ LANGUAGE plpgsql IMMUTABLE;
     """
   end
+
+  defp start_clippy_heartbeats() do
+    count = from(a in "activities", select: count(a.id)) |> Repo.one!()
+
+    if count > 5000 do
+      heartbeat_interval = :timer.minutes(2) + :timer.seconds(30)
+
+      all_tips =
+        Clippy.tips() ++
+          [
+            "The migration is still running, maybe it's time for another “tea”?",
+            "Happy rabbits practice a cute behavior known as a\n“binky:” they jump up in the air\nand twist\nand spin around!",
+            "Nothing and everything.\n\nI still work.",
+            "Pleroma runs on a Raspberry Pi!\n\n  … but this migration will take forever if you\nactually run on a raspberry pi",
+            "Status? Stati? Post? Note? Toot?\nRepeat? Reboost? Boost? Retweet? Retoot??\n\nI-I'm confused."
+          ]
+
+      heartbeat = fn heartbeat, runs, all_tips, tips ->
+        tips =
+          if Integer.is_even(runs) do
+            tips = if tips == [], do: all_tips, else: tips
+            [tip | tips] = Enum.shuffle(tips)
+            Clippy.puts(tip)
+            tips
+          else
+            IO.puts(
+              "\n -- #{DateTime.to_string(DateTime.utc_now())} Migration still running, please wait…\n"
+            )
+
+            tips
+          end
+
+        :timer.sleep(heartbeat_interval)
+        heartbeat.(heartbeat, runs + 1, all_tips, tips)
+      end
+
+      Clippy.puts([
+        [:red, :bright, "It looks like you are running an older instance!"],
+        [""],
+        [:bright, "This migration may take a long time", :reset, " -- so you probably should"],
+        ["go drink a cofe, or a tea, or a beer, a whiskey, a vodka,"],
+        ["while it runs to deal with your temporary fediverse pause!"]
+      ])
+
+      :timer.sleep(heartbeat_interval)
+      spawn_link(fn -> heartbeat.(heartbeat, 1, all_tips, []) end)
+    end
+  end
+
+  defp stop_clippy_heartbeats(pid) do
+    if pid do
+      Process.unlink(pid)
+      Process.exit(pid, :kill)
+      Clippy.puts([[:green, :bright, "Hurray!!", "", "", "Migration completed!"]])
+    end
+  end
 end