rack-singleshot 0.1.0

data/lib/rack/singleshot.rb

@@ -0,0 +1,132 @@
+require 'rack'
+
+module Rack
+  module Handler
+    class SingleShot
+      CRLF = "\r\n"
+
+      def self.run(app, options = {})
+        stdin   = options.fetch(:stdin, $stdin)
+        stdout  = options.fetch(:stdout, $stdout)
+
+        stdin.binmode = true  if stdin.respond_to?(:binmode=)
+        stdout.binmode = true if stdout.respond_to?(:binmode=)
+
+        new(app, stdin, stdout).run
+      end
+
+      def initialize(app, stdin, stdout)
+        @app, @stdin, @stdout = app, stdin, stdout
+      end
+
+      def run
+        request = read_request
+
+        status, headers, body = @app.call(request)
+
+        write_response(status, headers, body)
+      ensure
+        @stdout.close
+      end
+
+      def read_request
+        buffer, extra = drain(@stdin, CRLF * 2)
+
+        heading, buffer = buffer.split(CRLF, 2)
+
+        verb, path, version = heading.split(' ')
+
+        headers = parse_headers(buffer)
+
+        if length = request_body_length(verb, headers)
+          body = StringIO.new(extra + @stdin.read(length - extra.size))
+        else
+          body = StringIO.new(extra)
+        end
+
+        env_for(verb, path, version, headers, body)
+      end
+
+      def write_response(status, headers, body)
+        @stdout.write(['HTTP/1.1', status, Rack::Utils::HTTP_STATUS_CODES[status.to_i]].join(' ') << CRLF)
+
+        headers.each do |key, values|
+          values.split("\n").each do |value|
+            @stdout.write([key, value].join(": ") << CRLF)
+          end
+        end
+
+        @stdout.write(CRLF)
+
+        body.each do |chunk|
+          @stdout.write(chunk)
+        end
+      end
+
+      def parse_headers(raw_headers)
+        raw_headers.split(CRLF).inject({}) do |h, pair|
+          key, value = pair.split(": ")
+          h.update(header_key(key) => value)
+        end
+      end
+
+      def request_body_length(verb, headers)
+        return if %w[ POST PUT ].include?(verb.upcase)
+
+        if length = headers['CONTENT_LENGTH']
+          length.to_i
+        end
+      end
+
+      def drain(socket, stop_at, chunksize = 1024)
+        buffer = ''
+
+        while(chunk = socket.readpartial(chunksize))
+          buffer << chunk
+
+          if buffer.include?(stop_at)
+            buffer, extra = buffer.split(stop_at, 2)
+
+            return buffer, extra
+          end
+        end
+      rescue EOFError
+        return buffer, ''
+      end
+
+      def env_for(verb, path, version, headers, body)
+        env = headers
+
+        scheme = ['yes', 'on', '1'].include?(env['HTTPS']) ? 'https' : 'http'
+        host   = env['SERVER_NAME'] || env['HTTP_HOST']
+
+        uri = URI.parse([scheme, '://', host, path].join)
+
+        env.update 'REQUEST_METHOD' => verb
+        env.update 'SCRIPT_NAME'    => ''
+        env.update 'PATH_INFO'      => uri.path
+        env.update 'QUERY_STRING'   => uri.query || ''
+        env.update 'SERVER_NAME'    => uri.host
+        env.update 'SERVER_PORT'    => uri.port.to_s
+
+        env.update 'rack.version'       => Rack::VERSION
+        env.update 'rack.url_scheme'    => uri.scheme
+        env.update 'rack.input'         => body
+        env.update 'rack.errors'        => $stderr
+        env.update 'rack.multithread'   => false
+        env.update 'rack.multiprocess'  => false
+        env.update 'rack.run_once'      => true
+
+        env
+      end
+
+      def header_key(key)
+        key = key.upcase.gsub('-', '_')
+
+        %w[CONTENT_TYPE CONTENT_LENGTH SERVER_NAME].include?(key) ? key : "HTTP_#{key}"
+      end
+    end
+
+    register :singleshot, SingleShot
+  end
+end

data/spec/singleshot_spec.rb

@@ -0,0 +1,56 @@
+require 'spec_helper'
+
+describe Rack::Handler::SingleShot do
+  before(:each) do
+    @stdin, @in   = IO.pipe
+    @out, @stdout = IO.pipe
+
+    @app = Rack::Lint.new(lambda {|env| [200, {'Content-Type' => 'text/plain'}, []] })
+    @server = Rack::Handler::SingleShot.new(@app, @stdin, @stdout)
+  end
+
+  it 'can handle a simple request' do
+    @in << <<-REQUEST.gsub("\n", "\r\n")
+GET / HTTP/1.1
+Server-Name: localhost
+
+REQUEST
+
+    @server.run
+
+    @out.read.should == <<-RESPONSE.gsub("\n", "\r\n")
+HTTP/1.1 200 OK
+Content-Type: text/plain
+
+RESPONSE
+  end
+
+  describe "Sinatra App" do
+
+    class App < Sinatra::Base
+      get '/' do
+        'response body'
+      end
+    end
+
+    before(:each) do
+      @stdin, @in   = IO.pipe
+      @out, @stdout = IO.pipe
+
+      @app = Rack::Lint.new(App.new)
+      @server = Rack::Handler::SingleShot.new(@app, @stdin, @stdout)
+    end
+
+    it 'can handle a sinatra request' do
+      @in << <<-REQUEST.gsub("\n", "\r\n")
+GET / HTTP/1.1
+Server-Name: localhost
+
+REQUEST
+
+      @server.run
+
+      @out.read.should =~ /response body\Z/
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,8 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'rack/singleshot'
+require 'sinatra'
+
+RSpec.configure do |config|
+  config.color_enabled = true
+end

metadata

@@ -0,0 +1,129 @@
+--- !ruby/object:Gem::Specification 
+name: rack-singleshot
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Ben Burkert
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-02-14 00:00:00 +13:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rack
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: sinatra
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id004
+description: |
+  Handles a single request to/from STDIN/STDOUT. Exits when the response
+  has finished. Sutable for running via inetd.
+
+email: 
+- ben@benburkert.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- ./lib/rack/singleshot.rb
+- ./spec/singleshot_spec.rb
+- ./spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/benburkert/rack-singleshot
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: A single-shot rack handler.
+test_files: 
+- ./spec/singleshot_spec.rb
+- ./spec/spec_helper.rb