Fix repeated custom type can't be converted in message patcher (#41)

* Fix repeated custom type can't be converted in message patcher * Make ktlint happy
ButterCam · Jul 6, 2020 · 4f31749 · 4f31749
1 parent ea62519
commit 4f31749
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/sisyphus-protobuf/src/main/kotlin/com/bybutter/sisyphus/protobuf/MessagePatcher.kt b/lib/sisyphus-protobuf/src/main/kotlin/com/bybutter/sisyphus/protobuf/MessagePatcher.kt
@@ -44,7 +44,11 @@ class ValueNode : PatcherNode {
             else -> throw IllegalStateException()
         }
 
-        val propertyType = property.returnType.classifier as? KClass<*>
+        val propertyType = if (field.label == FieldDescriptorProto.Label.REPEATED) {
+            property.returnType.arguments.first().type?.classifier as? KClass<*>
+        } else {
+            property.returnType.classifier as? KClass<*>
+        }
         if (propertyType != null && propertyType.isSubclassOf(CustomProtoType::class)) {
             result = result.map {
                 val support = (propertyType.companionObjectInstance as CustomProtoTypeSupport<*, Any>)