Merge pull request #9 from vStone/feature/hostmanager-as-provisioner

Use hostmanager as provisioner

README.md

@@ -57,6 +57,17 @@ Vagrant.configure("2") do |config|
 end
 ```
 
+As a last option, you can also choose hostmanager as a provisioner.
+This allows you to use the provisioning order to ensure that hostmanager
+runs before or after provisioning. The provisioner will collect hosts from
+boxes with the same provider as the running box.
+
+Use:
+
+```ruby
+config.vm.provision :hostmanager
+```
+
 Contribute
 ----------
 Contributions are welcome.

lib/vagrant-hostmanager/hosts_file.rb

@@ -32,7 +32,7 @@ def generate(env, provider)
  host = machine.config.vm.hostname || name
  ip = get_ip_address.call(machine)
  host_aliases = machine.config.hostmanager.aliases.join("\s").chomp
- @logger.info "Adding /etc/hosts entry: #{ip} #{host} #{host_aliases}"
+ machine.env.ui.info "Adding /etc/hosts entry: #{ip} #{host} #{host_aliases}"
  file << "#{ip}\t#{host}\s#{host_aliases}\n"
  end
  end

lib/vagrant-hostmanager/plugin.rb

@@ -7,6 +7,8 @@ class Plugin < Vagrant.plugin('2')
  description <<-DESC
  This plugin manages the /etc/hosts file for guest machines. An entry is
  created for each active machine using the hostname attribute.
+
+ You can also use the hostmanager provisioner to update the hosts file.
  DESC
 
  config(:hostmanager) do
@@ -22,6 +24,11 @@ class Plugin < Vagrant.plugin('2')
  hook.append(Action::UpdateHostsFile)
  end
 
+ provisioner(:hostmanager) do
+ require_relative 'provisioner'
+ Provisioner
+ end
+
  command(:hostmanager) do
  require_relative 'command'
  Command

lib/vagrant-hostmanager/provisioner.rb

@@ -0,0 +1,12 @@
+module VagrantPlugins
+ module HostManager
+ class Provisioner < Vagrant.plugin('2', :provisioner)
+ include HostsFile
+
+ def provision
+ generate(@machine.env, @machine.box.provider.to_sym)
+ update(@machine)
+ end
+ end
+ end
+end