#9 Update codegen: implement update method

PatilShreyas · Sep 8, 2022 · 570fb30 · 570fb30
1 parent c6e2833
commit 570fb30
Show file tree

Hide file tree

Showing 4 changed files with 55 additions and 19 deletions.
diff --git a/mutekt-codegen/src/main/kotlin/dev/shreyaspatil/mutekt/codegen/codebuild/ClassNames.kt b/mutekt-codegen/src/main/kotlin/dev/shreyaspatil/mutekt/codegen/codebuild/ClassNames.kt
@@ -18,15 +18,17 @@ package dev.shreyaspatil.mutekt.codegen.codebuild
 import com.google.devtools.ksp.symbol.KSPropertyDeclaration
 import com.squareup.kotlinpoet.ClassName
 import com.squareup.kotlinpoet.ParameterizedTypeName.Companion.parameterizedBy
+import com.squareup.kotlinpoet.asClassName
 import com.squareup.kotlinpoet.ksp.toTypeName
-import dev.shreyaspatil.mutekt.core.StateFlowable
+import dev.shreyaspatil.mutekt.core.AtomicExecutor
+import dev.shreyaspatil.mutekt.core.MutektMutableState
 
 object ClassNames {
 
-    fun stateFlowableOf(clazz: ClassName) = ClassName(
-        StateFlowable::class.java.packageName,
-        StateFlowable::class.simpleName!!
-    ).parameterizedBy(clazz)
+    fun mutektMutableState(
+        immutableInterface: ClassName,
+        mutableInterface: ClassName
+    ) = MutektMutableState::class.asClassName().parameterizedBy(immutableInterface, mutableInterface)
 
     fun stateFlowOf(clazz: ClassName) = ClassName(
         "kotlinx.coroutines.flow",
@@ -47,4 +49,6 @@ object ClassNames {
         "kotlinx.coroutines.flow",
         "FlowCollector"
     ).parameterizedBy(clazz)
+
+    fun atomicExecutor() = AtomicExecutor::class.asClassName()
 }
diff --git a/mutekt-codegen/src/main/kotlin/dev/shreyaspatil/mutekt/codegen/codebuild/MemberNames.kt b/mutekt-codegen/src/main/kotlin/dev/shreyaspatil/mutekt/codegen/codebuild/MemberNames.kt
@@ -21,4 +21,5 @@ object MemberNames {
     val COROUTINE_SCOPE = MemberName("kotlinx.coroutines", "coroutineScope")
     val COMBINE = MemberName("kotlinx.coroutines.flow", "combine")
     val STATE_IN = MemberName("kotlinx.coroutines.flow", "stateIn")
+    val FILTER_NOT_NULL = MemberName("kotlinx.coroutines.flow", "filterNotNull")
 }
diff --git a/...in/dev/shreyaspatil/mutekt/codegen/codebuild/mutableModel/MutableInterfaceModelBuilder.kt b/...in/dev/shreyaspatil/mutekt/codegen/codebuild/mutableModel/MutableInterfaceModelBuilder.kt
@@ -35,12 +35,14 @@ class MutableInterfaceModelBuilder(
 
     fun build() = TypeSpec.interfaceBuilder(interfaceName)
         .addSuperinterface(immutableStateInterface)
-        .addSuperinterface(ClassNames.stateFlowableOf(immutableStateInterface))
+        .addSuperinterface(ClassNames.mutektMutableState(immutableStateInterface, thisClass()))
         .addKdoc("Mutable state model for [%L]", immutableStateInterface.simpleName)
         .addMutableStateModelFields()
         .build()
 
     private fun TypeSpec.Builder.addMutableStateModelFields() = apply {
         publicProperties.eachToProperty { mutable().addModifiers(KModifier.OVERRIDE) }.forEach { addProperty(it) }
     }
+
+    private fun thisClass() = ClassName(immutableStateInterface.packageName, interfaceName)
 }
diff --git a/...v/shreyaspatil/mutekt/codegen/codebuild/mutableModel/impl/MutableClassModelImplBuilder.kt b/...v/shreyaspatil/mutekt/codegen/codebuild/mutableModel/impl/MutableClassModelImplBuilder.kt
@@ -19,9 +19,12 @@ import com.google.devtools.ksp.symbol.KSPropertyDeclaration
 import com.squareup.kotlinpoet.ClassName
 import com.squareup.kotlinpoet.FunSpec
 import com.squareup.kotlinpoet.KModifier
+import com.squareup.kotlinpoet.LambdaTypeName
 import com.squareup.kotlinpoet.NOTHING
+import com.squareup.kotlinpoet.ParameterSpec
 import com.squareup.kotlinpoet.PropertySpec
 import com.squareup.kotlinpoet.TypeSpec
+import com.squareup.kotlinpoet.asClassName
 import com.squareup.kotlinpoet.buildCodeBlock
 import com.squareup.kotlinpoet.ksp.toTypeName
 import com.squareup.kotlinpoet.withIndent
@@ -44,9 +47,11 @@ class MutableClassModelImplBuilder(
         .addModifiers(KModifier.PRIVATE)
         .addSuperinterface(mutableModelInterfaceName)
         .primaryConstructor()
+        .privateAtomicExecutor()
         .addPrivateStateFlowAndGetterSetterFields()
         .immutableStateFlowImpl()
         .overrideFunAsStateFlow()
+        .overrideFunUpdate()
         .build()
 
     private fun TypeSpec.Builder.primaryConstructor() = apply {
@@ -58,6 +63,15 @@ class MutableClassModelImplBuilder(
         )
     }
 
+    private fun TypeSpec.Builder.privateAtomicExecutor() = apply {
+        addProperty(
+            PropertySpec.builder("_atomicExecutor", ClassNames.atomicExecutor())
+                .initializer("%T()", ClassNames.atomicExecutor())
+                .addModifiers(KModifier.PRIVATE)
+                .build()
+        )
+    }
+
     /**
      * Creates MutableStateFlow property from [property]
      */
@@ -120,6 +134,24 @@ class MutableClassModelImplBuilder(
                 .build()
         )
     }
+
+    private fun TypeSpec.Builder.overrideFunUpdate() = apply {
+        addFunction(
+            FunSpec.builder("update")
+                .addModifiers(KModifier.OVERRIDE)
+                .addParameter(
+                    ParameterSpec.builder(
+                        name = "mutate",
+                        type = LambdaTypeName.get(
+                            receiver = mutableModelInterfaceName,
+                            returnType = Unit::class.asClassName()
+                        )
+                    ).build()
+                )
+                .addStatement("_atomicExecutor.execute { mutate() }")
+                .build()
+        )
+    }
 }
 
 /**
@@ -191,27 +223,24 @@ class StateFlowImplBuilder(
                         beginControlFlow("%M", MemberNames.COROUTINE_SCOPE)
                         withIndent {
                             beginControlFlow(
-                                "%M(%L) { params ->",
+                                "%M(_atomicExecutor.executing, %L) { params ->",
                                 MemberNames.COMBINE,
                                 publicProperties.joinToString { "_${it.simpleName.asString()}" }
                             )
 
                             withIndent {
-                                addStatement("Immutable%L(", type.simpleName)
+                                addStatement("val isUpdating = params[0] as Boolean")
 
-                                withIndent {
-                                    publicProperties.forEachIndexed { index, field ->
-                                        addStatement(
-                                            "%L = params[%L] as %L,",
-                                            field.simpleName.asString(),
-                                            index,
-                                            field.type.toTypeName()
-                                        )
-                                    }
-                                }
-                                addStatement(")")
+                                beginControlFlow("if (!isUpdating)")
+                                addStatement("value")
+                                endControlFlow()
+
+                                beginControlFlow("else")
+                                addStatement("null")
+                                endControlFlow()
                             }
                             endControlFlow()
+                            addStatement(".%M()", MemberNames.FILTER_NOT_NULL)
                             addStatement(".%M(this)", MemberNames.STATE_IN)
                             addStatement(".collect(collector)")
                         }