summaryrefslogtreecommitdiff
path: root/lexer.py
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