##// END OF EJS Templates
repair.py: nodes are nodes, revs are revs...
Alexis S. L. Carvalho -
r5900:1206e3df default
parent child Browse files
Show More
@@ -9,7 +9,7 b''
9 import changegroup, os
9 import changegroup, os
10 from node import *
10 from node import *
11
11
12 def strip(ui, repo, rev, backup="all"):
12 def strip(ui, repo, node, backup="all"):
13 def limitheads(chlog, stop):
13 def limitheads(chlog, stop):
14 """return the list of all nodes that have no children"""
14 """return the list of all nodes that have no children"""
15 p = {}
15 p = {}
@@ -30,20 +30,20 b' def strip(ui, repo, rev, backup="all"):'
30 p[pn] = 1
30 p[pn] = 1
31 return h
31 return h
32
32
33 def bundle(repo, bases, heads, rev, suffix):
33 def bundle(repo, bases, heads, node, suffix):
34 cg = repo.changegroupsubset(bases, heads, 'strip')
34 cg = repo.changegroupsubset(bases, heads, 'strip')
35 backupdir = repo.join("strip-backup")
35 backupdir = repo.join("strip-backup")
36 if not os.path.isdir(backupdir):
36 if not os.path.isdir(backupdir):
37 os.mkdir(backupdir)
37 os.mkdir(backupdir)
38 name = os.path.join(backupdir, "%s-%s" % (short(rev), suffix))
38 name = os.path.join(backupdir, "%s-%s" % (short(node), suffix))
39 ui.warn("saving bundle to %s\n" % name)
39 ui.warn("saving bundle to %s\n" % name)
40 return changegroup.writebundle(cg, name, "HG10BZ")
40 return changegroup.writebundle(cg, name, "HG10BZ")
41
41
42 def stripall(revnum):
42 def stripall(striprev):
43 mm = repo.changectx(rev).manifest()
43 mm = repo.changectx(node).manifest()
44 seen = {}
44 seen = {}
45
45
46 for x in xrange(revnum, repo.changelog.count()):
46 for x in xrange(striprev, repo.changelog.count()):
47 for f in repo.changectx(x).files():
47 for f in repo.changectx(x).files():
48 if f in seen:
48 if f in seen:
49 continue
49 continue
@@ -66,12 +66,12 b' def strip(ui, repo, rev, backup="all"):'
66 filerev = ff.rev(filerev)
66 filerev = ff.rev(filerev)
67 else:
67 else:
68 filerev = 0
68 filerev = 0
69 ff.strip(filerev, revnum)
69 ff.strip(filerev, striprev)
70
70
71 chlog = repo.changelog
71 chlog = repo.changelog
72 # TODO delete the undo files, and handle undo of merge sets
72 # TODO delete the undo files, and handle undo of merge sets
73 pp = chlog.parents(rev)
73 pp = chlog.parents(node)
74 revnum = chlog.rev(rev)
74 striprev = chlog.rev(node)
75
75
76 # save is a list of all the branches we are truncating away
76 # save is a list of all the branches we are truncating away
77 # that we actually want to keep. changegroup will be used
77 # that we actually want to keep. changegroup will be used
@@ -79,7 +79,7 b' def strip(ui, repo, rev, backup="all"):'
79 saveheads = []
79 saveheads = []
80 savebases = {}
80 savebases = {}
81
81
82 heads = limitheads(chlog, rev)
82 heads = limitheads(chlog, node)
83 seen = {}
83 seen = {}
84
84
85 # search through all the heads, finding those where the revision
85 # search through all the heads, finding those where the revision
@@ -93,33 +93,33 b' def strip(ui, repo, rev, backup="all"):'
93 pp = chlog.parents(n)
93 pp = chlog.parents(n)
94 if pp[1] != nullid:
94 if pp[1] != nullid:
95 for p in pp:
95 for p in pp:
96 if chlog.rev(p) > revnum and p not in seen:
96 if chlog.rev(p) > striprev and p not in seen:
97 heads.append(p)
97 heads.append(p)
98 if pp[0] == nullid:
98 if pp[0] == nullid:
99 break
99 break
100 if chlog.rev(pp[0]) < revnum:
100 if chlog.rev(pp[0]) < striprev:
101 break
101 break
102 n = pp[0]
102 n = pp[0]
103 if n == rev:
103 if n == node:
104 break
104 break
105 r = chlog.reachable(h, rev)
105 r = chlog.reachable(h, node)
106 if rev not in r:
106 if node not in r:
107 saveheads.append(h)
107 saveheads.append(h)
108 for x in r:
108 for x in r:
109 if chlog.rev(x) > revnum:
109 if chlog.rev(x) > striprev:
110 savebases[x] = 1
110 savebases[x] = 1
111
111
112 # create a changegroup for all the branches we need to keep
112 # create a changegroup for all the branches we need to keep
113 if backup == "all":
113 if backup == "all":
114 bundle(repo, [rev], chlog.heads(), rev, 'backup')
114 bundle(repo, [node], chlog.heads(), node, 'backup')
115 if saveheads:
115 if saveheads:
116 chgrpfile = bundle(repo, savebases.keys(), saveheads, rev, 'temp')
116 chgrpfile = bundle(repo, savebases.keys(), saveheads, node, 'temp')
117
117
118 stripall(revnum)
118 stripall(striprev)
119
119
120 change = chlog.read(rev)
120 change = chlog.read(node)
121 chlog.strip(revnum, revnum)
121 chlog.strip(striprev, striprev)
122 repo.manifest.strip(repo.manifest.rev(change[0]), revnum)
122 repo.manifest.strip(repo.manifest.rev(change[0]), striprev)
123 if saveheads:
123 if saveheads:
124 ui.status("adding branch\n")
124 ui.status("adding branch\n")
125 f = open(chgrpfile, "rb")
125 f = open(chgrpfile, "rb")
General Comments 0
You need to be logged in to leave comments. Login now