import color, mode2
import lex
from lex2 import Grammar, PatternRule, RegionRule
# this might not be complete...
# see http://gcc.gnu.org/onlinedocs/gcc-2.95.3/cpp_3.html#SEC44

class MacroGrammar(Grammar):
    rules = [
        PatternRule(name='continued', pattern=r'.*\\ *$'),
    ]
class StringGrammar(Grammar):
    rules = [
        PatternRule(name=r'escaped', pattern=r'\\.'),
    ]

class CGrammar(Grammar):
    rules = [
        #PatternRule(name=r'macro1', pattern=r'#(?:define|import|include|undef)(?!=[a-zA-Z0-9_])'),
        RegionRule(
            name=r'macro2',
            start=r"#(?:assert|cpu|elif|else|error|endif|ident|ifdef|ifndef|if|include_next|line|machine|pragma|pragma_once|system|unassert|warning)(?!=[a-zA-Z0-9_])",
            grammar=Grammar(),
            end=r'(?<!\\) *$',
        ),
        PatternRule(name=r'include', pattern=r'#include(?!=[a-zA-Z0-9_])'),
        PatternRule(name=r'header', pattern=r'<[-A-Za-z/0-9_\.]+>|"[-A-Za-z/0-9_\.]+"'),
        RegionRule(
            name=r'macro',
            start=r'#(?:assert|cpu|define|elif|else|error|endif|ident|ifdef|ifndef|if|import|include_next|line|machine|pragma|pragma_once|system|unassert|undef|warning)(?!=[a-zA-Z0-9_])',
            grammar=MacroGrammar(),
            end=r'(?:.*[^\\])?$',
        ),
        PatternRule(name=r'label', pattern=r'[a-zA-Z_][a-zA-Z0-9_]*(?=:)'),
        RegionRule(name=r'comment1', start='/\*', grammar=Grammar(), end='\*/'),
        PatternRule(name=r'comment2', pattern=r'//.*$'),
        PatternRule(name=r'control', pattern=r"(?:break|case|continue|default|do|else|for|goto|if|return|switch|while)(?![a-zA-Z_])"),
        PatternRule(name=r'keyword', pattern=r"(?:auto|break|case|char|const|continue|default|do|double|else|enum|extern|float|for|goto|if|int|long|register|return|short|signed|sizeof|static|struct|switch|typedef|union|unsigned|void|volatile|while)(?![a-zA-z_])"),
        PatternRule(name=r'builtin', pattern=r"(?:NULL|TRUE|FALSE)"),
        PatternRule(name=r'identifier', pattern=r"[a-zA-Z_][a-zA-Z0-9_]*"),
        PatternRule(name=r"unop", pattern=r"\+=|-=|\*=|/=|//=|%=|&=\|\^=|>>=|<<=|\*\*="),
        PatternRule(name=r'binop', pattern=r"\+|<>|<<|<=|<|-|>>|>=|>|\*\*|&|\*|\||/|\^|==|//|~|!=|%"),
        # this is sketchy as hell
        PatternRule(name=r"delimiter", pattern=r"->|\.|\(|\)|\[|\]|{|}|@|,|:|`|;|=|\?"),
        PatternRule(name=r"integer", pattern=r"(?:0(?![x0-9])|[1-9][0-9]*|0[0-7]+|0[xX][0-9a-fA-F]+)[lL]?"),
        PatternRule(name=r"float", pattern=r"[0-9]+\.[0-9]*|\.[0-9]+|(?:[0-9]|[0-9]+\.[0-9]*|\.[0-9]+)[eE][\+-]?[0-9]+"),
        RegionRule(name=r'string', start='"', grammar=StringGrammar(), end='"'),
        PatternRule(name=r'char', pattern=r"'.'|'\\.'|'\\[0-7]{3}'")
    ]

class C(mode2.Fundamental):
    #tabbercls  = CTabber
    grammar = CGrammar()
    opentoken  = 'delimiter'
    opentags   = {'(': ')', '[': ']', '{': '}'}
    closetoken = 'delimiter'
    closetags  = {')': '(', ']': '[', '}': '{'}
    def __init__(self, w):
        mode2.Fundamental.__init__(self, w)
        self.add_bindings('close-paren', (')',))
        self.add_bindings('close-brace', ('}',))
        self.add_bindings('close-bracket', (']',))
        self.colors = {
            'include': color.build('blue', 'default'),
            'header':  color.build('green', 'default'),

            'macro.start':     color.build('blue', 'default', 'bold'),
            'macro.continued': color.build('default', 'default', 'bold'),
            'macro.end':       color.build('default', 'default', 'bold'),

            'control':        color.build('blue', 'default', 'bold'),
            'keyword':        color.build('cyan', 'default', 'bold'),
            'macro1':         color.build('blue', 'default', 'bold'),
            'macro2':         color.build('blue', 'default', 'bold'),
            'constant':       color.build('magenta', 'default', 'bold'),
            'header':         color.build('green', 'default', 'bold'),
            'label':          color.build('magenta', 'default', 'bold'),
            'char':           color.build('green', 'default'),
            'builtin':        color.build('magenta', 'default', 'bold'),
            'comment1.start': color.build('red', 'default'),
            'comment1.end':   color.build('red', 'default'),
            'comment1.null':  color.build('red', 'default'),
            'comment2':       color.build('red', 'default'),
            'string.start':   color.build('green', 'default'),
            'string.escaped': color.build('green', 'default'),
            'string.null':    color.build('green', 'default'),
            'string.end':     color.build('green', 'default'),
            'macro comment':  color.build('red', 'default'),
            'function name':  color.build('blue', 'default'),
            'integer':        color.build('green', 'default'),
            'float':          color.build('green', 'default'),
            'bizzaro':        color.build('magenta', 'green'),
        }
    def name(self):
        return "C"