Contents

require 'net/https'
require 'json'
require 'forwardable'
require 'active_support'
require 'active_support/core_ext'
require 'redash_exporter/query'

module RedashExporter
  class Queries
    extend Forwardable
    include Enumerable
    attr_accessor :api_key, :url, :queries
    def_delegators :@queries, :each, :size

    def initialize(url, api_key, dest = "#{__dir__}/../../dest")
      @url = url
      @url = "#{@url}/" unless @url.end_with?('/')
      @api_key = api_key
      @dest = dest
      fetch
    end

    def export_all
      dest_dir_path = File.expand_path(@dest)
      puts "Export SQL Files to #{dest_dir_path}"

      @queries.each do |query|
        query.export(dest_dir_path)
      end
      puts "done. Check out #{dest_dir_path}"
    end

    def fetch
      uri = URI.parse("#{@url}api/queries")
      header = {
        Authorization: "Key #{@api_key}"
      }
      http = Net::HTTP.new(uri.host, uri.port)
      http.use_ssl = true if uri.scheme == 'https'

      req = Net::HTTP::Get.new(uri.path, header)
      res = http.request(req)
      body = JSON.parse(res.body, symbolize_names: true)
      @queries = if body.is_a?(Array)
                   body.map { |q| Query.new(q) }
                 elsif body.is_a?(Hash)
                   body[:results].map { |q| Query.new(q) }
                 end
    end

    def reject!(*args, &block)
      @queries = reject(*args, &block)
    end

    def select!(*args, &block)
      @queries = select(*args, &block)
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
redash_exporter-0.1.0	lib/redash_exporter/queries.rb