chef-zero 1.4.0.alpha-x86-mingw32

data/lib/chef_zero/data_normalizer.rb

@@ -0,0 +1,142 @@
+require 'chef_zero'
+require 'chef_zero/rest_base'
+
+module ChefZero
+  class DataNormalizer
+    def self.normalize_client(client, name)
+      client['name'] ||= name
+      client['admin'] ||= false
+      client['public_key'] ||= PUBLIC_KEY
+      client['validator'] ||= false
+      client['json_class'] ||= "Chef::ApiClient"
+      client['chef_type'] ||= "client"
+      client
+    end
+
+    def self.normalize_user(user, name)
+      user['name'] ||= name
+      user['admin'] ||= false
+      user['public_key'] ||= PUBLIC_KEY
+      user
+    end
+
+    def self.normalize_data_bag_item(data_bag_item, data_bag_name, id, method)
+      if method == 'DELETE'
+        # TODO SERIOUSLY, WHO DOES THIS MANY EXCEPTIONS IN THEIR INTERFACE
+        if !(data_bag_item['json_class'] == 'Chef::DataBagItem' && data_bag_item['raw_data'])
+          data_bag_item['id'] ||= id
+          data_bag_item = { 'raw_data' => data_bag_item }
+          data_bag_item['chef_type'] ||= 'data_bag_item'
+          data_bag_item['json_class'] ||= 'Chef::DataBagItem'
+          data_bag_item['data_bag'] ||= data_bag_name
+          data_bag_item['name'] ||= "data_bag_item_#{data_bag_name}_#{id}"
+        end
+      else
+        # If it's not already wrapped with raw_data, wrap it.
+        if data_bag_item['json_class'] == 'Chef::DataBagItem' && data_bag_item['raw_data']
+          data_bag_item = data_bag_item['raw_data']
+        end
+        # Argh.  We don't do this on GET, but we do on PUT and POST????
+        if %w(PUT POST).include?(method)
+          data_bag_item['chef_type'] ||= 'data_bag_item'
+          data_bag_item['data_bag'] ||= data_bag_name
+        end
+        data_bag_item['id'] ||= id
+      end
+      data_bag_item
+    end
+
+    def self.normalize_environment(environment, name)
+      environment['name'] ||= name
+      environment['description'] ||= ''
+      environment['cookbook_versions'] ||= {}
+      environment['json_class'] ||= "Chef::Environment"
+      environment['chef_type'] ||= "environment"
+      environment['default_attributes'] ||= {}
+      environment['override_attributes'] ||= {}
+      environment
+    end
+
+    def self.normalize_cookbook(cookbook, name, version, base_uri, method)
+      # TODO I feel dirty
+      if method != 'PUT'
+        cookbook.each_pair do |key, value|
+          if value.is_a?(Array)
+            value.each do |file|
+              if file.is_a?(Hash) && file.has_key?('checksum')
+                file['url'] ||= RestBase::build_uri(base_uri, ['file_store', 'checksums', file['checksum']])
+              end
+            end
+          end
+        end
+        cookbook['name'] ||= "#{name}-#{version}"
+        # TODO this feels wrong, but the real chef server doesn't expand this default
+  #      cookbook['version'] ||= version
+        cookbook['cookbook_name'] ||= name
+        cookbook['frozen?'] ||= false
+        cookbook['metadata'] ||= {}
+        cookbook['metadata']['version'] ||= version
+        # Sad to not be expanding defaults just because Chef doesn't :(
+#        cookbook['metadata']['name'] ||= name
+#        cookbook['metadata']['description'] ||= "A fabulous new cookbook"
+        cookbook['metadata']['long_description'] ||= ""
+#        cookbook['metadata']['maintainer'] ||= "YOUR_COMPANY_NAME"
+#        cookbook['metadata']['maintainer_email'] ||= "YOUR_EMAIL"
+#        cookbook['metadata']['license'] ||= "none"
+        cookbook['metadata']['dependencies'] ||= {}
+        cookbook['metadata']['attributes'] ||= {}
+        cookbook['metadata']['recipes'] ||= {}
+      end
+      cookbook['json_class'] ||= 'Chef::CookbookVersion'
+      cookbook['chef_type'] ||= 'cookbook_version'
+      if method == 'MIN'
+        cookbook['metadata'].delete('attributes')
+        cookbook['metadata'].delete('long_description')
+      end
+      cookbook
+    end
+
+    def self.normalize_node(node, name)
+      node['name'] ||= name
+      node['json_class'] ||= 'Chef::Node'
+      node['chef_type'] ||= 'node'
+      node['chef_environment'] ||= '_default'
+      node['override'] ||= {}
+      node['normal'] ||= {}
+      node['default'] ||= {}
+      node['automatic'] ||= {}
+      node['run_list'] ||= []
+      node['run_list'] = normalize_run_list(node['run_list'])
+      node
+    end
+
+    def self.normalize_role(role, name)
+      role['name'] ||= name
+      role['description'] ||= ''
+      role['json_class'] ||= 'Chef::Role'
+      role['chef_type'] ||= 'role'
+      role['default_attributes'] ||= {}
+      role['override_attributes'] ||= {}
+      role['run_list'] ||= []
+      role['run_list'] = normalize_run_list(role['run_list'])
+      role['env_run_lists'] ||= {}
+      role['env_run_lists'].each_pair do |env, run_list|
+        role['env_run_lists'][env] = normalize_run_list(run_list)
+      end
+      role
+    end
+
+    def self.normalize_run_list(run_list)
+      run_list.map{|item|
+        case item
+        when /^recipe\[.*\]$/
+          item # explicit recipe
+        when /^role\[.*\]$/
+          item # explicit role
+        else
+          "recipe[#{item}]"
+        end
+      }.uniq
+    end
+  end
+end

