redash_to_google_docs 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cd0281de00fe298c934be62faf1a44cea5120d4a11f5db58c14f0c1b3b2a7d4c
-  data.tar.gz: 9fa192ef2a18a096935eb34f5c4c086c8a3edc7524f5630fe2301d5271f30a0e
+  metadata.gz: eb72009d25157a7ac5cf84df77bfb678105da72d4f681f6bcc1cc1b159e34da6
+  data.tar.gz: 6fdf0ff0bc5a329c328fbfc5bf9f729eb30d90e0088cfc7a07039af48397a354
 SHA512:
-  metadata.gz: 5bfca653c5137a96f2aa5aa41dd7b73ab6601818a422fa9b2c7f1b723803d59bb5c85b4bc6c77536d3c1288d60d8cb7b7dea464edcc74724904fa937149c0f1b
-  data.tar.gz: 89fd59052cd45901a647d29a0c45d1e20360707e9489478dd0a12ab14c383232941a00977aab6f19c4c028d51a70fc3191bfe5cf22eb758d4a79fbc582de278a
+  metadata.gz: dd3cc46dff7f60ac29a5ecadcc83f6c2bfd5d5c9a52e36f233c2d79d55e3c079a7a0849f1d22b7c0545be0a17045d0cb4801f86f558b062953d98057b742a70e
+  data.tar.gz: ac14731ceb830f4609de5e5ed77d9f54ff1b1d3ecb784be973311c260764fd9421a56631d98abecc5c22ed4a105b8b5ae1a5481de5c283500ff3acf0bc910189

data/exe/redash2gsheet

