git.squeep.com Git - akkoma/blob - priv/repo/migrations/20200907092050_move_tokens_expiration_into_oban.exs

   1 defmodule Pleroma.Repo.Migrations.MoveTokensExpirationIntoOban do
   2   use Ecto.Migration
   3
   4   import Ecto.Query, only: [from: 2]
   5
   6   def change do
   7     Pleroma.Config.Oban.warn()
   8
   9     Supervisor.start_link([{Oban, Pleroma.Config.get(Oban)}],
  10       strategy: :one_for_one,
  11       name: Pleroma.Supervisor
  12     )
  13
  14     if Pleroma.Config.get([:oauth2, :clean_expired_tokens]) do
  15       from(t in Pleroma.Web.OAuth.Token, where: t.valid_until > ^NaiveDateTime.utc_now())
  16       |> Pleroma.Repo.stream()
  17       |> Stream.each(fn token ->
  18         Pleroma.Workers.PurgeExpiredToken.enqueue(%{
  19           token_id: token.id,
  20           valid_until: DateTime.from_naive!(token.valid_until, "Etc/UTC"),
  21           mod: Pleroma.Web.OAuth.Token
  22         })
  23       end)
  24       |> Stream.run()
  25     end
  26
  27     from(t in Pleroma.MFA.Token, where: t.valid_until > ^NaiveDateTime.utc_now())
  28     |> Pleroma.Repo.stream()
  29     |> Stream.each(fn token ->
  30       Pleroma.Workers.PurgeExpiredToken.enqueue(%{
  31         token_id: token.id,
  32         valid_until: DateTime.from_naive!(token.valid_until, "Etc/UTC"),
  33         mod: Pleroma.MFA.Token
  34       })
  35     end)
  36     |> Stream.run()
  37   end
  38 end