chef-server-api 0.8.2

data/app/controllers/clients.rb

@@ -0,0 +1,111 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Nuo Yan (<nuo@opscode.com>)
+# Copyright:: Copyright (c) 2008 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/api_client'
+
+class ChefServerApi::Clients < ChefServerApi::Application
+  provides :json
+
+  before :authenticate_every
+  before :is_admin, :only => :index
+  before :is_correct_node, :only => [ :show, :create, :update, :destroy ]
+  
+  # GET /clients
+  def index
+    @list = Chef::ApiClient.cdb_list(true)
+    display(@list.inject({}) { |result, element| result[element.name] = absolute_slice_url(:client, :id => element.name); result })
+  end
+
+  # GET /clients/:id
+  def show
+    begin
+      @client = Chef::ApiClient.cdb_load(params[:id])
+    rescue Chef::Exceptions::CouchDBNotFound => e
+      raise NotFound, "Cannot load client #{params[:id]}"
+    end
+    #display({ :name => @client.name, :admin => @client.admin, :public_key => @client.public_key })
+    display @client
+  end
+
+  # POST /clients
+  def create
+    exists = true 
+    if params.has_key?(:inflated_object)
+      params[:name] ||= params[:inflated_object].name
+      params[:admin] ||= params[:inflated_object].admin
+    end
+
+    begin
+      Chef::ApiClient.cdb_load(params[:name])
+    rescue Chef::Exceptions::CouchDBNotFound
+      exists = false 
+    end
+    raise Conflict, "Client already exists" if exists
+
+    @client = Chef::ApiClient.new
+    @client.name(params[:name])
+    @client.admin(params[:admin]) if params[:admin]
+    @client.create_keys
+    @client.cdb_save
+    
+    self.status = 201
+    headers['Location'] = absolute_slice_url(:client, @client.name)
+    display({ :uri => absolute_slice_url(:client, @client.name), :private_key => @client.private_key })
+  end
+
+  # PUT /clients/:id
+  def update
+    if params.has_key?(:inflated_object)
+      params[:private_key] ||= params[:inflated_object].private_key
+      params[:admin] ||= params[:inflated_object].admin
+    end
+
+    begin
+      @client = Chef::ApiClient.cdb_load(params[:id])
+    rescue Chef::Exceptions::CouchDBNotFound => e
+      raise NotFound, "Cannot load client #{params[:id]}"
+    end
+    
+    @client.admin(params[:admin]) unless params[:admin].nil?
+
+    results = { :name => @client.name, :admin => @client.admin }
+
+    if params[:private_key] == true
+      @client.create_keys
+      results[:private_key] = @client.private_key
+    end
+
+    @client.cdb_save
+
+    display(results)
+  end
+
+  # DELETE /clients/:id
+  def destroy
+    begin
+      @client = Chef::ApiClient.cdb_load(params[:id])
+    rescue Chef::Exceptions::CouchDBNotFound => e
+      raise NotFound, "Cannot load client #{params[:id]}"
+    end
+    @client.cdb_destroy
+    display({ :name => @client.name })
+  end
+
+end
+

data/app/controllers/cookbooks.rb