data/lib/chef_zero/data_store/data_already_exists_error.rb

@@ -0,0 +1,29 @@
+#
+# Author:: John Keiser (<jkeiser@opscode.com>)
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef_zero/data_store/data_error'
+
+module ChefZero
+  module DataStore
+    class DataAlreadyExistsError < DataError
+      def initialize(path, cause = nil)
+        super
+      end
+    end
+  end
+end

data/lib/chef_zero/data_store/data_error.rb

@@ -0,0 +1,31 @@
+#
+# Author:: John Keiser (<jkeiser@opscode.com>)
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+module ChefZero
+  module DataStore
+    class DataError < StandardError
+      def initialize(path, cause = nil)
+        @path = path
+        @cause = cause
+      end
+
+      attr_reader :path
+      attr_reader :cause
+    end
+  end
+end

data/lib/chef_zero/data_store/data_not_found_error.rb

@@ -0,0 +1,29 @@
+#
+# Author:: John Keiser (<jkeiser@opscode.com>)
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef_zero/data_store/data_error'
+
+module ChefZero
+  module DataStore
+    class DataNotFoundError < DataError
+      def initialize(path, cause = nil)
+        super
+      end
+    end
+  end
+end

data/lib/chef_zero/data_store/memory_store.rb

@@ -0,0 +1,167 @@
+#
+# Author:: John Keiser (<jkeiser@opscode.com>)
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef_zero/data_store/data_already_exists_error'
+require 'chef_zero/data_store/data_not_found_error'
+
+module ChefZero
+  module DataStore
+    class MemoryStore
+      def initialize
+        clear
+      end
+
+      def clear
+        @data = {}
+
+        # Create containers
+        create_dir([], 'clients')
+        create_dir([], 'cookbooks')
+        create_dir([], 'data')
+        create_dir([], 'environments')
+        create_dir([], 'file_store')
+        create_dir(['file_store'], 'checksums')
+        create_dir([], 'nodes')
+        create_dir([], 'roles')
+        create_dir([], 'sandboxes')
+        create_dir([], 'users')
+
+        # Set defaults
+        create(['clients'], 'chef-validator', '{ "validator": true }')
+        create(['clients'], 'chef-webui', '{ "admin": true }')
+        create(['environments'], '_default', '{ "description": "The default Chef environment" }')
+        create(['users'], 'admin', '{ "admin": true }')
+      end
+
+      def create_dir(path, name, *options)
+        parent = _get(path, options.include?(:recursive))
+
+        if parent.has_key?(name)
+          if !options.include?(:recursive)
+            raise DataAlreadyExistsError.new(path + [name])
+          end
+        else
+          parent[name] = {}
+        end
+      end
+
+      def create(path, name, data, *options)
+        if !data.is_a?(String)
+          raise "set only works with strings"
+        end
+
+        parent = _get(path, options.include?(:create_dir))
+
+        if parent.has_key?(name)
+          raise DataAlreadyExistsError.new(path + [name])
+        end
+        parent[name] = data
+      end
+
+      def get(path, request=nil)
+        value = _get(path)
+        if value.is_a?(Hash)
+          raise "get() called on directory #{path.join('/')}"
+        end
+        value
+      end
+
+      def set(path, data, *options)
+        if !data.is_a?(String)
+          raise "set only works with strings: #{path} = #{data.inspect}"
+        end
+
+        # Get the parent
+        parent = _get(path[0..-2], options.include?(:create_dir))
+
+        if !options.include?(:create) && !parent[path[-1]]
+          raise DataNotFoundError.new(path)
+        end
+        parent[path[-1]] = data
+      end
+
+      def delete(path)
+        parent = _get(path[0,path.length-1])
+        if !parent.has_key?(path[-1])
+          raise DataNotFoundError.new(path)
+        end
+        if !parent[path[-1]].is_a?(String)
+          raise "delete only works with strings: #{path}"
+        end
+        parent.delete(path[-1])
+      end
+
+      def delete_dir(path)
+        parent = _get(path[0,path.length-1])
+        if !parent.has_key?(path[-1])
+          raise DataNotFoundError.new(path)
+        end
+        if !parent[path[-1]].is_a?(Hash)
+          raise "delete_dir only works with directories: #{path}"
+        end
+        parent.delete(path[-1])
+      end
+
+      def list(path)
+        dir = _get(path)
+        if !dir.is_a? Hash
+          raise "list only works with directories (#{path} = #{dir.class}"
+        end
+        dir.keys.sort
+      end
+
+      def exists?(path)
+        begin
+          get(path)
+          return true
+        rescue DataNotFoundError
+          return false
+        end
+      end
+
+      def exists_dir?(path)
+        begin
+          dir = _get(path)
+          if !dir.is_a? Hash
+            raise "exists_dir? only works with directories (#{path} = #{dir.class}"
+          end
+          return true
+        rescue DataNotFoundError
+          return false
+        end
+      end
+
+      private
+
+      def _get(path, create_dir=false)
+        value = @data
+        path.each_with_index do |path_part, index|
+          if !value.has_key?(path_part)
+            if create_dir
+              value[path_part] = {}
+            else
+              raise DataNotFoundError.new(path[0,index+1])
+            end
+          end
+          value = value[path_part]
+        end
+        value
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/actor_endpoint.rb

