blob: a16a38a1dbc29b5cc75b167c9f13524d400976eb (
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']
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
|