Added Portal remove property.

sjohnsonaz · Sep 20, 2018 · 0ac8241 · 0ac8241
1 parent 9b9c560
commit 0ac8241
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/scripts/dom/Portal.tsx b/src/scripts/dom/Portal.tsx
@@ -3,6 +3,7 @@ import { Component } from "./Component";
 
 export interface IPortalProps {
     element: HTMLElement;
+    remove?: boolean;
 }
 
 export default class Portal extends Component<IPortalProps> {
@@ -16,7 +17,13 @@ export default class Portal extends Component<IPortalProps> {
 
     afterRender(node: Node, updated: boolean) {
         if (!this.props.element.contains(node)) {
-            this.props.element.appendChild(node);
+            if (!this.props.remove) {
+                this.props.element.appendChild(node);
+            }
+        } else {
+            if (this.props.remove) {
+                this.props.element.removeChild(node);
+            }
         }
     }