From b1c3d6a630469ce6f54fbada559ac0c5b21d816f Mon Sep 17 00:00:00 2001 From: Thomas Forgione Date: Fri, 17 Feb 2023 11:25:36 +0100 Subject: [PATCH] Full support for classes --- index.js | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/index.js b/index.js index 53154cd..21b7368 100644 --- a/index.js +++ b/index.js @@ -80,12 +80,18 @@ async function main() { let shouldCreateSpan = query !== "ul" && query !== "ol" && query != "img"; // Query the considered element + let parents = await page.$$(query); let elements = await page.$$(query + (shouldCreateSpan ? ' > *:first-child' : '')); - for (let element of elements) { + for (let index = 0; index < elements.length; index ++) { + + let parent = parents[index]; + let element = elements[index]; let box = await element.boundingBox(); - let classNameAttr = await element.getProperty('className'); + + let classElement = shouldCreateSpan ? parent : element; + let classNameAttr = await classElement.getProperty('className'); let className = await classNameAttr.jsonValue(); // Scale the bounding box