Merge a755924 into c13f3e5

src/index.spec.ts

@@ -8,6 +8,11 @@ describe("index", () => {
         expect(select("div", dom)).toHaveLength(1);
     });
 
+    it("should find with a function", () => {
+        const dom = parseDOM("<div><p>First<p>Second") as Element[];
+        expect(select((elem) => elem.name === "p", dom)).toHaveLength(2);
+    });
+
     it("should support positionals", () => {
         const dom = parseDOM("<div><p>First<p>Second") as Element[];
         expect(select("p:first", dom)).toMatchInlineSnapshot(`

src/index.ts

@@ -125,10 +125,14 @@ function filterParsed(
 }
 
 export function select(
-    selector: string,
+    selector: string | ((el: Element) => boolean),
     root: Element | Element[],
     options: Options = {}
 ): Element[] {
+    if (typeof selector === "function") {
+        return find(root, selector);
+    }
+
     const [plain, filtered] = groupSelectors(parse(selector, options));
 
     const results: Element[][] = filtered.map((sel) =>
@@ -276,6 +280,11 @@ function findFilterElements(
           filterElements(result, [remainingSelector], remainingOpts);
 }
 
+interface CompiledQuery {
+    (el: Element): boolean;
+    shouldTestNextSiblings?: boolean;
+}
+
 function findElements(
     root: Element | Element[],
     sel: Selector[][],
@@ -284,8 +293,21 @@ function findElements(
 ): Element[] {
     if (limit === 0) return [];
 
-    // @ts-expect-error TS seems to mess up the type here ¯\_(ツ)_/¯
-    const query = compileToken<Node, Element>(sel, options, root);
+    const query: CompiledQuery = compileToken<Node, Element>(
+        sel,
+        // @ts-expect-error TS seems to mess up the type here ¯\_(ツ)_/¯
+        options,
+        root
+    );
+
+    return find(root, query, limit);
+}
+
+function find(
+    root: Element | Element[],
+    query: CompiledQuery,
+    limit = Infinity
+): Element[] {
     const elems = prepareContext<Node, Element>(
         root,
         DomUtils,