Change Object ID to FlakeId

4 files changed, 47 insertions, 2 deletions

diff --git a/lib/pleroma/chat/message_reference.ex b/lib/pleroma/chat/message_reference.ex
index 89537d155..06fdb3401 100644
--- a/lib/pleroma/chat/message_reference.ex
+++ b/lib/pleroma/chat/message_reference.ex
@@ -20,7 +20,7 @@ defmodule Pleroma.Chat.MessageReference do
   @primary_key {:id, FlakeId.Ecto.Type, autogenerate: true}
 
   schema "chat_message_references" do
-    belongs_to(:object, Object)
+    belongs_to(:object, Object, type: FlakeId.Ecto.CompatType)
     belongs_to(:chat, Chat, type: FlakeId.Ecto.CompatType)
 
     field(:unread, :boolean, default: true)
diff --git a/lib/pleroma/delivery.ex b/lib/pleroma/delivery.ex
index 511d5cf58..b1ed6a8fd 100644
--- a/lib/pleroma/delivery.ex
+++ b/lib/pleroma/delivery.ex
@@ -15,7 +15,7 @@ defmodule Pleroma.Delivery do
 
   schema "deliveries" do
     belongs_to(:user, User, type: FlakeId.Ecto.CompatType)
-    belongs_to(:object, Object)
+    belongs_to(:object, Object, type: FlakeId.Ecto.CompatType)
   end
 
   def changeset(delivery, params \\ %{}) do
diff --git a/lib/pleroma/object.ex b/lib/pleroma/object.ex
index c3ea1b98b..cbb694408 100644
--- a/lib/pleroma/object.ex
+++ b/lib/pleroma/object.ex
@@ -22,6 +22,8 @@ defmodule Pleroma.Object do
 
   @type t() :: %__MODULE__{}
 
+  @primary_key {:id, FlakeId.Ecto.CompatType, autogenerate: true}
+
   @derive {Jason.Encoder, only: [:data]}
 
   @cachex Pleroma.Config.get([:cachex, :provider], Cachex)
diff --git a/priv/repo/migrations/20211218181632_change_object_id_to_flake.exs b/priv/repo/migrations/20211218181632_change_object_id_to_flake.exs
new file mode 100644
index 000000000..3eebe3ef2
--- /dev/null
+++ b/priv/repo/migrations/20211218181632_change_object_id_to_flake.exs
@@ -0,0 +1,43 @@
+defmodule Pleroma.Repo.Migrations.ChangeObjectIdToFlake do
+  @moduledoc """
+  Convert object IDs to FlakeIds.
+  Fortunately only a few tables have a foreign key to objects. Update them.
+  """
+  use Ecto.Migration
+
+  def up do
+    # Switch object IDs to FlakeIds
+    execute("""
+    alter table objects
+    drop constraint objects_pkey cascade,
+    alter column id drop default,
+    alter column id set data type uuid using cast( lpad( to_hex(id), 32, '0') as uuid),
+    add primary key (id)
+    """)
+
+    # Update chat message foreign key
+    execute("""
+    alter table chat_message_references
+    alter column object_id set data type uuid using cast( lpad( to_hex(object_id), 32, '0') as uuid),
+    add constraint chat_message_references_object_id_fkey foreign key (object_id) references objects(id) on delete cascade
+    """)
+
+    # Update delivery foreign key
+    execute("""
+    alter table deliveries
+    alter column object_id set data type uuid using cast( lpad( to_hex(object_id), 32, '0') as uuid),
+    add constraint deliveries_object_id_fkey foreign key (object_id) references objects(id)
+    """)
+
+    # Update hashtag many-to-many foreign key
+    execute("""
+    alter table hashtags_objects
+    alter column object_id set data type uuid using cast( lpad( to_hex(object_id), 32, '0') as uuid),
+    add constraint hashtags_objects_object_id_fkey foreign key (object_id) references objects(id) on delete cascade
+    """)
+  end
+
+  def down do
+    :ok
+  end
+end