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'
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
|
