mbailey-chef 0.9.12.1

data/lib/chef/knife/configure_client.rb

@@ -0,0 +1,52 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 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/knife'
+
+class Chef
+  class Knife
+    class ConfigureClient < Knife
+      banner "knife configure client DIRECTORY"
+
+      def run
+        unless @config_dir = @name_args[0]
+          Chef::Log.fatal "You must provide the directory to put the files in"
+          show_usage
+          exit(1)
+        end
+
+        Chef::Log.info("Creating client configuration")
+        FileUtils.mkdir_p(@config_dir)
+        Chef::Log.info("Writing client.rb")
+        File.open(File.join(@config_dir, "client.rb"), "w") do |file|
+          file.puts('log_level        :info')
+          file.puts('log_location     STDOUT')
+          file.puts("chef_server_url  '#{Chef::Config[:chef_server_url]}'")
+          file.puts("validation_client_name '#{Chef::Config[:validation_client_name]}'")
+        end
+        Chef::Log.info("Writing validation.pem")
+        File.open(File.join(@config_dir, 'validation.pem'), "w") do |validation|
+          validation.puts(IO.read(Chef::Config[:validation_key]))
+        end
+      end
+
+    end
+  end
+end
+
+

data/lib/chef/knife/cookbook_bulk_delete.rb

@@ -0,0 +1,58 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Daniel DeLeo (<dan@opscode.com>)
+# Copyright:: Copyright (c) 2009, 2010 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/knife'
+require 'chef/knife/cookbook_delete'
+
+class Chef
+  class Knife
+    class CookbookBulkDelete < Knife
+
+      option :purge, :short => '-p', :long => '--purge', :boolean => true, :description => 'Permanently remove files from backing data store'
+      
+      banner "knife cookbook bulk delete REGEX (options)"
+
+      def run
+        unless regex_str = @name_args.first
+          Chef::Log.fatal("You must supply a regular expression to match the results against")
+          exit 42
+        end
+
+        regex = Regexp.new(regex_str)
+
+        all_cookbooks = Chef::CookbookVersion.list
+        cookbooks_names = all_cookbooks.keys.grep(regex)
+        cookbooks_to_delete = cookbooks_names.inject({}) { |hash, name| hash[name] = all_cookbooks[name];hash }
+        output(format_list_for_display(cookbooks_to_delete))
+
+        confirm("Do you really want to delete these cookbooks? All versions will be deleted. (Y/N) ", false)
+        
+        confirm("Files that are common to multiple cookbooks are shared, so purging the files may disable other cookbooks. Are you sure you want to purge files instead of just deleting the cookbooks") if config[:purge]
+        
+        cookbooks_names.each do |cookbook_name|
+          versions = rest.get_rest("cookbooks/#{cookbook_name}").values.flatten
+          versions.each do |version|
+            object = rest.delete_rest("cookbooks/#{cookbook_name}/#{version}#{config[:purge] ? "?purge=true" : ""}")
+            Chef::Log.info("Deleted cookbook  #{cookbook_name.ljust(25)} [#{version}]")
+          end
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/cookbook_create.rb

