Ugly fix for our specific use.

Gutenberg-Technology · Apr 12, 2016 · 8b3d6a4 · phloe · Apr 12, 2016 · Vadorequest
1 parent 69e872a
commit 8b3d6a4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,3 +3,4 @@ node_modules
 coverage
 lib
 dist
+.idea
diff --git a/src/HTML5Backend.js b/src/HTML5Backend.js
@@ -34,25 +34,26 @@ export default class HTML5Backend {
     this.endDragNativeItem = this.endDragNativeItem.bind(this);
   }
 
-  setup() {
-    if (typeof window === 'undefined') {
-      return;
+  setup(target = document.querySelector("iframe").contentWindow) {
+    if (typeof target === 'undefined') {
+      target = window;
     }
 
     if (this.constructor.isSetUp) {
       throw new Error('Cannot have two HTML5 backends at the same time.');
     }
     this.constructor.isSetUp = true;
-    this.addEventListeners(window);
+    this.addEventListeners(target);
   }
 
-  teardown() {
-    if (typeof window === 'undefined') {
-      return;
+  // TODO - Maybe save the target and use the same target for both setup and teardown to ensure the events are correctly removed in case the the target isn't provided.
+  teardown(target = document.querySelector("iframe").contentWindow) {
+    if (typeof target === 'undefined') {
+      target = window;
     }
 
     this.constructor.isSetUp = false;
-    this.removeEventListeners(window);
+    this.removeEventListeners(target);
     this.clearCurrentDragSourceNode();
   }