1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
## This Source Code Form is subject to the terms of the Mozilla Public
## License, v. 2.0. If a copy of the MPL was not distributed with this
## file, You can obtain one at https://mozilla.org/MPL/2.0/.
##
## Copyright (c) 2007-2023 Broadcom. All Rights Reserved. The term “Broadcom” refers to Broadcom Inc. and/or its subsidiaries. All rights reserved.
defmodule RabbitMQ.CLI.Plugins.Commands.DisableCommand do
alias RabbitMQ.CLI.Plugins.Helpers, as: PluginHelpers
alias RabbitMQ.CLI.Core.{DocGuide, Validators}
import RabbitMQ.CLI.Core.{CodePath, Paths}
@behaviour RabbitMQ.CLI.CommandBehaviour
def formatter(), do: RabbitMQ.CLI.Formatters.Plugins
def merge_defaults(args, opts) do
{args, Map.merge(%{online: false, offline: false, all: false}, opts)}
end
def distribution(%{offline: true}), do: :none
def distribution(%{offline: false}), do: :cli
def switches(), do: [online: :boolean, offline: :boolean, all: :boolean]
def validate([], %{all: false}) do
{:validation_failure, :not_enough_args}
end
def validate([_ | _], %{all: true}) do
{:validation_failure, {:bad_argument, "Cannot set both --all and a list of plugins"}}
end
def validate(_, %{online: true, offline: true}) do
{:validation_failure, {:bad_argument, "Cannot set both online and offline"}}
end
def validate(_args, _opts) do
:ok
end
def validate_execution_environment(args, opts) do
Validators.chain(
[
&PluginHelpers.can_set_plugins_with_mode/2,
&require_rabbit_and_plugins/2,
&PluginHelpers.enabled_plugins_file/2,
&plugins_dir/2
],
[args, opts]
)
end
def run(plugin_names, %{all: all_flag, node: node_name} = opts) do
plugins =
case all_flag do
false -> for s <- plugin_names, do: String.to_atom(s)
true -> PluginHelpers.plugin_names(PluginHelpers.list(opts))
end
enabled = PluginHelpers.read_enabled(opts)
all = PluginHelpers.list(opts)
implicit = :rabbit_plugins.dependencies(false, enabled, all)
to_disable_deps = :rabbit_plugins.dependencies(true, plugins, all)
plugins_to_set = MapSet.difference(MapSet.new(enabled), MapSet.new(to_disable_deps))
mode = PluginHelpers.mode(opts)
case PluginHelpers.set_enabled_plugins(MapSet.to_list(plugins_to_set), opts) do
{:ok, enabled_plugins} ->
{:stream,
Stream.concat([
[:rabbit_plugins.strictly_plugins(enabled_plugins, all)],
RabbitMQ.CLI.Core.Helpers.defer(fn ->
:timer.sleep(5000)
case PluginHelpers.update_enabled_plugins(enabled_plugins, mode, node_name, opts) do
%{set: new_enabled} = result ->
disabled = implicit -- new_enabled
filter_strictly_plugins(
Map.put(result, :disabled, :rabbit_plugins.strictly_plugins(disabled, all)),
all,
[:set, :started, :stopped]
)
other ->
other
end
end)
])}
{:error, _} = err ->
err
end
end
use RabbitMQ.CLI.Plugins.ErrorOutput
def banner([], %{all: true, node: node_name}) do
"Disabling ALL plugins on node #{node_name}"
end
def banner(plugins, %{node: node_name}) do
["Disabling plugins on node #{node_name}:" | plugins]
end
def usage, do: "disable <plugin1> [ <plugin2>] | --all [--offline] [--online]"
def usage_additional() do
[
["<plugin1> [ <plugin2>]", "names of plugins to disable separated by a space"],
[
"--online",
"contact target node to disable the plugins. Changes are applied immediately."
],
[
"--offline",
"update enabled plugins file directly without contacting target node. Changes will be delayed until the node is restarted."
],
["--all", "disable all currently enabled plugins"]
]
end
def usage_doc_guides() do
[
DocGuide.plugins()
]
end
def help_section(), do: :plugin_management
def description(), do: "Disables one or more plugins"
#
# Implementation
#
defp filter_strictly_plugins(map, _all, []) do
map
end
defp filter_strictly_plugins(map, all, [head | tail]) do
case map[head] do
nil ->
filter_strictly_plugins(map, all, tail)
other ->
value = :rabbit_plugins.strictly_plugins(other, all)
filter_strictly_plugins(Map.put(map, head, value), all, tail)
end
end
end
|
