lib/vagrant-notify.rb in vagrant-notify-0.0.1 vs lib/vagrant-notify.rb in vagrant-notify-0.1.0

@@ -1,32 +1,35 @@
 require 'vagrant'
 require 'socket'
 require 'erb'
 require 'ostruct'
 
+if File.exists?(File.join(File.expand_path('../../', __FILE__), '.git'))
+  $:.unshift(File.expand_path('../../lib', __FILE__))
+end
+
 require 'vagrant-notify/middleware'
 require 'vagrant-notify/server'
+require 'vagrant-notify/vagrant_ssh_ext'
 require "vagrant-notify/version"
 
 module Vagrant
   module Notify
     class << self
-      def server_port
-        @@server_port ||= 8081
-      end
-      def server_port=(server_port)
-        @@server_port = server_port
-      end
-
       def files_path
         @file_path ||= File.expand_path(File.dirname(__FILE__) + '/../files')
       end
     end
   end
 end
 
 Vagrant.actions[:start].tap do |start|
-  start.use          Vagrant::Notify::Middleware::StartServer
-  start.insert_after Vagrant::Action::VM::Boot, Vagrant::Notify::Middleware::InstallCommand
+  start.insert_after Vagrant::Action::VM::Boot,                Vagrant::Notify::Middleware::StartServer
+  start.insert_after Vagrant::Notify::Middleware::StartServer, Vagrant::Notify::Middleware::InstallCommand
 end
-Vagrant.actions[:halt].use Vagrant::Notify::Middleware::StopServer
+Vagrant.actions[:resume].tap do |start|
+  start.use Vagrant::Notify::Middleware::StartServer
+  start.use Vagrant::Notify::Middleware::InstallCommand
+end
+Vagrant.actions[:halt].use      Vagrant::Notify::Middleware::StopServer
+Vagrant.actions[:suspend].use   Vagrant::Notify::Middleware::StopServer
 Vagrant.actions[:provision].use Vagrant::Notify::Middleware::InstallCommand