knife-spork-berks 1.3.2

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

@@ -0,0 +1,46 @@
+require 'knife-spork/plugins/plugin'
+
+module KnifeSpork
+  module Plugins
+    class Foodcritic < Plugin
+      name :foodcritic
+      hooks :after_check, :before_upload
+
+      def perform
+        safe_require 'foodcritic'
+
+        if Gem::Specification.find_all_by_name("foodcritic", "~> 3.0.0").empty?
+          ui.fatal "The knife-spork foodcritic plugin requires foodcritic >= 3.0.0. Please install a more recent foodcritic version."
+          exit 1
+        end
+
+        tags = config.tags || []
+        fail_tags = config.fail_tags || ['any']
+        include_rules = config.include_rules || []
+
+        cookbooks.each do |cookbook|
+          ui.info "Running foodcritic against #{cookbook.name}@#{cookbook.version}..."
+
+          cookbook_path = cookbook.root_dir
+
+          ui.info cookbook_path
+
+          options = {:tags => tags, :fail_tags => fail_tags, :include_rules => include_rules, :cookbook_paths => [cookbook_path]}
+          review = ::FoodCritic::Linter.new.check(options)
+
+          if review.failed?
+            ui.error "Foodcritic failed!"
+            review.to_s.split("\n").each{ |r| ui.error r.to_s }
+            exit(1) if config.epic_fail
+          else
+            ui.info "Passed!"
+          end
+        end
+      end
+
+      def epic_fail?
+        config.epic_fail.nil? ? 'true' : config.epic_fail
+      end
+    end
+  end
+end

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

