import commands, dirutil, grp, method, mode, os.path, pwd, re import util import buffer, buffer.fs from window import Window from lex import Grammar, PatternRule, RegionRule, PatternMatchRule from point import Point from method import Method, Argument class PermGrammar(Grammar): rules = [ PatternRule('perm_sticky', '[tT]'), PatternRule('perm_setid', '[sS]'), PatternRule('perm_read', 'r'), PatternRule('perm_write', 'w'), PatternRule('perm_exec', 'x'), ] ds = r'([^ ]+)( +)([^ ]+)( +)([^ ]+)( +)([A-Za-z]{3} [ 0-9]{2} [0-9]{2}:[0-9]{2})( +)([^\n]+)' class PathGrammar(Grammar): rules = [ RegionRule('dir_perm', '(?<=^.)', PermGrammar, ' '), PatternMatchRule('x', ds, 'dir_owner', 'spaces', 'dir_group', 'spaces', 'dir_size', 'spaces', 'dir_mtime', 'spaces', 'dir_name', 'spaces'), ] class DirGrammar(Grammar): rules = [ RegionRule('dir_file', '^-', PathGrammar, r'\n'), RegionRule('dir_blk', '^b', PathGrammar, r'\n'), RegionRule('dir_chr', '^c', PathGrammar, r'\n'), RegionRule('dir_dir', '^d', PathGrammar, r'\n'), RegionRule('dir_lnk', '^l', PathGrammar, r'\n'), RegionRule('dir_fifo', '^p', PathGrammar, r'\n'), RegionRule('dir_sock', '^s', PathGrammar, r'\n'), RegionRule('dir_unk', '^\?', PathGrammar, r'\n'), ] class FsSettingBase(Method): msg = 'Settings have changed' def _doit(self, w, **vargs): pass def _execute(self, w, **vargs): self._doit(w, **vargs) w.buffer.reload() w.goto_beginning() w.set_error(self.msg) class HideDotFiles(FsSettingBase): msg = "Dotfiles are hidden" def _doit(self, w, **vargs): w.buffer.settings['hide-dot'] = True class ShowDotFiles(FsSettingBase): msg = "Dotfiles are visible" def _doit(self, w, **vargs): w.buffer.settings['hide-dot'] = False class SortType(FsSettingBase): msg = "Sorting files by type, name" def _doit(self, w, **vargs): w.buffer.settings['type-sort'] = True class SortName(FsSettingBase): msg = "Sorting files by name, type" def _doit(self, w, **vargs): w.buffer.settings['type-sort'] = False class DirRefresh(Method): def _execute(self, w, **vargs): t = dirutil.resolve_token(w) s = t.string w.buffer.reload() dirutil.find_name(w, s) class DirOpen(Method): def _execute(self, w, **vargs): path = dirutil.resolve_path(w) w.set_error("opening %r" % path) w.application.methods['open-file'].execute(w, filename=path) class DirGrep(Method): args = [Argument('pattern', datatype="str", prompt="Pattern: ")] def _execute(self, w, **vargs): cmd = 'grep -rEl "%s" %r' % (vargs['pattern'], w.buffer.path) (status, output) = commands.getstatusoutput(cmd) paths = [] suffixes = w.application.config['ignore_suffix'] for path in output.split('\n'): if not path or util.should_ignore_path(path, suffixes): continue paths.append(path) bufname = '*%s*' % self.name.title() b = buffer.fs.PathListBuffer(bufname, paths) b.modename = 'dir' b.open() Window(b, w.application, height=0, width=0) w.application.add_buffer(b) w.application.switch_buffer(b) w.set_error("grep exited with %d" % status) class DirCmd(Method): def _make_cmd(self, w, path, **vargs): return '' def _run(self, w, **vargs): basename = dirutil.resolve_name(w) path = os.path.join(w.buffer.path, basename) cmd = self._make_cmd(w, path, **vargs) (status, output) = commands.getstatusoutput(cmd) if status != 0: w.set_error("%s failed (exit %d)" % (self.name, status)) dirutil.find_name(w, basename) class DirChmod(DirCmd): args = [Argument('mode', type=type(''), prompt="New Mode: ")] octal_re = re.compile('^[0-7]{1,4}$') symbolic_re = re.compile('(?:[ugoa]*(?:[-+=](?:[rwxXst]*|[ugo]))+ *,?)+') def _make_cmd(self, w, path, **vargs): return 'chmod %r %r' % (vargs['mode'], path) def _execute(self, w, **vargs): if self.octal_re.match(vargs['mode']): pass elif self.symbolic_re.match(vargs['mode']): pass else: w.set_error("Not a valid mode: %r" % vargs['mode']) self._run(w, **vargs) class DirChown(DirCmd): args = [Argument('owner', type=type(''), prompt="New Owner: ")] def _make_cmd(self, w, path, **vargs): return 'chown %r %r' % (vargs['owner'], path) def _execute(self, w, **vargs): fields = vargs['owner'].split(':') if len(fields) == 1: (owner, group) = (fields[0], None) elif len(fields) == 2: (owner, group) = fields else: w.set_error("Malformed 'owner' argument: %r" % vargs['owner']) return if not dirutil.valid_owner(owner): w.set_error('User %r does not exist' % owner) return if group is not None and not dirutil.valid_group(group): w.set_error('Group %r does not exist' % group) return self._run(w, **vargs) class DirChgrp(DirCmd): args = [Argument('group', type=type(''), prompt="New Group: ")] def _make_cmd(self, w, path, **vargs): return 'chgrp %r %r' % (vargs['group'], path) def _execute(self, w, **vargs): if not dirutil.valid_group(vargs['group']): w.set_error('Group %r does not exist' % group) return self._run(w, **vargs) class DirTouch(Method): args = [Argument('filename', datatype="path", prompt="Touch File: ")] def _execute(self, w, **vargs): basename = vargs['filename'] path = os.path.join(w.buffer.path, basename) retval = os.system('touch %r' % path) w.application.methods['dir-refresh'].execute(w, filename=path) dirutil.find_name(w, basename) if retval != 0: w.set_error("touch %r failed (exit %d)" % (path, retval)) class DirRemove(Method): def _execute(self, w, **vargs): self._old_window = w self._old_path = dirutil.resolve_path(w) basename = os.path.basename(self._old_path) self._prompt = "Do you want to delete %r? " % basename w.application.open_mini_buffer(self._prompt, self._callback) def _callback(self, v): a = self._old_window.application if v == 'yes': self._doit() a.close_mini_buffer() return if v == 'no': a.close_mini_buffer() return a.open_mini_buffer(self._prompt, self._callback) a.set_error('Please type "yes" or "no"') def _doit(self): w = self._old_window path = self._old_path try: w.application.methods['previous-line'].execute(w) os.remove(path) w.set_error("deleted %r " % path) w.application.methods['dir-refresh'].execute(w, filename=path) except: w.set_error("failed to delete %r" % path) class Dir(mode.Fundamental): name = 'Dir' grammar = DirGrammar() colors = { 'dir_blk.start': ('cyan', 'default', 'bold'), 'dir_blk.dir_name': ('cyan', 'default', 'bold'), 'dir_chr.start': ('yellow', 'default', 'bold'), 'dir_chr.dir_name': ('yellow', 'default', 'bold'), 'dir_dir.start': ('blue', 'default', 'bold'), 'dir_dir.dir_name': ('blue', 'default', 'bold'), 'dir_lnk.start': ('green', 'default', 'bold'), 'dir_lnk.dir_name': ('green', 'default', 'bold'), 'dir_fifo.start': ('red', 'default', 'bold'), 'dir_fifo.dir_name': ('red', 'default', 'bold'), 'dir_sock.start': ('red', 'default', 'bold'), 'dir_sock.dir_name': ('red', 'default', 'bold'), 'dir_unk.start': ('magenta', 'default', 'bold'), 'dir_unk.dir_name': ('magenta', 'default', 'bold'), 'dir_perm.perm_setid': ('yellow', 'default', 'bold'), 'dir_perm.perm_sticky': ('yellow', 'default', 'bold'), 'dir_perm.perm_read': ('magenta', 'default', 'bold'), 'dir_perm.perm_write': ('magenta', 'default', 'bold'), 'dir_perm.perm_exec': ('magenta', 'default', 'bold'), 'dir_owner': ('cyan', 'default', 'bold'), 'dir_group': ('cyan', 'default', 'bold'), 'dir_size': ('yellow', 'default', 'bold'), 'dir_mtime': ('green', 'default', 'bold'), } actions = [DirRefresh, DirOpen, DirGrep, DirChmod, DirChown, DirChgrp, DirTouch, DirRemove, HideDotFiles, ShowDotFiles, SortName, SortType] def __init__(self, w): mode.Fundamental.__init__(self, w) self.add_bindings('dir-refresh', ('C-c r',)) self.add_bindings('dir-open', ('RETURN',)) self.add_bindings('dir-grep', ('C-c G',)) self.add_bindings('dir-chmod', ('C-c m',)) self.add_bindings('dir-chown', ('C-c o',)) self.add_bindings('dir-chgrp', ('C-c g',)) self.add_bindings('dir-touch', ('C-c t',)) self.add_bindings('dir-remove', ('DELETE', 'BACKSPACE', 'C-d')) install = Dir.install