head_chef 0.1.0

data/lib/head_chef/cookbook_diff.rb

@@ -0,0 +1,80 @@
+module HeadChef
+  class CookbookDiff
+    attr_reader :diff_hash
+
+    def initialize
+      @diff_hash = { add: [],
+                     update: [],
+                     remove: [],
+                     revert: [],
+                     conflict: [] }
+    end
+
+    # @TODO: cleanup
+    # @TODO: switch statements
+    def add(cookbook)
+      # Removal is the only operation that does not require a diff, as no
+      # cookbook will be uploaded
+      if cookbook.chef_version && !cookbook.berkshelf_version
+        @diff_hash[:remove] << cookbook and return
+      end
+
+      unless cookbook.diff
+        @diff_hash[:conflict] << cookbook and return
+      end
+
+      if cookbook.berkshelf_version && !cookbook.chef_version
+        @diff_hash[:add] << cookbook and return
+      end
+
+      berkshelf_version = Semantic::Version.new(cookbook.berkshelf_version)
+      chef_version = Semantic::Version.new(cookbook.chef_version)
+
+      if berkshelf_version > chef_version
+        @diff_hash[:update] << cookbook and return
+      elsif berkshelf_version < chef_version
+        @diff_hash[:revert] << cookbook and return
+      end
+    end
+
+    def conflicts
+      @diff_hash[:conflict]
+    end
+
+    def conflicts?
+      !@diff_hash[:conflict].empty?
+    end
+
+    def empty?
+      [:add, :update, :remove, :revert, :conflict].each do |method|
+        return false if !@diff_hash[method].empty?
+      end
+
+      true
+    end
+
+    def pretty_print
+      if self.empty?
+        HeadChef.ui.say("Berksfile and Chef environment are identical", :green)
+        return
+      end
+
+      colors = { add: :green,
+                 update: :green,
+                 remove: :red,
+                 revert: :red,
+                 conflict: :red }
+
+      [:add, :update, :remove, :revert, :conflict].each do |method|
+        color = colors[method]
+
+        unless @diff_hash[method].empty?
+          HeadChef.ui.say("#{method.to_s.upcase}:", color)
+          diff_hash[method].each do |cookbook|
+            HeadChef.ui.say("  #{cookbook.to_s}", color)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/head_chef/tasks/diff.rb

@@ -0,0 +1,42 @@
+module HeadChef
+  class Diff
+    def self.diff(environment)
+      HeadChef.ui.info("Loading environment #{environment} from chef server...")
+      chef_environment = HeadChef.chef_server.environment.find(environment)
+
+      if chef_environment
+        chef_versions = chef_environment.cookbook_versions
+      else
+        HeadChef.ui.error("Environment #{environment} not found on chef server.")
+        Kernel.exit(1337)
+      end
+
+      # Run berks install to populate cached cookbook list
+      # @NOTE: for now it is up to user to maintain Berksfile
+      HeadChef.ui.info('Loading cookbooks from berkshelf...')
+      cached_cookbooks = Berkshelf.ui.mute { HeadChef.berksfile.install }
+
+      HeadChef.ui.say('Calculating diff...', :cyan)
+      cookbook_diff = CookbookDiff.new
+
+      cached_cookbooks.each do |berkshelf_cookbook|
+        cookbook_name = berkshelf_cookbook.name.chomp("-#{berkshelf_cookbook.version}")
+
+        if chef_versions[cookbook_name]
+          chef_version = chef_versions[cookbook_name]
+        else
+          chef_version = nil
+        end
+
+        cookbook_diff.add(Cookbook.new(cookbook_name, berkshelf_cookbook.version, chef_version))
+        chef_versions.delete(cookbook_name)
+      end
+
+      chef_versions.each do |cookbook_name, cookbook_version|
+        cookbook_diff.add(Cookbook.new(cookbook_name, nil, cookbook_version))
+      end
+
+      cookbook_diff
+    end
+  end
+end

data/lib/head_chef/tasks/env.rb

