detect keywords in the lookahead

GrammaticalFramework · Mar 8, 2024 · 518e571 · 518e571
1 parent 5a2e1a8
commit 518e571
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/compiler/api/GF/Grammar/Lexer.x b/src/compiler/api/GF/Grammar/Lexer.x
@@ -358,13 +358,18 @@ start_of_tag s = isJust (match s)
 
     matchIdent s
       | BS.null s                = Nothing
-      | BS.elem (BS.head s) init = matchRest (BS.tail s)
+      | BS.elem (BS.head s) init = matchRest s 1 (BS.tail s)
       | otherwise                = Nothing
 
-    matchRest s
-      | BS.null s                = Just s
-      | BS.elem (BS.head s) rest = matchRest (BS.tail s)
-      | otherwise                = Just s
+    matchRest s0 i s
+      | BS.null s                = checkResWord (BS.take i s0) s
+      | BS.elem (BS.head s) rest = matchRest s0 (i+1) (BS.tail s)
+      | otherwise                = checkResWord (BS.take i s0) s
+
+    checkResWord w s =
+      case Map.lookup (identC (rawIdentC w)) resWords of
+        Just t  -> Nothing
+        Nothing -> Just s
 
 getPosn :: P Posn
 getPosn = P $ \ai2@(_,inp@(AI pos _ _)) -> POk ai2 pos