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'cheetah_placeholder', r'\$[a-zA-Z0-9_]+'),
        PatternRule(r'cheetah_directive', r'#.*?[#\n]'),
        RegionRule(r'cheetah_tag', r'</?', TagGrammar, r'/?>'),
    ]

class Template(mode.Fundamental):
    modename   = 'Cheetah'
    extensions = ['.tmpl']
    grammar    = TemplateGrammar
    colors     = {
        'cheetah_directive':        ('magenta', 'default'),
        'cheetah_placeholder':      ('magenta', 'default'),
        'cheetah_tag.start':        ('default', 'default'),
        'cheetah_tag.namespace':    ('magenta', 'default'),
        'cheetah_tag.name':         ('blue', 'default'),
        'cheetah_tag.attrname':     ('cyan', 'default'),
        'cheetah_tag.string.start': ('green', 'default'),
        'cheetah_tag.string.null':  ('green', 'default'),
        'cheetah_tag.string.end':   ('green', 'default'),
        'cheetah_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