Contents

require 'csv'

module Vcloud
  class Query

    def initialize(type=nil, options={}, query_runner = Vcloud::QueryRunner.new)
      @type = type
      @options = options
      @options[:output_format] ||= 'tsv'
      @query_runner = query_runner
    end

    def run()
      if @type.nil?
        output_available_query_types
      else
        output_query_results
      end
    end

    # <b>DEPRECATED:</b> Please use <tt>Vcloud::QueryRunner.run</tt> instead.
    def get_all_results
      warn "[DEPRECATION] `Vcloud::Query::get_all_results` is deprecated.  Please use `Vcloud::QueryRunner.run` instead."
      @query_runner.run(@type, @options)
    end

  private
    def output_query_results
      results = @query_runner.run(@type, @options)
      output_header(results)
      output_results(results)
    end

    def output_available_query_types
      @query_runner.available_query_types.each do |entity_type|
        puts entity_type
      end
    end

    def output_header(results)
      return if results.size == 0
      case @options[:output_format]
      when 'csv'
        csv_string = CSV.generate do |csv|
          csv << results.first.keys
        end
        puts csv_string
      when 'tsv'
        puts results.first.keys.join("\t")
      end
    end

    def output_results(results)
      return if results.size == 0

      case @options[:output_format]
      when 'yaml'
        puts YAML.dump(results)
      when 'csv'
        csv_string = CSV.generate do |csv|
          results.each do |record|
            csv << record.values
          end
        end
        puts csv_string
      when 'tsv'
        puts results.first.keys.join("\t") if @options[:page] == 1
        results.each do |record|
          puts record.values.join("\t")
        end
      else
        raise "Unsupported output format #{@options[:output_format]}"
      end
    end

  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
vcloud-core-0.1.0	lib/vcloud/core/query.rb
vcloud-core-0.0.13	lib/vcloud/core/query.rb