Add getNodeSelector page function

GoogleChrome · Dec 2, 2018 · 030fb89 · 030fb89
1 parent 5e341c0
commit 030fb89
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/lighthouse-core/lib/page-functions.js b/lighthouse-core/lib/page-functions.js
@@ -184,6 +184,38 @@ function getNodePath(node) {
   return path.join(',');
 }
 
+/**
+ * @param {Element} node
+ * @returns {string}
+ */
+function getNodeSelector(node) {
+  /**
+   * @param {Element} node
+   */
+  function getSelectorPart(node) {
+    let part = node.tagName.toLowerCase();
+    if (node.id) {
+      part += '#' + node.id;
+    } else if (node.classList.length > 0) {
+      part += '.' + node.classList[0];
+    }
+    return part;
+  }
+
+  const parts = [];
+  while (parts.length < 4) {
+    parts.unshift(getSelectorPart(node));
+    if (!node.parentElement) {
+      break;
+    }
+    node = node.parentElement;
+    if (node.tagName === 'HTML') {
+      break;
+    }
+  }
+  return parts.join(' > ');
+}
+
 module.exports = {
   wrapRuntimeEvalErrorInBrowserString: wrapRuntimeEvalErrorInBrowser.toString(),
   registerPerformanceObserverInPageString: registerPerformanceObserverInPage.toString(),
@@ -194,4 +226,5 @@ module.exports = {
   ultradumbBenchmark: ultradumbBenchmark,
   ultradumbBenchmarkString: ultradumbBenchmark.toString(),
   getNodePathString: getNodePath.toString(),
+  getNodeSelectorString: getNodeSelector.toString(),
 };