summaryrefslogtreecommitdiff
path: root/lib/chef/knife/client_bulk_delete.rb
blob: a5fe94681d35c0df79cb891da376c61dcc97e8e4 (plain)
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
#
# Author:: Adam Jacob (<adam@chef.io>)
# Copyright:: Copyright 2009-2016, Chef Software Inc.
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

require "chef/knife"

class Chef
  class Knife
    class ClientBulkDelete < Knife

      deps do
        require "chef/api_client_v1"
        require "chef/json_compat"
      end

      option :delete_validators,
       short: "-D",
       long: "--delete-validators",
       description: "Force deletion of clients if they're validators"

      banner "knife client bulk delete REGEX (options)"

      def run
        if name_args.length < 1
          ui.fatal("You must supply a regular expression to match the results against")
          exit 42
        end
        all_clients = Chef::ApiClientV1.list(true)

        matcher = /#{name_args[0]}/
        clients_to_delete = {}
        validators_to_delete = {}
        all_clients.each do |name, client|
          next unless name =~ matcher
          if client.validator
            validators_to_delete[client.name] = client
          else
            clients_to_delete[client.name] = client
          end
        end

        if clients_to_delete.empty? && validators_to_delete.empty?
          ui.info "No clients match the expression /#{name_args[0]}/"
          exit 0
        end

        check_and_delete_validators(validators_to_delete)
        check_and_delete_clients(clients_to_delete)
      end

      def check_and_delete_validators(validators)
        unless validators.empty?
          unless config[:delete_validators]
            ui.msg("The following clients are validators and will not be deleted:")
            print_clients(validators)
            ui.msg("You must specify --delete-validators to delete the validator clients")
          else
            ui.msg("The following validators will be deleted:")
            print_clients(validators)
            if ui.confirm_without_exit("Are you sure you want to delete these validators")
              destroy_clients(validators)
            end
          end
        end
      end

      def check_and_delete_clients(clients)
        unless clients.empty?
          ui.msg("The following clients will be deleted:")
          print_clients(clients)
          ui.confirm("Are you sure you want to delete these clients")
          destroy_clients(clients)
        end
      end

      def destroy_clients(clients)
        clients.sort.each do |name, client|
          client.destroy
          ui.msg("Deleted client #{name}")
        end
      end

      def print_clients(clients)
        ui.msg("")
        ui.msg(ui.list(clients.keys.sort, :columns_down))
        ui.msg("")
      end
    end
  end
end