@@ -0,0 +1,48 @@
+module HeadChef
+  class Env < Thor
+    class_option :environment,
+      aliases: '-e',
+      banner: '<environment>',
+      desc: 'Applies to the specified environment',
+      type: :string
+
+    desc 'diff', 'Shows cookbook diff between Berksfile and Chef <environment>'
+    long_desc <<-EOD
+      Shows cookbook version diff between Berksfile and Chef <environment>
+
+      By default, matches current git branch name against Chef environment.
+    EOD
+    def diff
+      environment = options[:environment] || HeadChef.current_branch
+
+      Diff.diff(environment).pretty_print
+    end
+
+    desc 'list', 'Lists cookbooks with versions from Chef <environment>.'
+    long_desc <<-EOD
+      Shows cookbook version diff between Berksfile and Chef <environment>
+
+      By default, matches current git branch name against Chef enviroment.
+    EOD
+    def list
+      environment = options[:environment] || HeadChef.current_branch
+
+      List.list(environment)
+    end
+
+    desc 'sync', 'Syncs Berksfile with Chef <environment>'
+    long_desc <<-EOD
+      Syncs Berksfile cookbook with Chef <environment>
+
+      By default, matches current git branch and against Chef enviroment. Chef
+      environment will be created if it does not exist.
+    EOD
+    option :force, banner: '', desc: 'Force upload of cookbooks to chef server'
+    def sync
+      environment = options[:environment] || HeadChef.current_branch
+      force = options[:force] ? true : false
+
+      Sync.sync(environment, force)
+    end
+  end
+end

data/lib/head_chef/tasks/list.rb

@@ -0,0 +1,17 @@
+module HeadChef
+  class List
+    def self.list(environment)
+      chef_environment = HeadChef.chef_server.environment.find(environment)
+
+      unless chef_environment
+        HeadChef.ui.error "Environment #{environment} not found on chef server."
+        Kernel.exit(1337)
+      end
+
+      HeadChef.ui.say("COOKBOOKS:")
+      chef_environment.cookbook_versions.sort.each do |cookbook, version|
+        HeadChef.ui.say("  #{cookbook}: #{version}")
+      end
+    end
+  end
+end

data/lib/head_chef/tasks/sync.rb

@@ -0,0 +1,38 @@
+module HeadChef
+  class Sync
+    def self.sync(environment, force)
+      # Check if environment exits, if not create it
+      # Perform first, if it fails no need to continue
+      unless HeadChef.chef_server.environment.find(environment)
+        HeadChef.chef_server.environment.create(name: environment)
+      end
+
+      # Diff now performs all Berkshelf/lockfile dependency operations
+      HeadChef.ui.say("Determing side effects of sync with chef environment "\
+                      "#{environment}...", :cyan)
+      cookbook_diff = HeadChef.ui.mute { Diff.diff(environment) }
+
+      unless force
+        if cookbook_diff.conflicts?
+          HeadChef.ui.error 'The following cookbooks are in conflict:'
+          cookbook_diff.conflicts.each do |cookbook|
+            HeadChef.ui.error "#{cookbook.name}: #{cookbook.berkshelf_version}"
+          end
+          HeadChef.ui.error 'Use --force to sync environment'
+          Kernel.exit(1337)
+        end
+      end
+
+      # Retrieve berksfile
+      berksfile = HeadChef.berksfile
+
+      HeadChef.ui.say('Uploading cookbooks to chef server...', :cyan)
+      berksfile.upload({force: force})
+
+      # Apply without lock options argument
+      HeadChef.ui.say("Applying Berksfile.lock cookbook version to " \
+                      "environment #{environment}...", :cyan)
+      berksfile.apply(environment, {})
+    end
+  end
+end

data/lib/head_chef/tasks.rb

@@ -0,0 +1,3 @@
+Dir["#{File.dirname(__FILE__)}/tasks/*.rb"].sort.each do |path|
+  require_relative "tasks/#{File.basename(path, '.rb')}"
+end

data/lib/head_chef/ui.rb

@@ -0,0 +1,23 @@
+module HeadChef
+  module UI
+    def mute!
+      @mute = true
+    end
+
+    def unmute!
+      @mute = false
+    end
+
+    def error(message, color = :red)
+      message = set_color(message, *color) if color
+      super(message)
+    end
+
+    def info(message, color = :cyan)
+      message = set_color(message, *color) if color
+      super(message)
+    end
+  end
+end
+
+Thor::Base.shell.send(:include, HeadChef::UI)

data/lib/head_chef/version.rb

@@ -0,0 +1,3 @@
+module HeadChef
+  VERSION = '0.1.0'
+end

data/lib/head_chef.rb

