mbailey-chef 0.9.12.1

data/lib/chef/knife/cookbook_test.rb

@@ -0,0 +1,82 @@
+#
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Matthew Kent (<mkent@magoazul.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# Copyright:: Copyright (c) 2010 Matthew Kent
+# 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/checksum_cache'
+
+class Chef
+  class Knife
+    class CookbookTest < Knife
+
+      banner "knife cookbook test [COOKBOOKS...] (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 => "Test 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 |key, cookbook|
+            test_cookbook(key)
+          end
+        else
+          @name_args.each do |cb|
+            test_cookbook(cb)
+          end
+        end
+      end
+
+      def test_cookbook(cookbook)
+        Chef::Log.info("Running syntax check on #{cookbook}")
+        Array(config[:cookbook_path]).reverse.each do |path|
+          syntax_checker = Chef::Cookbook::SyntaxCheck.for_cookbook(cookbook, path)
+          test_ruby(syntax_checker)
+          test_templates(syntax_checker)
+        end
+      end
+
+
+      def test_ruby(syntax_checker)
+        Chef::Log.info("Validating ruby files")
+        exit(1) unless syntax_checker.validate_ruby_files
+      end
+      
+      def test_templates(syntax_checker)
+        Chef::Log.info("Validating templates")
+        exit(1) unless syntax_checker.validate_templates
+      end
+
+    end
+  end
+end

data/lib/chef/knife/cookbook_upload.rb

@@ -0,0 +1,95 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Christopher Walters (<cw@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/cookbook_loader'
+require 'chef/cookbook_uploader'
+
+class Chef
+  class Knife
+    class CookbookUpload < Knife
+      include Chef::Mixin::ShellOut
+
+      banner "knife cookbook upload [COOKBOOKS...] (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
+
+        Chef::Cookbook::FileVendor.on_create { |manifest| Chef::Cookbook::FileSystemFileVendor.new(manifest) }
+
+        cl = Chef::CookbookLoader.new
+
+        humanize_auth_exceptions do
+          if config[:all]
+            cl.each do |cookbook_name, cookbook|
+              Chef::Log.info("** #{cookbook.name.to_s} **")
+              Chef::CookbookUploader.upload_cookbook(cookbook)
+            end
+          else
+            if @name_args.length < 1
+              show_usage
+              Chef::Log.fatal("You must specify the --all flag or at least one cookbook name")
+              exit 1
+            end
+            @name_args.each do |cookbook_name|
+              if cl.cookbook_exists?(cookbook_name)
+                Chef::CookbookUploader.upload_cookbook(cl[cookbook_name])
+              else
+                Chef::Log.error("Could not find cookbook #{cookbook_name} in your cookbook path, skipping it")
+              end
+            end
+          end
+        end
+      end
+
+      private
+
+      def humanize_auth_exceptions
+        begin
+          yield
+        rescue Net::HTTPServerException => e
+          case e.response.code
+          when "401"
+            Chef::Log.fatal "Request failed due to authentication (#{e}), check your client configuration (username, key)"
+            exit 18
+          else
+            raise
+          end
+        end
+      end
+
+
+    end
+  end
+end

data/lib/chef/knife/data_bag_create.rb

@@ -0,0 +1,59 @@
+#
+# 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 "knife data bag create BAG [ITEM] (options)"
+      category "data bag"
+
+      def run
+        @data_bag_name, @data_bag_item_name = @name_args
+
+        if @data_bag_name.nil?
+          show_usage
+          Chef::Log.fatal("You must specify a data bag name")
+          exit 1
+        end
+        
+        # create the data bag
+        begin
+          rest.post_rest("data", { "name" => @data_bag_name })
+          Chef::Log.info("Created data_bag[#{@data_bag_name}]")
+        rescue Net::HTTPServerException => e
+          raise unless e.to_s =~ /^409/
+          Chef::Log.info("Data bag #{@data_bag_name} already exists")
+        end
+        
+        # if an item is specified, create it, as well
+        if @data_bag_item_name
+          create_object({ "id" => @data_bag_item_name }, "data_bag_item[#{@data_bag_item_name}]") do |output|
+            rest.post_rest("data/#{@data_bag_name}", output)
+          end
+        end
+      end
+    end
+  end
+end
+
+
+

data/lib/chef/knife/data_bag_delete.rb

@@ -0,0 +1,48 @@
+#
+# 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 DataBagDelete < Knife
+
+      banner "knife data bag delete BAG [ITEM] (options)"
+      category "data bag"
+
+      def run 
+        if @name_args.length == 2
+          delete_object(Chef::DataBagItem, @name_args[1], "data_bag_item") do
+            rest.delete_rest("data/#{@name_args[0]}/#{@name_args[1]}")
+          end
+        elsif @name_args.length == 1
+          delete_object(Chef::DataBag, @name_args[0], "data_bag") do
+            rest.delete_rest("data/#{@name_args[0]}")
+          end
+        else
+          show_usage
+          Chef::Log.fatal("You must specify at least a data bag name")
+          exit 1
+        end
+      end
+    end
+  end
+end
+
+

data/lib/chef/knife/data_bag_edit.rb

@@ -0,0 +1,50 @@
+#
+# 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_item'
+
+class Chef
+  class Knife
+    class DataBagEdit < Knife
+
+      banner "knife data bag edit BAG ITEM (options)"
+      category "data bag"
+
+      def run 
+        if @name_args.length != 2
+          Chef::Log.fatal("You must supply the data bag and an item to edit!")
+          exit 42
+        else
+          object = Chef::DataBagItem.load(@name_args[0], @name_args[1])
+
+          output = edit_data(object)
+
+          rest.put_rest("data/#{@name_args[0]}/#{@name_args[1]}", output)
+
+          Chef::Log.info("Saved data_bag_item[#{@name_args[1]}]")
+
+          output(format_for_display(object)) if config[:print_after]
+        end
+      end
+    end
+  end
+end
+
+
+

data/lib/chef/knife/data_bag_from_file.rb

@@ -0,0 +1,44 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 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/data_bag'
+require 'chef/data_bag_item'
+
+class Chef
+  class Knife
+    class DataBagFromFile < Knife
+
+      banner "knife data bag from file BAG FILE (options)"
+      category "data bag"
+
+      def run 
+        updated = load_from_file(Chef::DataBagItem, @name_args[1], @name_args[0])
+        dbag = Chef::DataBagItem.new
+        dbag.data_bag(@name_args[0])
+        dbag.raw_data = updated
+        dbag.save
+        Chef::Log.info("Updated data_bag_item[#{@name_args[1]}]")
+      end
+    end
+  end
+end
+
+
+
+

data/lib/chef/knife/data_bag_list.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 DataBagList < Knife
+
+      banner "knife data bag list (options)"
+      category "data bag"
+
+      option :with_uri,
+        :short => "-w",
+        :long => "--with-uri",
+        :description => "Show corresponding URIs"
+
+      def run
+        output(format_list_for_display(Chef::DataBag.list))
+      end
+    end
+  end
+end
+
+
+
+

data/lib/chef/knife/data_bag_show.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 'chef/data_bag'
+
+class Chef
+  class Knife
+    class DataBagShow < Knife
+
+      banner "knife data bag show BAG [ITEM] (options)"
+      category "data bag"
+
+      def run
+        display = case @name_args.length
+                  when 2
+                    format_for_display(Chef::DataBagItem.load(@name_args[0], @name_args[1]))
+                  else
+                    format_list_for_display(Chef::DataBag.load(@name_args[0]))
+                  end
+        output(display)
+      end
+    end
+  end
+end
+