vagrant 0.1.0

data/config/default.rb

@@ -0,0 +1,29 @@
+Vagrant::Config.run do |config|
+  # default config goes here
+  config.vagrant.log_output = STDOUT
+  config.vagrant.dotfile_name = ".vagrant"
+  config.vagrant.home = "~/.vagrant"
+
+  config.ssh.username = "vagrant"
+  config.ssh.password = "vagrant"
+  config.ssh.host = "localhost"
+  config.ssh.forwarded_port_key = "ssh"
+  config.ssh.max_tries = 10
+
+  config.vm.box_ovf = "box.ovf"
+  config.vm.base_mac = "0800279C2E42"
+  config.vm.project_directory = "/vagrant"
+  config.vm.forward_port("ssh", 22, 2222)
+  config.vm.disk_image_format = 'VMDK'
+
+  config.package.name = 'vagrant'
+  config.package.extension = '.box'
+
+  config.chef.enabled = false
+  config.chef.cookbooks_path = "cookbooks"
+  config.chef.provisioning_path = "/tmp/vagrant-chef"
+  config.chef.json = {
+    :instance_role => "vagrant",
+    :recipes => ["vagrant_main"]
+  }
+end

data/lib/vagrant.rb

@@ -0,0 +1,14 @@
+libdir = File.dirname(__FILE__)
+$:.unshift(libdir)
+PROJECT_ROOT = File.join(libdir, '..') unless defined?(PROJECT_ROOT)
+
+# The libs which must be loaded prior to the rest
+%w{tempfile open-uri ftools json pathname logger uri net/http virtualbox net/ssh tarruby
+  net/scp fileutils vagrant/util vagrant/actions/base vagrant/downloaders/base}.each do |f|
+  require f
+end
+
+# Glob require the rest
+Dir[File.join(PROJECT_ROOT, "lib", "vagrant", "**", "*.rb")].each do |f|
+  require f
+end

data/lib/vagrant/actions/base.rb

@@ -0,0 +1,93 @@
+module Vagrant
+  module Actions
+    # Base class for any command actions.
+    #
+    # Actions are the smallest unit of functionality found within
+    # Vagrant. Vagrant composes many actions together to execute
+    # its complex tasks while keeping the individual pieces of a
+    # task as discrete reusable actions. Actions are ran exclusively
+    # by an {Runner action runner} which is simply a subclass of {Runner}.
+    #
+    # Actions work by implementing any or all of the following methods
+    # which a {Runner} executes:
+    #
+    # * `prepare` - Called once for each action before any action has `execute!`
+    #   called. This is meant for basic setup.
+    # * `execute!` - This is where the meat of the action typically goes;
+    #   the main code which executes the action.
+    # * `cleanup` - This is called exactly once for each action after every
+    #   other action is completed. It is meant for cleaning up any resources.
+    # * `rescue` - This is called if an exception occurs in _any action_. This
+    #   gives every other action a chance to clean itself up.
+    #
+    # For details of each step of an action, read the specific function call
+    # documentation below.
+    class Base
+      # The {Runner runner} which is executing the action
+      attr_reader :runner
+
+      # Included so subclasses don't need to include it themselves.
+      include Vagrant::Util
+
+      # Initialization of the action, passing any arguments which may have
+      # been given to the {Runner runner}. This method can be used by subclasses
+      # to save any of the configuration options which are passed in.
+      def initialize(runner, *args)
+        @runner = runner
+      end
+
+      # This method is called once per action, allowing the action
+      # to setup any callbacks, add more events, etc. Prepare is
+      # called in the order the actions are defined, and the action
+      # itself has no control over this.
+      #
+      # Examples of its usage:
+      #
+      # Perhaps we need an additional action only if a configuration is set:
+      #
+      #     def prepare
+      #       @vm.actions << FooAction if Vagrant.config[:foo] == :bar
+      #     end
+      #
+      def prepare; end
+
+      # This method is called once, after preparing, to execute the
+      # actual task. This method is responsible for calling any
+      # callbacks. Adding new actions here will have unpredictable
+      # effects and should never be done.
+      #
+      # Examples of its usage:
+      #
+      #     def execute!
+      #       @vm.invoke_callback(:before_oven, "cookies")
+      #       # Do lots of stuff here
+      #       @vm.invoke_callback(:after_oven, "more", "than", "one", "option")
+      #     end
+      #
+      def execute!; end
+
+      # This method is called after all actions have finished executing.
+      # It is meant as a place where final cleanup code can be done, knowing
+      # that all other actions are finished using your data.
+      def cleanup; end
+
+      # This method is only called if some exception occurs in the chain
+      # of actions. If an exception is raised in any action in the current
+      # chain, then every action part of that chain has {#rescue} called
+      # before raising the exception further. This method should be used to
+      # perform any cleanup necessary in the face of errors.
+      #
+      # **Warning:** Since this method is called when an exception is already
+      # raised, be _extra careful_ when implementing this method to handle
+      # all your own exceptions, otherwise it'll mask the initially raised
+      # exception.
+      def rescue(exception); end
+    end
+
+    # An exception which occured within an action. This should be used instead of
+    # {Vagrant::Util#error_and_exit error_and_exit}, since it allows the {Runner} to call
+    # {Base#rescue rescue} on all the actions and properly exit. Any message
+    # passed into the {ActionException} is then shown and and vagrant exits.
+    class ActionException < Exception; end
+  end
+end

