summaryrefslogtreecommitdiff
path: root/lexer.py
blob: 8a2cbb44df52d787bb3ef02b1b518965dc8034a5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#!/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 = [
        'arb',
        'fold',
        'foldmatch',
        'inj',
        'map',
        'max',
        'primrec',
        'prj',
        '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