git.squeep.com Git - akkoma/blob - lib/mix/tasks/pleroma/ecto/rollback.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-onl
   4
   5 defmodule Mix.Tasks.Pleroma.Ecto.Rollback do
   6   use Mix.Task
   7   require Logger
   8   @shortdoc "Wrapper on `ecto.rollback` task"
   9
  10   @aliases [
  11     n: :step,
  12     v: :to
  13   ]
  14
  15   @switches [
  16     all: :boolean,
  17     step: :integer,
  18     to: :integer,
  19     start: :boolean,
  20     quiet: :boolean,
  21     log_sql: :boolean,
  22     migrations_path: :string
  23   ]
  24
  25   @moduledoc """
  26   Changes `Logger` level to `:info` before start rollback.
  27   Changes level back when rollback ends.
  28
  29   ## Start rollback
  30
  31       mix pleroma.ecto.rollback
  32
  33   Options:
  34     - see https://hexdocs.pm/ecto/2.0.0/Mix.Tasks.Ecto.Rollback.html
  35   """
  36
  37   @impl true
  38   def run(args \\ []) do
  39     {opts, _} = OptionParser.parse!(args, strict: @switches, aliases: @aliases)
  40
  41     opts =
  42       if opts[:to] || opts[:step] || opts[:all],
  43         do: opts,
  44         else: Keyword.put(opts, :step, 1)
  45
  46     opts =
  47       if opts[:quiet],
  48         do: Keyword.merge(opts, log: false, log_sql: false),
  49         else: opts
  50
  51     path = Mix.Tasks.Pleroma.Ecto.ensure_migrations_path(Pleroma.Repo, opts)
  52
  53     level = Logger.level()
  54     Logger.configure(level: :info)
  55
  56     if Pleroma.Config.get(:env) == :test do
  57       Logger.info("Rollback succesfully")
  58     else
  59       {:ok, _, _} =
  60         Ecto.Migrator.with_repo(Pleroma.Repo, &Ecto.Migrator.run(&1, path, :down, opts))
  61     end
  62
  63     Logger.configure(level: level)
  64   end
  65 end