##// END OF EJS Templates
mq: Mention usage of hg add/remove/copy/rename in qrefresh help text.
Thomas Arendsen Hein -
r4048:5d6b3fa6 default
parent child Browse files
Show More
@@ -1,2185 +1,2188 b''
1 1 # queue.py - patch queues for mercurial
2 2 #
3 3 # Copyright 2005, 2006 Chris Mason <mason@suse.com>
4 4 #
5 5 # This software may be used and distributed according to the terms
6 6 # of the GNU General Public License, incorporated herein by reference.
7 7
8 8 '''patch management and development
9 9
10 10 This extension lets you work with a stack of patches in a Mercurial
11 11 repository. It manages two stacks of patches - all known patches, and
12 12 applied patches (subset of known patches).
13 13
14 14 Known patches are represented as patch files in the .hg/patches
15 15 directory. Applied patches are both patch files and changesets.
16 16
17 17 Common tasks (use "hg help command" for more details):
18 18
19 19 prepare repository to work with patches qinit
20 20 create new patch qnew
21 21 import existing patch qimport
22 22
23 23 print patch series qseries
24 24 print applied patches qapplied
25 25 print name of top applied patch qtop
26 26
27 27 add known patch to applied stack qpush
28 28 remove patch from applied stack qpop
29 29 refresh contents of top applied patch qrefresh
30 30 '''
31 31
32 32 from mercurial.i18n import _
33 33 from mercurial import commands, cmdutil, hg, patch, revlog, util, changegroup
34 34 import os, sys, re, errno
35 35
36 36 commands.norepo += " qclone qversion"
37 37
38 38 # Patch names looks like unix-file names.
39 39 # They must be joinable with queue directory and result in the patch path.
40 40 normname = util.normpath
41 41
42 42 class statusentry:
43 43 def __init__(self, rev, name=None):
44 44 if not name:
45 45 fields = rev.split(':', 1)
46 46 if len(fields) == 2:
47 47 self.rev, self.name = fields
48 48 else:
49 49 self.rev, self.name = None, None
50 50 else:
51 51 self.rev, self.name = rev, name
52 52
53 53 def __str__(self):
54 54 return self.rev + ':' + self.name
55 55
56 56 class queue:
57 57 def __init__(self, ui, path, patchdir=None):
58 58 self.basepath = path
59 59 self.path = patchdir or os.path.join(path, "patches")
60 60 self.opener = util.opener(self.path)
61 61 self.ui = ui
62 62 self.applied = []
63 63 self.full_series = []
64 64 self.applied_dirty = 0
65 65 self.series_dirty = 0
66 66 self.series_path = "series"
67 67 self.status_path = "status"
68 68 self.guards_path = "guards"
69 69 self.active_guards = None
70 70 self.guards_dirty = False
71 71 self._diffopts = None
72 72
73 73 if os.path.exists(self.join(self.series_path)):
74 74 self.full_series = self.opener(self.series_path).read().splitlines()
75 75 self.parse_series()
76 76
77 77 if os.path.exists(self.join(self.status_path)):
78 78 lines = self.opener(self.status_path).read().splitlines()
79 79 self.applied = [statusentry(l) for l in lines]
80 80
81 81 def diffopts(self):
82 82 if self._diffopts is None:
83 83 self._diffopts = patch.diffopts(self.ui)
84 84 return self._diffopts
85 85
86 86 def join(self, *p):
87 87 return os.path.join(self.path, *p)
88 88
89 89 def find_series(self, patch):
90 90 pre = re.compile("(\s*)([^#]+)")
91 91 index = 0
92 92 for l in self.full_series:
93 93 m = pre.match(l)
94 94 if m:
95 95 s = m.group(2)
96 96 s = s.rstrip()
97 97 if s == patch:
98 98 return index
99 99 index += 1
100 100 return None
101 101
102 102 guard_re = re.compile(r'\s?#([-+][^-+# \t\r\n\f][^# \t\r\n\f]*)')
103 103
104 104 def parse_series(self):
105 105 self.series = []
106 106 self.series_guards = []
107 107 for l in self.full_series:
108 108 h = l.find('#')
109 109 if h == -1:
110 110 patch = l
111 111 comment = ''
112 112 elif h == 0:
113 113 continue
114 114 else:
115 115 patch = l[:h]
116 116 comment = l[h:]
117 117 patch = patch.strip()
118 118 if patch:
119 119 if patch in self.series:
120 120 raise util.Abort(_('%s appears more than once in %s') %
121 121 (patch, self.join(self.series_path)))
122 122 self.series.append(patch)
123 123 self.series_guards.append(self.guard_re.findall(comment))
124 124
125 125 def check_guard(self, guard):
126 126 bad_chars = '# \t\r\n\f'
127 127 first = guard[0]
128 128 for c in '-+':
129 129 if first == c:
130 130 return (_('guard %r starts with invalid character: %r') %
131 131 (guard, c))
132 132 for c in bad_chars:
133 133 if c in guard:
134 134 return _('invalid character in guard %r: %r') % (guard, c)
135 135
136 136 def set_active(self, guards):
137 137 for guard in guards:
138 138 bad = self.check_guard(guard)
139 139 if bad:
140 140 raise util.Abort(bad)
141 141 guards = dict.fromkeys(guards).keys()
142 142 guards.sort()
143 143 self.ui.debug('active guards: %s\n' % ' '.join(guards))
144 144 self.active_guards = guards
145 145 self.guards_dirty = True
146 146
147 147 def active(self):
148 148 if self.active_guards is None:
149 149 self.active_guards = []
150 150 try:
151 151 guards = self.opener(self.guards_path).read().split()
152 152 except IOError, err:
153 153 if err.errno != errno.ENOENT: raise
154 154 guards = []
155 155 for i, guard in enumerate(guards):
156 156 bad = self.check_guard(guard)
157 157 if bad:
158 158 self.ui.warn('%s:%d: %s\n' %
159 159 (self.join(self.guards_path), i + 1, bad))
160 160 else:
161 161 self.active_guards.append(guard)
162 162 return self.active_guards
163 163
164 164 def set_guards(self, idx, guards):
165 165 for g in guards:
166 166 if len(g) < 2:
167 167 raise util.Abort(_('guard %r too short') % g)
168 168 if g[0] not in '-+':
169 169 raise util.Abort(_('guard %r starts with invalid char') % g)
170 170 bad = self.check_guard(g[1:])
171 171 if bad:
172 172 raise util.Abort(bad)
173 173 drop = self.guard_re.sub('', self.full_series[idx])
174 174 self.full_series[idx] = drop + ''.join([' #' + g for g in guards])
175 175 self.parse_series()
176 176 self.series_dirty = True
177 177
178 178 def pushable(self, idx):
179 179 if isinstance(idx, str):
180 180 idx = self.series.index(idx)
181 181 patchguards = self.series_guards[idx]
182 182 if not patchguards:
183 183 return True, None
184 184 default = False
185 185 guards = self.active()
186 186 exactneg = [g for g in patchguards if g[0] == '-' and g[1:] in guards]
187 187 if exactneg:
188 188 return False, exactneg[0]
189 189 pos = [g for g in patchguards if g[0] == '+']
190 190 exactpos = [g for g in pos if g[1:] in guards]
191 191 if pos:
192 192 if exactpos:
193 193 return True, exactpos[0]
194 194 return False, pos
195 195 return True, ''
196 196
197 197 def explain_pushable(self, idx, all_patches=False):
198 198 write = all_patches and self.ui.write or self.ui.warn
199 199 if all_patches or self.ui.verbose:
200 200 if isinstance(idx, str):
201 201 idx = self.series.index(idx)
202 202 pushable, why = self.pushable(idx)
203 203 if all_patches and pushable:
204 204 if why is None:
205 205 write(_('allowing %s - no guards in effect\n') %
206 206 self.series[idx])
207 207 else:
208 208 if not why:
209 209 write(_('allowing %s - no matching negative guards\n') %
210 210 self.series[idx])
211 211 else:
212 212 write(_('allowing %s - guarded by %r\n') %
213 213 (self.series[idx], why))
214 214 if not pushable:
215 215 if why:
216 216 write(_('skipping %s - guarded by %r\n') %
217 217 (self.series[idx], why))
218 218 else:
219 219 write(_('skipping %s - no matching guards\n') %
220 220 self.series[idx])
221 221
222 222 def save_dirty(self):
223 223 def write_list(items, path):
224 224 fp = self.opener(path, 'w')
225 225 for i in items:
226 226 print >> fp, i
227 227 fp.close()
228 228 if self.applied_dirty: write_list(map(str, self.applied), self.status_path)
229 229 if self.series_dirty: write_list(self.full_series, self.series_path)
230 230 if self.guards_dirty: write_list(self.active_guards, self.guards_path)
231 231
232 232 def readheaders(self, patch):
233 233 def eatdiff(lines):
234 234 while lines:
235 235 l = lines[-1]
236 236 if (l.startswith("diff -") or
237 237 l.startswith("Index:") or
238 238 l.startswith("===========")):
239 239 del lines[-1]
240 240 else:
241 241 break
242 242 def eatempty(lines):
243 243 while lines:
244 244 l = lines[-1]
245 245 if re.match('\s*$', l):
246 246 del lines[-1]
247 247 else:
248 248 break
249 249
250 250 pf = self.join(patch)
251 251 message = []
252 252 comments = []
253 253 user = None
254 254 date = None
255 255 format = None
256 256 subject = None
257 257 diffstart = 0
258 258
259 259 for line in file(pf):
260 260 line = line.rstrip()
261 261 if line.startswith('diff --git'):
262 262 diffstart = 2
263 263 break
264 264 if diffstart:
265 265 if line.startswith('+++ '):
266 266 diffstart = 2
267 267 break
268 268 if line.startswith("--- "):
269 269 diffstart = 1
270 270 continue
271 271 elif format == "hgpatch":
272 272 # parse values when importing the result of an hg export
273 273 if line.startswith("# User "):
274 274 user = line[7:]
275 275 elif line.startswith("# Date "):
276 276 date = line[7:]
277 277 elif not line.startswith("# ") and line:
278 278 message.append(line)
279 279 format = None
280 280 elif line == '# HG changeset patch':
281 281 format = "hgpatch"
282 282 elif (format != "tagdone" and (line.startswith("Subject: ") or
283 283 line.startswith("subject: "))):
284 284 subject = line[9:]
285 285 format = "tag"
286 286 elif (format != "tagdone" and (line.startswith("From: ") or
287 287 line.startswith("from: "))):
288 288 user = line[6:]
289 289 format = "tag"
290 290 elif format == "tag" and line == "":
291 291 # when looking for tags (subject: from: etc) they
292 292 # end once you find a blank line in the source
293 293 format = "tagdone"
294 294 elif message or line:
295 295 message.append(line)
296 296 comments.append(line)
297 297
298 298 eatdiff(message)
299 299 eatdiff(comments)
300 300 eatempty(message)
301 301 eatempty(comments)
302 302
303 303 # make sure message isn't empty
304 304 if format and format.startswith("tag") and subject:
305 305 message.insert(0, "")
306 306 message.insert(0, subject)
307 307 return (message, comments, user, date, diffstart > 1)
308 308
309 309 def printdiff(self, repo, node1, node2=None, files=None,
310 310 fp=None, changes=None, opts={}):
311 311 fns, matchfn, anypats = cmdutil.matchpats(repo, files, opts)
312 312
313 313 patch.diff(repo, node1, node2, fns, match=matchfn,
314 314 fp=fp, changes=changes, opts=self.diffopts())
315 315
316 316 def mergeone(self, repo, mergeq, head, patch, rev, wlock):
317 317 # first try just applying the patch
318 318 (err, n) = self.apply(repo, [ patch ], update_status=False,
319 319 strict=True, merge=rev, wlock=wlock)
320 320
321 321 if err == 0:
322 322 return (err, n)
323 323
324 324 if n is None:
325 325 raise util.Abort(_("apply failed for patch %s") % patch)
326 326
327 327 self.ui.warn("patch didn't work out, merging %s\n" % patch)
328 328
329 329 # apply failed, strip away that rev and merge.
330 330 hg.clean(repo, head, wlock=wlock)
331 331 self.strip(repo, n, update=False, backup='strip', wlock=wlock)
332 332
333 333 ctx = repo.changectx(rev)
334 334 ret = hg.merge(repo, rev, wlock=wlock)
335 335 if ret:
336 336 raise util.Abort(_("update returned %d") % ret)
337 337 n = repo.commit(None, ctx.description(), ctx.user(),
338 338 force=1, wlock=wlock)
339 339 if n == None:
340 340 raise util.Abort(_("repo commit failed"))
341 341 try:
342 342 message, comments, user, date, patchfound = mergeq.readheaders(patch)
343 343 except:
344 344 raise util.Abort(_("unable to read %s") % patch)
345 345
346 346 patchf = self.opener(patch, "w")
347 347 if comments:
348 348 comments = "\n".join(comments) + '\n\n'
349 349 patchf.write(comments)
350 350 self.printdiff(repo, head, n, fp=patchf)
351 351 patchf.close()
352 352 return (0, n)
353 353
354 354 def qparents(self, repo, rev=None):
355 355 if rev is None:
356 356 (p1, p2) = repo.dirstate.parents()
357 357 if p2 == revlog.nullid:
358 358 return p1
359 359 if len(self.applied) == 0:
360 360 return None
361 361 return revlog.bin(self.applied[-1].rev)
362 362 pp = repo.changelog.parents(rev)
363 363 if pp[1] != revlog.nullid:
364 364 arevs = [ x.rev for x in self.applied ]
365 365 p0 = revlog.hex(pp[0])
366 366 p1 = revlog.hex(pp[1])
367 367 if p0 in arevs:
368 368 return pp[0]
369 369 if p1 in arevs:
370 370 return pp[1]
371 371 return pp[0]
372 372
373 373 def mergepatch(self, repo, mergeq, series, wlock):
374 374 if len(self.applied) == 0:
375 375 # each of the patches merged in will have two parents. This
376 376 # can confuse the qrefresh, qdiff, and strip code because it
377 377 # needs to know which parent is actually in the patch queue.
378 378 # so, we insert a merge marker with only one parent. This way
379 379 # the first patch in the queue is never a merge patch
380 380 #
381 381 pname = ".hg.patches.merge.marker"
382 382 n = repo.commit(None, '[mq]: merge marker', user=None, force=1,
383 383 wlock=wlock)
384 384 self.applied.append(statusentry(revlog.hex(n), pname))
385 385 self.applied_dirty = 1
386 386
387 387 head = self.qparents(repo)
388 388
389 389 for patch in series:
390 390 patch = mergeq.lookup(patch, strict=True)
391 391 if not patch:
392 392 self.ui.warn("patch %s does not exist\n" % patch)
393 393 return (1, None)
394 394 pushable, reason = self.pushable(patch)
395 395 if not pushable:
396 396 self.explain_pushable(patch, all_patches=True)
397 397 continue
398 398 info = mergeq.isapplied(patch)
399 399 if not info:
400 400 self.ui.warn("patch %s is not applied\n" % patch)
401 401 return (1, None)
402 402 rev = revlog.bin(info[1])
403 403 (err, head) = self.mergeone(repo, mergeq, head, patch, rev, wlock)
404 404 if head:
405 405 self.applied.append(statusentry(revlog.hex(head), patch))
406 406 self.applied_dirty = 1
407 407 if err:
408 408 return (err, head)
409 409 return (0, head)
410 410
411 411 def patch(self, repo, patchfile):
412 412 '''Apply patchfile to the working directory.
413 413 patchfile: file name of patch'''
414 414 files = {}
415 415 try:
416 416 fuzz = patch.patch(patchfile, self.ui, strip=1, cwd=repo.root,
417 417 files=files)
418 418 except Exception, inst:
419 419 self.ui.note(str(inst) + '\n')
420 420 if not self.ui.verbose:
421 421 self.ui.warn("patch failed, unable to continue (try -v)\n")
422 422 return (False, files, False)
423 423
424 424 return (True, files, fuzz)
425 425
426 426 def apply(self, repo, series, list=False, update_status=True,
427 427 strict=False, patchdir=None, merge=None, wlock=None):
428 428 # TODO unify with commands.py
429 429 if not patchdir:
430 430 patchdir = self.path
431 431 err = 0
432 432 if not wlock:
433 433 wlock = repo.wlock()
434 434 lock = repo.lock()
435 435 tr = repo.transaction()
436 436 n = None
437 437 for patchname in series:
438 438 pushable, reason = self.pushable(patchname)
439 439 if not pushable:
440 440 self.explain_pushable(patchname, all_patches=True)
441 441 continue
442 442 self.ui.warn("applying %s\n" % patchname)
443 443 pf = os.path.join(patchdir, patchname)
444 444
445 445 try:
446 446 message, comments, user, date, patchfound = self.readheaders(patchname)
447 447 except:
448 448 self.ui.warn("Unable to read %s\n" % patchname)
449 449 err = 1
450 450 break
451 451
452 452 if not message:
453 453 message = "imported patch %s\n" % patchname
454 454 else:
455 455 if list:
456 456 message.append("\nimported patch %s" % patchname)
457 457 message = '\n'.join(message)
458 458
459 459 (patcherr, files, fuzz) = self.patch(repo, pf)
460 460 patcherr = not patcherr
461 461
462 462 if merge and files:
463 463 # Mark as merged and update dirstate parent info
464 464 repo.dirstate.update(repo.dirstate.filterfiles(files.keys()), 'm')
465 465 p1, p2 = repo.dirstate.parents()
466 466 repo.dirstate.setparents(p1, merge)
467 467 files = patch.updatedir(self.ui, repo, files, wlock=wlock)
468 468 n = repo.commit(files, message, user, date, force=1, lock=lock,
469 469 wlock=wlock)
470 470
471 471 if n == None:
472 472 raise util.Abort(_("repo commit failed"))
473 473
474 474 if update_status:
475 475 self.applied.append(statusentry(revlog.hex(n), patchname))
476 476
477 477 if patcherr:
478 478 if not patchfound:
479 479 self.ui.warn("patch %s is empty\n" % patchname)
480 480 err = 0
481 481 else:
482 482 self.ui.warn("patch failed, rejects left in working dir\n")
483 483 err = 1
484 484 break
485 485
486 486 if fuzz and strict:
487 487 self.ui.warn("fuzz found when applying patch, stopping\n")
488 488 err = 1
489 489 break
490 490 tr.close()
491 491 return (err, n)
492 492
493 493 def delete(self, repo, patches, opts):
494 494 realpatches = []
495 495 for patch in patches:
496 496 patch = self.lookup(patch, strict=True)
497 497 info = self.isapplied(patch)
498 498 if info:
499 499 raise util.Abort(_("cannot delete applied patch %s") % patch)
500 500 if patch not in self.series:
501 501 raise util.Abort(_("patch %s not in series file") % patch)
502 502 realpatches.append(patch)
503 503
504 504 appliedbase = 0
505 505 if opts.get('rev'):
506 506 if not self.applied:
507 507 raise util.Abort(_('no patches applied'))
508 508 revs = cmdutil.revrange(repo, opts['rev'])
509 509 if len(revs) > 1 and revs[0] > revs[1]:
510 510 revs.reverse()
511 511 for rev in revs:
512 512 if appliedbase >= len(self.applied):
513 513 raise util.Abort(_("revision %d is not managed") % rev)
514 514
515 515 base = revlog.bin(self.applied[appliedbase].rev)
516 516 node = repo.changelog.node(rev)
517 517 if node != base:
518 518 raise util.Abort(_("cannot delete revision %d above "
519 519 "applied patches") % rev)
520 520 realpatches.append(self.applied[appliedbase].name)
521 521 appliedbase += 1
522 522
523 523 if not opts.get('keep'):
524 524 r = self.qrepo()
525 525 if r:
526 526 r.remove(realpatches, True)
527 527 else:
528 528 for p in realpatches:
529 529 os.unlink(self.join(p))
530 530
531 531 if appliedbase:
532 532 del self.applied[:appliedbase]
533 533 self.applied_dirty = 1
534 534 indices = [self.find_series(p) for p in realpatches]
535 535 indices.sort()
536 536 for i in indices[-1::-1]:
537 537 del self.full_series[i]
538 538 self.parse_series()
539 539 self.series_dirty = 1
540 540
541 541 def check_toppatch(self, repo):
542 542 if len(self.applied) > 0:
543 543 top = revlog.bin(self.applied[-1].rev)
544 544 pp = repo.dirstate.parents()
545 545 if top not in pp:
546 546 raise util.Abort(_("queue top not at same revision as working directory"))
547 547 return top
548 548 return None
549 549 def check_localchanges(self, repo, force=False, refresh=True):
550 550 m, a, r, d = repo.status()[:4]
551 551 if m or a or r or d:
552 552 if not force:
553 553 if refresh:
554 554 raise util.Abort(_("local changes found, refresh first"))
555 555 else:
556 556 raise util.Abort(_("local changes found"))
557 557 return m, a, r, d
558 558 def new(self, repo, patch, msg=None, force=None):
559 559 if os.path.exists(self.join(patch)):
560 560 raise util.Abort(_('patch "%s" already exists') % patch)
561 561 m, a, r, d = self.check_localchanges(repo, force)
562 562 commitfiles = m + a + r
563 563 self.check_toppatch(repo)
564 564 wlock = repo.wlock()
565 565 insert = self.full_series_end()
566 566 if msg:
567 567 n = repo.commit(commitfiles, "[mq]: %s" % msg, force=True,
568 568 wlock=wlock)
569 569 else:
570 570 n = repo.commit(commitfiles,
571 571 "New patch: %s" % patch, force=True, wlock=wlock)
572 572 if n == None:
573 573 raise util.Abort(_("repo commit failed"))
574 574 self.full_series[insert:insert] = [patch]
575 575 self.applied.append(statusentry(revlog.hex(n), patch))
576 576 self.parse_series()
577 577 self.series_dirty = 1
578 578 self.applied_dirty = 1
579 579 p = self.opener(patch, "w")
580 580 if msg:
581 581 msg = msg + "\n"
582 582 p.write(msg)
583 583 p.close()
584 584 wlock = None
585 585 r = self.qrepo()
586 586 if r: r.add([patch])
587 587 if commitfiles:
588 588 self.refresh(repo, short=True)
589 589
590 590 def strip(self, repo, rev, update=True, backup="all", wlock=None):
591 591 def limitheads(chlog, stop):
592 592 """return the list of all nodes that have no children"""
593 593 p = {}
594 594 h = []
595 595 stoprev = 0
596 596 if stop in chlog.nodemap:
597 597 stoprev = chlog.rev(stop)
598 598
599 599 for r in xrange(chlog.count() - 1, -1, -1):
600 600 n = chlog.node(r)
601 601 if n not in p:
602 602 h.append(n)
603 603 if n == stop:
604 604 break
605 605 if r < stoprev:
606 606 break
607 607 for pn in chlog.parents(n):
608 608 p[pn] = 1
609 609 return h
610 610
611 611 def bundle(cg):
612 612 backupdir = repo.join("strip-backup")
613 613 if not os.path.isdir(backupdir):
614 614 os.mkdir(backupdir)
615 615 name = os.path.join(backupdir, "%s" % revlog.short(rev))
616 616 name = savename(name)
617 617 self.ui.warn("saving bundle to %s\n" % name)
618 618 return changegroup.writebundle(cg, name, "HG10BZ")
619 619
620 620 def stripall(revnum):
621 621 mm = repo.changectx(rev).manifest()
622 622 seen = {}
623 623
624 624 for x in xrange(revnum, repo.changelog.count()):
625 625 for f in repo.changectx(x).files():
626 626 if f in seen:
627 627 continue
628 628 seen[f] = 1
629 629 if f in mm:
630 630 filerev = mm[f]
631 631 else:
632 632 filerev = 0
633 633 seen[f] = filerev
634 634 # we go in two steps here so the strip loop happens in a
635 635 # sensible order. When stripping many files, this helps keep
636 636 # our disk access patterns under control.
637 637 seen_list = seen.keys()
638 638 seen_list.sort()
639 639 for f in seen_list:
640 640 ff = repo.file(f)
641 641 filerev = seen[f]
642 642 if filerev != 0:
643 643 if filerev in ff.nodemap:
644 644 filerev = ff.rev(filerev)
645 645 else:
646 646 filerev = 0
647 647 ff.strip(filerev, revnum)
648 648
649 649 if not wlock:
650 650 wlock = repo.wlock()
651 651 lock = repo.lock()
652 652 chlog = repo.changelog
653 653 # TODO delete the undo files, and handle undo of merge sets
654 654 pp = chlog.parents(rev)
655 655 revnum = chlog.rev(rev)
656 656
657 657 if update:
658 658 self.check_localchanges(repo, refresh=False)
659 659 urev = self.qparents(repo, rev)
660 660 hg.clean(repo, urev, wlock=wlock)
661 661 repo.dirstate.write()
662 662
663 663 # save is a list of all the branches we are truncating away
664 664 # that we actually want to keep. changegroup will be used
665 665 # to preserve them and add them back after the truncate
666 666 saveheads = []
667 667 savebases = {}
668 668
669 669 heads = limitheads(chlog, rev)
670 670 seen = {}
671 671
672 672 # search through all the heads, finding those where the revision
673 673 # we want to strip away is an ancestor. Also look for merges
674 674 # that might be turned into new heads by the strip.
675 675 while heads:
676 676 h = heads.pop()
677 677 n = h
678 678 while True:
679 679 seen[n] = 1
680 680 pp = chlog.parents(n)
681 681 if pp[1] != revlog.nullid and chlog.rev(pp[1]) > revnum:
682 682 if pp[1] not in seen:
683 683 heads.append(pp[1])
684 684 if pp[0] == revlog.nullid:
685 685 break
686 686 if chlog.rev(pp[0]) < revnum:
687 687 break
688 688 n = pp[0]
689 689 if n == rev:
690 690 break
691 691 r = chlog.reachable(h, rev)
692 692 if rev not in r:
693 693 saveheads.append(h)
694 694 for x in r:
695 695 if chlog.rev(x) > revnum:
696 696 savebases[x] = 1
697 697
698 698 # create a changegroup for all the branches we need to keep
699 699 if backup == "all":
700 700 backupch = repo.changegroupsubset([rev], chlog.heads(), 'strip')
701 701 bundle(backupch)
702 702 if saveheads:
703 703 backupch = repo.changegroupsubset(savebases.keys(), saveheads, 'strip')
704 704 chgrpfile = bundle(backupch)
705 705
706 706 stripall(revnum)
707 707
708 708 change = chlog.read(rev)
709 709 chlog.strip(revnum, revnum)
710 710 repo.manifest.strip(repo.manifest.rev(change[0]), revnum)
711 711 if saveheads:
712 712 self.ui.status("adding branch\n")
713 713 commands.unbundle(self.ui, repo, "file:%s" % chgrpfile,
714 714 update=False)
715 715 if backup != "strip":
716 716 os.unlink(chgrpfile)
717 717
718 718 def isapplied(self, patch):
719 719 """returns (index, rev, patch)"""
720 720 for i in xrange(len(self.applied)):
721 721 a = self.applied[i]
722 722 if a.name == patch:
723 723 return (i, a.rev, a.name)
724 724 return None
725 725
726 726 # if the exact patch name does not exist, we try a few
727 727 # variations. If strict is passed, we try only #1
728 728 #
729 729 # 1) a number to indicate an offset in the series file
730 730 # 2) a unique substring of the patch name was given
731 731 # 3) patchname[-+]num to indicate an offset in the series file
732 732 def lookup(self, patch, strict=False):
733 733 patch = patch and str(patch)
734 734
735 735 def partial_name(s):
736 736 if s in self.series:
737 737 return s
738 738 matches = [x for x in self.series if s in x]
739 739 if len(matches) > 1:
740 740 self.ui.warn(_('patch name "%s" is ambiguous:\n') % s)
741 741 for m in matches:
742 742 self.ui.warn(' %s\n' % m)
743 743 return None
744 744 if matches:
745 745 return matches[0]
746 746 if len(self.series) > 0 and len(self.applied) > 0:
747 747 if s == 'qtip':
748 748 return self.series[self.series_end(True)-1]
749 749 if s == 'qbase':
750 750 return self.series[0]
751 751 return None
752 752 if patch == None:
753 753 return None
754 754
755 755 # we don't want to return a partial match until we make
756 756 # sure the file name passed in does not exist (checked below)
757 757 res = partial_name(patch)
758 758 if res and res == patch:
759 759 return res
760 760
761 761 if not os.path.isfile(self.join(patch)):
762 762 try:
763 763 sno = int(patch)
764 764 except(ValueError, OverflowError):
765 765 pass
766 766 else:
767 767 if sno < len(self.series):
768 768 return self.series[sno]
769 769 if not strict:
770 770 # return any partial match made above
771 771 if res:
772 772 return res
773 773 minus = patch.rfind('-')
774 774 if minus >= 0:
775 775 res = partial_name(patch[:minus])
776 776 if res:
777 777 i = self.series.index(res)
778 778 try:
779 779 off = int(patch[minus+1:] or 1)
780 780 except(ValueError, OverflowError):
781 781 pass
782 782 else:
783 783 if i - off >= 0:
784 784 return self.series[i - off]
785 785 plus = patch.rfind('+')
786 786 if plus >= 0:
787 787 res = partial_name(patch[:plus])
788 788 if res:
789 789 i = self.series.index(res)
790 790 try:
791 791 off = int(patch[plus+1:] or 1)
792 792 except(ValueError, OverflowError):
793 793 pass
794 794 else:
795 795 if i + off < len(self.series):
796 796 return self.series[i + off]
797 797 raise util.Abort(_("patch %s not in series") % patch)
798 798
799 799 def push(self, repo, patch=None, force=False, list=False,
800 800 mergeq=None, wlock=None):
801 801 if not wlock:
802 802 wlock = repo.wlock()
803 803 patch = self.lookup(patch)
804 804 if patch and self.isapplied(patch):
805 805 raise util.Abort(_("patch %s is already applied") % patch)
806 806 if self.series_end() == len(self.series):
807 807 raise util.Abort(_("patch series fully applied"))
808 808 if not force:
809 809 self.check_localchanges(repo)
810 810
811 811 self.applied_dirty = 1;
812 812 start = self.series_end()
813 813 if start > 0:
814 814 self.check_toppatch(repo)
815 815 if not patch:
816 816 patch = self.series[start]
817 817 end = start + 1
818 818 else:
819 819 end = self.series.index(patch, start) + 1
820 820 s = self.series[start:end]
821 821 if mergeq:
822 822 ret = self.mergepatch(repo, mergeq, s, wlock)
823 823 else:
824 824 ret = self.apply(repo, s, list, wlock=wlock)
825 825 top = self.applied[-1].name
826 826 if ret[0]:
827 827 self.ui.write("Errors during apply, please fix and refresh %s\n" %
828 828 top)
829 829 else:
830 830 self.ui.write("Now at: %s\n" % top)
831 831 return ret[0]
832 832
833 833 def pop(self, repo, patch=None, force=False, update=True, all=False,
834 834 wlock=None):
835 835 def getfile(f, rev):
836 836 t = repo.file(f).read(rev)
837 837 repo.wfile(f, "w").write(t)
838 838
839 839 if not wlock:
840 840 wlock = repo.wlock()
841 841 if patch:
842 842 # index, rev, patch
843 843 info = self.isapplied(patch)
844 844 if not info:
845 845 patch = self.lookup(patch)
846 846 info = self.isapplied(patch)
847 847 if not info:
848 848 raise util.Abort(_("patch %s is not applied") % patch)
849 849 if len(self.applied) == 0:
850 850 raise util.Abort(_("no patches applied"))
851 851
852 852 if not update:
853 853 parents = repo.dirstate.parents()
854 854 rr = [ revlog.bin(x.rev) for x in self.applied ]
855 855 for p in parents:
856 856 if p in rr:
857 857 self.ui.warn("qpop: forcing dirstate update\n")
858 858 update = True
859 859
860 860 if not force and update:
861 861 self.check_localchanges(repo)
862 862
863 863 self.applied_dirty = 1;
864 864 end = len(self.applied)
865 865 if not patch:
866 866 if all:
867 867 popi = 0
868 868 else:
869 869 popi = len(self.applied) - 1
870 870 else:
871 871 popi = info[0] + 1
872 872 if popi >= end:
873 873 self.ui.warn("qpop: %s is already at the top\n" % patch)
874 874 return
875 875 info = [ popi ] + [self.applied[popi].rev, self.applied[popi].name]
876 876
877 877 start = info[0]
878 878 rev = revlog.bin(info[1])
879 879
880 880 # we know there are no local changes, so we can make a simplified
881 881 # form of hg.update.
882 882 if update:
883 883 top = self.check_toppatch(repo)
884 884 qp = self.qparents(repo, rev)
885 885 changes = repo.changelog.read(qp)
886 886 mmap = repo.manifest.read(changes[0])
887 887 m, a, r, d, u = repo.status(qp, top)[:5]
888 888 if d:
889 889 raise util.Abort("deletions found between repo revs")
890 890 for f in m:
891 891 getfile(f, mmap[f])
892 892 for f in r:
893 893 getfile(f, mmap[f])
894 894 util.set_exec(repo.wjoin(f), mmap.execf(f))
895 895 repo.dirstate.update(m + r, 'n')
896 896 for f in a:
897 897 try:
898 898 os.unlink(repo.wjoin(f))
899 899 except OSError, e:
900 900 if e.errno != errno.ENOENT:
901 901 raise
902 902 try: os.removedirs(os.path.dirname(repo.wjoin(f)))
903 903 except: pass
904 904 if a:
905 905 repo.dirstate.forget(a)
906 906 repo.dirstate.setparents(qp, revlog.nullid)
907 907 self.strip(repo, rev, update=False, backup='strip', wlock=wlock)
908 908 del self.applied[start:end]
909 909 if len(self.applied):
910 910 self.ui.write("Now at: %s\n" % self.applied[-1].name)
911 911 else:
912 912 self.ui.write("Patch queue now empty\n")
913 913
914 914 def diff(self, repo, pats, opts):
915 915 top = self.check_toppatch(repo)
916 916 if not top:
917 917 self.ui.write("No patches applied\n")
918 918 return
919 919 qp = self.qparents(repo, top)
920 920 if opts.get('git'):
921 921 self.diffopts().git = True
922 922 self.printdiff(repo, qp, files=pats, opts=opts)
923 923
924 924 def refresh(self, repo, pats=None, **opts):
925 925 if len(self.applied) == 0:
926 926 self.ui.write("No patches applied\n")
927 927 return 1
928 928 wlock = repo.wlock()
929 929 self.check_toppatch(repo)
930 930 (top, patchfn) = (self.applied[-1].rev, self.applied[-1].name)
931 931 top = revlog.bin(top)
932 932 cparents = repo.changelog.parents(top)
933 933 patchparent = self.qparents(repo, top)
934 934 message, comments, user, date, patchfound = self.readheaders(patchfn)
935 935
936 936 patchf = self.opener(patchfn, "w")
937 937 msg = opts.get('msg', '').rstrip()
938 938 if msg:
939 939 if comments:
940 940 # Remove existing message.
941 941 ci = 0
942 942 for mi in xrange(len(message)):
943 943 while message[mi] != comments[ci]:
944 944 ci += 1
945 945 del comments[ci]
946 946 comments.append(msg)
947 947 if comments:
948 948 comments = "\n".join(comments) + '\n\n'
949 949 patchf.write(comments)
950 950
951 951 if opts.get('git'):
952 952 self.diffopts().git = True
953 953 fns, matchfn, anypats = cmdutil.matchpats(repo, pats, opts)
954 954 tip = repo.changelog.tip()
955 955 if top == tip:
956 956 # if the top of our patch queue is also the tip, there is an
957 957 # optimization here. We update the dirstate in place and strip
958 958 # off the tip commit. Then just commit the current directory
959 959 # tree. We can also send repo.commit the list of files
960 960 # changed to speed up the diff
961 961 #
962 962 # in short mode, we only diff the files included in the
963 963 # patch already
964 964 #
965 965 # this should really read:
966 966 # mm, dd, aa, aa2, uu = repo.status(tip, patchparent)[:5]
967 967 # but we do it backwards to take advantage of manifest/chlog
968 968 # caching against the next repo.status call
969 969 #
970 970 mm, aa, dd, aa2, uu = repo.status(patchparent, tip)[:5]
971 971 changes = repo.changelog.read(tip)
972 972 man = repo.manifest.read(changes[0])
973 973 aaa = aa[:]
974 974 if opts.get('short'):
975 975 filelist = mm + aa + dd
976 976 else:
977 977 filelist = None
978 978 m, a, r, d, u = repo.status(files=filelist)[:5]
979 979
980 980 # we might end up with files that were added between tip and
981 981 # the dirstate parent, but then changed in the local dirstate.
982 982 # in this case, we want them to only show up in the added section
983 983 for x in m:
984 984 if x not in aa:
985 985 mm.append(x)
986 986 # we might end up with files added by the local dirstate that
987 987 # were deleted by the patch. In this case, they should only
988 988 # show up in the changed section.
989 989 for x in a:
990 990 if x in dd:
991 991 del dd[dd.index(x)]
992 992 mm.append(x)
993 993 else:
994 994 aa.append(x)
995 995 # make sure any files deleted in the local dirstate
996 996 # are not in the add or change column of the patch
997 997 forget = []
998 998 for x in d + r:
999 999 if x in aa:
1000 1000 del aa[aa.index(x)]
1001 1001 forget.append(x)
1002 1002 continue
1003 1003 elif x in mm:
1004 1004 del mm[mm.index(x)]
1005 1005 dd.append(x)
1006 1006
1007 1007 m = util.unique(mm)
1008 1008 r = util.unique(dd)
1009 1009 a = util.unique(aa)
1010 1010 filelist = filter(matchfn, util.unique(m + r + a))
1011 1011 patch.diff(repo, patchparent, files=filelist, match=matchfn,
1012 1012 fp=patchf, changes=(m, a, r, [], u),
1013 1013 opts=self.diffopts())
1014 1014 patchf.close()
1015 1015
1016 1016 repo.dirstate.setparents(*cparents)
1017 1017 copies = {}
1018 1018 for dst in a:
1019 1019 src = repo.dirstate.copied(dst)
1020 1020 if src is None:
1021 1021 continue
1022 1022 copies.setdefault(src, []).append(dst)
1023 1023 repo.dirstate.update(a, 'a')
1024 1024 # remember the copies between patchparent and tip
1025 1025 # this may be slow, so don't do it if we're not tracking copies
1026 1026 if self.diffopts().git:
1027 1027 for dst in aaa:
1028 1028 f = repo.file(dst)
1029 1029 src = f.renamed(man[dst])
1030 1030 if src:
1031 1031 copies[src[0]] = copies.get(dst, [])
1032 1032 if dst in a:
1033 1033 copies[src[0]].append(dst)
1034 1034 # we can't copy a file created by the patch itself
1035 1035 if dst in copies:
1036 1036 del copies[dst]
1037 1037 for src, dsts in copies.iteritems():
1038 1038 for dst in dsts:
1039 1039 repo.dirstate.copy(src, dst)
1040 1040 repo.dirstate.update(r, 'r')
1041 1041 # if the patch excludes a modified file, mark that file with mtime=0
1042 1042 # so status can see it.
1043 1043 mm = []
1044 1044 for i in xrange(len(m)-1, -1, -1):
1045 1045 if not matchfn(m[i]):
1046 1046 mm.append(m[i])
1047 1047 del m[i]
1048 1048 repo.dirstate.update(m, 'n')
1049 1049 repo.dirstate.update(mm, 'n', st_mtime=0)
1050 1050 repo.dirstate.forget(forget)
1051 1051
1052 1052 if not msg:
1053 1053 if not message:
1054 1054 message = "patch queue: %s\n" % patchfn
1055 1055 else:
1056 1056 message = "\n".join(message)
1057 1057 else:
1058 1058 message = msg
1059 1059
1060 1060 self.strip(repo, top, update=False, backup='strip', wlock=wlock)
1061 1061 n = repo.commit(filelist, message, changes[1], force=1, wlock=wlock)
1062 1062 self.applied[-1] = statusentry(revlog.hex(n), patchfn)
1063 1063 self.applied_dirty = 1
1064 1064 else:
1065 1065 self.printdiff(repo, patchparent, fp=patchf)
1066 1066 patchf.close()
1067 1067 added = repo.status()[1]
1068 1068 for a in added:
1069 1069 f = repo.wjoin(a)
1070 1070 try:
1071 1071 os.unlink(f)
1072 1072 except OSError, e:
1073 1073 if e.errno != errno.ENOENT:
1074 1074 raise
1075 1075 try: os.removedirs(os.path.dirname(f))
1076 1076 except: pass
1077 1077 # forget the file copies in the dirstate
1078 1078 # push should readd the files later on
1079 1079 repo.dirstate.forget(added)
1080 1080 self.pop(repo, force=True, wlock=wlock)
1081 1081 self.push(repo, force=True, wlock=wlock)
1082 1082
1083 1083 def init(self, repo, create=False):
1084 1084 if os.path.isdir(self.path):
1085 1085 raise util.Abort(_("patch queue directory already exists"))
1086 1086 os.mkdir(self.path)
1087 1087 if create:
1088 1088 return self.qrepo(create=True)
1089 1089
1090 1090 def unapplied(self, repo, patch=None):
1091 1091 if patch and patch not in self.series:
1092 1092 raise util.Abort(_("patch %s is not in series file") % patch)
1093 1093 if not patch:
1094 1094 start = self.series_end()
1095 1095 else:
1096 1096 start = self.series.index(patch) + 1
1097 1097 unapplied = []
1098 1098 for i in xrange(start, len(self.series)):
1099 1099 pushable, reason = self.pushable(i)
1100 1100 if pushable:
1101 1101 unapplied.append((i, self.series[i]))
1102 1102 self.explain_pushable(i)
1103 1103 return unapplied
1104 1104
1105 1105 def qseries(self, repo, missing=None, start=0, length=0, status=None,
1106 1106 summary=False):
1107 1107 def displayname(patchname):
1108 1108 if summary:
1109 1109 msg = self.readheaders(patchname)[0]
1110 1110 msg = msg and ': ' + msg[0] or ': '
1111 1111 else:
1112 1112 msg = ''
1113 1113 return '%s%s' % (patchname, msg)
1114 1114
1115 1115 def pname(i):
1116 1116 if status == 'A':
1117 1117 return self.applied[i].name
1118 1118 else:
1119 1119 return self.series[i]
1120 1120
1121 1121 applied = dict.fromkeys([p.name for p in self.applied])
1122 1122 if not length:
1123 1123 length = len(self.series) - start
1124 1124 if not missing:
1125 1125 for i in xrange(start, start+length):
1126 1126 pfx = ''
1127 1127 patch = pname(i)
1128 1128 if self.ui.verbose:
1129 1129 if patch in applied:
1130 1130 stat = 'A'
1131 1131 elif self.pushable(i)[0]:
1132 1132 stat = 'U'
1133 1133 else:
1134 1134 stat = 'G'
1135 1135 pfx = '%d %s ' % (i, stat)
1136 1136 self.ui.write('%s%s\n' % (pfx, displayname(patch)))
1137 1137 else:
1138 1138 msng_list = []
1139 1139 for root, dirs, files in os.walk(self.path):
1140 1140 d = root[len(self.path) + 1:]
1141 1141 for f in files:
1142 1142 fl = os.path.join(d, f)
1143 1143 if (fl not in self.series and
1144 1144 fl not in (self.status_path, self.series_path)
1145 1145 and not fl.startswith('.')):
1146 1146 msng_list.append(fl)
1147 1147 msng_list.sort()
1148 1148 for x in msng_list:
1149 1149 pfx = self.ui.verbose and ('D ') or ''
1150 1150 self.ui.write("%s%s\n" % (pfx, displayname(x)))
1151 1151
1152 1152 def issaveline(self, l):
1153 1153 if l.name == '.hg.patches.save.line':
1154 1154 return True
1155 1155
1156 1156 def qrepo(self, create=False):
1157 1157 if create or os.path.isdir(self.join(".hg")):
1158 1158 return hg.repository(self.ui, path=self.path, create=create)
1159 1159
1160 1160 def restore(self, repo, rev, delete=None, qupdate=None):
1161 1161 c = repo.changelog.read(rev)
1162 1162 desc = c[4].strip()
1163 1163 lines = desc.splitlines()
1164 1164 i = 0
1165 1165 datastart = None
1166 1166 series = []
1167 1167 applied = []
1168 1168 qpp = None
1169 1169 for i in xrange(0, len(lines)):
1170 1170 if lines[i] == 'Patch Data:':
1171 1171 datastart = i + 1
1172 1172 elif lines[i].startswith('Dirstate:'):
1173 1173 l = lines[i].rstrip()
1174 1174 l = l[10:].split(' ')
1175 1175 qpp = [ hg.bin(x) for x in l ]
1176 1176 elif datastart != None:
1177 1177 l = lines[i].rstrip()
1178 1178 se = statusentry(l)
1179 1179 file_ = se.name
1180 1180 if se.rev:
1181 1181 applied.append(se)
1182 1182 else:
1183 1183 series.append(file_)
1184 1184 if datastart == None:
1185 1185 self.ui.warn("No saved patch data found\n")
1186 1186 return 1
1187 1187 self.ui.warn("restoring status: %s\n" % lines[0])
1188 1188 self.full_series = series
1189 1189 self.applied = applied
1190 1190 self.parse_series()
1191 1191 self.series_dirty = 1
1192 1192 self.applied_dirty = 1
1193 1193 heads = repo.changelog.heads()
1194 1194 if delete:
1195 1195 if rev not in heads:
1196 1196 self.ui.warn("save entry has children, leaving it alone\n")
1197 1197 else:
1198 1198 self.ui.warn("removing save entry %s\n" % hg.short(rev))
1199 1199 pp = repo.dirstate.parents()
1200 1200 if rev in pp:
1201 1201 update = True
1202 1202 else:
1203 1203 update = False
1204 1204 self.strip(repo, rev, update=update, backup='strip')
1205 1205 if qpp:
1206 1206 self.ui.warn("saved queue repository parents: %s %s\n" %
1207 1207 (hg.short(qpp[0]), hg.short(qpp[1])))
1208 1208 if qupdate:
1209 1209 print "queue directory updating"
1210 1210 r = self.qrepo()
1211 1211 if not r:
1212 1212 self.ui.warn("Unable to load queue repository\n")
1213 1213 return 1
1214 1214 hg.clean(r, qpp[0])
1215 1215
1216 1216 def save(self, repo, msg=None):
1217 1217 if len(self.applied) == 0:
1218 1218 self.ui.warn("save: no patches applied, exiting\n")
1219 1219 return 1
1220 1220 if self.issaveline(self.applied[-1]):
1221 1221 self.ui.warn("status is already saved\n")
1222 1222 return 1
1223 1223
1224 1224 ar = [ ':' + x for x in self.full_series ]
1225 1225 if not msg:
1226 1226 msg = "hg patches saved state"
1227 1227 else:
1228 1228 msg = "hg patches: " + msg.rstrip('\r\n')
1229 1229 r = self.qrepo()
1230 1230 if r:
1231 1231 pp = r.dirstate.parents()
1232 1232 msg += "\nDirstate: %s %s" % (hg.hex(pp[0]), hg.hex(pp[1]))
1233 1233 msg += "\n\nPatch Data:\n"
1234 1234 text = msg + "\n".join([str(x) for x in self.applied]) + '\n' + (ar and
1235 1235 "\n".join(ar) + '\n' or "")
1236 1236 n = repo.commit(None, text, user=None, force=1)
1237 1237 if not n:
1238 1238 self.ui.warn("repo commit failed\n")
1239 1239 return 1
1240 1240 self.applied.append(statusentry(revlog.hex(n),'.hg.patches.save.line'))
1241 1241 self.applied_dirty = 1
1242 1242
1243 1243 def full_series_end(self):
1244 1244 if len(self.applied) > 0:
1245 1245 p = self.applied[-1].name
1246 1246 end = self.find_series(p)
1247 1247 if end == None:
1248 1248 return len(self.full_series)
1249 1249 return end + 1
1250 1250 return 0
1251 1251
1252 1252 def series_end(self, all_patches=False):
1253 1253 end = 0
1254 1254 def next(start):
1255 1255 if all_patches:
1256 1256 return start
1257 1257 i = start
1258 1258 while i < len(self.series):
1259 1259 p, reason = self.pushable(i)
1260 1260 if p:
1261 1261 break
1262 1262 self.explain_pushable(i)
1263 1263 i += 1
1264 1264 return i
1265 1265 if len(self.applied) > 0:
1266 1266 p = self.applied[-1].name
1267 1267 try:
1268 1268 end = self.series.index(p)
1269 1269 except ValueError:
1270 1270 return 0
1271 1271 return next(end + 1)
1272 1272 return next(end)
1273 1273
1274 1274 def appliedname(self, index):
1275 1275 pname = self.applied[index].name
1276 1276 if not self.ui.verbose:
1277 1277 p = pname
1278 1278 else:
1279 1279 p = str(self.series.index(pname)) + " " + pname
1280 1280 return p
1281 1281
1282 1282 def qimport(self, repo, files, patchname=None, rev=None, existing=None,
1283 1283 force=None, git=False):
1284 1284 def checkseries(patchname):
1285 1285 if patchname in self.series:
1286 1286 raise util.Abort(_('patch %s is already in the series file')
1287 1287 % patchname)
1288 1288 def checkfile(patchname):
1289 1289 if not force and os.path.exists(self.join(patchname)):
1290 1290 raise util.Abort(_('patch "%s" already exists')
1291 1291 % patchname)
1292 1292
1293 1293 if rev:
1294 1294 if files:
1295 1295 raise util.Abort(_('option "-r" not valid when importing '
1296 1296 'files'))
1297 1297 rev = cmdutil.revrange(repo, rev)
1298 1298 rev.sort(lambda x, y: cmp(y, x))
1299 1299 if (len(files) > 1 or len(rev) > 1) and patchname:
1300 1300 raise util.Abort(_('option "-n" not valid when importing multiple '
1301 1301 'patches'))
1302 1302 i = 0
1303 1303 added = []
1304 1304 if rev:
1305 1305 # If mq patches are applied, we can only import revisions
1306 1306 # that form a linear path to qbase.
1307 1307 # Otherwise, they should form a linear path to a head.
1308 1308 heads = repo.changelog.heads(repo.changelog.node(rev[-1]))
1309 1309 if len(heads) > 1:
1310 1310 raise util.Abort(_('revision %d is the root of more than one '
1311 1311 'branch') % rev[-1])
1312 1312 if self.applied:
1313 1313 base = revlog.hex(repo.changelog.node(rev[0]))
1314 1314 if base in [n.rev for n in self.applied]:
1315 1315 raise util.Abort(_('revision %d is already managed')
1316 1316 % rev[0])
1317 1317 if heads != [revlog.bin(self.applied[-1].rev)]:
1318 1318 raise util.Abort(_('revision %d is not the parent of '
1319 1319 'the queue') % rev[0])
1320 1320 base = repo.changelog.rev(revlog.bin(self.applied[0].rev))
1321 1321 lastparent = repo.changelog.parentrevs(base)[0]
1322 1322 else:
1323 1323 if heads != [repo.changelog.node(rev[0])]:
1324 1324 raise util.Abort(_('revision %d has unmanaged children')
1325 1325 % rev[0])
1326 1326 lastparent = None
1327 1327
1328 1328 if git:
1329 1329 self.diffopts().git = True
1330 1330
1331 1331 for r in rev:
1332 1332 p1, p2 = repo.changelog.parentrevs(r)
1333 1333 n = repo.changelog.node(r)
1334 1334 if p2 != revlog.nullrev:
1335 1335 raise util.Abort(_('cannot import merge revision %d') % r)
1336 1336 if lastparent and lastparent != r:
1337 1337 raise util.Abort(_('revision %d is not the parent of %d')
1338 1338 % (r, lastparent))
1339 1339 lastparent = p1
1340 1340
1341 1341 if not patchname:
1342 1342 patchname = normname('%d.diff' % r)
1343 1343 checkseries(patchname)
1344 1344 checkfile(patchname)
1345 1345 self.full_series.insert(0, patchname)
1346 1346
1347 1347 patchf = self.opener(patchname, "w")
1348 1348 patch.export(repo, [n], fp=patchf, opts=self.diffopts())
1349 1349 patchf.close()
1350 1350
1351 1351 se = statusentry(revlog.hex(n), patchname)
1352 1352 self.applied.insert(0, se)
1353 1353
1354 1354 added.append(patchname)
1355 1355 patchname = None
1356 1356 self.parse_series()
1357 1357 self.applied_dirty = 1
1358 1358
1359 1359 for filename in files:
1360 1360 if existing:
1361 1361 if filename == '-':
1362 1362 raise util.Abort(_('-e is incompatible with import from -'))
1363 1363 if not patchname:
1364 1364 patchname = normname(filename)
1365 1365 if not os.path.isfile(self.join(patchname)):
1366 1366 raise util.Abort(_("patch %s does not exist") % patchname)
1367 1367 else:
1368 1368 try:
1369 1369 if filename == '-':
1370 1370 if not patchname:
1371 1371 raise util.Abort(_('need --name to import a patch from -'))
1372 1372 text = sys.stdin.read()
1373 1373 else:
1374 1374 text = file(filename).read()
1375 1375 except IOError:
1376 1376 raise util.Abort(_("unable to read %s") % patchname)
1377 1377 if not patchname:
1378 1378 patchname = normname(os.path.basename(filename))
1379 1379 checkfile(patchname)
1380 1380 patchf = self.opener(patchname, "w")
1381 1381 patchf.write(text)
1382 1382 checkseries(patchname)
1383 1383 index = self.full_series_end() + i
1384 1384 self.full_series[index:index] = [patchname]
1385 1385 self.parse_series()
1386 1386 self.ui.warn("adding %s to series file\n" % patchname)
1387 1387 i += 1
1388 1388 added.append(patchname)
1389 1389 patchname = None
1390 1390 self.series_dirty = 1
1391 1391 qrepo = self.qrepo()
1392 1392 if qrepo:
1393 1393 qrepo.add(added)
1394 1394
1395 1395 def delete(ui, repo, *patches, **opts):
1396 1396 """remove patches from queue
1397 1397
1398 1398 With --rev, mq will stop managing the named revisions. The
1399 1399 patches must be applied and at the base of the stack. This option
1400 1400 is useful when the patches have been applied upstream.
1401 1401
1402 1402 Otherwise, the patches must not be applied.
1403 1403
1404 1404 With --keep, the patch files are preserved in the patch directory."""
1405 1405 q = repo.mq
1406 1406 q.delete(repo, patches, opts)
1407 1407 q.save_dirty()
1408 1408 return 0
1409 1409
1410 1410 def applied(ui, repo, patch=None, **opts):
1411 1411 """print the patches already applied"""
1412 1412 q = repo.mq
1413 1413 if patch:
1414 1414 if patch not in q.series:
1415 1415 raise util.Abort(_("patch %s is not in series file") % patch)
1416 1416 end = q.series.index(patch) + 1
1417 1417 else:
1418 1418 end = len(q.applied)
1419 1419 if not end:
1420 1420 return
1421 1421
1422 1422 return q.qseries(repo, length=end, status='A', summary=opts.get('summary'))
1423 1423
1424 1424 def unapplied(ui, repo, patch=None, **opts):
1425 1425 """print the patches not yet applied"""
1426 1426 q = repo.mq
1427 1427 if patch:
1428 1428 if patch not in q.series:
1429 1429 raise util.Abort(_("patch %s is not in series file") % patch)
1430 1430 start = q.series.index(patch) + 1
1431 1431 else:
1432 1432 start = q.series_end()
1433 1433 q.qseries(repo, start=start, summary=opts.get('summary'))
1434 1434
1435 1435 def qimport(ui, repo, *filename, **opts):
1436 1436 """import a patch
1437 1437
1438 1438 The patch will have the same name as its source file unless you
1439 1439 give it a new one with --name.
1440 1440
1441 1441 You can register an existing patch inside the patch directory
1442 1442 with the --existing flag.
1443 1443
1444 1444 With --force, an existing patch of the same name will be overwritten.
1445 1445
1446 1446 An existing changeset may be placed under mq control with --rev
1447 1447 (e.g. qimport --rev tip -n patch will place tip under mq control).
1448 1448 With --git, patches imported with --rev will use the git diff
1449 1449 format.
1450 1450 """
1451 1451 q = repo.mq
1452 1452 q.qimport(repo, filename, patchname=opts['name'],
1453 1453 existing=opts['existing'], force=opts['force'], rev=opts['rev'],
1454 1454 git=opts['git'])
1455 1455 q.save_dirty()
1456 1456 return 0
1457 1457
1458 1458 def init(ui, repo, **opts):
1459 1459 """init a new queue repository
1460 1460
1461 1461 The queue repository is unversioned by default. If -c is
1462 1462 specified, qinit will create a separate nested repository
1463 1463 for patches. Use qcommit to commit changes to this queue
1464 1464 repository."""
1465 1465 q = repo.mq
1466 1466 r = q.init(repo, create=opts['create_repo'])
1467 1467 q.save_dirty()
1468 1468 if r:
1469 1469 fp = r.wopener('.hgignore', 'w')
1470 1470 print >> fp, 'syntax: glob'
1471 1471 print >> fp, 'status'
1472 1472 print >> fp, 'guards'
1473 1473 fp.close()
1474 1474 r.wopener('series', 'w').close()
1475 1475 r.add(['.hgignore', 'series'])
1476 1476 return 0
1477 1477
1478 1478 def clone(ui, source, dest=None, **opts):
1479 1479 '''clone main and patch repository at same time
1480 1480
1481 1481 If source is local, destination will have no patches applied. If
1482 1482 source is remote, this command can not check if patches are
1483 1483 applied in source, so cannot guarantee that patches are not
1484 1484 applied in destination. If you clone remote repository, be sure
1485 1485 before that it has no patches applied.
1486 1486
1487 1487 Source patch repository is looked for in <src>/.hg/patches by
1488 1488 default. Use -p <url> to change.
1489 1489 '''
1490 1490 commands.setremoteconfig(ui, opts)
1491 1491 if dest is None:
1492 1492 dest = hg.defaultdest(source)
1493 1493 sr = hg.repository(ui, ui.expandpath(source))
1494 1494 qbase, destrev = None, None
1495 1495 if sr.local():
1496 1496 reposetup(ui, sr)
1497 1497 if sr.mq.applied:
1498 1498 qbase = revlog.bin(sr.mq.applied[0].rev)
1499 1499 if not hg.islocal(dest):
1500 1500 destrev = sr.parents(qbase)[0]
1501 1501 ui.note(_('cloning main repo\n'))
1502 1502 sr, dr = hg.clone(ui, sr, dest,
1503 1503 pull=opts['pull'],
1504 1504 rev=destrev,
1505 1505 update=False,
1506 1506 stream=opts['uncompressed'])
1507 1507 ui.note(_('cloning patch repo\n'))
1508 1508 spr, dpr = hg.clone(ui, opts['patches'] or (sr.url() + '/.hg/patches'),
1509 1509 dr.url() + '/.hg/patches',
1510 1510 pull=opts['pull'],
1511 1511 update=not opts['noupdate'],
1512 1512 stream=opts['uncompressed'])
1513 1513 if dr.local():
1514 1514 if qbase:
1515 1515 ui.note(_('stripping applied patches from destination repo\n'))
1516 1516 reposetup(ui, dr)
1517 1517 dr.mq.strip(dr, qbase, update=False, backup=None)
1518 1518 if not opts['noupdate']:
1519 1519 ui.note(_('updating destination repo\n'))
1520 1520 hg.update(dr, dr.changelog.tip())
1521 1521
1522 1522 def commit(ui, repo, *pats, **opts):
1523 1523 """commit changes in the queue repository"""
1524 1524 q = repo.mq
1525 1525 r = q.qrepo()
1526 1526 if not r: raise util.Abort('no queue repository')
1527 1527 commands.commit(r.ui, r, *pats, **opts)
1528 1528
1529 1529 def series(ui, repo, **opts):
1530 1530 """print the entire series file"""
1531 1531 repo.mq.qseries(repo, missing=opts['missing'], summary=opts['summary'])
1532 1532 return 0
1533 1533
1534 1534 def top(ui, repo, **opts):
1535 1535 """print the name of the current patch"""
1536 1536 q = repo.mq
1537 1537 t = len(q.applied)
1538 1538 if t:
1539 1539 return q.qseries(repo, start=t-1, length=1, status='A',
1540 1540 summary=opts.get('summary'))
1541 1541 else:
1542 1542 ui.write("No patches applied\n")
1543 1543 return 1
1544 1544
1545 1545 def next(ui, repo, **opts):
1546 1546 """print the name of the next patch"""
1547 1547 q = repo.mq
1548 1548 end = q.series_end()
1549 1549 if end == len(q.series):
1550 1550 ui.write("All patches applied\n")
1551 1551 return 1
1552 1552 return q.qseries(repo, start=end, length=1, summary=opts.get('summary'))
1553 1553
1554 1554 def prev(ui, repo, **opts):
1555 1555 """print the name of the previous patch"""
1556 1556 q = repo.mq
1557 1557 l = len(q.applied)
1558 1558 if l == 1:
1559 1559 ui.write("Only one patch applied\n")
1560 1560 return 1
1561 1561 if not l:
1562 1562 ui.write("No patches applied\n")
1563 1563 return 1
1564 1564 return q.qseries(repo, start=l-2, length=1, status='A',
1565 1565 summary=opts.get('summary'))
1566 1566
1567 1567 def new(ui, repo, patch, **opts):
1568 1568 """create a new patch
1569 1569
1570 1570 qnew creates a new patch on top of the currently-applied patch
1571 1571 (if any). It will refuse to run if there are any outstanding
1572 1572 changes unless -f is specified, in which case the patch will
1573 1573 be initialised with them.
1574 1574
1575 1575 -e, -m or -l set the patch header as well as the commit message.
1576 1576 If none is specified, the patch header is empty and the
1577 1577 commit message is 'New patch: PATCH'"""
1578 1578 q = repo.mq
1579 1579 message = commands.logmessage(opts)
1580 1580 if opts['edit']:
1581 1581 message = ui.edit(message, ui.username())
1582 1582 q.new(repo, patch, msg=message, force=opts['force'])
1583 1583 q.save_dirty()
1584 1584 return 0
1585 1585
1586 1586 def refresh(ui, repo, *pats, **opts):
1587 1587 """update the current patch
1588 1588
1589 1589 If any file patterns are provided, the refreshed patch will contain only
1590 1590 the modifications that match those patterns; the remaining modifications
1591 1591 will remain in the working directory.
1592
1593 hg add/remove/copy/rename work as usual, though you might want to use
1594 git-style patches (--git or [diff] git=1) to track copies and renames.
1592 1595 """
1593 1596 q = repo.mq
1594 1597 message = commands.logmessage(opts)
1595 1598 if opts['edit']:
1596 1599 if message:
1597 1600 raise util.Abort(_('option "-e" incompatible with "-m" or "-l"'))
1598 1601 patch = q.applied[-1].name
1599 1602 (message, comment, user, date, hasdiff) = q.readheaders(patch)
1600 1603 message = ui.edit('\n'.join(message), user or ui.username())
1601 1604 ret = q.refresh(repo, pats, msg=message, **opts)
1602 1605 q.save_dirty()
1603 1606 return ret
1604 1607
1605 1608 def diff(ui, repo, *pats, **opts):
1606 1609 """diff of the current patch"""
1607 1610 repo.mq.diff(repo, pats, opts)
1608 1611 return 0
1609 1612
1610 1613 def fold(ui, repo, *files, **opts):
1611 1614 """fold the named patches into the current patch
1612 1615
1613 1616 Patches must not yet be applied. Each patch will be successively
1614 1617 applied to the current patch in the order given. If all the
1615 1618 patches apply successfully, the current patch will be refreshed
1616 1619 with the new cumulative patch, and the folded patches will
1617 1620 be deleted. With -k/--keep, the folded patch files will not
1618 1621 be removed afterwards.
1619 1622
1620 1623 The header for each folded patch will be concatenated with
1621 1624 the current patch header, separated by a line of '* * *'."""
1622 1625
1623 1626 q = repo.mq
1624 1627
1625 1628 if not files:
1626 1629 raise util.Abort(_('qfold requires at least one patch name'))
1627 1630 if not q.check_toppatch(repo):
1628 1631 raise util.Abort(_('No patches applied'))
1629 1632
1630 1633 message = commands.logmessage(opts)
1631 1634 if opts['edit']:
1632 1635 if message:
1633 1636 raise util.Abort(_('option "-e" incompatible with "-m" or "-l"'))
1634 1637
1635 1638 parent = q.lookup('qtip')
1636 1639 patches = []
1637 1640 messages = []
1638 1641 for f in files:
1639 1642 p = q.lookup(f)
1640 1643 if p in patches or p == parent:
1641 1644 ui.warn(_('Skipping already folded patch %s') % p)
1642 1645 if q.isapplied(p):
1643 1646 raise util.Abort(_('qfold cannot fold already applied patch %s') % p)
1644 1647 patches.append(p)
1645 1648
1646 1649 for p in patches:
1647 1650 if not message:
1648 1651 messages.append(q.readheaders(p)[0])
1649 1652 pf = q.join(p)
1650 1653 (patchsuccess, files, fuzz) = q.patch(repo, pf)
1651 1654 if not patchsuccess:
1652 1655 raise util.Abort(_('Error folding patch %s') % p)
1653 1656 patch.updatedir(ui, repo, files)
1654 1657
1655 1658 if not message:
1656 1659 message, comments, user = q.readheaders(parent)[0:3]
1657 1660 for msg in messages:
1658 1661 message.append('* * *')
1659 1662 message.extend(msg)
1660 1663 message = '\n'.join(message)
1661 1664
1662 1665 if opts['edit']:
1663 1666 message = ui.edit(message, user or ui.username())
1664 1667
1665 1668 q.refresh(repo, msg=message)
1666 1669 q.delete(repo, patches, opts)
1667 1670 q.save_dirty()
1668 1671
1669 1672 def guard(ui, repo, *args, **opts):
1670 1673 '''set or print guards for a patch
1671 1674
1672 1675 Guards control whether a patch can be pushed. A patch with no
1673 1676 guards is always pushed. A patch with a positive guard ("+foo") is
1674 1677 pushed only if the qselect command has activated it. A patch with
1675 1678 a negative guard ("-foo") is never pushed if the qselect command
1676 1679 has activated it.
1677 1680
1678 1681 With no arguments, print the currently active guards.
1679 1682 With arguments, set guards for the named patch.
1680 1683
1681 1684 To set a negative guard "-foo" on topmost patch ("--" is needed so
1682 1685 hg will not interpret "-foo" as an option):
1683 1686 hg qguard -- -foo
1684 1687
1685 1688 To set guards on another patch:
1686 1689 hg qguard other.patch +2.6.17 -stable
1687 1690 '''
1688 1691 def status(idx):
1689 1692 guards = q.series_guards[idx] or ['unguarded']
1690 1693 ui.write('%s: %s\n' % (q.series[idx], ' '.join(guards)))
1691 1694 q = repo.mq
1692 1695 patch = None
1693 1696 args = list(args)
1694 1697 if opts['list']:
1695 1698 if args or opts['none']:
1696 1699 raise util.Abort(_('cannot mix -l/--list with options or arguments'))
1697 1700 for i in xrange(len(q.series)):
1698 1701 status(i)
1699 1702 return
1700 1703 if not args or args[0][0:1] in '-+':
1701 1704 if not q.applied:
1702 1705 raise util.Abort(_('no patches applied'))
1703 1706 patch = q.applied[-1].name
1704 1707 if patch is None and args[0][0:1] not in '-+':
1705 1708 patch = args.pop(0)
1706 1709 if patch is None:
1707 1710 raise util.Abort(_('no patch to work with'))
1708 1711 if args or opts['none']:
1709 1712 q.set_guards(q.find_series(patch), args)
1710 1713 q.save_dirty()
1711 1714 else:
1712 1715 status(q.series.index(q.lookup(patch)))
1713 1716
1714 1717 def header(ui, repo, patch=None):
1715 1718 """Print the header of the topmost or specified patch"""
1716 1719 q = repo.mq
1717 1720
1718 1721 if patch:
1719 1722 patch = q.lookup(patch)
1720 1723 else:
1721 1724 if not q.applied:
1722 1725 ui.write('No patches applied\n')
1723 1726 return 1
1724 1727 patch = q.lookup('qtip')
1725 1728 message = repo.mq.readheaders(patch)[0]
1726 1729
1727 1730 ui.write('\n'.join(message) + '\n')
1728 1731
1729 1732 def lastsavename(path):
1730 1733 (directory, base) = os.path.split(path)
1731 1734 names = os.listdir(directory)
1732 1735 namere = re.compile("%s.([0-9]+)" % base)
1733 1736 maxindex = None
1734 1737 maxname = None
1735 1738 for f in names:
1736 1739 m = namere.match(f)
1737 1740 if m:
1738 1741 index = int(m.group(1))
1739 1742 if maxindex == None or index > maxindex:
1740 1743 maxindex = index
1741 1744 maxname = f
1742 1745 if maxname:
1743 1746 return (os.path.join(directory, maxname), maxindex)
1744 1747 return (None, None)
1745 1748
1746 1749 def savename(path):
1747 1750 (last, index) = lastsavename(path)
1748 1751 if last is None:
1749 1752 index = 0
1750 1753 newpath = path + ".%d" % (index + 1)
1751 1754 return newpath
1752 1755
1753 1756 def push(ui, repo, patch=None, **opts):
1754 1757 """push the next patch onto the stack"""
1755 1758 q = repo.mq
1756 1759 mergeq = None
1757 1760
1758 1761 if opts['all']:
1759 1762 if not q.series:
1760 1763 raise util.Abort(_('no patches in series'))
1761 1764 patch = q.series[-1]
1762 1765 if opts['merge']:
1763 1766 if opts['name']:
1764 1767 newpath = opts['name']
1765 1768 else:
1766 1769 newpath, i = lastsavename(q.path)
1767 1770 if not newpath:
1768 1771 ui.warn("no saved queues found, please use -n\n")
1769 1772 return 1
1770 1773 mergeq = queue(ui, repo.join(""), newpath)
1771 1774 ui.warn("merging with queue at: %s\n" % mergeq.path)
1772 1775 ret = q.push(repo, patch, force=opts['force'], list=opts['list'],
1773 1776 mergeq=mergeq)
1774 1777 q.save_dirty()
1775 1778 return ret
1776 1779
1777 1780 def pop(ui, repo, patch=None, **opts):
1778 1781 """pop the current patch off the stack"""
1779 1782 localupdate = True
1780 1783 if opts['name']:
1781 1784 q = queue(ui, repo.join(""), repo.join(opts['name']))
1782 1785 ui.warn('using patch queue: %s\n' % q.path)
1783 1786 localupdate = False
1784 1787 else:
1785 1788 q = repo.mq
1786 1789 q.pop(repo, patch, force=opts['force'], update=localupdate, all=opts['all'])
1787 1790 q.save_dirty()
1788 1791 return 0
1789 1792
1790 1793 def rename(ui, repo, patch, name=None, **opts):
1791 1794 """rename a patch
1792 1795
1793 1796 With one argument, renames the current patch to PATCH1.
1794 1797 With two arguments, renames PATCH1 to PATCH2."""
1795 1798
1796 1799 q = repo.mq
1797 1800
1798 1801 if not name:
1799 1802 name = patch
1800 1803 patch = None
1801 1804
1802 1805 if patch:
1803 1806 patch = q.lookup(patch)
1804 1807 else:
1805 1808 if not q.applied:
1806 1809 ui.write(_('No patches applied\n'))
1807 1810 return
1808 1811 patch = q.lookup('qtip')
1809 1812 absdest = q.join(name)
1810 1813 if os.path.isdir(absdest):
1811 1814 name = normname(os.path.join(name, os.path.basename(patch)))
1812 1815 absdest = q.join(name)
1813 1816 if os.path.exists(absdest):
1814 1817 raise util.Abort(_('%s already exists') % absdest)
1815 1818
1816 1819 if name in q.series:
1817 1820 raise util.Abort(_('A patch named %s already exists in the series file') % name)
1818 1821
1819 1822 if ui.verbose:
1820 1823 ui.write('Renaming %s to %s\n' % (patch, name))
1821 1824 i = q.find_series(patch)
1822 1825 guards = q.guard_re.findall(q.full_series[i])
1823 1826 q.full_series[i] = name + ''.join([' #' + g for g in guards])
1824 1827 q.parse_series()
1825 1828 q.series_dirty = 1
1826 1829
1827 1830 info = q.isapplied(patch)
1828 1831 if info:
1829 1832 q.applied[info[0]] = statusentry(info[1], name)
1830 1833 q.applied_dirty = 1
1831 1834
1832 1835 util.rename(q.join(patch), absdest)
1833 1836 r = q.qrepo()
1834 1837 if r:
1835 1838 wlock = r.wlock()
1836 1839 if r.dirstate.state(name) == 'r':
1837 1840 r.undelete([name], wlock)
1838 1841 r.copy(patch, name, wlock)
1839 1842 r.remove([patch], False, wlock)
1840 1843
1841 1844 q.save_dirty()
1842 1845
1843 1846 def restore(ui, repo, rev, **opts):
1844 1847 """restore the queue state saved by a rev"""
1845 1848 rev = repo.lookup(rev)
1846 1849 q = repo.mq
1847 1850 q.restore(repo, rev, delete=opts['delete'],
1848 1851 qupdate=opts['update'])
1849 1852 q.save_dirty()
1850 1853 return 0
1851 1854
1852 1855 def save(ui, repo, **opts):
1853 1856 """save current queue state"""
1854 1857 q = repo.mq
1855 1858 message = commands.logmessage(opts)
1856 1859 ret = q.save(repo, msg=message)
1857 1860 if ret:
1858 1861 return ret
1859 1862 q.save_dirty()
1860 1863 if opts['copy']:
1861 1864 path = q.path
1862 1865 if opts['name']:
1863 1866 newpath = os.path.join(q.basepath, opts['name'])
1864 1867 if os.path.exists(newpath):
1865 1868 if not os.path.isdir(newpath):
1866 1869 raise util.Abort(_('destination %s exists and is not '
1867 1870 'a directory') % newpath)
1868 1871 if not opts['force']:
1869 1872 raise util.Abort(_('destination %s exists, '
1870 1873 'use -f to force') % newpath)
1871 1874 else:
1872 1875 newpath = savename(path)
1873 1876 ui.warn("copy %s to %s\n" % (path, newpath))
1874 1877 util.copyfiles(path, newpath)
1875 1878 if opts['empty']:
1876 1879 try:
1877 1880 os.unlink(q.join(q.status_path))
1878 1881 except:
1879 1882 pass
1880 1883 return 0
1881 1884
1882 1885 def strip(ui, repo, rev, **opts):
1883 1886 """strip a revision and all later revs on the same branch"""
1884 1887 rev = repo.lookup(rev)
1885 1888 backup = 'all'
1886 1889 if opts['backup']:
1887 1890 backup = 'strip'
1888 1891 elif opts['nobackup']:
1889 1892 backup = 'none'
1890 1893 update = repo.dirstate.parents()[0] != revlog.nullid
1891 1894 repo.mq.strip(repo, rev, backup=backup, update=update)
1892 1895 return 0
1893 1896
1894 1897 def select(ui, repo, *args, **opts):
1895 1898 '''set or print guarded patches to push
1896 1899
1897 1900 Use the qguard command to set or print guards on patch, then use
1898 1901 qselect to tell mq which guards to use. A patch will be pushed if it
1899 1902 has no guards or any positive guards match the currently selected guard,
1900 1903 but will not be pushed if any negative guards match the current guard.
1901 1904 For example:
1902 1905
1903 1906 qguard foo.patch -stable (negative guard)
1904 1907 qguard bar.patch +stable (positive guard)
1905 1908 qselect stable
1906 1909
1907 1910 This activates the "stable" guard. mq will skip foo.patch (because
1908 1911 it has a negative match) but push bar.patch (because it
1909 1912 has a positive match).
1910 1913
1911 1914 With no arguments, prints the currently active guards.
1912 1915 With one argument, sets the active guard.
1913 1916
1914 1917 Use -n/--none to deactivate guards (no other arguments needed).
1915 1918 When no guards are active, patches with positive guards are skipped
1916 1919 and patches with negative guards are pushed.
1917 1920
1918 1921 qselect can change the guards on applied patches. It does not pop
1919 1922 guarded patches by default. Use --pop to pop back to the last applied
1920 1923 patch that is not guarded. Use --reapply (which implies --pop) to push
1921 1924 back to the current patch afterwards, but skip guarded patches.
1922 1925
1923 1926 Use -s/--series to print a list of all guards in the series file (no
1924 1927 other arguments needed). Use -v for more information.'''
1925 1928
1926 1929 q = repo.mq
1927 1930 guards = q.active()
1928 1931 if args or opts['none']:
1929 1932 old_unapplied = q.unapplied(repo)
1930 1933 old_guarded = [i for i in xrange(len(q.applied)) if
1931 1934 not q.pushable(i)[0]]
1932 1935 q.set_active(args)
1933 1936 q.save_dirty()
1934 1937 if not args:
1935 1938 ui.status(_('guards deactivated\n'))
1936 1939 if not opts['pop'] and not opts['reapply']:
1937 1940 unapplied = q.unapplied(repo)
1938 1941 guarded = [i for i in xrange(len(q.applied))
1939 1942 if not q.pushable(i)[0]]
1940 1943 if len(unapplied) != len(old_unapplied):
1941 1944 ui.status(_('number of unguarded, unapplied patches has '
1942 1945 'changed from %d to %d\n') %
1943 1946 (len(old_unapplied), len(unapplied)))
1944 1947 if len(guarded) != len(old_guarded):
1945 1948 ui.status(_('number of guarded, applied patches has changed '
1946 1949 'from %d to %d\n') %
1947 1950 (len(old_guarded), len(guarded)))
1948 1951 elif opts['series']:
1949 1952 guards = {}
1950 1953 noguards = 0
1951 1954 for gs in q.series_guards:
1952 1955 if not gs:
1953 1956 noguards += 1
1954 1957 for g in gs:
1955 1958 guards.setdefault(g, 0)
1956 1959 guards[g] += 1
1957 1960 if ui.verbose:
1958 1961 guards['NONE'] = noguards
1959 1962 guards = guards.items()
1960 1963 guards.sort(lambda a, b: cmp(a[0][1:], b[0][1:]))
1961 1964 if guards:
1962 1965 ui.note(_('guards in series file:\n'))
1963 1966 for guard, count in guards:
1964 1967 ui.note('%2d ' % count)
1965 1968 ui.write(guard, '\n')
1966 1969 else:
1967 1970 ui.note(_('no guards in series file\n'))
1968 1971 else:
1969 1972 if guards:
1970 1973 ui.note(_('active guards:\n'))
1971 1974 for g in guards:
1972 1975 ui.write(g, '\n')
1973 1976 else:
1974 1977 ui.write(_('no active guards\n'))
1975 1978 reapply = opts['reapply'] and q.applied and q.appliedname(-1)
1976 1979 popped = False
1977 1980 if opts['pop'] or opts['reapply']:
1978 1981 for i in xrange(len(q.applied)):
1979 1982 pushable, reason = q.pushable(i)
1980 1983 if not pushable:
1981 1984 ui.status(_('popping guarded patches\n'))
1982 1985 popped = True
1983 1986 if i == 0:
1984 1987 q.pop(repo, all=True)
1985 1988 else:
1986 1989 q.pop(repo, i-1)
1987 1990 break
1988 1991 if popped:
1989 1992 try:
1990 1993 if reapply:
1991 1994 ui.status(_('reapplying unguarded patches\n'))
1992 1995 q.push(repo, reapply)
1993 1996 finally:
1994 1997 q.save_dirty()
1995 1998
1996 1999 def reposetup(ui, repo):
1997 2000 class mqrepo(repo.__class__):
1998 2001 def abort_if_wdir_patched(self, errmsg, force=False):
1999 2002 if self.mq.applied and not force:
2000 2003 parent = revlog.hex(self.dirstate.parents()[0])
2001 2004 if parent in [s.rev for s in self.mq.applied]:
2002 2005 raise util.Abort(errmsg)
2003 2006
2004 2007 def commit(self, *args, **opts):
2005 2008 if len(args) >= 6:
2006 2009 force = args[5]
2007 2010 else:
2008 2011 force = opts.get('force')
2009 2012 self.abort_if_wdir_patched(
2010 2013 _('cannot commit over an applied mq patch'),
2011 2014 force)
2012 2015
2013 2016 return super(mqrepo, self).commit(*args, **opts)
2014 2017
2015 2018 def push(self, remote, force=False, revs=None):
2016 2019 if self.mq.applied and not force and not revs:
2017 2020 raise util.Abort(_('source has mq patches applied'))
2018 2021 return super(mqrepo, self).push(remote, force, revs)
2019 2022
2020 2023 def tags(self):
2021 2024 if self.tagscache:
2022 2025 return self.tagscache
2023 2026
2024 2027 tagscache = super(mqrepo, self).tags()
2025 2028
2026 2029 q = self.mq
2027 2030 if not q.applied:
2028 2031 return tagscache
2029 2032
2030 2033 mqtags = [(patch.rev, patch.name) for patch in q.applied]
2031 2034 mqtags.append((mqtags[-1][0], 'qtip'))
2032 2035 mqtags.append((mqtags[0][0], 'qbase'))
2033 2036 for patch in mqtags:
2034 2037 if patch[1] in tagscache:
2035 2038 self.ui.warn('Tag %s overrides mq patch of the same name\n' % patch[1])
2036 2039 else:
2037 2040 tagscache[patch[1]] = revlog.bin(patch[0])
2038 2041
2039 2042 return tagscache
2040 2043
2041 2044 def _branchtags(self):
2042 2045 q = self.mq
2043 2046 if not q.applied:
2044 2047 return super(mqrepo, self)._branchtags()
2045 2048
2046 2049 self.branchcache = {} # avoid recursion in changectx
2047 2050 cl = self.changelog
2048 2051 partial, last, lrev = self._readbranchcache()
2049 2052
2050 2053 qbase = cl.rev(revlog.bin(q.applied[0].rev))
2051 2054 start = lrev + 1
2052 2055 if start < qbase:
2053 2056 # update the cache (excluding the patches) and save it
2054 2057 self._updatebranchcache(partial, lrev+1, qbase)
2055 2058 self._writebranchcache(partial, cl.node(qbase-1), qbase-1)
2056 2059 start = qbase
2057 2060 # if start = qbase, the cache is as updated as it should be.
2058 2061 # if start > qbase, the cache includes (part of) the patches.
2059 2062 # we might as well use it, but we won't save it.
2060 2063
2061 2064 # update the cache up to the tip
2062 2065 self._updatebranchcache(partial, start, cl.count())
2063 2066
2064 2067 return partial
2065 2068
2066 2069 if repo.local():
2067 2070 repo.__class__ = mqrepo
2068 2071 repo.mq = queue(ui, repo.join(""))
2069 2072
2070 2073 seriesopts = [('s', 'summary', None, _('print first line of patch header'))]
2071 2074
2072 2075 cmdtable = {
2073 2076 "qapplied": (applied, [] + seriesopts, 'hg qapplied [-s] [PATCH]'),
2074 2077 "qclone": (clone,
2075 2078 [('', 'pull', None, _('use pull protocol to copy metadata')),
2076 2079 ('U', 'noupdate', None, _('do not update the new working directories')),
2077 2080 ('', 'uncompressed', None,
2078 2081 _('use uncompressed transfer (fast over LAN)')),
2079 2082 ('e', 'ssh', '', _('specify ssh command to use')),
2080 2083 ('p', 'patches', '', _('location of source patch repo')),
2081 2084 ('', 'remotecmd', '',
2082 2085 _('specify hg command to run on the remote side'))],
2083 2086 'hg qclone [OPTION]... SOURCE [DEST]'),
2084 2087 "qcommit|qci":
2085 2088 (commit,
2086 2089 commands.table["^commit|ci"][1],
2087 2090 'hg qcommit [OPTION]... [FILE]...'),
2088 2091 "^qdiff": (diff,
2089 2092 [('g', 'git', None, _('use git extended diff format')),
2090 2093 ('I', 'include', [], _('include names matching the given patterns')),
2091 2094 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2092 2095 'hg qdiff [-I] [-X] [FILE]...'),
2093 2096 "qdelete|qremove|qrm":
2094 2097 (delete,
2095 2098 [('k', 'keep', None, _('keep patch file')),
2096 2099 ('r', 'rev', [], _('stop managing a revision'))],
2097 2100 'hg qdelete [-k] [-r REV]... PATCH...'),
2098 2101 'qfold':
2099 2102 (fold,
2100 2103 [('e', 'edit', None, _('edit patch header')),
2101 2104 ('k', 'keep', None, _('keep folded patch files'))
2102 2105 ] + commands.commitopts,
2103 2106 'hg qfold [-e] [-m <text>] [-l <file] PATCH...'),
2104 2107 'qguard': (guard, [('l', 'list', None, _('list all patches and guards')),
2105 2108 ('n', 'none', None, _('drop all guards'))],
2106 2109 'hg qguard [PATCH] [+GUARD...] [-GUARD...]'),
2107 2110 'qheader': (header, [],
2108 2111 _('hg qheader [PATCH]')),
2109 2112 "^qimport":
2110 2113 (qimport,
2111 2114 [('e', 'existing', None, 'import file in patch dir'),
2112 2115 ('n', 'name', '', 'patch file name'),
2113 2116 ('f', 'force', None, 'overwrite existing files'),
2114 2117 ('r', 'rev', [], 'place existing revisions under mq control'),
2115 2118 ('g', 'git', None, _('use git extended diff format'))],
2116 2119 'hg qimport [-e] [-n NAME] [-f] [-g] [-r REV]... FILE...'),
2117 2120 "^qinit":
2118 2121 (init,
2119 2122 [('c', 'create-repo', None, 'create queue repository')],
2120 2123 'hg qinit [-c]'),
2121 2124 "qnew":
2122 2125 (new,
2123 2126 [('e', 'edit', None, _('edit commit message')),
2124 2127 ('f', 'force', None, _('import uncommitted changes into patch'))
2125 2128 ] + commands.commitopts,
2126 2129 'hg qnew [-e] [-m TEXT] [-l FILE] [-f] PATCH'),
2127 2130 "qnext": (next, [] + seriesopts, 'hg qnext [-s]'),
2128 2131 "qprev": (prev, [] + seriesopts, 'hg qprev [-s]'),
2129 2132 "^qpop":
2130 2133 (pop,
2131 2134 [('a', 'all', None, 'pop all patches'),
2132 2135 ('n', 'name', '', 'queue name to pop'),
2133 2136 ('f', 'force', None, 'forget any local changes')],
2134 2137 'hg qpop [-a] [-n NAME] [-f] [PATCH | INDEX]'),
2135 2138 "^qpush":
2136 2139 (push,
2137 2140 [('f', 'force', None, 'apply if the patch has rejects'),
2138 2141 ('l', 'list', None, 'list patch name in commit text'),
2139 2142 ('a', 'all', None, 'apply all patches'),
2140 2143 ('m', 'merge', None, 'merge from another queue'),
2141 2144 ('n', 'name', '', 'merge queue name')],
2142 2145 'hg qpush [-f] [-l] [-a] [-m] [-n NAME] [PATCH | INDEX]'),
2143 2146 "^qrefresh":
2144 2147 (refresh,
2145 2148 [('e', 'edit', None, _('edit commit message')),
2146 2149 ('g', 'git', None, _('use git extended diff format')),
2147 2150 ('s', 'short', None, 'refresh only files already in the patch'),
2148 2151 ('I', 'include', [], _('include names matching the given patterns')),
2149 2152 ('X', 'exclude', [], _('exclude names matching the given patterns'))
2150 2153 ] + commands.commitopts,
2151 2154 'hg qrefresh [-I] [-X] [-e] [-m TEXT] [-l FILE] [-s] FILES...'),
2152 2155 'qrename|qmv':
2153 2156 (rename, [], 'hg qrename PATCH1 [PATCH2]'),
2154 2157 "qrestore":
2155 2158 (restore,
2156 2159 [('d', 'delete', None, 'delete save entry'),
2157 2160 ('u', 'update', None, 'update queue working dir')],
2158 2161 'hg qrestore [-d] [-u] REV'),
2159 2162 "qsave":
2160 2163 (save,
2161 2164 [('c', 'copy', None, 'copy patch directory'),
2162 2165 ('n', 'name', '', 'copy directory name'),
2163 2166 ('e', 'empty', None, 'clear queue status file'),
2164 2167 ('f', 'force', None, 'force copy')] + commands.commitopts,
2165 2168 'hg qsave [-m TEXT] [-l FILE] [-c] [-n NAME] [-e] [-f]'),
2166 2169 "qselect": (select,
2167 2170 [('n', 'none', None, _('disable all guards')),
2168 2171 ('s', 'series', None, _('list all guards in series file')),
2169 2172 ('', 'pop', None,
2170 2173 _('pop to before first guarded applied patch')),
2171 2174 ('', 'reapply', None, _('pop, then reapply patches'))],
2172 2175 'hg qselect [OPTION...] [GUARD...]'),
2173 2176 "qseries":
2174 2177 (series,
2175 2178 [('m', 'missing', None, 'print patches not in series')] + seriesopts,
2176 2179 'hg qseries [-ms]'),
2177 2180 "^strip":
2178 2181 (strip,
2179 2182 [('f', 'force', None, 'force multi-head removal'),
2180 2183 ('b', 'backup', None, 'bundle unrelated changesets'),
2181 2184 ('n', 'nobackup', None, 'no backups')],
2182 2185 'hg strip [-f] [-b] [-n] REV'),
2183 2186 "qtop": (top, [] + seriesopts, 'hg qtop [-s]'),
2184 2187 "qunapplied": (unapplied, [] + seriesopts, 'hg qunapplied [-s] [PATCH]'),
2185 2188 }
General Comments 0
You need to be logged in to leave comments. Login now