Refactor document colors provider

This uses the new virtual document abstraction to simplify the implementation of the provider. Now it operates only on things it cares about:
- utility class names
- CSS helper functions

Author: thecrypticace

packages/tailwindcss-language-service/src/documentColorProvider.ts

@@ -1,51 +1,40 @@
-import type { State } from './util/state'
-import {
-  findClassListsInDocument,
-  getClassNamesInClassList,
-  findHelperFunctionsInDocument,
-} from './util/find'
+import type { ColorInformation } from 'vscode-languageserver'
+import type { Document } from './documents/document'
 import { getColor, getColorFromValue, culoriColorToVscodeColor } from './util/color'
 import { stringToPath } from './util/stringToPath'
-import type { ColorInformation } from 'vscode-languageserver'
-import type { TextDocument } from 'vscode-languageserver-textdocument'
 import dlv from 'dlv'
 import { dedupeByRange } from './util/array'
 
-export async function getDocumentColors(
-  state: State,
-  document: TextDocument,
-): Promise<ColorInformation[]> {
+export function getDocumentColors(doc: Document): ColorInformation[] {
   let colors: ColorInformation[] = []
-  if (!state.enabled) return colors
 
-  let settings = await state.editor.getConfiguration(document.uri)
-  if (settings.tailwindCSS.colorDecorators === false) return colors
+  for (let className of doc.classNames()) {
+    let color = getColor(doc.state, className.className)
+    if (!color) continue
+    if (typeof color === 'string') continue
+    if ((color.alpha ?? 1) === 0) continue
 
-  let classLists = await findClassListsInDocument(state, document)
-  classLists.forEach((classList) => {
-    let classNames = getClassNamesInClassList(classList, state.blocklist)
-    classNames.forEach((className) => {
-      let color = getColor(state, className.className)
-      if (color === null || typeof color === 'string' || (color.alpha ?? 1) === 0) {
-        return
-      }
-      colors.push({
-        range: className.range,
-        color: culoriColorToVscodeColor(color),
-      })
+    colors.push({
+      range: className.range,
+      color: culoriColorToVscodeColor(color),
     })
-  })
+  }
 
-  let helperFns = findHelperFunctionsInDocument(state, document)
-  helperFns.forEach((fn) => {
+  for (let fn of doc.helperFns()) {
     let keys = stringToPath(fn.path)
     let base = fn.helper === 'theme' ? ['theme'] : []
-    let value = dlv(state.config, [...base, ...keys])
+    let value = dlv(doc.state.config, [...base, ...keys])
+
     let color = getColorFromValue(value)
-    if (color && typeof color !== 'string' && (color.alpha ?? 1) !== 0) {
-      colors.push({ range: fn.ranges.path, color: culoriColorToVscodeColor(color) })
-    }
-  })
+    if (!color) continue
+    if (typeof color === 'string') continue
+    if ((color.alpha ?? 1) === 0) continue
+
+    colors.push({
+      range: fn.ranges.path,
+      color: culoriColorToVscodeColor(color),
+    })
+  }
 
   return dedupeByRange(colors)
 }

packages/tailwindcss-language-service/src/service.ts

@@ -139,7 +139,7 @@ async function createLanguageDocument(
     async documentColors() {
       if (!state.enabled || !settings.tailwindCSS.colorDecorators) return []
 
-      return getDocumentColors(state, doc.storage)
+      return getDocumentColors(doc)
     },
 
     async colorPresentation(color: Color, range: Range) {