nixme-vagrant 0.7.2

data/lib/vagrant/downloaders/base.rb

@@ -0,0 +1,23 @@
+module Vagrant
+  module Downloaders
+    # Represents a base class for a downloader. A downloader handles
+    # downloading a box file to a temporary file.
+    class Base
+      include Vagrant::Util
+
+      # The environment which this downloader is operating.
+      attr_reader :env
+
+      def initialize(env)
+        @env = env
+      end
+
+      # Called prior to execution so any error checks can be done
+      def prepare(source_url); end
+
+      # Downloads the source file to the destination file. It is up to
+      # implementors of this class to handle the logic.
+      def download!(source_url, destination_file); end
+    end
+  end
+end

data/lib/vagrant/downloaders/file.rb

@@ -0,0 +1,22 @@
+require 'fileutils'
+
+module Vagrant
+  module Downloaders
+    # "Downloads" a file to a temporary file. Basically, this downloader
+    # simply does a file copy.
+    class File < Base
+      def self.match?(uri)
+        ::File.exists?(uri)
+      end
+
+      def prepare(source_url)
+        raise Errors::DownloaderFileDoesntExist if !::File.file?(source_url)
+      end
+
+      def download!(source_url, destination_file)
+        env.ui.info I18n.t("vagrant.downloaders.file.download")
+        FileUtils.cp(source_url, destination_file.path)
+      end
+    end
+  end
+end

data/lib/vagrant/downloaders/http.rb

@@ -0,0 +1,64 @@
+require 'net/http'
+require 'net/https'
+require 'open-uri'
+require 'uri'
+
+module Vagrant
+  module Downloaders
+    # Downloads a file from an HTTP URL to a temporary file. This
+    # downloader reports its progress to stdout while downloading.
+    class HTTP < Base
+      def self.match?(uri)
+        # URI.parse barfs on '<drive letter>:\\files \on\ windows'
+        extracted = URI.extract(uri).first
+        extracted && extracted.include?(uri)
+      end
+
+      def download!(source_url, destination_file)
+        proxy_uri = URI.parse(ENV["http_proxy"] || "")
+        uri = URI.parse(source_url)
+        http = Net::HTTP.new(uri.host, uri.port, proxy_uri.host, proxy_uri.port, proxy_uri.user, proxy_uri.password)
+
+        if uri.scheme == "https"
+          http.use_ssl = true
+          http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        end
+
+        http.start do |h|
+          env.ui.info I18n.t("vagrant.downloaders.http.download", :url => source_url)
+
+          h.request_get(uri.request_uri) do |response|
+            if response.is_a?(Net::HTTPRedirection)
+              # Follow the HTTP redirect.
+              # TODO: Error on some redirect limit
+              download!(response["Location"], destination_file)
+              return
+            end
+
+            total = response.content_length
+            progress = 0
+            segment_count = 0
+
+            response.read_body do |segment|
+              # Report the progress out
+              progress += segment.length
+              segment_count += 1
+
+              # Progress reporting is limited to every 25 segments just so
+              # we're not constantly updating
+              if segment_count % 25 == 0
+                env.ui.report_progress(progress, total)
+                segment_count = 0
+              end
+
+              # Store the segment
+              destination_file.write(segment)
+            end
+          end
+        end
+      rescue SocketError
+        raise Errors::DownloaderHTTPSocketError
+      end
+    end
+  end
+end

data/lib/vagrant/environment.rb

