properly register comments as parsed items

anoma · Dec 6, 2024 · d3f13c1 · d3f13c1
1 parent 7ae318d
commit d3f13c1
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