From 906d3cfb4ee92af42cd51e6744b8beeb9b16c54b Mon Sep 17 00:00:00 2001
From: Brad Cornes <bradlc41@gmail.com>
Date: Tue, 25 Aug 2020 15:21:12 +0100
Subject: [PATCH] Fix color parsing when alpha is 0 (#177)

---
 src/lsp/providers/completionProvider.ts    |  6 ++--
 src/lsp/providers/documentColorProvider.ts |  6 ++--
 src/lsp/util/color.ts                      | 42 +++++++++++-----------
 3 files changed, 29 insertions(+), 25 deletions(-)

diff --git a/src/lsp/providers/completionProvider.ts b/src/lsp/providers/completionProvider.ts
index 795127c..17fedcb 100644
--- a/src/lsp/providers/completionProvider.ts
+++ b/src/lsp/providers/completionProvider.ts
@@ -89,9 +89,11 @@ function completionsFromClassList(
           sortText = '-' + sortText // move to top
         } else {
           const color = getColor(state, [className])
-          if (color) {
+          if (color !== null) {
             kind = CompletionItemKind.Color
-            documentation = color.documentation
+            if (typeof color !== 'string' && color.a !== 0) {
+              documentation = color.toRgbString()
+            }
           }
         }
 
diff --git a/src/lsp/providers/documentColorProvider.ts b/src/lsp/providers/documentColorProvider.ts
index 688ee74..876f577 100644
--- a/src/lsp/providers/documentColorProvider.ts
+++ b/src/lsp/providers/documentColorProvider.ts
@@ -27,8 +27,10 @@ export function registerDocumentColorProvider(state: State) {
           let parts = getClassNameParts(state, className.className)
           if (!parts) return
           let color = getColor(state, parts)
-          if (!color) return
-          colors.push({ range: className.range, color: color.documentation })
+          if (color === null || typeof color === 'string' || color.a === 0) {
+            return
+          }
+          colors.push({ range: className.range, color: color.toRgbString() })
         })
       })
 
diff --git a/src/lsp/util/color.ts b/src/lsp/util/color.ts
index 31d56ec..436a94c 100644
--- a/src/lsp/util/color.ts
+++ b/src/lsp/util/color.ts
@@ -21,10 +21,14 @@ const COLOR_PROPS = [
   'text-decoration-color',
 ]
 
+function isKeyword(value: string): boolean {
+  return ['transparent', 'currentcolor'].includes(value.toLowerCase())
+}
+
 export function getColor(
   state: State,
   keys: string[]
-): { documentation?: string } {
+): TinyColor | string | null {
   const item = dlv(state.classNames.classNames, keys)
   if (!item.__rule) return null
   const props = Object.keys(removeMeta(item))
@@ -48,40 +52,36 @@ export function getColor(
   )
 
   // check that all of the values are valid colors
-  if (colors.some((color) => color !== 'transparent' && !color.isValid)) {
+  if (colors.some((color) => typeof color !== 'string' && !color.isValid)) {
     return null
   }
 
   // check that all of the values are the same color, ignoring alpha
   const colorStrings = dedupe(
     colors.map((color) =>
-      color === 'transparent'
-        ? 'transparent'
-        : `${color.r}-${color.g}-${color.b}`
+      typeof color === 'string' ? color : `${color.r}-${color.g}-${color.b}`
     )
   )
   if (colorStrings.length !== 1) {
     return null
   }
 
-  if (colorStrings[0] === 'transparent') {
-    return {
-      documentation: 'rgba(0, 0, 0, 0.01)',
-    }
+  if (isKeyword(colorStrings[0])) {
+    return colorStrings[0]
   }
 
-  const nonTransparentColors = colors.filter(
-    (color): color is TinyColor => color !== 'transparent'
+  const nonKeywordColors = colors.filter(
+    (color): color is TinyColor => typeof color !== 'string'
   )
 
-  const alphas = dedupe(nonTransparentColors.map((color) => color.a))
+  const alphas = dedupe(nonKeywordColors.map((color) => color.a))
 
-  if (alphas.length === 1 || (alphas.length === 2 && alphas.includes(0))) {
-    return {
-      documentation: nonTransparentColors
-        .find((color) => color.a !== 0)
-        .toRgbString(),
-    }
+  if (alphas.length === 1) {
+    return nonKeywordColors[0]
+  }
+
+  if (alphas.length === 2 && alphas.includes(0)) {
+    return nonKeywordColors.find((color) => color.a !== 0)
   }
 
   return null
@@ -99,9 +99,9 @@ export function getColorFromValue(value: unknown): string {
   return null
 }
 
-function createColor(str: string): TinyColor | 'transparent' {
-  if (str === 'transparent') {
-    return 'transparent'
+function createColor(str: string): TinyColor | string {
+  if (isKeyword(str)) {
+    return str
   }
 
   // matches: rgba(<r>, <g>, <b>, var(--bg-opacity))