From b35de4834bdb8be31f6f73332546ba46778b0547 Mon Sep 17 00:00:00 2001
From: "Rico Sta. Cruz" <rico@ambiescent.com>
Date: Sat, 16 Jun 2012 06:30:11 +0800
Subject: [PATCH] Restart the listener when reloading the server

---
 .../lib/middleman-core/preview_server.rb           | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/middleman-core/lib/middleman-core/preview_server.rb b/middleman-core/lib/middleman-core/preview_server.rb
index 7ead903c..434860b3 100644
--- a/middleman-core/lib/middleman-core/preview_server.rb
+++ b/middleman-core/lib/middleman-core/preview_server.rb
@@ -32,6 +32,8 @@ module Middleman
         )
         
         mount_instance(app)
+
+        start_file_watcher unless options[:"disable-watcher"]
         
         @initialized ||= false
         unless @initialized
@@ -39,8 +41,6 @@ module Middleman
           
           register_signal_handlers unless ::Middleman::WINDOWS
           
-          start_file_watcher unless options[:"disable-watcher"]
-
           # Save the last-used options so it may be re-used when
           # reloading later on.
           @last_options = options
@@ -83,7 +83,10 @@ module Middleman
 
           if added_and_modified.length > 0
             # See if the changed file is config.rb or lib/*.rb
-            return reload if needs_to_reload?(added_and_modified)
+            if needs_to_reload?(added_and_modified)
+              reload
+              return listener.stop
+            end
 
             # Otherwise forward to Middleman
             added_and_modified.each do |path|
@@ -93,7 +96,10 @@ module Middleman
       
           if removed.length > 0
             # See if the changed file is config.rb or lib/*.rb
-            return reload if needs_to_reload?(removed)
+            if needs_to_reload?(removed)
+              reload
+              return listener.stop
+            end
 
             # Otherwise forward to Middleman
             removed.each do |path|