##// END OF EJS Templates
repair.py: rename chlog to cl
Alexis S. L. Carvalho -
r5901:16f4129c default
parent child Browse files
Show More
@@ -1,131 +1,131
1 1 # repair.py - functions for repository repair for mercurial
2 2 #
3 3 # Copyright 2005, 2006 Chris Mason <mason@suse.com>
4 4 # Copyright 2007 Matt Mackall
5 5 #
6 6 # This software may be used and distributed according to the terms
7 7 # of the GNU General Public License, incorporated herein by reference.
8 8
9 9 import changegroup, os
10 10 from node import *
11 11
12 12 def strip(ui, repo, node, backup="all"):
13 def limitheads(chlog, stop):
13 def limitheads(cl, stop):
14 14 """return the list of all nodes that have no children"""
15 15 p = {}
16 16 h = []
17 17 stoprev = 0
18 if stop in chlog.nodemap:
19 stoprev = chlog.rev(stop)
18 if stop in cl.nodemap:
19 stoprev = cl.rev(stop)
20 20
21 for r in xrange(chlog.count() - 1, -1, -1):
22 n = chlog.node(r)
21 for r in xrange(cl.count() - 1, -1, -1):
22 n = cl.node(r)
23 23 if n not in p:
24 24 h.append(n)
25 25 if n == stop:
26 26 break
27 27 if r < stoprev:
28 28 break
29 for pn in chlog.parents(n):
29 for pn in cl.parents(n):
30 30 p[pn] = 1
31 31 return h
32 32
33 33 def bundle(repo, bases, heads, node, suffix):
34 34 cg = repo.changegroupsubset(bases, heads, 'strip')
35 35 backupdir = repo.join("strip-backup")
36 36 if not os.path.isdir(backupdir):
37 37 os.mkdir(backupdir)
38 38 name = os.path.join(backupdir, "%s-%s" % (short(node), suffix))
39 39 ui.warn("saving bundle to %s\n" % name)
40 40 return changegroup.writebundle(cg, name, "HG10BZ")
41 41
42 42 def stripall(striprev):
43 43 mm = repo.changectx(node).manifest()
44 44 seen = {}
45 45
46 46 for x in xrange(striprev, repo.changelog.count()):
47 47 for f in repo.changectx(x).files():
48 48 if f in seen:
49 49 continue
50 50 seen[f] = 1
51 51 if f in mm:
52 52 filerev = mm[f]
53 53 else:
54 54 filerev = 0
55 55 seen[f] = filerev
56 56 # we go in two steps here so the strip loop happens in a
57 57 # sensible order. When stripping many files, this helps keep
58 58 # our disk access patterns under control.
59 59 seen_list = seen.keys()
60 60 seen_list.sort()
61 61 for f in seen_list:
62 62 ff = repo.file(f)
63 63 filerev = seen[f]
64 64 if filerev != 0:
65 65 if filerev in ff.nodemap:
66 66 filerev = ff.rev(filerev)
67 67 else:
68 68 filerev = 0
69 69 ff.strip(filerev, striprev)
70 70
71 chlog = repo.changelog
71 cl = repo.changelog
72 72 # TODO delete the undo files, and handle undo of merge sets
73 pp = chlog.parents(node)
74 striprev = chlog.rev(node)
73 pp = cl.parents(node)
74 striprev = cl.rev(node)
75 75
76 76 # save is a list of all the branches we are truncating away
77 77 # that we actually want to keep. changegroup will be used
78 78 # to preserve them and add them back after the truncate
79 79 saveheads = []
80 80 savebases = {}
81 81
82 heads = limitheads(chlog, node)
82 heads = limitheads(cl, node)
83 83 seen = {}
84 84
85 85 # search through all the heads, finding those where the revision
86 86 # we want to strip away is an ancestor. Also look for merges
87 87 # that might be turned into new heads by the strip.
88 88 while heads:
89 89 h = heads.pop()
90 90 n = h
91 91 while True:
92 92 seen[n] = 1
93 pp = chlog.parents(n)
93 pp = cl.parents(n)
94 94 if pp[1] != nullid:
95 95 for p in pp:
96 if chlog.rev(p) > striprev and p not in seen:
96 if cl.rev(p) > striprev and p not in seen:
97 97 heads.append(p)
98 98 if pp[0] == nullid:
99 99 break
100 if chlog.rev(pp[0]) < striprev:
100 if cl.rev(pp[0]) < striprev:
101 101 break
102 102 n = pp[0]
103 103 if n == node:
104 104 break
105 r = chlog.reachable(h, node)
105 r = cl.reachable(h, node)
106 106 if node not in r:
107 107 saveheads.append(h)
108 108 for x in r:
109 if chlog.rev(x) > striprev:
109 if cl.rev(x) > striprev:
110 110 savebases[x] = 1
111 111
112 112 # create a changegroup for all the branches we need to keep
113 113 if backup == "all":
114 bundle(repo, [node], chlog.heads(), node, 'backup')
114 bundle(repo, [node], cl.heads(), node, 'backup')
115 115 if saveheads:
116 116 chgrpfile = bundle(repo, savebases.keys(), saveheads, node, 'temp')
117 117
118 118 stripall(striprev)
119 119
120 change = chlog.read(node)
121 chlog.strip(striprev, striprev)
120 change = cl.read(node)
121 cl.strip(striprev, striprev)
122 122 repo.manifest.strip(repo.manifest.rev(change[0]), striprev)
123 123 if saveheads:
124 124 ui.status("adding branch\n")
125 125 f = open(chgrpfile, "rb")
126 126 gen = changegroup.readbundle(f, chgrpfile)
127 127 repo.addchangegroup(gen, 'strip', 'bundle:' + chgrpfile)
128 128 f.close()
129 129 if backup != "strip":
130 130 os.unlink(chgrpfile)
131 131
General Comments 0
You need to be logged in to leave comments. Login now