class HTTP::Timeout::Global

def connect(socket_class, host, port)
  reset_timer
  ::Timeout.timeout(time_left, TimeoutError) do
    @socket = socket_class.open(host, port)
  end
  log_time
end

def connect_ssl
  reset_timer
  begin
    socket.connect_nonblock
  rescue IO::WaitReadable
    IO.select([socket], nil, nil, time_left)
    log_time
    retry
  rescue IO::WaitWritable
    IO.select(nil, [socket], nil, time_left)
    log_time
    retry
  end
end

def initialize(*args)
  super
  @time_left = connect_timeout + read_timeout + write_timeout
  @total_timeout = time_left
end

def log_time
  @time_left -= (Time.now - @started)
  if time_left <= 0
    fail TimeoutError, "Timed out after using the allocated #{total_timeout} seconds"
  end
  reset_timer
end

def readpartial(size)
  reset_timer
  begin
    socket.read_nonblock(size)
  rescue IO::WaitReadable
    IO.select([socket], nil, nil, time_left)
    log_time
    retry
  end
rescue EOFError
  :eof
end

def readpartial(size)
  reset_timer
  loop do
    result = socket.read_nonblock(size, :exception => false)
    if result.nil?
      return :eof
    elsif result != :wait_readable
      return result
    end
    IO.select([socket], nil, nil, time_left)
    log_time
  end
end

def reset_timer
  @started = Time.now
end

def write(data)
  reset_timer
  begin
    socket.write_nonblock(data)
  rescue IO::WaitWritable
    IO.select(nil, [socket], nil, time_left)
    log_time
    retry
  end
rescue EOFError
  :eof
end

def write(data)
  reset_timer
  loop do
    result = socket.write_nonblock(data, :exception => false)
    break unless result == :wait_writable
    IO.select(nil, [socket], nil, time_left)
    log_time
  end
end