@@ -0,0 +1,213 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Christopher Brown (<cb@opscode.com>)
+# Author:: Christopher Walters (<cw@opscode.com>)
+# Copyright:: Copyright (c) 2008, 2009 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' / 'cookbook_loader'
+require 'chef' / 'cookbook' / 'metadata'
+
+class ChefServerApi::Cookbooks < ChefServerApi::Application
+  
+  provides :json
+
+  before :authenticate_every
+
+  include Chef::Mixin::Checksum
+  include Merb::ChefServerApi::TarballHelper
+  
+  def index
+    cl = Chef::CookbookLoader.new
+    cookbook_list = Hash.new
+    cl.each do |cookbook|
+      cookbook_list[cookbook.name] = absolute_slice_url(:cookbook, :id => cookbook.name.to_s) 
+    end
+    display cookbook_list 
+  end
+
+  def show
+    cl = Chef::CookbookLoader.new
+    begin
+      cookbook = cl[params[:id]]
+    rescue ArgumentError => e
+      raise NotFound, "Cannot find a cookbook named #{params[:id]}"
+    end
+    results = load_cookbook_files(cookbook)
+    results[:name] = cookbook.name.to_s
+    results[:metadata] = cl.metadata[cookbook.name.to_sym]
+    display results 
+  end
+ 
+  def show_segment
+    cl = Chef::CookbookLoader.new
+    begin
+      cookbook = cl[params[:cookbook_id]]
+    rescue ArgumentError => e
+      raise NotFound, "Cannot find a cookbook named #{params[:cookbook_id]}" 
+    end
+    cookbook_files = load_cookbook_files(cookbook)
+    raise NotFound unless cookbook_files.has_key?(params[:segment].to_sym)
+
+    if params[:id]
+      case params[:segment]
+      when "templates","files"
+        if params[:recursive]
+          serve_directory_preferred(cookbook, params[:segment], cookbook_files[params[:segment].to_sym])
+        else
+          serve_segment_preferred(cookbook, params[:segment], cookbook_files[params[:segment].to_sym])
+        end
+      else
+        serve_segment_file(cookbook, params[:segment], cookbook_files[params[:segment].to_sym])
+      end
+    else
+      display cookbook_files[params[:segment].to_sym]
+    end
+  end
+
+  def serve_segment_preferred(cookbook, segment, files)
+
+    to_send = nil
+    
+    preferences.each do |pref|
+      unless to_send
+        Chef::Log.debug("Looking for a file with name `#{params[:id]}' and specificity #{pref}")
+        to_send = files.detect do |file| 
+          Chef::Log.debug("#{pref.inspect} #{file.inspect}")
+          file[:name] == params[:id] && file[:specificity] == pref
+          
+        end
+      end
+    end
+
+    raise NotFound, "Cannot find a suitable #{segment} file for #{params[:id]}!" unless to_send 
+    current_checksum = to_send[:checksum] 
+    Chef::Log.debug("#{to_send[:name]} Client Checksum: #{params[:checksum]}, Server Checksum: #{current_checksum}") 
+    if current_checksum == params[:checksum]
+      raise NotModified, "File #{to_send[:name]} has not changed"
+    else
+      file_name = nil
+      segment_files(segment.to_sym, cookbook).each do |f|
+        if f =~ /#{to_send[:specificity]}\/#{to_send[:name]}$/
+          file_name = File.expand_path(f)
+          break 
+        end
+      end
+      raise NotFound, "Cannot find the real file for #{to_send[:specificity]} #{to_send[:name]} - this is a 42 error (shouldn't ever happen)" unless file_name
+      send_file(file_name)
+    end
+  end
+  
+  def serve_directory_preferred(cookbook, segment, files)
+    preferred_dir_contents = []
+    preferences.each do |preference|
+      preferred_dir_contents = files.select { |file| file[:name] =~ /^#{params[:id]}/ && file[:specificity] == preference  }
+      break unless preferred_dir_contents.empty?
+    end
+    
+    raise NotFound, "Cannot find a suitable directory for #{params[:id]}" if preferred_dir_contents.empty?
+    
+    display preferred_dir_contents.map { |file| file[:name].sub(/^#{params[:id]}/, '') }
+  end
+  
+  def preferences
+    ["host-#{params[:fqdn]}",
+    "#{params[:platform]}-#{params[:version]}",
+    "#{params[:platform]}",
+    "default"]
+  end
+
+  def serve_segment_file(cookbook, segment, files)
+    to_send = files.detect { |f| f[:name] == params[:id] } 
+    raise NotFound, "Cannot find a suitable #{segment} file!" unless to_send 
+    current_checksum = to_send[:checksum] 
+    Chef::Log.debug("#{to_send[:name]} Client Checksum: #{params[:checksum]}, Server Checksum: #{current_checksum}") 
+    if current_checksum == params[:checksum]
+      raise NotModified, "File #{to_send[:name]} has not changed"
+    else
+      file_name = nil
+      segment_files(segment.to_sym, cookbook).each do |f|
+        next unless File.basename(f) == to_send[:name]
+        file_name = File.expand_path(f)
+      end
+      raise NotFound, "Cannot find the real file for #{to_send[:name]} - this is a 42 error (shouldn't ever happen)" unless file_name
+      send_file(file_name)
+    end
+  end
+  
+  def create
+    # validate name and file parameters and throw an error if a cookbook with the same name already exists
+    raise BadRequest, "missing required parameter: name" unless params[:name]
+    desired_name = params[:name]
+    raise BadRequest, "invalid parameter: name must be at least one character long and contain only letters, numbers, periods (.), underscores (_), and hyphens (-)" unless desired_name =~ /\A[\w.-]+\Z/
+    begin
+      validate_file_parameter(desired_name, params[:file])
+    rescue FileParameterException => te
+      raise BadRequest, te.message
+    end
+    
+    begin
+      Chef::CookbookLoader.new[desired_name]
+      raise BadRequest, "Cookbook with the name #{desired_name} already exists"
+    rescue ArgumentError
+    end
+    
+    expand_tarball_and_put_in_repository(desired_name, params[:file][:tempfile])
+    
+    # construct successful response
+    self.status = 201
+    location = absolute_slice_url(:cookbook, :id => desired_name)
+    headers['Location'] = location
+    result = { 'uri' => location }
+    display result
+  end
+  
+  def get_tarball
+    cookbook_name = params[:cookbook_id]
+    expected_location = cookbook_location(cookbook_name)
+    raise NotFound, "Cannot find cookbook named #{cookbook_name} at #{expected_location}. Note: Tarball generation only applies to cookbooks under the first directory in the server's Chef::Config.cookbook_path variable and does to apply overrides." unless File.directory? expected_location
+    
+    send_file(get_or_create_cookbook_tarball_location(cookbook_name))
+  end
+  
+  def update
+    cookbook_name = params[:cookbook_id]
+    cookbook_path = cookbook_location(cookbook_name)
+    raise NotFound, "Cannot find cookbook named #{cookbook_name}" unless File.directory? cookbook_path
+    begin
+      validate_file_parameter(cookbook_name, params[:file])
+    rescue FileParameterException => te
+      raise BadRequest, te.message
+    end
+    
+    expand_tarball_and_put_in_repository(cookbook_name, params[:file][:tempfile])
+    
+    display Hash.new
+  end
+  
+  def destroy
+    cookbook_name = params[:id]
+    cookbook_path = cookbook_location(cookbook_name)
+    raise NotFound, "Cannot find cookbook named #{cookbook_name}" unless File.directory? cookbook_path
+
+    FileUtils.rm_rf(cookbook_path)
+    FileUtils.rm_f(cookbook_tarball_location(cookbook_name))
+
+    display Hash.new
+  end
+  
+end
+

data/app/controllers/data.rb

@@ -0,0 +1,75 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Christopher Brown (<cb@opscode.com>)
+# Copyright:: Copyright (c) 2008 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/data_bag'
+
+class ChefServerApi::Data < ChefServerApi::Application
+  
+  provides :json
+  
+  before :authenticate_every
+  before :is_admin, :only => [ :create, :destroy ]
+  
+  def index
+    @bag_list = Chef::DataBag.cdb_list(false)
+    display(@bag_list.inject({}) { |r,b| r[b] = absolute_slice_url(:datum, :id => b); r })
+    
+  end
+
+  def show
+    begin
+      @data_bag = Chef::DataBag.cdb_load(params[:id])
+    rescue Chef::Exceptions::CouchDBNotFound => e
+      raise NotFound, "Cannot load data bag #{params[:id]}"
+    end
+    display(@data_bag.list.inject({}) { |res, i| res[i] = absolute_slice_url(:data_bag_item, :data_bag_id => @data_bag.name, :id => i); res })
+  end
+
+  def create
+    @data_bag = nil
+    if params.has_key?("inflated_object")
+      @data_bag = params["inflated_object"]
+    else
+      @data_bag = Chef::DataBag.new
+      @data_bag.name(params["name"])
+    end
+    exists = true 
+    begin
+      Chef::DataBag.cdb_load(@data_bag.name)
+    rescue Chef::Exceptions::CouchDBNotFound
+      exists = false
+    end
+    raise Conflict, "Data bag already exists" if exists
+    self.status = 201
+    @data_bag.cdb_save
+    display({ :uri => absolute_slice_url(:datum, :id => @data_bag.name) })
+  end
+
+  def destroy
+    begin
+      @data_bag = Chef::DataBag.cdb_load(params[:id])
+    rescue Chef::Exceptions::CouchDBNotFound => e 
+      raise NotFound, "Cannot load data bag #{params[:id]}"
+    end
+    @data_bag.cdb_destroy
+    @data_bag.couchdb_rev = nil
+    display @data_bag
+  end
+  
+end

data/app/controllers/data_item.rb

@@ -0,0 +1,108 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Christopher Brown (<cb@opscode.com>)
+# Author:: Nuo Yan (<nuo@opscode.com>)
+# Copyright:: Copyright (c) 2008 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/data_bag'
+require 'chef/data_bag_item'
+
+class ChefServerApi::DataItem < ChefServerApi::Application
+  
+  provides :json
+  
+  before :populate_data_bag
+  before :authenticate_every
+  before :is_admin, :only => [ :create, :update, :destroy ]
+
+  def populate_data_bag
+    begin
+      @data_bag = Chef::DataBag.cdb_load(params[:data_bag_id])
+    rescue Chef::Exceptions::CouchDBNotFound => e
+      raise NotFound, "Cannot load data bag #{params[:data_bag_id]}"
+    end
+  end
+  
+  def show
+    begin
+      @data_bag_item = Chef::DataBagItem.cdb_load(params[:data_bag_id], params[:id])
+    rescue Chef::Exceptions::CouchDBNotFound => e
+      raise NotFound, "Cannot load data bag #{params[:data_bag_id]} item #{params[:id]}"
+    end
+    display @data_bag_item.raw_data
+  end
+
+  def create
+    raw_data = nil
+    if params.has_key?("inflated_object")
+      raw_data = params["inflated_object"].raw_data
+    else
+      raw_data = params
+      raw_data.delete(:action)
+      raw_data.delete(:controller)
+      raw_data.delete(:data_bag_id)
+    end
+    @data_bag_item = nil
+    begin
+      @data_bag_item = Chef::DataBagItem.cdb_load(@data_bag.name, params[:id])
+    rescue Chef::Exceptions::CouchDBNotFound
+      @data_bag_item = Chef::DataBagItem.new
+      @data_bag_item.data_bag(@data_bag.name)
+    else 
+      raise Conflict, "Databag Item #{params[:id]} already exists" if @data_bag_item
+    end
+    @data_bag_item.raw_data = raw_data
+    @data_bag_item.cdb_save
+    display @data_bag_item.raw_data
+  end
+  
+  def update
+    raw_data = nil
+    if params.has_key?("inflated_object")
+      raw_data = params["inflated_object"].raw_data
+    else
+      raw_data = params
+      raw_data.delete(:action)
+      raw_data.delete(:controller)
+      raw_data.delete(:data_bag_id)
+    end
+        
+    begin 
+      @data_bag_item = Chef::DataBagItem.cdb_load(@data_bag.name, params[:id])
+    rescue Chef::Exceptions::CouchDBNotFound => e
+      raise NotFound, "Cannot load Databag Item #{params[:id]}"
+    end 
+    
+    @data_bag_item.raw_data = raw_data
+    @data_bag_item.cdb_save
+    display @data_bag_item.raw_data
+
+  end 
+  
+
+  def destroy
+    begin
+      @data_bag_item = Chef::DataBagItem.cdb_load(params[:data_bag_id], params[:id])
+    rescue Chef::Exceptions::CouchDBNotFound => e 
+      raise NotFound, "Cannot load data bag #{params[:data_bag_id]} item #{params[:id]}"
+    end
+    @data_bag_item.cdb_destroy
+    @data_bag_item.couchdb_rev = nil
+    display @data_bag_item
+  end
+  
+end