module Vcloud

  class QueryRunner
    def initialize
      @fsi = Vcloud::Fog::ServiceInterface.new
    end

    def run(type=nil, options={})
      get_all_results(type, options)
    end

    def available_query_types
      query_list = @fsi.get_execute_query()
      query_list[:Link].select do |link|
        link[:rel] == 'down'
      end.map do |link|
        href  = Nokogiri::XML.fragment(link[:href])
        query = CGI.parse(URI.parse(href.text).query)
        [query['type'].first, query['format'].first]
      end
    end

  private

    def get_all_results(type, options)
      results = []
      (1..get_num_pages(type, options)).each do |page|
        results += get_results_page(page, type, options) || []
      end
      results
    end

    def get_num_pages(type, options)
      body = @fsi.get_execute_query(type, options)
      last_page = body[:lastPage] || 1
      raise "Invalid lastPage (#{last_page}) in query results" unless last_page.is_a? Integer
      last_page.to_i
    end

    def get_results_page(page, type, options)
      body = @fsi.get_execute_query(type, options.merge({:page=>page}))

      record_key = key_of_first_record_or_reference(body)
      body[record_key] = [body[record_key]] if body[record_key].is_a?(Hash)
      body[record_key]
    end

    def key_of_first_record_or_reference(body)
      body.keys.detect { |key| key.to_s =~ /Record|Reference$/ }
    end
  end
end