knife-spork-berks 1.3.2

data/lib/chef/knife/spork-node-runlistset.rb

@@ -0,0 +1,42 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+
+module KnifeSpork
+  class SporkNodeRunListSet < Chef::Knife
+    include KnifeSpork::Runner
+
+    banner 'knife spork node run_list set NODE ENTRIES (options)'
+
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      if @name_args.size < 2
+        ui.error "You must supply both a node name and a run list."
+        show_usage
+        exit 1
+      end
+
+      @object_name = @name_args.first
+
+      @object_secondary_name = @name_args[1..-1].map do |entry|
+        entry.split(',').map { |e| e.strip }
+      end.flatten.to_s
+
+      run_plugins(:before_noderunlistset)
+      pre_node = load_node(@object_name)
+      node_runlistset
+      post_node = load_node(@object_name)
+      @object_difference = json_diff(pre_node,post_node).to_s
+      run_plugins(:after_noderunlistset)
+    end
+
+    private
+    def node_runlistset
+      nrla = Chef::Knife::NodeRunListSet.new
+      nrla.name_args = @name_args
+      nrla.config[:editor] = config[:editor]
+      nrla.run
+    end
+  end
+end

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

@@ -0,0 +1,107 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+
+begin
+  require 'berkshelf'
+rescue LoadError; end
+
+module KnifeSpork
+  class SporkOmni < Chef::Knife
+    include KnifeSpork::Runner
+
+    banner 'knife spork omni 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 :depends,
+           :short => '-D',
+           :long => '--include-dependencies',
+           :description => 'Also upload cookbook dependencies during the upload step'
+
+    option :bump_level,
+           :short => '-l',
+           :long  => '--bump-level [major|minor|patch]',
+           :description => 'Version level to bump the cookbook (defaults to patch)',
+           :default => nil
+
+    option :omni_environment,
+           :short => '-e',
+           :long  => '--environment ENVIRONMENT',
+           :description => 'Environment to promote the cookbook to',
+           :default => nil
+
+    option :remote,
+           :long  => '--remote',
+           :description => 'Make omni finish with promote --remote instead of a local promote',
+           :default => nil
+
+    if defined?(::Berkshelf)
+      option :berksfile,
+             :short => '-b',
+             :long => '--berksfile BERKSFILE',
+             :description => 'Path to a Berksfile to operate from',
+             :default => File.join(Dir.pwd, ::Berkshelf::DEFAULT_FILENAME)
+    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
+
+      cookbook = name_args.first
+
+      run_plugins(:before_omni)
+      omni(cookbook)
+      run_plugins(:after_omni)
+    end
+
+    private
+
+    def bump(cookbook)
+      ui.msg "OMNI: Bumping #{cookbook}"
+      bump = SporkBump.new
+      bump.name_args = [cookbook,config[:bump_level]]
+      bump.run
+    end
+
+    def upload(cookbook)
+      ui.msg "OMNI: Uploading #{cookbook}"
+      upload = SporkUpload.new
+      upload.name_args = [cookbook]
+      upload.config[:cookbook_path] = config[:cookbook_path]
+      upload.config[:depends] = config[:depends]
+      upload.run
+    end
+
+    def promote(cookbook)
+      ui.msg "OMNI: Promoting #{cookbook}"
+      promote = SporkPromote.new
+      if config[:omni_environment]
+        promote.name_args = [config[:omni_environment],cookbook]
+      else
+        promote.name_args = [cookbook]
+      end
+      promote.config[:remote] = config[:remote]
+      if defined?(::Berkshelf)
+        promote.config[:berksfile] = config[:berksfile]
+      end
+      promote.run
+    end
+
+    def omni(cookbook)
+      bump(cookbook)
+      ui.msg ""
+      upload(cookbook)
+      ui.msg ""
+      promote(cookbook)
+    end
+  end
+end

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

