Fix memory leak in #161 and regression in #194 (#192)

Co-authored-by: David Roman <[email protected]>

Author: okmyself

Introspect/UIKitIntrospectionView.swift

@@ -63,9 +63,18 @@ public struct UIKitIntrospectionView<TargetViewType: UIView>: UIViewRepresentabl
         return view
     }
 
+    /// In some cases, UIView does not call `moveToWindowHandler`,
+    /// so the `moveToWindowHandler` call in updateUIView must be preserved.
     public func updateUIView(
         _ uiView: IntrospectionUIView,
         context: UIViewRepresentableContext<UIKitIntrospectionView>
-    ) {}
+    ) {
+        uiView.moveToWindowHandler?()
+    }
+    
+    /// Avoid memory leaks.
+    public static func dismantleUIView(_ uiView: IntrospectionUIView, coordinator: ()) {
+        uiView.moveToWindowHandler = nil
+    }
 }
 #endif

Introspect/UIKitIntrospectionViewController.swift

@@ -52,9 +52,18 @@ public struct UIKitIntrospectionViewController<TargetViewControllerType: UIViewC
         return viewController
     }
 
+    /// If you find that the `moveToWindowHandler` is not called in certain situations (which has not been discovered yet),
+    /// you can add code to call the `moveToWindowHandler` in the function body.
     public func updateUIViewController(
-        _ uiViewController: IntrospectionUIViewController,
+        _ viewController: IntrospectionUIViewController,
         context: UIViewControllerRepresentableContext<UIKitIntrospectionViewController>
-    ) {}
+    ) {
+        (viewController.view as? IntrospectionUIView)?.moveToWindowHandler?()
+    }
+
+    /// Avoid memory leaks.
+    public static func dismantleUIViewController(_ viewController: IntrospectionUIViewController, coordinator: ()) {
+        (viewController.view as? IntrospectionUIView)?.moveToWindowHandler = nil
+    }
 }
 #endif