chef-zero 0.9

data/lib/chef_zero/endpoints/cookbooks_endpoint.rb

@@ -0,0 +1,12 @@
+require 'chef_zero/endpoints/cookbooks_base'
+
+module ChefZero
+  module Endpoints
+    # /cookbooks
+    class CookbooksEndpoint < CookbooksBase
+      def get(request)
+        json_response(200, format_cookbooks_list(request, data['cookbooks']))
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/data_bag_endpoint.rb

@@ -0,0 +1,50 @@
+require 'json'
+require 'chef_zero/endpoints/rest_list_endpoint'
+require 'chef_zero/endpoints/data_bag_item_endpoint'
+require 'chef_zero/rest_error_response'
+
+module ChefZero
+  module Endpoints
+    # /data/NAME
+    class DataBagEndpoint < RestListEndpoint
+      def initialize(server)
+        super(server, 'id')
+      end
+
+      def post(request)
+        key = JSON.parse(request.body, :create_additions => false)[identity_key]
+        response = super(request)
+        if response[0] == 201
+          already_json_response(201, DataBagItemEndpoint::populate_defaults(request, request.body, request.rest_path[1], key))
+        else
+          response
+        end
+      end
+
+      def get_key(contents)
+        data_bag_item = JSON.parse(contents, :create_additions => false)
+        if data_bag_item['json_class'] == 'Chef::DataBagItem' && data_bag_item['raw_data']
+          data_bag_item['raw_data']['id']
+        else
+          data_bag_item['id']
+        end
+      end
+
+      def delete(request)
+        key = request.rest_path[1]
+        container = data['data']
+        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)
+        json_response(200, {
+          'chef_type' => 'data_bag',
+          'json_class' => 'Chef::DataBag',
+          'name' => key
+        })
+      end
+    end
+  end
+end
+

data/lib/chef_zero/endpoints/data_bag_item_endpoint.rb

@@ -0,0 +1,25 @@
+require 'json'
+require 'chef_zero/endpoints/rest_object_endpoint'
+require 'chef_zero/endpoints/data_bag_item_endpoint'
+require 'chef_zero/data_normalizer'
+
+module ChefZero
+  module Endpoints
+    # /data/NAME/NAME
+    class DataBagItemEndpoint < RestObjectEndpoint
+      def initialize(server)
+        super(server, 'id')
+      end
+
+      def populate_defaults(request, response_json)
+        DataBagItemEndpoint::populate_defaults(request, response_json, request.rest_path[1], request.rest_path[2])
+      end
+
+      def self.populate_defaults(request, response_json, data_bag, data_bag_item)
+        response = JSON.parse(response_json, :create_additions => false)
+        response = DataNormalizer.normalize_data_bag_item(response, data_bag, data_bag_item, request.method)
+        JSON.pretty_generate(response)
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/data_bags_endpoint.rb

@@ -0,0 +1,21 @@
+require 'json'
+require 'chef_zero/endpoints/rest_list_endpoint'
+
+module ChefZero
+  module Endpoints
+    # /data
+    class DataBagsEndpoint < RestListEndpoint
+      def post(request)
+        container = get_data(request)
+        contents = request.body
+        name = JSON.parse(contents, :create_additions => false)[identity_key]
+        if container[name]
+          error(409, "Object already exists")
+        else
+          container[name] = {}
+          json_response(201, {"uri" => "#{build_uri(request.base_uri, request.rest_path + [name])}"})
+        end
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/environment_cookbook_endpoint.rb

@@ -0,0 +1,24 @@
+require 'json'
+require 'chef_zero/endpoints/cookbooks_base'
+
+module ChefZero
+  module Endpoints
+    # /environments/NAME/cookbooks/NAME
+    class EnvironmentCookbookEndpoint < CookbooksBase
+      def get(request)
+        cookbook_name = request.rest_path[3]
+        environment = JSON.parse(get_data(request, request.rest_path[0..1]), :create_additions => false)
+        constraints = environment['cookbook_versions'] || {}
+        cookbook = get_data(request, request.rest_path[2..3])
+        if request.query_params['num_versions'] == 'all'
+          num_versions = nil
+        elsif request.query_params['num_versions']
+          num_versions = request.query_params['num_versions'].to_i
+        else
+          num_versions = nil
+        end
+        json_response(200, format_cookbooks_list(request, { cookbook_name => cookbook }, constraints, num_versions))
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/environment_cookbook_versions_endpoint.rb

