webify 0.1.1 → 0.2.0

data/lib/webify.rb

@@ -1,47 +1,10 @@
-require 'webrick'
-require 'optparse'
+require 'webify/version'
+require 'webify/runner'
 
-class Webify
-  include WEBrick
-  
-  DEFAULT_PORT = 3000
-
-  def self.start(options={})
-    dir  = File.expand_path(options[:dir] || Dir::pwd)
-    port = options[:port] || DEFAULT_PORT
-    
-    msg = "Serving:  #{dir}\nURL:      http://#{Socket.gethostname}:#{port}"
-
-    puts "="*80
-    puts "#{msg}\n"
-    puts ("="*80)+"\n\n"
-    
-    server = HTTPServer.new(
-      :Port            => port,
-      :DocumentRoot    => dir
-    )
-    
-    trap('INT') { server.shutdown }
-    server.start
-  end
-end
-
-options = {
-  :port => Webify::DEFAULT_PORT,
-  :dir  => Dir.pwd
-}
-
-options_parser = OptionParser.new do |opts|
-  opts.banner = "Webify -- serve any directory to your browser\n"+
-                "Usage: webify [options]"
-
-  opts.separator ""
-  opts.separator "Server options:"
-
-  opts.on('-p', '--port PORT', "Use PORT (default: #{options[:port]})") {|val| options[:port] = val }
-  opts.on('-d', '--dir DIRECTORY', "Document root DIRECTORY (default: #{options[:dir]})") {|val| options[:dir] = val }
-  opts.on('-h', '--help', 'Display this help message') { puts opts; exit! }
+module Webify
+  DEFAULT_BACKEND = 'Webrick'
+  DEFAULT_PORT    = 3000
+  DEFAULT_DIR     = Dir::pwd
 end
-options_parser.parse!(ARGV)
 
-Webify.start(options)
+Webify::Runner.start!

data/lib/webify/backend/mongrel.rb

@@ -0,0 +1,26 @@
+begin
+  require 'mongrel'
+rescue LoadError => ex
+  $stderr.puts "Oops.. can't load Mongrel, try: gem install mongrel"
+  exit!
+end
+
+module Webify
+  module Backend
+    
+    class Mongrel
+      def self.start!(options={})
+        dir  = File.expand_path(options[:dir] || Webify::DEFAULT_DIR)
+        port = options[:port] || Webify::DEFAULT_PORT
+        throttle = options[:throttle] || 0
+        
+        server = ::Mongrel::HttpServer.new("0.0.0.0", port, 950, throttle.to_i, 60)
+        server.register("/", ::Mongrel::DirHandler.new(dir))
+        
+        trap('INT') { server.stop }
+        server.run.join
+      end
+    end
+    
+  end
+end

data/lib/webify/backend/webrick.rb

@@ -0,0 +1,24 @@
+require 'webrick'
+
+module Webify
+  module Backend
+    
+    class Webrick
+      include ::WEBrick
+
+      def self.start!(options={})
+        dir  = File.expand_path(options[:dir] || Webify::DEFAULT_DIR)
+        port = options[:port] || Webify::DEFAULT_PORT
+
+        server = HTTPServer.new(
+          :Port            => port,
+          :DocumentRoot    => dir
+        )
+
+        trap('INT') { server.shutdown }
+        server.start
+      end
+    end
+    
+  end
+end

data/lib/webify/runner.rb

@@ -0,0 +1,52 @@
+require 'optparse'
+
+module Webify
+  class Runner
+    
+    def self.start!
+      options = {
+        :backend  => Webify::DEFAULT_BACKEND,
+        :port     => Webify::DEFAULT_PORT,
+        :dir      => Webify::DEFAULT_DIR,
+        :throttle => 0
+      }
+
+      options_parser = OptionParser.new do |opts|
+        opts.banner = "Webify -- serve any directory to your browser\n"+
+                      "Usage: webify [options]"
+
+        opts.separator ""
+        opts.separator "Server options:"
+
+        opts.on('-m', '--mongrel', "Use Mongrel as the backend (default: #{options[:backend]})") { options[:backend] = 'Mongrel' }
+        opts.on('-d', '--dir DIRECTORY', "Document root DIRECTORY (default: #{options[:dir]})") {|val| options[:dir] = val }
+        opts.on('-p', '--port PORT', "Use PORT (default: #{options[:port]})") {|val| options[:port] = val }
+        opts.on('-h', '--help', 'Display this help message') { puts opts; exit! }
+        
+        opts.separator ""
+        opts.separator "Mongrel backend options:"
+        opts.on('-t', '--throttle TIME', "Time to pause (in hundredths of a second) between accepting clients. (default: #{options[:throttle]})") {|val| options[:throttle] = val }
+      end
+      options_parser.parse!(ARGV)
+      
+      # Load the backend
+      begin
+        require "webify/backend/#{options[:backend].downcase}"
+      rescue LoadError => ex
+        $stderr.puts "Error! Cannot load backend #{options[:backend]}"
+        exit
+      end
+      
+      backend_klass = instance_eval("Webify::Backend::" + options[:backend])
+      
+      # Booting up the server
+      msg = "Serving:  #{options[:dir]}\nURL:      http://#{Socket.gethostname}:#{options[:port]}"
+      puts "="*80
+      puts "#{msg}\n"
+      puts ("="*80)+"\n\n"
+      
+      backend_klass.start!(options)
+    end
+    
+  end
+end

data/lib/webify/version.rb

@@ -0,0 +1,3 @@
+module Webify
+  VERSION = '0.2.0'
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: webify
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -22,6 +22,10 @@ extensions: []
 extra_rdoc_files: []
 files:
 - README.md
+- lib/webify/backend/mongrel.rb
+- lib/webify/backend/webrick.rb
+- lib/webify/runner.rb
+- lib/webify/version.rb
 - lib/webify.rb
 - bin/webify
 homepage: http://github.com/nulayer/webify