|
@@ -3,16 +3,18 @@ const isDOMElement = require('./isDOMElement')
|
|
|
/**
|
|
|
* Find one or more DOM elements.
|
|
|
*
|
|
|
- * @param {string} element
|
|
|
- * @returns {Array|null}
|
|
|
+ * @param {string|Node} element
|
|
|
+ * @returns {Node[]|null}
|
|
|
*/
|
|
|
module.exports = function findAllDOMElements (element) {
|
|
|
if (typeof element === 'string') {
|
|
|
- const elements = [].slice.call(document.querySelectorAll(element))
|
|
|
- return elements.length > 0 ? elements : null
|
|
|
+ const elements = document.querySelectorAll(element)
|
|
|
+ return elements.length === 0 ? null : Array.from(elements)
|
|
|
}
|
|
|
|
|
|
if (typeof element === 'object' && isDOMElement(element)) {
|
|
|
return [element]
|
|
|
}
|
|
|
+
|
|
|
+ return null
|
|
|
}
|