@@ -0,0 +1,114 @@
+require 'json'
+require 'chef/exceptions' # Needed so Chef::Version/VersionConstraint load
+require 'chef/version_class'
+require 'chef/version_constraint'
+require 'chef_zero/rest_base'
+require 'chef_zero/rest_error_response'
+
+module ChefZero
+  module Endpoints
+    # /environments/NAME/cookbook_versions
+    class EnvironmentCookbookVersionsEndpoint < RestBase
+      def cookbooks
+        data['cookbooks']
+      end
+
+      def environments
+        data['environments']
+      end
+
+      def post(request)
+        # Get the list of cookbooks and versions desired by the runlist
+        desired_versions = {}
+        run_list = JSON.parse(request.body, :create_additions => false)['run_list']
+        run_list.each do |run_list_entry|
+          if run_list_entry =~ /(.+)\@(.+)/
+            raise RestErrorResponse.new(412, "No such cookbook: #{$1}") if !cookbooks[$1]
+            raise RestErrorResponse.new(412, "No such cookbook version for cookbook #{$1}: #{$2}") if !cookbooks[$1][$2]
+            desired_versions[$1] = [ $2 ]
+          else
+            raise RestErrorResponse.new(412, "No such cookbook: #{run_list_entry}") if !cookbooks[run_list_entry]
+            desired_versions[run_list_entry] = cookbooks[run_list_entry].keys
+          end
+        end
+
+        # Filter by environment constraints
+        environment = JSON.parse(get_data(request, request.rest_path[0..1]), :create_additions => false)
+        environment_constraints = environment['cookbook_versions']
+
+        desired_versions.each_key do |name|
+          desired_versions = filter_by_constraint(desired_versions, name, environment_constraints[name])
+        end
+
+        # Depsolve!
+        solved = depsolve(desired_versions.keys, desired_versions, environment_constraints)
+        if !solved
+          return raise RestErrorResponse.new(412, "Unsolvable versions!")
+        end
+
+        result = {}
+        solved.each_pair do |name, versions|
+          result[name] = JSON.parse(data['cookbooks'][name][versions[0]], :create_additions => false)
+        end
+        json_response(200, result)
+      end
+
+      def depsolve(unsolved, desired_versions, environment_constraints)
+        return nil if desired_versions.values.any? { |versions| versions.empty? }
+
+        # If everything is already
+        solve_for = unsolved[0]
+        return desired_versions if !solve_for
+
+        # Go through each desired version of this cookbook, starting with the latest,
+        # until we find one we can solve successfully with
+        sort_versions(desired_versions[solve_for]).each do |desired_version|
+          new_desired_versions = desired_versions.clone
+          new_desired_versions[solve_for] = [ desired_version ]
+          new_unsolved = unsolved[1..-1]
+
+          # Pick this cookbook, and add dependencies
+          cookbook_obj = JSON.parse(cookbooks[solve_for][desired_version], :create_additions => false)
+          dep_not_found = false
+          cookbook_obj['metadata']['dependencies'].each_pair do |dep_name, dep_constraint|
+            # If the dep is not already in the list, add it to the list to solve
+            # and bring in all environment-allowed cookbook versions to desired_versions
+            if !new_desired_versions.has_key?(dep_name)
+              new_unsolved = new_unsolved + [dep_name]
+              # If the dep is missing, we will try other versions of the cookbook that might not have the bad dep.
+              if !cookbooks[dep_name]
+                dep_not_found = true
+                break
+              end
+              new_desired_versions[dep_name] = cookbooks[dep_name].keys
+              new_desired_versions = filter_by_constraint(new_desired_versions, dep_name, environment_constraints[dep_name])
+            end
+            new_desired_versions = filter_by_constraint(new_desired_versions, dep_name, dep_constraint)
+          end
+
+          next if dep_not_found
+
+          # Depsolve children with this desired version!  First solution wins.
+          result = depsolve(new_unsolved, new_desired_versions, environment_constraints)
+          return result if result
+        end
+        return nil
+      end
+
+      def sort_versions(versions)
+        result = versions.sort_by { |version| Chef::Version.new(version) }
+        result.reverse
+      end
+
+      def filter_by_constraint(versions, cookbook_name, constraint)
+        return versions if !constraint
+        constraint = Chef::VersionConstraint.new(constraint)
+        new_versions = versions[cookbook_name]
+        new_versions = new_versions.select { |version| constraint.include?(version) }
+        result = versions.clone
+        result[cookbook_name] = new_versions
+        result
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/environment_cookbooks_endpoint.rb

@@ -0,0 +1,22 @@
+require 'json'
+require 'chef_zero/endpoints/cookbooks_base'
+
+module ChefZero
+  module Endpoints
+    # /environments/NAME/cookbooks
+    class EnvironmentCookbooksEndpoint < CookbooksBase
+      def get(request)
+        environment = JSON.parse(get_data(request, request.rest_path[0..1]), :create_additions => false)
+        constraints = environment['cookbook_versions'] || {}
+        if request.query_params['num_versions'] == 'all'
+          num_versions = nil
+        elsif request.query_params['num_versions']
+          num_versions = request.query_params['num_versions'].to_i
+        else
+          num_versions = 1
+        end
+        json_response(200, format_cookbooks_list(request, data['cookbooks'], constraints, num_versions))
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/environment_endpoint.rb

