lifecylle) Expand LifeCycle::anchorTo to latch onto other `LifeCycl…

…e`-likes
knockout · Nov 6, 2019 · 90f1223 · 90f1223
1 parent e86bb47
commit 90f1223
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/packages/lifecycle/src/LifeCycle.js b/packages/lifecycle/src/LifeCycle.js
@@ -73,9 +73,13 @@ export default class LifeCycle {
     this.addDisposable({ dispose })
   }
 
-  anchorTo (node) {
-    addDisposeCallback(node, () => this.dispose())
-    this[ANCHOR_NODE] = node
+  anchorTo (nodeOrLifeCycle) {
+    if ('addDisposable' in nodeOrLifeCycle) {
+      nodeOrLifeCycle.addDisposable.push(this)
+    } else {
+      this[ANCHOR_NODE] = nodeOrLifeCycle
+      addDisposeCallback(nodeOrLifeCycle, () => this[ANCHOR_NODE] === nodeOrLifeCycle && this.dispose())
+    }
   }
 
   dispose () {