forked from AkkomaGang/akkoma
113 lines
2.9 KiB
Elixir
113 lines
2.9 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.Activity
|
|
alias Pleroma.Object
|
|
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 of old ap_id to new ap_id and vice reversa
|
|
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 ->
|
|
acc
|
|
|> Map.put(String.trim(old_ap_id), String.trim(new_ap_id))
|
|
|> Map.put(String.trim(new_ap_id), String.trim(old_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)
|
|
|
|
# Fix users
|
|
query =
|
|
from(u in User,
|
|
where: ^old_follower_address in u.following,
|
|
update: [
|
|
push: [following: ^user.follower_address]
|
|
]
|
|
)
|
|
|
|
Repo.update_all(query, [])
|
|
|
|
# Fix activities
|
|
query =
|
|
from(a in Activity,
|
|
where: fragment("?->>'actor' = ?", a.data, ^mapping[user.ap_id]),
|
|
update: [
|
|
set: [
|
|
data:
|
|
fragment(
|
|
"jsonb_set(jsonb_set(?, '{actor}', ?), '{to}', (?->'to')::jsonb || ?)",
|
|
a.data,
|
|
^user.ap_id,
|
|
a.data,
|
|
^[user.follower_address]
|
|
),
|
|
actor: ^user.ap_id
|
|
],
|
|
push: [
|
|
recipients: ^user.follower_address
|
|
]
|
|
]
|
|
)
|
|
|
|
Repo.update_all(query, [])
|
|
|
|
# Fix objects
|
|
query =
|
|
from(a in Object,
|
|
where: fragment("?->>'actor' = ?", a.data, ^mapping[user.ap_id]),
|
|
update: [
|
|
set: [
|
|
data:
|
|
fragment(
|
|
"jsonb_set(jsonb_set(?, '{actor}', ?), '{to}', (?->'to')::jsonb || ?)",
|
|
a.data,
|
|
^user.ap_id,
|
|
a.data,
|
|
^[user.follower_address]
|
|
)
|
|
]
|
|
]
|
|
)
|
|
|
|
Repo.update_all(query, [])
|
|
end)
|
|
end
|
|
end
|