wavefront-cli 5.1.1 → 7.2.0

data/lib/wavefront-cli/display/proxy.rb

@@ -7,6 +7,16 @@ module WavefrontDisplay
   # Format human-readable output for proxies.
   #
   class Proxy < Base
+    def do_list
+      filter_inactive_proxies! if options[:active]
+      super
+    end
+
+    def do_list_brief
+      filter_inactive_proxies! if options[:active]
+      super
+    end
+
     def do_describe
       readable_time(:lastCheckInTime)
       long_output
@@ -15,5 +25,11 @@ module WavefrontDisplay
     def do_versions
       multicolumn(:id, :version, :name)
     end
+
+    private
+
+    def filter_inactive_proxies!
+      data.delete_if { |p| p[:status] != 'ACTIVE' }
+    end
   end
 end

data/lib/wavefront-cli/display/role.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module WavefrontDisplay
+  #
+  # Format human-readable output for role command
+  #
+  class Role < Base
+    def do_list_brief
+      data.map! do |d|
+        d.merge(acct_count: "#{d[:linkedAccountsCount]} accounts",
+                group_count: "#{d[:linkedGroupsCount]} groups")
+      end
+      multicolumn(:id, :name, :acct_count, :group_count)
+    end
+
+    def do_accounts
+      if data.empty?
+        puts "No accounts have role '#{options[:'<id>']}'."
+      else
+        multicolumn(:identifier)
+      end
+    end
+
+    def do_groups
+      if data.empty?
+        puts "No groups have role '#{options[:'<id>']}'."
+      else
+        multicolumn(:id, :name)
+      end
+    end
+
+    def do_permissions
+      if data[:permissions].empty?
+        puts "Role '#{options[:'<id>']}' has no permissions."
+      else
+        puts data[:permissions]
+      end
+    end
+
+    def do_grant
+      puts format("Granted '%<perm>s' permission to '%<id>s'.",
+                  perm: options[:'<permission>'],
+                  id: options[:'<id>'])
+    end
+
+    def do_revoke
+      puts format("Revoked '%<perm>s' permission from '%<id>s'.",
+                  perm: options[:'<permission>'],
+                  id: options[:'<id>'])
+    end
+
+    def do_give_to
+      puts format("Gave '%<role>s' to %<members>s.",
+                  members: quoted(options[:'<member>']),
+                  role: options[:'<id>']).fold(TW, 0)
+    end
+
+    def do_take_from
+      puts format("Took '%<role>s' from %<members>s.",
+                  members: quoted(options[:'<member>']),
+                  role: options[:'<id>']).fold(TW, 0)
+    end
+  end
+end

data/lib/wavefront-cli/display/settings.rb

@@ -8,6 +8,7 @@ module WavefrontDisplay
   #
   class Settings < Base
     def do_list_permissions
+      data.sort_by! { |p| p[:groupName] }
       options[:long] ? long_output : multicolumn(:groupName)
     end
 

data/lib/wavefront-cli/display/usergroup.rb

@@ -15,28 +15,28 @@ module WavefrontDisplay
       puts "Deleted user group '#{options[:'<id>']}'."
     end
 
-    def do_add_user
+    def do_add_to
       puts format("Added %<quoted_user>s to '%<group_id>s'.",
                   quoted_user: quoted(options[:'<user>']),
                   group_id: options[:'<id>']).fold(TW, 0)
     end
 
-    def do_remove_user
+    def do_remove_from
       puts format("Removed %<quoted_user>s from '%<group_id>s'.",
                   quoted_user: quoted(options[:'<user>']),
                   group_id: options[:'<id>']).fold(TW, 0)
     end
 
-    def do_grant
-      puts format("Granted '%<perm>s' permission to '%<group_id>s'.",
-                  perm: options[:'<permission>'],
-                  group_id: options[:'<id>'])
+    def do_add_role
+      puts format("Added %<quoted_role>s to '%<group_id>s'.",
+                  quoted_role: quoted(options[:'<role>']),
+                  group_id: options[:'<id>']).fold(TW, 0)
     end
 
