rack-mongrel2 0.0.0 → 0.1.0

data/VERSION

@@ -1 +1 @@
-0.0.0
+0.1.0

data/lib/mongrel2/connection.rb

@@ -0,0 +1,35 @@
+require 'zmq'
+require 'mongrel2/request'
+require 'mongrel2/response'
+
+module Mongrel2
+  class Connection
+    CTX = ZMQ::Context.new(1)
+
+    def initialize(ident, sub, pub, block = true)
+      @ident, @sub, @pub, @block = ident, sub, pub, block
+
+      # Connect to receive requests
+      @reqs = CTX.socket(ZMQ::UPSTREAM)
+      @reqs.connect(sub)
+
+      # Connect to send responses
+      @resp = CTX.socket(ZMQ::PUB)
+      @resp.connect(pub)
+      @resp.setsockopt(ZMQ::IDENTITY, ident)
+    end
+
+    def recv
+      Request.parse(@reqs.recv)
+    end
+
+    def reply(req, body, status = 200, headers = {})
+      Response.new(@resp).send_http(req, body, status, headers)
+    end
+
+    def close
+      @reqs.close rescue nil
+      @resp.close rescue nil
+    end
+  end
+end

data/lib/mongrel2/request.rb

@@ -0,0 +1,34 @@
+module Mongrel2
+  class Request
+    attr_reader :headers, :body, :uuid, :conn_id
+
+    class << self
+      def parse(msg)
+        # UUID CONN_ID PATH SIZE:HEADERS,SIZE:BODY,
+        uuid, conn_id, path, rest = msg.split(' ', 4)
+        headers, rest = parse_netstring(rest)
+        body, _ = parse_netstring(rest)
+        headers = Yajl::Parser.parse(headers)
+        new(uuid, conn_id, path, headers, body)
+      end
+
+      def parse_netstring(ns)
+        # SIZE:HEADERS,
+
+        len, rest = ns.split(':', 2)
+        len = len.to_i
+        raise "Netstring did not end in ','" unless rest[len].chr == ','
+        [rest[0, len], rest[(len + 1)..-1]]
+      end
+    end
+
+    def initialize(uuid, conn_id, path, headers, body)
+      @uuid, @conn_id, @path, @headers, @body = uuid, conn_id, path, headers, body
+      @data = headers['METHOD'] == 'JSON' ? Yajl::Parser.parse(body) : {}
+    end
+
+    def disconnect?
+      headers['METHOD'] == 'JSON' && @data['type'] == 'disconnect'
+    end
+  end
+end

data/lib/mongrel2/response.rb

@@ -0,0 +1,66 @@
+module Mongrel2
+  class Response
+    StatusMessage = {
+      100 => 'Continue',
+      101 => 'Switching Protocols',
+      200 => 'OK',
+      201 => 'Created',
+      202 => 'Accepted',
+      203 => 'Non-Authoritative Information',
+      204 => 'No Content',
+      205 => 'Reset Content',
+      206 => 'Partial Content',
+      300 => 'Multiple Choices',
+      301 => 'Moved Permanently',
+      302 => 'Found',
+      303 => 'See Other',
+      304 => 'Not Modified',
+      305 => 'Use Proxy',
+      307 => 'Temporary Redirect',
+      400 => 'Bad Request',
+      401 => 'Unauthorized',
+      402 => 'Payment Required',
+      403 => 'Forbidden',
+      404 => 'Not Found',
+      405 => 'Method Not Allowed',
+      406 => 'Not Acceptable',
+      407 => 'Proxy Authentication Required',
+      408 => 'Request Timeout',
+      409 => 'Conflict',
+      410 => 'Gone',
+      411 => 'Length Required',
+      412 => 'Precondition Failed',
+      413 => 'Request Entity Too Large',
+      414 => 'Request-URI Too Large',
+      415 => 'Unsupported Media Type',
+      416 => 'Request Range Not Satisfiable',
+      417 => 'Expectation Failed',
+      500 => 'Internal Server Error',
+      501 => 'Not Implemented',
+      502 => 'Bad Gateway',
+      503 => 'Service Unavailable',
+      504 => 'Gateway Timeout',
+      505 => 'HTTP Version Not Supported'
+    }
+
+    def initialize(resp)
+      @resp = resp
+    end
+
+    def send_http(req, body, status, headers)
+      send_resp(req.uuid, req.conn_id, build_http_response(body, status, headers))
+    end
+
+  private
+
+    def send_resp(uuid, conn_id, data)
+      @resp.send('%s %d:%s, %s' % [uuid, conn_id.size, conn_id, data])
+    end
+
+    def build_http_response(body, status, headers)
+      headers['Content-Length'] = body.size.to_s
+      headers = headers.map{ |k, v| '%s: %s' % [k,v] }.join("\r\n")
+      "HTTP/1.1 #{status} #{StatusMessage[status.to_i]}\r\n#{headers}\r\n\r\n#{body}"
+    end
+  end
+end

data/lib/rack/handler/mongrel2.rb

