git.squeep.com Git - akkoma/blob - priv/repo/migrations/20220905011454_generate_unset_user_keys.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Repo.Migrations.GenerateUnsetUserKeys do
   6   use Ecto.Migration
   7   import Ecto.Query
   8   alias Pleroma.Keys
   9   alias Pleroma.Repo
  10   alias Pleroma.User
  11
  12   def change do
  13     query =
  14       from(u in User,
  15         where: u.local == true,
  16         where: is_nil(u.keys),
  17         select: u.id
  18       )
  19
  20     Repo.stream(query)
  21     |> Enum.each(fn user ->
  22       with {:ok, pem} <- Keys.generate_rsa_pem() do
  23         Ecto.Changeset.cast(%User{id: user}, %{keys: pem}, [:keys])
  24         |> Repo.update(returning: false)
  25       end
  26     end)
  27   end
  28 end