Add css class name completions

Author: krukas

djlsp/index.py

@@ -50,6 +50,7 @@ class WorkspaceIndex:
     libraries: dict[str, Library] = field(default_factory=dict)
     templates: dict[str, Template] = field(default_factory=dict)
     global_template_context: dict[str, str] = field(default_factory=dict)
+    css_class_names: list = field(default_factory=list)
 
     def update(self, django_data: dict):
         self.file_watcher_globs = django_data.get(
@@ -96,3 +97,4 @@ def update(self, django_data: dict):
         }
 
         self.global_template_context = django_data.get("global_template_context", {})
+        self.css_class_names = django_data.get("css_class_names", [])

djlsp/parser.py

@@ -207,7 +207,8 @@ def completions(self, line, character):
                         ),
                     )
                 )
-        return []
+        # Check CSS
+        return self.get_css_class_name_completions(line, character)
 
     def get_load_completions(self, match: Match, **kwargs):
         prefix = match.group(1).split(" ")[-1]
@@ -418,6 +419,37 @@ def get_sort_text(comp):
                 if var.startswith(prefix)
             ]
 
+    def get_css_class_name_completions(self, line, character):
+        # Flatten text to one line and remove Django template
+        one_line_html = "".join(self.document.lines)
+        one_line_html = re.sub(
+            r"\{\%.*?\%\}", lambda m: " " * len(m.group(0)), one_line_html
+        )
+        one_line_html = re.sub(
+            r"\{\{.*?\}\}", lambda m: " " * len(m.group(0)), one_line_html
+        )
+
+        abs_position = sum(len(self.document.lines[i]) for i in range(line)) + character
+        class_attr_pattern = re.compile(r'class=["\']([^"\']*)["\']', re.DOTALL)
+
+        for match in class_attr_pattern.finditer(one_line_html):
+            start_idx, end_idx = match.span(1)
+
+            if start_idx <= abs_position <= end_idx:
+                class_value = match.group(1)
+                relative_pos = abs_position - start_idx
+
+                prefix_match = re.search(r"\b[\w-]*$", class_value[:relative_pos])
+                prefix = prefix_match.group(0) if prefix_match else ""
+
+                return [
+                    CompletionItem(label=class_name)
+                    for class_name in self.workspace_index.css_class_names
+                    if class_name.startswith(prefix)
+                ]
+
+        return []
+
     ###################################################################################
     # Hover
     ###################################################################################

djlsp/scripts/django-collector.py

@@ -196,6 +196,7 @@ def __init__(self, project_src_path):
         self.libraries = {}
         self.templates: dict[str, Template] = {}
         self.global_template_context = {}
+        self.css_class_names = []
 
     def collect(self):
         self.file_watcher_globs = self.get_file_watcher_globs()
@@ -204,6 +205,7 @@ def collect(self):
         self.urls = self.get_urls()
         self.libraries = self.get_libraries()
         self.global_template_context = self.get_global_template_context()
+        self.css_class_names = self.get_css_class_names()
 
         # Third party collectors
         self.collect_for_wagtail()
@@ -217,6 +219,7 @@ def to_json(self):
                 "libraries": self.libraries,
                 "templates": self.templates,
                 "global_template_context": self.global_template_context,
+                "css_class_names": self.css_class_names,
             },
             indent=4,
         )
@@ -573,6 +576,23 @@ def collect_for_wagtail(self):
                         model.context_object_name
                     ] = (model.__module__ + "." + model.__name__)
 
+    # CSS class names
+    # ---------------------------------------------------------------------------------
+    def get_css_class_names(self):
+        class_pattern = re.compile(r"\.(?!\d)([a-zA-Z0-9_-]+)")
+        class_names = set()
+
+        for finder in get_finders():
+            for path, file_storage in finder.list(None):
+                if path.endswith(".css") and not path.startswith("admin"):
+                    try:
+                        with file_storage.open(path, "r") as f:
+                            content = f.read()
+                            class_names.update(class_pattern.findall(content))
+                    except Exception as e:
+                        logger.error(f"Could not parse CSS file: {e}")
+        return list(class_names)
+
 
 #######################################################################################
 # CLI

tests/django_test/test-static-folder/website.css

@@ -0,0 +1,3 @@
+.blog .item {
+  color: red;
+}