summaryrefslogtreecommitdiff
path: root/priv/repo/migrations/20200406100225_users_add_emoji.exs
blob: f248108defbd11d8cc1080cd7e3298eba38af5b1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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" => "Emoji"} -> true
          _ -> false
        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