akkoma/lib/pleroma/spc_fixes/spc_fixes.ex

# Pleroma: A lightweight social networking server
# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only

alias Pleroma.Repo
alias Pleroma.User
import Ecto.Query

defmodule Pleroma.SpcFixes do
  def upgrade_users do
    query =
      from(u in User,
        where: fragment("? like ?", u.ap_id, "https://shitposter.club/user/%")
      )

    {:ok, file} = File.read("lib/pleroma/spc_fixes/users_conversion.txt")

    mapping =
      file
      |> String.trim()
      |> String.split("\n")
      |> Enum.map(fn line ->
        line
        |> String.split("\t")
      end)
      |> Enum.reduce(%{}, fn [_id, old_ap_id, new_ap_id], acc ->
        Map.put(acc, old_ap_id, String.trim(new_ap_id))
      end)

    # First, refetch all the old users.
    _old_users =
      query
      |> Repo.all()
      |> Enum.each(fn user ->
        with ap_id when is_binary(ap_id) <- mapping[user.ap_id] do
          # This fetches and updates the user.
          User.get_or_fetch_by_ap_id(ap_id)
        end
      end)

    # Now, fix follow relationships.
    query =
      from(u in User,
        where: fragment("? like ?", u.ap_id, "https://shitposter.club/users/%")
      )

    query
    |> Repo.all()
    |> Enum.each(fn user ->
      old_follower_address = User.ap_followers(user)

      query =
        from(u in User,
          where: ^old_follower_address in u.following,
          update: [
            push: [following: ^user.follower_address]
          ]
        )

      Repo.update_all(query, [])
    end)
  end
end