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 'json'
require 'chef_zero/endpoints/rest_object_endpoint'
require 'chef_zero/rest_error_response'
require 'chef_zero/data_normalizer'
module ChefZero
module Endpoints
# /cookbooks/NAME/VERSION
class CookbookVersionEndpoint < RestObjectEndpoint
def get(request)
if request.rest_path[2] == "_latest"
request.rest_path[2] = latest_version(get_data(request, request.rest_path[0..1]).keys)
end
super(request)
end
def put(request)
name = request.rest_path[1]
version = request.rest_path[2]
data['cookbooks'][name] = {} if !data['cookbooks'][name]
existing_cookbook = data['cookbooks'][name][version]
# Honor frozen
if existing_cookbook
existing_cookbook_json = JSON.parse(existing_cookbook, :create_additions => false)
if existing_cookbook_json['frozen?']
if request.query_params['force'] != "true"
raise RestErrorResponse.new(409, "The cookbook #{name} at version #{version} is frozen. Use the 'force' option to override.")
end
# For some reason, you are forever unable to modify "frozen?" on a frozen cookbook.
request_body = JSON.parse(request.body, :create_additions => false)
if !request_body['frozen?']
request_body['frozen?'] = true
request.body = JSON.pretty_generate(request_body)
end
end
end
# Set the cookbook
data['cookbooks'][name][version] = request.body
# If the cookbook was updated, check for deleted files and clean them up
if existing_cookbook
missing_checksums = get_checksums(existing_cookbook) - get_checksums(request.body)
if missing_checksums.size > 0
hoover_unused_checksums(missing_checksums)
end
end
already_json_response(existing_cookbook ? 200 : 201, populate_defaults(request, data['cookbooks'][name][version]))
end
def delete(request)
deleted_cookbook = get_data(request, request.rest_path)
response = super(request)
cookbook_name = request.rest_path[1]
data['cookbooks'].delete(cookbook_name) if data['cookbooks'][cookbook_name].size == 0
# Hoover deleted files, if they exist
hoover_unused_checksums(get_checksums(deleted_cookbook))
response
end
def get_checksums(cookbook)
result = []
JSON.parse(cookbook, :create_additions => false).each_pair do |key, value|
if value.is_a?(Array)
value.each do |file|
if file.is_a?(Hash) && file.has_key?('checksum')
result << file['checksum']
end
end
end
end
result
end
def hoover_unused_checksums(deleted_checksums)
data['cookbooks'].each_pair do |cookbook_name, versions|
versions.each_pair do |cookbook_version, cookbook|
deleted_checksums = deleted_checksums - get_checksums(cookbook)
end
end
deleted_checksums.each do |checksum|
data['file_store'].delete(checksum)
end
end
def populate_defaults(request, response_json)
# Inject URIs into each cookbook file
cookbook = JSON.parse(response_json, :create_additions => false)
cookbook = DataNormalizer.normalize_cookbook(cookbook, request.rest_path[1], request.rest_path[2], request.base_uri, request.method)
JSON.pretty_generate(cookbook)
end
def latest_version(versions)
sorted = versions.sort_by { |version| Chef::Version.new(version) }
sorted[-1]
end
end
end
end
|