data/lib/vagrant/actions/box/add.rb

@@ -0,0 +1,22 @@
+module Vagrant
+  module Actions
+    module Box
+      # A meta-action which adds a box by downloading and unpackaging it.
+      # This action downloads and unpackages a box with a given URI. This
+      # is a _meta action_, meaning it simply adds more actions to the
+      # action chain, and those actions do the work.
+      #
+      # This is the action called by {Box#add}.
+      class Add < Base
+        def prepare
+          if File.exists?(@runner.directory)
+            raise ActionException.new("A box with the name '#{@runner.name}' already exists, please use another name or use `vagrant box remove #{@runner.name}`")
+          end
+
+          @runner.add_action(Download)
+          @runner.add_action(Unpackage)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/actions/box/destroy.rb

@@ -0,0 +1,14 @@
+module Vagrant
+  module Actions
+    module Box
+      # Action to destroy a box. This action is not reversible and expects
+      # to be called by a {Box} object.
+      class Destroy < Base
+        def execute!
+          logger.info "Deleting box directory..."
+          FileUtils.rm_rf(@runner.directory)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/actions/box/download.rb

@@ -0,0 +1,63 @@
+module Vagrant
+  module Actions
+    module Box
+      # An action which acts on a box by downloading the box file from
+      # the given URI into a temporary location. This action parses a
+      # given URI and handles downloading it via one of the many Vagrant
+      # downloads (such as {Vagrant::Downloaders::File}).
+      #
+      # This action cleans itself up by removing the downloaded box file.
+      class Download < Base
+        BASENAME = "box"
+        BUFFERSIZE = 1048576 # 1 MB
+
+        attr_reader :downloader
+
+        def prepare
+          # Parse the URI given and prepare a downloader
+          uri = URI.parse(@runner.uri)
+          uri_map = [[URI::HTTP, Downloaders::HTTP], [URI::Generic, Downloaders::File]]
+
+          uri_map.find do |uri_type, downloader_klass|
+            if uri.is_a?(uri_type)
+              logger.info "#{uri_type} for URI, downloading via #{downloader_klass}..."
+              @downloader = downloader_klass.new
+            end
+          end
+
+          raise ActionException.new("Unknown URI type for box download.") unless @downloader
+        end
+
+        def execute!
+          with_tempfile do |tempfile|
+            download_to(tempfile)
+            @runner.temp_path = tempfile.path
+          end
+        end
+
+        def cleanup
+          if @runner.temp_path && File.exist?(@runner.temp_path)
+            logger.info "Cleaning up downloaded box..."
+            File.unlink(@runner.temp_path)
+          end
+        end
+
+        def rescue(exception)
+          cleanup
+        end
+
+        def with_tempfile
+          logger.info "Creating tempfile for storing box file..."
+          Tempfile.open(BASENAME, Env.tmp_path) do |tempfile|
+            yield tempfile
+          end
+        end
+
+        def download_to(f)
+          logger.info "Copying box to temporary location..."
+          downloader.download!(@runner.uri, f)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/actions/box/unpackage.rb

@@ -0,0 +1,49 @@
+module Vagrant
+  module Actions
+    module Box
+      # This action unpackages a downloaded box file into its final
+      # box destination within the vagrant home folder.
+      class Unpackage < Base
+        TAR_OPTIONS = [File::RDONLY, 0644, Tar::GNU]
+
+        def execute!
+          @runner.invoke_around_callback(:unpackage) do
+            setup_box_dir
+            decompress
+          end
+        end
+
+        def rescue(exception)
+          if File.directory?(box_dir)
+            logger.info "An error occurred, rolling back box unpackaging..."
+            FileUtils.rm_rf(box_dir)
+          end
+        end
+
+        def setup_box_dir
+          if File.directory?(box_dir)
+            error_and_exit(<<-msg)
+This box appears to already exist! Please call `vagrant box remove #{@runner.name}`
+and then try to add it again.
+msg
+          end
+
+          FileUtils.mkdir_p(box_dir)
+        end
+
+        def box_dir
+          @runner.directory
+        end
+
+        def decompress
+          Dir.chdir(box_dir) do
+            logger.info "Extracting box to #{box_dir}..."
+            Tar.open(@runner.temp_path, *TAR_OPTIONS) do |tar|
+              tar.extract_all
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/actions/runner.rb

