properly register comments as parsed items

anoma · Dec 6, 2024 · 7bbefb1 · 7bbefb1
1 parent e06356d
commit 7bbefb1
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/Juvix/Compiler/Concrete/Translation/FromSource/ParserResultBuilder.hs b/src/Juvix/Compiler/Concrete/Translation/FromSource/ParserResultBuilder.hs
@@ -95,12 +95,14 @@ runParserResultBuilder s =
       registerItem' i
     RegisterSpaceSpan g -> do
       modify' (over parserStateComments (g :))
-      forM_ (g ^.. spaceSpan . each . _SpaceComment) $ \c ->
-        registerItem'
-          ParsedItem
-            { _parsedLoc = getLoc c,
-              _parsedTag = ParsedTagComment
-            }
+      forM_ (g ^.. spaceSpan . each . _SpaceComment) $ \c -> do
+        let i =
+              ParsedItem
+                { _parsedLoc = getLoc c,
+                  _parsedTag = ParsedTagComment
+                }
+        highlightParsedItem i
+        registerItem' i
 
 ignoreParserResultBuilder :: Sem (ParserResultBuilder ': r) a -> Sem r a
 ignoreParserResultBuilder = interpret $ \case