Allowing elements to be passed to service.

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