import os, commands, re, sets, tempfile from subprocess import Popen, PIPE, STDOUT import buffer, default, dirutil, regex, util, window from point import Point from method import DATATYPES, Method, Argument class OpenFile(Method): '''Open file in a new buffer, or go to file's open buffer''' args = [Argument('filename', datatype="path", prompt="Open File: ", default=default.path_dirname, load_default=True)] def _execute(self, w, **vargs): b = w.application.open_path(vargs['filename']) SwitchBuffer().execute(w, buffername=b.name()) class OpenAesFile(Method): '''Open AES encrypted file in a new buffer, or go to file's open buffer''' args = [Argument('filename', datatype="path", prompt="Open AES File: "), Argument('password', prompt="Use AES Password: ")] def _execute(self, w, **vargs): b = w.application.open_path(vargs['filename'], 'aes', vargs['password']) SwitchBuffer().execute(w, buffername=b.name()) return class ViewBufferParent(Method): def _execute(self, w, **vargs): b = w.buffer if not hasattr(b, 'path'): w.set_error('Buffer has no path') elif b.path == '/': w.set_error("Root directory has no parent") else: path = os.path.dirname(b.path) w.application.methods['open-file'].execute(w, filename=path) class SwitchBuffer(Method): '''Switch to a different''' args = [Argument('buffername', datatype="buffer", prompt="Switch To Buffer: ", default=default.last_buffer)] def _pre_execute(self, w, **vargs): a = w.application if len(a.bufferlist.buffers) < 1: raise Exception, "No other buffers" def _execute(self, w, **vargs): name = vargs['buffername'] buf = None if w.application.has_buffer_name(name): b = w.application.bufferlist.get_buffer_by_name(name) w.application.switch_buffer(b) else: w.set_error("buffer %r was not found" % name) class KillBuffer(Method): '''Close the current buffer''' force=False args = [Argument('buffername', datatype="buffer", prompt="Kill Buffer: ", default=default.current_buffer)] def _execute(self, w, **vargs): name = vargs['buffername'] a = w.application assert name in a.bufferlist.buffer_names, "Buffer %r does not exist" % name assert name != '*Scratch*', "Can't kill scratch buffer" self._to_kill = a.bufferlist.buffer_names[name] self._old_window = w if self.force or not self._to_kill.changed(): self._doit() else: self._prompt = "Buffer has unsaved changes; kill anyway? " a.open_mini_buffer(self._prompt, self._callback) def _doit(self): a = self._old_window.application b = self._to_kill if a.bufferlist.is_buffer_visible(b): a.bufferlist.set_slot(a.active_slot, a.bufferlist.hidden_buffers[0]) a.bufferlist.remove_buffer(b) b.close() def _callback(self, v): a = self._old_window.application if v == 'yes': self._doit() a.close_mini_buffer() elif v == 'no': a.close_mini_buffer() else: a.close_mini_buffer() a.set_error('Please type "yes" or "no"') class ForceKillBuffer(KillBuffer): force=True args = [Argument('buffername', datatype="buffer", prompt="Force Kill Buffer: ", default=default.current_buffer)] class ListBuffers(Method): '''List all open buffers in a new buffer''' def _execute(self, w, **vargs): bl = w.application.bufferlist bnames = [b.name() for b in bl.buffers] bnames.sort() data = '\n'.join(bnames) w.application.data_buffer("*Buffers*", data, switch_to=True) class SaveBufferAs(Method): '''Save the contents of a buffer to the specified path''' args = [Argument('path', datatype="path", prompt="Write file: ", default=default.current_working_dir, load_default=True)] def _execute(self, w, **vargs): curr_buffer = w.buffer curr_buffer_name = curr_buffer.name() data = curr_buffer.make_string() path = os.path.realpath(os.path.expanduser(vargs['path'])) w.set_error("got %r (%d)" % (path, len(data))) if w.application.has_buffer_name(path): w.set_error("buffer for %r is already open" % path) return w.application.file_buffer(path, data, switch_to=True) if curr_buffer_name != '*Scratch*': w.application.methods['kill-buffer'].execute(w, buffername=curr_buffer_name) else: curr_buffer.set_data('') w.set_error('Wrote %r' % path) class SaveBuffer(Method): '''Save the contents of a buffer''' def _execute(self, w, **vargs): if w.buffer.changed(): w.buffer.save() w.set_error("Wrote %s" % (w.buffer.path)) else: w.set_error("(No changes need to be saved)")