Wire up signers

Author: nateprewitt

codegen/core/src/main/java/software/amazon/smithy/python/codegen/integrations/RestJsonProtocolGenerator.java

@@ -397,13 +397,25 @@ public void wrapInputStream(GenerationContext context, PythonWriter writer) {
         writer.addImport("smithy_core.aio.types", "AsyncBytesReader");
         writer.addImport("smithy_core.types", "TimestampFormat");
         writer.addImport("aws_event_stream.aio", "AWSInputEventStream");
+        writer.addImport("aws_sdk_signers", "AsyncEventSigner");
         writer.write(
                 """
+                        # TODO - Move this out of the RestJSON generator
+                        ctx = request_context
+                        signer_properties = ctx.properties.get("signer_properties")  # type: ignore
+                        identity = ctx.properties.get("identity")  # type: ignore
+                        signature = ctx.properties.get("signature")  # type: ignore
+                        signer = AsyncEventSigner(
+                            signing_properties=signer_properties,  # type: ignore
+                            identity=identity,  # type: ignore
+                            initial_signature=signature,  # type: ignore
+                        )
                         codec = JSONCodec(default_timestamp_format=TimestampFormat.EPOCH_SECONDS)
                         return AWSInputEventStream[Any, Any](
                             payload_codec=codec,
                             awaitable_output=awaitable_output,
                             async_writer=request_context.transport_request.body,  # type: ignore
+                            signer=signer,  # type: ignore
                         )
                         """);
     }
@@ -439,15 +451,28 @@ public void wrapDuplexStream(GenerationContext context, PythonWriter writer) {
         writer.addImport("smithy_core.aio.types", "AsyncBytesReader");
         writer.addImport("smithy_core.types", "TimestampFormat");
         writer.addImport("aws_event_stream.aio", "AWSDuplexEventStream");
+        writer.addImport("aws_sdk_signers", "AsyncEventSigner");
         writer.write(
                 """
+                        # TODO - Move this out of the RestJSON generator
+                        ctx = request_context
+                        signer_properties = ctx.properties.get("signer_properties")  # type: ignore
+                        identity = ctx.properties.get("identity")  # type: ignore
+                        signature = ctx.properties.get("signature")  # type: ignore
+                        signer = AsyncEventSigner(
+                            signing_properties=signer_properties,  # type: ignore
+                            identity=identity,  # type: ignore
+                            initial_signature=signature,  # type: ignore
+                        )
+
                         codec = JSONCodec(default_timestamp_format=TimestampFormat.EPOCH_SECONDS)
                         return AWSDuplexEventStream[Any, Any, Any](
                             payload_codec=codec,
                             async_writer=request_context.transport_request.body,  # type: ignore
                             awaitable_output=awaitable_output,
                             awaitable_response=response_future,
                             deserializer=event_deserializer,  # type: ignore
+                            signer=signer,  # type: ignore
                         )
                         """);
     }

packages/aws-sdk-signers/src/aws_sdk_signers/signers.py

@@ -353,7 +353,7 @@ def _should_sha256_sign_payload(
         if request.destination.scheme != "https":
             return True
 
-        return signing_properties.get("payload_signing_enabled", True)
+        return signing_properties.get("payload_signing_enabled", False)
 
     def _format_canonical_payload(
         self,
@@ -364,7 +364,7 @@ def _format_canonical_payload(
         payload_hash = self._compute_payload_hash(
             request=request, signing_properties=signing_properties
         )
-        if signing_properties.get("content_checksum_enabled", False):
+        if signing_properties.get("content_checksum_enabled", True):
             request.fields.set_field(
                 Field(name="X-Amz-Content-SHA256", values=[payload_hash])
             )
@@ -721,7 +721,7 @@ async def _should_sha256_sign_payload(
         if request.destination.scheme != "https":
             return True
 
-        return signing_properties.get("payload_signing_enabled", True)
+        return signing_properties.get("payload_signing_enabled", False)
 
     async def _format_canonical_payload(
         self,
@@ -732,7 +732,7 @@ async def _format_canonical_payload(
         payload_hash = await self._compute_payload_hash(
             request=request, signing_properties=signing_properties
         )
-        if signing_properties.get("content_checksum_enabled", False):
+        if signing_properties.get("content_checksum_enabled", True):
             request.fields.set_field(
                 Field(name="X-Amz-Content-SHA256", values=[payload_hash])
             )