@@ -0,0 +1,166 @@
+require 'chef/knife'
+require 'chef/exceptions'
+require 'knife-spork/runner'
+
+begin
+  require 'berkshelf'
+rescue LoadError; end
+
+module KnifeSpork
+  class SporkPromote < Chef::Knife
+    include KnifeSpork::Runner
+
+    banner 'knife spork promote ENVIRONMENT COOKBOOK (options)'
+
+    option :version,
+      :short => '-v',
+      :long  => '--version VERSION',
+      :description => 'Set the environment\'s version constraint to the specified version',
+      :default => nil
+
+    option :remote,
+      :long  => '--remote',
+      :description => 'Save the environment to the chef server in addition to the local JSON file',
+      :default => nil
+
+    if defined?(::Berkshelf)
+      option :berksfile,
+        :short => '-b',
+        :long => '--berksfile 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)
+
+      if @name_args.empty?
+        show_usage
+        ui.error("You must specify the cookbook and environment to promote to")
+        exit 1
+      end
+
+      #First load so plugins etc know what to work with
+      @environments, @cookbook = load_environments_and_cookbook
+
+      run_plugins(:before_promote)
+
+      #Reload cookbook and env in case a VCS plugin found updates
+      @environments, @cookbook = load_environments_and_cookbook
+
+      check_cookbook_uploaded(@cookbook)
+
+      @environments.each do |e|
+        environment = load_environment_from_file(e)
+
+
+        promote(environment, @cookbook)
+
+        ui.msg "Saving changes to #{e}.json"
+
+        new_environment_json = pretty_print_json(environment.to_hash)
+        save_environment_changes(e, new_environment_json)
+
+        if config[:remote] || spork_config.always_promote_remote
+          ui.msg "Uploading #{environment.name}.json to Chef Server"
+          save_environment_changes_remote(e)
+          ui.info "Promotion complete at #{Time.now}!"
+        else
+          ui.info "Promotion complete. Don't forget to upload your changed #{environment.name}.json to Chef Server"
+        end
+      end
+      run_plugins(:after_promote_local)
+      if config[:remote] || spork_config.always_promote_remote
+        run_plugins(:after_promote_remote)
+      end
+    end
+
+    def update_version_constraints(environment, cookbook, new_version)
+      validate_version!(new_version)
+      environment.cookbook_versions[cookbook] = "= #{new_version}"
+    end
+
+    def save_environment_changes_remote(environment)
+      local_environment = load_environment_from_file(environment)
+      remote_environment = load_remote_environment(environment)
+      @environment_diffs ||= Hash.new
+      @environment_diffs["#{environment}"] = environment_diff(local_environment, remote_environment)
+
+      version_change_threshold = spork_config.version_change_threshold || 2
+      env_constraints_diff = constraints_diff(@environment_diffs["#{environment}"]).select{|k,v| v > version_change_threshold}
+
+      if env_constraints_diff.size !=0 then
+        ui.warn 'You\'re about to promote a significant version number change to 1 or more cookbooks:'
+        ui.warn @environment_diffs["#{environment}"].select{|k,v|env_constraints_diff.has_key?(k)}.collect{|k,v| "\t#{k}: #{v}"}.join("\n")
+
+        begin
+          ui.confirm('Are you sure you want to continue?')
+        rescue SystemExit => e
+          if e.status == 3
+            ui.confirm("Would you like to reset your local #{environment}.json to match the remote server?")
+            tmp = Chef::Environment.load(environment)
+            save_environment_changes(environment, pretty_print_json(tmp))
+            ui.info "#{environment}.json was reset"
+          end
+
+          raise
+        end
+      end
+
+      if @environment_diffs["#{environment}"].size > 1
+        ui.msg ""
+        ui.warn "You're about to promote changes to several cookbooks at once:"
+        ui.warn @environment_diffs["#{environment}"].collect{|k,v| "\t#{k}: #{v}"}.join("\n")
+
+        begin
+          ui.confirm('Are you sure you want to continue?')
+        rescue SystemExit => e
+          if e.status == 3
+            ui.confirm("Would you like to reset your local #{environment}.json to match the remote server?")
+            tmp = Chef::Environment.load(environment)
+            save_environment_changes(environment, pretty_print_json(tmp))
+            ui.info "#{environment}.json was reset"
+          end
+
+          raise
+        end
+      end
+
+      local_environment.save
+    end
+
+    def save_environment_changes(environment, json)
+      environments_path = spork_config[:environment_path] || cookbook_path.gsub('cookbooks', 'environments')
+      environment_path = File.expand_path( File.join(environments_path, "#{environment}.json") )
+
+      File.open(environment_path, 'w'){ |f| f.puts(json) }
+    end
+
+    def promote(environment, cookbook_names)
+      cookbook_names = [cookbook_names].flatten
+
+      cookbook_names.each do |cookbook_name|
+        validate_version!(config[:version])
+        version = config[:version] || load_cookbook(cookbook_name).version
+
+        ui.msg "Adding version constraint #{cookbook_name} = #{version}"
+        update_version_constraints(environment, cookbook_name, version)
+      end
+    end
+
+    def check_cookbook_uploaded(cookbook_name)
+      validate_version!(config[:version])
+      version = config[:version] || load_cookbook(cookbook_name).version
+
+      environment = config[:environment]
+      api_endpoint = environment ? "environments/#{environment}/cookbooks/#{cookbook_name}/#{version}" : "cookbooks/#{cookbook_name}/#{version}"
+
+      begin
+        cookbooks = rest.get_rest(api_endpoint)
+      rescue Net::HTTPServerException => e
+        ui.error "#{cookbook_name}@#{version} does not exist on Chef Server! Upload the cookbook first by running:\n\n\tknife spork upload #{cookbook_name}\n\n"
+        exit(1)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,43 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+
+module KnifeSpork
+  class SporkRoleCreate < Chef::Knife
+    include KnifeSpork::Runner
+
+    banner 'knife spork role create ROLE (options)'
+
+    option :description,
+           :short => "-d DESC",
+           :long => "--description DESC",
+           :description => "The role description"
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      if @name_args.empty?
+        show_usage
+        ui.error("You must specify a role name")
+        exit 1
+      end
+
+      @object_name = @name_args.first
+
+      run_plugins(:before_rolecreate)
+      pre_role = {}
+      role_create
+      post_role = load_role(@object_name)
+      @object_difference = json_diff(pre_role,post_role).to_s
+      run_plugins(:after_rolecreate)
+    end
+
+    private
+    def role_create
+      rc = Chef::Knife::RoleCreate.new
+      rc.name_args = @name_args
+      rc.config[:editor] = config[:editor]
+      rc.config[:description] = config[:description]
+      rc.run
+    end
+  end
+end

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

