1 files changed, 65 insertions, 0 deletions
diff --git a/lib/chef_zero/endpoints/rest_object_endpoint.rb b/lib/chef_zero/endpoints/rest_object_endpoint.rb
new file mode 100644
index 0000000..bd45afe
--- /dev/null
+++ b/lib/chef_zero/endpoints/rest_object_endpoint.rb
@@ -0,0 +1,65 @@
+require 'json'
+require 'chef_zero/rest_base'
+require 'chef_zero/rest_error_response'
+
+module ChefZero
+  module Endpoints
+    # Typical REST leaf endpoint (/roles/NAME or /data/BAG/NAME)
+    class RestObjectEndpoint < RestBase
+      def initialize(server, identity_key = 'name')
+        super(server)
+        @identity_key = identity_key
+      end
+
+      attr_reader :identity_key
+
+      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
+        old_body = get_data(request)
+        request_json = JSON.parse(request.body, :create_additions => false)
+        key = request_json[identity_key] || request.rest_path[-1]
+        container = get_data(request, request.rest_path[0..-2])
+        # If it's a rename, check for conflict and delete the old value
+        rename = key != request.rest_path[-1]
+        if rename
+          if container.has_key?(key)
+            return error(409, "Cannot rename '#{request.rest_path[-1]}' to '#{key}': '#{key}' already exists")
+          end
+          container.delete(request.rest_path[-1])
+        end
+        container[key] = request.body
+        already_json_response(200, populate_defaults(request, request.body))
+      end
+
+      def delete(request)
+        key = request.rest_path[-1]
+        container = get_data(request, request.rest_path[0..-2])
+        if !container.has_key?(key)
+          raise RestErrorResponse.new(404, "Object not found: #{build_uri(request.base_uri, request.rest_path)}")
+        end
+        result = container[key]
+        container.delete(key)
+        already_json_response(200, populate_defaults(request, result))
+      end
+
+      def patch_request_body(request)
+        container = get_data(request, request.rest_path[0..-2])
+        existing_value = container[request.rest_path[-1]]
+        if existing_value
+          request_json = JSON.parse(request.body, :create_additions => false)
+          existing_json = JSON.parse(existing_value, :create_additions => false)
+          merged_json = existing_json.merge(request_json)
+          if merged_json.size > request_json.size
+            return JSON.pretty_generate(merged_json)
+          end
+        end
+        request.body
+      end
+    end
+  end
+end
+