Hardcoding unwrap()s for now

smithy-lang · Sep 19, 2024 · da16064 · da16064
1 parent 75ecb60
commit da16064
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/...src/main/java/software/amazon/polymorph/smithyrust/generator/RustAwsSdkShimGenerator.java b/...src/main/java/software/amazon/polymorph/smithyrust/generator/RustAwsSdkShimGenerator.java
@@ -20,6 +20,7 @@
 import software.amazon.smithy.model.traits.BoxTrait;
 import software.amazon.smithy.model.traits.DefaultTrait;
 import software.amazon.smithy.model.traits.EnumTrait;
+import software.amazon.smithy.model.traits.RequiredTrait;
 import software.amazon.smithy.model.traits.UnitTypeTrait;
 
 import java.nio.file.Path;
@@ -419,6 +420,7 @@ protected TokenTree operationRequestFromDafnyFunction(
       "fluentMemberSetters",
       fluentMemberSettersForStructure(inputShape).toString()
     );
+    variables.put("unwrapIfNecessary", ".unwrap()");
 
     return TokenTree.of(
       evalTemplate(
@@ -432,7 +434,7 @@ pub fn from_dafny(
             $sdkCrate:L::operation::$snakeCaseOperationName:L::$sdkOperationInputStruct:L::builder()
                   $fluentMemberSetters:L
                   .build()
-                  .unwrap()
+                  $unwrapIfNecessary:L
         }
         """,
         variables
@@ -512,6 +514,9 @@ protected TokenTree operationResponseFromDafnyFunction(
       "fluentMemberSetters",
       fluentMemberSettersForStructure(outputShape).toString()
     );
+    variables.put("unwrapIfNecessary",
+      // TODO: Can't figure out why this one is fallible but not other similar output structures
+      outputShape.getId().equals(ShapeId.from("com.amazonaws.dynamodb#DescribeEndpointsResponse")) ? ".unwrap()" : "");
 
     return TokenTree.of(
       evalTemplate(
@@ -525,6 +530,8 @@ pub fn from_dafny(
             $sdkCrate:L::operation::$snakeCaseOperationName:L::$sdkOperationOutputStruct:L::builder()
                   $fluentMemberSetters:L
                   .build()
+                  $unwrapIfNecessary:L
+                  
         }
         """,
         variables