Parcelable: Support custom Parcelers in compiler plugin

2017-09-14 23:00:33 +03:00
parent 7dbefc1613
commit a63aca08f2
16 changed files with 603 additions and 38 deletions
@@ -0,0 +1,55 @@
+// WITH_RUNTIME
+
+@file:JvmName("TestKt")
+package test
+
+import kotlinx.android.parcel.*
+import android.os.Parcel
+import android.os.Parcelable
+import java.util.Arrays
+
+object Parceler1 : Parceler<Int> {
+    override fun create(parcel: Parcel) = -parcel.readInt()
+
+    override fun Int.write(parcel: Parcel, flags: Int) {
+        parcel.writeInt(this)
+    }
+}
+
+object Parceler2 : Parceler<Long> {
+    override fun create(parcel: Parcel) = parcel.readString().length.toLong()
+
+    override fun Long.write(parcel: Parcel, flags: Int) {
+        parcel.writeString("Abc")
+    }
+}
+
+@Parcelize
+data class Test(
+        val a: Int,
+        @TypeParceler<Int, Parceler1> val b: Int,
+        @TypeParceler<Long, Parceler2> val c: Long,
+        @TypeParceler<Int, Parceler1> val d: List<Int>,
+        @TypeParceler<Long, Parceler2> val e: LongArray
+)
+
+fun box() = parcelTest { parcel ->
+    val test = Test(5, 5, 50L, listOf(1, 2, 3), longArrayOf(3, 2, 1))
+    test.writeToParcel(parcel, 0)
+
+    val bytes = parcel.marshall()
+    parcel.unmarshall(bytes, 0, bytes.size)
+
+    val test2 = readFromParcel<Test>(parcel)
+
+    println(test.toString())
+    println(test2.toString())
+
+    with (test) {
+        assert(a == 5 && b == 5 && c == 50L && d == listOf(1, 2, 3) && Arrays.equals(e, longArrayOf(3, 2, 1)))
+    }
+
+    with (test2) {
+        assert(a == 5 && b == -5 && c == 3L && d == listOf(-1, -2, -3) && Arrays.equals(e, longArrayOf(3, 3, 3)))
+    }
+}
@@ -0,0 +1,48 @@
+// WITH_RUNTIME
+
+@file:JvmName("TestKt")
+package test
+
+import kotlinx.android.parcel.*
+import android.os.Parcel
+import android.os.Parcelable
+
+object Parceler1 : Parceler<String> {
+    override fun create(parcel: Parcel) = parcel.readInt().toString()
+
+    override fun String.write(parcel: Parcel, flags: Int) {
+        parcel.writeInt(length)
+    }
+}
+
+typealias Parceler2 = Parceler1
+
+object Parceler3 : Parceler<String> {
+    override fun create(parcel: Parcel) = parcel.readString().toUpperCase()
+
+    override fun String.write(parcel: Parcel, flags: Int) {
+        parcel.writeString(this)
+    }
+}
+
+@Parcelize
+@TypeParceler<String, Parceler2>
+data class Test(
+        val a: String,
+        @TypeParceler<String, Parceler1> val b: String,
+        @TypeParceler<String, Parceler3> val c: CharSequence,
+        val d: @WriteWith<Parceler3> String
+)
+
+fun box() = parcelTest { parcel ->
+    val test = Test("Abc", "Abc", "Abc", "Abc")
+    test.writeToParcel(parcel, 0)
+
+    val bytes = parcel.marshall()
+    parcel.unmarshall(bytes, 0, bytes.size)
+
+    val test2 = readFromParcel<Test>(parcel)
+
+    assert(test.a == "Abc" && test.b == "Abc" && test.c == "Abc" && test.d == "Abc")
+    assert(test2.a == "3" && test2.b == "3" && test2.c == "Abc" && test2.d == "ABC")
+}
@@ -0,0 +1,51 @@
+// WITH_RUNTIME
+
+@file:JvmName("TestKt")
+package test
+
+import kotlinx.android.parcel.*
+import android.os.Parcel
+import android.os.Parcelable
+
+object Parceler1 : Parceler<String> {
+    override fun create(parcel: Parcel) = parcel.readInt().toString()
+
+    override fun String.write(parcel: Parcel, flags: Int) {
+        parcel.writeInt(length)
+    }
+}
+
+object Parceler2 : Parceler<List<String>> {
+    override fun create(parcel: Parcel) = listOf(parcel.readString())
+
+    override fun List<String>.write(parcel: Parcel, flags: Int) {
+        parcel.writeString(this.joinToString(","))
+    }
+}
+
+@Parcelize
+data class Test(
+        val a: String,
+        val b: @WriteWith<Parceler1> String,
+        val c: List<@WriteWith<Parceler1> String>,
+        val d: @WriteWith<Parceler2> List<String>,
+        val e: @WriteWith<Parceler2> List<@WriteWith<Parceler1> String>
+)
+
+fun box() = parcelTest { parcel ->
+    val test = Test("Abc", "Abc", listOf("A", "bc"), listOf("A", "bc"), listOf("A", "bc"))
+    test.writeToParcel(parcel, 0)
+
+    val bytes = parcel.marshall()
+    parcel.unmarshall(bytes, 0, bytes.size)
+
+    val test2 = readFromParcel<Test>(parcel)
+
+    with (test) {
+        assert(a == "Abc" && b == "Abc" && c == listOf("A", "bc") && d == listOf("A", "bc") && e == listOf("A", "bc"))
+    }
+
+    with (test2) {
+        assert(a == "Abc" && b == "3" && c == listOf("A", "bc") && d == listOf("A,bc") && e == listOf("A,bc"))
+    }
+}