Fix Kotlin Serialization

Author: JavierSegoviaCordoba

CHANGELOG.md

@@ -10,6 +10,8 @@
 
 ### Fixed
 
+- Kotlin Serialization
+
 ### Removed
 
 ### Updated

kopy-compiler/main/kotlin/com/javiersc/kotlin/kopy/compiler/fir/generation/FirKopyDeclarationGenerationExtension.kt

@@ -25,7 +25,9 @@ import com.javiersc.kotlin.kopy.compiler.kopyFunctionUpdateEachClassId
 import com.javiersc.kotlin.kopy.compiler.kopyOptInClassId
 import com.javiersc.kotlin.kopy.compiler.measureExecution
 import com.javiersc.kotlin.kopy.compiler.measureKey
+import com.javiersc.kotlin.kopy.compiler.serializableAnnotationClassId
 import com.javiersc.kotlin.kopy.compiler.setName
+import com.javiersc.kotlin.kopy.compiler.transientAnnotationClassId
 import com.javiersc.kotlin.kopy.compiler.underscoreAtomicName
 import com.javiersc.kotlin.kopy.compiler.updateEachName
 import com.javiersc.kotlin.kopy.compiler.updateName
@@ -130,16 +132,17 @@ internal class FirKopyDeclarationGenerationExtension(
 
             val atomicProperty: FirProperty =
                 createMemberProperty(
-                    owner = context.owner,
-                    key = Key,
-                    name = callableId.callableName,
-                    returnType = atomicRefType,
-                    config = {
-                        status { isOverride = false }
-                        modality = Modality.FINAL
-                        visibility = atomicVisibility
-                    },
-                )
+                        owner = context.owner,
+                        key = Key,
+                        name = callableId.callableName,
+                        returnType = atomicRefType,
+                        config = {
+                            status { isOverride = false }
+                            modality = Modality.FINAL
+                            visibility = atomicVisibility
+                        },
+                    )
+                    .apply { addTransientAnnotationIfOwnedBySerializable(owner = owner) }
             return listOf(atomicProperty.symbol)
         }
 
@@ -423,6 +426,17 @@ internal class FirKopyDeclarationGenerationExtension(
             ?.toFirTypeRef()
             ?.let(::createFirAnnotation)
 
+    private fun FirProperty.addTransientAnnotationIfOwnedBySerializable(owner: FirClassSymbol<*>) {
+        val isSerializable: Boolean = owner.hasAnnotation(serializableAnnotationClassId, session)
+        if (isSerializable) {
+            val replacedAnnotations: List<FirAnnotation> = buildList {
+                addAll(annotations)
+                createAnnotation(classId = transientAnnotationClassId)?.let(::add)
+            }
+            replaceAnnotations(replacedAnnotations)
+        }
+    }
+
     private fun calculateVisibility(classSymbol: FirClassSymbol<*>): Visibility {
         val visibility: Visibility =
             classSymbol.primaryConstructorSymbol(session)?.visibility ?: return Visibilities.Public

kopy-compiler/main/kotlin/com/javiersc/kotlin/kopy/compiler/values.kt

@@ -21,6 +21,8 @@ private const val iterable: String = "kotlin.collections.Iterable"
 private const val list: String = "kotlin.collections.List"
 private const val map: String = "kotlin.collections.map"
 private const val also: String = "kotlin.also"
+private const val serializableAnnotation: String = "kotlinx.serialization.Serializable"
+private const val transientAnnotation: String = "kotlinx.serialization.Transient"
 
 internal val kopyFqName: FqName = kopy.toFqName()
 internal val kopyFunctionCopyFqName: FqName = kopyFunctionCopy.toFqName()
@@ -39,6 +41,8 @@ internal val kopyFunctionUpdateEachClassId: ClassId = kopyFunctionUpdateEach.toC
 internal val atomicReferenceClassId: ClassId = atomicReference.toClassId()
 internal val iterableClassId: ClassId = iterable.toClassId()
 internal val listClassId: ClassId = list.toClassId()
+internal val serializableAnnotationClassId: ClassId = serializableAnnotation.toClassId()
+internal val transientAnnotationClassId: ClassId = transientAnnotation.toClassId()
 
 internal val underscoreAtomicName: Name = "_atomic".toName()
 internal val loadName: Name = "load".toName()