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
|
require 'ffi_yajl'
require 'chef_zero/endpoints/rest_list_endpoint'
module ChefZero
module Endpoints
# /users, /organizations/ORG/clients or /organizations/ORG/users
class ActorsEndpoint < RestListEndpoint
DEFAULT_PUBLIC_KEY_NAME = "default"
def get(request)
# TODO Refactor this
response = super(request)
if request.query_params['email']
results = FFI_Yajl::Parser.parse(response[2], :create_additions => false)
new_results = {}
results.each do |name, url|
record = get_data(request, request.rest_path + [ name ], :nil)
if record
record = FFI_Yajl::Parser.parse(record, :create_additions => false)
new_results[name] = url if record['email'] == request.query_params['email']
end
end
response[2] = FFI_Yajl::Encoder.encode(new_results, :pretty => true)
end
if request.query_params['verbose'] && !client?(request)
results = FFI_Yajl::Parser.parse(response[2], :create_additions => false)
results.each do |name, url|
record = get_data(request, request.rest_path + [ name ], :nil)
if record
record = FFI_Yajl::Parser.parse(record, :create_additions => false)
record = ChefData::DataNormalizer.normalize_user(data_store, record, name, identity_keys, server.options[:osc_compat])
results[name] = record
end
end
response[2] = FFI_Yajl::Encoder.encode(results, :pretty => true)
end
response
end
def post(request)
request_body = FFI_Yajl::Parser.parse(request.body, :create_additions => false)
client_or_user_name = request_body[ client?(request) ? "name" : "username" ]
public_key = request_body["public_key"]
# Did the user post a public_key? If not, generate one.
unless public_key
private_key, public_key = server.gen_key_pair
end
request_body.delete('public_key')
request.body = FFI_Yajl::Encoder.encode(request_body, :pretty => true)
result = super(request)
if result[0] == 201
# Store the received or generated public key
store_default_public_key!(request, client_or_user_name, public_key)
# If we generated a key, stuff it in the response.
response = FFI_Yajl::Parser.parse(result[2], :create_additions => false)
response['private_key'] = private_key if private_key
response['public_key'] = public_key unless request.rest_path[0] == 'users'
json_response(201, response)
else
result
end
end
private
# Store the public key in user_keys
def store_default_public_key!(request, client_or_user_name, public_key)
path =
if client?(request)
[ *request.rest_path[0..1], "client_keys" ]
else
[ "user_keys" ]
end
.push(client_or_user_name, "keys")
data = FFI_Yajl::Encoder.encode(
"name" => DEFAULT_PUBLIC_KEY_NAME,
"public_key" => public_key,
"expiration_date" => "infinity"
)
create_data(request, path, DEFAULT_PUBLIC_KEY_NAME, data, :create_dir)
end
def client?(request)
request.rest_path[2] == "clients"
end
end
end
end
|