capnp_lexer.py 2.07 KB
Newer Older
Kenton Varda's avatar
Kenton Varda committed
1 2 3 4 5 6 7 8 9 10 11 12 13
#! /usr/bin/env python

from pygments.lexer import RegexLexer
from pygments.token import *

class CapnpLexer(RegexLexer):
    name = "Cap'n Proto lexer"
    aliases = ['capnp']
    filenames = ['*.capnp']

    tokens = {
        'root': [
            (r'#.*?$', Comment.Single),
14
            (r'@[0-9a-zA-Z]*', Name.Decorator),
Kenton Varda's avatar
Kenton Varda committed
15 16
            (r'=', Literal, 'expression'),
            (r':', Name.Class, 'type'),
17
            (r'\$', Name.Attribute, 'annotation'),
18
            (r'(struct|enum|interface|union|import|using|const|annotation|extends|in|of|on|as|with|from|fixed|bulk|realtime)\b',
Kenton Varda's avatar
Kenton Varda committed
19 20
                Token.Keyword),
            (r'[a-zA-Z0-9_.]+', Token.Name),
21
            (r'[^#@=:$a-zA-Z0-9_]+', Text),
Kenton Varda's avatar
Kenton Varda committed
22 23
        ],
        'type': [
24
            (r'[^][=;,(){}$]+', Name.Class),
Kenton Varda's avatar
Kenton Varda committed
25 26 27 28 29 30 31 32 33 34
            (r'[[(]', Name.Class, 'parentype'),
            (r'', Name.Class, '#pop')
        ],
        'parentype': [
            (r'[^][;()]+', Name.Class),
            (r'[[(]', Name.Class, '#push'),
            (r'[])]', Name.Class, '#pop'),
            (r'', Name.Class, '#pop')
        ],
        'expression': [
35
            (r'[^][;,(){}$]+', Literal),
Kenton Varda's avatar
Kenton Varda committed
36 37 38 39 40 41 42 43 44
            (r'[[(]', Literal, 'parenexp'),
            (r'', Literal, '#pop')
        ],
        'parenexp': [
            (r'[^][;()]+', Literal),
            (r'[[(]', Literal, '#push'),
            (r'[])]', Literal, '#pop'),
            (r'', Literal, '#pop')
        ],
45 46 47 48 49 50 51 52 53 54 55
        'annotation': [
            (r'[^][;,(){}=:]+', Name.Attribute),
            (r'[[(]', Name.Attribute, 'annexp'),
            (r'', Name.Attribute, '#pop')
        ],
        'annexp': [
            (r'[^][;()]+', Name.Attribute),
            (r'[[(]', Name.Attribute, '#push'),
            (r'[])]', Name.Attribute, '#pop'),
            (r'', Name.Attribute, '#pop')
        ],
Kenton Varda's avatar
Kenton Varda committed
56 57 58 59 60 61 62 63 64
    }

if __name__ == "__main__":
    from setuptools import setup, find_packages
    setup(name = "CapnpPygmentsLexer",
          version = "0.1",
          packages = find_packages(),
          py_modules = [ 'capnp_lexer' ],
          entry_points = {'pygments.lexers': 'capnp = capnp_lexer:CapnpLexer'})