blob: f6d93e73760b1b49b4fc89a1323a3f4e64cc0950 (
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
|
require 'json'
require 'chef_zero/endpoints/rest_object_endpoint'
require 'chef_zero/data_normalizer'
module ChefZero
module Endpoints
# /clients/* and /users/*
class ActorEndpoint < RestObjectEndpoint
def put(request)
# Find out if we're updating the public key.
request_body = JSON.parse(request.body, :create_additions => false)
if request_body['public_key'].nil?
# If public_key is null, then don't overwrite it. Weird patchiness.
body_modified = true
request_body.delete('public_key')
else
updating_public_key = true
end
# Generate private_key if requested.
if request_body.has_key?('private_key')
body_modified = true
if request_body['private_key']
private_key, public_key = server.gen_key_pair
updating_public_key = true
request_body['public_key'] = public_key
end
request_body.delete('private_key')
end
# Save request
request.body = JSON.pretty_generate(request_body) if body_modified
# PUT /clients is patchy
request.body = patch_request_body(request)
result = super(request)
# Inject private_key into response, delete public_key/password if applicable
if result[0] == 200 || result[0] == 201
response = JSON.parse(result[2], :create_additions => false)
response['private_key'] = private_key if private_key
response.delete('public_key') if !updating_public_key && request.rest_path[2] == 'users'
response.delete('password')
json_response(result[0], response)
else
result
end
end
def populate_defaults(request, response_json)
response = JSON.parse(response_json, :create_additions => false)
if request.rest_path[2] == 'clients'
response = DataNormalizer.normalize_client(response, request.rest_path[3])
else
response = DataNormalizer.normalize_user(response, request.rest_path[3])
end
JSON.pretty_generate(response)
end
end
end
end
|