vagrant-chef-zero 0.2.10

data/lib/vagrant-chef-zero/config.rb

@@ -0,0 +1,61 @@
+require "vagrant"
+
+module VagrantPlugins
+  module ChefZero
+    class Config < Vagrant.plugin("2", :config)
+      # Path to role fixtures
+      #
+      # @return [String]
+      attr_accessor :roles
+
+      # Path to environment fixtures
+      #
+      # @return [String]
+      attr_accessor :environments
+
+      # Path to node fixtures
+      #
+      # @return [String]
+      attr_accessor :nodes
+
+      # Path to cookbooks
+      #
+      # @return [String]
+      attr_accessor :cookbooks
+
+      # Path to data bag fixtures
+      #
+      # @return [String]
+      attr_accessor :data_bags
+
+      # Is the plugin enabled?
+      #
+      # @return [TrueClass, FalseClass]
+      attr_accessor :enabled
+
+      def initialize
+        @roles        = UNSET_VALUE
+        @environments = UNSET_VALUE
+        @nodes        = UNSET_VALUE
+        @cookbooks    = UNSET_VALUE
+        @data_bags    = UNSET_VALUE
+        @enabled      = UNSET_VALUE
+      end
+
+      def validate(machine)
+        { "Chef Zero" => [] }
+      end
+
+      def finalize!
+        @enabled = true if @enabled == UNSET_VALUE
+        @roles = nil if @roles == UNSET_VALUE
+        @environments = nil if @environments == UNSET_VALUE
+        @nodes = nil if @nodes == UNSET_VALUE
+        @cookbooks = nil if @cookbooks == UNSET_VALUE
+        @data_bags = nil if @data_bags == UNSET_VALUE
+        {}
+      end
+
+    end
+  end
+end

data/lib/vagrant-chef-zero/env.rb

@@ -0,0 +1,15 @@
+module VagrantPlugins
+  module ChefZero
+    # @author Jamie Winsor <reset@riotgames.com>
+    #
+    # Environment data to build up and persist through the middleware chain
+    class Env
+      # @return [Vagrant::UI::Colored]
+      attr_accessor :ui
+
+      def initialize
+        @ui = ::Vagrant::UI::Colored.new.scope('Chef Zero')
+      end
+    end
+  end
+end

data/lib/vagrant-chef-zero/env_helpers.rb

@@ -0,0 +1,105 @@
+module VagrantPlugins
+  module ChefZero
+    module EnvHelpers
+
+      def server_info(env)
+        dict = { host: nil, client_name: nil, client_key: nil }
+        provisioners(:chef_client, env).each do |provisioner|
+          host = provisioner.config.chef_server_url ||= nil
+          client_name = provisioner.config.validation_client_name ||= nil
+          client_key = provisioner.config.validation_key_path ||= nil
+          dict = { host: host, client_name: client_name, client_key: client_key }
+        end
+        dict
+      end
+
+      def set_config(config_field, new_value, env)
+        provisioners(:chef_client, env).each do |provisioner|
+          provisioner.config.instance_variable_set(config_field, new_value)
+        end
+      end
+
+      def provisioners(name, env)
+        env[:machine].config.vm.provisioners.select { |prov| prov.name == name }
+      end
+
+      def get_validation_client_name(env)
+        current_client_name = server_info(env)[:client_name]
+        if ! current_client_name || current_client_name.empty?
+          current_client_name = "dummy-validator"
+        end
+        current_client_name
+      end
+
+      def get_chef_server_url(env)
+        current_chef_server_url = server_info(env)[:host]
+        if ! current_chef_server_url || current_chef_server_url.empty?
+          require 'socket'
+          ip_address = Socket.ip_address_list.detect{|intf| intf.ipv4_private?}.ip_address
+          current_chef_server_url = "http://#{ip_address}:4000"
+        end
+        current_chef_server_url
+      end
+
+      def get_key_path(env)
+        current_key_path = server_info(env)[:client_key]
+        if ! current_key_path || ! ::File.exists?(current_key_path)
+          current_key_path = get_fake_key_path
+        end
+        current_key_path
+      end
+
+      def get_fake_key_path
+        require 'openssl'
+        require 'tmpdir'
+        fake_key = OpenSSL::PKey::RSA.new(2048)
+        fake_directory = File.join(Dir.tmpdir, "fake_key")
+        fake_key_path = File.join(fake_directory, "fake.pem")
+        Dir.mkdir(fake_directory) unless File.exists?(fake_directory)
+        File.open(fake_key_path,"w") {|f| f.puts fake_key } unless File.exists?(fake_key_path)
+        fake_key_path
+      end
+
+      def get_host(env)
+        url = server_info(env)[:host]
+        if url
+          # Some terrible string manipulation to get the ip
+          return url.split('//').last.split(':').first
+        end
+      end
+
+      def get_port(env)
+        url = server_info(env)[:host]
+        # Same with the port
+        if url
+          p = url.split(':').last
+          if p && p != ""
+            port = p
+          end
+        else
+          port = "4000"
+        end
+        port
+      end
+
+      def chef_client?(env)
+        provisioners(:chef_client, env).any?
+      end
+
+      def chef_zero_enabled?(env)
+        env[:global_config].chef_zero.enabled && chef_client?(env)
+      end
+
+      def berkshelf_enabled?(env)
+        env[:global_config].berkshelf.enabled == true && chef_client?(env)
+      end
+
+      def set_berkshelf_client_key(value)
+        ObjectSpace.each_object(Berkshelf::Config).each do |o|
+          o.chef.client_key = value
+        end
+      end
+
+    end
+  end
+end

