Oh no! More fixes!

lib/mix/tasks/fix_ap_users.ex

@@ -14,7 +14,7 @@ def run([]) do
 
     Enum.each(users, fn(user) ->
       IO.puts("Fetching #{user.nickname}")
-      Pleroma.Web.ActivityPub.Transmogrifier.upgrade_user_from_ap_id(user.ap_id)
+      Pleroma.Web.ActivityPub.Transmogrifier.upgrade_user_from_ap_id(user.ap_id, false)
     end)
   end
 end

lib/pleroma/web/activity_pub/transmogrifier.ex

@@ -197,18 +197,8 @@ def prepare_attachments(object) do
     |> Map.put("attachment", attachments)
   end
 
-  def upgrade_user_from_ap_id(ap_id) do
-    with %User{} = user <- User.get_by_ap_id(ap_id),
-         {:ok, data} <- ActivityPub.fetch_and_prepare_user_from_ap_id(ap_id) do
-      data = data
-      |> Map.put(:info, Map.merge(user.info, data[:info]))
-
+  defp user_upgrade_task(user) do
     old_follower_address = User.ap_followers(user)
-      {:ok, user} = User.upgrade_changeset(user, data)
-      |> Repo.update()
-
-      # This could potentially take a long time, do it in the background
-      Task.start(fn ->
     q  = from u in User,
     where: ^old_follower_address in u.following,
     update: [set: [following: fragment("array_replace(?,?,?)", u.following, ^old_follower_address, ^user.follower_address)]]
@@ -221,7 +211,25 @@ def upgrade_user_from_ap_id(ap_id) do
     where: a.id > ^since,
     update: [set: [recipients: fragment("array_replace(?,?,?)", a.recipients, ^old_follower_address, ^user.follower_address)]]
     Repo.update_all(q, [])
+  end
+
+  def upgrade_user_from_ap_id(ap_id, async \\ true) do
+    with %User{} = user <- User.get_by_ap_id(ap_id),
+         {:ok, data} <- ActivityPub.fetch_and_prepare_user_from_ap_id(ap_id) do
+      data = data
+      |> Map.put(:info, Map.merge(user.info, data[:info]))
+
+      {:ok, user} = User.upgrade_changeset(user, data)
+      |> Repo.update()
+
+      # This could potentially take a long time, do it in the background
+      if async do
+        Task.start(fn ->
+          user_upgrade_task(user)
         end)
+      else
+        user_upgrade_task(user)
+      end
 
       {:ok, user}
     else