pmacs3/mode/haskell.py

import commands, os.path, string, sys, traceback
import color, completer, default, mode, method, regex, tab
import buffer.emul
import window
from point import Point
from lex import Grammar, PatternRule, RegionRule, OverridePatternRule, NocasePatternRule

class CommentGrammar(Grammar): pass
CommentGrammar.rules = [
    RegionRule(r'comment', r'{-', CommentGrammar, r'-}'),
    PatternRule(r'data', r'(?:[^{-]|{(?!-)|-(?!}))+'),
]

class StringGrammar(Grammar):
    rules = [
        RegionRule(r'gap', r'\\ *\n$', Grammar, r'^ *[^ \n]'),
        NocasePatternRule(r'escaped', r'\\(?:NUL|SOH|STX|ETX|EOT|ENQ|ACK|BEL|BS|HT|LF|VT|FF|CR|SO|SI|DLE|DC1|DC2|DC3|DC4|NAK|SYN|ETB|CAN|EM|SUB|ESC|FS|GS|RS|US|SP|DEL|[&abfnrtv\\"\']|\^[a-z]|\d{1,3}|o[0-7]{0,3}|x(?:[0-9a-f]){0,2})'),
        PatternRule(r'data', r'[^\\"]+'),
    ]

class HaskellGrammar(Grammar):
    rules = [
        PatternRule(r'comment', r'--(?!>).*$'),
        RegionRule(r'comment', r'{-', CommentGrammar, r'-}'),

        PatternRule(r'integer', r'[0-9]+|0[oO][0-7]+|0[xX][0-7]+'),
        PatternRule(r'float', r'[0-9]+\.[0-9]+(?:[eE][+-]?[0-9]+)?|[0-9]+[eE][+-]?[0-9]+'),
        PatternRule(r'char', r"'[^'\\]'"),
        PatternRule(r'char', r"'(?:[^'\\]|\\(?:NUL|SOH|STX|ETX|EOT|ENQ|ACK|BEL|BS|HT|LF|VT|FF|CR|SO|SI|DLE|DC1|DC2|DC3|DC4|NAK|SYN|ETB|CAN|EM|SUB|ESC|FS|GS|RS|US|SP|DEL|[&abfnrtv\\\"\']|\^[a-z]|\d{1,3}|o[0-7]{0,3}|x(?:[0-9a-f]){0,2}))'"),

        PatternRule(r'spaces', r' +'),
        PatternRule(r'eol', r'\n'),
        PatternRule(r'delimiter', r'[\[\](){};]'),

        RegionRule(r'string', r'"', StringGrammar, r'"'),
        PatternRule(r'reserved', r"(?:_|where|type|then|of|newtype|module|let|instance|infixr|infixl|infix|in|import|if|else|do|deriving|default|data|class|case)(?![a-zA-Z0-9_'])"),
        PatternRule(r'hs_declaration', r"^[a-z_][a-zA-Z0-9_']*"),
        PatternRule(r'hs_variable', r"[a-z_][a-zA-Z0-9_']*"),
        PatternRule(r'hs_constructor', r"[A-Z][a-zA-Z0-9_']*"),
        PatternRule(r'hs_operator', r'`[a-z_][a-zA-Z0-9_\']*`'),
        PatternRule(r'operator', r'@|!|>@>|>>=|>>|=>|::|->|;|<-|\\\\|\.\.|!!|:|\+\+|\||\.|\\|>=|>|/=|==|<=|<|\|\||&&|\^\^|\*\*|##|\^|/|\*|-|\+|='),
    ]

#class HaskellOpenHugs(method.Method):
#    '''Evaluate haskell expressions'''
#    bname = '*Hugs*'
#    def _execute(self, w, **vargs):
#        a = w.application
#        if not a.has_buffer_name(self.bname):
#            b = buffer.emul.XTermBuffer(a, '/usr/bin/hugs', [], name=self.bname)
#            a.add_buffer(b)
#            window.Window(b, a)
#        b = a.bufferlist.get_buffer_by_name(self.bname)
#        if a.window().buffer is not b:
#            a.switch_buffer(b)
class HaskellOpenHugs(method.shell.Interact):
    args = []
    def _execute(self, w, **vargs):
        method.shell.Interact._execute(self, w, bname='*Hugs', cmd='hugs')

class HaskellTabber(tab.Tabber):
    pass

class Haskell(mode.Fundamental):
    modename     = 'Haskell'
    extensions   = ['.hs']
    tabwidth     = 4
    commentc     = '--'
    #tabbercls   = mode.lisp.LispTabber
    grammar      = HaskellGrammar
    opentokens  = ('delimiter',)
    opentags    = {'(': ')', '{': '}', '[': ']'}
    closetokens = ('delimiter',)
    closetags   = {')': '(', '}': '{', ']': '['}
    colors       = {
        'hs_constructor':   ('magenta', 'default', 'bold'),
        'hs_declaration':   ('blue', 'default', 'bold'),
        'hs_variable':      ('yellow', 'default', 'bold'),
        'hs_operator':      ('cyan', 'default', 'bold'),
        'string.gap.start': ('red', 'default', 'bold'),
        'string.gap.null':  ('red', 'default', 'bold'),
        'string.gap.end':   ('red', 'default', 'bold'),
    }
    _bindings = {
        'close-paren':   (')',),
        'close-brace':   ('}',),
        'close-bracket': (']',),
    }
    actions  = [HaskellOpenHugs]


install = Haskell.install