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