optimize document color algorithm

Author: CppCXY

crates/code_analysis/src/config/configs/document_color.rs

@@ -0,0 +1,22 @@
+use schemars::JsonSchema;
+use serde::{Deserialize, Serialize};
+
+#[derive(Serialize, Deserialize, Debug, JsonSchema)]
+#[serde(rename_all = "camelCase")]
+pub struct EmmyrcDocumentColor {
+    /// Whether to enable document color.
+    #[serde(default = "default_true")]
+    pub enable: bool,
+}
+
+impl Default for EmmyrcDocumentColor {
+    fn default() -> Self {
+        Self {
+            enable: default_true(),
+        }
+    }
+}
+
+fn default_true() -> bool {
+    true
+}

crates/code_analysis/src/config/configs/mod.rs

@@ -10,6 +10,7 @@ mod strict;
 mod semantictoken;
 mod references;
 mod hover;
+mod document_color;
 
 
 pub use completion::{EmmyrcCompletion, EmmyrcFilenameConvention};
@@ -23,4 +24,5 @@ pub use codelen::EmmyrcCodeLen;
 pub use strict::EmmyrcStrict;
 pub use semantictoken::EmmyrcSemanticToken;
 pub use references::EmmyrcReference;
-pub use hover::EmmyrcHover;
+pub use hover::EmmyrcHover;
+pub use document_color::EmmyrcDocumentColor;

crates/code_analysis/src/config/mod.rs

@@ -6,6 +6,7 @@ use std::{collections::HashSet, path::PathBuf};
 
 use crate::{semantic::LuaInferConfig, FileId};
 pub use config_loader::load_configs;
+use configs::EmmyrcDocumentColor;
 pub use configs::EmmyrcFilenameConvention;
 pub use configs::EmmyrcLuaVersion;
 use configs::{
@@ -49,6 +50,8 @@ pub struct Emmyrc {
     pub references: EmmyrcReference,
     #[serde(default)]
     pub hover: EmmyrcHover,
+    #[serde(default)]
+    pub document_color: EmmyrcDocumentColor,
 }
 
 impl Emmyrc {

crates/emmylua_ls/src/handlers/document_color/build_color.rs

@@ -32,6 +32,16 @@ fn try_build_color_information(
     let mut i = 0;
     while i + 6 <= len {
         if bytes[i].is_ascii_hexdigit() {
+            let is_start_boundary = if i == 0 {
+                true
+            } else {
+                !bytes[i - 1].is_ascii_alphanumeric()
+            };
+            if !is_start_boundary {
+                i += 1;
+                continue;
+            }
+
             let mut j = i + 1;
             while j < len && bytes[j].is_ascii_hexdigit() {
                 j += 1;
@@ -106,4 +116,4 @@ pub fn convert_color_to_hex(color: Color, len: usize) -> String {
         }
         _ => "".to_string(),
     }
-}
+}

crates/emmylua_ls/src/handlers/document_color/mod.rs

@@ -29,6 +29,10 @@ pub async fn on_document_color(
             return vec![];
         };
 
+    if !semantic_model.get_emmyrc().document_color.enable {
+        return vec![];
+    }
+
     let document = semantic_model.get_document();
     let root = semantic_model.get_root();
     build_colors(root.syntax().clone(), &document)

resources/schema.json

@@ -50,6 +50,16 @@
         }
       ]
     },
+    "documentColor": {
+      "default": {
+        "enable": true
+      },
+      "allOf": [
+        {
+          "$ref": "#/definitions/EmmyrcDocumentColor"
+        }
+      ]
+    },
     "hint": {
       "default": {
         "enable": true,
@@ -470,6 +480,16 @@
         }
       }
     },
+    "EmmyrcDocumentColor": {
+      "type": "object",
+      "properties": {
+        "enable": {
+          "description": "Whether to enable document color.",
+          "default": true,
+          "type": "boolean"
+        }
+      }
+    },
     "EmmyrcFilenameConvention": {
       "oneOf": [
         {