lib/vagrant_dns_server.rb in vagrant-dns-server-0.4 vs lib/vagrant_dns_server.rb in vagrant-dns-server-0.4.1

@@ -1,12 +1,23 @@
-require 'em-zeromq'
-require 'eventmachine'
-require 'vagrant_dns/consts'
-require 'vagrant_dns/resolver'
+require 'vagrant'
+require 'zmq'
+# client
 require 'moneta'
-require 'forwardable'
+require 'vagrant_dns/consts'
 require 'vagrant_dns/configuration'
-require 'rubydns'
-require 'vagrant_dns/dns'
-require 'vagrant_dns/server'
+require 'vagrant_dns/port'
+require 'vagrant_dns/middleware/update'
+require 'vagrant_dns/middleware/remove'
+require 'vagrant_dns/middleware/network_binder'
+
+#see http://tinyurl.com/bfuawuj  
+class Plugin < Vagrant.plugin("2")
+  name 'Vagrant DNS plugin'
+  
+  action_hook(:vagrant_dns_plugin) do |hook|
+    hook.after(::Vagrant::Action::Builtin::Provision, VagrantDns::Middleware::Update)
+    hook.after(VagrantPlugins::ProviderVirtualBox::Action::Boot, VagrantDns::Middleware::Update)
+  end
+
+end