feat(experimentalIdentityAndAuth): add `getEndpointParameterInstructi…

…ons()` to smithy context
smithy-lang · Sep 23, 2023 · b42b666 · b42b666
1 parent c7640d3
commit b42b666
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/...ipt-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java b/...ipt-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java
@@ -361,6 +361,12 @@ private void generateCommandMiddlewareResolver(String configType) {
                 writer.openBlock("[SMITHY_CONTEXT_KEY]: {", "},", () -> {
                     writer.write("service: $S,", service.toShapeId().getName());
                     writer.write("operation: $S,", operation.toShapeId().getName());
+                    if (service.hasTrait(EndpointRuleSetTrait.class)) {
+                        writer.openBlock("endpointRuleSet: {", "},", () -> {
+                            writer.write("getEndpointParameterInstructions: $L.getEndpointParameterInstructions,",
+                                symbol.getName());
+                        });
+                    }
                 });
             });
             writer.write("const { requestHandler } = configuration;");