@@ -0,0 +1,209 @@
+#
+# Author:: Nuo Yan (<nuo@opscode.com>)
+# Copyright:: Copyright (c) 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/knife'
+require 'json'
+require 'uri'
+require 'chef/mixin/shell_out'
+
+class Chef
+  class Knife
+    class CookbookCreate < Knife
+      include Chef::Mixin::ShellOut
+
+      banner "knife cookbook create COOKBOOK (options)"
+
+      option :cookbook_path,
+        :short => "-o PATH",
+        :long => "--cookbook-path PATH",
+        :description => "The directory where the cookbook will be created"
+
+      option :readme_format,
+        :short => "-r FORMAT",
+        :long => "--readme-format FORMAT",
+        :description => "Format of the README file, supported formats are 'md' (markdown) and 'rdoc' (rdoc)",
+        :default => "rdoc"
+
+      option :cookbook_license,
+        :short => "-I LICENSE",
+        :long => "--license LICENSE",
+        :description => "License for cookbook, apachev2 or none"
+
+      option :cookbook_copyright,
+        :short => "-C COPYRIGHT",
+        :long => "--copyright COPYRIGHT",
+        :description => "Name of Copyright holder"
+
+      option :cookbook_email,
+        :short => "-E EMAIL",
+        :long => "--email EMAIL",
+        :description => "Email address of cookbook maintainer"
+
+      def run
+        self.config = Chef::Config.merge!(config)
+        if @name_args.length < 1
+          show_usage
+          Chef::Log.fatal("You must specify a cookbook name")
+          exit 1
+        end
+
+        if default_cookbook_path_empty? && parameter_empty?(config[:cookbook_path])
+          raise ArgumentError, "Default cookbook_path is not specified in the knife.rb config file, and a value to -o is not provided. Nowhere to write the new cookbook to."
+        end
+
+        cookbook_path = Array(config[:cookbook_path]).first
+        cookbook_name = @name_args.first
+        copyright = config[:cookbook_copyright] || "YOUR_COMPANY_NAME"
+        email = config[:cookbook_email] || "YOUR_EMAIL"
+        license = ((config[:cookbook_license] != "false") && config[:cookbook_license]) || "none"
+        create_cookbook(cookbook_path,cookbook_name, copyright, license)
+        create_readme(cookbook_path,cookbook_name)
+        create_metadata(cookbook_path,cookbook_name, copyright, email, license)
+    end
+
+    def create_cookbook(dir, cookbook_name, copyright, license)
+      msg("** Creating cookbook #{cookbook_name}")
+      shell_out "mkdir -p #{File.join(dir, cookbook_name, "attributes")}"
+      shell_out "mkdir -p #{File.join(dir, cookbook_name, "recipes")}"
+      shell_out "mkdir -p #{File.join(dir, cookbook_name, "definitions")}"
+      shell_out "mkdir -p #{File.join(dir, cookbook_name, "libraries")}"
+      shell_out "mkdir -p #{File.join(dir, cookbook_name, "resources")}"
+      shell_out "mkdir -p #{File.join(dir, cookbook_name, "providers")}"
+      shell_out "mkdir -p #{File.join(dir, cookbook_name, "files", "default")}"
+      shell_out "mkdir -p #{File.join(dir, cookbook_name, "templates", "default")}"
+      unless File.exists?(File.join(dir, cookbook_name, "recipes", "default.rb"))
+        open(File.join(dir, cookbook_name, "recipes", "default.rb"), "w") do |file|
+          file.puts <<-EOH
+#
+# Cookbook Name:: #{cookbook_name}
+# Recipe:: default
+#
+# Copyright #{Time.now.year}, #{copyright}
+#
+EOH
+          case license
+          when "apachev2"
+            file.puts <<-EOH
+# 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.
+#
+EOH
+          when "none"
+            file.puts <<-EOH
+# All rights reserved - Do Not Redistribute
+#
+EOH
+          end
+        end
+      end
+    end
+
+    def create_readme(dir, cookbook_name)
+      msg("** Creating README for cookbook: #{cookbook_name}")
+      unless File.exists?(File.join(dir, cookbook_name, "README.#{config[:readme_format]}"))
+        open(File.join(dir, cookbook_name, "README.#{config[:readme_format]}"), "w") do |file|
+          case config[:readme_format]
+          when "rdoc"
+            file.puts <<-EOH
+= DESCRIPTION:
+
+= REQUIREMENTS:
+
+= ATTRIBUTES:
+
+= USAGE:
+
+EOH
+          when "md","mkd","txt"
+            file.puts <<-EOH
+Description
+===========
+
+Requirements
+============
+
+Attributes
+==========
+
+Usage
+=====
+
+EOH
+          else
+            file.puts <<-EOH
+Description
+
+Requirements
+
+Attributes
+
+Usage
+
+EOH
+          end
+        end
+      end
+    end
+
+    def create_metadata(dir, cookbook_name, copyright, email, license)
+      msg("** Creating metadata for cookbook: #{cookbook_name}")
+
+      license_name = case license
+                     when "apachev2"
+                       "Apache 2.0"
+                     when "none"
+                       "All rights reserved"
+                     end
+
+      unless File.exists?(File.join(dir, cookbook_name, "metadata.rb"))
+        open(File.join(dir, cookbook_name, "metadata.rb"), "w") do |file|
+          if File.exists?(File.join(dir, cookbook_name, "README.#{config[:readme_format]}"))
+            long_description = "long_description IO.read(File.join(File.dirname(__FILE__), 'README.#{config[:readme_format]}'))"
+          end
+          file.puts <<-EOH
+maintainer       "#{copyright}"
+maintainer_email "#{email}"
+license          "#{license_name}"
+description      "Installs/Configures #{cookbook_name}"
+#{long_description}
+version          "0.0.1"
+EOH
+        end
+      end
+    end
+
+    private
+      def default_cookbook_path_empty?
+        Chef::Config[:cookbook_path].nil? || Chef::Config[:cookbook_path].empty?
+      end
+
+      def parameter_empty?(parameter)
+        parameter.nil? || parameter.empty?
+      end
+    end
+  end
+end