@@ -0,0 +1,128 @@
+module Vagrant
+  module Actions
+    # Base class for any class which will act as a runner
+    # for actions. A runner handles queueing up and executing actions,
+    # and executing the methods of an action in the proper order. The
+    # action runner also handles invoking callbacks that actions may
+    # request.
+    #
+    # # Executing Actions
+    #
+    # Actions can be executed by adding them and executing them all
+    # at once:
+    #
+    #     runner = Vagrant::Actions::Runner.new
+    #     runner.add_action(FooAction)
+    #     runner.add_action(BarAction)
+    #     runner.add_action(BazAction)
+    #     runner.execute!
+    #
+    # Single actions have a shorthand to be executed:
+    #
+    #     Vagrant::Actions::Runner.execute!(FooAction)
+    #
+    # Arguments may be passed into added actions by adding them after
+    # the action class:
+    #
+    #     runner.add_action(FooAction, "many", "arguments", "may", "follow")
+    #
+    class Runner
+      include Vagrant::Util
+
+      class << self
+        # Executes a specific action, optionally passing in any arguments to that
+        # action. This method is shorthand to initializing a runner, adding a single
+        # action, and executing it.
+        def execute!(action_klass, *args)
+          runner = new
+          runner.add_action(action_klass, *args)
+          runner.execute!
+        end
+      end
+
+      # Returns an array of all the actions in queue. Because this
+      # will persist accross calls (calling {#actions} twice will yield
+      # exactly the same object), to clear or modify it, use the ruby
+      # array methods which act on `self`, such as `Array#clear`.
+      #
+      # @return [Array]
+      def actions
+        @actions ||= []
+      end
+
+      # Returns the first action instance which matches the given class.
+      #
+      # @param [Class] action_klass The action to search for in the queue
+      # @return [Object]
+      def find_action(action_klass)
+        actions.find { |a| a.is_a?(action_klass) }
+      end
+
+      # Add an action to the list of queued actions to execute. This method
+      # appends the given action class to the end of the queue. Any arguments
+      # given after the class are passed into the class constructor.
+      def add_action(action_klass, *args)
+        actions << action_klass.new(self, *args)
+      end
+
+      # Execute the actions in queue. This method can also optionally be used
+      # to execute a single action on an instance. The syntax for executing a
+      # single method on an instance is the same as the {execute!} class method.
+      def execute!(single_action=nil, *args)
+        if single_action
+          actions.clear
+          add_action(single_action, *args)
+        end
+
+        # Call the prepare method on each once its
+        # initialized, then call the execute! method
+        begin
+          [:prepare, :execute!, :cleanup].each do |method|
+            actions.each do |action|
+              action.send(method)
+            end
+          end
+        rescue Exception => e
+          # Run the rescue code to do any emergency cleanup
+          actions.each do |action|
+            action.rescue(e)
+          end
+
+          # If its an ActionException, error and exit the message
+          if e.is_a?(ActionException)
+            error_and_exit(e.message)
+            return
+          end
+
+          # Finally, reraise the exception
+          raise
+        end
+
+        # Clear the actions
+        actions.clear
+      end
+
+      # Invokes an "around callback" which invokes before_name and
+      # after_name for the given callback name, yielding a block between
+      # callback invokations.
+      def invoke_around_callback(name, *args)
+        invoke_callback("before_#{name}".to_sym, *args)
+        yield
+        invoke_callback("after_#{name}".to_sym, *args)
+      end
+
+      # Invokes a single callback. This method will go through each action
+      # and call the method given in the parameter `name` if the action
+      # responds to it.
+      def invoke_callback(name, *args)
+        # Attempt to call the method for the callback on each of the
+        # actions
+        results = []
+        actions.each do |action|
+          results << action.send(name, *args) if action.respond_to?(name)
+        end
+        results
+      end
+    end
+  end
+end

data/lib/vagrant/actions/vm/destroy.rb

@@ -0,0 +1,14 @@
+module Vagrant
+  module Actions
+    module VM
+      class Destroy < Base
+        def execute!
+          @runner.invoke_around_callback(:destroy) do
+            logger.info "Destroying VM and associated drives..."
+            @runner.vm.destroy(:destroy_image => true)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/actions/vm/down.rb

@@ -0,0 +1,12 @@
+module Vagrant
+  module Actions
+    module VM
+      class Down < Base
+        def prepare
+          @runner.add_action(Halt) if @runner.vm.running?
+          @runner.add_action(Destroy)
+        end
+      end
+    end
+  end
+end