@@ -0,0 +1,197 @@
+require 'knife-spork/plugins/plugin'
+
+module KnifeSpork
+  module Plugins
+    class Git < Plugin
+      name :git
+
+      def perform; end
+
+      def before_bump
+        git_pull(environment_path) unless cookbook_path.include?(environment_path.gsub"/environments","")
+        git_pull_submodules(environment_path) unless cookbook_path.include?(environment_path.gsub"/environments","")
+        cookbooks.each do |cookbook|
+          git_pull(cookbook_path_for(cookbook))
+          git_pull_submodules(cookbook_path_for(cookbook))
+        end
+      end
+
+      def before_upload
+        git_pull(environment_path) unless cookbook_path.include?(environment_path.gsub"/environments","")
+        git_pull_submodules(environment_path) unless cookbook_path.include?(environment_path.gsub"/environments","")
+        cookbooks.each do |cookbook|
+          git_pull(cookbook_path_for(cookbook))
+          git_pull_submodules(cookbook_path_for(cookbook))
+        end
+      end
+
+      def before_promote
+        cookbooks.each do |cookbook|
+          git_pull(environment_path) unless cookbook_path_for(cookbook).include?(environment_path.gsub"/environments","")
+          git_pull_submodules(environment_path) unless cookbook_path_for(cookbook).include?(environment_path.gsub"/environments","")
+          git_pull(cookbook_path_for(cookbook))
+          git_pull_submodules(cookbook_path_for(cookbook))
+        end
+      end
+
+      def after_bump
+        cookbooks.each do |cookbook|
+          git_add(cookbook_path_for(cookbook),"metadata.rb")
+        end
+      end
+
+      def after_promote_local
+        environments.each do |environment|
+          git_add(environment_path,"#{environment}.json")
+        end
+      end
+
+      private
+      def git
+        safe_require 'git'
+        log = Logger.new(STDOUT)
+        log.level = Logger::WARN
+        @git ||= begin
+          ::Git.open('.', :log => log)
+        rescue
+          ui.error 'You are not currently in a git repository. Please ensure you are in a git repo, a repo subdirectory, or remove the git plugin from your KnifeSpork configuration!'
+          exit(0)
+        end
+      end
+
+      # In this case, a git pull will:
+      #   - Stash local changes
+      #   - Pull from the remote
+      #   - Pop the stash
+      def git_pull(path)
+        if is_repo?(path)
+          ui.msg "Git: Pulling latest changes from #{path}"
+          output = IO.popen("git pull 2>&1")
+          Process.wait
+          exit_code = $?
+          if !exit_code.exitstatus ==  0
+            ui.error "#{output.read()}\n"
+            exit 1
+          end
+        end
+      end
+
+      def git_pull_submodules(path)
+        if is_repo?(path)
+          ui.msg "Pulling latest changes from git submodules (if any)"
+          top_level = `cd #{path} && git rev-parse --show-toplevel 2>&1`.chomp
+          if is_submodule?(top_level)
+            top_level = get_parent_dir(top_level)
+          end
+          output = IO.popen("cd #{top_level} && git submodule foreach git pull 2>&1")
+          Process.wait
+          exit_code = $?
+          if !exit_code.exitstatus ==  0
+              ui.error "#{output.read()}\n"
+              exit 1
+          end
+        end
+      end
+
+      def git_add(filepath,filename)
+        if is_repo?(filepath)
+          ui.msg "Git add'ing #{filepath}/#{filename}"
+          output = IO.popen("cd #{filepath} && git add #{filename}")
+          Process.wait
+          exit_code = $?
+          if !exit_code.exitstatus ==  0
+              ui.error "#{output.read()}\n"
+              exit 1
+          end
+        end
+      end
+
+      # Commit changes, if any
+      def git_commit
+        begin
+          git.add('.')
+          `git ls-files --deleted`.chomp.split("\n").each{ |f| git.remove(f) }
+          git.commit_all "[KnifeSpork] Bumping cookbooks:\n#{cookbooks.collect{|c| "  #{c.name}@#{c.version}"}.join("\n")}"
+        rescue ::Git::GitExecuteError; end
+      end
+
+      def git_push(tags = false)
+        begin
+          git.push remote, branch, tags
+        rescue ::Git::GitExecuteError => e
+          ui.error "Could not push to remote #{remote}/#{branch}. Does it exist?"
+        end
+      end
+
+      def git_tag(tag)
+        begin
+          git.add_tag(tag)
+        rescue ::Git::GitExecuteError => e
+          ui.error "Could not tag #{tag_name}. Does it already exist?"
+          ui.error 'You may need to delete the tag before running promote again.'
+        end
+      end
+
+      def is_repo?(path)
+        output = IO.popen("cd #{path} && git rev-parse --git-dir 2>&1")
+        Process.wait
+        if $? != 0
+            ui.warn "#{path} is not a git repo, skipping..."
+            return false
+        else
+            return true
+        end
+      end
+
+      def is_submodule?(path)
+        top_level = `cd #{path} && git rev-parse --show-toplevel 2>&1`.chomp
+        output = IO.popen("cd #{top_level}/.. && git rev-parse --show-toplevel 2>&1")
+        Process.wait
+        if $? != 0
+          return false
+        else
+          return true
+        end
+      end
+
+      def get_parent_dir(path)
+        top_level = path
+        return_code = 0
+        while return_code == 0
+          output = IO.popen("cd #{top_level}/.. && git rev-parse --show-toplevel 2>&1")
+          Process.wait
+          return_code = $?
+          cmd_output = output.read.chomp
+          #cygwin, I hate you for making me do this
+          if cmd_output.include?("fatal: Not a git repository")
+            return_code = 1
+          end
+          if return_code == 0
+            top_level = cmd_output
+          end
+        end
+        top_level
+      end
+
+      def remote
+        config.remote || 'origin'
+      end
+
+      def branch
+        config.branch || 'master'
+      end
+
+      def tag_name
+        cookbooks.collect{|c| "#{c.name}@#{c.version}"}.join('-')
+      end
+
+      def cookbook_path_for cookbook
+        if defined?(Berkshelf) and cookbook.is_a? Berkshelf::CachedCookbook
+          cookbook.path.to_s
+        else
+          cookbook.root_dir
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,25 @@
+require 'knife-spork/plugins/plugin'
+
+module KnifeSpork
+  module Plugins
+    class Graphite < Plugin
+      name :graphite
+      hooks :after_promote_remote
+
+      def perform
+        environments.each do |environment|
+          begin
+            message = "deploys.chef.#{environment} 1 #{Time.now.to_i}\n"
+            socket = TCPSocket.open(config.server, config.port)
+            socket.write(message)
+          rescue Exception => e
+            ui.error 'Graphite was unable to process the request.'
+            ui.error e.to_s
+          ensure
+            socket.close unless socket.nil?
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,161 @@
+require 'knife-spork/plugins/plugin'
+
+module KnifeSpork
+  module Plugins
+    class Grove < Plugin
+      name :grove
+
+      def perform; end
+
+      def after_upload
+        grove <<-EOH
+#{current_user} froze the following cookbooks on Chef Server: #{cookbooks.collect{|c| "#{c.name}@#{c.version}"}.join(', ')}
+        EOH
+      end
+
+      def after_promote_remote
+        grove <<-EOH
+#{current_user} promoted #{cookbooks.collect{|c| "#{c.name}@#{c.version}"}.join(', ')} on #{environments.collect{|e| "#{e.name}"}.join(', ')}
+        EOH
+      end
+
+      def after_environmentfromfile
+        grove <<-EOH
+#{current_user} uploaded environment #{object_name}
+        EOH
+      end
+
+      def after_environmentedit
+        grove <<-EOH
+#{current_user} edited environment #{object_name}
+        EOH
+      end
+
+      def after_environmentcreate
+        grove <<-EOH
+#{current_user} created environment #{object_name}
+        EOH
+      end
+
+      def after_environmentdelete
+        grove <<-EOH
+#{current_user} deleted environment #{object_name}
+        EOH
+      end
+
+      def after_rolefromfile
+        grove <<-EOH
+#{current_user} uploaded role #{object_name}
+        EOH
+      end
+
+      def after_roleedit
+        grove <<-EOH
+#{current_user} edited role #{object_name}
+        EOH
+      end
+
+      def after_rolecreate
+        grove <<-EOH
+#{current_user} created role #{object_name}
+        EOH
+      end
+
+      def after_roledelete
+        grove <<-EOH
+#{current_user} deleted role #{object_name}
+        EOH
+      end
+
+      def after_databagcreate
+        grove <<-EOH
+#{current_user} created data bag #{object_name}
+        EOH
+      end
+
+      def after_databagedit
+        grove <<-EOH
+#{current_user} edited data bag item #{object_name}:#{object_secondary_name}
+        EOH
+      end
+
+      def after_databagitemdelete
+        grove <<-EOH
+#{current_user} edited data bag item #{object_name}:#{object_secondary_name}
+        EOH
+      end
+
+      def after_databagdelete
+        grove <<-EOH
+#{current_user} edited data bag #{object_name}
+        EOH
+      end
+
+      def after_databagfromfile
+        grove <<-EOH
+#{current_user} uploaded data bag item #{object_name}:#{object_secondary_name}
+        EOH
+      end
+
+      def after_nodeedit
+        grove <<-EOH
+#{current_user} edited node #{object_name}
+        EOH
+      end
+
+      def after_nodedelete
+        grove <<-EOH
+#{current_user} deleted node #{object_name}
+        EOH
+      end
+
+    def after_nodecreate
+        grove <<-EOH
+#{current_user} created node #{object_name}
+        EOH
+      end
+
+      def after_nodefromfile
+        grove <<-EOH
+#{current_user} uploaded node #{object_name}
+        EOH
+      end
+
+      def after_noderunlistadd
+        grove <<-EOH
+#{current_user} added run_list items to #{object_name}: #{object_secondary_name}
+        EOH
+      end
+
+      def after_noderunlistremove
+        grove <<-EOH
+#{current_user} removed run_list items from #{object_name}: #{object_secondary_name}
+        EOH
+      end
+
+      def after_noderunlistset
+        grove <<-EOH
+#{current_user} set the run_list for #{object_name} to #{object_secondary_name}
+        EOH
+      end
+
+      private
+      def grove(message)
+        safe_require 'rest_client'
+
+        config.tokens.each do |token|
+          # Grove can't handle multi-line messages, so let's split by line
+          message.split("\n").flatten.delete_if(&:empty?).each do |line|
+            RestClient.post "https://grove.io/api/notice/#{token}/",
+                            :message => line,
+                            :service => 'knife-spork'
+          end
+        end
+      end
+
+      def tokens
+        Array(config.token || config.tokens)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,131 @@
+require 'knife-spork/plugins/plugin'
+
+module KnifeSpork
+  module Plugins
+    class HipChat < Plugin
+      name :hipchat
+
+      def perform; end
+
+      def after_upload
+        hipchat "#{organization}#{current_user} uploaded the following cookbooks:\n#{cookbooks.collect{ |c| "  #{c.name}@#{c.version}" }.join("\n")}"
+      end
+
+      def after_promote_remote
+        hipchat "#{organization}#{current_user} promoted the following cookbooks:\n#{cookbooks.collect{ |c| "  #{c.name}@#{c.version}" }.join("\n")} to #{environments.collect{ |e| "#{e.name}" }.join(", ")}"
+      end
+
+      def after_environmentfromfile
+        hipchat "#{organization}#{current_user} uploaded environment #{object_name}"
+      end
+
+      def after_environmentedit
+        hipchat "#{organization}#{current_user} edited environment #{object_name}"
+      end
+
+      def after_environmentcreate
+        hipchat "#{organization}#{current_user} created environment #{object_name}"
+      end
+
+      def after_environmentdelete
+        hipchat "#{organization}#{current_user} deleted environment #{object_name}"
+      end
+
+      def after_rolefromfile
+        hipchat "#{organization}#{current_user} uploaded role #{object_name}"
+      end
+
+      def after_roleedit
+        hipchat "#{organization}#{current_user} edited role #{object_name}"
+      end
+
+      def after_rolecreate
+        hipchat "#{organization}#{current_user} created role #{object_name}"
+      end
+
+      def after_roledelete
+        hipchat "#{organization}#{current_user} deleted role #{object_name}"
+      end
+
+      def after_databagedit
+        hipchat "#{organization}#{current_user} edited data bag item #{object_name}:#{object_secondary_name}"
+      end
+
+      def after_databagcreate
+        hipchat "#{organization}#{current_user} created data bag #{object_name}"
+      end
+
+      def after_databagdelete
+        hipchat "#{organization}#{current_user} deleted data bag item #{object_name}"
+      end
+
+      def after_databagitemdelete
+        hipchat "#{organization}#{current_user} deleted data bag item #{object_name}:#{object_secondary_name}"
+      end
+
+      def after_databagfromfile
+        hipchat "#{organization}#{current_user} uploaded data bag item #{object_name}:#{object_secondary_name}"
+      end
+
+      def after_nodeedit
+        hipchat "#{organization}#{current_user} edited node #{object_name}"
+      end
+
+      def after_nodedelete
+        hipchat "#{organization}#{current_user} deleted node #{object_name}"
+      end
+
+      def after_nodecreate
+        hipchat "#{organization}#{current_user} created node #{object_name}"
+      end
+
+      def after_nodefromfile
+        hipchat "#{organization}#{current_user} uploaded node #{object_name}"
+      end
+
+      def after_noderunlistadd
+        hipchat "#{organization}#{current_user} added run_list items to #{object_name}: #{object_secondary_name}"
+      end
+
+      def after_noderunlistremove
+        hipchat "#{organization}#{current_user} removed run_list items from #{object_name}: #{object_secondary_name}"
+      end
+
+      def after_noderunlistset
+        hipchat "#{organization}#{current_user} set the run_list for #{object_name} to #{object_secondary_name}"
+      end
+
+
+      private
+      def hipchat(message)
+        safe_require 'hipchat'
+
+        rooms.each do |room_name|
+          begin
+            client = ::HipChat::Client.new(config.api_token)
+            client[room_name].send(nickname, message, :notify => notify, :color =>color)
+          rescue Exception => e
+            ui.error 'Something went wrong sending to HipChat.'
+            ui.error e.to_s
+          end
+        end
+      end
+
+      def rooms
+        [ config.room || config.rooms ].flatten
+      end
+
+      def nickname
+        config.nickname || 'KnifeSpork'
+      end
+
+      def notify
+        config.notify.nil? ? true : config.notify
+      end
+
+      def color
+        config.color || 'yellow'
+      end
+    end
+  end
+end