Contents

require 'forwardable'

module Reel
  class Request
    extend Forwardable
    include RequestMixin

    UPGRADE   = 'Upgrade'.freeze
    WEBSOCKET = 'websocket'.freeze

    # Array#include? seems slow compared to Hash lookup
    request_methods = Http::METHODS.map { |m| m.to_s.upcase }
    REQUEST_METHODS = Hash[request_methods.zip(request_methods)].freeze

    def self.read(connection)
      parser = connection.parser

      begin
        data = connection.socket.readpartial(Connection::BUFFER_SIZE)
        parser << data
      end until parser.headers

      REQUEST_METHODS[parser.http_method] ||
        raise(ArgumentError, "Unknown Request Method: %s" % parser.http_method)

      upgrade = parser.headers[UPGRADE]
      if upgrade && upgrade.downcase == WEBSOCKET
        WebSocket.new(parser, connection.socket)
      else
        Request.new(parser, connection)
      end
    end

    def_delegators :@connection, :respond, :finish_response, :close, :read

    def initialize(http_parser, connection = nil)
      @http_parser, @connection = http_parser, connection
    end

    def body
      @body ||= begin
        raise "no connection given" unless @connection

        body = "" unless block_given?
        while (chunk = @connection.readpartial)
          if block_given?
            yield chunk
          else
            body << chunk
          end
        end
        body unless block_given?
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
reel-0.4.0.pre	lib/reel/request.rb
reel-0.3.0	lib/reel/request.rb
reel-0.3.0.pre	lib/reel/request.rb