summaryrefslogtreecommitdiff
path: root/lib/chef/knife/node_bulk_delete.rb
blob: 2ca63da51235972a01002c749c7a6cf5523b9fa0 (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
#
# 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 NodeBulkDelete < Knife

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

      banner "knife node 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

        nodes_to_delete = {}
        matcher = /#{name_args[0]}/

        all_nodes.each do |name, node|
          next unless name =~ matcher
          nodes_to_delete[name] = node
        end

        if nodes_to_delete.empty?
          ui.msg "No nodes match the expression /#{name_args[0]}/"
          exit 0
        end

        ui.msg("The following nodes will be deleted:")
        ui.msg("")
        ui.msg(ui.list(nodes_to_delete.keys.sort, :columns_down))
        ui.msg("")
        ui.confirm("Are you sure you want to delete these nodes")

        nodes_to_delete.sort.each do |name, node|
          node.destroy
          ui.msg("Deleted node #{name}")
        end
      end

      def all_nodes
        node_uris_by_name = Chef::Node.list

        node_uris_by_name.keys.inject({}) do |nodes_by_name, name|
          nodes_by_name[name] = Chef::Node.new.tap { |n| n.name(name) }
          nodes_by_name
        end
      end

    end
  end
end