@@ -0,0 +1,63 @@
+# external requires
+require 'berkshelf'
+require 'ridley'
+require 'grit'
+require 'thor'
+require 'semantic'
+require 'pathname'
+
+# internal requires
+require_relative 'head_chef/tasks'
+require_relative 'head_chef/cookbook'
+require_relative 'head_chef/cookbook_diff'
+require_relative 'head_chef/ui'
+require_relative 'head_chef/version'
+
+#@TODO: establish head_chef exit codes
+#Create custom errors
+module HeadChef
+
+  BERKSFILE_LOCATION = 'Berksfile'.freeze
+  BERKSFILE_COOKBOOK_DIR = '.head_chef'.freeze
+
+  class << self
+    def root
+      @root ||= Pathname.new(File.expand_path('../', File.dirname(__FILE__)))
+    end
+
+    def ui
+      @ui ||= Thor::Base.shell.new
+    end
+
+    def chef_server
+      @chef_server ||= Ridley.from_chef_config()
+    end
+
+    # @TODO: refactor?
+    # Is grit necessary to get current branch, is shell command sufficient?
+    # This can look up dir's until it finds .git dir
+    def master_cookbook
+      begin
+        @master_cookbook ||= Grit::Repo.new('.')
+      rescue Grit::InvalidGitRepositoryError
+        puts Dir.pwd
+        HeadChef.ui.error 'head_chef must be run in root of git repo'
+        Kernel.exit(1337)
+      end
+    end
+
+    def current_branch
+      master_cookbook.head.name
+    end
+
+    def berksfile
+      @berksfile ||= Berkshelf::Berksfile.from_file(BERKSFILE_LOCATION)
+    end
+
+    def cleanup
+      if Dir.exists?(BERKSFILE_COOKBOOK_DIR)
+        FileUtils.rm_rf(BERKSFILE_COOKBOOK_DIR)
+      end
+    end
+  end
+end

data/spec/cookbook_diff_spec.rb

@@ -0,0 +1,77 @@
+require 'spec_helper'
+
+describe HeadChef::CookbookDiff do
+  let(:empty_diff_hash) do
+    { add: [],
+      update: [],
+      remove: [],
+      revert: [],
+      conflict: [] }
+  end
+
+  shared_examples_for 'correct add to CookbookDiff' do |cookbook, method|
+    it "appends #{cookbook.name} cookbook to :#{method} list" do
+        subject.add(cookbook)
+        expect(subject.diff_hash[method]).to include(cookbook)
+    end
+  end
+
+  describe 'ClassMethods' do
+    describe '#new' do
+      it 'initializes hash' do
+        expect(subject.diff_hash).to eq(empty_diff_hash)
+      end
+    end
+  end
+
+  describe 'InstanceMethods' do
+    describe '#add(HeadChef::Cookbook)' do
+
+      context 'with cookbook content conflict' do
+        before(:each) do
+          HeadChef::Cookbook.any_instance.stub(:diff).and_return(false)
+        end
+
+        it_should_behave_like 'correct add to CookbookDiff',
+          HeadChef::Cookbook.new('add_test', '0.0.1', nil), :conflict
+
+        it_should_behave_like 'correct add to CookbookDiff',
+          HeadChef::Cookbook.new('update_test', '0.0.2', '0.0.1'), :conflict
+
+        it_should_behave_like 'correct add to CookbookDiff',
+          HeadChef::Cookbook.new('remove_test', nil, '0.0.1'), :remove
+
+        it_should_behave_like 'correct add to CookbookDiff',
+          HeadChef::Cookbook.new('revert_test', '0.0.1', '0.0.2'), :conflict
+
+        it_should_behave_like 'correct add to CookbookDiff',
+          HeadChef::Cookbook.new('conflict_test', '0.0.1', '0.0.1'), :conflict
+      end
+
+      context 'without cookbook content conflict' do
+        before(:each) do
+          HeadChef::Cookbook.any_instance.stub(:diff).and_return(true)
+        end
+
+        it_should_behave_like 'correct add to CookbookDiff',
+          HeadChef::Cookbook.new('add_test', '0.0.1', nil), :add
+
+        it_should_behave_like 'correct add to CookbookDiff',
+          HeadChef::Cookbook.new('update_test', '0.0.2', '0.0.1'), :update
+
+        it_should_behave_like 'correct add to CookbookDiff',
+          HeadChef::Cookbook.new('remove_test', nil, '0.0.1'), :remove
+
+        it_should_behave_like 'correct add to CookbookDiff',
+          HeadChef::Cookbook.new('revert_test', '0.0.1', '0.0.2'), :revert
+
+        it 'does nothing when berkshelf version == chef version' do
+          cookbook = HeadChef::Cookbook.new('conflict_test', '0.0.1', '0.0.1')
+          subject.add(cookbook)
+          expect(subject.diff_hash).to eq(empty_diff_hash)
+        end
+
+      end
+    end
+  end
+end

