forked from AkkomaGang/akkoma
36 lines
994 B
Elixir
36 lines
994 B
Elixir
|
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", [])
|
||
|
|
|> Enum.filter(fn %{"type" => t} -> t == "Emoji" end)
|
||
|
|
|> 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
|