comparison mercurial/patch.py @ 14370:17cea10c343e

patch: add a workingbackend dirstate layer on top of fsbackend

_updatedir() is no longer used by internalpatch()

The change in test-mq-missingfiles.t comes from workingbackend not considering
the missing 'b' file as changed, thus not calling addremove() on it.

14369:f8932d540088

    def exists(self, fname):
        return os.path.lexists(self._join(fname))

    def setmode(self, fname, islink, isexec):
        util.setflags(self._join(fname), islink, isexec)

# @@ -start,len +start,len @@ or @@ -start +start @@ if len is 1
unidesc = re.compile('@@ -(\d+)(,(\d+))? \+(\d+)(,(\d+))? @@')
contextdesc = re.compile('(---|\*\*\*) (\d+)(,(\d+))? (---|\*\*\*)')
eolmodes = ['strict', 'crlf', 'lf', 'auto']

    found and 1 if there was any fuzz.

    If 'eolmode' is 'strict', the patch content and patched file are
    read in binary mode. Otherwise, line endings are ignored when
    patching then normalized according to 'eolmode'.

    Callers probably want to call '_updatedir' after this to
    apply certain categories of changes not done by this function.
    """
    return _applydiff(ui, fp, patchfile, backend, changed, strip=strip,
                      eolmode=eolmode)

def _applydiff(ui, fp, patcher, backend, changed, strip=1, eolmode='strict'):

    if code:
        raise PatchError(_("patch command failed: %s") %
                         util.explainexit(code)[0])
    return fuzz

def internalpatch(ui, repo, patchobj, strip, cwd, files=None, eolmode='strict',
                  similarity=0):
    """use builtin patch to apply <patchobj> to the working directory.
    returns whether patch was applied with fuzz factor."""

    if files is None:

        eolmode = ui.config('patch', 'eol', 'strict')
    if eolmode.lower() not in eolmodes:
        raise util.Abort(_('unsupported line endings type: %s') % eolmode)
    eolmode = eolmode.lower()

    backend = fsbackend(ui, cwd)
    try:
        fp = open(patchobj, 'rb')
    except TypeError:
        fp = patchobj
    try:
        ret = applydiff(ui, fp, files, backend, strip=strip, eolmode=eolmode)
    finally:
        if fp != patchobj:
            fp.close()
        touched = _updatedir(ui, repo, files, similarity)
        files.update(dict.fromkeys(touched))
    if ret < 0:
        raise PatchError(_('patch failed to apply'))
    return ret > 0

def patch(ui, repo, patchname, strip=1, cwd=None, files=None, eolmode='strict',

                return _externalpatch(patcher, patchname, ui, strip, cwd,
                                      files)
            finally:
                touched = _updatedir(ui, repo, files, similarity)
                files.update(dict.fromkeys(touched))
        return internalpatch(ui, repo, patchname, strip, cwd, files, eolmode,
                             similarity)
    except PatchError, err:
        raise util.Abort(str(err))

def changedfiles(ui, repo, patchpath, strip=1):

14370:17cea10c343e

    def exists(self, fname):
        return os.path.lexists(self._join(fname))

    def setmode(self, fname, islink, isexec):
        util.setflags(self._join(fname), islink, isexec)

class workingbackend(fsbackend):
    def __init__(self, ui, repo, similarity):
        super(workingbackend, self).__init__(ui, repo.root)
        self.repo = repo
        self.similarity = similarity
        self.removed = set()
        self.changed = set()
        self.copied = []

    def writelines(self, fname, lines, mode):
        super(workingbackend, self).writelines(fname, lines, mode)
        self.changed.add(fname)

    def unlink(self, fname):
        super(workingbackend, self).unlink(fname)
        self.removed.add(fname)
        self.changed.add(fname)

    def copy(self, src, dst):
        super(workingbackend, self).copy(src, dst)
        self.copied.append((src, dst))
        self.changed.add(dst)

    def setmode(self, fname, islink, isexec):
        super(workingbackend, self).setmode(fname, islink, isexec)
        self.changed.add(fname)

    def close(self):
        wctx = self.repo[None]
        addremoved = set(self.changed)
        for src, dst in self.copied:
            scmutil.dirstatecopy(self.ui, self.repo, wctx, src, dst)
            addremoved.discard(src)
        if (not self.similarity) and self.removed:
            wctx.remove(sorted(self.removed))
        if addremoved:
            cwd = self.repo.getcwd()
            if cwd:
                addremoved = [util.pathto(self.repo.root, cwd, f)
                              for f in addremoved]
            scmutil.addremove(self.repo, addremoved, similarity=self.similarity)
        return sorted(self.changed)

# @@ -start,len +start,len @@ or @@ -start +start @@ if len is 1
unidesc = re.compile('@@ -(\d+)(,(\d+))? \+(\d+)(,(\d+))? @@')
contextdesc = re.compile('(---|\*\*\*) (\d+)(,(\d+))? (---|\*\*\*)')
eolmodes = ['strict', 'crlf', 'lf', 'auto']

    found and 1 if there was any fuzz.

    If 'eolmode' is 'strict', the patch content and patched file are
    read in binary mode. Otherwise, line endings are ignored when
    patching then normalized according to 'eolmode'.
    """
    return _applydiff(ui, fp, patchfile, backend, changed, strip=strip,
                      eolmode=eolmode)

def _applydiff(ui, fp, patcher, backend, changed, strip=1, eolmode='strict'):

    if code:
        raise PatchError(_("patch command failed: %s") %
                         util.explainexit(code)[0])
    return fuzz

def internalpatch(ui, repo, patchobj, strip, files=None, eolmode='strict',
                  similarity=0):
    """use builtin patch to apply <patchobj> to the working directory.
    returns whether patch was applied with fuzz factor."""

    if files is None:

        eolmode = ui.config('patch', 'eol', 'strict')
    if eolmode.lower() not in eolmodes:
        raise util.Abort(_('unsupported line endings type: %s') % eolmode)
    eolmode = eolmode.lower()

    backend = workingbackend(ui, repo, similarity)
    try:
        fp = open(patchobj, 'rb')
    except TypeError:
        fp = patchobj
    try:
        ret = applydiff(ui, fp, files, backend, strip=strip, eolmode=eolmode)
    finally:
        if fp != patchobj:
            fp.close()
        files.update(dict.fromkeys(backend.close()))
    if ret < 0:
        raise PatchError(_('patch failed to apply'))
    return ret > 0

def patch(ui, repo, patchname, strip=1, cwd=None, files=None, eolmode='strict',

                return _externalpatch(patcher, patchname, ui, strip, cwd,
                                      files)
            finally:
                touched = _updatedir(ui, repo, files, similarity)
                files.update(dict.fromkeys(touched))
        return internalpatch(ui, repo, patchname, strip, files, eolmode,
                             similarity)
    except PatchError, err:
        raise util.Abort(str(err))

def changedfiles(ui, repo, patchpath, strip=1):