require "http/headers" module HTTP class Request class Writer # CRLF is the universal HTTP delimiter CRLF = "\r\n".freeze # Chunked data termintaor. ZERO = "0".freeze # Chunked transfer encoding CHUNKED = "chunked".freeze # End of a chunked transfer CHUNKED_END = "#{ZERO}#{CRLF}#{CRLF}".freeze # Types valid to be used as body source VALID_BODY_TYPES = [String, NilClass, Enumerable] def initialize(socket, body, headers, headline) @body = body @socket = socket @headers = headers @request_header = [headline] validate_body_type! end # Adds headers to the request header from the headers array def add_headers @headers.each do |field, value| @request_header << "#{field}: #{value}" end end # Stream the request to a socket def stream add_headers add_body_type_headers send_request end # Send headers needed to connect through proxy def connect_through_proxy add_headers write(join_headers) end # Adds the headers to the header array for the given request body we are working # with def add_body_type_headers if @body.is_a?(String) && !@headers[Headers::CONTENT_LENGTH] @request_header << "#{Headers::CONTENT_LENGTH}: #{@body.bytesize}" elsif @body.is_a?(Enumerable) && CHUNKED != @headers[Headers::TRANSFER_ENCODING] fail(RequestError, "invalid transfer encoding") end end # Joins the headers specified in the request into a correctly formatted # http request header string def join_headers # join the headers array with crlfs, stick two on the end because # that ends the request header @request_header.join(CRLF) + (CRLF) * 2 end def send_request headers = join_headers case @body when NilClass write(headers) when String write(headers << @body) when Enumerable write(headers) @body.each do |chunk| write(chunk.bytesize.to_s(16) << CRLF) write(chunk << CRLF) end write(CHUNKED_END) else fail TypeError, "invalid body type: #{@body.class}" end end private def write(data) until data.empty? length = @socket.write(data) if data.length > length data = data[length..-1] else break end end end def validate_body_type! return if VALID_BODY_TYPES.any? { |type| @body.is_a? type } fail RequestError, "body of wrong type: #{@body.class}" end end end end