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

class DirectiveGrammar(Grammar):
    rules = [
        PatternRule(r'comment', r'#(?:[^%]|%(?!\]))*'),
        PatternRule(r'keyword', r'(?:BLOCK|CALL|CASE|CATCH|CLEAR|DEBUG|DEFAULT|FINAL|FILTER|FOREACH|ELSIF|ELSE|END|GET|IF|INCLUDE|INSERT|IN|LAST|MACRO|META|NEXT|PERL|PROCESS|RAWPERL|RETURN|SET|STOP|SWITCH|TAGS|THROW|TRY|UNLESS|USE|WHILE|WRAPPER)(?![A-Z_])'),
        PatternRule(r'attribute', r'[a-zA-Z_][a-zA-Z0-9_]*(?= *=)'),
        RegionRule(r'string', r'"', StringGrammar, r'"'),
        RegionRule(r'string', r"'", StringGrammar, r"'"),
    ]

class TagGrammar(Grammar):
    rules = [
        RegionRule(r'directive', r'\[\%', DirectiveGrammar, r'%%\]'),
        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 = [
        PatternRule(r'escaped', r'&[^;]+;'),
        RegionRule(r'comment', r'<!--', Grammar, r'-->'),
        RegionRule(r'directive', r'\[\%', DirectiveGrammar, r'%%\]'),
        RegionRule(r'tt_tag', r'</?', TagGrammar, r'/?>'),
    ]

class Template(mode.Fundamental):
    name       = 'Template'
    extensions = ['.tt']
    grammar    = TemplateGrammar
    colors     = {
        'directive.start':          ('magenta', 'default', 'bold'),
        'directive.comment':        ('red', 'default', 'bold'),
        'directive.keyword':        ('yellow', 'default', 'bold'),
        'directive.attribute':      ('magenta', 'default', 'bold'),
        'directive.string.start':   ('green', 'default', 'bold'),
        'directive.string.escaped': ('magenta', 'default', 'bold'),
        'directive.string.octal':   ('magenta', 'default', 'bold'),
        'directive.string.null':    ('green', 'default', 'bold'),
        'directive.string.end':     ('green', 'default', 'bold'),
        'directive.null':           ('magenta', 'default', 'bold'),
        'directive.end':            ('magenta', 'default', 'bold'),

        'tt_tag.start':            ('default', 'default', 'bold'),
        'tt_tag.namespace':        ('magenta', 'default', 'bold'),
        'tt_tag.name':             ('blue', 'default', 'bold'),
        'tt_tag.attrname':         ('cyan', 'default', 'bold'),
        'tt_tag.string.start':     ('green', 'default', 'bold'),
        'tt_tag.string.null':      ('green', 'default', 'bold'),
        'tt_tag.string.end':       ('green', 'default', 'bold'),
        'tt_tag.end':              ('default', 'default', 'bold'),
    }
    _bindings = {
        'close-paren':   (')',),
        'close-brace':   ('}',),
        'close-bracket': (']',),
    }
install = Template.install