lib/vagrant-guests-openbsd/plugin.rb in vagrant-guests-openbsd-0.0.2 vs lib/vagrant-guests-openbsd/plugin.rb in vagrant-guests-openbsd-0.0.3

@@ -4,27 +4,34 @@
   module GuestOpenBSD
     class Plugin < Vagrant.plugin("2")
       name "OpenBSD guest"
       description "OpenBSD guest support."
 
-      config("openbsd") do
-        require_relative "config"
-        Config
-      end
+      %w{openbsd openbsd_v2}.each do |os|
+        guest(os) do
+          require_relative "guest"
+          Guest
+        end
 
-      guest("openbsd") do
-        require_relative "guest"
-        Guest
-      end
+        guest_capability(os, "change_host_name") do
+          require_relative "cap/change_host_name"
+          Cap::ChangeHostName
+        end
 
-      config("openbsd_v2") do
-        require_relative "config"
-        Config
-      end
+        guest_capability(os, "configure_networks") do
+          require_relative "cap/configure_networks"
+          Cap::ConfigureNetworks
+        end
 
-      guest("openbsd_v2") do
-        require_relative "guest"
-        Guest
+        guest_capability(os, "halt") do
+          require_relative "cap/halt"
+          Cap::Halt
+        end
+
+        guest_capability(os, "mount_nfs_folder") do
+          require_relative "cap/mount_nfs_folder"
+          Cap::MountNFSFolder
+        end
       end
     end
   end
 end