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
|
#
# Author:: Sander Botman (<sbotman@schubergphilis.com>)
# Copyright:: Copyright 2013-2016, Sander Botman.
# 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 EnvironmentCompare < Knife
deps do
require "chef/environment"
end
banner "knife environment compare [ENVIRONMENT..] (options)"
option :all,
:short => "-a",
:long => "--all",
:description => "Show all cookbooks",
:boolean => true
option :mismatch,
:short => "-m",
:long => "--mismatch",
:description => "Only show mismatching versions",
:boolean => true
def run
# Get the commandline environments or all if none are provided.
environments = environment_list
# Get a list of all cookbooks that have constraints and their environment.
constraints = constraint_list(environments)
# Get the total list of cookbooks that have constraints
cookbooks = cookbook_list(constraints)
# If we cannot find any cookbooks, we can stop here.
if cookbooks.nil? || cookbooks.empty?
ui.error "Cannot find any environment cookbook constraints"
exit 1
end
# Get all cookbooks so we can compare them all
cookbooks = rest.get("/cookbooks?num_versions=1") if config[:all]
# display matrix view of in the requested format.
if config[:format] == "summary"
matrix = matrix_output(cookbooks, constraints)
ui.output(matrix)
else
ui.output(constraints)
end
end
private
def environment_list
environments = []
unless @name_args.nil? || @name_args.empty?
@name_args.each { |name| environments << name }
else
environments = Chef::Environment.list
end
end
def constraint_list(environments)
constraints = {}
environments.each do |env,url|
# Because you cannot modify the default environment I filter it out here.
unless env == "_default"
envdata = Chef::Environment.load(env)
ver = envdata.cookbook_versions
constraints[env] = ver
end
end
constraints
end
def cookbook_list(constraints)
result = {}
constraints.each { |env, cb| result.merge!(cb) }
result
end
def matrix_output(cookbooks, constraints)
rows = [ "" ]
environments = []
constraints.each { |e,v| environments << e.to_s }
columns = environments.count + 1
environments.each { |env| rows << ui.color(env, :bold) }
cookbooks.each do |c,v|
total = []
environments.each { |n| total << constraints[n][c]}
if total.uniq.count == 1
next if config[:mismatch]
color = :white
else
color = :yellow
end
rows << ui.color(c, :bold)
environments.each do |e|
tag = constraints[e][c] || "latest"
rows << ui.color(tag, color)
end
end
ui.list(rows, :uneven_columns_across, columns)
end
end
end
end
|
