gazer 0.3.2 → 0.3.3

data/lib/gzr/modules/project.rb

@@ -0,0 +1,106 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2023 Mike DeAngelo Google, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+module Gzr
+  module Project
+
+    def all_projects(fields: 'id')
+      data = []
+      begin
+        data = @sdk.all_projects({ fields: fields })
+      rescue LookerSDK::NotFound => e
+        return []
+      rescue LookerSDK::Error => e
+        say_error "Error querying all_projects(#{fields})"
+        say_error e
+        raise
+      end
+      data
+    end
+
+    def cat_project(project_id)
+      begin
+        return @sdk.project(project_id)&.to_attrs
+      rescue LookerSDK::NotFound => e
+        return nil
+      rescue LookerSDK::Error => e
+        say_error "Error getting project(#{project_id})"
+        say_error e
+        raise
+      end
+    end
+
+    def trim_project(data)
+      data.select do |k,v|
+        keys_to_keep('create_project').include? k
+      end
+    end
+
+    def create_project(body)
+      begin
+        return @sdk.create_project(body)&.to_attrs
+      rescue LookerSDK::Error => e
+        say_error "Error running create_project(#{JSON.pretty_generate(body)})"
+        say_error e
+        raise
+      end
+    end
+
+    def update_project(id,body)
+      begin
+        return @sdk.update_project(id,body)&.to_attrs
+      rescue LookerSDK::NotFound => e
+        return nil
+      rescue LookerSDK::Error => e
+        say_error "Error running update_project(#{id},#{JSON.pretty_generate(body)})"
+        say_error e
+        raise
+      end
+    end
+
+    def git_deploy_key(id)
+      begin
+        return @sdk.git_deploy_key(id)
+      rescue LookerSDK::NotFound => e
+        return nil
+      rescue LookerSDK::Error => e
+        say_error "Error running git_deploy_key(#{id})"
+        say_error e
+        raise
+      end
+    end
+
+    def create_git_deploy_key(id)
+      begin
+        return @sdk.create_git_deploy_key(id)
+      rescue LookerSDK::NotFound => e
+        return nil
+      rescue LookerSDK::Error => e
+        say_error "Error running create_git_deploy_key(#{id})"
+        say_error e
+        raise
+      end
+    end
+
+  end
+end

data/lib/gzr/modules/session.rb

@@ -55,6 +55,38 @@ module Gzr
       !versions.drop_while {|v| v < minimum_version}.reverse.drop_while {|v| v > given_version}.empty?
     end
 
+    def token_file
+      "#{ENV["HOME"]}/.gzr_auth"
+    end
+
+    def read_token_data
+      return nil unless File.exist?(token_file)
+      s = File.stat(token_file)
+      if !(s.mode.to_s(8)[3..5] == "600")
+        say_error "#{token_file} mode is #{s.mode.to_s(8)[3..5]}. It must be 600. Ignoring."
+        return nil
+      end
+      token_data = nil
+      file = nil
+      begin
+        file = File.open(token_file)
+        token_data = JSON.parse(file.read,{:symbolize_names => true})
+      ensure
+        file.close if file
+      end
+      token_data
+    end
+
+    def write_token_data(token_data)
+      file = nil
+      begin
+        file = File.new(token_file, "wt")
+        file.chmod(0600)
+        file.write JSON.pretty_generate(token_data)
+      ensure
+        file.close if file
+      end
+    end
 
     def build_connection_hash(api_version=nil)
       conn_hash = Hash.new
@@ -86,6 +118,9 @@ module Gzr
         }
       end
       conn_hash[:user_agent] = "Gazer #{Gzr::VERSION}"
+
+      return conn_hash if @options[:token] || @options[:token_file]
+
       if @options[:client_id] then
         conn_hash[:client_id] = @options[:client_id]
         if @options[:client_secret] then
@@ -103,12 +138,14 @@ module Gzr
     end
 
     def login(min_api_version="4.0")
-      if (@options[:client_id].nil? && ENV["LOOKERSDK_CLIENT_ID"])
-        @options[:client_id] = ENV["LOOKERSDK_CLIENT_ID"]
-      end
+      if !@options[:token] && !@options[:token_file]
+        if (@options[:client_id].nil? && ENV["LOOKERSDK_CLIENT_ID"])
+          @options[:client_id] = ENV["LOOKERSDK_CLIENT_ID"]
+        end
 
-      if (@options[:client_secret].nil? && ENV["LOOKERSDK_CLIENT_SECRET"])
-        @options[:client_secret] = ENV["LOOKERSDK_CLIENT_SECRET"]
+        if (@options[:client_secret].nil? && ENV["LOOKERSDK_CLIENT_SECRET"])
+          @options[:client_secret] = ENV["LOOKERSDK_CLIENT_SECRET"]
+        end
       end
 
       if (@options[:verify_ssl] && ENV["LOOKERSDK_VERIFY_SSL"])
@@ -183,6 +220,31 @@ module Gzr
         @sdk = LookerSDK::Client.new(conn_hash.merge(faraday: faraday)) unless @sdk
 
         say_ok "check for connectivity: #{@sdk.alive?}" if @options[:debug]