data/lib/vagrant-chef-zero/plugin.rb

@@ -0,0 +1,53 @@
+require 'ridley'
+
+# This is a sanity check to make sure no one is attempting to install
+# this into an early Vagrant version.
+if Vagrant::VERSION < "1.2.0"
+  raise "The Vagrant Chef Zero plugin is only compatible with Vagrant 1.2+"
+end
+
+module VagrantPlugins
+  module ChefZero
+    class Plugin < Vagrant.plugin("2")
+
+      class << self
+        def reconfig(hook)
+          hook.before(::Vagrant::Action::Builtin::ConfigValidate, VagrantPlugins::ChefZero::Action.chef_zero_ui_setup)
+          hook.before(::Vagrant::Action::Builtin::ConfigValidate, VagrantPlugins::ChefZero::Action.chef_zero_reconfig)
+        end
+
+        def provision(hook)
+          hook.before(::Vagrant::Action::Builtin::Provision, VagrantPlugins::ChefZero::Action.chef_zero_provision)
+        end
+
+        def destroy(hook)
+          hook.after(VagrantPlugins::ProviderVirtualBox::Action::DestroyUnusedNetworkInterfaces, VagrantPlugins::ChefZero::Action.chef_zero_destroy)
+        end
+
+      end
+
+      name "chef_zero"
+      description <<-DESC
+      This plugin adds configuration options to allow Vagrant to interact with
+      Chef Zero
+      DESC
+
+      # Deep magic of not binding to any specific action.. so we can bind to them all
+      action_hook(:chef_zero_reconfig, &method(:reconfig))
+
+      action_hook(:chef_zero_up, :machine_action_up, &method(:provision))
+
+      action_hook(:chef_zero_provision, :machine_action_provision, &method(:provision))
+
+      action_hook(:chef_zero_provision, :machine_action_reload, &method(:provision))
+
+      action_hook(:chef_zero_destroy, :machine_action_destroy, &method(:destroy))
+
+      config(:chef_zero) do
+        require_relative "config"
+        Config
+      end
+
+    end
+  end
+end

data/lib/vagrant-chef-zero/server_helpers.rb

