weakref for control events

Author: Alex009

widgets/src/iosMain/kotlin/dev/icerock/moko/widgets/core/factory/ButtonWithIconViewFactory.kt

@@ -100,7 +100,7 @@ actual class ButtonWithIconViewFactory actual constructor(
 
         button.setEventHandler(
             controlEvent = UIControlEventTouchUpInside,
-            action = widget.onTap
+            action = { widget.onTap }
         )
 
         val contentAttribute: UISemanticContentAttribute

widgets/src/iosMain/kotlin/dev/icerock/moko/widgets/core/utils/UIControlExt.kt

@@ -20,8 +20,13 @@ import platform.objc.OBJC_ASSOCIATION_RETAIN
 import platform.objc.objc_setAssociatedObject
 import kotlin.native.ref.WeakReference
 
-fun UIControl.setEventHandler(controlEvent: UIControlEvents, action: () -> Unit) {
-    val target = LambdaTarget(action)
+fun <V: UIControl> V.setEventHandler(controlEvent: UIControlEvents, action: (V) -> Unit) {
+    val weakReference: WeakReference<V> = WeakReference(this)
+    val target = LambdaTarget {
+        val strongRef: V = weakReference.get() ?: return@LambdaTarget
+
+        action(strongRef)
+    }
 
     addTarget(
         target = target,