vodorod/parcer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: get all descendant selectors

Browse Source
This commit is contained in:
karishmas6
2024-09-03 05:50:04 +05:30
parent 0dfb3f6673
commit e727d11861
1 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
server/src/workflow-management/selector.ts
View File

@@ -784,12 +784,9 @@ export const getNonUniqueSelectors = async (page: Page, coordinates: Coordinates
return { generalSelector: '' };
}
};
interface SelectorNode {
selector: string;
children: SelectorNode[];
}
export const getChildSelectors = async (page: Page, parentSelector: string, maxDepth: number = 2): Promise<SelectorNode[]> => {
export const getChildSelectors = async (page: Page, parentSelector: string, maxDepth: number = 2): Promise<string[]> => {
try {
const childSelectors = await page.evaluate(({ parentSelector, maxDepth }: { parentSelector: string, maxDepth: number }) => {
function getNonUniqueSelector(element: HTMLElement): string {
@@ -822,22 +819,24 @@ export const getChildSelectors = async (page: Page, parentSelector: string, maxD
return path.join(' > ');
}
function getDescendantSelectors(element: HTMLElement, currentDepth: number = 0): SelectorNode[] {
function getAllDescendantSelectors(element: HTMLElement, currentDepth: number = 0): string[] {
if (currentDepth >= maxDepth) return [];
let selectors: string[] = [];
const children = Array.from(element.children) as HTMLElement[];
return children.map(child => {
return {
selector: getSelectorPath(child),
children: getDescendantSelectors(child, currentDepth + 1)
};
});
for (const child of children) {
selectors.push(getSelectorPath(child));
selectors = selectors.concat(getAllDescendantSelectors(child, currentDepth + 1));
}
return selectors;
}
const parentElement = document.querySelector(parentSelector) as HTMLElement;
if (!parentElement) return [];
return getDescendantSelectors(parentElement);
return getAllDescendantSelectors(parentElement);
}, { parentSelector, maxDepth });
return childSelectors || [];