wavefront-sdk 3.7.1 → 5.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f8a2c03d2fff24c9d9c986c89203a60dd5535d93c5df5018fc7fdb9fcd4a50a8
-  data.tar.gz: 2957c881c681837d8e90f0c83b635a62e59d5806091f74f9861a4561f8422a83
+  metadata.gz: fc04400fbf51a67e94d228b3145a9fdd5fdb484c24035f1c83ff583e755d705e
+  data.tar.gz: 3acd358086967628300c65bff916b541e28004a1a5ca55b44b764a7ddb277d1e
 SHA512:
-  metadata.gz: 80ffca53940c2d25cdaf6c5248fa0a5eddcbeeed7fc390d1f79405a2bf55fc9b1981624082891aa994e5e243238a37046ddb470829c5eb1140f37fef4dc6a520
-  data.tar.gz: 4b68cae0043168eba7232296c15916a5d7e872cec758a217e8824e78a6a4945569d7d8d8e030a999b2cfb83aab2636617eb94fda12352c6985ce532a7206b416
+  metadata.gz: 665e0f09c85149b0879b868ec51336b88ac9c9c1af542df52ce0e00b932249a06236268e6398e6b828b7a7deb4421669611cdcaaf838965dd2eb560a720f0181
+  data.tar.gz: 519dc5ca20b01d81454bf8255815bd85ba3e016a4639eda688f1e2544b6d6c6e26ce6ff097e125fdf1eb40ab3ac10530307e0df3a31e590fac24279c89965b64

data/.rubocop.yml

@@ -1,7 +1,48 @@
 ---
 
 AllCops:
-  TargetRubyVersion: 2.3
+  TargetRubyVersion: 2.4
 
 Metrics/ClassLength:
-  Max: 124
+  Max: 150
+
+# New cops
+#
+Lint/RaiseException:
+  Enabled: true
+Lint/StructNewOverride:
+  Enabled: true
+Style/ExponentialNotation:
+  Enabled: true
+Style/HashEachMethods:
+  Enabled: true
+Style/HashTransformKeys:
+  Enabled: true
+Style/HashTransformValues:
+  Enabled: true
+Layout/EmptyLinesAroundAttributeAccessor:
+  Enabled: true
+Layout/SpaceAroundMethodCallOperator:
+  Enabled: true
+Style/SlicingWithRange:
+  Enabled: true
+Lint/DeprecatedOpenSSLConstant:
+  Enabled: true
+Lint/MixedRegexpCaptureTypes:
+  Enabled: true
+Style/RedundantRegexpCharacterClass:
+  Enabled: true
+Style/RedundantRegexpEscape:
+  Enabled: true
+Style/AccessorGrouping:
+  Enabled: true
+Style/BisectedAttrAccessor:
+  Enabled: true
+Style/RedundantAssignment:
+  Enabled: true
+Style/RedundantFetchBlock:
+  Enabled: true
+
+# Is nothing sacred?
+Layout/LineLength:
+  Max: 80

data/.travis.yml

@@ -1,7 +1,6 @@
 language: ruby
 cache: bundler
 rvm:
-  - 2.3.8
   - 2.4.9
   - 2.5.7
   - 2.6.5

data/HISTORY.md

@@ -1,5 +1,38 @@
 # Changelog
 
