Update the lexer for it to be slightly more flexible

agagniere · Mar 7, 2024 · 877f691 · 877f691
1 parent 3ca7207
commit 877f691
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/extra/pygment/lexer.py b/extra/pygment/lexer.py
@@ -18,29 +18,35 @@ class CustomLexer(RegexLexer):
             (r'//.*?$', Comment.Single),
         ],
         'detect_prepro' : [
-            (r'^(#)(include)(\s+)([<"][\w/.]+[">])',
+            (r'^(#)(\s*)(include)(\s+)([<"][\w/.]+[">])(\s*)([/].*)?',
              bygroups(Comment.Hashbang,
+                      Whitespace,
                       Keyword,
                       Whitespace,
-                      Comment.PreprocFile)),
-            (r'^(#)(define)(\s+)(\w+)(\s+)',
+                      Comment.PreprocFile,
+                      Whitespace,
+                      Whitespace)),
+            (r'^(#)(\s*)(define)(\s+)(\w+)(\s+)',
              bygroups(Comment.Hashbang,
+                      Whitespace,
                       Keyword,
                       Whitespace,
                       Name.Constant,
                       Whitespace),
              'prepro'),
-            (r'^(#)(define)(\s+)(\w+)([(])',
+            (r'^(#)(\s*)(define)(\s+)(\w+)([(])',
              bygroups(Comment.Hashbang,
+                      Whitespace,
                       Keyword,
                       Whitespace,
                       Name.Function,
                       Punctuation),
              ('prepro', 'param_list')),
-            (r'^(#)(\w*if\w*|el[si][ef])(\s+)$',
+            (r'^(#)(\s*)(\w*if\w*|el[si][ef])(.*)$',
              bygroups(Comment.Hashbang,
+                      Whitespace,
                       Keyword,
-                      Whitespace)),
+                      Comment.Preproc)),
             (r'^(#)(.*)$',
              bygroups(Comment.Hashbang,
                       Comment.Preproc)),