@@ -0,0 +1,64 @@
+require 'mongrel2/connection'
+require 'stringio'
+
+module Rack
+  module Handler
+    class Mongrel2
+      class << self
+        def run(app, options = {})
+          options = {
+            :recv => 'tcp://127.0.0.1:9997' || ENV['RACK_MONGREL2_RECV'],
+            :send => 'tcp://127.0.0.1:9996' || ENV['RACK_MONGREL2_SEND'],
+            :uuid => ENV['RACK_MONGREL2_UUID']
+          }.merge(options)
+          raise ArgumentError.new('Must specify an :uuid or set RACK_MONGREL2_UUID') if options[:uuid].nil?
+
+          conn = ::Mongrel2::Connection.new(options[:uuid], options[:recv], options[:send])
+
+          running = true
+          trap('SIGINT') do
+            running = false
+            conn.close
+          end
+
+          while running
+            req = conn.recv
+            next if req.disconnect?
+            return unless running
+
+            script_name = ENV['RACK_RELATIVE_URL_ROOT'] || req.headers['PATTERN'].split('(', 2).first.gsub(/\/$/, '')
+            env = {
+              'rack.version' => Rack::VERSION,
+              'rack.url_scheme' => 'http', # Only HTTP for now
+              'rack.input' => StringIO.new(req.body),
+              'rack.errors' => $stderr,
+              'rack.multithread' => true,
+              'rack.multiprocess' => true,
+              'rack.run_once' => false,
+              'mongrel2.pattern' => req.headers['PATTERN'],
+              'REQUEST_METHOD' => req.headers['METHOD'],
+              'SCRIPT_NAME' => script_name,
+              'PATH_INFO' => req.headers['PATH'].gsub(script_name, ''),
+              'QUERY_STRING' => req.headers['QUERY']
+            }
+
+            env['SERVER_NAME'], env['SERVER_PORT'] = req.headers['host'].split(':', 2)
+            req.headers.each do |key, val|
+              unless key =~ /content_(type|length)/i
+                key = "HTTP_#{key.upcase}"
+              end
+              env[key] = val
+            end
+
+            status, headers, rack_response = app.call(env)
+            body = ''
+            rack_response.each { |b| body << b }
+            conn.reply(req, body, status, headers)
+          end
+        ensure
+          conn.close if conn.respond_to?(:close)
+        end
+      end
+    end
+  end
+end

data/rack-mongrel2.gemspec

@@ -0,0 +1,74 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{rack-mongrel2}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Daniel Huckstep"]
+  s.date = %q{2010-10-23}
+  s.description = %q{A Rack handler for the Mongrel2 web server, by Zed Shaw. http://mongrel2.org/}
+  s.email = %q{darkhelmet@darkhelmetlive.com}
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     ".rvmrc",
+     "Gemfile",
+     "Gemfile.lock",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "lib/mongrel2/connection.rb",
+     "lib/mongrel2/request.rb",
+     "lib/mongrel2/response.rb",
+     "lib/rack-mongrel2.rb",
+     "lib/rack/handler/mongrel2.rb",
+     "rack-mongrel2.gemspec",
+     "spec/rack-mongrel2_spec.rb",
+     "spec/spec.opts",
+     "spec/spec_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/darkhelmet/rack-mongrel2}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.7}
+  s.summary = %q{The only Mongrel2 Rack handler you'll ever need.}
+  s.test_files = [
+    "spec/rack-mongrel2_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<yajl-ruby>, ["~> 0.7.8"])
+      s.add_runtime_dependency(%q<zmq>, ["~> 2.0.9"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.4.0"])
+      s.add_development_dependency(%q<rspec>, ["~> 2.0.1"])
+      s.add_development_dependency(%q<yard>, ["~> 0.6.1"])
+    else
+      s.add_dependency(%q<yajl-ruby>, ["~> 0.7.8"])
+      s.add_dependency(%q<zmq>, ["~> 2.0.9"])
+      s.add_dependency(%q<jeweler>, ["~> 1.4.0"])
+      s.add_dependency(%q<rspec>, ["~> 2.0.1"])
+      s.add_dependency(%q<yard>, ["~> 0.6.1"])
+    end
+  else
+    s.add_dependency(%q<yajl-ruby>, ["~> 0.7.8"])
+    s.add_dependency(%q<zmq>, ["~> 2.0.9"])
+    s.add_dependency(%q<jeweler>, ["~> 1.4.0"])
+    s.add_dependency(%q<rspec>, ["~> 2.0.1"])
+    s.add_dependency(%q<yard>, ["~> 0.6.1"])
+  end
+end
+

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: rack-mongrel2
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 27
   prerelease: false
   segments: 
   - 0
+  - 1
   - 0
-  - 0
-  version: 0.0.0
+  version: 0.1.0
 platform: ruby
 authors: 
 - Daniel Huckstep
@@ -117,7 +117,12 @@ files:
 - README.rdoc
 - Rakefile
 - VERSION
+- lib/mongrel2/connection.rb
+- lib/mongrel2/request.rb
+- lib/mongrel2/response.rb
 - lib/rack-mongrel2.rb
+- lib/rack/handler/mongrel2.rb
+- rack-mongrel2.gemspec
 - spec/rack-mongrel2_spec.rb
 - spec/spec.opts
 - spec/spec_helper.rb