import color, mode
from lex import Grammar, PatternRule, RegionRule, PatternMatchRule
from mode.python import StringGrammar1, StringGrammar2, StringGrammar3, StringGrammar4

chr1 = '[a-zA-Z_.?]'
chr2 = '[a-zA-Z0-9_.?$#@~]'
word = chr1 + chr2 + '*'

class NasmGrammar(Grammar):
    rules = [
        PatternRule('continuation', r'\\\n$'),
        PatternRule('nasm_keyword', "(?:section|global|extern)(?!" + chr2 + ")"),
        PatternRule('macros', "%(?:define|undef|assign|strlen|macro|endmacro|if|elif|else|endif|ifdef|ifndef|include|push|pop|stacksize)(?!" + chr2 + ")"),
        PatternRule('instructions', "(?:jeq|jne|ja|jmp|push|pushad|pushfd|call|ret|sub|add|pop|popa|popad|popfd|call|and|cwd|cdq|cmp|cmpxchg|cpuid|div|divpd|enter|leave|fadd|fld|fmul|fsqrt|fsub|hlt|imul|inc|int|int3|lea|mov|movd|mul|neg|not|nop|or|sal|sar|shl|shr|shld|shrd|syscall|sysenter|sysexit|test|xchg|xadd|xor)(?!" + chr2 + ")"),
        PatternRule('registers', "(?:eax|ax|ah|al|ebx|bx|bh|bl|ecx|cx|ch|cl|esi|edi|esp|ebp)(?!" + chr2 + ")"),
        PatternRule('prefix', "(?:dword|word|lock)(?!" + chr2 + ")"),
        PatternMatchRule('x', '(' + word + ')(:)', 'nasm_label', 'delimiter'),
        #PatternRule('nasm_label', word + ":"),
        PatternRule("identifier", r'\$?' + word),
        PatternRule("integer", "(0|[1-9][0-9]*|0[0-7]+|0[xX][0-9a-fA-F]+)[lL]?"),
        PatternRule("float", r"[0-9]+\.[0-9]*|\.[0-9]+|([0-9]|[0-9]+\.[0-9]*|\.[0-9]+)[eE][\+-]?[0-9]+"),
        RegionRule('string', "'''", StringGrammar3, "'''"),
        RegionRule('string', '"""', StringGrammar4, '"""'),
        RegionRule('string', "'", StringGrammar1, "'"),
        RegionRule('string', '"', StringGrammar2, '"'),
        PatternRule('comment', ';.*$'),
    ]

class Nasm(mode.Fundamental):
    name       = 'nasm'
    extensions = ['.s']
    grammar    = NasmGrammar
    commentc   = ';'
    colors     = {
        'nasm_keyword': ('cyan', 'default', 'bold'),
        'macros':       ('blue', 'default', 'bold'),
        'registers':    ('yellow', 'default', 'bold'),
        'instructions': ('magenta', 'default', 'bold'),
        'nasm_label':   ('blue', 'default', 'bold'),
    }

install = Nasm.install