1 files changed, 67 insertions, 0 deletions
diff --git a/lib/chef_zero/endpoints/cookbook_artifacts_cookbook_identifier.rb b/lib/chef_zero/endpoints/cookbook_artifacts_cookbook_identifier.rb
new file mode 100644
index 0000000..4d22d08
--- /dev/null
+++ b/lib/chef_zero/endpoints/cookbook_artifacts_cookbook_identifier.rb
@@ -0,0 +1,67 @@
+require 'chef_zero/chef_data/data_normalizer'
+
+module ChefZero
+  module Endpoints
+    class CookbookArtifactsCookbookIdentifierEndpoint < ChefZero::Endpoints::CookbookVersionEndpoint
+      # these endpoints are almost, but not quite, not entirely unlike the corresponding /cookbooks endpoints.
+      # it could all be refactored for maximum reuse, but they're short REST methods with well-defined
+      # behavioral specs (pedant), so there's not a huge benefit.
+
+      # GET /organizations/ORG/cookbook_artifacts/NAME/IDENTIFIER
+      def get(request)
+        cookbook_data = normalize(request, parse_json(get_data(request)))
+        return json_response(200, cookbook_data)
+      end
+
+      # PUT /organizations/ORG/cookbook_artifacts/COOKBOOK/IDENTIFIER
+      def put(request)
+        if exists_data?(request)
+          return error(409, "Cookbooks cannot be modified, and a cookbook with this identifier already exists.")
+        end
+
+        set_data(request, nil, request.body, :create_dir)
+
+        return already_json_response(201, request.body)
+      end
+
+      # DELETE /organizations/ORG/cookbook_artifacts/COOKBOOK/IDENTIFIER
+      def delete(request)
+        begin
+          doomed_cookbook_json = get_data(request)
+          identified_cookbook_data = normalize(request, parse_json(doomed_cookbook_json))
+          delete_data(request)
+
+          # go through the recipes and delete stuff in the file store.
+          hoover_unused_checksums(get_checksums(doomed_cookbook_json), request, 'cookbook_artifacts')
+
+          # if this was the last revision, delete the directory so future requests will 404, instead of
+          # returning 200 with an empty list.
+          artifact_path = request.rest_path[0..-2]
+          if list_data(request, artifact_path).size == 0
+            delete_data_dir(request, artifact_path)
+          end
+
+          json_response(200, identified_cookbook_data)
+        rescue RestErrorResponse => ex
+          if ex.response_code == 404
+            error(404, "not_found")
+          else
+            raise
+          end
+        end
+      end
+
+      private
+
+      def make_file_store_path(rest_path, recipe)
+        rest_path.first(2) + ["file_store", "checksums", recipe["checksum"]]
+      end
+
+      def normalize(request, cookbook_artifact_data)
+        ChefData::DataNormalizer.normalize_cookbook(self, request.rest_path[0..1],
+                                                    cookbook_artifact_data, request.rest_path[3], request.rest_path[4],
+                                                    request.base_uri, request.method, true)
+      end
+    end
+  end
+end