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
|
require "ffi_yajl"
require_relative "../rest_base"
require_relative "../rest_error_response"
module ChefZero
module Endpoints
# Typical REST leaf endpoint (/roles/NAME or /data/BAG/NAME)
class RestObjectEndpoint < RestBase
def initialize(server, identity_keys = [ "name" ])
super(server)
identity_keys = [ identity_keys ] if identity_keys.is_a?(String)
@identity_keys = identity_keys
end
attr_reader :identity_keys
def get(request)
already_json_response(200, populate_defaults(request, get_data(request)))
end
def put(request)
# We grab the old body to trigger a 404 if it doesn't exist
get_data(request)
# If it's a rename, check for conflict and delete the old value
if is_rename?(request)
key = identity_key_value(request)
begin
create_data(request, request.rest_path[0..-2], key, request.body, :data_store_exceptions)
rescue DataStore::DataAlreadyExistsError
return error(409, "Cannot rename '#{request.rest_path[-1]}' to '#{key}': '#{key}' already exists")
end
delete_data(request)
already_json_response(201, populate_defaults(request, request.body))
else
set_data(request, request.rest_path, request.body)
already_json_response(200, populate_defaults(request, request.body))
end
end
def delete(request)
result = get_data(request)
delete_data(request)
already_json_response(200, populate_defaults(request, result))
end
def patch_request_body(request)
existing_value = get_data(request, nil, :nil)
if existing_value
request_json = FFI_Yajl::Parser.parse(request.body)
existing_json = FFI_Yajl::Parser.parse(existing_value)
merged_json = existing_json.merge(request_json)
if merged_json.size > request_json.size
return FFI_Yajl::Encoder.encode(merged_json, pretty: true)
end
end
request.body
end
private
# Get the value of the (first existing) identity key from the request body or nil
def identity_key_value(request)
request_json = parse_json(request.body)
identity_keys.map { |k| request_json[k] }.compact.first
end
# Does this request change the value of the identity key?
def is_rename?(request)
return false unless (key = identity_key_value(request))
key != request.rest_path[-1]
end
end
end
end
|