@@ -0,0 +1,44 @@
+module VagrantPlugins
+  module ChefZero
+    module ServerHelpers
+
+      def start_chef_zero(env)
+        port = get_port(env)
+        host = get_host(env)
+        if ! chef_zero_server_running?(port)
+          vagrant_gems = ENV['GEM_PATH'].split(':').select { |gp| gp.include?('vagrant.d')}.first
+          chef_zero_binary = ::File.join(vagrant_gems, "bin", "chef-zero")
+          proc = IO.popen("#{chef_zero_binary} --host #{host} --port #{port} 2>&1 > /dev/null")
+          env[:chef_zero].ui.info("Starting Chef Zero at http://#{host}:#{port}")
+        end
+        while ! chef_zero_server_running?(port)
+          sleep 1
+          env[:chef_zero].ui.warn("Waiting for Chef Zero to start")
+        end
+      end
+
+      def stop_chef_zero(env)
+        port = get_port(env)
+        if chef_zero_server_running?(port)
+          pid = get_chef_zero_server_pid(port)
+          env[:chef_zero].ui.info("Stopping Chef Zero")
+          system("kill #{pid}")
+        end
+      end
+
+      def chef_zero_server_running?(port)
+        pid = get_chef_zero_server_pid(port)
+        !! pid
+      end
+
+      def get_chef_zero_server_pid(port)
+        pid = %x[ lsof -i tcp:#{port} | grep -E 'ruby|chef-zero' | awk '{print $2}' ]
+        if pid && pid != ""
+          return pid
+        end
+        return nil
+      end
+
+    end
+  end
+end

data/lib/vagrant-chef-zero/version.rb

@@ -0,0 +1,5 @@
+module VagrantPlugins
+  module ChefZero
+    VERSION = "0.2.10"
+  end
+end

data/lib/vagrant-chef-zero.rb

@@ -0,0 +1,15 @@
+begin
+  require "vagrant"
+rescue LoadError
+  raise "The Vagrant Chef-Zero plugin must be run within Vagrant."
+end
+
+module VagrantPlugins
+  module ChefZero
+    autoload :Env, 'vagrant-chef-zero/env'
+    autoload :Action, 'vagrant-chef-zero/action'
+    autoload :Config, 'vagrant-chef-zero/config'
+  end
+end
+
+require "vagrant-chef-zero/plugin"

data/spec/lib/server_helpers_spec.rb

@@ -0,0 +1,65 @@
+require_relative '../spec_helper'
+
+describe 'VagrantPlugins::ChefZero::ServerHelpers' do
+
+  before do
+  class DummyClass
+    include VagrantPlugins::ChefZero::ServerHelpers
+  end
+
+  @d = DummyClass.new
+  end
+
+  describe "chef_zero_server_running?" do
+
+    it "should have an chef_zero_server_running method" do
+      @d.must_respond_to(:chef_zero_server_running?)
+    end
+
+    describe "server is running" do
+
+      before do
+        @d.stubs(:get_chef_zero_server_pid).returns("bar")
+      end
+
+      it "should show that the server is running" do
+        port = "foo"
+        @d.chef_zero_server_running?(port).must_equal true
+      end
+
+    end
+
+    describe "server not running" do
+
+      before do
+        @d.stubs(:get_chef_zero_server_pid).returns(nil)
+      end
+
+      it "should show that the server is running" do
+        port = "foo"
+        @d.chef_zero_server_running?(port).must_equal false
+      end
+
+    end
+
+  end
+
+  describe "stop_chef_zero" do
+
+    it "should have an stop_chef_zero method" do
+      @d.must_respond_to(:stop_chef_zero)
+    end
+
+    describe "server is running" do
+
+      before do
+        @d.stubs(:chef_zero_server_running?).returns(true)
+        @d.stubs(:`).returns
+
+      end
+
+    end
+
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,21 @@
+require 'simplecov'
+SimpleCov.formatter = SimpleCov::Formatter::SimpleFormatter
+SimpleCov.start
+
+require_relative "../lib/vagrant-chef-zero"
+require_relative "../lib/vagrant-chef-zero/server_helpers.rb"
+require_relative "../lib/vagrant-chef-zero.rb"
+
+require 'minitest/autorun'
+require 'minitest/spec'
+require 'minitest/mock'
+require 'mocha/setup' # This should be after any minitest requires
+require 'turn'
+require 'json'
+
+Turn.config do |c|
+  # :outline  - turn's original case/test outline mode [default]
+  c.format  = :outline
+  # use humanized test names (works only with :outline format)
+  c.natural = true
+end

data/spec/vagrant-chef-zero/fixtures/cookbooks/blah/metadata.rb

@@ -0,0 +1,3 @@
+version "1.5.1"
+
+depends "blork"

data/spec/vagrant-chef-zero/fixtures/cookbooks/blah/recipes/default.rb

@@ -0,0 +1 @@
+template "/tmp/hii"

data/spec/vagrant-chef-zero/fixtures/cookbooks/blah/templates/default/hi

@@ -0,0 +1 @@
+helloooooooooo world

data/spec/vagrant-chef-zero/fixtures/cookbooks/blork/recipes/default.rb

@@ -0,0 +1 @@
+template "/tmp/hii"

data/spec/vagrant-chef-zero/fixtures/cookbooks/blork/templates/default/hi

@@ -0,0 +1 @@
+hhellooooooo world

data/spec/vagrant-chef-zero/fixtures/data_bags/foo/bar.json

@@ -0,0 +1 @@
+{ "id": "bar" }

data/spec/vagrant-chef-zero/fixtures/data_bags/foo/baz.json

@@ -0,0 +1 @@
+{ "id": "baz" }

data/spec/vagrant-chef-zero/fixtures/data_bags/foo/blarghle.json

@@ -0,0 +1 @@
+{ "id": "blarghle", "name": "sldfkja" }

data/spec/vagrant-chef-zero/fixtures/data_bags/foom/x.json

@@ -0,0 +1,5 @@
+{
+  "id": "x",
+  "foo": "barr",
+  "wow": "ee"
+}

data/spec/vagrant-chef-zero/fixtures/data_bags/widdle/blank.json

@@ -0,0 +1 @@
+{ "id": "blank" }

data/spec/vagrant-chef-zero/fixtures/data_bags/widdle/wow.json

@@ -0,0 +1 @@
+{ "id": "wow" }

data/spec/vagrant-chef-zero/fixtures/environments/desert.json

@@ -0,0 +1,13 @@
+{
+  "name": "desert",
+  "default_attributes": {
+  },
+  "json_class": "Chef::Environment",
+  "description": "This is your just desert",
+  "cookbook_versions": {
+    "bork": "> 2.0"
+  },
+  "override_attributes": {
+  },
+  "chef_type": "environment"
+}

data/spec/vagrant-chef-zero/fixtures/environments/rainforest.json

@@ -0,0 +1,13 @@
+{
+  "name": "rainforest",
+  "description": "Aaaaagh!  Save me!",
+  "cookbook_versions": {
+  },
+  "json_class": "Chef::Environment",
+  "chef_type": "environment",
+  "default_attributes": {
+    "a": "b"
+  },
+  "override_attributes": {
+  }
+}

data/spec/vagrant-chef-zero/fixtures/environments/semi_arid_plains.json

@@ -0,0 +1,13 @@
+{
+  "name": "semi_arid_plains",
+  "description": "The plains, they are so arid",
+  "cookbook_versions": {
+  },
+  "json_class": "Chef::Environment",
+  "chef_type": "environment",
+  "default_attributes": {
+    "derp": true
+  },
+  "override_attributes": {
+  }
+}

data/spec/vagrant-chef-zero/fixtures/nodes/blah.json

@@ -0,0 +1,7 @@
+{
+  "chef_type": "node",
+  "json_class": "Chef::Node",
+  "name": "blah",
+  "chef_environment": "_default",
+  "run_list": []
+}

data/spec/vagrant-chef-zero/fixtures/nodes/blarrrrgh.json

@@ -0,0 +1,7 @@
+{
+  "chef_type": "node",
+  "json_class": "Chef::Node",
+  "name": "blarrrrgh",
+  "chef_environment": "_default",
+  "run_list": []
+}

data/spec/vagrant-chef-zero/fixtures/nodes/boxer.json

@@ -0,0 +1,7 @@
+{
+  "chef_type": "node",
+  "json_class": "Chef::Node",
+  "name": "boxer",
+  "chef_environment": "_default",
+  "run_list": []
+}

data/spec/vagrant-chef-zero/fixtures/nodes/camel.json

@@ -0,0 +1,7 @@
+{
+  "chef_type": "node",
+  "json_class": "Chef::Node",
+  "name": "camel",
+  "chef_environment": "desert",
+  "run_list": []
+}

data/spec/vagrant-chef-zero/fixtures/nodes/monkey.json

@@ -0,0 +1,7 @@
+{
+  "chef_type": "node",
+  "json_class": "Chef::Node",
+  "name": "monkey",
+  "chef_environment": "rainforest",
+  "run_list": []
+}

data/vagrant-chef-zero.gemspec

@@ -0,0 +1,74 @@
+$:.unshift File.expand_path("../lib", __FILE__)
+require "vagrant-chef-zero/version"
+
+Gem::Specification.new do |s|
+  s.name          = "vagrant-chef-zero"
+  s.version       = VagrantPlugins::ChefZero::VERSION
+  s.platform      = Gem::Platform::RUBY
+  s.authors       = "Andrew Gross"
+  s.email         = "andrew.w.gross@gmail.com"
+  s.homepage      = "http://github.com/andrewgross/vagrant-chef-zero"
+  s.summary       = "Enables Vagrant to interact with Chef Zero."
+  s.description   = "Enables Vagrant to interact with Chef Zero"
+
+  s.required_rubygems_version = ">= 1.3.6"
+  s.rubyforge_project         = "vagrant-chef-zero"
+  s.license = "MIT"
+
+  s.add_dependency "chef-zero", "~> 1.3"
+  s.add_dependency "ridley", ">= 1.0.0"
+
+
+  # Stolen from Vagrant Berkshelf because Ruby hates dependency resolution
+  # activesupport 3.2.13 contains an incompatible hard lock on i18n (= 0.6.1)
+  s.add_dependency 'activesupport', '>= 3.2.0', '< 3.2.13'
+
+  # Explicit locks to ensure we activate the proper gem versions for Vagrant
+  s.add_dependency 'i18n', '~> 0.6.0'
+  s.add_dependency 'net-ssh', '~> 2.6.6'
+  s.add_dependency 'net-scp', '~> 1.1.0'
+
+  s.add_development_dependency "mocha"
+  s.add_development_dependency "simplecov"
+  s.add_development_dependency "rake"
+  s.add_development_dependency "turn"
+  s.add_development_dependency "minitest", '< 5.0'
+  s.add_development_dependency "minitest-reporters"
+
+  # The following block taken from @mitchellh 's vagrant-aws code
+
+  # The following block of code determines the files that should be included
+  # in the gem. It does this by reading all the files in the directory where
+  # this gemspec is, and parsing out the ignored files from the gitignore.
+  # Note that the entire gitignore(5) syntax is not supported, specifically
+  # the "!" syntax, but it should mostly work correctly.
+  root_path      = File.dirname(__FILE__)
+  all_files      = Dir.chdir(root_path) { Dir.glob("**/{*,.*}") }
+  all_files.reject! { |file| [".", ".."].include?(File.basename(file)) }
+  gitignore_path = File.join(root_path, ".gitignore")
+  gitignore      = File.readlines(gitignore_path)
+  gitignore.map!    { |line| line.chomp.strip }
+  gitignore.reject! { |line| line.empty? || line =~ /^(#|!)/ }
+
+  unignored_files = all_files.reject do |file|
+    # Ignore any directories, the gemspec only cares about files
+    next true if File.directory?(file)
+
+    # Ignore any paths that match anything in the gitignore. We do
+    # two tests here:
+    #
+    #   - First, test to see if the entire path matches the gitignore.
+    #   - Second, match if the basename does, this makes it so that things
+    #     like '.DS_Store' will match sub-directories too (same behavior
+    #     as git).
+    #
+    gitignore.any? do |ignore|
+      File.fnmatch(ignore, file, File::FNM_PATHNAME) ||
+        File.fnmatch(ignore, File.basename(file), File::FNM_PATHNAME)
+    end
+  end
+
+  s.files         = unignored_files
+  s.executables   = unignored_files.map { |f| f[/^bin\/(.*)/, 1] }.compact
+  s.require_path  = 'lib'
+end