@@ -0,0 +1,36 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+
+module KnifeSpork
+  class SporkRoleDelete < Chef::Knife
+    include KnifeSpork::Runner
+
+    banner 'knife spork role delete ROLENAME (options)'
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      if @name_args.empty?
+        show_usage
+        ui.error("You must specify a role name")
+        exit 1
+      end
+
+      @object_name = @name_args.first
+
+      run_plugins(:before_roledelete)
+      pre_role = load_role(@object_name)
+      role_delete
+      post_role = {}
+      @object_difference = json_diff(pre_role,post_role).to_s
+      run_plugins(:after_roledelete)
+    end
+
+    private
+    def role_delete
+      rd = Chef::Knife::RoleDelete.new
+      rd.name_args = @name_args
+      rd.run
+    end
+  end
+end

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

@@ -0,0 +1,37 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+
+module KnifeSpork
+  class SporkRoleEdit < Chef::Knife
+    include KnifeSpork::Runner
+
+    banner 'knife spork role edit ROLENAME (options)'
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      if @name_args.empty?
+        show_usage
+        ui.error("You must specify a role name")
+        exit 1
+      end
+
+      @object_name = @name_args.first
+
+      run_plugins(:before_roleedit)
+      pre_role = load_role(@object_name)
+      role_edit
+      post_role = load_role(@object_name)
+      @object_difference = json_diff(pre_role,post_role).to_s
+      run_plugins(:after_roleedit)
+    end
+
+    private
+    def role_edit
+      re = Chef::Knife::RoleEdit.new
+      re.name_args = @name_args
+      re.config[:editor] = config[:editor]
+      re.run
+    end
+  end
+end

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

@@ -0,0 +1,42 @@
+require 'chef/knife'
+require 'knife-spork/runner'
+require 'json'
+
+module KnifeSpork
+  class SporkRoleFromFile < Chef::Knife
+    include KnifeSpork::Runner
+
+    deps do
+      require 'chef/knife/role_from_file'
+    end
+
+    banner 'knife spork role from file FILENAME (options)'
+
+    def run
+      self.config = Chef::Config.merge!(config)
+
+      if @name_args.empty?
+        show_usage
+        ui.error("You must specify a role name")
+        exit 1
+      end
+
+      @name_args.each do |arg|
+          @object_name = arg.split("/").last
+          run_plugins(:before_rolefromfile)
+          pre_role = load_role(@object_name.gsub(".json","").gsub(".rb",""))
+          role_from_file
+          post_role = load_role(@object_name.gsub(".json","").gsub(".rb",""))
+          @object_difference = json_diff(pre_role,post_role).to_s
+          run_plugins(:after_rolefromfile)
+      end
+    end
+
+    private
+    def role_from_file
+      rff = Chef::Knife::RoleFromFile.new
+      rff.name_args = @name_args
+      rff.run
+    end
+  end
+end