data/spec/cookbook_spec.rb

@@ -0,0 +1,60 @@
+require 'spec_helper'
+
+describe HeadChef::Cookbook do
+  let(:cookbook_resource) { double('Ridley::CookbookResouce') }
+  let(:cached_cookbooks) { [] }
+  let(:cached_cookbook) { double('Berkshelf::CachedCookbook') }
+
+  subject { HeadChef::Cookbook.new('test', '0.0.1', '0.0.2') }
+
+  describe 'ClassMethods' do
+    describe '::new' do
+      it 'reads name' do
+        expect(subject.name).to eq('test')
+      end
+
+      it 'reads berkshelf version' do
+        expect(subject.berkshelf_version).to eq('0.0.1')
+      end
+
+      it 'reads chef version' do
+        expect(subject.chef_version).to eq('0.0.2')
+      end
+    end
+  end
+
+  describe 'InstanceMethods' do
+    describe '#diff' do
+      before(:each) do
+        HeadChef.stub_chain(:chef_server, :cookbook).
+          and_return(cookbook_resource)
+
+        allow(cookbook_resource).to receive(:find).
+          with(subject.name, subject.berkshelf_version).
+          and_return(cookbook_resource)
+        allow(cookbook_resource).to receive(:manifest).and_return(Hashie::Mash.new)
+
+        HeadChef.stub_chain(:berksfile, :cached_cookbooks).
+          and_return(cached_cookbooks)
+        allow(cached_cookbooks).to receive(:find).and_return(cached_cookbook)
+
+        allow(cached_cookbook).to receive(:path)
+        allow(subject).to receive(:remove_ignored_files).and_return([])
+      end
+
+      after(:each) do
+        subject.diff
+      end
+
+      it 'retrieves cookbook checksums from chef server' do
+        expect(cookbook_resource).to receive(:find).
+          with(subject.name, subject.berkshelf_version)
+      end
+
+      it 'loads cookbook from berkshelf cache' do
+        expect(cached_cookbooks).to receive(:find)
+      end
+    end
+  end
+
+end

data/spec/diff_spec.rb

@@ -0,0 +1,38 @@
+require 'spec_helper'
+
+describe HeadChef::Diff do
+  describe 'ClassMethods' do
+    let(:berksfile) { double('Berkshelf::Berksfile') }
+    let(:chef_environment) { double('Hashie::Mash') }
+    let(:environment) { 'test_env' }
+
+    describe '::diff' do
+      before(:each) do
+        HeadChef.stub_chain(:chef_server, :environment, :find).
+          with(environment).
+          and_return(chef_environment)
+
+        allow(chef_environment).to receive(:cookbook_versions).and_return({})
+
+        allow(HeadChef).to receive(:berksfile).and_return(berksfile)
+        allow(berksfile).to receive(:install).and_return([])
+      end
+
+      after(:each) do
+        described_class.diff(environment)
+      end
+
+      it 'loads chef environment' do
+        expect(chef_environment).to receive(:cookbook_versions)
+      end
+
+      it 'calls Berksfile#install to load berkshelf cookbooks into cache' do
+        expect(berksfile).to receive(:install)
+      end
+
+      it 'returns CookbookDiff' do
+        expect(described_class.diff(environment)).to be_an_instance_of(HeadChef::CookbookDiff)
+      end
+    end
+  end
+end

data/spec/env_spec.rb

