vagrant-hosts-provisioner 1.0.1 → 2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b5ec1e1905eebb2cba20f837e02d69768aae2be1
-  data.tar.gz: f94c9dfb5897c79b6abee8d059df11f4f12692fa
+  metadata.gz: b9682f369371d62fb9d52045d0b5349578adb9da
+  data.tar.gz: acabdb3efdbc7691a848fee0bcca1ce86c235adc
 SHA512:
-  metadata.gz: d8da804b098709c4b7d0b09e195dc164835593477e5964353a8bb07f13673156a0c7dfa3952039999ea9800fa61f1dd00f41db0600141f03f8a9a1ad08d70a64
-  data.tar.gz: c84600221267b52ecb30c12ff3bc0133ab29b74dba35189936f7caaaf7b6c572a6b0c233b22965b9e9e9fceede35f0022834e27ec704aa034243cc253930e9db
+  metadata.gz: 62771fda68d297ee2b465f9ae7beaa93a47c15a16f12dd0a504cfcdc23292cc336563d00e60d3314f04acc33537afd27c80cdf96e94cac816370c3e816b3b9b5
+  data.tar.gz: 61875020970a84ba8facc499bc5422b501b01e0129b6daa447776e2b765cbc7344623984206591101231835ae66270e5c2e2c9ba9b116bca37619d515d483dcb

data/README.md

@@ -1,7 +1,6 @@
 # VagrantPlugins::HostsProvisioner
 
