class Typhoeus::Hydra

def self.hydra
  @hydra ||= new
end

def self.hydra=(val)
  @hydra = val
end

def cache_getter(&block)
  @cache_getter = block
end

def cache_setter(&block)
  @cache_setter = block
end

def clear_cache_callbacks
  @cache_setter = nil
  @cache_getter = nil
end

def disable_memoization
  @memoize_requests = false
end

def fire_and_forget
  @queued_requests.each {|r| queue(r, false)}
  @multi.fire_and_forget
end

def get_easy_object(request)
  @running_requests += 1
  easy = @easy_pool.pop || Easy.new
  easy.verbose          = request.verbose
  if request.username || request.password
    auth = { :username => request.username, :password => request.password }
    auth[:method] = Typhoeus::Easy::AUTH_TYPES["CURLAUTH_#{request.auth_method.to_s.upcase}".to_sym] if request.auth_method
    easy.auth = auth
  end
  easy.url          = request.url
  easy.method       = request.method
  easy.params       = request.params  if request.method == :post && !request.params.nil?
  easy.headers      = request.headers if request.headers
  easy.request_body = request.body    if request.body
  easy.timeout      = request.timeout if request.timeout
  easy.connect_timeout = request.connect_timeout if request.connect_timeout
  easy.follow_location = request.follow_location if request.follow_location
  easy.max_redirects = request.max_redirects if request.max_redirects
  easy.proxy = request.proxy if request.proxy
  easy.disable_ssl_peer_verification if request.disable_ssl_peer_verification
  easy.ssl_cert         = request.ssl_cert
  easy.ssl_cert_type    = request.ssl_cert_type
  easy.ssl_key          = request.ssl_key
  easy.ssl_key_type     = request.ssl_key_type
  easy.ssl_key_password = request.ssl_key_password
  easy.ssl_cacert       = request.ssl_cacert
  easy.ssl_capath       = request.ssl_capath
  easy.verbose          = request.verbose
  easy.on_success do |easy|
    queue_next
    handle_request(request, response_from_easy(easy, request))
    release_easy_object(easy)
  end
  easy.on_failure do |easy|
    queue_next
    handle_request(request, response_from_easy(easy, request))
    release_easy_object(easy)
  end
  easy.set_headers
  easy
end

def get_from_cache_or_queue(request)
  if @cache_getter
    val = @cache_getter.call(request)
    if val
      @retrieved_from_cache[request.url] = val
      handle_request(request, val, false)
    else
      @multi.add(get_easy_object(request))
    end
  else
    @multi.add(get_easy_object(request))
  end
end

def handle_request(request, response, live_request = true)
  request.response = response
  self.class.run_global_hooks_for(:after_request_before_on_complete,
                                  request)
  if live_request && request.cache_timeout && @cache_setter
    @cache_setter.call(request)
  end
  @on_complete.call(response) if @on_complete
  request.call_handlers
  if requests = @memoized_requests[request.url]
    requests.each do |r|
      r.response = response
      r.call_handlers
    end
  end
end

def initialize(options = {})
  @memoize_requests = true
  @multi       = Multi.new
  @easy_pool   = []
  initial_pool_size = options[:initial_pool_size] || 10
  @max_concurrency = options[:max_concurrency] || 200
  initial_pool_size.times { @easy_pool << Easy.new }
  @memoized_requests = {}
  @retrieved_from_cache = {}
  @queued_requests = []
  @running_requests = 0
  self.stubs = []
  @active_stubs = []
end

def on_complete(&block)
  @on_complete = block
end

def on_complete=(proc)
  @on_complete = proc
end

def queue(request, obey_concurrency_limit = true)
  return if assign_to_stub(request)
  # At this point, we are running over live HTTP. Make sure we haven't
  # disabled live requests.
  check_allow_net_connect!(request)
  if @running_requests >= @max_concurrency && obey_concurrency_limit
    @queued_requests << request
  else
    if request.method == :get
      if @memoize_requests && @memoized_requests.has_key?(request.url)
        if response = @retrieved_from_cache[request.url]
          request.response = response
          request.call_handlers
        else
          @memoized_requests[request.url] << request
        end
      else
        @memoized_requests[request.url] = [] if @memoize_requests
        get_from_cache_or_queue(request)
      end
    else
      get_from_cache_or_queue(request)
    end
  end
end

def queue_next
  @running_requests -= 1
  queue(@queued_requests.pop) unless @queued_requests.empty?
end

def release_easy_object(easy)
  easy.reset
  @easy_pool.push easy
end

def response_from_easy(easy, request)
  Response.new(:code    => easy.response_code,
               :headers => easy.response_header,
               :body    => easy.response_body,
               :time    => easy.total_time_taken,
               :effective_url => easy.effective_url,
               :request => request)
end

def run
  while !@active_stubs.empty?
    m = @active_stubs.first
    while request = m.requests.shift
      response = m.response
      response.request = request
      handle_request(request, response)
    end
    @active_stubs.delete(m)
  end
  @multi.perform
  @memoized_requests = {}
  @retrieved_from_cache = {}
end