Use context parameter for the parentNode

features/share/api/src/main/kotlin/io/element/android/features/share/api/ShareEntryPoint.kt

@@ -17,7 +17,8 @@ import io.element.android.libraries.matrix.api.core.RoomId
 interface ShareEntryPoint : FeatureEntryPoint {
     data class Params(val intent: Intent)
 
-    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(roomIds: List<RoomId>)

features/share/impl/src/main/kotlin/io/element/android/features/share/impl/DefaultShareEntryPoint.kt

@@ -16,7 +16,8 @@ import io.element.android.libraries.di.SessionScope
 
 @ContributesBinding(SessionScope::class)
 class DefaultShareEntryPoint : ShareEntryPoint {
-    override fun createNode(parentNode: Node, buildContext: BuildContext, params: ShareEntryPoint.Params, callback: ShareEntryPoint.Callback): Node {
+    context(parentNode: Node)
+    override fun createNode(buildContext: BuildContext, params: ShareEntryPoint.Params, callback: ShareEntryPoint.Callback): Node {
         return parentNode.createNode<ShareNode>(
             buildContext = buildContext,
             plugins = listOf(

features/share/impl/src/main/kotlin/io/element/android/features/share/impl/ShareNode.kt

@@ -67,7 +67,6 @@ class ShareNode(
         }
 
         return roomSelectEntryPoint.createNode(
-            parentNode = this,
             buildContext = buildContext,
             params = RoomSelectEntryPoint.Params(mode = RoomSelectMode.Share),
             callback = callback,

features/share/impl/src/test/kotlin/io/element/android/features/share/impl/DefaultShareEntryPointTest.kt

@@ -38,8 +38,8 @@ class DefaultShareEntryPointTest {
                 plugins = plugins,
                 presenterFactory = { createSharePresenter() },
                 roomSelectEntryPoint = object : RoomSelectEntryPoint {
+                    context(parentNode: Node)
                     override fun createNode(
-                        parentNode: Node,
                         buildContext: BuildContext,
                         params: RoomSelectEntryPoint.Params,
                         callback: RoomSelectEntryPoint.Callback,
@@ -53,12 +53,13 @@ class DefaultShareEntryPointTest {
         val params = ShareEntryPoint.Params(
             intent = Intent(),
         )
-        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(ShareNode::class.java)
         assertThat(result.plugins).contains(ShareNode.Inputs(params.intent))
         assertThat(result.plugins).contains(callback)