Skip to content

Commit 6c05df3

Browse files
authored
feat(codegen): add getAwsChunkedEncodingStream to config (#3349)
1 parent 50b48fe commit 6c05df3

File tree

2 files changed

+30
-6
lines changed

2 files changed

+30
-6
lines changed

codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AddHttpChecksumDependency.java

Lines changed: 23 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,11 @@ public void addConfigInterfaceFields(
7676
+ "that computes SHA1 hashes.\n"
7777
+ "@internal");
7878
writer.write("sha1?: __HashConstructor;\n");
79+
80+
writer.addImport("GetAwsChunkedEncodingStream", "GetAwsChunkedEncodingStream", "@aws-sdk/types");
81+
writer.writeDocs("A function that returns Readable Stream which follows aws-chunked encoding stream.\n"
82+
+ "@internal");
83+
writer.write("getAwsChunkedEncodingStream?: GetAwsChunkedEncodingStream;\n");
7984
}
8085

8186
@Override
@@ -93,9 +98,9 @@ public Map<String, Consumer<TypeScriptWriter>> getRuntimeConfigWriters(
9398
case NODE:
9499
return MapUtils.of(
95100
"streamHasher", writer -> {
96-
writer.addDependency(TypeScriptDependency.STREAM_HASHER_NODE);
101+
writer.addDependency(AwsDependency.STREAM_HASHER_NODE);
97102
writer.addImport("readableStreamHasher", "streamHasher",
98-
TypeScriptDependency.STREAM_HASHER_NODE.packageName);
103+
AwsDependency.STREAM_HASHER_NODE.packageName);
99104
writer.write("streamHasher");
100105
},
101106
"md5", writer -> {
@@ -109,26 +114,38 @@ public Map<String, Consumer<TypeScriptWriter>> getRuntimeConfigWriters(
109114
writer.addImport("HashConstructor", "__HashConstructor",
110115
TypeScriptDependency.AWS_SDK_TYPES.packageName);
111116
writer.write("Hash.bind(null, \"sha1\")");
117+
},
118+
"getAwsChunkedEncodingStream", writer -> {
119+
writer.addDependency(AwsDependency.UTIL_STREAM_NODE);
120+
writer.addImport("getAwsChunkedEncodingStream", "getAwsChunkedEncodingStream",
121+
AwsDependency.UTIL_STREAM_NODE.packageName);
122+
writer.write("getAwsChunkedEncodingStream");
112123
}
113124
);
114125
case BROWSER:
115126
return MapUtils.of(
116127
"streamHasher", writer -> {
117-
writer.addDependency(TypeScriptDependency.STREAM_HASHER_BROWSER);
128+
writer.addDependency(AwsDependency.STREAM_HASHER_BROWSER);
118129
writer.addImport("blobHasher", "streamHasher",
119-
TypeScriptDependency.STREAM_HASHER_BROWSER.packageName);
130+
AwsDependency.STREAM_HASHER_BROWSER.packageName);
120131
writer.write("streamHasher");
121132
},
122133
"md5", writer -> {
123-
writer.addDependency(TypeScriptDependency.MD5_BROWSER);
124-
writer.addImport("Md5", "Md5", TypeScriptDependency.MD5_BROWSER.packageName);
134+
writer.addDependency(AwsDependency.MD5_BROWSER);
135+
writer.addImport("Md5", "Md5", AwsDependency.MD5_BROWSER.packageName);
125136
writer.write("Md5");
126137
},
127138
"sha1", writer -> {
128139
writer.addDependency(AwsDependency.AWS_CRYPTO_SHA1_BROWSER);
129140
writer.addImport("Sha1",
130141
"Sha1", AwsDependency.AWS_CRYPTO_SHA1_BROWSER.packageName);
131142
writer.write("Sha1");
143+
},
144+
"getAwsChunkedEncodingStream", writer -> {
145+
writer.addDependency(AwsDependency.UTIL_STREAM_BROWSER);
146+
writer.addImport("getAwsChunkedEncodingStream", "getAwsChunkedEncodingStream",
147+
AwsDependency.UTIL_STREAM_BROWSER.packageName);
148+
writer.write("getAwsChunkedEncodingStream");
132149
}
133150
);
134151
default:

codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AwsDependency.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,13 @@ public enum AwsDependency implements SymbolDependencyContainer {
7272
AWS_SDK_UTIL_USER_AGENT_NODE(NORMAL_DEPENDENCY, "@aws-sdk/util-user-agent-node"),
7373
MIDDLEWARE_ENDPOINT_DISCOVERY(NORMAL_DEPENDENCY, "@aws-sdk/middleware-endpoint-discovery"),
7474
AWS_CRYPTO_SHA1_BROWSER(NORMAL_DEPENDENCY, "@aws-crypto/sha1-browser", "2.0.0"),
75+
76+
// Conditionally added when httpChecksum trait exists
77+
MD5_BROWSER(NORMAL_DEPENDENCY, "@aws-sdk/md5-js"),
78+
STREAM_HASHER_NODE(NORMAL_DEPENDENCY, "@aws-sdk/hash-stream-node"),
79+
STREAM_HASHER_BROWSER(NORMAL_DEPENDENCY, "@aws-sdk/hash-blob-browser"),
80+
UTIL_STREAM_NODE(NORMAL_DEPENDENCY, "@aws-sdk/util-stream-node"),
81+
UTIL_STREAM_BROWSER(NORMAL_DEPENDENCY, "@aws-sdk/util-stream-browser"),
7582
FLEXIBLE_CHECKSUMS_MIDDLEWARE(NORMAL_DEPENDENCY, "@aws-sdk/middleware-flexible-checksums");
7683

7784
public final String packageName;

0 commit comments

Comments
 (0)