+        if @options[:token_file]
+          entry = read_token_data&.fetch(@options[:host].to_sym,nil)&.fetch(@options[:su]&.to_sym || :default,nil)
+          if entry.nil?
+            say_error "No token found for host #{@options[:host]} and user #{@options[:su] || "default"}"
+            say_error "login with `gzr session login --host #{@options[:host]}` to set a token"
+            raise LookerSDK::Unauthorized.new
+          end
+          (day, time, tz) = entry[:expiration].split(' ')
+          day_parts = day.split('-')
+          time_parts = time.split(':')
+          date_time_parts = day_parts + time_parts + [tz]
+          expiration = Time.new(*date_time_parts)
+          if expiration < (Time.now + 300)
+            if expiration < Time.now
+              say_error "token expired at #{expiration}"
+            else
+              say_error "token expires at #{expiration}, which is in the next 5 minutes"
+            end
+            say_error "login again with `gzr session login --host #{@options[:host]}`"
+            raise LookerSDK::Unauthorized.new
+          end
+          @sdk.access_token = entry[:token]
+        elsif @options[:token]
+          @sdk.access_token = @options[:token]
+        end
         say_ok "verify authentication: #{@sdk.authenticated?}" if @options[:debug]
       rescue LookerSDK::Unauthorized => e
         say_error "Unauthorized - credentials are not valid"
@@ -196,7 +258,7 @@ module Gzr
       raise Gzr::CLI::Error, "Invalid credentials" unless @sdk.authenticated?
 
 
-      if @options[:su] then
+      if @options[:su] && !(@options[:token] || @options[:token_file])then
         say_ok "su to user #{@options[:su]}" if @options[:debug]
         @access_token_stack.push(@sdk.access_token)
         begin
@@ -247,8 +309,20 @@ module Gzr
         e.backtrace.each { |b| say_error b } if @options[:debug]
         raise Gzr::CLI::Error, e.message
       ensure
-        logout_all
+        logout_all unless @options[:token] || @options[:token_file]
       end
     end
+
+    def update_auth(workspace_id)
+      body = {}
+      body[:workspace_id] = workspace_id
+      begin
+        return @sdk.update_session(body)&.to_attrs
+      rescue LookerSDK::Error => e
+        say_error "Unable to run update_session(#{JSON.pretty_generate(body)})"
+        say_error e
+      end
+    end
+
   end
 end

data/lib/gzr/version.rb

@@ -20,5 +20,5 @@
 # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
 module Gzr
-  VERSION = '0.3.2'.freeze
+  VERSION = '0.3.3'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gazer
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.3.3
 platform: ruby
 authors:
 - Mike DeAngelo
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-04-21 00:00:00.000000000 Z
+date: 2023-05-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tty-reader
@@ -291,8 +291,12 @@ files:
 - lib/gzr/commands/attribute/rm.rb
 - lib/gzr/commands/attribute/set_group_value.rb
 - lib/gzr/commands/connection.rb
+- lib/gzr/commands/connection/cat.rb
 - lib/gzr/commands/connection/dialects.rb
+- lib/gzr/commands/connection/import.rb
 - lib/gzr/commands/connection/ls.rb
+- lib/gzr/commands/connection/rm.rb
+- lib/gzr/commands/connection/test.rb
 - lib/gzr/commands/dashboard.rb
 - lib/gzr/commands/dashboard/cat.rb
 - lib/gzr/commands/dashboard/import.rb
@@ -328,6 +332,12 @@ files:
 - lib/gzr/commands/plan/ls.rb
 - lib/gzr/commands/plan/rm.rb
 - lib/gzr/commands/plan/run.rb
+- lib/gzr/commands/project.rb
+- lib/gzr/commands/project/cat.rb
+- lib/gzr/commands/project/deploy_key.rb
+- lib/gzr/commands/project/import.rb
+- lib/gzr/commands/project/ls.rb
+- lib/gzr/commands/project/update.rb
 - lib/gzr/commands/query.rb
 - lib/gzr/commands/query/runquery.rb
 - lib/gzr/commands/role.rb
@@ -340,6 +350,11 @@ files:
 - lib/gzr/commands/role/user_add.rb
 - lib/gzr/commands/role/user_ls.rb
 - lib/gzr/commands/role/user_rm.rb
+- lib/gzr/commands/session.rb
+- lib/gzr/commands/session/get.rb
+- lib/gzr/commands/session/login.rb
+- lib/gzr/commands/session/logout.rb
+- lib/gzr/commands/session/update.rb
 - lib/gzr/commands/subcommandbase.rb
 - lib/gzr/commands/user.rb
 - lib/gzr/commands/user/cat.rb
@@ -359,6 +374,7 @@ files:
 - lib/gzr/modules/model.rb
 - lib/gzr/modules/permissions.rb
 - lib/gzr/modules/plan.rb
+- lib/gzr/modules/project.rb
 - lib/gzr/modules/role.rb
 - lib/gzr/modules/session.rb
 - lib/gzr/modules/user.rb