Ione Rack server using Puma's HttpParser

http_server.rb

# encoding: utf-8

require 'ione'
require 'ione/http_client'
require 'puma/puma_http11'


module Ione
  class HttpServer
    def initialize(port, app)
      @port = port
      @app = app
      @reactor = Io::IoReactor.new
    end

    def start
      f = @reactor.start
      f = f.flat_map do
        @reactor.bind('0.0.0.0', @port, 5) do |acceptor|
          acceptor.on_accept do |connection|
            accept_connection(connection)
          end
        end
      end
      f.map(self)
    end

    def stop
      @reactor.stop.map(self)
    end

    private

    def accept_connection(connection)
      HttpConnection.new(connection, @app)
    end
  end

  class HttpConnection
    def initialize(connection, app)
      @http_parser = Puma::HttpParser.new
      @env = {}
      @app = app
      @connection = connection
      @connection.on_data(&method(:handle_data))
      @connection.on_closed(&method(:handle_closed))
    end

    def handle_data(data)
      consumed_bytes = @http_parser.execute(@env, data, 0)
      status, headers, body = @app.call(@env)
      @connection.write do |buffer|
        buffer << "HTTP/1.1 200 OK\r\n"
        headers.each do |header, value|
          buffer << "#{header}: #{value}\r\n"
        end
        buffer << "Transfer-Encoding: chunked\r\n"
        buffer << "\r\n"
        body.each do |part|
          chunk = part.to_s
          buffer << "#{chunk.bytesize.to_s(16)}\r\n#{chunk}\r\n"
        end
        buffer << "0\r\n\r\n"
      end
      @connection.drain
    end

    def handle_closed(cause=nil)
    end
  end
end

http_server_spec.rb

# encoding: utf-8

require 'ione/http_server'
require 'ione/http_client'


module Ione
  describe HttpServer do
    let :server do
      described_class.new(port, app)
    end

    let :port do
      rand(2**15) + 2**15
    end

    let :app do
      lambda { |env| [200, {}, [env['REQUEST_PATH']]] }
    end

    let :client do
      HttpClient.new
    end

    context 'handles a GET request' do
      before do
        server.start.value
        client.start.value
      end

      after do
        client.stop.value
        server.stop.value
      end

      it 'responds with the requested path' do
        f = client.get("http://localhost:#{port}/helloworld")
        response = f.value
        response.status.should == 200
        response.body.should == '/helloworld'
      end
    end
  end
end