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
|
require "chef/knife"
require "chef/http"
class Chef
class Knife
class Raw < Chef::Knife
banner "knife raw REQUEST_PATH"
deps do
require "chef/json_compat"
require "chef/config"
require "chef/http"
require "chef/http/authenticator"
require "chef/http/cookie_manager"
require "chef/http/decompressor"
require "chef/http/json_output"
end
option :method,
:long => "--method METHOD",
:short => "-m METHOD",
:default => "GET",
:description => "Request method (GET, POST, PUT or DELETE). Default: GET"
option :pretty,
:long => "--[no-]pretty",
:boolean => true,
:default => true,
:description => "Pretty-print JSON output. Default: true"
option :input,
:long => "--input FILE",
:short => "-i FILE",
:description => "Name of file to use for PUT or POST"
option :proxy_auth,
:long => "--proxy-auth",
:boolean => true,
:default => false,
:description => "Use webui proxy authentication. Client key must be the webui key."
class RawInputServerAPI < Chef::HTTP
def initialize(options = {})
options[:client_name] ||= Chef::Config[:node_name]
options[:signing_key_filename] ||= Chef::Config[:client_key]
super(Chef::Config[:chef_server_url], options)
end
use Chef::HTTP::JSONOutput
use Chef::HTTP::CookieManager
use Chef::HTTP::Decompressor
use Chef::HTTP::Authenticator
use Chef::HTTP::RemoteRequestID
end
def run
if name_args.length == 0
show_usage
ui.fatal("You must provide the path you want to hit on the server")
exit(1)
elsif name_args.length > 1
show_usage
ui.fatal("Only one path accepted for knife raw")
exit(1)
end
path = name_args[0]
data = false
if config[:input]
data = IO.read(config[:input])
end
begin
method = config[:method].to_sym
headers = { "Content-Type" => "application/json" }
if config[:proxy_auth]
headers["x-ops-request-source"] = "web"
end
if config[:pretty]
chef_rest = RawInputServerAPI.new
result = chef_rest.request(method, name_args[0], headers, data)
unless result.is_a?(String)
result = Chef::JSONCompat.to_json_pretty(result)
end
else
chef_rest = RawInputServerAPI.new(:raw_output => true)
result = chef_rest.request(method, name_args[0], headers, data)
end
output result
rescue Timeout::Error => e
ui.error "Server timeout"
exit 1
rescue Net::HTTPServerException => e
ui.error "Server responded with error #{e.response.code} \"#{e.response.message}\""
ui.error "Error Body: #{e.response.body}" if e.response.body && e.response.body != ""
exit 1
end
end
end # class Raw
end
end
|
