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
|
#
# Author:: Daniel DeLeo (<dan@kallistec.com>)
# Copyright:: Copyright 2009-2016, Daniel DeLeo
# 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 IndexRebuild < Knife
banner "knife index rebuild (options)"
option :yes,
:short => "-y",
:long => "--yes",
:boolean => true,
:description => "don't bother to ask if I'm sure"
def run
api_info = grab_api_info
if unsupported_version?(api_info)
unsupported_server_message(api_info)
exit 1
else
deprecated_server_message
nag
output rest.post("/search/reindex", {})
end
end
def grab_api_info
# Since we don't yet have any endpoints that implement an
# OPTIONS handler, we need to get our version header
# information in a more roundabout way. We'll try to query
# for a node we know won't exist; the 404 response that comes
# back will give us what we want
dummy_node = "knife_index_rebuild_test_#{rand(1000000)}"
rest.get("/nodes/#{dummy_node}")
rescue Net::HTTPServerException => exception
r = exception.response
parse_api_info(r)
end
# Only Chef 11+ servers will have version information in their
# headers, and only those servers will lack an API endpoint for
# index rebuilding.
def unsupported_version?(api_info)
!!api_info["version"]
end
def unsupported_server_message(api_info)
ui.error("Rebuilding the index is not available via knife for #{server_type(api_info)}s version 11.0.0 and above.")
ui.info("Instead, run the '#{ctl_command(api_info)} reindex' command on the server itself.")
end
def deprecated_server_message
ui.warn("'knife index rebuild' has been removed for Chef 11+ servers. It will continue to work for prior versions, however.")
end
def nag
ui.info("This operation is destructive. Rebuilding the index may take some time.")
ui.confirm("Continue")
end
# Chef 11 (and above) servers return various pieces of
# information about the server in an +x-ops-api-info+ header.
# This is a +;+ delimited string of key / value pairs, separated
# by +=+.
#
# Given a Net::HTTPResponse object, this method extracts this
# information (if present), and returns it as a hash. If no
# such header is found, an empty hash is returned.
def parse_api_info(response)
value = response["x-ops-api-info"]
if value
kv = value.split(";")
kv.inject({}) do |acc, pair|
k, v = pair.split("=")
acc[k] = v
acc
end
else
{}
end
end
# Given an API info hash (see +#parse_api_info(response)+),
# return a string describing the kind of server we're
# interacting with (based on the +flavor+ field)
def server_type(api_info)
case api_info["flavor"]
when "osc"
"Open Source Chef Server"
when "opc"
"Private Chef Server"
else
# Generic fallback
"Chef Server"
end
end
# Given an API info hash (see +#parse_api_info(response)+),
# return the name of the "server-ctl" command for the kind of
# server we're interacting with (based on the +flavor+ field)
def ctl_command(api_info)
case api_info["flavor"]
when "osc"
"chef-server-ctl"
when "opc"
"private-chef-ctl"
else
# Generic fallback
"chef-server-ctl"
end
end
end
end
end
|
