knife-spork-berks 1.3.2

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require 'bundler/gem_tasks'

data/knife-spork-berks.gemspec

@@ -0,0 +1,22 @@
+$:.push File.expand_path('../lib', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.name          = 'knife-spork-berks'
+  gem.version       = '1.3.2'
+  gem.authors       = ["Jon Cowie"]
+  gem.email         = 'jonlives@gmail.com'
+  gem.homepage      = 'https://github.com/jonlives/knife-spork'
+  gem.summary       = "A workflow plugin to help many devs work with the same chef repo/server. Includes fixes for using with Berkshelf."
+  gem.description       = "A workflow plugin to help many devs work with the same chef repo/server. Includes fixes for using with Berkshelf."
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "knife-spork-berks"
+  gem.require_paths = ["lib"]
+
+  gem.add_runtime_dependency 'chef', '>= 11.0.0'
+  gem.add_runtime_dependency 'git', '>= 1.2.5'
+  gem.add_runtime_dependency 'app_conf', '>= 0.4.0'
+  gem.add_runtime_dependency 'diffy', '>= 3.0.1'
+end

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

@@ -0,0 +1,86 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+
+module KnifeSpork
+  class SporkBump < Chef::Knife
+    include KnifeSpork::Runner
+
+    TYPE_INDEX = { :major => 0, :minor => 1, :patch => 2, :manual => 3 }.freeze
+
+    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(':') }
+
+    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)
+
+      option :skip_dependencies,
+        :short => '-s',
+        :long => '--skip-dependencies',
+        :description => 'Berksfile skips resolving source cookbook dependencies',
+        :default => true
+    end
+
+    banner 'knife spork bump COOKBOOK [major|minor|patch|manual]'
+
+    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 at least a cookbook name")
+        exit 1
+      end
+
+      #First load so plugins etc know what to work with
+      @cookbook = load_cookbook(name_args.first)
+
+      run_plugins(:before_bump)
+
+      #Reload cookbook in case a VCS plugin found updates
+      @cookbook = load_cookbook(name_args.first)
+      bump
+      run_plugins(:after_bump)
+    end
+
+    private
+    def bump
+      old_version = @cookbook.version
+
+      if bump_type == 3
+        # manual bump
+        version_array = manual_bump_version.split('.')
+      else
+        # major, minor, or patch bump
+        version_array = old_version.split('.').collect{ |i| i.to_i }
+        version_array[bump_type] += 1
+        ((bump_type+1)..2).each{ |i| version_array[i] = 0 } # reset all lower version numbers to 0
+      end
+
+      new_version = version_array.join('.')
+
+      metadata_file = "#{@cookbook.root_dir}/metadata.rb"
+      new_contents = File.read(metadata_file).gsub(/(version\s+['"])[0-9\.]+(['"])/, "\\1#{new_version}\\2")
+      File.open(metadata_file, 'w'){ |f| f.write(new_contents) }
+
+      ui.info "Successfully bumped #{@cookbook.name} to v#{new_version}!"
+    end
+
+    def bump_type
+      TYPE_INDEX[(name_args[1] || 'patch').to_sym]
+    end
+
+    def manual_bump_version
+      version = name_args.last
+      validate_version!(version)
+      version
+    end
+  end
+end

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

@@ -0,0 +1,149 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+
+module KnifeSpork
+  class SporkCheck < Chef::Knife
+    include KnifeSpork::Runner
+
+    banner 'knife spork check COOKBOOK (options)'
+
+    option :all,
+      :short => '--a',
+      :long => '--all',
+      :description => 'Show all uploaded versions of the cookbook'
+      
+      option :fail,
+       	 :long => "--fail",
+       	 :description => "If the check fails exit with non-zero exit code"
+
+    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(':') }
+
+    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)
+
+      option :skip_dependencies,
+        :short => '-s',
+        :long => '--skip-dependencies',
+        :description => 'Berksfile skips resolving source cookbook dependencies',
+        :default => true
+    end
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      if name_args.empty?
+        ui.fatal 'You must specify a cookbook name!'
+        show_usage
+        exit(1)
+      end
+
+      #First load so plugins etc know what to work with
+      initial_load
+      
+      run_plugins(:before_check)
+
+      #Reload cookbook in case a VCS plugin found updates
+      initial_load
+
+      check
+      run_plugins(:after_check)
+    end
+
+    private
+    def check
+      ui.msg "Checking versions for cookbook #{@cookbook.name}..."
+      ui.msg ""
+      ui.msg "Local Version:"
+      ui.msg "  #{local_version}"
+      ui.msg ""
+      ui.msg "Remote Versions: (* indicates frozen)"
+      remote_versions.each do |remote_version|
+        if frozen?(remote_version)
+          ui.msg " *#{remote_version}"
+        else
+          ui.msg "  #{remote_version}"
+        end
+      end
+      ui.msg ""
+
+      remote_versions.each do |remote_version|
+        if remote_version == local_version
+          if frozen?(remote_version)
+            message = "Your local version (#{local_version}) is frozen on the remote server. You'll need to bump before you can upload."
+            if config[:fail]
+              fail_and_exit("#{message}")
+            else
+              ui.warn("#{message}")
+              answer = ui.ask("Would you like to perform a patch-level bump on the #{@cookbook.name} cookbook now? (Y/N)")
+              if answer == "Y" or answer == "y"
+                bump = SporkBump.new
+                bump.name_args = [@cookbook.name]
+                bump.run
+              else
+                ui.info "Skipping bump..."
+              end
+            end
+          else
+            message =  "The version #{local_version} exists on the server and is not frozen. Uploading will overwrite!"
+            config[:fail] ? fail_and_exit("#{message}") : ui.error("#{message}")
+          end
+
+          return
+        end
+      end
+
+      ui.msg 'Everything looks good!'
+    end
+
+    def initial_load
+      begin
+        @cookbook = load_cookbook(name_args.first)
+      rescue Chef::Exceptions::CookbookNotFoundInRepo => e
+        ui.error "#{name_args.first} does not exist locally in your cookbook path(s), Exiting."
+        exit(1)
+      end
+    end
+
+    def local_version
+      @cookbook.version
+    end
+
+    def remote_versions
+      @remote_versions ||= begin
+        environment = config[:environment]
+        api_endpoint = environment ? "environments/#{environment}/cookbooks/#{@cookbook.name}" : "cookbooks/#{@cookbook.name}"
+        cookbooks = rest.get_rest(api_endpoint)
+
+        versions = cookbooks[@cookbook.name.to_s]['versions']
+        (config[:all] ? versions : versions[0..4]).collect{|v| v['version']}
+      rescue Net::HTTPServerException => e
+        ui.info "#{@cookbook.name} does not yet exist on the Chef Server!"
+        return []
+      end
+    end
+
+    def frozen?(version)
+      @versions_cache ||= {}
+
+      @versions_cache[version.to_sym] ||= begin
+        environment = config[:environment]
+        api_endpoint = environment ? "environments/#{environment}/cookbooks/#{@cookbook.name}" : "cookbooks/#{@cookbook.name}/#{version}"
+        rest.get_rest(api_endpoint).to_hash['frozen?']
+      end
+    end
+    
+    def fail_and_exit(message, options={})
+     	ui.fatal message
+     	show_usage if options[:show_usage]
+     	exit 1
+    end
+  end
+end

data/lib/chef/knife/spork-databag-create.rb

@@ -0,0 +1,53 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+require 'json'
+
+module KnifeSpork
+  class SporkDataBagCreate < Chef::Knife
+    include KnifeSpork::Runner
+
+    deps do
+      require 'chef/knife/data_bag_create'
+    end
+
+    banner 'knife data bag create BAG [ITEM] (options)'
+
+    option :secret,
+           :short => "-s SECRET",
+           :long  => "--secret ",
+           :description => "The secret key to use to encrypt data bag item values"
+
+    option :secret_file,
+           :long => "--secret-file SECRET_FILE",
+           :description => "A file containing the secret key to use to encrypt data bag item values"
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      if @name_args.nil?
+        ui.error "You must specify a data bag name"
+        ui.error opt_parser
+        exit 1
+      end
+
+      @object_name = @name_args.first
+
+      run_plugins(:before_databagcreate)
+      pre_databag = {}
+      databag_create
+      post_databag = load_databag(@object_name)
+      @object_difference = json_diff(pre_databag,post_databag).to_s
+      run_plugins(:after_databagcreate)
+    end
+
+    private
+    def databag_create
+      dbc = Chef::Knife::DataBagCreate.new
+      dbc.name_args = @name_args
+      dbc.config[:editor] = config[:editor]
+      dbc.config[:secret] = config[:secret]
+      dbc.config[:secret_file] = config[:secret_file]
+      dbc.run
+    end
+  end
+end

data/lib/chef/knife/spork-databag-delete.rb

@@ -0,0 +1,51 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+require 'json'
+
+module KnifeSpork
+  class SporkDataBagDelete < Chef::Knife
+    include KnifeSpork::Runner
+
+    deps do
+      require 'chef/knife/data_bag_delete'
+    end
+
+    banner 'knife data bag delete BAG [ITEM] (options)'
+
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      if @name_args.length == 2
+        @object_name = @name_args.first
+        @object_secondary_name = @name_args.last
+        run_plugins(:before_databagitemdelete)
+        pre_databag = load_databag_item(@object_name,@object_secondary_name)
+        databag_delete
+        post_databag = {}
+        @object_difference = json_diff(pre_databag,post_databag).to_s
+        run_plugins(:after_databagitemdelete)
+      elsif @name_args.length == 1
+        @object_name = @name_args.first
+        run_plugins(:before_databagdelete)
+        pre_databag = load_databag(@object_name)
+        databag_delete
+        post_databag =  {}
+        @object_difference = json_diff(pre_databag,post_databag).to_s
+        run_plugins(:after_databagdelete)
+      else
+        show_usage
+        ui.fatal("You must specify at least a data bag name")
+        exit 1
+      end
+    end
+
+    private
+    def databag_delete
+      dbd = Chef::Knife::DataBagDelete.new
+      dbd.name_args = @name_args
+      dbd.config[:editor] = config[:editor]
+      dbd.run
+    end
+  end
+end

data/lib/chef/knife/spork-databag-edit.rb

@@ -0,0 +1,54 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+require 'json'
+
+module KnifeSpork
+  class SporkDataBagEdit < Chef::Knife
+    include KnifeSpork::Runner
+
+    deps do
+      require 'chef/knife/data_bag_edit'
+    end
+
+    banner 'knife spork data bag edit BAG ITEM (options)'
+
+    option :secret,
+           :short => "-s SECRET",
+           :long  => "--secret ",
+           :description => "The secret key to use to encrypt data bag item values"
+
+    option :secret_file,
+           :long => "--secret-file SECRET_FILE",
+           :description => "A file containing the secret key to use to encrypt data bag item values"
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      if @name_args.length != 2
+        ui.error "You must supply the data bag and an item to edit!"
+        ui.error opt_parser
+        exit 1
+      end
+
+      @object_name = @name_args.first
+      @object_secondary_name = @name_args.last
+
+      run_plugins(:before_databagedit)
+      pre_databag = load_databag_item(@object_name, @object_secondary_name)
+      databag_edit
+      post_databag = load_databag_item(@object_name, @object_secondary_name)
+      @object_difference = json_diff(pre_databag,post_databag).to_s
+      run_plugins(:after_databagedit)
+    end
+
+    private
+    def databag_edit
+      dbe = Chef::Knife::DataBagEdit.new
+      dbe.name_args = @name_args
+      dbe.config[:editor] = config[:editor]
+      dbe.config[:secret] = config[:secret]
+      dbe.config[:secret_file] = config[:secret_file]
+      dbe.run
+    end
+  end
+end

data/lib/chef/knife/spork-databag-fromfile.rb

@@ -0,0 +1,82 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+require 'json'
+
+module KnifeSpork
+  class SporkDataBagFromFile < Chef::Knife
+    include KnifeSpork::Runner
+
+    deps do
+      require 'chef/knife/data_bag_from_file'
+    end
+
+    banner 'knife data bag from file BAG FILE|FOLDER [FILE|FOLDER..] (options)'
+
+    option :secret,
+           :short => "-s SECRET",
+           :long  => "--secret ",
+           :description => "The secret key to use to encrypt data bag item values"
+
+    option :secret_file,
+           :long => "--secret-file SECRET_FILE",
+           :description => "A file containing the secret key to use to encrypt data bag item values"
+
+    option :all,
+           :short => "-a",
+           :long  => "--all",
+           :description => "Upload all data bags"
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      @object_name = @name_args.first
+
+      if config[:all] == true
+        test = Chef::Knife::DataBagFromFile.new
+        test.config[:verbosity] = 3
+        databags = test.send(:find_all_data_bags)
+        databags.each do |bag|
+          test.send(:find_all_data_bag_items,bag).each do |item|
+            @object_name = bag
+            @object_secondary_name = item.split("/").last
+            run_plugins(:before_databagfromfile)
+            begin
+              pre_databag = load_databag_item(@object_name, @object_secondary_name.gsub(".json",""))
+            rescue
+              pre_databag = {}
+            end
+            databag_from_file([@object_name,@object_secondary_name])
+            post_databag = load_databag_item(@object_name, @object_secondary_name.gsub(".json",""))
+            @object_difference = json_diff(pre_databag,post_databag).to_s
+            run_plugins(:after_databagfromfile)
+          end
+        end
+
+      else
+        @name_args[1..-1].each do |arg|
+            @object_secondary_name = arg.split("/").last
+            run_plugins(:before_databagfromfile)
+            begin
+              pre_databag = load_databag_item(@object_name, @object_secondary_name.gsub(".json",""))
+            rescue
+              pre_databag = {}
+            end
+            databag_from_file
+            post_databag = load_databag_item(@object_name, @object_secondary_name.gsub(".json",""))
+            @object_difference = json_diff(pre_databag,post_databag).to_s
+            run_plugins(:after_databagfromfile)
+        end
+      end
+    end
+
+    private
+    def databag_from_file(data_bag_names=nil)
+      dbff = Chef::Knife::DataBagFromFile.new
+      dbff.name_args = data_bag_names || @name_args
+      dbff.config[:editor] = config[:editor]
+      dbff.config[:secret] = config[:secret]
+      dbff.config[:secret_file] = config[:secret_file]
+      dbff.run
+    end
+  end
+end