git.squeep.com Git - akkoma/blob - priv/repo/migrations/20190710125158_add_following_address_from_source_data.exs

   1 defmodule Pleroma.Repo.Migrations.AddFollowingAddressFromSourceData do
   2   alias Pleroma.User
   3   import Ecto.Query
   4   require Logger
   5   use Ecto.Migration
   6
   7   def change do
   8     query =
   9       User.Query.build(%{
  10         external: true,
  11         legacy_active: true,
  12         order_by: :id
  13       })
  14       |> select([u], struct(u, [:id, :ap_id, :info]))
  15
  16     Pleroma.Repo.stream(query)
  17     |> Enum.each(fn
  18       %{info: %{source_data: source_data}} = user ->
  19         Ecto.Changeset.cast(user, %{following_address: source_data["following"]}, [
  20           :following_address
  21         ])
  22         |> Pleroma.Repo.update()
  23
  24       user ->
  25         Logger.warn("User #{user.id} / #{user.nickname} does not seem to have source_data")
  26     end)
  27   end
  28 end