pmacs3/mode/replace.py

import re, sets, string

import color, method, minibuffer, mode, searchutil
from point import Point

subgroup_re = re.compile(r'((?:\\\\)*)\\(0|[1-9][0-9]*)')

class ReplaceOne(method.Method):
    def execute(self, w, **vargs):
        m = w.buffer.method
        _replace(m)
        _find_next(m, False)
        _finish(m, w)

class ReplaceDone(method.Method):
    def execute(self, w, **vargs):
        m = w.buffer.method
        _replace(m)
        _end(w)
        w.set_error("Replace done")

class SkipReplace(method.Method):
    def execute(self, w, **vargs):
        m = w.buffer.method
        _find_next(m, True)
        _finish(m, w)

class ReplaceAll(method.Method):
    def execute(self, w, **vargs):
        m = w.buffer.method
        while m.p1 is not None:
            _replace(m)
            _find_next(m, False)
        _end(w)
        w.set_error("Replace ended")

class CancelReplace(method.Method):
    def execute(self, w, **vargs):
        _end(w)
        w.set_error("Replace cancelled")

def _find_next(m, move=False):
    s = m.before
    w = m.old_window
    c = w.logical_cursor()
    try:
        if m.is_literal:
            r = re.compile(searchutil.escape_literal(s))
        else:
            r = re.compile(s)
    except:
        (m.p1, m.p2) = (None, None)
        return False
        
    newc = searchutil.find_next(r, w, move, start=c.add(0, 0))
    if newc:
        (m.p1, m.p2, m.match) = newc
        return True
    else:
        (m.p1, m.p2, m.match) = (None, None, None)
        return False

def _get_before(m):
    if m.match is None:
        return m.before
    else:
        return m.match.group(0)
    
def _get_after(m):
    if m.after is None:
        return None
    elif m.match is None:
        return m.after

    def _repl(match):
        (pre, num) = (match.group(1), int(match.group(2)))
        if num == 0 or m.match.lastindex and num <= m.match.lastindex:
            return pre + m.match.group(num)
        else:
            return match.group(0)

    return subgroup_re.sub(_repl, m.after)

def _set_prompt(m):
    w = m.old_window
    if m.p1 is None:
        w.application.mini_prompt = '%r was not found' % m.before
        return
    (x, y) = m.p1.xy()
    count = 0
    while y < len(w.buffer.lines):
        count += w.buffer.lines[y][x:].count(m.before)
        y += 1
        x = 0

    after  = _get_after(m)
    before = _get_before(m)

    if count > 1:
        p = 'Replace %r with %r [ynadq] (%d occurances)?' % (before, after, count)
    else:
        p = 'Replace %r with %r [ynadq] (1 occurance)?' % (before, after)
    w.application.mini_prompt = p

def _replace(m):
    m.old_window.buffer.delete(m.p1, m.p2)
    if m.after:
        after = _get_after(m)
        m.old_window.buffer.insert_string(m.p1, after)

def _finish(m, w):
    if m.p1 is None:
        _end(w)
        w.set_error("Replace ended")
    else:
        _set_prompt(m)

def _end(w):
    w.application.close_mini_buffer()
    w.application.clear_highlighted_ranges()
    w.buffer.method.old_cursor = None
    w.buffer.method.old_window = None
    assert not w.application.mini_active

class Replace(mode.Fundamental):
    modename = 'Replace'
    actions = [ReplaceAll, ReplaceDone, ReplaceOne, SkipReplace, CancelReplace]
    def __init__(self, w):
        mode.Fundamental.__init__(self, w)

        self.actions = {}
        self.bindings = {}
        self.add_bindings('replace-all', ('a', '!',))
        self.add_bindings('replace-done', ('d',))
        self.add_bindings('replace-one', ('y', 'SPACE',))
        self.add_bindings('skip-replace', ('n', 'DELETE',))
        self.add_bindings('cancel-replace', ('q', 'RETURN', 'C-]', 'C-n', 'C-p', 'C-a', 'C-e', 'C-f', 'C-b'))

        m = w.buffer.method
        found = _find_next(m, False)
        if not found:
            w.set_error('%r was not found' % m.before)
            raise minibuffer.MiniBufferError
        _set_prompt(m)

install = Replace.install