@@ -0,0 +1,177 @@
+#!/usr/bin/env ruby
+
+require 'bundler/setup'
+
+require 'dry/cli'
+require 'net/http'
+require 'uri'
+require 'redash_to_google_docs'
+require 'json'
+
+require 'tty-prompt'
+
+
+require 'google/apis/sheets_v4'
+require 'googleauth'
+require 'googleauth/stores/file_token_store'
+
+
+require "fileutils"
+
+Prompt = TTY::Prompt.new
+
+
+
+module RedashToGoogleSheet
+  module Clients
+    class Redash
+      ResultPending = Struct.new(:job_id)
+      ResultAvailable = Struct.new(:rows)
+
+      attr_reader :client, :api_host, :api_key
+
+      def initialize(api_host, api_key)
+        @api_host = api_host
+        @api_key = api_key
+        @client = Net::HTTP
+      end
+
+      def query_result(query_id)
+        response = client.post(query_results_url(query_id), '', { 'Authorization' => "Key #{api_key}" })
+
+        data = JSON.load(response.body)
+
+        if data.key?('job')
+          ResultPending.new(data.dig('job', 'id'))
+        else
+          ResultAvailable.new(data.dig('query_result', 'data', 'rows'))
+        end
+      end
+
+      def query_results_url(query_id)
+        URI("#{api_host}/api/queries/#{query_id}/results")
+      end
+    end
+
+    class Sheets
+      OOB_URI = "urn:ietf:wg:oauth:2.0:oob".freeze
+      CREDENTIALS_PATH = "credentials.json".freeze
+      TOKEN_PATH = "token.yaml".freeze
+      SCOPE = Google::Apis::SheetsV4::AUTH_SPREADSHEETS
+
+      attr_reader :prompt
+
+      def initialize(prompt)
+        @prompt = prompt
+      end
+
+      ##
+      # Ensure valid credentials, either by restoring from the saved credentials
+      # files or intitiating an OAuth2 authorization. If authorization is required,
+      # the user's default browser will be launched to approve the request.
+      #
+      # @return [Google::Auth::UserRefreshCredentials] OAuth2 credentials
+      def authorize
+        client_id = Google::Auth::ClientId.from_file CREDENTIALS_PATH
+        token_store = Google::Auth::Stores::FileTokenStore.new file: TOKEN_PATH
+        authorizer = Google::Auth::UserAuthorizer.new client_id, SCOPE, token_store
+        user_id = "default"
+        credentials = authorizer.get_credentials user_id
+
+        if credentials.nil?
+          url = authorizer.get_authorization_url base_url: OOB_URI
+          code = prompt.ask(<<~PROMPT)
+            Open the following URL in the browser and enter the resulting code after authorization:
+            #{url}
+          PROMPT
+
+          credentials = authorizer.get_and_store_credentials_from_code(
+            user_id: user_id, code: code, base_url: OOB_URI
+          )
+        end
+        credentials
+      end
+    end
+  end
+
+  module Commands
+    extend Dry::CLI::Registry
+
+    require 'redash_to_google_docs/version'
+    class Version < Dry::CLI::Command
+      def call(**)
+        puts "redash2gsheet version #{RedashToGoogleDocs::VERSION}"
+      end
+    end
+
+    class Sync < Dry::CLI::Command
+      option :redash_key, desc: "Redash API key"
+      option :redash_url, desc: 'Redash URL. Example: https://app.redash.io/company'
+      option :query_id, desc: "Redash query ID. Example: for https://app.redash.io/company/queries/444111, ID is 444111"
+      option :spreadsheet_id, desc: "Spreadsheet URL"
+
+      def call(redash_key:, query_id:, spreadsheet_id:, redash_url: , **)
+        redash_response = fetch_result_from_redash(redash_url, redash_key, query_id)
+
+        case redash_response
+        when Clients::Redash::ResultPending
+          prompt.say "Result is refreshing. Please run this command in a minute"
+        when Clients::Redash::ResultAvailable
+          prompt.say "Alright, we've got the result. Upload it to google sheets"
+
+          service = authenticate_and_authorize
+          result = replace_google_sheet(service, spreadsheet_id, redash_response)
+
+          Prompt.say("Updated #{result.updated_range}")
+        end
+      end
+
+      private
+
+      def prompt
+        Prompt
+      end
+
+      def fetch_result_from_redash(redash_url, redash_key, query_id)
+        client = Clients::Redash.new(redash_url, redash_key)
+        client.query_result(query_id)
+      end
+
+
+      def authenticate_and_authorize
+        Google::Apis::SheetsV4::SheetsService.new.tap do |service|
+          service.authorization = Clients::Sheets.new(prompt).authorize
+        end
+      end
+
+      def replace_google_sheet(service, spreadsheet_id, result)
+        values = [
+          result.rows.first.keys,
+          *result.rows.map(&:values)
+        ]
+
+        clear_values(service, spreadsheet_id)
+
+        upload_values(service, spreadsheet_id, values)
+      end
+
+      def clear_values(service, spreadsheet_id)
+        request_body = Google::Apis::SheetsV4::ClearValuesRequest.new
+        service.clear_values(spreadsheet_id, 'A:Z', request_body)
+      end
+
+      def upload_values(service, spreadsheet_id, values)
+        value_range_object = Google::Apis::SheetsV4::ValueRange.new(values: values)
+        service.update_spreadsheet_value(spreadsheet_id, 'A1', value_range_object, value_input_option: 'RAW')
+      end
+    end
+
+    register 'version', Version, aliases: %w[v]
+
+    register 'sync', Sync, aliases: %w[s]
+  end
+end
+
+
+Dry::CLI.new(RedashToGoogleSheet::Commands).call
+

data/lib/redash_to_google_docs/version.rb

@@ -1,3 +1,3 @@
 module RedashToGoogleDocs
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: redash_to_google_docs
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Igor S. Morozov
@@ -55,7 +55,8 @@ dependencies:
 description:
 email:
 - igor@morozov.is
-executables: []
+executables:
+- redash2gsheet
 extensions: []
 extra_rdoc_files: []
 files:
@@ -67,6 +68,7 @@ files:
 - Rakefile
 - bin/console
 - bin/setup
+- exe/redash2gsheet
 - lib/redash_to_google_docs.rb
 - lib/redash_to_google_docs/version.rb
 - redash_to_google_docs.gemspec