Mercurial > hg > mercurial-crew
annotate hgext/transplant.py @ 4959:97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
| author | Matt Mackall <mpm@selenic.com> |
|---|---|
| date | Sat, 21 Jul 2007 16:02:10 -0500 |
| parents | 059bdc8dfb9d |
| children | 126f527b3ba3 |
| rev | line source |
|---|---|
| 3714 | 1 # Patch transplanting extension for Mercurial |
| 2 # | |
|
4635
63b9d2deed48
Updated copyright notices and add "and others" to "hg version"
Thomas Arendsen Hein <thomas@intevation.de>
parents:
4516
diff
changeset
|
3 # Copyright 2006, 2007 Brendan Cully <brendan@kublai.com> |
| 3714 | 4 # |
| 5 # This software may be used and distributed according to the terms | |
| 6 # of the GNU General Public License, incorporated herein by reference. | |
| 7 | |
| 3893 | 8 from mercurial.i18n import _ |
|
3886
abaee83ce0a6
Replace demandload with new demandimport
Matt Mackall <mpm@selenic.com>
parents:
3827
diff
changeset
|
9 import os, tempfile |
|
4049
863465381028
transplant: update to current writebundle API
Brendan Cully <brendan@kublai.com>
parents:
4035
diff
changeset
|
10 from mercurial import bundlerepo, changegroup, cmdutil, commands, hg, merge |
|
863465381028
transplant: update to current writebundle API
Brendan Cully <brendan@kublai.com>
parents:
4035
diff
changeset
|
11 from mercurial import patch, revlog, util |
| 3714 | 12 |
| 13 '''patch transplanting tool | |
| 14 | |
| 15 This extension allows you to transplant patches from another branch. | |
| 16 | |
| 17 Transplanted patches are recorded in .hg/transplant/transplants, as a map | |
| 18 from a changeset hash to its hash in the source repository. | |
| 19 ''' | |
| 20 | |
| 21 class transplantentry: | |
| 22 def __init__(self, lnode, rnode): | |
| 23 self.lnode = lnode | |
| 24 self.rnode = rnode | |
| 25 | |
| 26 class transplants: | |
| 27 def __init__(self, path=None, transplantfile=None, opener=None): | |
| 28 self.path = path | |
| 29 self.transplantfile = transplantfile | |
| 30 self.opener = opener | |
| 31 | |
| 32 if not opener: | |
| 33 self.opener = util.opener(self.path) | |
| 34 self.transplants = [] | |
| 35 self.dirty = False | |
| 36 self.read() | |
| 37 | |
| 38 def read(self): | |
| 39 abspath = os.path.join(self.path, self.transplantfile) | |
| 40 if self.transplantfile and os.path.exists(abspath): | |
| 41 for line in self.opener(self.transplantfile).read().splitlines(): | |
| 42 lnode, rnode = map(revlog.bin, line.split(':')) | |
| 43 self.transplants.append(transplantentry(lnode, rnode)) | |
| 44 | |
| 45 def write(self): | |
| 46 if self.dirty and self.transplantfile: | |
| 47 if not os.path.isdir(self.path): | |
| 48 os.mkdir(self.path) | |
| 49 fp = self.opener(self.transplantfile, 'w') | |
| 50 for c in self.transplants: | |
| 51 l, r = map(revlog.hex, (c.lnode, c.rnode)) | |
| 52 fp.write(l + ':' + r + '\n') | |
| 53 fp.close() | |
| 54 self.dirty = False | |
| 55 | |
| 56 def get(self, rnode): | |
| 57 return [t for t in self.transplants if t.rnode == rnode] | |
| 58 | |
| 59 def set(self, lnode, rnode): | |
| 60 self.transplants.append(transplantentry(lnode, rnode)) | |
| 61 self.dirty = True | |
| 62 | |
| 63 def remove(self, transplant): | |
| 64 del self.transplants[self.transplants.index(transplant)] | |
| 65 self.dirty = True | |
| 66 | |
| 67 class transplanter: | |
| 68 def __init__(self, ui, repo): | |
| 69 self.ui = ui | |
| 70 self.path = repo.join('transplant') | |
| 71 self.opener = util.opener(self.path) | |
| 72 self.transplants = transplants(self.path, 'transplants', opener=self.opener) | |
| 73 | |
| 74 def applied(self, repo, node, parent): | |
| 75 '''returns True if a node is already an ancestor of parent | |
| 76 or has already been transplanted''' | |
| 77 if hasnode(repo, node): | |
| 78 if node in repo.changelog.reachable(parent, stop=node): | |
| 79 return True | |
| 80 for t in self.transplants.get(node): | |
| 81 # it might have been stripped | |
| 82 if not hasnode(repo, t.lnode): | |
| 83 self.transplants.remove(t) | |
| 84 return False | |
| 85 if t.lnode in repo.changelog.reachable(parent, stop=t.lnode): | |
| 86 return True | |
| 87 return False | |
| 88 | |
| 89 def apply(self, repo, source, revmap, merges, opts={}): | |
| 90 '''apply the revisions in revmap one by one in revision order''' | |
| 91 revs = revmap.keys() | |
| 92 revs.sort() | |
| 93 | |
| 94 p1, p2 = repo.dirstate.parents() | |
| 95 pulls = [] | |
| 96 diffopts = patch.diffopts(self.ui, opts) | |
| 97 diffopts.git = True | |
| 98 | |
|
4959
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
99 lock = wlock = None |
| 3714 | 100 try: |
|
4959
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
101 wlock = repo.wlock() |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
102 lock = repo.lock() |
| 3714 | 103 for rev in revs: |
| 104 node = revmap[rev] | |
| 105 revstr = '%s:%s' % (rev, revlog.short(node)) | |
| 106 | |
| 107 if self.applied(repo, node, p1): | |
| 108 self.ui.warn(_('skipping already applied revision %s\n') % | |
| 109 revstr) | |
| 110 continue | |
| 111 | |
| 112 parents = source.changelog.parents(node) | |
| 113 if not opts.get('filter'): | |
| 114 # If the changeset parent is the same as the wdir's parent, | |
| 115 # just pull it. | |
| 116 if parents[0] == p1: | |
| 117 pulls.append(node) | |
| 118 p1 = node | |
| 119 continue | |
| 120 if pulls: | |
| 121 if source != repo: | |
| 122 repo.pull(source, heads=pulls, lock=lock) | |
|
4035
c8f26bd24e86
Update transplant's merge.update call
Brendan Cully <brendan@kublai.com>
parents:
3991
diff
changeset
|
123 merge.update(repo, pulls[-1], False, False, None, |
|
c8f26bd24e86
Update transplant's merge.update call
Brendan Cully <brendan@kublai.com>
parents:
3991
diff
changeset
|
124 wlock=wlock) |
| 3714 | 125 p1, p2 = repo.dirstate.parents() |
| 126 pulls = [] | |
| 127 | |
| 128 domerge = False | |
| 129 if node in merges: | |
| 130 # pulling all the merge revs at once would mean we couldn't | |
| 131 # transplant after the latest even if transplants before them | |
| 132 # fail. | |
| 133 domerge = True | |
| 134 if not hasnode(repo, node): | |
| 135 repo.pull(source, heads=[node], lock=lock) | |
| 136 | |
| 137 if parents[1] != revlog.nullid: | |
| 138 self.ui.note(_('skipping merge changeset %s:%s\n') | |
| 139 % (rev, revlog.short(node))) | |
| 140 patchfile = None | |
| 141 else: | |
| 142 fd, patchfile = tempfile.mkstemp(prefix='hg-transplant-') | |
| 143 fp = os.fdopen(fd, 'w') | |
|
3759
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
144 patch.diff(source, parents[0], node, fp=fp, opts=diffopts) |
| 3714 | 145 fp.close() |
| 146 | |
| 147 del revmap[rev] | |
| 148 if patchfile or domerge: | |
| 149 try: | |
| 150 n = self.applyone(repo, node, source.changelog.read(node), | |
| 151 patchfile, merge=domerge, | |
| 152 log=opts.get('log'), | |
| 153 filter=opts.get('filter'), | |
| 154 lock=lock, wlock=wlock) | |
|
4251
e76e52145c3d
transplant: fix ignoring empty changesets (eg after filter)
Brendan Cully <brendan@kublai.com>
parents:
4072
diff
changeset
|
155 if n and domerge: |
| 3714 | 156 self.ui.status(_('%s merged at %s\n') % (revstr, |
| 157 revlog.short(n))) | |
|
4251
e76e52145c3d
transplant: fix ignoring empty changesets (eg after filter)
Brendan Cully <brendan@kublai.com>
parents:
4072
diff
changeset
|
158 elif n: |
| 3714 | 159 self.ui.status(_('%s transplanted to %s\n') % (revlog.short(node), |
| 160 revlog.short(n))) | |
| 161 finally: | |
| 162 if patchfile: | |
| 163 os.unlink(patchfile) | |
| 164 if pulls: | |
| 165 repo.pull(source, heads=pulls, lock=lock) | |
|
4035
c8f26bd24e86
Update transplant's merge.update call
Brendan Cully <brendan@kublai.com>
parents:
3991
diff
changeset
|
166 merge.update(repo, pulls[-1], False, False, None, wlock=wlock) |
| 3714 | 167 finally: |
| 168 self.saveseries(revmap, merges) | |
| 169 self.transplants.write() | |
|
4959
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
170 del lock, wlock |
| 3714 | 171 |
| 172 def filter(self, filter, changelog, patchfile): | |
| 173 '''arbitrarily rewrite changeset before applying it''' | |
| 174 | |
|
3752
f902f409cd81
transplant: "filtering %s\n"
Brendan Cully <brendan@kublai.com>
parents:
3726
diff
changeset
|
175 self.ui.status('filtering %s\n' % patchfile) |
|
3759
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
176 user, date, msg = (changelog[1], changelog[2], changelog[4]) |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
177 |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
178 fd, headerfile = tempfile.mkstemp(prefix='hg-transplant-') |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
179 fp = os.fdopen(fd, 'w') |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
180 fp.write("# HG changeset patch\n") |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
181 fp.write("# User %s\n" % user) |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
182 fp.write("# Date %d %d\n" % date) |
|
3827
2db191165e9a
transplant: don't add extra newlines to changelog entry in filter
Brendan Cully <brendan@kublai.com>
parents:
3759
diff
changeset
|
183 fp.write(changelog[4]) |
|
3759
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
184 fp.close() |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
185 |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
186 try: |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
187 util.system('%s %s %s' % (filter, util.shellquote(headerfile), |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
188 util.shellquote(patchfile)), |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
189 environ={'HGUSER': changelog[1]}, |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
190 onerr=util.Abort, errprefix=_('filter failed')) |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
191 user, date, msg = self.parselog(file(headerfile))[1:4] |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
192 finally: |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
193 os.unlink(headerfile) |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
194 |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
195 return (user, date, msg) |
| 3714 | 196 |
| 197 def applyone(self, repo, node, cl, patchfile, merge=False, log=False, | |
| 198 filter=None, lock=None, wlock=None): | |
| 199 '''apply the patch in patchfile to the repository as a transplant''' | |
| 200 (manifest, user, (time, timezone), files, message) = cl[:5] | |
| 201 date = "%d %d" % (time, timezone) | |
| 202 extra = {'transplant_source': node} | |
| 203 if filter: | |
|
3759
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
204 (user, date, message) = self.filter(filter, cl, patchfile) |
| 3714 | 205 |
| 206 if log: | |
| 207 message += '\n(transplanted from %s)' % revlog.hex(node) | |
| 208 | |
| 209 self.ui.status(_('applying %s\n') % revlog.short(node)) | |
| 210 self.ui.note('%s %s\n%s\n' % (user, date, message)) | |
| 211 | |
| 212 if not patchfile and not merge: | |
| 213 raise util.Abort(_('can only omit patchfile if merging')) | |
| 214 if patchfile: | |
| 215 try: | |
| 216 files = {} | |
|
3726
752884db5037
transplant: recover added/removed files after failed application
Brendan Cully <brendan@kublai.com>
parents:
3725
diff
changeset
|
217 try: |
|
752884db5037
transplant: recover added/removed files after failed application
Brendan Cully <brendan@kublai.com>
parents:
3725
diff
changeset
|
218 fuzz = patch.patch(patchfile, self.ui, cwd=repo.root, |
|
752884db5037
transplant: recover added/removed files after failed application
Brendan Cully <brendan@kublai.com>
parents:
3725
diff
changeset
|
219 files=files) |
|
752884db5037
transplant: recover added/removed files after failed application
Brendan Cully <brendan@kublai.com>
parents:
3725
diff
changeset
|
220 if not files: |
|
752884db5037
transplant: recover added/removed files after failed application
Brendan Cully <brendan@kublai.com>
parents:
3725
diff
changeset
|
221 self.ui.warn(_('%s: empty changeset') % revlog.hex(node)) |
|
4251
e76e52145c3d
transplant: fix ignoring empty changesets (eg after filter)
Brendan Cully <brendan@kublai.com>
parents:
4072
diff
changeset
|
222 return None |
|
3726
752884db5037
transplant: recover added/removed files after failed application
Brendan Cully <brendan@kublai.com>
parents:
3725
diff
changeset
|
223 finally: |
|
752884db5037
transplant: recover added/removed files after failed application
Brendan Cully <brendan@kublai.com>
parents:
3725
diff
changeset
|
224 files = patch.updatedir(self.ui, repo, files, wlock=wlock) |
| 3714 | 225 except Exception, inst: |
| 226 if filter: | |
| 227 os.unlink(patchfile) | |
|
3757
faed44bab17b
transplant: clobber old series when transplant fails
Brendan Cully <brendan@kublai.com>
parents:
3752
diff
changeset
|
228 seriespath = os.path.join(self.path, 'series') |
|
faed44bab17b
transplant: clobber old series when transplant fails
Brendan Cully <brendan@kublai.com>
parents:
3752
diff
changeset
|
229 if os.path.exists(seriespath): |
|
faed44bab17b
transplant: clobber old series when transplant fails
Brendan Cully <brendan@kublai.com>
parents:
3752
diff
changeset
|
230 os.unlink(seriespath) |
| 3714 | 231 p1 = repo.dirstate.parents()[0] |
| 232 p2 = node | |
|
3725
ccc7a9eb0e5e
transplant: preserve filter changes in --continue log
Brendan Cully <brendan@kublai.com>
parents:
3724
diff
changeset
|
233 self.log(user, date, message, p1, p2, merge=merge) |
| 3714 | 234 self.ui.write(str(inst) + '\n') |
| 235 raise util.Abort(_('Fix up the merge and run hg transplant --continue')) | |
| 236 else: | |
| 237 files = None | |
| 238 if merge: | |
| 239 p1, p2 = repo.dirstate.parents() | |
| 240 repo.dirstate.setparents(p1, node) | |
| 241 | |
| 242 n = repo.commit(files, message, user, date, lock=lock, wlock=wlock, | |
| 243 extra=extra) | |
| 244 if not merge: | |
| 245 self.transplants.set(n, node) | |
| 246 | |
| 247 return n | |
| 248 | |
| 249 def resume(self, repo, source, opts=None): | |
| 250 '''recover last transaction and apply remaining changesets''' | |
| 251 if os.path.exists(os.path.join(self.path, 'journal')): | |
| 252 n, node = self.recover(repo) | |
|
3724
ea523d6f5f1a
transplant: fix --continue; add --continue test
Brendan Cully <brendan@kublai.com>
parents:
3723
diff
changeset
|
253 self.ui.status(_('%s transplanted as %s\n') % (revlog.short(node), |
|
ea523d6f5f1a
transplant: fix --continue; add --continue test
Brendan Cully <brendan@kublai.com>
parents:
3723
diff
changeset
|
254 revlog.short(n))) |
| 3714 | 255 seriespath = os.path.join(self.path, 'series') |
| 256 if not os.path.exists(seriespath): | |
|
3758
889f7e74a0d9
transplant: log source node when recovering too.
Brendan Cully <brendan@kublai.com>
parents:
3757
diff
changeset
|
257 self.transplants.write() |
| 3714 | 258 return |
| 259 nodes, merges = self.readseries() | |
| 260 revmap = {} | |
| 261 for n in nodes: | |
| 262 revmap[source.changelog.rev(n)] = n | |
| 263 os.unlink(seriespath) | |
| 264 | |
| 265 self.apply(repo, source, revmap, merges, opts) | |
| 266 | |
| 267 def recover(self, repo): | |
| 268 '''commit working directory using journal metadata''' | |
| 269 node, user, date, message, parents = self.readlog() | |
| 270 merge = len(parents) == 2 | |
| 271 | |
| 272 if not user or not date or not message or not parents[0]: | |
| 273 raise util.Abort(_('transplant log file is corrupt')) | |
| 274 | |
|
3758
889f7e74a0d9
transplant: log source node when recovering too.
Brendan Cully <brendan@kublai.com>
parents:
3757
diff
changeset
|
275 extra = {'transplant_source': node} |
| 3714 | 276 wlock = repo.wlock() |
|
4959
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
277 try: |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
278 p1, p2 = repo.dirstate.parents() |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
279 if p1 != parents[0]: |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
280 raise util.Abort( |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
281 _('working dir not at transplant parent %s') % |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
282 revlog.hex(parents[0])) |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
283 if merge: |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
284 repo.dirstate.setparents(p1, parents[1]) |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
285 n = repo.commit(None, message, user, date, wlock=wlock, |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
286 extra=extra) |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
287 if not n: |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
288 raise util.Abort(_('commit failed')) |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
289 if not merge: |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
290 self.transplants.set(n, node) |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
291 self.unlog() |
| 3714 | 292 |
|
4959
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
293 return n, node |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
294 finally: |
|
97b734fb9c6f
Use try/finally pattern to cleanup locks and transactions
Matt Mackall <mpm@selenic.com>
parents:
4680
diff
changeset
|
295 del wlock |
| 3714 | 296 |
| 297 def readseries(self): | |
| 298 nodes = [] | |
| 299 merges = [] | |
| 300 cur = nodes | |
| 301 for line in self.opener('series').read().splitlines(): | |
| 302 if line.startswith('# Merges'): | |
| 303 cur = merges | |
| 304 continue | |
| 305 cur.append(revlog.bin(line)) | |
| 306 | |
| 307 return (nodes, merges) | |
| 308 | |
| 309 def saveseries(self, revmap, merges): | |
| 310 if not revmap: | |
| 311 return | |
| 312 | |
| 313 if not os.path.isdir(self.path): | |
| 314 os.mkdir(self.path) | |
| 315 series = self.opener('series', 'w') | |
| 316 revs = revmap.keys() | |
| 317 revs.sort() | |
| 318 for rev in revs: | |
| 319 series.write(revlog.hex(revmap[rev]) + '\n') | |
| 320 if merges: | |
| 321 series.write('# Merges\n') | |
| 322 for m in merges: | |
| 323 series.write(revlog.hex(m) + '\n') | |
| 324 series.close() | |
| 325 | |
|
3759
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
326 def parselog(self, fp): |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
327 parents = [] |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
328 message = [] |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
329 node = revlog.nullid |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
330 inmsg = False |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
331 for line in fp.read().splitlines(): |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
332 if inmsg: |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
333 message.append(line) |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
334 elif line.startswith('# User '): |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
335 user = line[7:] |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
336 elif line.startswith('# Date '): |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
337 date = line[7:] |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
338 elif line.startswith('# Node ID '): |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
339 node = revlog.bin(line[10:]) |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
340 elif line.startswith('# Parent '): |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
341 parents.append(revlog.bin(line[9:])) |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
342 elif not line.startswith('#'): |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
343 inmsg = True |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
344 message.append(line) |
|
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
345 return (node, user, date, '\n'.join(message), parents) |
|
4516
96d8a56d4ef9
Removed trailing whitespace and tabs from python files
Thomas Arendsen Hein <thomas@intevation.de>
parents:
4251
diff
changeset
|
346 |
|
3725
ccc7a9eb0e5e
transplant: preserve filter changes in --continue log
Brendan Cully <brendan@kublai.com>
parents:
3724
diff
changeset
|
347 def log(self, user, date, message, p1, p2, merge=False): |
| 3714 | 348 '''journal changelog metadata for later recover''' |
| 349 | |
| 350 if not os.path.isdir(self.path): | |
| 351 os.mkdir(self.path) | |
| 352 fp = self.opener('journal', 'w') | |
|
3725
ccc7a9eb0e5e
transplant: preserve filter changes in --continue log
Brendan Cully <brendan@kublai.com>
parents:
3724
diff
changeset
|
353 fp.write('# User %s\n' % user) |
|
ccc7a9eb0e5e
transplant: preserve filter changes in --continue log
Brendan Cully <brendan@kublai.com>
parents:
3724
diff
changeset
|
354 fp.write('# Date %s\n' % date) |
| 3714 | 355 fp.write('# Node ID %s\n' % revlog.hex(p2)) |
| 356 fp.write('# Parent ' + revlog.hex(p1) + '\n') | |
| 357 if merge: | |
| 358 fp.write('# Parent ' + revlog.hex(p2) + '\n') | |
|
3725
ccc7a9eb0e5e
transplant: preserve filter changes in --continue log
Brendan Cully <brendan@kublai.com>
parents:
3724
diff
changeset
|
359 fp.write(message.rstrip() + '\n') |
| 3714 | 360 fp.close() |
| 361 | |
| 362 def readlog(self): | |
|
3759
e96f97ca0358
transplant: split filter args into changelog entry and patch
Brendan Cully <brendan@kublai.com>
parents:
3758
diff
changeset
|
363 return self.parselog(self.opener('journal')) |
| 3714 | 364 |
| 365 def unlog(self): | |
| 366 '''remove changelog journal''' | |
| 367 absdst = os.path.join(self.path, 'journal') | |
| 368 if os.path.exists(absdst): | |
| 369 os.unlink(absdst) | |
| 370 | |
| 371 def transplantfilter(self, repo, source, root): | |
| 372 def matchfn(node): | |
| 373 if self.applied(repo, node, root): | |
| 374 return False | |
| 375 if source.changelog.parents(node)[1] != revlog.nullid: | |
| 376 return False | |
| 377 extra = source.changelog.read(node)[5] | |
| 378 cnode = extra.get('transplant_source') | |
| 379 if cnode and self.applied(repo, cnode, root): | |
| 380 return False | |
| 381 return True | |
| 382 | |
| 383 return matchfn | |
| 384 | |
| 385 def hasnode(repo, node): | |
| 386 try: | |
| 387 return repo.changelog.rev(node) != None | |
| 388 except revlog.RevlogError: | |
| 389 return False | |
| 390 | |
| 391 def browserevs(ui, repo, nodes, opts): | |
| 392 '''interactively transplant changesets''' | |
| 393 def browsehelp(ui): | |
| 394 ui.write('y: transplant this changeset\n' | |
| 395 'n: skip this changeset\n' | |
| 396 'm: merge at this changeset\n' | |
| 397 'p: show patch\n' | |
| 398 'c: commit selected changesets\n' | |
| 399 'q: cancel transplant\n' | |
| 400 '?: show this help\n') | |
| 401 | |
|
3723
c828fca6f38a
transplant: show_changeset moved to cmdutil
Brendan Cully <brendan@kublai.com>
parents:
3714
diff
changeset
|
402 displayer = cmdutil.show_changeset(ui, repo, opts) |
| 3714 | 403 transplants = [] |
| 404 merges = [] | |
| 405 for node in nodes: | |
| 406 displayer.show(changenode=node) | |
| 407 action = None | |
| 408 while not action: | |
| 409 action = ui.prompt(_('apply changeset? [ynmpcq?]:')) | |
| 410 if action == '?': | |
| 411 browsehelp(ui) | |
| 412 action = None | |
| 413 elif action == 'p': | |
| 414 parent = repo.changelog.parents(node)[0] | |
| 415 patch.diff(repo, parent, node) | |
| 416 action = None | |
| 417 elif action not in ('y', 'n', 'm', 'c', 'q'): | |
| 418 ui.write('no such option\n') | |
| 419 action = None | |
| 420 if action == 'y': | |
| 421 transplants.append(node) | |
| 422 elif action == 'm': | |
| 423 merges.append(node) | |
| 424 elif action == 'c': | |
| 425 break | |
| 426 elif action == 'q': | |
| 427 transplants = () | |
| 428 merges = () | |
| 429 break | |
| 430 return (transplants, merges) | |
| 431 | |
| 432 def transplant(ui, repo, *revs, **opts): | |
| 433 '''transplant changesets from another branch | |
| 434 | |
| 435 Selected changesets will be applied on top of the current working | |
| 436 directory with the log of the original changeset. If --log is | |
| 437 specified, log messages will have a comment appended of the form: | |
| 438 | |
| 439 (transplanted from CHANGESETHASH) | |
| 440 | |
| 441 You can rewrite the changelog message with the --filter option. | |
| 442 Its argument will be invoked with the current changelog message | |
| 443 as $1 and the patch as $2. | |
| 444 | |
| 445 If --source is specified, selects changesets from the named | |
| 446 repository. If --branch is specified, selects changesets from the | |
| 447 branch holding the named revision, up to that revision. If --all | |
| 448 is specified, all changesets on the branch will be transplanted, | |
| 449 otherwise you will be prompted to select the changesets you want. | |
| 450 | |
| 451 hg transplant --branch REVISION --all will rebase the selected branch | |
| 452 (up to the named revision) onto your current working directory. | |
| 453 | |
| 454 You can optionally mark selected transplanted changesets as | |
| 455 merge changesets. You will not be prompted to transplant any | |
| 456 ancestors of a merged transplant, and you can merge descendants | |
| 457 of them normally instead of transplanting them. | |
| 458 | |
| 459 If no merges or revisions are provided, hg transplant will start | |
| 460 an interactive changeset browser. | |
| 461 | |
| 462 If a changeset application fails, you can fix the merge by hand and | |
| 463 then resume where you left off by calling hg transplant --continue. | |
| 464 ''' | |
| 465 def getoneitem(opts, item, errmsg): | |
| 466 val = opts.get(item) | |
| 467 if val: | |
| 468 if len(val) > 1: | |
| 469 raise util.Abort(errmsg) | |
| 470 else: | |
| 471 return val[0] | |
| 472 | |
| 473 def getremotechanges(repo, url): | |
| 474 sourcerepo = ui.expandpath(url) | |
| 475 source = hg.repository(ui, sourcerepo) | |
| 476 incoming = repo.findincoming(source, force=True) | |
| 477 if not incoming: | |
| 478 return (source, None, None) | |
| 479 | |
| 480 bundle = None | |
| 481 if not source.local(): | |
| 482 cg = source.changegroup(incoming, 'incoming') | |
|
4049
863465381028
transplant: update to current writebundle API
Brendan Cully <brendan@kublai.com>
parents:
4035
diff
changeset
|
483 bundle = changegroup.writebundle(cg, None, 'HG10UN') |
| 3714 | 484 source = bundlerepo.bundlerepository(ui, repo.root, bundle) |
| 485 | |
| 486 return (source, incoming, bundle) | |
| 487 | |
| 488 def incwalk(repo, incoming, branches, match=util.always): | |
| 489 if not branches: | |
| 490 branches=None | |
| 491 for node in repo.changelog.nodesbetween(incoming, branches)[0]: | |
| 492 if match(node): | |
| 493 yield node | |
| 494 | |
| 495 def transplantwalk(repo, root, branches, match=util.always): | |
| 496 if not branches: | |
| 497 branches = repo.heads() | |
| 498 ancestors = [] | |
| 499 for branch in branches: | |
| 500 ancestors.append(repo.changelog.ancestor(root, branch)) | |
| 501 for node in repo.changelog.nodesbetween(ancestors, branches)[0]: | |
| 502 if match(node): | |
| 503 yield node | |
| 504 | |
| 505 def checkopts(opts, revs): | |
| 506 if opts.get('continue'): | |
| 507 if filter(lambda opt: opts.get(opt), ('branch', 'all', 'merge')): | |
| 508 raise util.Abort(_('--continue is incompatible with branch, all or merge')) | |
| 509 return | |
| 510 if not (opts.get('source') or revs or | |
| 511 opts.get('merge') or opts.get('branch')): | |
| 512 raise util.Abort(_('no source URL, branch tag or revision list provided')) | |
| 513 if opts.get('all'): | |
| 514 if not opts.get('branch'): | |
| 515 raise util.Abort(_('--all requires a branch revision')) | |
| 516 if revs: | |
| 517 raise util.Abort(_('--all is incompatible with a revision list')) | |
| 518 | |
| 519 checkopts(opts, revs) | |
| 520 | |
| 521 if not opts.get('log'): | |
| 522 opts['log'] = ui.config('transplant', 'log') | |
| 523 if not opts.get('filter'): | |
| 524 opts['filter'] = ui.config('transplant', 'filter') | |
| 525 | |
| 526 tp = transplanter(ui, repo) | |
| 527 | |
| 528 p1, p2 = repo.dirstate.parents() | |
| 529 if p1 == revlog.nullid: | |
| 530 raise util.Abort(_('no revision checked out')) | |
| 531 if not opts.get('continue'): | |
| 532 if p2 != revlog.nullid: | |
| 533 raise util.Abort(_('outstanding uncommitted merges')) | |
| 534 m, a, r, d = repo.status()[:4] | |
| 535 if m or a or r or d: | |
| 536 raise util.Abort(_('outstanding local changes')) | |
| 537 | |
| 538 bundle = None | |
| 539 source = opts.get('source') | |
| 540 if source: | |
| 541 (source, incoming, bundle) = getremotechanges(repo, source) | |
| 542 else: | |
| 543 source = repo | |
| 544 | |
| 545 try: | |
| 546 if opts.get('continue'): | |
|
3724
ea523d6f5f1a
transplant: fix --continue; add --continue test
Brendan Cully <brendan@kublai.com>
parents:
3723
diff
changeset
|
547 tp.resume(repo, source, opts) |
| 3714 | 548 return |
| 549 | |
| 550 tf=tp.transplantfilter(repo, source, p1) | |
| 551 if opts.get('prune'): | |
| 552 prune = [source.lookup(r) | |
| 553 for r in cmdutil.revrange(source, opts.get('prune'))] | |
| 554 matchfn = lambda x: tf(x) and x not in prune | |
| 555 else: | |
| 556 matchfn = tf | |
| 557 branches = map(source.lookup, opts.get('branch', ())) | |
| 558 merges = map(source.lookup, opts.get('merge', ())) | |
| 559 revmap = {} | |
| 560 if revs: | |
| 561 for r in cmdutil.revrange(source, revs): | |
| 562 revmap[int(r)] = source.lookup(r) | |
| 563 elif opts.get('all') or not merges: | |
| 564 if source != repo: | |
| 565 alltransplants = incwalk(source, incoming, branches, match=matchfn) | |
| 566 else: | |
| 567 alltransplants = transplantwalk(source, p1, branches, match=matchfn) | |
| 568 if opts.get('all'): | |
| 569 revs = alltransplants | |
| 570 else: | |
| 571 revs, newmerges = browserevs(ui, source, alltransplants, opts) | |
| 572 merges.extend(newmerges) | |
| 573 for r in revs: | |
| 574 revmap[source.changelog.rev(r)] = r | |
| 575 for r in merges: | |
| 576 revmap[source.changelog.rev(r)] = r | |
| 577 | |
| 578 revs = revmap.keys() | |
| 579 revs.sort() | |
| 580 pulls = [] | |
| 581 | |
| 582 tp.apply(repo, source, revmap, merges, opts) | |
| 583 finally: | |
| 584 if bundle: | |
|
4072
e916bc0dfdd6
transplant: remote bundle source was not closed before deleting the fetched bundle.
Patrick Mezard <pmezard@gmail.com>
parents:
4049
diff
changeset
|
585 source.close() |
| 3714 | 586 os.unlink(bundle) |
| 587 | |
| 588 cmdtable = { | |
| 589 "transplant": | |
| 590 (transplant, | |
| 591 [('s', 'source', '', _('pull patches from REPOSITORY')), | |
| 592 ('b', 'branch', [], _('pull patches from branch BRANCH')), | |
| 593 ('a', 'all', None, _('pull all changesets up to BRANCH')), | |
| 594 ('p', 'prune', [], _('skip over REV')), | |
| 595 ('m', 'merge', [], _('merge at REV')), | |
| 596 ('', 'log', None, _('append transplant info to log message')), | |
| 597 ('c', 'continue', None, _('continue last transplant session after repair')), | |
| 598 ('', 'filter', '', _('filter changesets through FILTER'))], | |
|
3991
da3dc89f1e9a
Corrected synopsis for transplant.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
3893
diff
changeset
|
599 _('hg transplant [-s REPOSITORY] [-b BRANCH [-a]] [-p REV] [-m REV] [REV]...')) |
| 3714 | 600 } |