44 lines
1.4 KiB
TypeScript
44 lines
1.4 KiB
TypeScript
|
import { State } from './util/state'
|
||
|
|
import {
|
||
|
|
findClassListsInDocument,
|
||
|
|
getClassNamesInClassList,
|
||
|
|
findHelperFunctionsInDocument,
|
||
|
|
} from './util/find'
|
||
|
|
import { getClassNameParts } from './util/getClassNameAtPosition'
|
||
|
|
import { getColor, getColorFromValue } from './util/color'
|
||
|
|
import { stringToPath } from './util/stringToPath'
|
||
|
|
import type { TextDocument } from 'vscode-languageserver'
|
||
|
|
const dlv = require('dlv')
|
||
|
|
|
||
|
|
export function getDocumentColors(state: State, document: TextDocument) {
|
||
|
|
let colors = []
|
||
|
|
if (!state.enabled) return colors
|
||
|
|
|
||
|
|
let classLists = findClassListsInDocument(state, document)
|
||
|
|
classLists.forEach((classList) => {
|
||
|
|
let classNames = getClassNamesInClassList(classList)
|
||
|
|
classNames.forEach((className) => {
|
||
|
|
let parts = getClassNameParts(state, className.className)
|
||
|
|
if (!parts) return
|
||
|
|
let color = getColor(state, parts)
|
||
|
|
if (color === null || typeof color === 'string' || color.a === 0) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
colors.push({ range: className.range, color: color.toRgbString() })
|
||
|
|
})
|
||
|
|
})
|
||
|
|
|
||
|
|
let helperFns = findHelperFunctionsInDocument(state, document)
|
||
|
|
helperFns.forEach((fn) => {
|
||
|
|
let keys = stringToPath(fn.value)
|
||
|
|
let base = fn.helper === 'theme' ? ['theme'] : []
|
||
|
|
let value = dlv(state.config, [...base, ...keys])
|
||
|
|
let color = getColorFromValue(value)
|
||
|
|
if (color) {
|
||
|
|
colors.push({ range: fn.valueRange, color })
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
return colors
|
||
|
|
}
|