wrapper/tailwindcss
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Use DefaultMap instead of plain-old Map

Browse source
This commit is contained in:
Philipp Spiess 2024-08-30 17:04:54 +02:00
parent 3dbf264e5f
commit 4b92050c17
1 changed files with 22 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

33
packages/@tailwindcss-vite/src/index.ts
View file

@ -24,7 +24,7 @@ export default function tailwindcss(): Plugin[] {
// to manually rebuild the css file after the compilation is done.
let cssPlugins: readonly Plugin[] = []
let roots: Map<string, Root> = new Map()
let roots: DefaultMap<string, Root> = new DefaultMap((id) => new Root(id))
// The Vite extension has two types of sources for candidates:
//
@ -341,12 +341,7 @@ export default function tailwindcss(): Plugin[] {
async transform(src, id, options) {
if (!isPotentialCssRootFile(id)) return
// TODO: Use DefaultMap
let root = roots.get(id)
if (!root) {
root = new Root(id)
roots.set(id, root)
}
root.invalidate()
if (!options?.ssr) {
@ -378,12 +373,7 @@ export default function tailwindcss(): Plugin[] {
async transform(src, id) {
if (!isPotentialCssRootFile(id)) return
// TODO: Use DefaultMap
let root = roots.get(id)
if (!root) {
root = new Root(id)
roots.set(id, root)
}
root.invalidate()
// We do a first pass to generate valid CSS for the downstream plugins.
@ -476,3 +466,24 @@ function optimizeCss(
function idToPath(id: string) {
return path.resolve(id.replace(/\?.*$/, ''))
}
/**
* A Map that can generate default values for keys that don't exist.
* Generated default values are added to the map to avoid recomputation.
*/
class DefaultMap<K, V> extends Map<K, V> {
constructor(private factory: (key: K, self: DefaultMap<K, V>) => V) {
super()
}
get(key: K): V {
let value = super.get(key)
if (value === undefined) {
value = this.factory(key, this)
this.set(key, value)
}
return value
}
}