chore: Add imports to model files where needed (#722)

smithy-lang · May 10, 2024 · c2d1346 · c2d1346
1 parent 1fbe84e
commit c2d1346
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/...n/software/amazon/smithy/swift/codegen/integration/serde/readwrite/WritingClosureUtils.kt b/...n/software/amazon/smithy/swift/codegen/integration/serde/readwrite/WritingClosureUtils.kt
@@ -8,6 +8,7 @@ import software.amazon.smithy.model.shapes.TimestampShape
 import software.amazon.smithy.model.traits.SparseTrait
 import software.amazon.smithy.model.traits.TimestampFormatTrait
 import software.amazon.smithy.model.traits.XmlFlattenedTrait
+import software.amazon.smithy.swift.codegen.SwiftDependency
 import software.amazon.smithy.swift.codegen.SwiftWriter
 import software.amazon.smithy.swift.codegen.integration.ProtocolGenerator
 import software.amazon.smithy.swift.codegen.integration.serde.json.TimestampUtils
@@ -32,6 +33,7 @@ class WritingClosureUtils(
     }
 
     private fun makeWritingClosure(shape: Shape, memberTimestampFormatTrait: TimestampFormatTrait?, isSparse: Boolean): String {
+        writer.addImport(SwiftDependency.SMITHY_READ_WRITE.target)
         val base = when (shape) {
             is MapShape -> {
                 val keyNodeInfo = nodeInfoUtils.nodeInfo(shape.key)

diff --git a/...in/software/amazon/smithy/swift/codegen/integration/serde/struct/StructEncodeGenerator.kt b/...in/software/amazon/smithy/swift/codegen/integration/serde/struct/StructEncodeGenerator.kt
@@ -22,7 +22,6 @@ import software.amazon.smithy.swift.codegen.integration.serde.readwrite.requestW
 import software.amazon.smithy.swift.codegen.integration.serde.readwrite.responseWireProtocol
 import software.amazon.smithy.swift.codegen.model.ShapeMetadata
 import software.amazon.smithy.swift.codegen.model.isError
-import java.lang.Exception
 
 class StructEncodeGenerator(
     private val ctx: ProtocolGenerator.GenerationContext,

diff --git a/...tlin/software/amazon/smithy/swift/codegen/integration/serde/union/UnionDecodeGenerator.kt b/...tlin/software/amazon/smithy/swift/codegen/integration/serde/union/UnionDecodeGenerator.kt
@@ -15,7 +15,7 @@ import software.amazon.smithy.swift.codegen.integration.serde.member.MemberShape
 import software.amazon.smithy.swift.codegen.integration.serde.readwrite.AWSProtocol
 import software.amazon.smithy.swift.codegen.integration.serde.readwrite.addImports
 import software.amazon.smithy.swift.codegen.integration.serde.readwrite.awsProtocol
-import software.amazon.smithy.swift.codegen.integration.serde.readwrite.requestWireProtocol
+import software.amazon.smithy.swift.codegen.integration.serde.readwrite.responseWireProtocol
 import software.amazon.smithy.swift.codegen.integration.serde.struct.readerSymbol
 import software.amazon.smithy.swift.codegen.model.getTrait
 
@@ -27,7 +27,7 @@ class UnionDecodeGenerator(
 ) : MemberShapeDecodeGenerator(ctx, writer, shapeContainingMembers) {
 
     fun render() {
-        writer.addImports(ctx.service.requestWireProtocol)
+        writer.addImports(ctx.service.responseWireProtocol)
         val symbol = ctx.symbolProvider.toSymbol(shapeContainingMembers)
         writer.openBlock(
             "static func read(from reader: \$N) throws -> \$N {", "}",