2020-04-03 05:03:32 -06:00
|
|
|
defmodule Pleroma.Repo.Migrations.UsersPopulateEmoji do
|
|
|
|
use Ecto.Migration
|
|
|
|
|
|
|
|
import Ecto.Query
|
|
|
|
|
|
|
|
alias Pleroma.User
|
|
|
|
alias Pleroma.Repo
|
|
|
|
|
|
|
|
def up do
|
|
|
|
execute("ALTER TABLE users ALTER COLUMN emoji SET DEFAULT '{}'::jsonb")
|
|
|
|
execute("UPDATE users SET emoji = DEFAULT WHERE emoji = '[]'::jsonb")
|
|
|
|
|
|
|
|
from(u in User)
|
|
|
|
|> select([u], struct(u, [:id, :ap_id, :source_data]))
|
|
|
|
|> Repo.stream()
|
|
|
|
|> Enum.each(fn user ->
|
|
|
|
emoji =
|
|
|
|
user.source_data
|
|
|
|
|> Map.get("tag", [])
|
2020-04-17 15:55:56 -06:00
|
|
|
|> Enum.filter(fn
|
2020-04-17 16:48:40 -06:00
|
|
|
%{"type" => "Emoji"} -> true
|
2020-04-17 15:55:56 -06:00
|
|
|
_ -> false
|
|
|
|
end)
|
2020-04-03 05:03:32 -06:00
|
|
|
|> Enum.reduce(%{}, fn %{"icon" => %{"url" => url}, "name" => name}, acc ->
|
|
|
|
Map.put(acc, String.trim(name, ":"), url)
|
|
|
|
end)
|
|
|
|
|
|
|
|
user
|
|
|
|
|> Ecto.Changeset.cast(%{emoji: emoji}, [:emoji])
|
|
|
|
|> Repo.update()
|
|
|
|
end)
|
|
|
|
end
|
|
|
|
|
|
|
|
def down do
|
|
|
|
execute("ALTER TABLE users ALTER COLUMN emoji SET DEFAULT '[]'::jsonb")
|
|
|
|
execute("UPDATE users SET emoji = DEFAULT WHERE emoji = '{}'::jsonb")
|
|
|
|
end
|
|
|
|
end
|