aws-sdk-kms 1.88.0 → 1.105.0

data/lib/aws-sdk-kms/plugins/endpoints.rb

@@ -15,11 +15,11 @@ module Aws::KMS
         :endpoint_provider,
         doc_type: 'Aws::KMS::EndpointProvider',
         rbs_type: 'untyped',
-        docstring: 'The endpoint provider used to resolve endpoints. Any '\
-                   'object that responds to `#resolve_endpoint(parameters)` '\
-                   'where `parameters` is a Struct similar to '\
-                   '`Aws::KMS::EndpointParameters`'
-      ) do |cfg|
+        docstring: <<~DOCS) do |_cfg|
+The endpoint provider used to resolve endpoints. Any object that responds to
+`#resolve_endpoint(parameters)` where `parameters` is a Struct similar to
+`Aws::KMS::EndpointParameters`.
+        DOCS
         Aws::KMS::EndpointProvider.new
       end
 
@@ -27,7 +27,7 @@ module Aws::KMS
       class Handler < Seahorse::Client::Handler
         def call(context)
           unless context[:discovered_endpoint]
-            params = parameters_for_operation(context)
+            params = Aws::KMS::Endpoints.parameters_for_operation(context)
             endpoint = context.config.endpoint_provider.resolve_endpoint(params)
 
             context.http_request.endpoint = endpoint.url
@@ -40,11 +40,23 @@ module Aws::KMS
           context[:auth_scheme] =
             Aws::Endpoints.resolve_auth_scheme(context, endpoint)
 
-          @handler.call(context)
+          with_metrics(context) { @handler.call(context) }
         end
 
         private
 
+        def with_metrics(context, &block)
+          metrics = []
+          metrics << 'ENDPOINT_OVERRIDE' unless context.config.regional_endpoint
+          if context[:auth_scheme] && context[:auth_scheme]['name'] == 'sigv4a'
+            metrics << 'SIGV4A_SIGNING'
+          end
+          if context.config.credentials&.credentials&.account_id
+            metrics << 'RESOLVED_ACCOUNT_ID'
+          end
+          Aws::Plugins::UserAgent.metric(*metrics, &block)
+        end
+
         def apply_endpoint_headers(context, headers)
           headers.each do |key, values|
             value = values
@@ -55,117 +67,6 @@ module Aws::KMS
             context.http_request.headers[key] = value
           end
         end
