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