Forward oneof properties to backing storage

When we have a storage type present.
Faire · Aug 19, 2020 · 72c5e30 · 72c5e30
1 parent ea2d9f1
commit 72c5e30
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/wire-library/wire-swift-generator/src/main/java/com/squareup/wire/swift/SwiftGenerator.kt b/wire-library/wire-swift-generator/src/main/java/com/squareup/wire/swift/SwiftGenerator.kt
@@ -644,6 +644,13 @@ class SwiftGenerator private constructor(
       val enumName = oneOfEnumNames.getValue(oneOf)
 
       addProperty(PropertySpec.varBuilder(oneOf.name, enumName.makeOptional(), PUBLIC)
+          .getter(FunctionSpec.getterBuilder()
+              .addStatement("%N.%N", storageName, oneOf.name)
+              .build())
+          .setter(FunctionSpec.setterBuilder()
+              .addStatement("copyStorage()")
+              .addStatement("%N.%N = newValue", storageName, oneOf.name)
+              .build())
           .apply {
             if (oneOf.documentation.isNotBlank()) {
               addKdoc("%N\n", oneOf.documentation.sanitizeDoc())