pmacs3/method/buffers.py

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)")