knife-vagrant 0.0.1

data/README.md

@@ -0,0 +1,47 @@
+knife-vagrant
+========
+First pass at knife-vagrant, a knife plugin that will create a Vagrant box, then run a set of Chef recipes in it.  Eventually will also run cucumber tests and report the results.
+
+Requirements
+-------------------
+
+- Virtualbox 4.1.X
+- Vagrant (http://vagrantup.com/)
+- chef gem install (client only)
+
+Installation
+-------------------
+#### Script Install
+Copy https://github.com/garrettux/knife-vagrant/blob/master/lib/chef/knife/vagrant_test.rb to your .chef/plugins/knife directory.
+
+#### Gem Install
+Not ready yet, will be available on rubygems.org later.
+
+Usage
+-------------------
+knife vagrant test (options)
+
+    -b, --box BOX                    Name of vagrant box to be provisioned
+
+    -U, --box-url URL                URL of pre-packaged vbox template.  Can be a local path or an HTTP URL.  Defaults to ./package.box
+
+    -l, --chef-loglevel LEVEL        Logging level for the chef-client process that runs inside the provisioned VM.  Default is INFO
+
+    -x, --destroy                    Destroy vagrant box and delete chef node/client when finished
+
+    -E, --environment ENVIRONMENT    Set the Chef environment
+
+    -H, --hostname HOSTNAME          Hostname to be set as hostname on vagrant box when provisioned
+
+    -m, --memsize MEMORY             Amount of RAM to allocate to provisioned VM, in MB.  Defaults to 1024
+
+    -D, --vagrant-dir PATH           Path to vagrant project directory.  Defaults to cwd (/home/mgarrett/knife-vagrant) if not specified
+
+    -r, --vagrant-run-list RUN_LIST  Comma separated list of roles/recipes to apply
+
+    -y, --yes                        Say yes to all prompts for confirmation
+
+Disclaimer
+-------------------
+
+This is my first stab at it, so I can't make any promises as to how well it works just yet.

data/lib/chef/knife/vagrant_test.rb

@@ -0,0 +1,134 @@
+# knife-vagrant
+# knife plugin for spinning up a vagrant instance and testing a runlist.
+
+module KnifePlugins
+  class VagrantTest < Chef::Knife
+
+    banner "knife vagrant test (options)"
+
+    deps do
+      require 'rubygems'
+      require 'pp'
+      require 'vagrant'
+      require 'vagrant/cli'
+      require 'chef/node'
+      require 'chef/api_client'
+    end
+ 
+    # Default is nil here because if :cwd passed to the Vagrant::Environment object is nil,
+    # it defaults to Dir.pwd, which is the cwd of the running process.
+    option :vagrant_dir,
+      :short => '-D PATH',
+      :long => '--vagrant-dir PATH',
+      :description => "Path to vagrant project directory.  Defaults to cwd (#{Dir.pwd}) if not specified",
+      :default => nil
+      
+    option :vagrant_run_list,
+      :short => "-r RUN_LIST",
+      :long => "--vagrant-run-list RUN_LIST",
+      :description => "Comma separated list of roles/recipes to apply",
+      :proc => lambda { |o| o.split(/[\s,]+/) },
+      :default => []
+ 
+    option :box,
+      :short => '-b BOX',
+      :long => '--box BOX',
+      :description => 'Name of vagrant box to be provisioned',
+      :default => false
+
+    option :hostname,
+      :short => '-H HOSTNAME',
+      :long => '--hostname HOSTNAME',
+      :description => 'Hostname to be set as hostname on vagrant box when provisioned',
+      :default => 'vagrant_test'
+
+    option :box_url,
+      :short => '-U URL',
+      :long => '--box-url URL',
+      :description => 'URL of pre-packaged vbox template.  Can be a local path or an HTTP URL.  Defaults to ./package.box',
+      :default => "#{Dir.pwd}/package.box"
+
+    option :memsize,
+      :short => '-m MEMORY',
+      :long => '--memsize MEMORY',
+      :description => 'Amount of RAM to allocate to provisioned VM, in MB.  Defaults to 1024',
+      :default => 1024
+  
+    option :chef_loglevel,
+      :short => '-l LEVEL',
+      :long => '--chef-loglevel LEVEL',
+      :description => 'Logging level for the chef-client process that runs inside the provisioned VM.  Default is INFO',
+      :default => 'INFO'
+
+    option :destroy,
+      :short => '-x',
+      :long => '--destroy',
+      :description => 'Destroy vagrant box and delete chef node/client when finished',
+      :default => false
+
+    # TODO - hook into chef/runlist
+    def build_runlist(runlist)
+      runlist.collect { |i| "\"#{i}\"" }.join(",\n")
+    end
+
+    # TODO:  see if there's a way to pass this whole thing in as an object or hash or something, instead of writing a file to disk.
+    def build_vagrantfile
+      file = <<-EOF
+        Vagrant::Config.run do |config|
+          config.vm.forward_port(22, 2222)
+          config.vm.box = "#{config[:box]}"
+          config.vm.host_name = "#{config[:hostname]}"
+          config.vm.customize do |vm|
+            vm.memory_size = #{config[:memsize]}
+            vm.name = "#{config[:box]}"
+          end
+          config.vm.box_url = "#{config[:box_url]}"
+          config.vm.provision :chef_client do |chef|
+            chef.chef_server_url = "#{Chef::Config[:chef_server_url]}"
+            chef.validation_key_path = "#{Chef::Config[:validation_key]}"
+            chef.validation_client_name = "#{Chef::Config[:validation_client_name]}"
+            chef.node_name = "#{config[:hostname]}"
+            chef.provisioning_path = "#{Chef::Config[:provisioning_path]}"
+            chef.log_level = :#{config[:chef_loglevel].downcase}
+            chef.environment = "#{Chef::Config[:environment]}"
+            chef.run_list = [
+              #{build_runlist(config[:vagrant_run_list])}
+            ]
+          end
+        end
+      EOF
+      file
+    end
+      
+    def write_vagrantfile(path, content)
+      File.open(path, 'w') { |f| f.write(content) }
+    end
+  
+    def cleanup(path)
+      File.delete(path)
+    end
+    
+    def run
+      ui.msg('Loading vagrant environment..')
+      Dir.chdir(config[:vagrant_dir])
+      vagrantfile = "#{config[:vagrant_dir]}/Vagrantfile"
+      write_vagrantfile(vagrantfile, build_vagrantfile)
+      @vagrant_env = Vagrant::Environment.new(:cwd => config[:vagrant_dir], :ui_class => Vagrant::UI::Colored)
+      @vagrant_env.load!
+      begin
+        @vagrant_env.cli("up")
+      rescue
+        raise # I'll put some error handling here later.
+      ensure
+        if config[:destroy]
+          ui.confirm("Destroy vagrant box #{config[:box]} and delete chef node and client")
+          config[:yes] = true unless config[:yes]
+          @vagrant_env.cli("destroy")
+          delete_object(Chef::Node, config[:hostname])
+          delete_object(Chef::ApiClient, config[:hostname])
+        end
+      end
+    end
+
+  end
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: knife-vagrant
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Michael Garrett
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-06-01 00:00:00.000000000 Z
+dependencies: []
+description: knife-vagrant, a knife plugin that will create a Vagrant box, then run
+  a set of Chef recipes in it.
+email: garrettux@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- lib/chef/knife/vagrant_test.rb
+homepage: https://github.com/garrettux/knife-vagrant
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: knife-vagrant, a knife plugin that will create a Vagrant box, then run a
+  set of Chef recipes in it.
+test_files: []