forked from AkkomaGang/akkoma
64 lines
1.6 KiB
Elixir
64 lines
1.6 KiB
Elixir
# 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
|
|
# alias Pleroma.Web.ActivityPub.Transmogrifier
|
|
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
|