@@ -0,0 +1,68 @@
+require 'json'
+require 'chef_zero/endpoints/rest_object_endpoint'
+require 'chef_zero/data_normalizer'
+
+module ChefZero
+  module Endpoints
+    # /clients/* and /users/*
+    class ActorEndpoint < RestObjectEndpoint
+      def put(request)
+        # Find out if we're updating the public key.
+        request_body = JSON.parse(request.body, :create_additions => false)
+        if request_body['public_key'].nil?
+          # If public_key is null, then don't overwrite it.  Weird patchiness.
+          body_modified = true
+          request_body.delete('public_key')
+        else
+          updating_public_key = true
+        end
+
+        # Generate private_key if requested.
+        if request_body.has_key?('private_key')
+          body_modified = true
+          if request_body['private_key']
+            private_key, public_key = server.gen_key_pair
+            updating_public_key = true
+            request_body['public_key'] = public_key
+          end
+          request_body.delete('private_key')
+        end
+
+        # Save request
+        request.body = JSON.pretty_generate(request_body) if body_modified
+
+        # PUT /clients is patchy
+        request.body = patch_request_body(request)
+
+        result = super(request)
+
+        # Inject private_key into response, delete public_key/password if applicable
+        if result[0] == 200
+          response = JSON.parse(result[2], :create_additions => false)
+          response['private_key'] = private_key if private_key
+          response.delete('public_key') if !updating_public_key && request.rest_path[0] == 'users'
+          response.delete('password')
+          # For PUT /clients, a rename returns 201.
+          if request_body['name'] && request.rest_path[1] != request_body['name']
+            json_response(201, response)
+          else
+            json_response(200, response)
+          end
+        else
+          result
+        end
+      end
+
+      def populate_defaults(request, response_json)
+        response = JSON.parse(response_json, :create_additions => false)
+        if request.rest_path[0] == 'clients'
+          response = DataNormalizer.normalize_client(response, request.rest_path[1])
+        else
+          response = DataNormalizer.normalize_user(response, request.rest_path[1])
+        end
+        JSON.pretty_generate(response)
+      end
+    end
+  end
+end
+