import color, mode
from lex import Grammar, PatternRule, RegionRule
from mode.perl import StringGrammar

class TagGrammar(Grammar):
    rules = [
        PatternRule(r'placeholder', r'\$[a-zA-Z0-9_]+'),
        RegionRule(r'string', r'"', Grammar, r'"'),
        RegionRule(r'string', r"'", Grammar, r"'"),
        PatternRule(r'namespace', r'[a-zA-Z_]+:'),
        PatternRule(r'attrname', r'[^ =>\n]+(?==)'),
        PatternRule(r'name', r'[^\[\] =>\n]+'),
    ]

class TemplateGrammar(Grammar):
    rules = [
        RegionRule(r'comment', r'#\*', Grammar, r'\*#'),
        PatternRule(r'comment', r'##.*\n'),
        PatternRule(r'placeholder', r'\$[a-zA-Z0-9_]+'),
        PatternRule(r'directive', r'#.*?[#\n]'),
        RegionRule(r'tag', r'</?', TagGrammar, r'/?>'),
    ]

class Template(mode.Fundamental):
    modename   = 'Cheetah'
    extensions = ['.tmpl']
    grammar    = TemplateGrammar
    colors     = {
        'comment':       ('red', 'default'),
        'comment.start': ('red', 'default'),
        'comment.null':  ('red', 'default'),
        'comment.end':   ('red', 'default'),

        'directive':   ('magenta', 'default'),
        'placeholder': ('magenta', 'default'),

        'tag.start':            ('default', 'default'),
        'tag.namespace':        ('magenta', 'default'),
        'tag.name':             ('blue', 'default'),
        'tag.attrname':         ('cyan', 'default'),
        'tag.string.start':     ('green', 'default'),
        'tag.string.null':      ('green', 'default'),
        'tag.string.end':       ('green', 'default'),
        'tag.end':              ('default', 'default'),
    }
    def __init__(self, w):
        mode.Fundamental.__init__(self, w)
        self.add_bindings('close-paren', (')',))
        self.add_bindings('close-brace', ('}',))
        self.add_bindings('close-bracket', (']',))

install = Template.install