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