blob: 6963b77d0e04fc186d2cf87d9cb128cb7eb08663 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#!/usr/bin/env python3
from pygments.lexers.ml import OcamlLexer
from pygments.token import Name, Keyword
from pathlib import Path
class SystemTLexer(OcamlLexer):
name = 'System T'
aliases = ['syst']
filenames = ['*.syst']
EXTRA_KEYWORDS = ['primrec', 'fold', 'foldmatch', 'roll', 'roll2']
def get_tokens_unprocessed(self, text):
for index, token, value in OcamlLexer.get_tokens_unprocessed(self, text):
if token is Name and value in self.EXTRA_KEYWORDS:
yield index, Keyword, value
else:
yield index, token, value
|