Use a callback strategy to short circuit the functions and print a nice error
This commit is contained in:
Mark Felder
parent
a7b5280b5b
commit
13947999ad
2 changed files with 205 additions and 171 deletions
|
|
@ -14,149 +14,158 @@ defmodule Mix.Tasks.Pleroma.Config do
|
|||
@moduledoc File.read!("docs/administration/CLI_tasks/config.md")
|
||||
|
||||
def run(["migrate_to_db"]) do
|
||||
check_configdb()
|
||||
start_pleroma()
|
||||
migrate_to_db()
|
||||
check_configdb(fn ->
|
||||
start_pleroma()
|
||||
migrate_to_db()
|
||||
end)
|
||||
end
|
||||
|
||||
def run(["migrate_from_db" | options]) do
|
||||
check_configdb()
|
||||
start_pleroma()
|
||||
check_configdb(fn ->
|
||||
start_pleroma()
|
||||
|
||||
{opts, _} =
|
||||
OptionParser.parse!(options,
|
||||
strict: [env: :string, delete: :boolean],
|
||||
aliases: [d: :delete]
|
||||
)
|
||||
{opts, _} =
|
||||
OptionParser.parse!(options,
|
||||
strict: [env: :string, delete: :boolean],
|
||||
aliases: [d: :delete]
|
||||
)
|
||||
|
||||
migrate_from_db(opts)
|
||||
migrate_from_db(opts)
|
||||
end)
|
||||
end
|
||||
|
||||
def run(["dump"]) do
|
||||
check_configdb()
|
||||
start_pleroma()
|
||||
check_configdb(fn ->
|
||||
start_pleroma()
|
||||
|
||||
header = config_header()
|
||||
header = config_header()
|
||||
|
||||
settings =
|
||||
ConfigDB
|
||||
|> Repo.all()
|
||||
|> Enum.sort()
|
||||
settings =
|
||||
ConfigDB
|
||||
|> Repo.all()
|
||||
|> Enum.sort()
|
||||
|
||||
unless settings == [] do
|
||||
shell_info("#{header}")
|
||||
unless settings == [] do
|
||||
shell_info("#{header}")
|
||||
|
||||
settings |> Enum.each(&dump(&1))
|
||||
else
|
||||
shell_error("No settings in ConfigDB.")
|
||||
end
|
||||
settings |> Enum.each(&dump(&1))
|
||||
else
|
||||
shell_error("No settings in ConfigDB.")
|
||||
end
|
||||
end)
|
||||
end
|
||||
|
||||
def run(["dump", group, key]) do
|
||||
check_configdb()
|
||||
start_pleroma()
|
||||
check_configdb(fn ->
|
||||
start_pleroma()
|
||||
|
||||
group = maybe_atomize(group)
|
||||
key = maybe_atomize(key)
|
||||
group = maybe_atomize(group)
|
||||
key = maybe_atomize(key)
|
||||
|
||||
dump_key(group, key)
|
||||
dump_key(group, key)
|
||||
end)
|
||||
end
|
||||
|
||||
def run(["dump", group]) do
|
||||
check_configdb()
|
||||
start_pleroma()
|
||||
check_configdb(fn ->
|
||||
start_pleroma()
|
||||
|
||||
group = maybe_atomize(group)
|
||||
group = maybe_atomize(group)
|
||||
|
||||
dump_group(group)
|
||||
dump_group(group)
|
||||
end)
|
||||
end
|
||||
|
||||
def run(["groups"]) do
|
||||
check_configdb()
|
||||
start_pleroma()
|
||||
check_configdb(fn ->
|
||||
start_pleroma()
|
||||
|
||||
groups =
|
||||
ConfigDB
|
||||
|> Repo.all()
|
||||
|> Enum.map(fn x -> x.group end)
|
||||
|> Enum.sort()
|
||||
|> Enum.uniq()
|
||||
groups =
|
||||
ConfigDB
|
||||
|> Repo.all()
|
||||
|> Enum.map(fn x -> x.group end)
|
||||
|> Enum.sort()
|
||||
|> Enum.uniq()
|
||||
|
||||
if length(groups) > 0 do
|
||||
shell_info("The following configuration groups are set in ConfigDB:\r\n")
|
||||
groups |> Enum.each(fn x -> shell_info("- #{x}") end)
|
||||
shell_info("\r\n")
|
||||
end
|
||||
if length(groups) > 0 do
|
||||
shell_info("The following configuration groups are set in ConfigDB:\r\n")
|
||||
groups |> Enum.each(fn x -> shell_info("- #{x}") end)
|
||||
shell_info("\r\n")
|
||||
end
|
||||
end)
|
||||
end
|
||||
|
||||
def run(["reset"]) do
|
||||
check_configdb()
|
||||
start_pleroma()
|
||||
check_configdb(fn ->
|
||||
start_pleroma()
|
||||
|
||||
shell_info("The following settings will be permanently removed:")
|
||||
shell_info("The following settings will be permanently removed:")
|
||||
|
||||
ConfigDB
|
||||
|> Repo.all()
|
||||
|> Enum.sort()
|
||||
|> Enum.each(&dump(&1))
|
||||
ConfigDB
|
||||
|> Repo.all()
|
||||
|> Enum.sort()
|
||||
|> Enum.each(&dump(&1))
|
||||
|
||||
shell_error("\nTHIS CANNOT BE UNDONE!")
|
||||
shell_error("\nTHIS CANNOT BE UNDONE!")
|
||||
|
||||
if shell_prompt("Are you sure you want to continue?", "n") in ~w(Yn Y y) do
|
||||
Ecto.Adapters.SQL.query!(Repo, "TRUNCATE config;")
|
||||
Ecto.Adapters.SQL.query!(Repo, "ALTER SEQUENCE config_id_seq RESTART;")
|
||||
if shell_prompt("Are you sure you want to continue?", "n") in ~w(Yn Y y) do
|
||||
Ecto.Adapters.SQL.query!(Repo, "TRUNCATE config;")
|
||||
Ecto.Adapters.SQL.query!(Repo, "ALTER SEQUENCE config_id_seq RESTART;")
|
||||
|
||||
shell_info("The ConfigDB settings have been removed from the database.")
|
||||
else
|
||||
shell_error("No changes made.")
|
||||
end
|
||||
shell_info("The ConfigDB settings have been removed from the database.")
|
||||
else
|
||||
shell_error("No changes made.")
|
||||
end
|
||||
end)
|
||||
end
|
||||
|
||||
def run(["delete", group]) do
|
||||
check_configdb()
|
||||
start_pleroma()
|
||||
check_configdb(fn ->
|
||||
start_pleroma()
|
||||
|
||||
group = maybe_atomize(group)
|
||||
group = maybe_atomize(group)
|
||||
|
||||
if group_exists?(group) do
|
||||
shell_info("The following settings will be removed from ConfigDB:\n")
|
||||
if group_exists?(group) do
|
||||
shell_info("The following settings will be removed from ConfigDB:\n")
|
||||
|
||||
dump_group(group)
|
||||
dump_group(group)
|
||||
|
||||
if shell_prompt("Are you sure you want to continue?", "n") in ~w(Yn Y y) do
|
||||
ConfigDB
|
||||
|> Repo.all()
|
||||
|> Enum.filter(fn x ->
|
||||
if x.group == group do
|
||||
x |> delete(true)
|
||||
end
|
||||
end)
|
||||
else
|
||||
shell_error("No changes made.")
|
||||
end
|
||||
else
|
||||
shell_error("No settings in ConfigDB for #{inspect(group)}. Aborting.")
|
||||
end
|
||||
end)
|
||||
end
|
||||
|
||||
def run(["delete", group, key]) do
|
||||
check_configdb(fn ->
|
||||
start_pleroma()
|
||||
|
||||
group = maybe_atomize(group)
|
||||
key = maybe_atomize(key)
|
||||
|
||||
if shell_prompt("Are you sure you want to continue?", "n") in ~w(Yn Y y) do
|
||||
ConfigDB
|
||||
|> Repo.all()
|
||||
|> Enum.filter(fn x ->
|
||||
if x.group == group do
|
||||
if x.group == group and x.key == key do
|
||||
x |> delete(true)
|
||||
end
|
||||
end)
|
||||
else
|
||||
shell_error("No changes made.")
|
||||
end
|
||||
else
|
||||
shell_error("No settings in ConfigDB for #{inspect(group)}. Aborting.")
|
||||
end
|
||||
end
|
||||
|
||||
def run(["delete", group, key]) do
|
||||
check_configdb()
|
||||
start_pleroma()
|
||||
|
||||
group = maybe_atomize(group)
|
||||
key = maybe_atomize(key)
|
||||
|
||||
if shell_prompt("Are you sure you want to continue?", "n") in ~w(Yn Y y) do
|
||||
ConfigDB
|
||||
|> Repo.all()
|
||||
|> Enum.filter(fn x ->
|
||||
if x.group == group and x.key == key do
|
||||
x |> delete(true)
|
||||
end
|
||||
end)
|
||||
else
|
||||
shell_error("No changes made.")
|
||||
end
|
||||
end)
|
||||
end
|
||||
|
||||
@spec migrate_to_db(Path.t() | nil) :: any()
|
||||
|
|
@ -282,7 +291,7 @@ defp dump(%Pleroma.ConfigDB{} = config) do
|
|||
end
|
||||
|
||||
defp configdb_not_enabled do
|
||||
raise(
|
||||
shell_error(
|
||||
"ConfigDB not enabled. Please check the value of :configurable_from_database in your configuration."
|
||||
)
|
||||
end
|
||||
|
|
@ -336,9 +345,9 @@ defp maybe_atomize(arg) when is_binary(arg) do
|
|||
end
|
||||
end
|
||||
|
||||
defp check_configdb() do
|
||||
defp check_configdb(callback) do
|
||||
with true <- Pleroma.Config.get([:configurable_from_database]) do
|
||||
:ok
|
||||
callback.()
|
||||
else
|
||||
_ -> configdb_not_enabled()
|
||||
end
|
||||
|
|
|
|||
|
|
@ -22,8 +22,6 @@ defmodule Mix.Tasks.Pleroma.ConfigTest do
|
|||
:ok
|
||||
end
|
||||
|
||||
setup_all do: clear_config(:configurable_from_database, true)
|
||||
|
||||
test "error if file with custom settings doesn't exist" do
|
||||
Mix.Tasks.Pleroma.Config.migrate_to_db("config/not_existance_config_file.exs")
|
||||
|
||||
|
|
@ -36,6 +34,7 @@ test "error if file with custom settings doesn't exist" do
|
|||
|
||||
describe "migrate_to_db/1" do
|
||||
setup do
|
||||
clear_config(:configurable_from_database, true)
|
||||
initial = Application.get_env(:quack, :level)
|
||||
on_exit(fn -> Application.put_env(:quack, :level, initial) end)
|
||||
end
|
||||
|
|
@ -83,6 +82,7 @@ test "config table is truncated before migration" do
|
|||
|
||||
describe "with deletion temp file" do
|
||||
setup do
|
||||
clear_config(:configurable_from_database, true)
|
||||
temp_file = "config/temp.exported_from_db.secret.exs"
|
||||
|
||||
on_exit(fn ->
|
||||
|
|
@ -187,89 +187,114 @@ test "load a settings with large values and pass to file", %{temp_file: temp_fil
|
|||
end
|
||||
end
|
||||
|
||||
test "dumping a specific group" do
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: :instance,
|
||||
value: [
|
||||
name: "Pleroma Test"
|
||||
]
|
||||
)
|
||||
describe "operations on database config" do
|
||||
setup do: clear_config(:configurable_from_database, true)
|
||||
|
||||
insert(:config,
|
||||
group: :web_push_encryption,
|
||||
key: :vapid_details,
|
||||
value: [
|
||||
subject: "mailto:administrator@example.com",
|
||||
public_key:
|
||||
"BOsPL-_KjNnjj_RMvLeR3dTOrcndi4TbMR0cu56gLGfGaT5m1gXxSfRHOcC4Dd78ycQL1gdhtx13qgKHmTM5xAI",
|
||||
private_key: "Ism6FNdS31nLCA94EfVbJbDdJXCxAZ8cZiB1JQPN_t4"
|
||||
]
|
||||
)
|
||||
test "dumping a specific group" do
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: :instance,
|
||||
value: [
|
||||
name: "Pleroma Test"
|
||||
]
|
||||
)
|
||||
|
||||
Mix.Tasks.Pleroma.Config.run(["dump", "pleroma"])
|
||||
insert(:config,
|
||||
group: :web_push_encryption,
|
||||
key: :vapid_details,
|
||||
value: [
|
||||
subject: "mailto:administrator@example.com",
|
||||
public_key:
|
||||
"BOsPL-_KjNnjj_RMvLeR3dTOrcndi4TbMR0cu56gLGfGaT5m1gXxSfRHOcC4Dd78ycQL1gdhtx13qgKHmTM5xAI",
|
||||
private_key: "Ism6FNdS31nLCA94EfVbJbDdJXCxAZ8cZiB1JQPN_t4"
|
||||
]
|
||||
)
|
||||
|
||||
assert_receive {:mix_shell, :info,
|
||||
["config :pleroma, :instance, [name: \"Pleroma Test\"]\r\n\r\n"]}
|
||||
Mix.Tasks.Pleroma.Config.run(["dump", "pleroma"])
|
||||
|
||||
refute_receive {
|
||||
:mix_shell,
|
||||
:info,
|
||||
[
|
||||
"config :web_push_encryption, :vapid_details, [subject: \"mailto:administrator@example.com\", public_key: \"BOsPL-_KjNnjj_RMvLeR3dTOrcndi4TbMR0cu56gLGfGaT5m1gXxSfRHOcC4Dd78ycQL1gdhtx13qgKHmTM5xAI\", private_key: \"Ism6FNdS31nLCA94EfVbJbDdJXCxAZ8cZiB1JQPN_t4\"]\r\n\r\n"
|
||||
]
|
||||
}
|
||||
assert_receive {:mix_shell, :info,
|
||||
["config :pleroma, :instance, [name: \"Pleroma Test\"]\r\n\r\n"]}
|
||||
|
||||
refute_receive {
|
||||
:mix_shell,
|
||||
:info,
|
||||
[
|
||||
"config :web_push_encryption, :vapid_details, [subject: \"mailto:administrator@example.com\", public_key: \"BOsPL-_KjNnjj_RMvLeR3dTOrcndi4TbMR0cu56gLGfGaT5m1gXxSfRHOcC4Dd78ycQL1gdhtx13qgKHmTM5xAI\", private_key: \"Ism6FNdS31nLCA94EfVbJbDdJXCxAZ8cZiB1JQPN_t4\"]\r\n\r\n"
|
||||
]
|
||||
}
|
||||
end
|
||||
|
||||
test "dumping a specific key in a group" do
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: :instance,
|
||||
value: [
|
||||
name: "Pleroma Test"
|
||||
]
|
||||
)
|
||||
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: Pleroma.Captcha,
|
||||
value: [
|
||||
enabled: false
|
||||
]
|
||||
)
|
||||
|
||||
Mix.Tasks.Pleroma.Config.run(["dump", "pleroma", "Pleroma.Captcha"])
|
||||
|
||||
refute_receive {:mix_shell, :info,
|
||||
["config :pleroma, :instance, [name: \"Pleroma Test\"]\r\n\r\n"]}
|
||||
|
||||
assert_receive {:mix_shell, :info,
|
||||
["config :pleroma, Pleroma.Captcha, [enabled: false]\r\n\r\n"]}
|
||||
end
|
||||
|
||||
test "dumps all configuration successfully" do
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: :instance,
|
||||
value: [
|
||||
name: "Pleroma Test"
|
||||
]
|
||||
)
|
||||
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: Pleroma.Captcha,
|
||||
value: [
|
||||
enabled: false
|
||||
]
|
||||
)
|
||||
|
||||
Mix.Tasks.Pleroma.Config.run(["dump"])
|
||||
|
||||
assert_receive {:mix_shell, :info,
|
||||
["config :pleroma, :instance, [name: \"Pleroma Test\"]\r\n\r\n"]}
|
||||
|
||||
assert_receive {:mix_shell, :info,
|
||||
["config :pleroma, Pleroma.Captcha, [enabled: false]\r\n\r\n"]}
|
||||
end
|
||||
end
|
||||
|
||||
test "dumping a specific key in a group" do
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: :instance,
|
||||
value: [
|
||||
name: "Pleroma Test"
|
||||
]
|
||||
)
|
||||
describe "when configdb disabled" do
|
||||
test "refuses to dump" do
|
||||
clear_config(:configurable_from_database, false)
|
||||
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: Pleroma.Captcha,
|
||||
value: [
|
||||
enabled: false
|
||||
]
|
||||
)
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: :instance,
|
||||
value: [
|
||||
name: "Pleroma Test"
|
||||
]
|
||||
)
|
||||
|
||||
Mix.Tasks.Pleroma.Config.run(["dump", "pleroma", "Pleroma.Captcha"])
|
||||
Mix.Tasks.Pleroma.Config.run(["dump"])
|
||||
|
||||
refute_receive {:mix_shell, :info,
|
||||
["config :pleroma, :instance, [name: \"Pleroma Test\"]\r\n\r\n"]}
|
||||
|
||||
assert_receive {:mix_shell, :info,
|
||||
["config :pleroma, Pleroma.Captcha, [enabled: false]\r\n\r\n"]}
|
||||
end
|
||||
|
||||
test "dumps all configuration successfully" do
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: :instance,
|
||||
value: [
|
||||
name: "Pleroma Test"
|
||||
]
|
||||
)
|
||||
|
||||
insert(:config,
|
||||
group: :pleroma,
|
||||
key: Pleroma.Captcha,
|
||||
value: [
|
||||
enabled: false
|
||||
]
|
||||
)
|
||||
|
||||
Mix.Tasks.Pleroma.Config.run(["dump"])
|
||||
|
||||
assert_receive {:mix_shell, :info,
|
||||
["config :pleroma, :instance, [name: \"Pleroma Test\"]\r\n\r\n"]}
|
||||
|
||||
assert_receive {:mix_shell, :info,
|
||||
["config :pleroma, Pleroma.Captcha, [enabled: false]\r\n\r\n"]}
|
||||
assert_receive {:mix_shell, :error,
|
||||
[
|
||||
"ConfigDB not enabled. Please check the value of :configurable_from_database in your configuration."
|
||||
]}
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
|||
Loading…
Reference in a new issue