git.squeep.com Git - akkoma/blob - lib/mix/tasks/pleroma/config.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 Mix.Tasks.Pleroma.Config do
   6   use Mix.Task
   7
   8   import Mix.Pleroma
   9
  10   alias Pleroma.ConfigDB
  11   alias Pleroma.Repo
  12
  13   @shortdoc "Manages the location of the config"
  14   @moduledoc File.read!("docs/administration/CLI_tasks/config.md")
  15
  16   def run(["migrate_to_db"]) do
  17     start_pleroma()
  18     migrate_to_db()
  19   end
  20
  21   def run(["migrate_from_db" | options]) do
  22     start_pleroma()
  23
  24     {opts, _} =
  25       OptionParser.parse!(options,
  26         strict: [env: :string, delete: :boolean],
  27         aliases: [d: :delete]
  28       )
  29
  30     migrate_from_db(opts)
  31   end
  32
  33   @spec migrate_to_db(Path.t() | nil) :: any()
  34   def migrate_to_db(file_path \\ nil) do
  35     if Pleroma.Config.get([:configurable_from_database]) do
  36       config_file =
  37         if file_path do
  38           file_path
  39         else
  40           if Pleroma.Config.get(:release) do
  41             Pleroma.Config.get(:config_path)
  42           else
  43             "config/#{Pleroma.Config.get(:env)}.secret.exs"
  44           end
  45         end
  46
  47       do_migrate_to_db(config_file)
  48     else
  49       migration_error()
  50     end
  51   end
  52
  53   defp do_migrate_to_db(config_file) do
  54     if File.exists?(config_file) do
  55       custom_config =
  56         config_file
  57         |> read_file()
  58         |> elem(0)
  59
  60       custom_config
  61       |> Keyword.keys()
  62       |> Enum.each(&create(&1, custom_config))
  63     else
  64       shell_info("To migrate settings, you must define custom settings in #{config_file}.")
  65     end
  66   end
  67
  68   defp create(group, settings) do
  69     group
  70     |> Pleroma.Config.Loader.filter_group(settings)
  71     |> Enum.each(fn {key, value} ->
  72       key = inspect(key)
  73       {:ok, _} = ConfigDB.update_or_create(%{group: inspect(group), key: key, value: value})
  74
  75       shell_info("Settings for key #{key} migrated.")
  76     end)
  77
  78     shell_info("Settings for group :#{group} migrated.")
  79   end
  80
  81   defp migrate_from_db(opts) do
  82     if Pleroma.Config.get([:configurable_from_database]) do
  83       env = opts[:env] || "prod"
  84
  85       config_path =
  86         if Pleroma.Config.get(:release) do
  87           :config_path
  88           |> Pleroma.Config.get()
  89           |> Path.dirname()
  90         else
  91           "config"
  92         end
  93         |> Path.join("#{env}.exported_from_db.secret.exs")
  94
  95       file = File.open!(config_path, [:write, :utf8])
  96
  97       IO.write(file, config_header())
  98
  99       ConfigDB
 100       |> Repo.all()
 101       |> Enum.each(&write_and_delete(&1, file, opts[:delete]))
 102
 103       :ok = File.close(file)
 104       System.cmd("mix", ["format", config_path])
 105     else
 106       migration_error()
 107     end
 108   end
 109
 110   defp migration_error do
 111     shell_error(
 112       "Migration is not allowed in config. You can change this behavior by setting `configurable_from_database` to true."
 113     )
 114   end
 115
 116   if Code.ensure_loaded?(Config.Reader) do
 117     defp config_header, do: "import Config\r\n\r\n"
 118     defp read_file(config_file), do: Config.Reader.read_imports!(config_file)
 119   else
 120     defp config_header, do: "use Mix.Config\r\n\r\n"
 121     defp read_file(config_file), do: Mix.Config.eval!(config_file)
 122   end
 123
 124   defp write_and_delete(config, file, delete?) do
 125     config
 126     |> write(file)
 127     |> delete(delete?)
 128   end
 129
 130   defp write(config, file) do
 131     value =
 132       config.value
 133       |> ConfigDB.from_binary()
 134       |> inspect(limit: :infinity)
 135
 136     IO.write(file, "config #{config.group}, #{config.key}, #{value}\r\n\r\n")
 137
 138     config
 139   end
 140
 141   defp delete(config, true) do
 142     {:ok, _} = Repo.delete(config)
 143     shell_info("#{config.key} deleted from DB.")
 144   end
 145
 146   defp delete(_config, _), do: :ok
 147 end