repair.py
145 lines
| 4.8 KiB
| text/x-python
|
PythonLexer
/ mercurial / repair.py
|
r4702 | # repair.py - functions for repository repair for mercurial | ||
# | ||||
# Copyright 2005, 2006 Chris Mason <mason@suse.com> | ||||
# Copyright 2007 Matt Mackall | ||||
# | ||||
|
r8225 | # This software may be used and distributed according to the terms of the | ||
# GNU General Public License version 2, incorporated herein by reference. | ||||
|
r4702 | |||
|
r8312 | import changegroup | ||
|
r6211 | from node import nullrev, short | ||
|
r6953 | from i18n import _ | ||
|
r8312 | import os | ||
|
r4702 | |||
|
r5909 | def _bundle(repo, bases, heads, node, suffix, extranodes=None): | ||
|
r5905 | """create a bundle with the specified revisions as a backup""" | ||
|
r5909 | cg = repo.changegroupsubset(bases, heads, 'strip', extranodes) | ||
|
r5905 | backupdir = repo.join("strip-backup") | ||
if not os.path.isdir(backupdir): | ||||
os.mkdir(backupdir) | ||||
name = os.path.join(backupdir, "%s-%s" % (short(node), suffix)) | ||||
|
r6953 | repo.ui.warn(_("saving bundle to %s\n") % name) | ||
|
r5905 | return changegroup.writebundle(cg, name, "HG10BZ") | ||
|
r4702 | |||
|
r5910 | def _collectfiles(repo, striprev): | ||
"""find out the filelogs affected by the strip""" | ||||
|
r8462 | files = set() | ||
|
r4702 | |||
|
r6750 | for x in xrange(striprev, len(repo)): | ||
|
r8479 | files.update(repo[x].files()) | ||
|
r5902 | |||
|
r8462 | return sorted(files) | ||
|
r5902 | |||
|
r5909 | def _collectextranodes(repo, files, link): | ||
"""return the nodes that have to be saved before the strip""" | ||||
def collectone(revlog): | ||||
extra = [] | ||||
|
r6750 | startrev = count = len(revlog) | ||
|
r5909 | # find the truncation point of the revlog | ||
for i in xrange(0, count): | ||||
|
r7361 | lrev = revlog.linkrev(i) | ||
|
r5909 | if lrev >= link: | ||
startrev = i + 1 | ||||
break | ||||
# see if any revision after that point has a linkrev less than link | ||||
# (we have to manually save these guys) | ||||
for i in xrange(startrev, count): | ||||
node = revlog.node(i) | ||||
|
r7361 | lrev = revlog.linkrev(i) | ||
|
r5909 | if lrev < link: | ||
extra.append((node, cl.node(lrev))) | ||||
return extra | ||||
extranodes = {} | ||||
cl = repo.changelog | ||||
extra = collectone(repo.manifest) | ||||
if extra: | ||||
extranodes[1] = extra | ||||
for fname in files: | ||||
f = repo.file(fname) | ||||
extra = collectone(f) | ||||
if extra: | ||||
extranodes[fname] = extra | ||||
return extranodes | ||||
|
r5905 | def strip(ui, repo, node, backup="all"): | ||
|
r5901 | cl = repo.changelog | ||
|
r4702 | # TODO delete the undo files, and handle undo of merge sets | ||
|
r5901 | striprev = cl.rev(node) | ||
|
r4702 | |||
|
r6147 | # Some revisions with rev > striprev may not be descendants of striprev. | ||
# We have to find these revisions and put them in a bundle, so that | ||||
# we can restore them after the truncations. | ||||
# To create the bundle we use repo.changegroupsubset which requires | ||||
# the list of heads and bases of the set of interesting revisions. | ||||
# (head = revision in the set that has no descendant in the set; | ||||
# base = revision in the set that has no ancestor in the set) | ||||
|
r8462 | tostrip = set((striprev,)) | ||
saveheads = set() | ||||
|
r6147 | savebases = [] | ||
|
r6750 | for r in xrange(striprev + 1, len(cl)): | ||
|
r6147 | parents = cl.parentrevs(r) | ||
if parents[0] in tostrip or parents[1] in tostrip: | ||||
# r is a descendant of striprev | ||||
|
r8462 | tostrip.add(r) | ||
|
r6147 | # if this is a merge and one of the parents does not descend | ||
# from striprev, mark that parent as a savehead. | ||||
if parents[1] != nullrev: | ||||
for p in parents: | ||||
if p not in tostrip and p > striprev: | ||||
|
r8462 | saveheads.add(p) | ||
|
r6147 | else: | ||
# if no parents of this revision will be stripped, mark it as | ||||
# a savebase | ||||
if parents[0] < striprev and parents[1] < striprev: | ||||
savebases.append(cl.node(r)) | ||||
|
r4702 | |||
|
r8479 | saveheads.difference_update(parents) | ||
|
r8462 | saveheads.add(r) | ||
|
r4702 | |||
|
r6147 | saveheads = [cl.node(r) for r in saveheads] | ||
|
r5910 | files = _collectfiles(repo, striprev) | ||
|
r5909 | |||
|
r5910 | extranodes = _collectextranodes(repo, files, striprev) | ||
|
r5909 | |||
|
r4702 | # create a changegroup for all the branches we need to keep | ||
if backup == "all": | ||||
|
r5905 | _bundle(repo, [node], cl.heads(), node, 'backup') | ||
|
r5909 | if saveheads or extranodes: | ||
|
r6147 | chgrpfile = _bundle(repo, savebases, saveheads, node, 'temp', | ||
|
r5909 | extranodes) | ||
|
r4702 | |||
|
r8073 | fs = [repo.file(name) for name in files] | ||
mfst = repo.manifest | ||||
tr = repo.transaction() | ||||
offset = len(tr.entries) | ||||
|
r8363 | tr.startgroup() | ||
|
r8073 | cl.strip(striprev, tr) | ||
mfst.strip(striprev, tr) | ||||
for f in fs: | ||||
f.strip(striprev, tr) | ||||
|
r8363 | tr.endgroup() | ||
|
r8073 | |||
try: | ||||
for i in xrange(offset, len(tr.entries)): | ||||
file, troffset, ignore = tr.entries[i] | ||||
repo.sopener(file, 'a').truncate(troffset) | ||||
tr.close() | ||||
except: | ||||
tr.abort() | ||||
raise | ||||
|
r4702 | |||
|
r5909 | if saveheads or extranodes: | ||
|
r6953 | ui.status(_("adding branch\n")) | ||
|
r5898 | f = open(chgrpfile, "rb") | ||
gen = changegroup.readbundle(f, chgrpfile) | ||||
|
r5909 | repo.addchangegroup(gen, 'strip', 'bundle:' + chgrpfile, True) | ||
|
r5898 | f.close() | ||
|
r4702 | if backup != "strip": | ||
os.unlink(chgrpfile) | ||||