git.squeep.com Git - akkoma/blob - lib/pleroma/config/transfer_task.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Config.TransferTask do
   6   use Task
   7
   8   alias Pleroma.ConfigDB
   9   alias Pleroma.Repo
  10
  11   require Logger
  12
  13   def start_link(_) do
  14     load_and_update_env()
  15     if Pleroma.Config.get(:env) == :test, do: Ecto.Adapters.SQL.Sandbox.checkin(Repo)
  16     :ignore
  17   end
  18
  19   def load_and_update_env do
  20     with true <- Pleroma.Config.get(:configurable_from_database),
  21          true <- Ecto.Adapters.SQL.table_exists?(Repo, "config"),
  22          started_applications <- Application.started_applications() do
  23       # We need to restart applications for loaded settings take effect
  24       ConfigDB
  25       |> Repo.all()
  26       |> Enum.map(&update_env(&1))
  27       |> Enum.uniq()
  28       # TODO: some problem with prometheus after restart!
  29       |> Enum.reject(&(&1 in [:pleroma, nil, :prometheus]))
  30       |> Enum.each(&restart(started_applications, &1))
  31     end
  32   end
  33
  34   defp update_env(setting) do
  35     try do
  36       key = ConfigDB.from_string(setting.key)
  37       group = ConfigDB.from_string(setting.group)
  38       value = ConfigDB.from_binary(setting.value)
  39
  40       if group != :phoenix and key != :serve_endpoints do
  41         :ok = Application.put_env(group, key, value)
  42       end
  43
  44       group
  45     rescue
  46       e ->
  47         Logger.warn(
  48           "updating env causes error, key: #{inspect(setting.key)}, error: #{inspect(e)}"
  49         )
  50
  51         nil
  52     end
  53   end
  54
  55   defp restart(started_applications, app) do
  56     with {^app, _, _} <- List.keyfind(started_applications, app, 0),
  57          :ok <- Application.stop(app) do
  58       :ok = Application.start(app)
  59     else
  60       nil -> Logger.warn("#{app} is not started.")
  61       error -> Logger.warn(inspect(error))
  62     end
  63   end
  64 end