Process StringArray parameter in ClientContextParams

smithy-lang · Aug 27, 2024 · dee40fc · dee40fc
1 parent b04921e
commit dee40fc
Showing 1 changed file with 17 additions and 4 deletions.
diff --git a/...in/java/software/amazon/smithy/typescript/codegen/endpointsV2/RuleSetParameterFinder.java b/...in/java/software/amazon/smithy/typescript/codegen/endpointsV2/RuleSetParameterFinder.java
@@ -28,6 +28,7 @@
 import software.amazon.smithy.model.shapes.OperationShape;
 import software.amazon.smithy.model.shapes.ServiceShape;
 import software.amazon.smithy.model.shapes.Shape;
+import software.amazon.smithy.model.shapes.ShapeType;
 import software.amazon.smithy.rulesengine.traits.ClientContextParamsTrait;
 import software.amazon.smithy.rulesengine.traits.ContextParamTrait;
 import software.amazon.smithy.rulesengine.traits.EndpointRuleSetTrait;
@@ -68,10 +69,22 @@ public Map<String, String> getClientContextParams() {
         if (trait.isPresent()) {
             ClientContextParamsTrait clientContextParamsTrait = trait.get();
             clientContextParamsTrait.getParameters().forEach((name, definition) -> {
-                map.put(
-                    name,
-                    definition.getType().toString().toLowerCase() // "boolean" and "string" are directly usable in TS.
-                );
+                ShapeType shapeType = definition.getType();
+                if (shapeType.isShapeType(ShapeType.STRING) || shapeType.isShapeType(ShapeType.BOOLEAN)) {
+                    map.put(
+                        name,
+                        definition.getType().toString().toLowerCase() // "boolean" and "string" are directly usable in TS.
+                    );
+                } else if (shapeType.isShapeType(ShapeType.LIST)) {
+                    map.put(
+                        name,
+                        "string[]" // Only string lists are supported.
+                    );
+                } else {
+                    throw new RuntimeException("unexpected type "
+                        + definition.getType().toString()
+                        + " received as clientContextParam.");
+                }
             });
         }
         return map;