@@ -0,0 +1,392 @@
+require 'pathname'
+require 'fileutils'
+
+module Vagrant
+  # Represents a single Vagrant environment. A "Vagrant environment" is
+  # defined as basically a folder with a "Vagrantfile." This class allows
+  # access to the VMs, CLI, etc. all in the scope of this environment.
+  class Environment
+    ROOTFILE_NAME = "Vagrantfile"
+    HOME_SUBDIRS = ["tmp", "boxes", "logs"]
+    DEFAULT_VM = :default
+    DEFAULT_HOME = "~/.vagrant"
+
+    # Parent environment (in the case of multi-VMs)
+    attr_reader :parent
+
+    # The `cwd` that this environment represents
+    attr_reader :cwd
+
+    # The single VM that this environment represents, in the case of
+    # multi-VM.
+    attr_accessor :vm
+
+    # The {UI} object to communicate with the outside world.
+    attr_writer :ui
+
+    # The {Config} object representing the Vagrantfile loader
+    attr_reader :config_loader
+
+    #---------------------------------------------------------------
+    # Class Methods
+    #---------------------------------------------------------------
+    class << self
+      # Verifies that VirtualBox is installed and that the version of
+      # VirtualBox installed is high enough.
+      def check_virtualbox!
+        version = VirtualBox.version
+        raise Errors::VirtualBoxNotDetected if version.nil?
+        raise Errors::VirtualBoxInvalidVersion, :version => version.to_s if version.to_f < 4.0
+      rescue Errors::VirtualBoxNotDetected
+        # On 64-bit Windows, show a special error. This error is a subclass
+        # of VirtualBoxNotDetected, so libraries which use Vagrant can just
+        # rescue VirtualBoxNotDetected.
+        raise Errors::VirtualBoxNotDetected_Win64 if Util::Platform.windows? && Util::Platform.bit64?
+
+        # Otherwise, reraise the old error
+        raise
+      end
+    end
+
+    # Initializes a new environment with the given options. The options
+    # is a hash where the main available key is `cwd`, which defines where
+    # the environment represents. There are other options available but
+    # they shouldn't be used in general. If `cwd` is nil, then it defaults
+    # to the `Dir.pwd` (which is the cwd of the executing process).
+    def initialize(opts=nil)
+      opts = {
+        :parent => nil,
+        :vm => nil,
+        :cwd => nil,
+      }.merge(opts || {})
+
+      opts[:cwd] ||= Dir.pwd
+      opts[:cwd] = Pathname.new(opts[:cwd])
+
+      opts.each do |key, value|
+        instance_variable_set("@#{key}".to_sym, opts[key])
+      end
+
+      @loaded = false
+    end
+
+    #---------------------------------------------------------------
+    # Helpers
+    #---------------------------------------------------------------
+
+    # The path to the `dotfile`, which contains the persisted UUID of
+    # the VM if it exists.
+    #
+    # @return [Pathname]
+    def dotfile_path
+      root_path.join(config.vagrant.dotfile_name) rescue nil
+    end
+
+    # The path to the home directory and converted into a Pathname object.
+    #
+    # @return [Pathname]
+    def home_path
+      @_home_path ||= Pathname.new(File.expand_path(ENV["VAGRANT_HOME"] || DEFAULT_HOME))
+    end
+
+    # The path to the Vagrant tmp directory
+    #
+    # @return [Pathname]
+    def tmp_path
+      home_path.join("tmp")
+    end
+
+    # The path to the Vagrant boxes directory
+    #
+    # @return [Pathname]
+    def boxes_path
+      home_path.join("boxes")
+    end
+
+    # Path to the Vagrant logs directory
+    #
+    # @return [Pathname]
+    def log_path
+      home_path.join("logs")
+    end
+
+    # Returns the name of the resource which this environment represents.
+    # The resource is the VM name if there is a VM it represents, otherwise
+    # it defaults to "vagrant"
+    #
+    # @return [String]
+    def resource
+      result = vm.name rescue nil
+      result || "vagrant"
+    end
+
+    # Returns the collection of boxes for the environment.
+    #
+    # @return [BoxCollection]
+    def boxes
+      return parent.boxes if parent
+      @_boxes ||= BoxCollection.new(self)
+    end
+
+    # Returns the box that this environment represents.
+    #
+    # @return [Box]
+    def box
+      boxes.find(config.vm.box)
+    end
+
+    # Returns the VMs associated with this environment.
+    #
+    # @return [Hash<Symbol,VM>]
+    def vms
+      return parent.vms if parent
+      load! if !loaded?
+      @vms ||= load_vms!
+    end
+
+    # Returns the VMs associated with this environment, in the order
+    # that they were defined.
+    #
+    # @return [Array<VM>]
+    def vms_ordered
+      @vms_enum ||= config.vm.defined_vm_keys.map { |name| @vms[name] }
+    end
+
+    # Returns the primary VM associated with this environment. This
+    # method is only applicable for multi-VM environments. This can
+    # potentially be nil if no primary VM is specified.
+    #
+    # @return [VM]
+    def primary_vm
+      return vms.values.first if !multivm?
+      return parent.primary_vm if parent
+
+      config.vm.defined_vms.each do |name, subvm|
+        return vms[name] if subvm.options[:primary]
+      end
+
+      nil
+    end
+
+    # Returns a boolean whether this environment represents a multi-VM
+    # environment or not. This will work even when called on child
+    # environments.
+    #
+    # @return [Bool]
+    def multivm?
+      if parent
+        parent.multivm?
+      else
+        vms.length > 1 || vms.keys.first != DEFAULT_VM
+      end
+    end
+
+    # Makes a call to the CLI with the given arguments as if they
+    # came from the real command line (sometimes they do!). An example:
+    #
+    #     env.cli("package", "--vagrantfile", "Vagrantfile")
+    #
+    def cli(*args)
+      CLI.start(args.flatten, :env => self)
+    end
+
+    # Returns the {UI} for the environment, which is responsible
+    # for talking with the outside world.
+    #
+    # @return [UI]
+    def ui
+      @ui ||= if parent
+        result = parent.ui.clone
+        result.env = self
+        result
+      else
+        UI.new(self)
+      end
+    end
+
+    # Returns the host object associated with this environment.
+    #
+    # @return [Hosts::Base]
+    def host
+      @host ||= Hosts::Base.load(self, config.vagrant.host)
+    end
+
+    # Returns the {Action} class for this environment which allows actions
+    # to be executed (middleware chains) in the context of this environment.
+    #
+    # @return [Action]
+    def actions
+      @actions ||= Action.new(self)
+    end
+
+    # Loads on initial access and reads data from the global data store.
+    # The global data store is global to Vagrant everywhere (in every environment),
+    # so it can be used to store system-wide information. Note that "system-wide"
+    # typically means "for this user" since the location of the global data
+    # store is in the home directory.
+    #
+    # @return [DataStore]
+    def global_data
+      return parent.global_data if parent
+      @global_data ||= DataStore.new(File.expand_path("global_data.json", home_path))
+    end
+
+    # Loads (on initial access) and reads data from the local data
+    # store. This file is always at the root path as the file "~/.vagrant"
+    # and contains a JSON dump of a hash. See {DataStore} for more
+    # information.
+    #
+    # @return [DataStore]
+    def local_data
+      return parent.local_data if parent
+      @local_data ||= DataStore.new(dotfile_path)
+    end
+
+    # Accesses the logger for Vagrant. This logger is a _detailed_
+    # logger which should be used to log internals only. For outward
+    # facing information, use {#ui}.
+    #
+    # @return [Logger]
+    def logger
+      return parent.logger if parent
+      @logger ||= Util::ResourceLogger.new(resource, self)
+    end
+
+    # The root path is the path where the top-most (loaded last)
+    # Vagrantfile resides. It can be considered the project root for
+    # this environment.
+    #
+    # @return [String]
+    def root_path
+      return @root_path if defined?(@root_path)
+
+      root_finder = lambda do |path|
+        return path if File.exist?(File.join(path.to_s, ROOTFILE_NAME))
+        return nil if path.root? || !File.exist?(path)
+        root_finder.call(path.parent)
+      end
+
+      @root_path = root_finder.call(cwd)
+    end
+
+    #---------------------------------------------------------------
+    # Config Methods
+    #---------------------------------------------------------------
+
+    # The configuration object represented by this environment. This
+    # will trigger the environment to load if it hasn't loaded yet (see
+    # {#load!}).
+    #
+    # @return [Config::Top]
+    def config
+      load! if !loaded?
+      @config
+    end
+
+    #---------------------------------------------------------------
+    # Load Methods
+    #---------------------------------------------------------------
+
+    # Returns a boolean representing if the environment has been
+    # loaded or not.
+    #
+    # @return [Bool]
+    def loaded?
+      !!@loaded
+    end
+
+    # Loads this entire environment, setting up the instance variables
+    # such as `vm`, `config`, etc. on this environment. The order this
+    # method calls its other methods is very particular.
+    def load!
+      if !loaded?
+        @loaded = true
+        self.class.check_virtualbox!
+        load_config!
+      end
+
+      self
+    end
+
+    # Reloads the configuration of this environment.
+    def reload_config!
+      @config = nil
+      @config_loader = nil
+      load_config!
+      self
+    end
+
+    # Loads this environment's configuration and stores it in the {#config}
+    # variable. The configuration loaded by this method is specified to
+    # this environment, meaning that it will use the given root directory
+    # to load the Vagrantfile into that context.
+    def load_config!
+      first_run = @config.nil?
+
+      # First load the initial, non config-dependent Vagrantfiles
+      @config_loader ||= Config.new(parent ? parent.config_loader : nil)
+      @config_loader.load_order = [:default, :box, :home, :root, :sub_vm]
+      @config_loader.set(:default, File.expand_path("config/default.rb", Vagrant.source_root))
+      @config_loader.set(:box, File.join(box.directory, ROOTFILE_NAME)) if !first_run && vm && box
+      @config_loader.set(:home, File.join(home_path, ROOTFILE_NAME)) if !first_run && home_path
+      @config_loader.set(:root, File.join(root_path, ROOTFILE_NAME)) if root_path
+
+      # If this environment is representing a sub-VM, then we push that
+      # proc on as the last configuration.
+      if vm
+        subvm = parent.config.vm.defined_vms[vm.name]
+        @config_loader.set(:sub_vm, subvm.proc_stack) if subvm
+      end
+
+      # Execute the configuration stack and store the result as the final
+      # value in the config ivar.
+      @config = @config_loader.load(self)
+
+      # (re)load the logger
+      @logger = nil
+
+      if first_run
+        # After the first run we want to load the configuration again since
+        # it can change due to box Vagrantfiles and home directory Vagrantfiles
+        load_home_directory!
+        load_config!
+      end
+    end
+
+    # Loads the home directory path and creates the necessary subdirectories
+    # within the home directory if they're not already created.
+    def load_home_directory!
+      # Setup the array of necessary home directories
+      dirs = [home_path]
+      dirs += HOME_SUBDIRS.collect { |subdir| home_path.join(subdir) }
+
+      # Go through each required directory, creating it if it doesn't exist
+      dirs.each do |dir|
+        next if File.directory?(dir)
+
+        ui.info I18n.t("vagrant.general.creating_home_dir", :directory => dir)
+        FileUtils.mkdir_p(dir)
+      end
+    end
+
+    # Loads the persisted VM (if it exists) for this environment.
+    def load_vms!
+      result = {}
+
+      # Load the VM UUIDs from the local data store
+      (local_data[:active] || {}).each do |name, uuid|
+        result[name.to_sym] = Vagrant::VM.find(uuid, self, name.to_sym)
+      end
+
+      # For any VMs which aren't created, create a blank VM instance for
+      # them
+      all_keys = config.vm.defined_vm_keys
+      all_keys = [DEFAULT_VM] if all_keys.empty?
+      all_keys.each do |name|
+        result[name] = Vagrant::VM.new(:name => name, :env => self) if !result.has_key?(name)
+      end
+
+      result
+    end
+  end
+end