Skip to content

Commit d41e953

Browse files
committed
chore(codegen): populate requestAlgorithmMemberHttpHeader
1 parent cb36048 commit d41e953

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

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

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@
3030
import software.amazon.smithy.model.knowledge.TopDownIndex;
3131
import software.amazon.smithy.model.shapes.OperationShape;
3232
import software.amazon.smithy.model.shapes.ServiceShape;
33+
import software.amazon.smithy.model.shapes.StructureShape;
34+
import software.amazon.smithy.model.traits.HttpHeaderTrait;
3335
import software.amazon.smithy.typescript.codegen.LanguageTarget;
3436
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
3537
import software.amazon.smithy.typescript.codegen.TypeScriptSettings;
@@ -201,6 +203,14 @@ private static Map<String, Object> getPluginFunctionParams(
201203
params.put("requestChecksumRequired", httpChecksumTrait.isRequestChecksumRequired());
202204
httpChecksumTrait.getRequestAlgorithmMember().ifPresent(requestAlgorithmMember -> {
203205
params.put("requestAlgorithmMember", requestAlgorithmMember);
206+
operation.getInput().ifPresent(inputShapeId -> {
207+
StructureShape inputShape = model.expectShape(inputShapeId, StructureShape.class);
208+
inputShape.getMember(requestAlgorithmMember).ifPresent(requestAlgorithmMemberShape -> {
209+
requestAlgorithmMemberShape.getTrait(HttpHeaderTrait.class).ifPresent(httpHeaderTrait -> {
210+
params.put("requestAlgorithmMemberHttpHeader", httpHeaderTrait.getValue());
211+
});
212+
});
213+
});
204214
});
205215
httpChecksumTrait.getRequestValidationModeMember().ifPresent(requestValidationModeMember -> {
206216
params.put("requestValidationModeMember", requestValidationModeMember);

0 commit comments

Comments
 (0)