pmacs3/mode_nasm.py

import color, mode2
from lex3 import Grammar, PatternRule, RegionRule

class StringGrammar(Grammar):
    rules = [
        PatternRule(r'octal', r'\\[0-7]{3}'),
        PatternRule(r'escaped', r'\\.'),
    ]

class NasmGrammar(Grammar):
    rules = [
        PatternRule(r'keyword', r"(?:section|global|extern)(?![a-zA-Z_])"),
        PatternRule(r'macros', r"%(?:define|undef|assign|strlen|macro|endmacro|if|elif|else|endif|ifdef|ifndef|include|push|pop|stacksize)(?![a-zA-Z_])"),
        PatternRule(r'instructions', r"(?: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)(?![a-zA-Z_])"),
        PatternRule(r'registers', r"(?:eax|ax|ah|al|ebx|bx|bh|bl|ecx|cx|ch|cl|esi|edi|esp|ebp)(?![a-zA-Z_])"),
        PatternRule(r'prefix', r"(?:dword|word|lock)(?![a-zA-Z_])"),
        PatternRule(r'label', r"[a-zA-Z_.][a-zA-Z0-9_.]*:"),
        PatternRule(r"identifier", r"[a-zA-Z_][a-zA-Z0-9_]*"),
        PatternRule(r"integer", r"(0|[1-9][0-9]*|0[0-7]+|0[xX][0-9a-fA-F]+)[lL]?"),
        PatternRule(r"float", r"[0-9]+\.[0-9]*|\.[0-9]+|([0-9]|[0-9]+\.[0-9]*|\.[0-9]+)[eE][\+-]?[0-9]+"),
        RegionRule(r'string', r'"""', StringGrammar, r'"""'),
        RegionRule(r'string', r"'''", StringGrammar, r"'''"),
        RegionRule(r'string', r'"', StringGrammar, r'"'),
        RegionRule(r'string', r"'", StringGrammar, r"'"),
        PatternRule(r'comment', r';.*$'),
    ]


class Nasm(mode2.Fundamental):
    grammar = NasmGrammar
    def __init__(self, w):
        mode2.Fundamental.__init__(self, w)
        self.colors = {
            'keyword':      color.build('cyan', 'default', 'bold'),
            'macros':       color.build('blue', 'default', 'bold'),
            'string.start': color.build('green', 'default'),
            'string.null':  color.build('green', 'default'),
            'string.end':   color.build('green', 'default'),
            'comment':      color.build('red', 'default'),
            'registers':    color.build('yellow', 'default'),
            'instructions': color.build('magenta', 'default'),
            'label':        color.build('blue', 'default'),
        }
    def name(self):
        return "Nasm"