knife-spork-berks 1.3.2

data/lib/chef/knife/spork-upload.rb

@@ -0,0 +1,129 @@
+require 'chef/knife'
+require 'chef/exceptions'
+require 'chef/cookbook_loader'
+require 'chef/cookbook_uploader'
+require 'knife-spork/runner'
+require 'socket'
+
+module KnifeSpork
+  class SporkUpload < Chef::Knife
+    include KnifeSpork::Runner
+
+    CHECKSUM = 'checksum'
+    MATCH_CHECKSUM = /[0-9a-f]{32,}/
+
+    banner 'knife spork 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 :freeze,
+      :long => '--freeze',
+      :description => 'Freeze this version of the cookbook so that it cannot be overwritten',
+      :boolean => true
+
+    option :depends,
+      :short => '-D',
+      :long => '--include-dependencies',
+      :description => 'Also upload cookbook dependencies'
+
+    if defined?(::Berkshelf)
+      option :berksfile,
+        :short => '-b',
+        :long => 'berksfile',
+        :description => 'Path to a Berksfile to operate off of',
+        :default => File.join(Dir.pwd, ::Berkshelf::DEFAULT_FILENAME)
+    end
+
+    def run
+      self.config = Chef::Config.merge!(config)
+      config[:cookbook_path] ||= Chef::Config[:cookbook_path]
+
+      if @name_args.empty?
+        show_usage
+        ui.error("You must specify the --all flag or at least one cookbook name")
+        exit 1
+      end
+
+      #First load so plugins etc know what to work with
+      @cookbooks = load_cookbooks(name_args)
+      include_dependencies if config[:depends]
+
+      run_plugins(:before_upload)
+
+      #Reload cookbook in case a VCS plugin found updates
+      @cookbooks = load_cookbooks(name_args)
+      include_dependencies if config[:depends]
+
+      upload
+      run_plugins(:after_upload)
+    end
+
+    private
+    def include_dependencies
+      @cookbooks.each do |cookbook|
+        @cookbooks.concat(load_cookbooks(cookbook.metadata.dependencies.keys))
+      end
+
+      @cookbooks.uniq!
+    end
+
+    def upload
+      # upload cookbooks in reverse so that dependencies are satisfied first
+      @cookbooks.reverse.each do |cookbook|
+        begin
+          check_dependencies(cookbook)
+          if name_args.include?(cookbook.name.to_s)
+            uploader = Chef::CookbookUploader.new(cookbook, ::Chef::Config.cookbook_path)
+            if uploader.respond_to?(:upload_cookbooks)
+              # Chef >= 10.14.0
+              uploader.upload_cookbooks
+              ui.info "Freezing #{cookbook.name} at #{cookbook.version}..."
+              cookbook.freeze_version
+              uploader.upload_cookbooks
+            else
+              uploader.upload_cookbook
+              ui.info "Freezing #{cookbook.name} at #{cookbook.version}..."
+              cookbook.freeze_version
+              uploader.upload_cookbook
+
+            end
+          end
+        rescue Net::HTTPServerException => e
+          if e.response.code == '409'
+            ui.error "#{cookbook.name}@#{cookbook.version} is frozen. Please bump your version number before continuing!"
+            exit(1)
+          else
+            raise
+          end
+        end
+      end
+
+      ui.msg "Successfully uploaded #{@cookbooks.collect{|c| "#{c.name}@#{c.version}"}.join(', ')}!"
+    end
+
+    # Ensures that all the cookbooks dependencies are either already on the server or being uploaded in this pass
+    def check_dependencies(cookbook)
+      cookbook.metadata.dependencies.each do |cookbook_name, version|
+        unless server_side_cookbooks(cookbook_name, version)
+          ui.error "#{cookbook.name} depends on #{cookbook_name} (#{version}), which is not currently being uploaded and cannot be found on the server!"
+          exit(1)
+        end
+      end
+    end
+
+    def server_side_cookbooks(cookbook_name, version)
+      if Chef::CookbookVersion.respond_to?(:list_all_versions)
+        @server_side_cookbooks ||= Chef::CookbookVersion.list_all_versions
+      else
+        @server_side_cookbooks ||= Chef::CookbookVersion.list
+      end
+
+      hash = @server_side_cookbooks[cookbook_name]
+      hash && hash['versions'] && hash['versions'].any?{ |v| Chef::VersionConstraint.new(version).include?(v['version']) }
+    end
+  end
+end

data/lib/knife-spork.rb

@@ -0,0 +1,3 @@
+module KnifeSpork
+  require 'knife-spork/plugins'
+end

data/lib/knife-spork/plugins.rb