-    def do_revoke
-      puts format("Revoked '%<perm>s' permission from '%<group_id>s'.",
-                  perm: options[:'<permission>'],
-                  group_id: options[:'<id>'])
+    def do_remove_role
+      puts format("Removed %<quoted_role>s from '%<group_id>s'.",
+                  quoted_role: quoted(options[:'<role>']),
+                  group_id: options[:'<id>']).fold(TW, 0)
     end
 
     def do_users
@@ -47,12 +47,16 @@ module WavefrontDisplay
            end)
     end
 
-    def do_permissions
-      puts(if !data.include?(:permissions) || data[:permissions].empty?
-             "Group '#{options[:'<id>']}' has no permissions."
+    def do_roles
+      puts(if !data.include?(:roles) || data[:roles].empty?
+             "Group '#{options[:'<id>']}' has no roles attached."
            else
-             data[:permissions]
+             data[:roles].map { |r| r[:id] }
            end)
     end
+
+    def do_permissions
+      puts data[:roles].map { |r| r[:permissions] }.flatten.sort.uniq
+    end
   end
 end

data/lib/wavefront-cli/exception_handler.rb

@@ -0,0 +1,89 @@
+# frozen_string_literal: true
+
+module WavefrontCli
+  #
+  # Handle fatal errors.
+  #
+  module ExceptionMixins
+    # rubocop:disable Metrics/MethodLength
+    # rubocop:disable Metrics/AbcSize
+    # rubocop:disable Metrics/CyclomaticComplexity
+    # rubocop:disable Metrics/PerceivedComplexity
+    def exception_handler(exception)
+      case exception
+      when WavefrontCli::Exception::UnhandledCommand
+        abort 'Fatal error. Unsupported command. Please open a Github issue.'
+      when WavefrontCli::Exception::InvalidInput
+        abort "Invalid input. #{exception.message}"
+      when Interrupt
+        abort "\nOperation aborted at user request."
+      when WavefrontCli::Exception::ConfigFileNotFound
+        abort "Configuration file #{exception}' not found."
+      when WavefrontCli::Exception::CredentialError
+        handle_missing_credentials(exception)
+      when WavefrontCli::Exception::MandatoryValue
+        abort 'A value must be supplied.'
+      when Wavefront::Exception::NetworkTimeout
+        abort 'Connection timed out.'
+      when Wavefront::Exception::InvalidPermission
+        abort "'#{exception}' is not a valid Wavefront permission."
+      when Wavefront::Exception::InvalidUserGroupId
+        abort "'#{exception}' is not a valid user group ID."
+      when Wavefront::Exception::InvalidAccountId
+        abort "'#{exception}' is not a valid system or user account ID."
+      when Wavefront::Exception::InvalidAwsExternalId
+        abort "'#{exception}' is not a valid AWS external ID."
+      when Wavefront::Exception::InvalidRoleId
+        abort "'#{exception}' is not a valid role ID."
+      when Wavefront::Exception::InvalidApiTokenId
+        abort "'#{exception}' is not a valid API token ID."
+      when Wavefront::Exception::InvalidIngestionPolicyId
+        abort "'#{exception}' is not a valid ingestion policy ID."
+      when WavefrontCli::Exception::InvalidValue
+        abort "Invalid value for #{exception}."
+      when WavefrontCli::Exception::ProfileExists
+        abort "Profile '#{exception}' already exists."
+      when WavefrontCli::Exception::ProfileNotFound
+        abort "Profile '#{exception}' not found."
+      when WavefrontCli::Exception::FileNotFound
+        abort 'File not found.'
+      when WavefrontCli::Exception::InsufficientData
+        abort "Insufficient data. #{exception.message}"
+      when WavefrontCli::Exception::InvalidQuery
+        abort "Invalid query. API message: '#{exception.message}'."
+      when WavefrontCli::Exception::SystemError
+        abort "Host system error. #{exception.message}"
+      when WavefrontCli::Exception::UnparseableInput
+        abort "Cannot parse input. #{exception.message}"
+      when WavefrontCli::Exception::UnparseableSearchPattern
+        abort 'Searches require a key, a value, and a match operator.'
+      when WavefrontCli::Exception::UnsupportedFileFormat
+        abort 'Unsupported file format.'
+      when WavefrontCli::Exception::UnsupportedOperation
+        abort "Unsupported operation.\n#{exception.message}"
+      when WavefrontCli::Exception::UnsupportedOutput
+        abort exception.message
+      when WavefrontCli::Exception::UnsupportedNoop
+        abort 'Multiple API call operations cannot be performed as no-ops.'
+      when WavefrontCli::Exception::UserGroupNotFound
+        abort "Cannot find user group '#{exception.message}'."
+      when Wavefront::Exception::UnsupportedWriter
+        abort "Unsupported writer '#{exception.message}'."
+      when WavefrontCli::Exception::UserError
+        abort "User error: #{exception.message}."
+      when WavefrontCli::Exception::ImpossibleSearch
+        abort 'Search on non-existent key. Please use a top-level field.'
+      when Wavefront::Exception::InvalidSamplingValue
+        abort 'Sampling rates must be between 0 and 0.05.'
+      else
+        warn "general error: #{exception}"
+        backtrace_message(exception)
+        abort
+      end
+    end
+    # rubocop:enable Metrics/MethodLength
+    # rubocop:enable Metrics/AbcSize
+    # rubocop:enable Metrics/PerceivedComplexity
+    # rubocop:enable Metrics/CyclomaticComplexity
+  end
+end

data/lib/wavefront-cli/output/hcl/base.rb

@@ -92,7 +92,7 @@ module WavefrontHclOutput
     def quote_value(val)
       case val.class.to_s.to_sym
       when :String
-        format('"%<value>s"', value: val.gsub(/\"/, '\"'))
+        format('"%<value>s"', value: val.gsub(/"/, '\"'))
       else
         val
       end

data/lib/wavefront-cli/output/hcl/dashboard.rb

@@ -91,7 +91,7 @@ module WavefrontHclOutput
     end
 
     def quote_value(val)
-      val.gsub!(/\$/, '$$') if v.is_a?(String)
+      val.gsub!(/\$/, '$$') if val.is_a?(String)
       super
     end
   end

data/lib/wavefront-cli/proxy.rb

@@ -7,6 +7,11 @@ module WavefrontCli
   # CLI coverage for the v2 'proxy' API.
   #
   class Proxy < WavefrontCli::Base
+    def do_list
+      options[:all] = true if options[:active]
+      super
+    end
+
     def no_api_response
       %w[do_versions]
     end

data/lib/wavefront-cli/query.rb

@@ -51,20 +51,26 @@ module WavefrontCli
 
     # @return [Hash] options for the SDK query method
     #
-    # rubocop:disable Metrics/AbcSize
     def q_opts
+      basic_q_opts.tap do |o|
+        o[:n] = options[:name]
+        o[:p] = options[:points]
+        o[:view] = 'HISTOGRAM' if options[:histogramview]
+        o[:cached] = false if options[:nocache]
+      end.compact
+    end
+
+    # Every query gets these options. They're modified by q_opts
+    #
+    def basic_q_opts
       { autoEvents: options[:events],
         i: options[:inclusive],
         summarization: options[:summarize] || 'mean',
         listMode: true,
-        strict: true,
+        strict: !options[:nostrict],
         includeObsoleteMetrics: options[:obsolete],
-        sorted: true }.tap do |o|
-          o[:n] = options[:name] if options[:name]
-          o[:p] = options[:points] if options[:points]
-        end
+        sorted: true }
     end
-    # rubocop:enable Metrics/AbcSize
 
     # @return [Integer] start of query window. If one has been
     #   given, that; if not, ten minutes ago

data/lib/wavefront-cli/role.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module WavefrontCli
+  #
+  # CLI coverage for the v2 'role' API.
+  #
+  class Role < WavefrontCli::Base
+    alias do_permissions do_describe
+
+    def do_create
+      wf.create({ name: options[:'<name>'],
+                  description: options[:description],
+                  permissions: options[:permission] }.compact)
+    end
+
+    def do_accounts
+      things_with_role(:account, options[:'<id>'])
+    end
+
+    def do_groups
+      things_with_role(:usergroup, options[:'<id>'])
+    end
+
+    def do_give_to
+      wf.add_assignees(options[:'<id>'], options[:'<member>'])
+    end
+
+    def do_take_from
+      wf.remove_assignees(options[:'<id>'], options[:'<member>'])
+    end
+
+    def do_grant
+      wf.grant(options[:'<permission>'], Array(options[:'<id>']))
+    end
+
+    def do_revoke
+      wf.revoke(options[:'<permission>'], Array(options[:'<id>']))
+    end
+
+    private
+
+    # Search for objects of the given type with the given role
+    #
+    def things_with_role(thing, role)
+      require 'wavefront-sdk/search'
+      wfs = Wavefront::Search.new(mk_creds, mk_opts)
+      wfs.search(thing,
+                 conds_to_query(["roles~#{role}"]),
+                 limit: :all, sort_field: :id)
+    end
+  end
+end

data/lib/wavefront-cli/serviceaccount.rb

@@ -173,20 +173,14 @@ module WavefrontCli
 
     def validate_groups
       options[:group].each { |g| wf_usergroup_id?(g) }
-    rescue Wavefront::Exception::InvalidUserGroupId => e
-      raise e, 'Invalid usergroup ID'
     end
 
     def validate_tokens
       options[:usertoken].each { |t| wf_apitoken_id?(t) }
-    rescue Wavefront::Exception::InvalidApiTokenId => e
-      raise e, 'Invalid API token'
     end
 
     def validate_perms
       options[:permission].each { |p| wf_permission?(p) }
-    rescue Wavefront::Exception::InvalidPermission => e
-      raise e, 'Invalid permission'
     end
 
     def descriptive_name

data/lib/wavefront-cli/spy.rb

@@ -42,14 +42,6 @@ module WavefrontCli
 
     private
 
-    def require_sdk_class
-      require 'wavefront-sdk/unstable/spy'
-    end
-
-    def _sdk_class
-      'Wavefront::Unstable::Spy'
-    end
-
     def rate
       return 0.01 unless options[:rate]
 

data/lib/wavefront-cli/usergroup.rb

@@ -12,27 +12,27 @@ module WavefrontCli
     end
 
     alias do_users do_describe
+    alias do_roles do_describe
     alias do_permissions do_describe
 
     def do_create
-      wf.create(name: options[:'<name>'],
-                permissions: options[:permission])
+      wf.create(name: options[:'<name>'], roleIDs: options[:roleid])
     end
 
-    def do_add_user
+    def do_add_to
       wf.add_users_to_group(options[:'<id>'], options[:'<user>'])
     end
 
-    def do_remove_user
+    def do_remove_from
       wf.remove_users_from_group(options[:'<id>'], options[:'<user>'])
     end
 
-    def do_grant
-      wf.grant(options[:'<permission>'], Array(options[:'<id>']))
+    def do_add_role
+      wf.add_roles_to_group(options[:'<id>'], options[:'<role>'])
     end
 
-    def do_revoke
-      wf.revoke(options[:'<permission>'], Array(options[:'<id>']))
+    def do_remove_role
+      wf.remove_roles_from_group(options[:'<id>'], options[:'<role>'])
     end
 
     def import_to_create(raw)