aws-sdk-ec2 1.341.0 → 1.343.0

data/lib/aws-sdk-ec2/plugins/copy_encrypted_snapshot.rb

@@ -21,6 +21,10 @@ module Aws
 
           def call(context)
             params = context.params
+            unless params[:source_region]
+              raise ArgumentError,
+                    'Source region is required for copy_snapshot'
+            end
             params[:destination_region] = context.config.region
             params[:presigned_url] = presigned_url(context, params)
             @handler.call(context)
@@ -33,25 +37,26 @@ module Aws
             param_list.set('Action', 'CopySnapshot')
             param_list.set('DestinationRegion', context.config.region)
             param_list.set('Version', context.config.api.version)
-            Aws::Query::EC2ParamBuilder.new(param_list).apply(context.operation.input, params)
+            Aws::Query::EC2ParamBuilder.new(param_list).apply(
+              context.operation.input, params
+            )
 
-            signer = Aws::Sigv4::Signer.new(
-              service: 'ec2',
+            endpoint_params = Aws::EC2::EndpointParameters.new(
               region: params[:source_region],
-              credentials_provider: context.config.credentials
+              use_dual_stack: context.config.use_dualstack_endpoint,
+              use_fips: context.config.use_fips_endpoint
             )
-            url = Aws::Partitions::EndpointProvider.resolve(
-              signer.region, 'ec2', 'regional',
-              {
-                dualstack: context.config.use_dualstack_endpoint,
-                fips: context.config.use_fips_endpoint
-              }
+            endpoint = context.config.endpoint_provider
+                              .resolve_endpoint(endpoint_params)
+            auth_scheme = Aws::Endpoints.resolve_auth_scheme(context, endpoint)
+
+            signer = Aws::Plugins::Sign.signer_for(
+              auth_scheme, context.config, params[:source_region]
             )
-            url += "?#{param_list.to_s}"
 
             signer.presign_url(
               http_method: 'GET',
-              url: url,
+              url: "#{endpoint.url}?#{param_list}",
               body: '',
               expires_in: 3600
             ).to_s