+## 5.2.0 (2020-09-03)
+* Add `:raise_on_no_profile` option to `Wavefront::Credentials` constructor
+  options. If this is true and a specific config stanza is requested but not
+  found, `Wavefront::Exception::MissingConfigProfile` is thrown.
+
+## 5.1.0 (2020-08-15)
+* Add `create_aws_external_id`, `delete_aws_external_id`, and
+  `confirm_aws_external_id` methods to `Wavefront::CloudIntegration`.
+
+## 5.0.1 (2020-07-08)
+* Reinstate `Wavefront::Role#grant` and `Wavefront::Role#revoke`, which were
+  accidentally removed prior to release of 5.0.0.
+
+## 5.0.0 (2020-07-08)
+* Remove `Wavefront::UserGroup#grant` and `Wavefront::UserGroup#revoke` as [the
+ API paths they used have been
+ removed](https://docs.wavefront.com/2020.06.x_release_notes.html#obsolete-and-deprecated-apis).
+ (Breaking change.)
+* Remove `Wavefront::MonitoredCluster` class, as it has been removed from the
+  public API.
+* Deprecate `Wavefront::User` class, as [the user API is now
+  deprecated](https://docs.wavefront.com/2020.06.x_release_notes.html#obsolete-and-deprecated-apis)
+* Add `Wavefront::Role` class, for managing roles.
+* Promote `Wavefront::Spy` class from unstable. It is now an official API.
+
+## 4.0.0 (2020-02-17)
+* Drop support for Ruby 2.3. (Breaking change.)
+* Add `Wavefront::MonitoredCluster` class.
+* Add `Wavefront::Unstable::Spy` class to speak to (undocumented) spy
+  interface.
+* Add `Wavefront::Unstable::Chart` class to speak to (undocumented) chart
+  interface.
+
 ## 3.7.1 (2020-02-09)
 * `Response` object returned by `Wavefront::Write#write` includes a valid HTTP
   code, rather than `nil`.

data/README.md

@@ -24,7 +24,7 @@ or to build locally,
 $ gem build wavefront-sdk.gemspec
 ```
 
-`wavefront-sdk` requires Ruby >= 2.3. All its dependencies are pure
+`wavefront-sdk` requires Ruby >= 2.4. All its dependencies are pure
 Ruby, right the way down, so a compiler should never be required to
 install it.
 
@@ -100,10 +100,11 @@ when you ask for a list of objects.
 
 You can set an offset and a limit when you list, but setting the
 limit to the magic value `:all` will return all items, without you
-having to deal with pagination.
+having to deal with pagination. When you do that, `offset` is repurposed as
+the number of results fetched with each call to the API.
 
 Calling a method with the limit set to `:lazy` returns a lazy
-enumerable.
+enumerable. Again, `offset` is the chunk size.
 
 ```ruby
 wf = Wavefront::Alert.new(creds.all)

data/lib/wavefront-sdk/account.rb

@@ -44,6 +44,18 @@ module Wavefront
       api.get(id)
     end
 
+    # POST /api/v2/account/{id}/addRoles
+    # Add specific roles to the account (user or service account)
+    # @param id [String] ID of the account
+    # @param role_list [Array[String]] list of roles to add
+    # @return [Wavefront::Response]
+    #
+    def add_roles(id, role_list)
+      wf_account_id?(id)
+      validate_role_list(role_list)
+      api.post([id, 'addRoles'].uri_concat, role_list, 'application/json')
+    end
+
     # POST /api/v2/account/{id}/addUserGroups
     # Adds specific user groups to the account (user or service account)
     # @param id [String] ID of the account
@@ -68,10 +80,22 @@ module Wavefront
       api.get([id, 'businessFunctions'].uri_concat)
     end
 
+    # POST /api/v2/account/{id}/removeRoles
+    # Removes specific roles from the account (user or service account)
+    # @param id [String] ID of the account
+    # @param role_list [Array[String]] list of roles to remove
+    # @return [Wavefront::Response]
+    #
+    def remove_roles(id, role_list)
+      wf_account_id?(id)
+      validate_role_list(role_list)
+      api.post([id, 'removeRoles'].uri_concat, role_list, 'application/json')
+    end
+
     # POST /api/v2/account/{id}/removeUserGroups
     # Removes specific user groups from the account (user or service account)
     # @param id [String] ID of the account
-    # @param group_list [Array[String]] list of groups to add
+    # @param group_list [Array[String]] list of groups to remove
     # @return [Wavefront::Response]
     #
     def remove_user_groups(id, group_list)
@@ -132,7 +156,8 @@ module Wavefront
     end
 
     # POST /api/v2/account/removeingestionpolicies
-    # Removes ingestion policies from multiple accounts
+    # Removes ingestion policies from multiple accounts. The API path says
+    # "policies" but I've made the method name "policy" for consistency.
     # @param policy_id [String] ID of the ingestion policy
     # @param id_list [Array[String]] list of accounts to be put in policy
     # @return [Wavefront::Response]
@@ -140,7 +165,7 @@ module Wavefront
     def remove_ingestion_policy(policy_id, id_list)
       wf_ingestionpolicy_id?(policy_id)
       validate_account_list(id_list)
-      api.post('removeingestionpolicy',
+      api.post('removeingestionpolicies',
                { ingestionPolicyId: policy_id,
                  accounts: id_list },
                'application/json')
@@ -157,6 +182,78 @@ module Wavefront
       api.post('deleteAccounts', id_list, 'application/json')
     end
 
+    # GET /api/v2/account/user
+    # Get all user accounts
+    # @param offset [Int] user account at which the list begins
+    # @param limit [Int] the number of user accounts to return
+    # @return [Wavefront::Response]
+    #
+    def user_list(offset = 0, limit = 100)
+      api.get('user', offset: offset, limit: limit)
+    end
+
+    def user_create(body, send_email = false)
+      raise ArgumentError unless body.is_a?(Hash)
+
+      uri = send_email ? "?sendEmail=#{send_email}" : 'user'
+
+      api.post(uri, body, 'application/json')
+    end
+
+    # POST /api/v2/account/user
+    # Creates or updates a user account
+    # @param id [String] a Wavefront user ID
+    # @param body [Hash] key-value hash of the parameters you wish to change
+    # @param modify [true, false] if true, use {#describe()} to get a hash
+    #   describing the existing object, and modify that with the new body. If
+    #   false, pass the new body straight through.
+    # @return [Wavefront::Response]
+    def user_update(body, modify = true)
+      raise ArgumentError unless body.is_a?(Hash)
+
+      return api.post('user', body, 'application/json') unless modify
+
+      api.post('user',
+               hash_for_update(describe(id).response, body),
+               'application/json')
+    end
+
+    # GET /api/v2/account/user/{id}
+    # Retrieves a user by identifier (email address)
+    # @param id [String] ID of the proxy
+    # @return [Wavefront::Response]
+    #
+    def user_describe(id)
+      wf_user_id?(id)
+      api.get(['user', id].uri_concat)
+    end
+
+    # POST /api/v2/account/user/invite
+    # Invite user accounts with given user groups and permissions.
+    # @param body [Array[Hash]] array of hashes, each hash describing a user.
+    #   See API docs for more details. It is your responsibility to validate
+    #   the data which describes each user.
+    # @return [Wavefront::Response]
+    #
+    def user_invite(body)
+      raise ArgumentError unless body.is_a?(Array)
+      raise ArgumentError unless body.first.is_a?(Hash)
+
+      api.post('user/invite', body, 'application/json')
+    end
+
+    # POST /api/v2/account/validateAccounts
+    # Returns valid accounts (users and service accounts), also invalid
+    # identifiers from the given list
+    # @param id_list [Array[String]] list of user IDs
+    # @return [Wavefront::Response]
+    #
+    def validate_accounts(id_list)
+      raise ArgumentError unless id_list.is_a?(Array)
+
+      api.post('validateAccounts', id_list, 'application/json')
+    end
+
     private
 
     # @param id [String] ID of the user
@@ -198,5 +295,9 @@ module Wavefront
       wf_permission?(permission)
       api.post(['revoke', permission].uri_concat, id_list, 'application/json')
     end
+
+    def update_keys
+      %i[identifier groups userGroups roles ingestionPolicyId]
+    end
   end
 end

data/lib/wavefront-sdk/api_mixins/user.rb

@@ -35,6 +35,16 @@ module Wavefront
 
         list.each { |id| wf_account_id?(id) }
       end
+
+      # Validate a list of roles
+      # @param list [Array[String]] list of role IDs
+      # @raise Wavefront::Exception::InvalidRole
+      #
+      def validate_role_list(list)
+        raise ArgumentError unless list.is_a?(Array)
+
+        list.each { |id| wf_role_id?(id) }
+      end
     end
   end
 end

data/lib/wavefront-sdk/cloudintegration.rb

@@ -101,5 +101,32 @@ module Wavefront
       wf_cloudintegration_id?(id)
       api.post([id, 'undelete'].uri_concat)
     end
+
+    # POST /api/v2/cloudintegration/awsExternalIdCreate an external id
+    # @return [Wavefront::Response]
+    #
+    def create_aws_external_id
+      api.post('awsExternalId', nil, 'application/json')
+    end
+
+    # DELETE /api/v2/cloudintegration/awsExternalId/{id}DELETEs an external id
+    # that was created by Wavefront
+    # @param id [String] AWS external ID
+    # @return [Wavefront::Response]
+    #
+    def delete_aws_external_id(external_id)
+      wf_aws_external_id?(external_id)
+      api.delete(['awsExternalId', external_id].uri_concat)
+    end
+
+    # GET /api/v2/cloudintegration/awsExternalId/{id}GETs (confirms) a valid
+    # external id that was created by Wavefront
+    # @param id [String] AWS external ID
+    # @return [Wavefront::Response]
+    #
+    def confirm_aws_external_id(external_id)
+      wf_aws_external_id?(external_id)
+      api.get(['awsExternalId', external_id].uri_concat)
+    end
   end
 end

data/lib/wavefront-sdk/core/api_caller.rb

@@ -6,6 +6,7 @@ require 'addressable'
 require_relative 'response'
 require_relative '../defs/version'
 require_relative '../support/mixins'
+require_relative '../stdlib/time'
 
 module Wavefront
   #
@@ -78,14 +79,47 @@ module Wavefront
     # was cleaner to add this method. Parameters are same as #get.
     #
     def get_flat_params(path, query = {})
-      conn = mk_conn(path,
-                     {},
-                     request: {
-                       params_encoder: Faraday::FlatParamsEncoder
-                     },
-                     params: query)
+      make_call(flat_param_conn(path, query), :get)
+    end
+
+    # This is used by the Wavefront::Unstable::Spy methods to stream data.
+    # It prints to standard out.
+    # @param path [String] path to be appended to the net[:api_base] path.
+    # @param query [Hash] optional key-value pairs with will be made into a
+    #   query string
+    # @param opts [Hash] keys:
+    #   timestamp_chunks -- prints a timestamp before each chunk of streamed
+    #     data
+    #   timeout -- after approximately this many seconds, return. It will be
+    #     the first chunk *after* the given time
+    # @return
+    #
+    def get_stream(path, query = {}, opts = {})
+      conn = flat_param_conn(path, query)
+      verbosity(conn, :get, query)
+      stream_connection(conn, query, opts)
+    rescue Faraday::TimeoutError
+      raise Wavefront::Exception::NetworkTimeout
+    rescue StopIteration
+      nil
+    end
+
+    def stream_connection(conn, query, opts)
+      t_end = end_time(opts)
+
+      conn.get do |req|
+        req.params = query
+        req.options.on_data = proc do |chunk, _size|
+          raise StopIteration if t_end && Time.right_now >= t_end
 
-      make_call(conn, :get)
+          puts Time.now if opts[:timestamp_chunks]
+          puts chunk
+        end
+      end
+    end
+
+    def end_time(opts)
+      Time.right_now + opts[:timeout] if opts[:timeout]&.positive?
     end
 
     # Make a POST call to the Wavefront API and return the result as
@@ -240,5 +274,14 @@ module Wavefront
         end
       end
     end
+
+    def flat_param_conn(path, query)
+      mk_conn(path,
+              {},
+              request: {
+                params_encoder: Faraday::FlatParamsEncoder
+              },
+              params: query)
+    end
   end
 end

data/lib/wavefront-sdk/core/exception.rb

@@ -12,6 +12,7 @@ module Wavefront
     class InvalidAlertId < RuntimeError; end
     class InvalidAlertSeverity < RuntimeError; end
     class InvalidApiTokenId < RuntimeError; end
+    class InvalidAwsExternalId < RuntimeError; end
     class InvalidConfigFile < RuntimeError; end
     class InvalidCloudIntegrationId < RuntimeError; end
     class InvalidDashboardId < RuntimeError; end
@@ -28,6 +29,7 @@ module Wavefront
     class InvalidIntegrationId < RuntimeError; end
     class InvalidLinkTemplate < RuntimeError; end
     class InvalidMaintenanceWindowId < RuntimeError; end
+    class InvalidMonitoredClusterId < RuntimeError; end
     class InvalidMessageId < RuntimeError; end
     class InvalidMetricName < RuntimeError; end
     class InvalidMetricValue < RuntimeError; end
@@ -37,7 +39,9 @@ module Wavefront
     class InvalidPoint < RuntimeError; end
     class InvalidPrefixLength < RuntimeError; end
     class InvalidProxyId < RuntimeError; end
+    class InvalidRoleId < RuntimeError; end
     class InvalidRelativeTime < RuntimeError; end
+    class InvalidSamplingValue < RuntimeError; end
     class InvalidSavedSearchEntity < RuntimeError; end
     class InvalidSavedSearchId < RuntimeError; end
     class InvalidServiceAccountId < RuntimeError; end
@@ -51,6 +55,8 @@ module Wavefront
     class InvalidUserGroupId < RuntimeError; end
     class InvalidVersion < RuntimeError; end
     class InvalidWebhookId < RuntimeError; end
+    class MissingConfigProfile < RuntimeError; end
+    class NetworkTimeout < RuntimeError; end
     class NotImplemented < RuntimeError; end
     class SocketError < RuntimeError; end
     class UnparseableResponse < RuntimeError; end

data/lib/wavefront-sdk/credentials.rb

@@ -31,11 +31,16 @@ module Wavefront
     # @param options [Hash] keys may be 'file', which
     #   specifies a config file which will be loaded and parsed. If
     #   no file is supplied, those listed above will be used.;
-    #   and/or 'profile' which select a profile section from 'file'
+    #   and/or 'profile' which select a profile section from 'file'.
+    #   Specify the key :raise_noprofile to have an exception thrown
+    #   if a given profile cannot be found. Otherwise that is ignored and
+    #   options are built from other sources.
     #
     def initialize(options = {})
-      raw = load_from_file(cred_files(options), options[:profile] ||
-                           'default')
+      @raise_noprofile = options[:raise_on_no_profile] || false
+      raw = load_from_file(real_files(cred_files(options)),
+                           options[:profile] || 'default')
+
       populate(env_override(raw))
     end
 
@@ -80,31 +85,45 @@ module Wavefront
       end
     end
 
+    def real_files(files)
+      files.select { |f| f.exist? && f.file? }
+    end
+
     # @param files [Array][Pathname] a list of ini-style config files
     # @param profile [String] a profile name
+    # @param disallow_missing [Bool] whether or not to raise an exception if
+    #   we are given a profile but can't find it.
     # @return [Hash] the given profile from the given list of files.
     #   If multiple files match, the last one will be used
     #
     def load_from_file(files, profile = 'default')
       ret = {}
+      profiles_found = conf_files_found = 0
 
       files.each do |f|
-        next unless f.exist? && f.file?
-
         ret = load_profile(f, profile)
+        conf_files_found += 1
+        profiles_found += 1 unless ret.empty?
         ret[:file] = f
       end
 
+      raise_on_missing_profile(profile, profiles_found, conf_files_found)
       ret
     end
 
-    # Load in an (optionally) given section of an ini-style
-    # configuration file not there, we don't consider that an error.
+    def raise_on_missing_profile(profile, profiles_found, conf_files_found)
+      return true unless @raise_noprofile
+      return true unless profiles_found.zero? && conf_files_found.positive?
+
+      raise Wavefront::Exception::MissingConfigProfile, profile
+    end
+
+    # Load in an (optionally) given section of an ini-style configuration
+    # file. If the section is not there, we don't consider that an error.
     #
     # @param file [Pathname] the file to read
     # @param profile [String] the section in the config to read
-    # @return [Hash] options loaded from file. Each key becomes a
-    #   symbol
+    # @return [Hash] options loaded from file. Each key becomes a symbol
     #
     def load_profile(file, profile = 'default')
       IniFile.load(file)[profile].each_with_object({}) do |(k, v), memo|