Use context parameter for the parentNode

features/verifysession/api/src/main/kotlin/io/element/android/features/verifysession/api/IncomingVerificationEntryPoint.kt

@@ -19,7 +19,8 @@ interface IncomingVerificationEntryPoint : FeatureEntryPoint {
         val verificationRequest: VerificationRequest.Incoming,
     ) : NodeInputs
 
-    fun createNode(parentNode: Node, buildContext: BuildContext, params: Params, callback: Callback): Node
+    context(parentNode: Node)
+    fun createNode(buildContext: BuildContext, params: Params, callback: Callback): Node
 
     interface Callback : Plugin {
         fun onDone()

features/verifysession/api/src/main/kotlin/io/element/android/features/verifysession/api/OutgoingVerificationEntryPoint.kt

@@ -20,7 +20,8 @@ interface OutgoingVerificationEntryPoint : FeatureEntryPoint {
         val verificationRequest: VerificationRequest.Outgoing,
     ) : NodeInputs
 
-    fun createNode(parentNode: Node, buildContext: BuildContext, params: Params, callback: Callback): Node
+    context(parentNode: Node)
+    fun createNode(buildContext: BuildContext, params: Params, callback: Callback): Node
 
     interface Callback : Plugin {
         fun navigateToLearnMoreAboutEncryption()

features/verifysession/impl/src/main/kotlin/io/element/android/features/verifysession/impl/incoming/DefaultIncomingVerificationEntryPoint.kt

@@ -16,8 +16,8 @@ import io.element.android.libraries.architecture.createNode
 
 @ContributesBinding(AppScope::class)
 class DefaultIncomingVerificationEntryPoint : IncomingVerificationEntryPoint {
+    context(parentNode: Node)
     override fun createNode(
-        parentNode: Node,
         buildContext: BuildContext,
         params: IncomingVerificationEntryPoint.Params,
         callback: IncomingVerificationEntryPoint.Callback,

features/verifysession/impl/src/main/kotlin/io/element/android/features/verifysession/impl/outgoing/DefaultOutgoingVerificationEntryPoint.kt

@@ -16,8 +16,8 @@ import io.element.android.libraries.architecture.createNode
 
 @ContributesBinding(AppScope::class)
 class DefaultOutgoingVerificationEntryPoint : OutgoingVerificationEntryPoint {
+    context(parentNode: Node)
     override fun createNode(
-        parentNode: Node,
         buildContext: BuildContext,
         params: OutgoingVerificationEntryPoint.Params,
         callback: OutgoingVerificationEntryPoint.Callback,

features/verifysession/impl/src/test/kotlin/io/element/android/features/verifysession/impl/incoming/DefaultIncomingVerificationEntryPointTest.kt

@@ -37,12 +37,13 @@ class DefaultIncomingVerificationEntryPointTest {
         val params = IncomingVerificationEntryPoint.Params(
             verificationRequest = anIncomingSessionVerificationRequest()
         )
-        val result = entryPoint.createNode(
-            parentNode = parentNode,
-            buildContext = BuildContext.root(null),
-            params = params,
-            callback = callback,
-        )
+        val result = with(parentNode) {
+            entryPoint.createNode(
+                buildContext = BuildContext.root(null),
+                params = params,
+                callback = callback,
+            )
+        }
         assertThat(result).isInstanceOf(IncomingVerificationNode::class.java)
         assertThat(result.plugins).contains(params)
         assertThat(result.plugins).contains(callback)

features/verifysession/impl/src/test/kotlin/io/element/android/features/verifysession/impl/outgoing/DefaultOutgoingVerificationEntryPointTest.kt

@@ -42,12 +42,13 @@ class DefaultOutgoingVerificationEntryPointTest {
             showDeviceVerifiedScreen = true,
             verificationRequest = anOutgoingSessionVerificationRequest(),
         )
-        val result = entryPoint.createNode(
-            parentNode = parentNode,
-            buildContext = BuildContext.root(null),
-            params = params,
-            callback = callback,
-        )
+        val result = with(parentNode) {
+            entryPoint.createNode(
+                buildContext = BuildContext.root(null),
+                params = params,
+                callback = callback,
+            )
+        }
         assertThat(result).isInstanceOf(OutgoingVerificationNode::class.java)
         assertThat(result.plugins).contains(params)
         assertThat(result.plugins).contains(callback)