@@ -0,0 +1,94 @@
+require 'spec_helper'
+
+describe HeadChef::Env do
+  let(:current_branch) { 'test_branch' }
+  let(:environment) { 'option_environment' }
+
+  shared_examples_for "HeadChef::Env command" do |klass, method, return_value|
+
+    context 'defaults' do
+      it 'uses branch name for environment' do
+        expect(klass).to receive(method) do |*args|
+          args[0].should eq(current_branch)
+        end.and_return(return_value)
+      end
+    end
+
+    context 'with --environment' do
+      it 'uses environment option value' do
+        subject.options[:environment] = environment
+
+        expect(klass).to receive(method) do |*args|
+          args[0].should eq(environment)
+        end.and_return(return_value)
+      end
+    end
+  end
+
+  describe 'commands' do
+
+    before(:each) do
+      allow(HeadChef).to receive(:current_branch).and_return(current_branch)
+
+      # Unfreeze Thor::CoreExt::HashWithIndifferentAccess
+      subject.options = subject.options.dup
+    end
+
+
+    describe '::diff' do
+      let(:cookbook_diff) { HeadChef::CookbookDiff.new }
+
+      after(:each) do
+        subject.diff
+      end
+
+      it_should_behave_like 'HeadChef::Env command',
+        HeadChef::Diff, :diff, HeadChef::CookbookDiff.new
+
+      context 'defaults' do
+        it 'outputs CookbookDiff' do
+          allow(HeadChef::Diff).to receive(:diff).
+            with(any_args).and_return(cookbook_diff)
+
+          expect(cookbook_diff).to receive(:pretty_print)
+        end
+      end
+    end
+
+    describe '::list' do
+      after(:each) do
+        subject.list
+      end
+
+      it_should_behave_like 'HeadChef::Env command',
+        HeadChef::List, :list, nil
+    end
+
+    describe '::sync' do
+      after(:each) do
+        subject.sync
+      end
+
+      it_should_behave_like 'HeadChef::Env command',
+        HeadChef::Sync, :sync, nil
+
+      context 'defaults' do
+        it 'uses false for force option' do
+          expect(HeadChef::Sync).to receive(:sync) do |*args|
+            args[1].should eq(false)
+          end
+        end
+      end
+
+      context 'with --force' do
+        it 'sets force option to true' do
+          subject.options[:force] = true
+
+          expect(HeadChef::Sync).to receive(:sync) do |*args|
+            args[1].should eq(true)
+          end
+        end
+      end
+    end
+  end
+end

data/spec/fixtures/Berksfiles/default

@@ -0,0 +1,3 @@
+site :opscode
+
+cookbook 'java'

data/spec/fixtures/Berksfiles/template.erb

@@ -0,0 +1,5 @@
+site :opscode
+
+<% @cookbooks.each do |cookbook| %>
+<%= "cookbook '#{cookbook[0]}', '#{cookbook[1]}', #{cookbook[2]}" %>
+<% end %>

data/spec/fixtures/cookbooks/test_cookbook/metadata.rb

@@ -0,0 +1,2 @@
+name             'test_cookbook'
+version          '0.1.0'

data/spec/fixtures/cookbooks/test_cookbook/recipes/default.rb

@@ -0,0 +1,8 @@
+#
+# Cookbook Name:: test_cookbook
+# Recipe:: default
+#
+# Copyright 2013, YOUR_COMPANY_NAME
+#
+# All rights reserved - Do Not Redistribute
+#

data/spec/fixtures/cookbooks/test_cookbook_file_content_conflict/metadata.rb

@@ -0,0 +1,2 @@
+name             'test_cookbook'
+version          '0.1.0'

data/spec/fixtures/cookbooks/test_cookbook_file_content_conflict/recipes/default.rb

@@ -0,0 +1,10 @@
+#
+# Cookbook Name:: test_cookbook
+# Recipe:: default
+#
+# Copyright 2013, YOUR_COMPANY_NAME
+#
+# All rights reserved - Do Not Redistribute
+#
+#
+### DIFFERENT CONTENT

data/spec/fixtures/cookbooks/test_cookbook_file_list_conflict/chefignore

@@ -0,0 +1,2 @@
+chefignore
+recipes/default.rb

data/spec/fixtures/cookbooks/test_cookbook_file_list_conflict/metadata.rb

@@ -0,0 +1,2 @@
+name             'test_cookbook'
+version          '0.1.0'

data/spec/fixtures/cookbooks/test_cookbook_file_list_conflict/recipes/default.rb

@@ -0,0 +1,10 @@
+#
+# Cookbook Name:: test_cookbook
+# Recipe:: default
+#
+# Copyright 2013, YOUR_COMPANY_NAME
+#
+# All rights reserved - Do Not Redistribute
+#
+#
+### DIFFERENT CONTENT

data/spec/fixtures/cookbooks/test_cookbook_file_list_conflict/recipes/not_default.rb

@@ -0,0 +1 @@
+### DIFFERENT FILE