tailwindcss/src/jit/lib/optimizeUniversalRules.js
2021-06-30 18:38:39 +02:00

73 lines
1.8 KiB
JavaScript

import selectorParser from 'postcss-selector-parser'
let elementSelectorParser = selectorParser((selectors) => {
return selectors.map((s) => {
return s
.split((n) => n.type === 'combinator')
.pop()
.filter((n) => n.type !== 'pseudo' || n.value.startsWith('::'))
.join('')
.trim()
})
})
let cache = new Map()
function extractElementSelector(selector) {
if (!cache.has(selector)) {
cache.set(selector, elementSelectorParser.transformSync(selector))
}
return cache.get(selector)
}
export default function collapseAdjacentRules() {
return (root) => {
// Walk all custom properties
// Collect into a map, where key is the custom property name
// Value is list of nodes
// Walk '*::tailwind' rules
// Get list of selectors per custom property
// Compute element selector (remove pseudo-classes, leading selectors, etc.)
// Collect selectors into Set
// Replace '*::tailwind' with selectors from Set
let variableNodeMap = new Map()
let universals = new Set()
root.walkDecls((decl) => {
if (!decl.prop.startsWith('--')) {
return
}
if (decl.parent.selector === '*::tailwind') {
universals.add(decl.parent)
return
}
let variable = decl.prop
if (!variableNodeMap.has(variable)) {
variableNodeMap.set(variable, new Set())
}
variableNodeMap.get(variable).add(decl.parent)
})
for (let universal of universals) {
let selectors = new Set()
universal.walkDecls((decl) => {
let variable = decl.prop
for (let node of variableNodeMap.get(variable) ?? []) {
for (let selector of extractElementSelector(node.selector)) {
selectors.add(selector)
}
}
})
universal.selectors = [...selectors]
}
}
}