runa-chef 0.8.0.1

data/lib/chef/knife/cookbook_download.rb

@@ -0,0 +1,57 @@
+#
+# 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 CookbookDownload < Knife
+
+      banner "Sub-Command: cookbook download COOKBOOK (options)"
+
+      option :file,
+       :short => "-f FILE",
+       :long => "--file FILE",
+       :description => "The filename to write to"
+
+      def run 
+        cookbook = rest.get_rest("cookbooks/#{@name_args[0]}")
+        version = cookbook["metadata"]["version"] 
+        Chef::Log.info("Downloading #{@name_args[0]} cookbook version #{version}")
+        rest.sign_on_redirect = false
+        tf = rest.get_rest("cookbooks/#{@name_args[0]}/_content", true)
+        rest.sign_on_redirect = true 
+        unless config[:file]
+          if version
+            config[:file] = File.join(Dir.pwd, "#{@name_args[0]}-#{version}.tar.gz")
+          else
+            config[:file] = File.join(Dir.pwd, "#{@name_args[0]}.tar.gz")
+          end
+        end
+        FileUtils.cp(tf.path, config[:file])
+        Chef::Log.info("Cookbook saved: #{config[:file]}")
+      end
+
+    end
+  end
+end
+
+
+
+
+

data/lib/chef/knife/cookbook_list.rb

@@ -0,0 +1,41 @@
+#
+# 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'
+require 'json'
+
+class Chef
+  class Knife
+    class CookbookList < Knife
+
+      banner "Sub-Command: cookbook list (options)"
+
+      option :with_uri,
+        :short => "-w",
+        :long => "--with-uri",
+        :description => "Show corresponding URIs"
+
+      def run 
+        json_pretty_print(format_list_for_display(rest.get_rest('cookbooks')))
+      end
+    end
+  end
+end
+
+
+

data/lib/chef/knife/cookbook_metadata.rb

@@ -0,0 +1,87 @@
+#
+#
+# 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 CookbookMetadata < Knife
+
+      banner "Sub-Command: cookbook metadata COOKBOOK (options)"
+
+      option :cookbook_path,
+        :short => "-o PATH:PATH",
+        :long => "--cookbook-path PATH:PATH",
+        :description => "A colon-separated path to look for cookbooks in",
+        :proc => lambda { |o| o.split(":") }
+
+      option :all,
+        :short => "-a",
+        :long => "--all",
+        :description => "Generate metadata for all cookbooks, rather than just a single cookbook"
+
+      def run 
+        if config[:cookbook_path]
+          Chef::Config[:cookbook_path] = config[:cookbook_path]
+        else
+          config[:cookbook_path] = Chef::Config[:cookbook_path]
+        end
+
+        if config[:all] 
+          cl = Chef::CookbookLoader.new
+          cl.each do |cookbook|
+            generate_metadata(cookbook.name.to_s)
+          end
+        else
+          generate_metadata(@name_args[0]) 
+        end
+      end
+
+      def generate_metadata(cookbook)
+        Chef::Log.info("Generating metadata for #{cookbook}")
+        config[:cookbook_path].reverse.each do |path|
+          file = File.join(path, cookbook, 'metadata.rb')
+          if File.exists?(file)
+            Chef::Log.info("Generating from #{file}")
+            md = Chef::Cookbook::Metadata.new
+            md.name(cookbook)
+            md.from_file(file)
+            json_file = File.join(File.dirname(file), 'metadata.json')
+            File.open(json_file, "w") do |f|
+              f.write(JSON.pretty_generate(md))
+            end
+            generated = true 
+            Chef::Log.info("Generated #{json_file}")
+          else
+            Chef::Log.debug("No #{file} found; skipping!")
+          end
+        end
+      end
+
+    end
+  end
+end
+
+
+
+
+
+
+
+

data/lib/chef/knife/cookbook_show.rb