data/lib/chef/knife/cookbook_delete.rb

@@ -0,0 +1,143 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 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/knife'
+
+class Chef
+  class Knife
+    class CookbookDelete < Knife
+      
+      option :all, :short => '-a', :long => '--all', :boolean => true, :description => 'delete all versions'
+
+      option :purge, :short => '-p', :long => '--purge', :boolean => true, :description => 'Permanently remove files from backing data store'
+
+      banner "knife cookbook delete COOKBOOK VERSION (options)"
+
+      def run
+        confirm("Files that are common to multiple cookbooks are shared, so purging the files may disable other cookbooks. Are you sure you want to purge files instead of just deleting the cookbook") if config[:purge]
+        @cookbook_name, @version = name_args
+        if @cookbook_name && @version
+          delete_explicit_version
+        elsif @cookbook_name && config[:all]
+          delete_all_versions
+        elsif @cookbook_name && @version.nil?
+          delete_without_explicit_version
+        elsif @cookbook_name.nil?
+          show_usage
+          Chef::Log.fatal("You must provide the name of the cookbook to delete")
+          exit(1)
+        end
+      end
+
+      def delete_explicit_version
+        delete_object(Chef::CookbookVersion, "#{@cookbook_name} version #{@version}", "cookbook") do
+          delete_request("cookbooks/#{@cookbook_name}/#{@version}")
+        end
+      end
+
+      def delete_all_versions
+        confirm("Do you really want to delete all versions of #{@cookbook_name}")
+        delete_all_without_confirmation
+      end
+
+      def delete_all_without_confirmation
+        # look up the available versions again just in case the user
+        # got to the list of versions to delete and selected 'all'
+        # and also a specific version
+        @available_versions = nil
+        Array(available_versions).each do |version|
+          delete_version_without_confirmation(version)
+        end
+      end
+
+      def delete_without_explicit_version
+        if available_versions.nil?
+          # we already logged an error or 2 about it, so just bail
+          exit(1)
+        elsif available_versions.size == 1
+          @version = available_versions.first
+          delete_explicit_version
+        else
+          versions_to_delete = ask_which_versions_to_delete
+          delete_versions_without_confirmation(versions_to_delete)
+        end
+      end
+
+      def available_versions
+        @available_versions ||= rest.get_rest("cookbooks/#{@cookbook_name}").values.flatten
+      rescue Net::HTTPServerException => e
+        if e.to_s =~ /^404/
+          Chef::Log.error("Cannot find a cookbook named #{@cookbook_name} to delete")
+          nil
+        else
+          raise
+        end
+      end
+
+      def ask_which_versions_to_delete
+        question = "Which version(s) do you want to delete?\n"
+        valid_responses = {}
+        available_versions.each_with_index do |version, index|
+          valid_responses[(index + 1).to_s] = version
+          question << "#{index + 1}. #{@cookbook_name} #{version}\n"
+        end
+        valid_responses[(available_versions.size + 1).to_s] = :all
+        question << "#{available_versions.size + 1}. All versions\n\n"
+        responses = ask_question(question).split(',').map { |response| response.strip }
+        
+        if responses.empty?
+          Chef::Log.error("No versions specified, exiting")
+          exit(1)
+        end
+        versions = responses.map do |response|
+          if version = valid_responses[response]
+            version
+          else
+            Chef::Log.error("#{response} is not a valid choice, skipping it")
+          end
+        end
+        versions.compact
+      end
+
+      def delete_version_without_confirmation(version)
+        object = delete_request("cookbooks/#{@cookbook_name}/#{version}")
+        output(format_for_display(object)) if config[:print_after]
+        Chef::Log.info("Deleted cookbook[#{@cookbook_name}][#{version}]")
+      end
+
+      def delete_versions_without_confirmation(versions)
+        versions.each do |version|
+          if version == :all
+            delete_all_without_confirmation
+            break
+          else
+            delete_version_without_confirmation(version)
+          end
+        end
+      end
+      
+      private
+      
+      def delete_request(path)
+        path += "?purge=true" if config[:purge]
+        rest.delete_rest(path)
+      end
+
+    end
+  end
+end