-
-        def parameters_for_operation(context)
-          case context.operation_name
-          when :cancel_key_deletion
-            Aws::KMS::Endpoints::CancelKeyDeletion.build(context)
-          when :connect_custom_key_store
-            Aws::KMS::Endpoints::ConnectCustomKeyStore.build(context)
-          when :create_alias
-            Aws::KMS::Endpoints::CreateAlias.build(context)
-          when :create_custom_key_store
-            Aws::KMS::Endpoints::CreateCustomKeyStore.build(context)
-          when :create_grant
-            Aws::KMS::Endpoints::CreateGrant.build(context)
-          when :create_key
-            Aws::KMS::Endpoints::CreateKey.build(context)
-          when :decrypt
-            Aws::KMS::Endpoints::Decrypt.build(context)
-          when :delete_alias
-            Aws::KMS::Endpoints::DeleteAlias.build(context)
-          when :delete_custom_key_store
-            Aws::KMS::Endpoints::DeleteCustomKeyStore.build(context)
-          when :delete_imported_key_material
-            Aws::KMS::Endpoints::DeleteImportedKeyMaterial.build(context)
-          when :derive_shared_secret
-            Aws::KMS::Endpoints::DeriveSharedSecret.build(context)
-          when :describe_custom_key_stores
-            Aws::KMS::Endpoints::DescribeCustomKeyStores.build(context)
-          when :describe_key
-            Aws::KMS::Endpoints::DescribeKey.build(context)
-          when :disable_key
-            Aws::KMS::Endpoints::DisableKey.build(context)
-          when :disable_key_rotation
-            Aws::KMS::Endpoints::DisableKeyRotation.build(context)
-          when :disconnect_custom_key_store
-            Aws::KMS::Endpoints::DisconnectCustomKeyStore.build(context)
-          when :enable_key
-            Aws::KMS::Endpoints::EnableKey.build(context)
-          when :enable_key_rotation
-            Aws::KMS::Endpoints::EnableKeyRotation.build(context)
-          when :encrypt
-            Aws::KMS::Endpoints::Encrypt.build(context)
-          when :generate_data_key
-            Aws::KMS::Endpoints::GenerateDataKey.build(context)
-          when :generate_data_key_pair
-            Aws::KMS::Endpoints::GenerateDataKeyPair.build(context)
-          when :generate_data_key_pair_without_plaintext
-            Aws::KMS::Endpoints::GenerateDataKeyPairWithoutPlaintext.build(context)
-          when :generate_data_key_without_plaintext
-            Aws::KMS::Endpoints::GenerateDataKeyWithoutPlaintext.build(context)
-          when :generate_mac
-            Aws::KMS::Endpoints::GenerateMac.build(context)
-          when :generate_random
-            Aws::KMS::Endpoints::GenerateRandom.build(context)
-          when :get_key_policy
-            Aws::KMS::Endpoints::GetKeyPolicy.build(context)
-          when :get_key_rotation_status
-            Aws::KMS::Endpoints::GetKeyRotationStatus.build(context)
-          when :get_parameters_for_import
-            Aws::KMS::Endpoints::GetParametersForImport.build(context)
-          when :get_public_key
-            Aws::KMS::Endpoints::GetPublicKey.build(context)
-          when :import_key_material
-            Aws::KMS::Endpoints::ImportKeyMaterial.build(context)
-          when :list_aliases
-            Aws::KMS::Endpoints::ListAliases.build(context)
-          when :list_grants
-            Aws::KMS::Endpoints::ListGrants.build(context)
-          when :list_key_policies
-            Aws::KMS::Endpoints::ListKeyPolicies.build(context)
-          when :list_key_rotations
-            Aws::KMS::Endpoints::ListKeyRotations.build(context)
-          when :list_keys
-            Aws::KMS::Endpoints::ListKeys.build(context)
-          when :list_resource_tags
-            Aws::KMS::Endpoints::ListResourceTags.build(context)
-          when :list_retirable_grants
-            Aws::KMS::Endpoints::ListRetirableGrants.build(context)
-          when :put_key_policy
-            Aws::KMS::Endpoints::PutKeyPolicy.build(context)
-          when :re_encrypt
-            Aws::KMS::Endpoints::ReEncrypt.build(context)
-          when :replicate_key
-            Aws::KMS::Endpoints::ReplicateKey.build(context)
-          when :retire_grant
-            Aws::KMS::Endpoints::RetireGrant.build(context)
-          when :revoke_grant
-            Aws::KMS::Endpoints::RevokeGrant.build(context)
-          when :rotate_key_on_demand
-            Aws::KMS::Endpoints::RotateKeyOnDemand.build(context)
-          when :schedule_key_deletion
-            Aws::KMS::Endpoints::ScheduleKeyDeletion.build(context)
-          when :sign
-            Aws::KMS::Endpoints::Sign.build(context)
-          when :tag_resource
-            Aws::KMS::Endpoints::TagResource.build(context)
-          when :untag_resource
-            Aws::KMS::Endpoints::UntagResource.build(context)
-          when :update_alias
-            Aws::KMS::Endpoints::UpdateAlias.build(context)
-          when :update_custom_key_store
-            Aws::KMS::Endpoints::UpdateCustomKeyStore.build(context)
-          when :update_key_description
-            Aws::KMS::Endpoints::UpdateKeyDescription.build(context)
-          when :update_primary_region
-            Aws::KMS::Endpoints::UpdatePrimaryRegion.build(context)
-          when :verify
-            Aws::KMS::Endpoints::Verify.build(context)
-          when :verify_mac
-            Aws::KMS::Endpoints::VerifyMac.build(context)
-          end
-        end
       end
 
       def add_handlers(handlers, _config)