feat: multi elements support for nth-of-type

fczbkk · Aug 15, 2021 · f481ca6 · f481ca6
1 parent 74443c2
commit f481ca6
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/src/selector-nth-of-type.ts b/src/selector-nth-of-type.ts
@@ -1,10 +1,12 @@
 import {getTagSelector} from './selector-tag'
-import {CssSelector} from './types'
+import {CssSelector, SelectorNeedle} from './types';
+import {sanitizeSelectorNeedle} from './utilities-selectors';
+import {getIntersection} from './utilities-data';
 
 /**
  * Get nth-of-type selector for an element.
  */
-export function getNthOfTypeSelector (element: Element): Array<CssSelector> {
+export function getElementNthOfTypeSelector (element: Element): Array<CssSelector> {
   const tag = getTagSelector(element)[0]
   const parentElement = element.parentElement
 
@@ -18,3 +20,11 @@ export function getNthOfTypeSelector (element: Element): Array<CssSelector> {
 
   return []
 }
+
+/**
+ * Get Nth-of-type selector matching all elements.
+ */
+export function getNthOfTypeSelector (needle: SelectorNeedle): Array<CssSelector> {
+  const elements = sanitizeSelectorNeedle(needle)
+  return getIntersection(elements.map(getElementNthOfTypeSelector))
+}
diff --git a/test/selector-nth-of-type.spec.js b/test/selector-nth-of-type.spec.js
@@ -71,4 +71,14 @@ describe('selector - nth-of-type', function () {
     assert.doesNotThrow(fn)
   })
 
+  it('should generate nth-of-type selector for multiple elements', () => {
+    root.innerHTML = `
+      <div><div></div><span></span></div>
+      <div><span></span></div>
+    `
+    const elements = [...root.querySelectorAll('span')]
+    const result = getNthOfTypeSelector(elements)
+    assert.sameMembers(result, ['span:nth-of-type(1)'])
+  })
+
 })