@@ -0,0 +1,75 @@
+#
+# 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'
+require 'json'
+require 'uri'
+
+class Chef
+  class Knife
+    class CookbookShow < Knife
+
+      banner "Sub-Command: cookbook show COOKBOOK [PART] [FILENAME] (options)"
+
+      option :fqdn,
+       :short => "-f FQDN",
+       :long => "--fqdn FQDN",
+       :description => "The FQDN of the host to see the file for"
+
+      option :platform,
+       :short => "-p PLATFORM",
+       :long => "--platform PLATFORM",
+       :description => "The platform to see the file for"
+
+      option :platform_version,
+       :short => "-V VERSION",
+       :long => "--platform-version VERSION",
+       :description => "The platform version to see the file for"
+
+      def run 
+        case @name_args.length
+        when 3 # We are showing a specific file
+          arguments = { :id => @name_args[2] } 
+          arguments[:fqdn] = config[:fqdn] if config.has_key?(:fqdn)
+          arguments[:platform] = config[:platform] if config.has_key?(:platform)
+          arguments[:version] = config[:platform_version] if config.has_key?(:platform_version)
+          result = rest.get_rest("cookbooks/#{@name_args[0]}/#{@name_args[1]}?#{make_query_params(arguments)}")
+          pretty_print(result)
+        when 2 # We are showing a specific part of the cookbook
+          result = rest.get_rest("cookbooks/#{@name_args[0]}")
+          json_pretty_print(result[@name_args[1]])
+        when 1 # We are showing the whole cookbook data
+          json_pretty_print(rest.get_rest("cookbooks/#{@name_args[0]}"))
+        end
+      end
+
+      def make_query_params(req_opts)
+        query_part = Array.new 
+        req_opts.keys.sort { |a,b| a.to_s <=> b.to_s }.each do |key|
+          query_part << "#{key}=#{URI.escape(req_opts[key])}"
+        end
+        query_part.join("&")
+      end
+
+    end
+  end
+end
+
+
+
+

data/lib/chef/knife/cookbook_upload.rb