@@ -0,0 +1,33 @@
+require 'json'
+require 'chef_zero/endpoints/rest_object_endpoint'
+require 'chef_zero/data_normalizer'
+
+module ChefZero
+  module Endpoints
+    # /environments/NAME
+    class EnvironmentEndpoint < RestObjectEndpoint
+      def delete(request)
+        if request.rest_path[1] == "_default"
+          # 405, really?
+          error(405, "The '_default' environment cannot be modified.")
+        else
+          super(request)
+        end
+      end
+
+      def put(request)
+        if request.rest_path[1] == "_default"
+          error(405, "The '_default' environment cannot be modified.")
+        else
+          super(request)
+        end
+      end
+
+      def populate_defaults(request, response_json)
+        response = JSON.parse(response_json, :create_additions => false)
+        response = DataNormalizer.normalize_environment(response, request.rest_path[1])
+        JSON.pretty_generate(response)
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/environment_nodes_endpoint.rb

@@ -0,0 +1,23 @@
+require 'json'
+require 'chef_zero/rest_base'
+
+module ChefZero
+  module Endpoints
+    # /environment/NAME/nodes
+    class EnvironmentNodesEndpoint < RestBase
+      def get(request)
+        # 404 if environment does not exist
+        get_data(request, request.rest_path[0..1])
+
+        result = {}
+        data['nodes'].each_pair do |name, node|
+          node_json = JSON.parse(node, :create_additions => false)
+          if node['chef_environment'] == request.rest_path[1]
+            result[name] = build_uri(request.base_uri, 'nodes', name)
+          end
+        end
+        json_response(200, result)
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/environment_recipes_endpoint.rb

@@ -0,0 +1,22 @@
+require 'json'
+require 'chef_zero/endpoints/cookbooks_base'
+
+module ChefZero
+  module Endpoints
+    # /environment/NAME/recipes
+    class EnvironmentRecipesEndpoint < CookbooksBase
+      def get(request)
+        environment = JSON.parse(get_data(request, request.rest_path[0..1]), :create_additions => false)
+        constraints = environment['cookbook_versions'] || {}
+        result = []
+        filter_cookbooks(data['cookbooks'], constraints, 1) do |name, versions|
+          if versions.size > 0
+            cookbook = JSON.parse(data['cookbooks'][name][versions[0]], :create_additions => false)
+            result += recipe_names(name, cookbook)
+          end
+        end
+        json_response(200, result.sort)
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/environment_role_endpoint.rb

@@ -0,0 +1,35 @@
+require 'json'
+require 'chef_zero/endpoints/cookbooks_base'
+
+module ChefZero
+  module Endpoints
+    # /environments/NAME/roles/NAME
+    # /roles/NAME/environments/NAME
+    class EnvironmentRoleEndpoint < CookbooksBase
+      def get(request)
+        # 404 if environment does not exist
+        if request.rest_path[0] == 'environments'
+          environment_path = request.rest_path[0..1]
+          role_path = request.rest_path[2..3]
+        else
+          environment_path = request.rest_path[2..3]
+          role_path = request.rest_path[0..1]
+        end
+        get_data(request, environment_path)
+
+        role = JSON.parse(get_data(request, role_path), :create_additions => false)
+        environment_name = environment_path[1]
+        if environment_name == '_default'
+          run_list = role['run_list']
+        else
+          if role['env_run_lists']
+            run_list = role['env_run_lists'][environment_name]
+          else
+            run_list = nil
+          end
+        end
+        json_response(200, { 'run_list' => run_list })
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/file_store_file_endpoint.rb

@@ -0,0 +1,22 @@
+require 'chef_zero/rest_base'
+
+module ChefZero
+  module Endpoints
+    # The minimum amount of S3 necessary to support cookbook upload/download
+    # /file_store/FILE
+    class FileStoreFileEndpoint < RestBase
+      def json_only
+        false
+      end
+
+      def get(request)
+        [200, {"Content-Type" => 'application/x-binary'}, get_data(request) ]
+      end
+
+      def put(request)
+        data['file_store'][request.rest_path[1]] = request.body
+        json_response(200, {})
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/node_endpoint.rb

@@ -0,0 +1,17 @@
+require 'json'
+require 'chef_zero/endpoints/rest_object_endpoint'
+require 'chef_zero/data_normalizer'
+
+module ChefZero
+  module Endpoints
+    # /nodes/ID
+    class NodeEndpoint < RestObjectEndpoint
+      def populate_defaults(request, response_json)
+        node = JSON.parse(response_json, :create_additions => false)
+        node = DataNormalizer.normalize_node(node, request.rest_path[1])
+        JSON.pretty_generate(node)
+      end
+    end
+  end
+end
+

data/lib/chef_zero/endpoints/not_found_endpoint.rb

@@ -0,0 +1,9 @@
+module ChefZero
+  module Endpoints
+    class NotFoundEndpoint
+      def call(env)
+        return [404, {"Content-Type" => "application/json"}, "Object not found: #{env['REQUEST_PATH']}"]
+      end
+    end
+  end
+end