chef-zero 1.4.0.alpha-x86-mingw32

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, all_cookbooks_list, 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 = {}
+        list_data(request, ['nodes']).each do |name|
+          node = JSON.parse(get_data(request, ['nodes', name]), :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(all_cookbooks_list, constraints, 1) do |name, versions|
+          if versions.size > 0
+            cookbook = JSON.parse(get_data(request, ['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,36 @@
+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
+        # Verify that the environment exists
+        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_store.set(request.rest_path, request.body, :create, :create_dir)
+        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,11 @@
+require 'json'
+
+module ChefZero
+  module Endpoints
+    class NotFoundEndpoint
+      def call(request)
+        return [404, {"Content-Type" => "application/json"}, JSON.pretty_generate({"error" => ["Object not found: #{request.env['REQUEST_PATH']}"]})]
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/principal_endpoint.rb

@@ -0,0 +1,30 @@
+require 'json'
+require 'chef_zero'
+require 'chef_zero/rest_base'
+
+module ChefZero
+  module Endpoints
+    # /principals/NAME
+    class PrincipalEndpoint < RestBase
+      def get(request)
+        name = request.rest_path[-1]
+        json = get_data(request, [ 'users', name ], :nil)
+        if json
+          type = 'user'
+        else
+          json = get_data(request, [ 'clients', name ], :nil)
+          type = 'client'
+        end
+        if json
+          json_response(200, {
+            'name' => name,
+            'type' => type,
+            'public_key' => JSON.parse(json)['public_key'] || PUBLIC_KEY
+          })
+        else
+          error(404, 'Principal not found')
+        end
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/rest_list_endpoint.rb

@@ -0,0 +1,40 @@
+require 'json'
+require 'chef_zero/rest_base'
+
+module ChefZero
+  module Endpoints
+    # Typical REST list endpoint (/roles or /data/BAG)
+    class RestListEndpoint < RestBase
+      def initialize(server, identity_key = 'name')
+        super(server)
+        @identity_key = identity_key
+      end
+
+      attr_reader :identity_key
+
+      def get(request)
+        # Get the result
+        result_hash = {}
+        list_data(request).sort.each do |name|
+          result_hash[name] = "#{build_uri(request.base_uri, request.rest_path + [name])}"
+        end
+        json_response(200, result_hash)
+      end
+
+      def post(request)
+        contents = request.body
+        key = get_key(contents)
+        if key.nil?
+          error(400, "Must specify '#{identity_key}' in JSON")
+        else
+          create_data(request, request.rest_path, key, contents)
+          json_response(201, {'uri' => "#{build_uri(request.base_uri, request.rest_path + [key])}"})
+        end
+      end
+
+      def get_key(contents)
+        JSON.parse(contents, :create_additions => false)[identity_key]
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/rest_object_endpoint.rb

@@ -0,0 +1,61 @@
+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]
+        # If it's a rename, check for conflict and delete the old value
+        rename = key != request.rest_path[-1]
+        if rename
+          begin
+            data_store.create(request.rest_path[0..-2], key, request.body)
+          rescue DataStore::DataAlreadyExistsError
+            return error(409, "Cannot rename '#{request.rest_path[-1]}' to '#{key}': '#{key}' already exists")
+          end
+          delete_data(request)
+        else
+          set_data(request, request.rest_path, request.body)
+        end
+        already_json_response(200, populate_defaults(request, request.body))
+      end
+
+      def delete(request)
+        result = get_data(request)
+        delete_data(request)
+        already_json_response(200, populate_defaults(request, result))
+      end
+
+      def patch_request_body(request)
+        existing_value = get_data(request, nil, :nil)
+        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
+

data/lib/chef_zero/endpoints/role_endpoint.rb

@@ -0,0 +1,16 @@
+require 'json'
+require 'chef_zero/endpoints/rest_object_endpoint'
+require 'chef_zero/data_normalizer'
+
+module ChefZero
+  module Endpoints
+    # /roles/NAME
+    class RoleEndpoint < RestObjectEndpoint
+      def populate_defaults(request, response_json)
+        role = JSON.parse(response_json, :create_additions => false)
+        role = DataNormalizer.normalize_role(role, request.rest_path[1])
+        JSON.pretty_generate(role)
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/role_environments_endpoint.rb

@@ -0,0 +1,14 @@
+require 'json'
+require 'chef_zero/rest_base'
+
+module ChefZero
+  module Endpoints
+    # /roles/NAME/environments
+    class RoleEnvironmentsEndpoint < RestBase
+      def get(request)
+        role = JSON.parse(get_data(request, request.rest_path[0..1]), :create_additions => false)
+        json_response(200, [ '_default' ] + (role['env_run_lists'].keys || []))
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/sandbox_endpoint.rb

@@ -0,0 +1,27 @@
+require 'chef_zero/rest_base'
+require 'chef_zero/rest_error_response'
+require 'json'
+
+module ChefZero
+  module Endpoints
+    # /sandboxes/ID
+    class SandboxEndpoint < RestBase
+      def put(request)
+        existing_sandbox = JSON.parse(get_data(request), :create_additions => false)
+        existing_sandbox['checksums'].each do |checksum|
+          if !exists_data?(request, ['file_store', 'checksums', checksum])
+            raise RestErrorResponse.new(503, "Checksum not uploaded: #{checksum}")
+          end
+        end
+        delete_data(request)
+        json_response(200, {
+          :guid => request.rest_path[1],
+          :name => request.rest_path[1],
+          :checksums => existing_sandbox['checksums'],
+          :create_time => existing_sandbox['create_time'],
+          :is_completed => true
+        })
+      end
+    end
+  end
+end

data/lib/chef_zero/endpoints/sandboxes_endpoint.rb

@@ -0,0 +1,51 @@
+require 'json'
+require 'chef_zero/rest_base'
+
+module ChefZero
+  module Endpoints
+    # /sandboxes
+    class SandboxesEndpoint < RestBase
+      def initialize(server)
+        super(server)
+        @next_id = 1
+      end
+
+      def post(request)
+        sandbox_checksums = []
+
+        needed_checksums = JSON.parse(request.body, :create_additions => false)['checksums']
+        result_checksums = {}
+        needed_checksums.keys.each do |needed_checksum|
+          if list_data(request, ['file_store', 'checksums']).include?(needed_checksum)
+            result_checksums[needed_checksum] = { :needs_upload => false }
+          else
+            result_checksums[needed_checksum] = {
+              :needs_upload => true,
+              :url => build_uri(request.base_uri, ['file_store', 'checksums', needed_checksum])
+            }
+            sandbox_checksums << needed_checksum
+          end
+        end
+
+        # There is an obvious race condition here.
+        id = @next_id.to_s
+        @next_id+=1
+
+        time_str = Time.now.utc.strftime('%Y-%m-%dT%H:%M:%S%z')
+        time_str = "#{time_str[0..21]}:#{time_str[22..23]}"
+
+        create_data(request, request.rest_path, id, JSON.pretty_generate({
+          :create_time => time_str,
+          :checksums => sandbox_checksums
+        }))
+
+        json_response(201, {
+          :uri => build_uri(request.base_uri, request.rest_path + [id]),
+          :checksums => result_checksums,
+          :sandbox_id => id
+        })
+      end
+    end
+  end
+end
+