@@ -0,0 +1,25 @@
+module KnifeSpork
+  module Plugins
+    # Load each of the drop-in plugins
+    Dir[File.expand_path('../plugins/**/*.rb', __FILE__)].each { |f| require f }
+
+    def self.run(options = {})
+      hook = options[:hook].to_sym
+
+      #Load each of the drop-in plugins specified in the custom plugin path
+      Dir[File.expand_path("#{options[:config][:custom_plugin_path]}/*.rb")].each { |f| require f }
+
+      klasses.each do |klass|
+        plugin = klass.new(options)
+        plugin.send(hook) if plugin.respond_to?(hook) && plugin.enabled?
+      end
+    end
+
+    # Get and return a list of all subclasses (plugins) that are not the base plugin
+    def self.klasses
+      @@klasses ||= self.constants.collect do |c|
+        self.const_get(c) if self.const_get(c).is_a?(Class) && self.const_get(c) != KnifeSpork::Plugins::Plugin
+      end.compact
+    end
+  end
+end

data/lib/knife-spork/plugins/campfire.rb

@@ -0,0 +1,211 @@
+require 'knife-spork/plugins/plugin'
+
+module KnifeSpork
+  module Plugins
+    class Campfire < Plugin
+      name :campfire
+
+      def perform; end
+
+      def after_upload
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} froze the following cookbooks on Chef Server:
+#{cookbooks.collect{|c| "  #{c.name}@#{c.version}"}.join("\n")}
+EOH
+        end
+      end
+
+      def after_promote_remote
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} promoted cookbooks on Chef Server:
+
+cookbooks:
+#{cookbooks.collect{|c| "  #{c.name}@#{c.version}"}.join("\n")}
+
+environments:
+#{environments.collect{|e| "  #{e.name}"}.join("\n")}
+EOH
+        end
+      end
+
+      def after_environmentfromfile
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} uploaded environment #{object_name}
+          EOH
+        end
+      end
+
+      def after_environmentedit
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} edited environment #{object_name}
+          EOH
+        end
+      end
+
+      def after_environmentcreate
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} created environment #{object_name}
+          EOH
+        end
+      end
+
+      def after_environmentdelete
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} deleted environment #{object_name}
+          EOH
+        end
+      end
+
+      def after_rolefromfile
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} uploaded role #{object_name}
+          EOH
+        end
+      end
+
+      def after_roleedit
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} edited role #{object_name}
+          EOH
+        end
+      end
+
+      def after_rolecreate
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} created role #{object_name}
+          EOH
+        end
+      end
+
+      def after_roledelete
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} deleted role #{object_name}
+          EOH
+        end
+      end
+
+      def after_databagedit
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} edited data bag item #{object_name}:#{object_secondary_name}
+          EOH
+        end
+      end
+
+      def after_databagdelete
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} deleted data bag #{object_name}}
+          EOH
+        end
+      end
+
+      def after_databagitemdelete
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} deleted data bag item #{object_name}:#{object_secondary_name}
+          EOH
+        end
+      end
+
+      def after_databagcreate
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} created data bag #{object_name}
+          EOH
+        end
+      end
+
+      def after_databagfromfile
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} uploaded data bag item #{object_name}:#{object_secondary_name}
+          EOH
+        end
+      end
+
+      def after_nodeedit
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} edited node #{object_name}
+          EOH
+        end
+      end
+
+      def after_nodedelete
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} deleted node #{object_name}
+          EOH
+        end
+      end
+
+      def after_nodecreate
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} created node #{object_name}
+          EOH
+        end
+      end
+
+      def after_nodefromfile
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} uploaded node #{object_name}
+          EOH
+        end
+      end
+
+      def after_noderunlistadd
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} added run_list items to #{object_name}: #{object_secondary_name}
+          EOH
+        end
+      end
+
+      def after_noderunlistremove
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} removed run_list items from #{object_name}: #{object_secondary_name}
+          EOH
+        end
+      end
+
+      def after_noderunlistset
+        campfire do |rooms|
+          rooms.paste <<-EOH
+#{organization}#{current_user} set the run_list for #{object_name} to #{object_secondary_name}
+          EOH
+        end
+      end
+
+      private
+      def campfire(&block)
+        safe_require 'campy'
+
+        rooms = [config.rooms || config.room].flatten.compact
+        campfire = Campy::Room.new(:account => config.account, :token => config.token)
+
+        rooms.each do |room_name|
+          room = Campy::Room.new(
+            :account => config.account,
+            :token => config.token,
+            :room => room_name
+          )
+          yield(room) unless room.nil?
+        end
+      end
+    end
+  end
+end

data/lib/knife-spork/plugins/eventinator.rb

