pmacs3/mode/sh.py

import commands
import color, mode, tab
from lex import Grammar, PatternRule, RegionRule
from method import Method

class StringGrammar(Grammar):
    rules = [
        PatternRule(r'escaped', r'\\.'),
        PatternRule(r'variable', r"\${(?:[a-zA-Z0-9_]+|\?\$)}"),
        PatternRule(r"variable", r"\$[^({][a-zA-Z0-9_]*"),
        PatternRule(r'variable', r"\$(?=\()"),
    ]

class EvalGrammar(Grammar):
    rules = [
        RegionRule(r'string', "'", Grammar, "'"),
        RegionRule(r'string', '"', StringGrammar, '"'),
        PatternRule(r'escaped', r'\\.'),
        PatternRule(r'variable', r"\${(?:[a-zA-Z0-9_]+|\?\$)}"),
        PatternRule(r"variable", r"\$[^({][a-zA-Z0-9_]*"),
        PatternRule(r'variable', r"\$(?=\()"),
    ]

class CaseGrammar(Grammar):
    rules = [
        RegionRule(r'stanza', r'.+\)', Grammar, r';;'),
    ]

class TestGrammar(Grammar):
    rules = [
        PatternRule(r'sh_builtin', r"(?:source|alias|bg|bind|break|builtin|cd|command|compgen|complete|declare|dirs|disown|echo|enable|eval|exec|exit|export|fc|fg|getops|hash|help|history|jobs|kill|let|local|logout|popd|printf|pushd|pwd|readonly|read|return|set|shift|shopt|suspend|test|times|trap|type|ulimit|umask|unalias|unset|wait)(?![a-zA-Z0-9_=/])"),
        PatternRule(r'binop', r'==|=|!=|<|>'),
        PatternRule(r'binop', r'-(?:nt|ot|ef|eq|ne|lt|gt|le|ge)(?![a-zA-Z0-9_])'),
        PatternRule(r'unop', r'-[a-zA-Z](?![a-zA-Z0-9_])'),
        PatternRule(r'continuation', r'\\\n$'),
        PatternRule(r'delimiter', r";;|[();{}|&><]"),
        RegionRule(r'eval', r'`', StringGrammar, r'`'),
        RegionRule(r'neval', r'\$\(', StringGrammar, r'\)'),
        PatternRule(r'variable', r"(?:^|(?<= ))[a-zA-Z_][a-zA-Z0-9_]*(?==)"),
        PatternRule(r'variable', r"\${(?:[a-zA-Z0-9_]+|\?\$)}"),
        PatternRule(r"variable", r"\$[^({][a-zA-Z0-9_]*"),
        PatternRule(r'variable', r"\$(?=\()"),
        RegionRule(r'string', "'", Grammar, "'"),
        RegionRule(r'string', '"', StringGrammar, '"'),
        PatternRule(r'bareword', r'[-a-zA-Z0-9_.]+'),
    ]

class ShGrammar(Grammar):
    rules = [
        RegionRule(r'heredoc', r"<<(?P<heredoc>[a-zA-Z0-9_]+)", StringGrammar, r'^%(heredoc)s$'),
        PatternRule(r'sh_function', r'[a-zA-Z_][a-zA-Z0-9_]*(?= *\(\))'),
        PatternRule(r'sh_reserved', r"(?:done|do|elif|else|esac|fi|for|function|if|in|select|then|until|while|time)(?![a-zA-Z0-9_=/])"),
        RegionRule(r'case', r'case', Grammar, 'in', CaseGrammar, r'esac'),
        PatternRule(r'sh_builtin', r"(?:source|alias|bg|bind|break|builtin|cd|command|compgen|complete|declare|dirs|disown|echo|enable|eval|exec|exit|export|fc|fg|getops|hash|help|history|jobs|kill|let|local|logout|popd|printf|pushd|pwd|readonly|read|return|set|shift|shopt|suspend|times|trap|type|ulimit|umask|unalias|unset|wait)(?![a-zA-Z0-9_=/])"),
        RegionRule(r'test', r'test', TestGrammar, r'\n'),
        RegionRule(r'test2', r'\[', TestGrammar, r'\]'),
        PatternRule(r'delimiter', r";;|[();{}|&><]"),
        #RegionRule(r'eval', r'`', StringGrammar, r'`'),
        #RegionRule(r'neval', r'\$\(', StringGrammar, r'\)'),
        RegionRule(r'eval', r'`', EvalGrammar, r'`'),
        RegionRule(r'neval', r'\$\(', EvalGrammar, r'\)'),
        PatternRule(r'variable', r"(?:^|(?<= ))[a-zA-Z_][a-zA-Z0-9_]*(?==)"),
        PatternRule(r'variable', r"\${(?:[a-zA-Z0-9_]+|\?\$)}"),
        PatternRule(r"variable", r"\$[^({][a-zA-Z0-9_]*"),
        PatternRule(r'variable', r"\$(?=\()"),
        RegionRule(r'string', "'", Grammar, "'"),
        RegionRule(r'string', '"', StringGrammar, '"'),
        PatternRule(r'comment', r'#.*$'),
        PatternRule(r'bareword', r'[-a-zA-Z0-9_.]+'),
        PatternRule(r'continuation', r'\\\n$'),
        PatternRule(r'eol', r'\n$'),
    ]

