inject sdk stream utility function to the stream response

Author: AllanZhengYP

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/SymbolVisitor.java

@@ -138,9 +138,11 @@ public String toMemberName(MemberShape shape) {
     @Override
     public Symbol blobShape(BlobShape shape) {
         if (shape.hasTrait(StreamingTrait.class)) {
-            // Note: `Readable` needs an import and a dependency.
-            return createSymbolBuilder(shape, "Readable | ReadableStream | Blob", null)
+            // Note: `Readable` and `SdkStream` needs an import and a dependency.
+            return createSymbolBuilder(shape, "SdkStream<Readable | ReadableStream | Blob>", null)
                     .addReference(Symbol.builder().name("Readable").namespace("stream", "/").build())
+                    .addReference(Symbol.builder().name("SdkStream").namespace(
+                            TypeScriptDependency.AWS_SDK_TYPES.packageName, "/").build())
                     .build();
         }
 

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpBindingProtocolGenerator.java

@@ -30,6 +30,7 @@
 import java.util.function.Consumer;
 import java.util.logging.Logger;
 import java.util.stream.Collectors;
+
 import software.amazon.smithy.codegen.core.CodegenException;
 import software.amazon.smithy.codegen.core.Symbol;
 import software.amazon.smithy.codegen.core.SymbolProvider;
@@ -2300,8 +2301,8 @@ private HttpBinding readPayload(
 
         // Handle streaming shapes differently.
         if (target.hasTrait(StreamingTrait.class)) {
-            // If payload is streaming, return raw low-level stream directly.
-            writer.write("const data: any = output.body;");
+            // If payload is streaming, return low-level stream with the stream utility functions mixin.
+            writer.write("const data: any = context.sdkStreamMixin(output.body);");
         } else if (target instanceof BlobShape) {
             // If payload is non-streaming Blob, only need to collect stream to binary data (Uint8Array).
             writer.write("const data: any = await collectBody(output.body, context);");

smithy-typescript-codegen/src/main/resources/META-INF/services/software.amazon.smithy.typescript.codegen.integration.TypeScriptIntegration

@@ -3,3 +3,4 @@ software.amazon.smithy.typescript.codegen.integration.AddChecksumRequiredDepende
 software.amazon.smithy.typescript.codegen.integration.AddDefaultsModeDependency
 software.amazon.smithy.typescript.codegen.integration.AddHttpApiKeyAuthPlugin
 software.amazon.smithy.typescript.codegen.integration.AddBaseServiceExceptionClass
+software.amazon.smithy.typescript.codegen.integration.SdkStreamUtilsMixin