Add WEBrick server to dev mode (closes #23)

bin/stasis

@@ -6,7 +6,7 @@ gem "slop", "~> 2.1.0"
 require 'slop'
 
 slop = Slop.parse :help => true do
-  on :d, :development, "Development mode\t\t(auto-regenerate)"
+  on :d, :development, "Development mode\t\t(auto-regenerate)", :optional => true, :as => Integer
   on :o, :only, "Only generate specific files\t(comma-separated)", :optional => true, :as => Array
   on :p, :public, "Public directory path", :optional => true
   on :s, :server, "Server mode\t\t\t(default redis host: localhost:6379/0)", :optional => true, :default => "localhost:6379/0"

lib/stasis/dev_mode.rb

@@ -1,6 +1,9 @@
 gem "directory_watcher", "~> 1.4.1"
 require 'directory_watcher'
 
+require 'logger'
+require 'webrick'
+
 class Stasis
   class DevMode
 
@@ -31,7 +34,26 @@ def initialize(dir, options={})
       dw.add_observer { render }
       dw.start
 
-      loop { sleep 1000 }
+      if options[:development]
+        mime_types = WEBrick::HTTPUtils::DefaultMimeTypes
+        mime_types.store 'js', 'application/javascript'
+
+        server = WEBrick::HTTPServer.new(
+          :AccessLog => [ nil, nil ],
+          :DocumentRoot => @stasis.destination,
+          :Logger => WEBrick::Log.new("/dev/null"),
+          :MimeTypes => mime_types,
+          :Port => options[:development] || 3000
+        )
+
+        ['INT', 'TERM'].each do |signal|
+          trap(signal) { server.shutdown }
+        end
+
+        server.start
+      else
+        loop { sleep 1000 }
+      end
     end
 
     private