@@ -0,0 +1,173 @@
+#
+#
+# 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'
+require 'chef/streaming_cookbook_uploader'
+require 'chef/cache/checksum'
+
+class Chef
+  class Knife
+    class CookbookUpload < Knife
+
+      banner "Sub-Command: cookbook upload COOKBOOK (options)"
+
+      option :cookbook_path,
+        :short => "-o PATH:PATH",
+        :long => "--cookbook-path PATH:PATH",
+        :description => "A colon-separated path to look for cookbooks in",
+        :proc => lambda { |o| o.split(":") }
+
+      option :all,
+        :short => "-a",
+        :long => "--all",
+        :description => "Upload all cookbooks, rather than just a single cookbook"
+
+      def run 
+        if config[:cookbook_path]
+          Chef::Config[:cookbook_path] = config[:cookbook_path]
+        else
+          config[:cookbook_path] = Chef::Config[:cookbook_path]
+        end
+
+        if config[:all] 
+          cl = Chef::CookbookLoader.new
+          cl.each do |cookbook|
+            Chef::Log.info("** #{cookbook.name.to_s} **")
+            upload_cookbook(cookbook.name.to_s)
+          end
+        else
+          upload_cookbook(@name_args[0]) 
+        end
+      end
+
+      def test_ruby(cookbook_dir)
+        Dir[File.join(cookbook_dir, '**', '*.rb')].each do |ruby_file|
+          Chef::Log.info("Testing #{ruby_file} for syntax errors...")
+          Chef::Mixin::Command.run_command(:command => "ruby -c #{ruby_file}")
+        end
+      end
+
+      def test_templates(cookbook_dir)
+        Dir[File.join(cookbook_dir, '**', '*.erb')].each do |erb_file|
+          Chef::Log.info("Testing template #{erb_file} for syntax errors...")
+          Chef::Mixin::Command.run_command(:command => "sh -c 'erubis -x #{erb_file} | ruby -c'")
+        end
+      end
+
+      def upload_cookbook(cookbook_name)
+
+        if cookbook_name =~ /^#{File::SEPARATOR}/
+          child_folders = cookbook_name 
+          cookbook_name = File.basename(cookbook_name)
+        else
+          child_folders = config[:cookbook_path].inject([]) do |r, e| 
+            r << File.join(e, cookbook_name)
+            r
+          end
+        end
+
+        tmp_cookbook_tarball = Tempfile.new("chef-#{cookbook_name}")
+        tmp_cookbook_tarball.close
+        tarball_name = "#{tmp_cookbook_tarball.path}.tar.gz"
+        File.unlink(tmp_cookbook_tarball.path)
+
+        tmp_cookbook_path = Tempfile.new("chef-#{cookbook_name}-build")
+        tmp_cookbook_path.close
+        tmp_cookbook_dir = tmp_cookbook_path.path
+        File.unlink(tmp_cookbook_dir)
+        FileUtils.mkdir_p(tmp_cookbook_dir)
+
+        Chef::Log.debug("Staging at #{tmp_cookbook_dir}")
+
+        found_cookbook = false 
+
+        child_folders.each do |file_path|
+          if File.directory?(file_path)
+            found_cookbook = true 
+            Chef::Log.info("Copying from #{file_path} to #{tmp_cookbook_dir}")
+            FileUtils.cp_r(file_path, tmp_cookbook_dir, :remove_destination => true)
+          else
+            Chef::Log.info("Nothing to copy from #{file_path}")
+          end
+        end 
+
+        unless found_cookbook
+          Chef::Log.fatal("Could not find cookbook #{cookbook_name}!")
+          exit 17
+        end
+
+        test_ruby(tmp_cookbook_dir)
+        test_templates(tmp_cookbook_dir)
+
+        # First, generate metadata
+        kcm = Chef::Knife::CookbookMetadata.new
+        kcm.config[:cookbook_path] = [ tmp_cookbook_dir ]
+        kcm.name_args = [ cookbook_name ]
+        kcm.run
+
+        Chef::Log.info("Creating tarball at #{tarball_name}")
+        Chef::Mixin::Command.run_command(
+          :command => "tar -C #{tmp_cookbook_dir} -cvzf #{tarball_name} ./#{cookbook_name}"
+        )
+
+        begin
+          cb = rest.get_rest("cookbooks/#{cookbook_name}")
+          cookbook_uploaded = true
+        rescue Net::HTTPServerException
+          cookbook_uploaded = false
+        end
+        Chef::Log.info("Uploading #{cookbook_name} (#{cookbook_uploaded ? 'new version' : 'first time'})")
+        if cookbook_uploaded
+          Chef::StreamingCookbookUploader.put(
+            "#{Chef::Config[:chef_server_url]}/cookbooks/#{cookbook_name}/_content", 
+            Chef::Config[:node_name], 
+            Chef::Config[:client_key], 
+            {
+              :file => File.new(tarball_name), 
+              :name => cookbook_name
+            }
+          )
+        else
+          Chef::StreamingCookbookUploader.post(
+            "#{Chef::Config[:chef_server_url]}/cookbooks", 
+            Chef::Config[:node_name], 
+            Chef::Config[:client_key], 
+            {
+              :file => File.new(tarball_name), 
+              :name => cookbook_name
+            }
+          )
+        end
+        Chef::Log.info("Upload complete!")
+        Chef::Log.debug("Removing local tarball at #{tarball_name}")
+        FileUtils.rm_rf tarball_name 
+        Chef::Log.debug("Removing local staging directory at #{tmp_cookbook_dir}")
+        FileUtils.rm_rf tmp_cookbook_dir
+      end
+
+    end
+  end
+end
+
+
+
+
+
+
+

data/lib/chef/knife/data_bag_create.rb

@@ -0,0 +1,43 @@
+#
+# 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'
+require 'chef/data_bag'
+
+class Chef
+  class Knife
+    class DataBagCreate < Knife
+
+      banner "Sub-Command: data bag create BAG [ITEM] (options)"
+
+      def run 
+        if @name_args.length == 2
+          create_object({ "id" => @name_args[1] }, "data_bag_item[#{@name_args[1]}]") do |output|
+            rest.put_rest("data/#{@name_args[0]}/#{@name_args[1]}", output)
+          end
+        else
+          rest.post_rest("data", { "name" => @name_args[0] })
+          Chef::Log.info("Created data_bag[#{@name_args[0]}]")
+        end
+      end
+    end
+  end
+end
+
+
+