# hacks to get some circular grammar refs
CaseGrammar.rules[0].pairs[0] = (ShGrammar, CaseGrammar.rules[0].pairs[0][1])
#ShGrammar.rules[2].pairs[0]   = (ShGrammar, ShGrammar.rules[0].pairs[0][1])

class ShTabber(tab.StackTabber):
    def is_base(self, y):
        if y == 0:
            return True
        highlighter = self.mode.window.buffer.highlights[self.mode.name()]
        if not highlighter.tokens[y]:
            return False
        t = highlighter.tokens[y][0]
        return t.name == 'sh_function'
    def _handle_close_token(self, currlvl, y, i):
        s = self.get_token(y, i).string
        if s == ')' and self.markers and self._peek_name() == "case":
            # we have to ignore ) when used in "case" statements.
            return currlvl
        else:
            return tab.StackTabber._handle_close_token(self, currlvl, y, i)
    def _handle_other_token(self, currlvl, y, i):
        token  = self.get_token(y, i)
        fqname = token.fqname()
        if token.name == 'continuation':
            self._opt_append("cont", currlvl + 4)
        elif token.name == 'sh_reserved' and token.string == 'else':
            currlvl -= 4
        elif token.name == 'eol':
            self._opt_pop("cont")
        return currlvl

class Sh(mode.Fundamental):
    modename    = 'sh'
    paths       = ['/etc/profile']
    basenames   = ['.bashrc', '.bash_profile', '.profile']
    extensions  = ['.bash', '.sh']
    detection   = ['sh', 'bash']
    grammar     = ShGrammar
    tabbercls   = ShTabber
    opentokens  = ('delimiter', 'sh_reserved', 'case.start')
    opentags    = {'(': ')', '[': ']', '{': '}', 'do': 'done', 'then': 'fi', 'case': 'esac'}
    closetokens = ('delimiter', 'sh_reserved', 'case.end') 
    closetags   = {')': '(', ']': '[', '}': '{',
                   'done': 'do', 'fi': 'then', 'esac': 'case'}
    colors = {
        'sh_builtin':         ('cyan', 'default', 'bold'),
        'sh_function':        ('magenta', 'default', 'bold'),
        'sh_reserved':        ('magenta', 'default', 'bold'),
        'variable':        ('yellow', 'default', 'bold'),

        # case statements
        'case.start':   ('magenta', 'default', 'bold'),
        'case.middle0': ('magenta', 'default', 'bold'),
        'case.end':     ('magenta', 'default', 'bold'),

        #'delimiter':  ('default', 'default', 'bold'),
        'test.start': ('cyan', 'default', 'bold'),
        'binop':      ('magenta', 'default', 'bold'),
        'unop':       ('magenta', 'default', 'bold'),

        #'string.start':    ('green', 'default'),
        #'string.variable': ('yellow', 'default'),
        #'string.escaped':  ('magenta', 'default'),
        #'string.null':     ('green', 'default'),
        #'string.end':      ('green', 'default'),

        'eval.start':      ('cyan', 'default'),
        'eval.variable':   ('yellow', 'default'),
        'eval.null':       ('cyan', 'default'),
        'eval.end':        ('cyan', 'default'),

        'neval.start':      ('yellow', 'default'),
        'neval.variable':   ('yellow', 'default'),
        'neval.null':       ('cyan', 'default'),
        'neval.end':        ('yellow', 'default'),

        #'comment':         ('red', 'default'),
        #'continuation':    ('red', 'default'),
    }
    def __init__(self, w):
        mode.Fundamental.__init__(self, w)
        self.add_action_and_bindings(ShCheckSyntax(), ('C-c s',))

class ShCheckSyntax(Method):
    '''Check the syntax of a shell script'''
    def _execute(self, w, **vargs):
        app = w.application
        cmd = "bash -n %r" % w.buffer.path
        (status, output) = commands.getstatusoutput(cmd)
        if status == 0:
            app.set_error("Syntax OK")
            app.data_buffer("*Sh-Check-Syntax*", output, switch_to=False)
        else:
            app.data_buffer("*Sh-Check-Syntax*", output)

install = Sh.install