Improve desensitizer: generate .serializer() for .toByteArray

2025-01-05 23:50:08 +08:00 · 2021-09-30 22:37:11 +01:00 · 2021-09-30 22:37:11 +01:00 · 6b63c323e5
commit 6b63c323e5
parent a2e602adfd
1 changed files with 6 additions and 1 deletions
--- a/mirai-core/src/commonTest/kotlin/notice/Desensitizer.kt
+++ b/mirai-core/src/commonTest/kotlin/notice/Desensitizer.kt
@ -195,9 +195,14 @@ private class DesensitizationVisitor(

                val result = instance.cast<NestedStructureDesensitizer<ProtocolStruct, ProtocolStruct>>()
                    .deserialize(desc.origin as ProtocolStruct, value.origin as ByteArray)
+                    ?: desc.origin

                val generate = ConstructorCallCodegenFacade.generateAndDesensitize(result)
-                PlainValueDesc(desc, "$generate.toByteArray()", value.origin)
+                PlainValueDesc(
+                    desc,
+                    "$generate.toByteArray(${result::class.qualifiedName}.serializer())",
+                    value.origin
+                )
            } else value
        }
    }