Add missing vertex detection

1 files changed, 21 insertions, 4 deletions

diff --git a/Map/Business/WmapSyntaxHighlighter/SyntaxHighlighter.swift b/Map/Business/WmapSyntaxHighlighter/SyntaxHighlighter.swift
index e0aaee7..d51a80f 100644
--- a/Map/Business/WmapSyntaxHighlighter/SyntaxHighlighter.swift
+++ b/Map/Business/WmapSyntaxHighlighter/SyntaxHighlighter.swift
@@ -16,11 +16,13 @@ import AppKit
 
 extension Wmap {
   class SyntaxHighlighter {
-    private var cachedSyntaxElements: [SyntaxElement] = []
+    var cachedSyntaxElements: [SyntaxElement] = []
     private var lineToElementsMap: [Int: [SyntaxElement]] = [:]
+    private var cachedVertexLabels: Set<String> = Set()
 
-    func updateSyntaxElements(from parsedMap: [Entity]) {
-      cachedSyntaxElements = parsedMap.flatMap { $0.syntaxElements }
+    func updateSyntaxElements(from parsedMap: Wmap.ParsedMap) {
+      cachedSyntaxElements = parsedMap.entities.flatMap { $0.syntaxElements }
+      cachedVertexLabels = parsedMap.vertexLabels
 
       // Build line-to-elements mapping for fast lookup
       lineToElementsMap.removeAll()
@@ -55,7 +57,22 @@ extension Wmap {
         }
 
         let color = colorForSyntaxType(element.type)
-        textStorage.addAttribute(.foregroundColor, value: color, range: nsRange)
+        var attributes: [NSAttributedString.Key: Any] = [
+          .foregroundColor: color
+        ]
+
+        if UserDefaults.standard.bool(forKey: "useSmartEditor") {
+          if element.type == .vertexLabel && !cachedVertexLabels.contains(element.value) {
+            if UserDefaults.standard.bool(forKey: "highlightMissingComponents") {
+              attributes[.underlineStyle] =
+                NSUnderlineStyle.thick.rawValue | NSUnderlineStyle.patternDot.rawValue
+              attributes[.underlineColor] = NSColor.Theme.Syntax.error
+              attributes[.init("MissingComponent")] = element.value
+            }
+          }
+        }
+
+        textStorage.addAttributes(attributes, range: nsRange)
       }
     }