2009-03-29 20:50:27 -04:00
|
|
|
import os.path
|
2008-10-06 12:32:18 -04:00
|
|
|
from subprocess import Popen, PIPE, STDOUT
|
2009-03-29 20:50:27 -04:00
|
|
|
from mode import Fundamental
|
|
|
|
import default
|
|
|
|
from lex import Grammar, PatternRule, RegionRule, PatternMatchRule
|
2008-10-01 16:58:43 -04:00
|
|
|
from mode.python import StringGrammar2
|
2008-10-02 12:49:22 -04:00
|
|
|
from tab import StackTabber2
|
2008-10-04 12:09:15 -04:00
|
|
|
from method import Method, Argument, arg
|
|
|
|
from method.shell import Exec, Pipe
|
2008-10-01 16:58:43 -04:00
|
|
|
|
2009-03-29 20:50:27 -04:00
|
|
|
chr1 = '[a-zA-Z_]'
|
|
|
|
chr2 = '[a-zA-Z0-9_]'
|
|
|
|
word = chr1 + chr2 + '*'
|
|
|
|
|
2008-10-01 18:10:11 -04:00
|
|
|
class RegexGrammar(Grammar):
|
|
|
|
rules = [
|
|
|
|
PatternRule(r'escaped', r'\\.'),
|
|
|
|
PatternRule(r'data', r'[^\\/]+'),
|
|
|
|
]
|
|
|
|
|
2008-10-01 16:58:43 -04:00
|
|
|
class AwkGrammar(Grammar):
|
|
|
|
rules = [
|
2009-04-06 11:21:43 -04:00
|
|
|
PatternRule('comment', r'#.*\n$'),
|
|
|
|
PatternRule('spaces', ' +'),
|
|
|
|
RegionRule('awk.regex', '/(?! )', RegexGrammar, '/'),
|
2009-03-29 20:50:27 -04:00
|
|
|
|
2009-04-06 11:21:43 -04:00
|
|
|
PatternMatchRule('x', '(function)( +)(' + word + ')',
|
|
|
|
'awk.keyword', 'spaces', 'awk.function'),
|
2009-03-29 20:50:27 -04:00
|
|
|
|
2009-04-06 11:21:43 -04:00
|
|
|
PatternRule('awk.global', '(?:TEXTDOMAIN|SUBSEP|RLENGTH|RSTART|RT|RS|PROCINFO|ORS|OFS|OFMT|NR|NF|LINT|IGNORECASE|FS|FNR|FILENAME|FIELDWIDTHS|ERRNO|ENVIRON|CONVFMT|BINMODE|ARGV|ARGIND|ARGC)(?!' + chr2 + ')'),
|
|
|
|
PatternRule('delimiter', r'(?:[\{\}()\[\]?:;,]|=(?!=)|\+=|-=|\*=|/=|\%=|\^=)'),
|
|
|
|
PatternRule('awk.keyword', '(?:BEGIN|END|function|if|else|while|do|for|break|continue|delete|exit)(?!' + chr2 + ')'),
|
|
|
|
PatternRule('awk.builtin', '(?:return|close|getline|nextfile|next|printf|print|system|fflush|atan2|cos|exp|int|log|rand|sin|sqrt|srand|asorti|asort|gensub|gsub|index|length|match|split|sprintf|strtonum|substr|sub|tolower|toupper|mktime|strftime|systime|and|compl|lshift|or|xor|rshift|bindtextdomain|dcgettext|dcngettext|extension)(?!' + chr2 + ')'),
|
2008-10-01 23:59:30 -04:00
|
|
|
|
2009-04-06 11:21:43 -04:00
|
|
|
PatternRule('awk.field', r'\$\d*'),
|
2008-10-01 18:15:52 -04:00
|
|
|
|
2009-04-06 11:21:43 -04:00
|
|
|
PatternRule('awk.number', '-?0x[0-9A-Fa-f]+'),
|
|
|
|
PatternRule('awk.number', '-?0[0-7]*'),
|
|
|
|
PatternRule('awk.number', r'-?[0-9]+\.?[0-9]*'),
|
|
|
|
PatternRule('awk.number', r'-?\.[0-9]+'),
|
2008-10-01 18:15:52 -04:00
|
|
|
|
2009-04-06 11:21:43 -04:00
|
|
|
PatternRule('awk.operator', r'!(?![=~])|--|\+\+'),
|
|
|
|
PatternRule('awk.operator', r'(?:&&|\|\||<=|>=|!=|!~|==|\^|%|[-~/+*<>])'),
|
2009-03-29 20:50:27 -04:00
|
|
|
|
2009-04-06 11:21:43 -04:00
|
|
|
RegionRule('awk.string', '"', StringGrammar2, '"'),
|
|
|
|
PatternRule('awk.function', word + r'(?=\()'),
|
|
|
|
PatternRule('awk_identifier', word),
|
2008-10-01 18:15:52 -04:00
|
|
|
|
2009-04-06 11:21:43 -04:00
|
|
|
PatternRule('continuation', r'\\\n$'),
|
|
|
|
PatternRule('eol', r'\n'),
|
2008-10-01 16:58:43 -04:00
|
|
|
]
|
|
|
|
|
2008-10-02 12:49:22 -04:00
|
|
|
class AwkTabber(StackTabber2):
|
|
|
|
open_tokens = {'delimiter': {'{': '}', '(': ')', '[': ']'}}
|
|
|
|
close_tokens = {'delimiter': {'}': '{', ')': '(', ']': '['}}
|
|
|
|
control_tokens = {
|
2009-04-06 11:21:43 -04:00
|
|
|
'awk.keyword': set(['if', 'else', 'while', 'do', 'for'])
|
2008-10-02 12:49:22 -04:00
|
|
|
}
|
|
|
|
end_at_eof = True
|
|
|
|
end_at_tokens = {}
|
2008-10-01 23:59:30 -04:00
|
|
|
def _is_base(self, y):
|
2009-03-29 20:50:27 -04:00
|
|
|
if y == 0: return True
|
2008-10-02 09:35:28 -04:00
|
|
|
t = self._get_tokens(y)[0]
|
2009-04-06 11:21:43 -04:00
|
|
|
if t.fqisa('awk.regex.start'): return True
|
|
|
|
if t.isa('awk.field', 'awk.global'): return True
|
|
|
|
if t.matchs('awk.keyword', ('BEGIN', 'END')): return True
|
|
|
|
|
2009-03-29 20:50:27 -04:00
|
|
|
return False
|
2009-04-06 11:21:43 -04:00
|
|
|
def _is_indent(self, t): return t.isa('spaces')
|
|
|
|
def _is_ignored(self, t): return t.isa('spaces', 'eol', 'comment')
|
2008-10-02 09:35:28 -04:00
|
|
|
|
2008-10-04 12:09:15 -04:00
|
|
|
class AwkFilterFile(Exec):
|
2009-03-27 23:40:09 -04:00
|
|
|
'''Filter a file through the current buffer's AWK program'''
|
2008-10-04 12:09:15 -04:00
|
|
|
show_success = True
|
2009-03-29 20:50:27 -04:00
|
|
|
args = [arg('path', dt="path", p="Filter File: ", dv=default.path_dirname,
|
|
|
|
ld=True, h="file to open")]
|
2008-10-04 12:09:15 -04:00
|
|
|
def _execute(self, w, **vargs):
|
|
|
|
if not hasattr(w.buffer, 'path'):
|
2008-10-06 12:32:18 -04:00
|
|
|
w.set_error("Buffer %r has no program" % w.buffer.name())
|
2008-10-04 12:09:15 -04:00
|
|
|
return
|
|
|
|
elif w.buffer.changed():
|
2008-10-06 12:32:18 -04:00
|
|
|
w.set_error("Buffer %r is not saved" % w.buffer.name())
|
2008-10-04 12:09:15 -04:00
|
|
|
return
|
2008-10-06 12:32:18 -04:00
|
|
|
elif not os.path.exists(vargs['path']):
|
|
|
|
w.set_error("File %r not found" % vargs['path'])
|
|
|
|
return
|
|
|
|
|
|
|
|
cmd = 'awk -f %r %r' % (w.buffer.path, vargs['path'])
|
|
|
|
pipe = Popen(cmd, shell=True, stdout=PIPE, stderr=STDOUT)
|
|
|
|
outdata = pipe.stdout.read()
|
|
|
|
status = pipe.wait() >> 8
|
|
|
|
w.application.data_buffer('*Awk-Output*', outdata, switch_to=True)
|
|
|
|
w.set_error("awk exited with status %d" % status)
|
2008-10-04 12:09:15 -04:00
|
|
|
|
2008-10-06 00:55:38 -04:00
|
|
|
class AwkFilterBuffer(Pipe):
|
2009-03-27 23:40:09 -04:00
|
|
|
'''Filter a buffer through the current buffer's AWK program'''
|
2008-10-06 12:32:18 -04:00
|
|
|
args = [arg('name', dt="buffer", p="Filter Buffer: ", h="name of the buffer to switch to")]
|
|
|
|
def _execute(self, w, **vargs):
|
|
|
|
if not hasattr(w.buffer, 'path'):
|
|
|
|
w.set_error("Buffer %r has no program" % w.buffer.name())
|
|
|
|
return
|
|
|
|
elif w.buffer.changed():
|
|
|
|
w.set_error("Buffer %r is not saved" % w.buffer.name())
|
|
|
|
return
|
|
|
|
elif not w.application.has_buffer_name(vargs['name']):
|
|
|
|
w.set_error("Buffer %r not found" % vargs['name'])
|
|
|
|
return
|
|
|
|
|
|
|
|
b = w.application.bufferlist.get_buffer_by_name(vargs['name'])
|
|
|
|
opts = w.application.config['awk.cmd-opts']
|
|
|
|
cmd = 'awk %s -f %r' % (opts, w.buffer.path)
|
|
|
|
pipe = Popen(cmd, shell=True, stdin=PIPE, stdout=PIPE, stderr=STDOUT)
|
|
|
|
pipe.stdin.write(b.make_string())
|
|
|
|
pipe.stdin.close()
|
|
|
|
output = pipe.stdout.read()
|
|
|
|
status = pipe.wait() >> 8
|
|
|
|
w.application.data_buffer('*Awk-Output*', output, switch_to=True)
|
|
|
|
w.set_error("awk exited with status %d" % status)
|
|
|
|
|
2008-10-04 12:09:15 -04:00
|
|
|
class AwkFilterInput(Method):
|
2009-03-27 23:40:09 -04:00
|
|
|
'''Filter input through the current buffer's AWK program'''
|
2008-10-06 12:32:18 -04:00
|
|
|
args = [arg('input', p="Data To Filter: ", h="data to filter")]
|
|
|
|
def _execute(self, w, **vargs):
|
|
|
|
if not hasattr(w.buffer, 'path'):
|
|
|
|
w.set_error("Buffer %r has no program" % w.buffer.name())
|
|
|
|
return
|
|
|
|
elif w.buffer.changed():
|
|
|
|
w.set_error("Buffer %r is not saved" % w.buffer.name())
|
|
|
|
return
|
|
|
|
|
|
|
|
opts = w.application.config['awk.cmd-opts']
|
|
|
|
cmd = 'awk %s -f %r' % (opts, w.buffer.path)
|
|
|
|
pipe = Popen(cmd, shell=True, stdin=PIPE, stdout=PIPE, stderr=STDOUT)
|
|
|
|
pipe.stdin.write(vargs['input'])
|
|
|
|
pipe.stdin.close()
|
|
|
|
output = pipe.stdout.read()
|
|
|
|
status = pipe.wait() >> 8
|
|
|
|
w.application.data_buffer('*Awk-Output*', output, switch_to=True)
|
|
|
|
w.set_error("awk exited with status %d" % status)
|
2008-10-01 23:59:30 -04:00
|
|
|
|
2009-03-29 20:50:27 -04:00
|
|
|
class Awk(Fundamental):
|
2009-03-17 15:24:10 -04:00
|
|
|
name = 'awk'
|
2008-10-02 09:35:28 -04:00
|
|
|
tabbercls = AwkTabber
|
2009-02-15 12:06:35 -05:00
|
|
|
extensions = ['.awk']
|
|
|
|
grammar = AwkGrammar
|
|
|
|
opentokens = ('delimiter',)
|
|
|
|
opentags = {'(': ')', '[': ']', '{': '}'}
|
|
|
|
closetokens = ('delimiter',)
|
|
|
|
closetags = {')': '(', ']': '[', '}': '{'}
|
|
|
|
config = {'awk.cmd-opts': ""}
|
|
|
|
actions = [AwkFilterFile, AwkFilterBuffer, AwkFilterInput]
|
|
|
|
colors = {
|
2009-04-06 11:21:43 -04:00
|
|
|
'awk.global': ('yellow', 'default', 'bold'),
|
|
|
|
'awk.function': ('magenta', 'default', 'bold'),
|
|
|
|
'awk.field': ('yellow', 'default', 'bold'),
|
|
|
|
'awk.regex.start': ('cyan', 'default', 'bold'),
|
|
|
|
'awk.regex.null': ('cyan', 'default', 'bold'),
|
|
|
|
'awk.regex.data': ('cyan', 'default', 'bold'),
|
|
|
|
'awk.regex.end': ('cyan', 'default', 'bold'),
|
2008-10-01 16:58:43 -04:00
|
|
|
}
|
2009-02-15 12:06:35 -05:00
|
|
|
_bindings = {
|
|
|
|
'close-paren': (')',),
|
|
|
|
'close-brace': ('}',),
|
|
|
|
'close-bracket': (']',),
|
|
|
|
}
|
2008-10-01 16:58:43 -04:00
|
|
|
install = Awk.install
|