From 81a7557459f8ace2b24c1b78d763ee59031723c8 Mon Sep 17 00:00:00 2001
From: Thomas Reynolds <tdreyno@gmail.com>
Date: Wed, 13 Jul 2011 18:58:22 -0700
Subject: [PATCH] Use Thor for mm-server

---
 bin/mm-server          | 68 ++++++++++++++++++++++--------------------
 lib/middleman/guard.rb | 28 +++++++++--------
 2 files changed, 51 insertions(+), 45 deletions(-)

diff --git a/bin/mm-server b/bin/mm-server
index 68208736..fc58f566 100755
--- a/bin/mm-server
+++ b/bin/mm-server
@@ -1,43 +1,45 @@
 #!/usr/bin/env ruby
 
-require 'optparse'
+require "thor"
+require "thor/group"
 
 # Require Middleman
 require File.join(File.dirname(__FILE__), '..', 'lib', 'middleman')
 
-env = ENV['MM_ENV'] || ENV['RACK_ENV'] || 'development'
-options = {}
-livereload_options = {}
+module Middleman
+  class GuardServer < ::Thor::Group
+    include Thor::Actions
+    
+    class_option :environment, :aliases => "-e", :default => ENV['MM_ENV'] || ENV['RACK_ENV'] || 'development'
+    
+    class_option :port, :aliases => "-p", :default => "4567"
+    class_option :"livereload-port", :default => "35729"
+    class_option :"livereload", :default => true, :type => :boolean
+    
+    def start_guard
+      if !File.exists?("config.rb")
+        $stderr.puts "== Error: Could not find a Middleman project config, perhaps you are in the wrong folder?"
+        return
+      end
 
-# TODO: Switch to Thor
-OptionParser.new { |opts|
-  opts.banner = "Usage: mm-server [rack options]"
-  opts.separator ""
-  opts.separator "Rack options:"
-  opts.on("-p", "--port PORT", "use PORT (default: 4567)") { |port|
-    options[:port] = port
-  }
-  opts.on("-E", "--env ENVIRONMENT", "use ENVIRONMENT for defaults (default: development)") { |e|
-    env = e
-  }
-  opts.on("--livereload_port PORT", "use PORT (default: 35729)") { |port|
-    livereload_options[:port] = port
-  }
+      # If the old directories exists, use it, but issue warning
+      if File.exists?("views") || File.exists?("public")
+        $stderr.puts "== Error: The views and public folders are have been combined. Create a new 'source' folder, add the contents of views and public to it and then remove the empty views and public folders."
+        return
+      end
 
-  opts.parse! ARGV
-}
-
-ENV['RACK_ENV'] = env
-
-if !File.exists?("config.rb")
-  $stderr.puts "== Error: Could not find a Middleman project config, perhaps you are in the wrong folder?"
-  exit
+      ENV['RACK_ENV'] = options[:environment]
+      
+      livereload_options = {
+        :port => options[:"livereload-port"]
+      }
+      livereload_options = nil unless options[:"livereload"] 
+      
+      ::Middleman::Guard.start({
+        :port => options[:port],
+      }, livereload_options)
+    end
+  end
 end
 
-# If the old directories exists, use it, but issue warning
-if File.exists?("views") || File.exists?("public")
-  $stderr.puts "== Error: The views and public folders are have been combined. Create a new 'source' folder, add the contents of views and public to it and then remove the empty views and public folders."
-  exit
-end
-
-Middleman::Guard.start(options, livereload_options)
\ No newline at end of file
+Middleman::GuardServer.start
\ No newline at end of file
diff --git a/lib/middleman/guard.rb b/lib/middleman/guard.rb
index dfb59945..30f37b8c 100644
--- a/lib/middleman/guard.rb
+++ b/lib/middleman/guard.rb
@@ -8,23 +8,27 @@ module Middleman::Guard
     options.each do |k,v|
       options_hash << ", :#{k} => '#{v}'"
     end
-  
-    livereload_options_hash = ""
-    livereload.each do |k,v|
-      livereload_options_hash << ", :#{k} => '#{v}'"
-    end
     
-    ::Guard.start({
-      :guardfile_contents => %Q{
-        guard 'middleman'#{options_hash} do 
-          watch("config.rb")
-        end
-        
+    guardfile_contents = %Q{
+      guard 'middleman'#{options_hash} do 
+        watch("config.rb")
+      end
+    }
+    
+    if livereload
+      livereload_options_hash = ""
+      livereload.each do |k,v|
+        livereload_options_hash << ", :#{k} => '#{v}'"
+      end
+      
+      guardfile_contents << %Q{
         guard 'livereload'#{livereload_options_hash} do 
           watch(%r{^source/(.*)$})
         end
       }
-    })
+    end
+    
+    ::Guard.start({ :guardfile_contents => guardfile_contents })
   end
 end