Allowing elements to be passed to service.

EmberSherpa · Dec 31, 2018 · 0be3b99 · 0be3b99
1 parent 258d3ed
commit 0be3b99
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/addon/services/resize-detector.js b/addon/services/resize-detector.js
@@ -16,7 +16,14 @@ export default Ember.Service.extend({
   },
 
   setup(selector, callback) {
-    let element = document.querySelector(selector)
+    let element;
+
+    if (selector instanceof HTMLElement) {
+      element = selector;
+    } else {
+      element = document.querySelector(selector);
+    }
+
     if (!element) {
       error(`service:resize-detector - could not find an element matching ${selector}`);
       return;
@@ -25,7 +32,14 @@ export default Ember.Service.extend({
   },
 
   teardown(selector, callback) {
-    let element = document.querySelector(selector)
+    let element;
+
+    if (selector instanceof HTMLElement) {
+      element = selector;
+    } else {
+      element = document.querySelector(selector);
+    }
+
     if (element) {
       this.detector.removeListener(element, callback);
     }