Memory leaks fixes

Source/animation/types/animation_generators/Cache/AnimationCache.swift

@@ -19,7 +19,7 @@ class AnimationCache {
         }
     }
 
-    let sceneLayer: CALayer
+    weak var sceneLayer: CALayer?
     var layerCache = [Node: CachedLayer]()
 
     required init(sceneLayer: CALayer) {
@@ -74,10 +74,10 @@ class AnimationCache {
             layer.contentsScale = calculateAnimationScale(animation: animation)
 
             layer.setNeedsDisplay()
-            sceneLayer.addSublayer(layer)
+            sceneLayer?.addSublayer(layer)
 
             layerCache[node] = CachedLayer(layer: layer, animation: animation)
-            sceneLayer.setNeedsDisplay()
+            sceneLayer?.setNeedsDisplay()
 
             return layer
         }
@@ -138,7 +138,7 @@ class AnimationCache {
 
         let layer = cachedLayer.layer
         layerCache.removeValue(forKey: node)
-        sceneLayer.setNeedsDisplay()
+        sceneLayer?.setNeedsDisplay()
         layer.removeFromSuperlayer()
     }
 

Source/bindings/Variable.swift

@@ -36,7 +36,7 @@ open class Variable<T> {
     @discardableResult open func onChange(_ f: @escaping ((T) -> Void)) -> Disposable {
         let handler = ChangeHandler<T>(f)
         handlers.append(handler)
-        return Disposable { [weak self] in
+        return Disposable { [weak self, unowned handler] in
             guard let index = self?.handlers.index(of: handler) else {
                 return
             }

Source/model/scene/Group.swift

@@ -4,6 +4,11 @@ open class Group: Node {
     open var contents: [Node] {
         get { return contentsVar.value }
         set(val) {
+
+            contentsVar.value.forEach { subNode in
+                nodesMap.remove(subNode)
+            }
+
             contentsVar.value = val
 
             if let view = nodesMap.getView(self) {
@@ -35,6 +40,10 @@ open class Group: Node {
         self.contentsVar.node = self
     }
 
+    deinit {
+        nodesMap.remove(self)
+    }
+
     // Searching
 
     override public func nodeBy(tag: String) -> Node? {

Source/model/scene/Node.swift

@@ -85,7 +85,7 @@ open class Node: Drawable {
         let handler = ChangeHandler<TouchEvent>(f)
         touchPressedHandlers.append(handler)
 
-        return Disposable { [weak self]  in
+        return Disposable { [weak self, unowned handler]  in
             guard let index = self?.touchPressedHandlers.index(of: handler) else {
                 return
             }
@@ -98,7 +98,7 @@ open class Node: Drawable {
         let handler = ChangeHandler<TouchEvent>(f)
         touchMovedHandlers.append(handler)
 
-        return Disposable { [weak self] in
+        return Disposable { [weak self, unowned handler] in
             guard let index = self?.touchMovedHandlers.index(of: handler) else {
                 return
             }
@@ -111,7 +111,7 @@ open class Node: Drawable {
         let handler = ChangeHandler<TouchEvent>(f)
         touchReleasedHandlers.append(handler)
 
-        return Disposable { [weak self] in
+        return Disposable { [weak self, unowned handler] in
             guard let index = self?.touchReleasedHandlers.index(of: handler) else {
                 return
             }
@@ -128,7 +128,7 @@ open class Node: Drawable {
             tapHandlers[tapCount] = [handler]
         }
 
-        return Disposable { [weak self]  in
+        return Disposable { [weak self, unowned handler]  in
             guard let index = self?.tapHandlers[tapCount]?.index(of: handler) else {
                 return
             }
@@ -141,7 +141,7 @@ open class Node: Drawable {
         let handler = ChangeHandler<TapEvent>(f)
         longTapHandlers.append(handler)
 
-        return Disposable { [weak self] in
+        return Disposable { [weak self, unowned handler] in
             guard let index = self?.longTapHandlers.index(of: handler) else {
                 return
             }
@@ -154,7 +154,7 @@ open class Node: Drawable {
         let handler = ChangeHandler<PanEvent>(f)
         panHandlers.append(handler)
 
-        return Disposable { [weak self] in
+        return Disposable { [weak self, unowned handler] in
             guard let index = self?.panHandlers.index(of: handler) else {
                 return
             }
@@ -167,7 +167,7 @@ open class Node: Drawable {
         let handler = ChangeHandler<RotateEvent>(f)
         rotateHandlers.append(handler)
 
-        return Disposable { [weak self] in
+        return Disposable { [weak self, unowned handler] in
             guard let index = self?.rotateHandlers.index(of: handler) else {
                 return
             }
@@ -180,7 +180,7 @@ open class Node: Drawable {
         let handler = ChangeHandler<PinchEvent>(f)
         pinchHandlers.append(handler)
 
-        return Disposable { [weak self] in
+        return Disposable { [weak self, unowned handler] in
             guard let index = self?.pinchHandlers.index(of: handler) else {
                 return
             }

Source/render/GroupRenderer.swift

@@ -18,6 +18,10 @@ class GroupRenderer: NodeRenderer {
         updateRenderers()
     }
 
+    deinit {
+        dispose()
+    }
+
     override func doAddObservers() {
         super.doAddObservers()
 
@@ -53,7 +57,7 @@ class GroupRenderer: NodeRenderer {
     override func dispose() {
         super.dispose()
         renderers.forEach { renderer in renderer.dispose() }
-        renderers = []
+        renderers.removeAll()
     }
 
     private func updateRenderers() {

Source/render/ImageRenderer.swift

@@ -18,6 +18,10 @@ class ImageRenderer: NodeRenderer {
         super.init(node: image, view: view, animationCache: animationCache)
     }
 
+    deinit {
+        dispose()
+    }
+
     override func node() -> Node? {
         return image
     }

Source/render/NodeRenderer.swift

@@ -28,7 +28,7 @@ class NodeRenderer {
         self.view = view
         self.animationCache = animationCache
 
-        onNodeChange = {
+        onNodeChange = { [unowned node, weak view] in
             guard let isAnimating = animationCache?.isAnimating(node) else {
                 return
             }
@@ -43,6 +43,10 @@ class NodeRenderer {
         addObservers()
     }
 
+    deinit {
+        disposables.dispose()
+    }
+
     func doAddObservers() {
         guard let node = node() else {
             return

Source/render/ShapeRenderer.swift

@@ -15,6 +15,10 @@ class ShapeRenderer: NodeRenderer {
         super.init(node: shape, view: view, animationCache: animationCache)
     }
 
+    deinit {
+        dispose()
+    }
+
     override func node() -> Node? {
         return shape
     }

Source/render/TextRenderer.swift

@@ -14,6 +14,10 @@ class TextRenderer: NodeRenderer {
         super.init(node: text, view: view, animationCache: animationCache)
     }
 
+    deinit {
+        dispose()
+    }
+
     override func node() -> Node? {
         return text
     }

Source/views/NodesMap.swift

@@ -29,6 +29,12 @@ class NodesMap {
     func remove(_ node: Node) {
         map.removeObject(forKey: node)
         parentsMap.removeValue(forKey: node)
+
+        if let group = node as? Group {
+            group.contents.forEach { child in
+                self.remove(child)
+            }
+        }
     }
 
     // MARK: - Parents