From 707c0c3f16e11a867cc09f5a5478fb792d5c30ab Mon Sep 17 00:00:00 2001
From: Ilya Grigorik <ilya@igvita.com>
Date: Wed, 20 Jul 2016 22:46:55 -0700
Subject: [PATCH] remove controller when video is removed from DOM

Closes #119.
---
 inject.js | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/inject.js b/inject.js
index 0770fdc..2cb0382 100644
--- a/inject.js
+++ b/inject.js
@@ -166,15 +166,23 @@ chrome.extension.sendMessage({}, function(response) {
         return false;
       }, true);
 
-      function checkForVideo(node, parent) {
+      function checkForVideo(node, parent, added) {
         if (node.nodeName === 'VIDEO') {
-          if (!node.classList.contains('vsc-initialized')) {
-            new tc.videoController(node, parent);
+          if (added) {
+            if (!node.classList.contains('vsc-initialized')) {
+              new tc.videoController(node, parent);
+            }
+          } else {
+            if (node.classList.contains('vsc-initialized')) {
+              var id = node.dataset['vscid'];
+              document.querySelector(`div[data-vscid="${id}"]`).remove();
+            }
           }
         } else if (node.children != undefined) {
           for (var i = 0; i < node.children.length; i++) {
             checkForVideo(node.children[i],
-                          node.children[i].parentNode || parent);
+                          node.children[i].parentNode || parent,
+                          added);
           }
         }
       }
@@ -183,8 +191,12 @@ chrome.extension.sendMessage({}, function(response) {
           forEach.call(mutation.addedNodes, function(node) {
             if (typeof node === "function")
               return;
-
-            checkForVideo(node, node.parentNode || mutation.target);
+            checkForVideo(node, node.parentNode || mutation.target, true);
+          })
+          forEach.call(mutation.removedNodes, function(node) {
+            if (typeof node === "function")
+              return;
+            checkForVideo(node, node.parentNode || mutation.target, false);
           })
         });
       });