feat: add getter for root element

stfsy · Jul 7, 2021 · acb9c31 · acb9c31
1 parent 4c0538e
commit acb9c31
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/lib/node.js b/lib/node.js
@@ -205,6 +205,14 @@ class Node {
         return this._wrapOrNull(next)
     }
 
+    get root() {
+        let root = this.parent
+        while (root && root.parent.type !== 'root') {
+            root = root.parent
+        }
+        return Node.of(root)
+    }
+
     /**
      * @private
      */

diff --git a/test/spec/node.spec.js b/test/spec/node.spec.js
@@ -114,6 +114,21 @@ describe('Node', () => {
         expect(node.length).to.equal(3)
     })
 
+    it('returns the current DOMS root element', () => {
+        const string = [
+            '<header>',
+            '<meta name="viewport" content="width=device-width, user-scalable=no">',
+            '<meta name="viewport" content="width=device-width, user-scalable=no">',
+            '<meta name="viewport" content="width=device-width, user-scalable=no">',
+            '</header>'
+        ].join('')
+
+        const node = Node.fromString(string)        
+        const meta = node.find('meta')[0]
+        const root = meta.root
+        expect(root.name).to.equal('header')
+    })
+
     it('returns the elements children wrapped as Nodes', () => {
         const string = [
             '<header>',