@@ -0,0 +1,317 @@
+require 'knife-spork/plugins/plugin'
+require 'json'
+require 'uri'
+
+module KnifeSpork
+  module Plugins
+    class Eventinator < Plugin
+      name :eventinator
+
+      def perform; end
+
+      def after_upload
+        cookbooks.each do |cookbook|
+          event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} uploaded and froze #{cookbook.name}@#{cookbook.version}",
+            :metadata => {
+              :cookbook_name => cookbook.name,
+              :cookbook_version => cookbook.version
+            }.to_json
+          }
+          eventinate(event_data)
+        end
+      end
+
+      def after_promote_remote
+        environments.each do |environment|
+          cookbooks.each do |cookbook|
+            event_data = {
+              :tag => 'knife',
+              :username => current_user,
+              :status => "#{organization}#{current_user} promoted #{cookbook.name}(#{cookbook.version}) to #{environment.name}",
+              :metadata => {
+                :cookbook_name => cookbook.name,
+                :cookbook_version => cookbook.version
+              }.to_json
+            }
+            eventinate(event_data)
+          end
+        end
+      end
+
+      def after_environmentfromfile
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} uploaded environment #{object_name}",
+            :metadata => {
+                :environment_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_environmentedit
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} edited environment #{object_name}",
+            :metadata => {
+                :environment_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_environmentcreate
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} created environment #{object_name}",
+            :metadata => {
+                :environment_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_environmentdelete
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} deleted environment #{object_name}",
+            :metadata => {
+                :environment_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_rolefromfile
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} uploaded role #{object_name}",
+            :metadata => {
+                :role_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_roleedit
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} edited role #{object_name}",
+            :metadata => {
+                :role_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_rolecreate
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} created role #{object_name}",
+            :metadata => {
+                :role_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_roledelete
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} deleted role #{object_name}",
+            :metadata => {
+                :role_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_databagedit
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} edited data bag item #{object_name}:#{object_secondary_name}",
+            :metadata => {
+                :databag_name => object_name,
+                :databag_item => object_secondary_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_databagcreate
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} created data bag #{object_name}",
+            :metadata => {
+                :databag_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_databagdelete
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} deleted data bag item #{object_name}",
+            :metadata => {
+                :databag_name => object_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_databagitemdelete
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} deleted data bag item #{object_name}:#{object_secondary_name}",
+            :metadata => {
+                :databag_name => object_name,
+                :databag_item => object_secondary_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_databagfromfile
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} uploaded data bag item #{object_name}:#{object_secondary_name}",
+            :metadata => {
+                :databag_name => object_name,
+                :databag_item => object_secondary_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_nodeedit
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} edited node #{object_name}",
+            :metadata => {
+                :node_name => object_name
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_nodedelete
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} deleted node #{object_name}",
+            :metadata => {
+                :node_name => object_name
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_nodecreate
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} created node #{object_name}",
+            :metadata => {
+                :node_name => object_name
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_nodefromfile
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} uploaded node #{object_name}",
+            :metadata => {
+                :node_name => object_name
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_noderunlistadd
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} added run_list items to #{object_name}: #{object_secondary_name}",
+            :metadata => {
+                :node_name => object_name,
+                :run_list_items => object_secondary_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_noderunlistremove
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} removed run_list items from #{object_name}: #{object_secondary_name}",
+            :metadata => {
+                :node_name => object_name,
+                :run_list_items => object_secondary_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def after_noderunlistset
+        event_data = {
+            :tag => 'knife',
+            :username => current_user,
+            :status => "#{organization}#{current_user} set the run_list for #{object_name} to #{object_secondary_name}",
+            :metadata => {
+                :node_name => object_name,
+                :run_list_items => object_secondary_name,
+            }.to_json
+        }
+        eventinate(event_data)
+      end
+
+      def eventinate(event_data)
+        begin
+            uri = URI.parse(config.url)
+          rescue Exception => e
+            ui.error 'Could not parse URI for Eventinator.'
+            ui.error e.to_s
+            return
+          end
+
+          http = Net::HTTP.new(uri.host, uri.port)
+          http.read_timeout = config.read_timeout || 5
+
+          request = Net::HTTP::Post.new(uri.request_uri)
+          request.set_form_data(event_data)
+
+          begin
+            response = http.request(request)
+            ui.error "Eventinator at #{config.url} did not receive a good response from the server" if response.code != '200'
+          rescue Timeout::Error
+            ui.error "Eventinator timed out connecting to #{config.url}. Is that URL accessible?"
+          rescue Exception => e
+            ui.error 'Eventinator error.'
+            ui.error e.to_s
+          end
+      end
+    end
+  end
+end