-A Vagrant provisioner for managing the /etc/hosts file of the host and guest machines. This plugin is currently being used by [#Dashboard VM](https://github.com/mdkholy/Dashboard)..
-
+A Vagrant provisioner for managing the /etc/hosts file of the host and guest machines.
 ## Installation
 
 Install into vagrant's isolated RubyGems instance using:
@@ -10,7 +9,22 @@ Install into vagrant's isolated RubyGems instance using:
 
 ## Usage
 
-TODO: Write usage instructions here
+Example configuration:
+
+```ruby
+config.vm.provision :hostsupdate, run: 'always' do |host|
+	host.hostname = 'demo-hostname'
+	host.manage_guest = true
+	host.manage_host = true
+	host.aliases = [
+		'hostname-aliase1',
+		'hostname-aliase2'
+	]
+	host.files = [
+		'config/hosts.json'
+	]
+end
+```
 
 ## Contributing
 

data/lib/vagrant-hosts-provisioner.rb

@@ -5,8 +5,6 @@ require "vagrant-hosts-provisioner/version"
 module VagrantPlugins
   module HostsProvisioner
     # This returns the path to the source of this plugin.
-    #
-    # @return [Pathname]
     def self.source_root
       @source_root ||= Pathname.new(File.expand_path('../../', __FILE__))
     end

data/lib/vagrant-hosts-provisioner/action.rb

@@ -0,0 +1,25 @@
+require_relative 'action/add'
+require_relative 'action/remove'
+
+module VagrantPlugins
+  module HostsProvisioner
+    module Action
+      include Vagrant::Action::Builtin
+
+      def self.add
+        Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Add
+        end
+      end
+
+      def self.remove
+        Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Remove
+        end
+      end
+
+    end
+  end
+end

data/lib/vagrant-hosts-provisioner/action/add.rb

@@ -0,0 +1,24 @@
+module VagrantPlugins
+  module HostsProvisioner
+    module Action
+      class Add
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @config = @machine.env.vagrantfile.config
+        end
+
+        def call(env)
+          @config.vm.provisioners.each do |provisioner|
+            if provisioner.name == :hostsupdate
+              Hosts.new(@machine, provisioner.config).add
+            end
+          end
+          @app.call(env)
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-hosts-provisioner/action/remove.rb

@@ -0,0 +1,24 @@
+module VagrantPlugins
+  module HostsProvisioner
+    module Action
+      class Remove
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @config = @machine.env.vagrantfile.config
+        end
+
+        def call(env)
+          @config.vm.provisioners.each do |provisioner|
+            if provisioner.name == :hostsupdate
+              Hosts.new(@machine, provisioner.config).remove
+            end
+          end
+          @app.call(env)
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-hosts-provisioner/config.rb

@@ -1,6 +1,7 @@
 module VagrantPlugins
   module HostsProvisioner
     class Config < Vagrant.plugin("2", :config)
+      attr_accessor :id
       attr_accessor :hostname
       attr_accessor :manage_guest
       attr_accessor :manage_host
@@ -8,6 +9,7 @@ module VagrantPlugins
       attr_accessor :files
 
       def initialize
+        @id                 = UNSET_VALUE
         @hostname           = UNSET_VALUE
         @manage_guest       = UNSET_VALUE
         @manage_host        = UNSET_VALUE
@@ -16,6 +18,7 @@ module VagrantPlugins
       end
 
       def finalize!
+        @id                 = 0 if @id == UNSET_VALUE
         @hostname           = nil if @hostname == UNSET_VALUE
         @manage_guest       = false if @manage_guest == UNSET_VALUE
         @manage_host        = false if @manage_host == UNSET_VALUE
@@ -29,16 +32,18 @@ module VagrantPlugins
       def validate(machine)
         errors = []
 
+        errors << validate_number('id', id)
+        errors << validate_bool_or_string('hostname', hostname)
         errors << validate_bool('manage_guest', manage_guest)
         errors << validate_bool('manage_host', manage_host)
-        errors << validate_array_or_string('aliases', aliases)
         errors << validate_array_or_string('files', files)
+        errors << validate_array_or_string('aliases', aliases)
         errors.compact!
 
         { "HostsProvisioner configuration" => errors }
       end
 
-      # Checks if a option is boolean
+      # Checks if a option is Boolean
       def validate_bool(key, value)
         if ![TrueClass, FalseClass].include?(value.class) && value != UNSET_VALUE
           I18n.t('vagrant_hostsprovisioner.error.invalid_bool', {
@@ -50,9 +55,21 @@ module VagrantPlugins
         end
       end
 
-      # Checks if a option is an Array
+      # Checks if a option is a Number
+      def validate_number(key, value)
+        if !value.kind_of?(Fixnum) && !value.kind_of?(NilClass)
+          I18n.t('vagrant_hostsprovisioner.error.not_a_number', {
+            :config_key    => key,
+            :invalid_class => value.class.to_s
+          })
+        else
+          nil
+        end
+      end
+
+      # Checks if a option is an Array or String
       def validate_array_or_string(key, value)
-        if !aliases.kind_of?(Array) && !aliases.kind_of?(String)
+        if !value.kind_of?(Array) && !value.kind_of?(String) && !value.kind_of?(NilClass)
           I18n.t('vagrant_hostsprovisioner.error.not_an_array_or_string', {
             :config_key    => key,
             :invalid_class => value.class.to_s
@@ -62,6 +79,30 @@ module VagrantPlugins
         end
       end
 
+      # Checks if a option is a String or Boolean
+      def validate_bool_or_string(key, value)
+        if ![TrueClass, FalseClass].include?(value.class) && !value.kind_of?(String) && !value.kind_of?(NilClass)
+          I18n.t('vagrant_hostsprovisioner.error.not_a_bool_or_string', {
+            :config_key    => key,
+            :invalid_class => value.class.to_s
+          })
+        else
+          nil
+        end
+      end
+
+      # Checks if a option is a String
+      def validate_string(key, value)
+        if !value.kind_of?(String) && !value.kind_of?(NilClass)
+          I18n.t('vagrant_hostsprovisioner.error.not_a_string', {
+            :config_key    => key,
+            :invalid_class => value.class.to_s
+          })
+        else
+          nil
+        end
+      end
+
     end
   end
 end

data/lib/vagrant-hosts-provisioner/hosts.rb

@@ -0,0 +1,264 @@
+require 'tempfile'
+
+module VagrantPlugins
+  module HostsProvisioner
+    class Hosts
+
+      def initialize(machine, config)
+        @machine = machine
+        @config = config
+      end
+
+      def add
+        # Update the guest machine if manage_guest is enabled
+        if @config.manage_guest
+          update_guest
+        end
+
+        # Update the host machine if manage_host is enabled
+        if @config.manage_host
+          update_host(false)
+        end
+      end
+
+      def remove
+        if @config.manage_host
+          update_host(true)
+        end
+      end
+
+      def update_guest
+        return unless @machine.communicate.ready?
+
+        handle_comm(:stdout, I18n.t("vagrant_hostsprovisioner.provisioner.update_guest"))
+
+        if (@machine.communicate.test("uname -s | grep SunOS"))
+          realhostfile = '/etc/inet/hosts'
+          move_cmd = 'mv'
+        elsif (@machine.communicate.test("test -d $Env:SystemRoot"))
+          windir = ""
+          @machine.communicate.execute("echo %SYSTEMROOT%", {:shell => :cmd}) do |type, contents|
+            windir << contents.gsub("\r\n", '') if type == :stdout
+          end
+          realhostfile = "#{windir}\\System32\\drivers\\etc\\hosts"
+          move_cmd = 'mv -force'
+        else
+          realhostfile = '/etc/hosts'
+          move_cmd = 'mv -f'
+        end
+
+        # download and modify file with Vagrant-managed entries
+        file = @machine.env.tmp_path.join("hosts.#{@machine.name}")
+        @machine.communicate.download(realhostfile, file)
+        if update_file(file, false, false)
+          # upload modified file and remove temporary file
+          @machine.communicate.upload(file, '/tmp/hosts')
+          @machine.communicate.sudo("#{move_cmd} /tmp/hosts #{realhostfile}")
+          handle_comm(:stdout, I18n.t("vagrant_hostsprovisioner.provisioner.hosts_file_updated", {:file => realhostfile}))
+        end
+
+        begin
+          FileUtils.rm(file)
+         rescue Exception => e
+        end
+      end
+
+      def update_host(clean)
+        # copy and modify hosts file on host with Vagrant-managed entries
+        file = @machine.env.tmp_path.join('hosts.local')
+
+        if clean == true
+          handle_comm(:stdout, I18n.t("vagrant_hostsprovisioner.provisioner.clean_host"))
+        else
+          handle_comm(:stdout, I18n.t("vagrant_hostsprovisioner.provisioner.update_host"))
+        end
+
+        if WindowsSupport.windows?
+          # lazily include windows Module
+          class << self
+            include WindowsSupport unless include? WindowsSupport
+          end
+
+          hosts_location = "#{ENV['WINDIR']}\\System32\\drivers\\etc\\hosts"
+          copy_proc = Proc.new { windows_copy_file(file, hosts_location) }
+        else
+          hosts_location = '/etc/hosts'
+          copy_proc = Proc.new { `sudo cp #{file} #{hosts_location}` }
+        end
+
+        FileUtils.cp(hosts_location, file)
+        if update_file(file, true, clean)
+          copy_proc.call
+          handle_comm(:stdout, I18n.t("vagrant_hostsprovisioner.provisioner.hosts_file_updated", {:file => hosts_location}))
+        end
+      end
+
+      def update_file(file, include_id, clean)
+        file = Pathname.new(file)
+        old_file_content = file.read
+        new_file_content = update_content(old_file_content, include_id, clean)
+        file.open('w') { |io| io.write(new_file_content) }
+        old_file_content != new_file_content
+      end
+
+      def update_content(file_content, include_id, clean)
+        id = include_id ? " id: #{read_or_create_id}" : ""
+        header = "## vagrant-hosts-provisioner-start#{id}\n"
+        footer = "## vagrant-hosts-provisioner-end\n"
+        body = clean ? "" : get_hosts_file_entry
+        get_new_content(header, footer, body, file_content)
+      end
+
+      def get_hosts_file_entry
+        # Get the vm ip address
+        ip = get_ip_address
+
+        # Return empy string if we don't have an ip address
+        if ip === nil
+          handle_comm(:stderr, I18n.t("vagrant_hostsprovisioner.error.no_vm_ip"))
+          return ''
+        end
+
+        hosts = []
+
+        # Add the machine hostname
+        unless @config.hostname === false
+          hosts.push(@config.hostname || @machine.config.vm.hostname || @machine.name)
+        end
+
+        # Add the defined aliases
+        hosts.concat(@config.aliases)
+
+        # Add the contents of the defined hosts files
+        if @config.files.count > 0
+          hosts.concat(get_files_data)
+        end
+
+        # Remove duplicates
+        hosts = hosts.uniq
+
+        # Limit the number of hosts per line to 8
+        lines = []
+        hosts.each_slice(8) do |chnk|
+          lines.push("#{ip}\t" + chnk.join(' ').strip)
+        end
+
+        # Join lines
+        hosts = lines.join("\n").strip
+
+        "#{hosts}\n"
+      end
+
+      def get_files_data
+        require 'json'
+        data = []
+        @config.files.each do |file|
+          if file.kind_of?(String) && file != ""
+            file_path = File.join(@machine.env.root_path, file)
+            if File.exist?(file_path)
+              file_data = JSON.parse(File.read(file_path))
+              data.concat([ file_data ].flatten)
+            else
+              handle_comm(:stderr, I18n.t("vagrant_hostsprovisioner.error.file_not_found", {:file => file.to_s}))
+            end
+          end
+        end
+        data.collect(&:strip)
+      end
+
+      def get_ip_address
+        ip = nil
+        @machine.config.vm.networks.each do |network|
+          key, options = network[0], network[1]
+          ip = options[:ip] if key == :private_network
+          break if ip
+        end
+        # If no ip is defined in private_network then use the ssh host ip instead
+        ip || (@machine.ssh_info ? @machine.ssh_info[:host] : nil)
+      end
+
+      def get_new_content(header, footer, body, old_content)
+        if body.empty?
+          block = "\n"
+        else
+          block = "\n\n" + header + body + footer + "\n"
+        end
+        # Pattern for finding existing block
+        header_pattern = Regexp.quote(header)
+        footer_pattern = Regexp.quote(footer)
+        pattern = Regexp.new("\n*#{header_pattern}.*?#{footer_pattern}\n*", Regexp::MULTILINE)
+        # Replace existing block or append
+        old_content.match(pattern) ? old_content.sub(pattern, block) : old_content.rstrip + block
+      end
+
+      def read_or_create_id
+        file = Pathname.new("#{@machine.env.local_data_path}/hostsprovisioner/#{@machine.name}")
+        if (file.file?)
+          id = file.read.strip
+        else
+          id = SecureRandom.uuid
+          file.dirname.mkpath
+          file.open('w') { |f| f.write(id) }
+        end
+        id + "-" + @config.id.to_s
+      end
+
+      ## Windows support for copying files, requesting elevated privileges if necessary
+      module WindowsSupport
+        require 'rbconfig'
+
+        def self.windows?
+          RbConfig::CONFIG['host_os'] =~ /mswin|mingw|cygwin/
+        end
+
+        require 'win32ole' if windows?
+
+        def windows_copy_file(source, dest)
+          begin
+            # First, try Ruby copy
+            FileUtils.cp(source, dest)
+          rescue Errno::EACCES
+            # Access denied, try with elevated privileges
+            windows_copy_file_elevated(source, dest)
+          end
+        end
+
+        private
+
+        def windows_copy_file_elevated(source, dest)
+          # copy command only supports backslashes as separators
+          source, dest = [source, dest].map { |s| s.to_s.gsub(/\//, '\\') }
+
+          # run 'cmd /C copy ...' with elevated privilege, minimized
+          copy_cmd = "copy \"#{source}\" \"#{dest}\""
+          WIN32OLE.new('Shell.Application').ShellExecute('cmd', "/C #{copy_cmd}", nil, 'runas', 7)
+
+          # Unfortunately, ShellExecute does not give us a status code,
+          # and it is non-blocking so we can't reliably compare the file contents
+          # to see if they were copied.
+          #
+          # If the user rejects the UAC prompt, vagrant will silently continue
+          # without updating the hostsfile.
+        end
+      end
+
+      # This handles outputting the communication data back to the UI
+      def handle_comm(type, data)
+        if [:stderr, :stdout].include?(type)
+          # Output the data with the proper color based on the stream.
+          color = type == :stdout ? :green : :red
+
+          # Clear out the newline since we add one
+          data = data.chomp
+          return if data.empty?
+
+          options = {}
+          options[:color] = color
+
+          @machine.ui.info(data.chomp, options)
+        end
+      end
+
+    end
+  end
+end

data/lib/vagrant-hosts-provisioner/plugin.rb

@@ -15,14 +15,35 @@ module VagrantPlugins
       DESC
 
       config(:hostsupdate, :provisioner) do
-        require File.expand_path("../config", __FILE__)
+        require_relative 'config'
         Config
       end
 
       provisioner(:hostsupdate) do
-        require File.expand_path("../provisioner", __FILE__)
+        require_relative 'provisioner'
         Provisioner
       end
+
+      action_hook(:hostsupdate, :machine_action_resume) do |hook|
+        require_relative 'action'
+        hook.append(Action.add)
+      end
+
+      action_hook(:hostsupdate, :machine_action_suspend) do |hook|
+        require_relative 'action'
+        hook.prepend(Action.remove)
+      end
+
+      action_hook(:hostsupdate, :machine_action_halt) do |hook|
+        require_relative 'action'
+        hook.prepend(Action.remove)
+      end
+
+      action_hook(:hostsupdate, :machine_action_destroy) do |hook|
+        require_relative 'action'
+        hook.prepend(Action.remove)
+      end
+
     end
   end
 end

data/lib/vagrant-hosts-provisioner/provisioner.rb

@@ -1,239 +1,16 @@
-require 'tempfile'
+require File.expand_path("../hosts", __FILE__)
 
 module VagrantPlugins
   module HostsProvisioner
     class Provisioner < Vagrant.plugin('2', :provisioner)
 
       def initialize(machine, config)
+        @hosts = Hosts.new(machine, config)
         super
       end
 
       def provision
-        # Update the guest machine if manage_guest is enabled
-        if @config.manage_guest
-          update_guest
-        end
-        
-        # Update the host machine if manage_host is enabled
-        if @config.manage_host
-          update_host
-        end
-      end
-
-      def update_guest
-        return unless @machine.communicate.ready?
-
-        handle_comm(:stdout, I18n.t("vagrant_hostsprovisioner.provisioner.update_guest"))
-
-        if (@machine.communicate.test("uname -s | grep SunOS"))
-          realhostfile = '/etc/inet/hosts'
-          move_cmd = 'mv'
-        elsif (@machine.communicate.test("test -d $Env:SystemRoot"))
-          windir = ""
-          @machine.communicate.execute("echo %SYSTEMROOT%", {:shell => :cmd}) do |type, contents|
-            windir << contents.gsub("\r\n", '') if type == :stdout
-          end
-          realhostfile = "#{windir}\\System32\\drivers\\etc\\hosts"
-          move_cmd = 'mv -force'
-        else 
-          realhostfile = '/etc/hosts'
-          move_cmd = 'mv -f'
-        end
-
-        # download and modify file with Vagrant-managed entries
-        file = @machine.env.tmp_path.join("hosts.#{@machine.name}")
-        @machine.communicate.download(realhostfile, file)
-        if update_file(file, false)
-          # upload modified file and remove temporary file
-          @machine.communicate.upload(file, '/tmp/hosts')
-          @machine.communicate.sudo("#{move_cmd} /tmp/hosts #{realhostfile}")
-          handle_comm(:stdout, I18n.t("vagrant_hostsprovisioner.provisioner.hosts_file_updated", {:file => realhostfile}))
-        end
-
-        begin
-          FileUtils.rm(file) 
-        rescue Exception => e
-        end
-      end
-
-      def update_host
-        # copy and modify hosts file on host with Vagrant-managed entries
-        file = @machine.env.tmp_path.join('hosts.local')
-
-        handle_comm(:stdout, I18n.t("vagrant_hostsprovisioner.provisioner.update_host"))
-
-        if WindowsSupport.windows?
-          # lazily include windows Module
-          class << self
-            include WindowsSupport unless include? WindowsSupport
-          end
-
-          hosts_location = "#{ENV['WINDIR']}\\System32\\drivers\\etc\\hosts"
-          copy_proc = Proc.new { windows_copy_file(file, hosts_location) }
-        else
-          hosts_location = '/etc/hosts'
-          copy_proc = Proc.new { `sudo cp #{file} #{hosts_location}` }
-        end
-
-        FileUtils.cp(hosts_location, file)
-        if update_file(file, true)
-          copy_proc.call
-          handle_comm(:stdout, I18n.t("vagrant_hostsprovisioner.provisioner.hosts_file_updated", {:file => hosts_location}))
-        end
-      end
-
-      def update_file(file, include_id)
-        file = Pathname.new(file)
-        old_file_content = file.read
-        new_file_content = update_content(old_file_content, include_id)
-        file.open('w') { |io| io.write(new_file_content) }
-        old_file_content != new_file_content
-      end
-
-      def update_content(file_content, include_id)
-        id = include_id ? " id: #{read_or_create_id}" : ""
-        header = "## vagrant-hosts-provisioner-start#{id}\n"
-        footer = "## vagrant-hosts-provisioner-end\n"
-        body = get_hosts_file_entry
-        get_new_content(header, footer, body, file_content) 
-      end
-
-      def get_hosts_file_entry
-        # Get the vm ip address
-        ip = get_ip_address
-
-        # Return empy string if we don't have an ip address
-        if ip === nil
-          handle_comm(:stderr, I18n.t("vagrant_hostsprovisioner.error.no_vm_ip"))
-          return ''
-        end
-
-        # Add the machine hostname
-        hosts = []
-        unless @config.hostname === false
-          hosts.push(@config.hostname || @machine.config.vm.hostname || @machine.name)
-        end
-
-        # Add the defined aliases
-        hosts.concat(@config.aliases)
-
-        # Add the contents of the defined hosts files
-        if @config.files.count > 0
-          hosts.concat(get_files_data)
-        end
-
-        # Remove duplicates
-        hosts = hosts.uniq.join(' ').strip
-
-        "#{ip}\t#{hosts}\n"
-      end
-
-      def get_files_data
-        require 'json'
-        data = []
-        @config.files.each do |file|
-          file_path = File.join(@machine.env.root_path, file)
-          if File.exist?(file_path)
-            file_data = JSON.parse(File.read(file_path))
-            data.concat([ file_data ].flatten)
-          else
-            handle_comm(:stderr, I18n.t("vagrant_hostsprovisioner.error.file_not_found", {:file => file.to_s}))
-          end
-        end
-        data.collect(&:strip)
-      end
-
-      def get_ip_address
-        ip = nil
-        @machine.config.vm.networks.each do |network|
-          key, options = network[0], network[1]
-          ip = options[:ip] if key == :private_network
-          break if ip
-        end
-        # If no ip is defined in private_network then use the ssh host ip instead
-        ip || (@machine.ssh_info ? @machine.ssh_info[:host] : nil)
-      end
-
-      def get_new_content(header, footer, body, old_content)
-        if body.empty?
-          block = "\n"
-        else
-          block = "\n\n" + header + body + footer + "\n"
-        end
-        # Pattern for finding existing block
-        header_pattern = Regexp.quote(header)
-        footer_pattern = Regexp.quote(footer)
-        pattern = Regexp.new("\n*#{header_pattern}.*?#{footer_pattern}\n*", Regexp::MULTILINE)
-        # Replace existing block or append
-        old_content.match(pattern) ? old_content.sub(pattern, block) : old_content.rstrip + block
-      end
-
-      def read_or_create_id
-        file = Pathname.new("#{@machine.env.local_data_path}/hostsprovisioner/#{@machine.name}")
-        if (file.file?)
-          id = file.read.strip
-        else
-          id = SecureRandom.uuid
-          file.dirname.mkpath
-          file.open('w') { |f| f.write(id) }
-        end
-        id
-      end
-
-      ## Windows support for copying files, requesting elevated privileges if necessary
-      module WindowsSupport
-        require 'rbconfig'
-
-        def self.windows?
-          RbConfig::CONFIG['host_os'] =~ /mswin|mingw|cygwin/
-        end
-
-        require 'win32ole' if windows?
-
-        def windows_copy_file(source, dest)
-          begin
-            # First, try Ruby copy
-            FileUtils.cp(source, dest)
-          rescue Errno::EACCES
-            # Access denied, try with elevated privileges
-            windows_copy_file_elevated(source, dest)
-          end
-        end
-
-        private 
-
-        def windows_copy_file_elevated(source, dest)
-          # copy command only supports backslashes as separators
-          source, dest = [source, dest].map { |s| s.to_s.gsub(/\//, '\\') }
-          
-          # run 'cmd /C copy ...' with elevated privilege, minimized
-          copy_cmd = "copy \"#{source}\" \"#{dest}\""        
-          WIN32OLE.new('Shell.Application').ShellExecute('cmd', "/C #{copy_cmd}", nil, 'runas', 7)
-
-          # Unfortunately, ShellExecute does not give us a status code,
-          # and it is non-blocking so we can't reliably compare the file contents
-          # to see if they were copied.
-          #
-          # If the user rejects the UAC prompt, vagrant will silently continue 
-          # without updating the hostsfile.
-        end
-      end
-
-      # This handles outputting the communication data back to the UI
-      def handle_comm(type, data)
-        if [:stderr, :stdout].include?(type)
-          # Output the data with the proper color based on the stream.
-          color = type == :stdout ? :green : :red
-
-          # Clear out the newline since we add one
-          data = data.chomp
-          return if data.empty?
-
-          options = {}
-          options[:color] = color
-
-          @machine.ui.info(data.chomp, options)
-        end
+        @hosts.add
       end
 
     end

data/lib/vagrant-hosts-provisioner/version.rb

@@ -1,5 +1,5 @@
 module VagrantPlugins
   module HostsProvisioner
-    VERSION = "1.0.1"
+    VERSION = "2.0"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vagrant-hosts-provisioner
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: '2.0'
 platform: ruby
 authors:
 - Mohamed Elkholy
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-08-31 00:00:00.000000000 Z
+date: 2015-03-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -52,7 +52,11 @@ files:
 - README.md
 - Rakefile
 - lib/vagrant-hosts-provisioner.rb
+- lib/vagrant-hosts-provisioner/action.rb
+- lib/vagrant-hosts-provisioner/action/add.rb
+- lib/vagrant-hosts-provisioner/action/remove.rb
 - lib/vagrant-hosts-provisioner/config.rb
+- lib/vagrant-hosts-provisioner/hosts.rb
 - lib/vagrant-hosts-provisioner/plugin.rb
 - lib/vagrant-hosts-provisioner/provisioner.rb
 - lib/vagrant-hosts-provisioner/version.rb
@@ -84,7 +88,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: vagrant-hosts-provisioner
-rubygems_version: 2.4.1
+rubygems_version: 2.4.5
 signing_key: 
 specification_version: 4
 summary: A Vagrant provisioner for managing the /etc/hosts file of the host and guest