SB05-065: add lkql lexer

Author: raph-amiard

user_manual/source/conf.py

@@ -19,6 +19,9 @@
 
 import os
 import sys
+from pygments.lexer import RegexLexer, words
+from pygments import token
+from sphinx.highlighting import lexers
 
 dir_path = os.path.dirname(os.path.realpath(__file__))
 sys.path.append(dir_path)
@@ -28,6 +31,35 @@
 author = 'Raphael Amiard'
 
 
+class LKQLPygmentsLexer(RegexLexer):
+    """
+    Pygments lexer for LKQL
+    """
+    name = 'LKQL'
+    filenames = ['*.lkql']
+
+    tokens = {
+        'root': [
+            (words(('select', 'let', 'when', 'val', 'fun', 'selector',
+                    'match', 'rec', 'skip', 'is', 'in', 'true', 'false',
+                    'if', 'else', 'then', 'not', 'null'),
+                   prefix=r'\b', suffix=r'\b'),
+             token.Keyword),
+            (r"#(.?)+", token.Comment),
+            (r"(\-\>|=|\=\>|\<\=|\>\=|\=|\!\=|\+|\-|\*|\/|\&|"
+             r"\@|\||\>|\<)", token.Operator),
+            (r"\b(and|or|not)\b", token.Operator),
+            (r"\{|\}|\(|\)|\[|\]|;|\.|,", token.Punctuation),
+            (r"\"[^\"]*\"", token.String),
+            (r'[0-9]+', token.Number),
+            (r'_?[a-zA-Z][\w\']*', token.Name),
+            (r'\n', token.Text),
+            (r'[^\S\n]+', token.Text),
+        ]
+    }
+
+lexers['lkql'] = LKQLPygmentsLexer()
+
 # -- General configuration ---------------------------------------------------
 
 # Add any Sphinx extension module names here, as strings. They can be