##// END OF EJS Templates
mq: align columns in verbose qseries output.
Dan Villiom Podlaski Christiansen -
r9016:894c5b4b default
parent child Browse files
Show More
@@ -1,2625 +1,2630 b''
1 1 # mq.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 of the
6 6 # GNU General Public License version 2, incorporated herein by reference.
7 7
8 8 '''manage a stack of patches
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.node import bin, hex, short, nullid, nullrev
34 34 from mercurial.lock import release
35 35 from mercurial import commands, cmdutil, hg, patch, util
36 36 from mercurial import repair, extensions, url, error
37 37 import os, sys, re, errno
38 38
39 39 commands.norepo += " qclone"
40 40
41 41 # Patch names looks like unix-file names.
42 42 # They must be joinable with queue directory and result in the patch path.
43 43 normname = util.normpath
44 44
45 45 class statusentry(object):
46 46 def __init__(self, rev, name=None):
47 47 if not name:
48 48 fields = rev.split(':', 1)
49 49 if len(fields) == 2:
50 50 self.rev, self.name = fields
51 51 else:
52 52 self.rev, self.name = None, None
53 53 else:
54 54 self.rev, self.name = rev, name
55 55
56 56 def __str__(self):
57 57 return self.rev + ':' + self.name
58 58
59 59 class patchheader(object):
60 60 def __init__(self, pf):
61 61 def eatdiff(lines):
62 62 while lines:
63 63 l = lines[-1]
64 64 if (l.startswith("diff -") or
65 65 l.startswith("Index:") or
66 66 l.startswith("===========")):
67 67 del lines[-1]
68 68 else:
69 69 break
70 70 def eatempty(lines):
71 71 while lines:
72 72 l = lines[-1]
73 73 if re.match('\s*$', l):
74 74 del lines[-1]
75 75 else:
76 76 break
77 77
78 78 message = []
79 79 comments = []
80 80 user = None
81 81 date = None
82 82 format = None
83 83 subject = None
84 84 diffstart = 0
85 85
86 86 for line in file(pf):
87 87 line = line.rstrip()
88 88 if line.startswith('diff --git'):
89 89 diffstart = 2
90 90 break
91 91 if diffstart:
92 92 if line.startswith('+++ '):
93 93 diffstart = 2
94 94 break
95 95 if line.startswith("--- "):
96 96 diffstart = 1
97 97 continue
98 98 elif format == "hgpatch":
99 99 # parse values when importing the result of an hg export
100 100 if line.startswith("# User "):
101 101 user = line[7:]
102 102 elif line.startswith("# Date "):
103 103 date = line[7:]
104 104 elif not line.startswith("# ") and line:
105 105 message.append(line)
106 106 format = None
107 107 elif line == '# HG changeset patch':
108 108 format = "hgpatch"
109 109 elif (format != "tagdone" and (line.startswith("Subject: ") or
110 110 line.startswith("subject: "))):
111 111 subject = line[9:]
112 112 format = "tag"
113 113 elif (format != "tagdone" and (line.startswith("From: ") or
114 114 line.startswith("from: "))):
115 115 user = line[6:]
116 116 format = "tag"
117 117 elif format == "tag" and line == "":
118 118 # when looking for tags (subject: from: etc) they
119 119 # end once you find a blank line in the source
120 120 format = "tagdone"
121 121 elif message or line:
122 122 message.append(line)
123 123 comments.append(line)
124 124
125 125 eatdiff(message)
126 126 eatdiff(comments)
127 127 eatempty(message)
128 128 eatempty(comments)
129 129
130 130 # make sure message isn't empty
131 131 if format and format.startswith("tag") and subject:
132 132 message.insert(0, "")
133 133 message.insert(0, subject)
134 134
135 135 self.message = message
136 136 self.comments = comments
137 137 self.user = user
138 138 self.date = date
139 139 self.haspatch = diffstart > 1
140 140
141 141 def setuser(self, user):
142 142 if not self.updateheader(['From: ', '# User '], user):
143 143 try:
144 144 patchheaderat = self.comments.index('# HG changeset patch')
145 145 self.comments.insert(patchheaderat + 1,'# User ' + user)
146 146 except ValueError:
147 147 self.comments = ['From: ' + user, ''] + self.comments
148 148 self.user = user
149 149
150 150 def setdate(self, date):
151 151 if self.updateheader(['# Date '], date):
152 152 self.date = date
153 153
154 154 def setmessage(self, message):
155 155 if self.comments:
156 156 self._delmsg()
157 157 self.message = [message]
158 158 self.comments += self.message
159 159
160 160 def updateheader(self, prefixes, new):
161 161 '''Update all references to a field in the patch header.
162 162 Return whether the field is present.'''
163 163 res = False
164 164 for prefix in prefixes:
165 165 for i in xrange(len(self.comments)):
166 166 if self.comments[i].startswith(prefix):
167 167 self.comments[i] = prefix + new
168 168 res = True
169 169 break
170 170 return res
171 171
172 172 def __str__(self):
173 173 if not self.comments:
174 174 return ''
175 175 return '\n'.join(self.comments) + '\n\n'
176 176
177 177 def _delmsg(self):
178 178 '''Remove existing message, keeping the rest of the comments fields.
179 179 If comments contains 'subject: ', message will prepend
180 180 the field and a blank line.'''
181 181 if self.message:
182 182 subj = 'subject: ' + self.message[0].lower()
183 183 for i in xrange(len(self.comments)):
184 184 if subj == self.comments[i].lower():
185 185 del self.comments[i]
186 186 self.message = self.message[2:]
187 187 break
188 188 ci = 0
189 189 for mi in self.message:
190 190 while mi != self.comments[ci]:
191 191 ci += 1
192 192 del self.comments[ci]
193 193
194 194 class queue(object):
195 195 def __init__(self, ui, path, patchdir=None):
196 196 self.basepath = path
197 197 self.path = patchdir or os.path.join(path, "patches")
198 198 self.opener = util.opener(self.path)
199 199 self.ui = ui
200 200 self.applied_dirty = 0
201 201 self.series_dirty = 0
202 202 self.series_path = "series"
203 203 self.status_path = "status"
204 204 self.guards_path = "guards"
205 205 self.active_guards = None
206 206 self.guards_dirty = False
207 207 self._diffopts = None
208 208
209 209 @util.propertycache
210 210 def applied(self):
211 211 if os.path.exists(self.join(self.status_path)):
212 212 lines = self.opener(self.status_path).read().splitlines()
213 213 return [statusentry(l) for l in lines]
214 214 return []
215 215
216 216 @util.propertycache
217 217 def full_series(self):
218 218 if os.path.exists(self.join(self.series_path)):
219 219 return self.opener(self.series_path).read().splitlines()
220 220 return []
221 221
222 222 @util.propertycache
223 223 def series(self):
224 224 self.parse_series()
225 225 return self.series
226 226
227 227 @util.propertycache
228 228 def series_guards(self):
229 229 self.parse_series()
230 230 return self.series_guards
231 231
232 232 def invalidate(self):
233 233 for a in 'applied full_series series series_guards'.split():
234 234 if a in self.__dict__:
235 235 delattr(self, a)
236 236 self.applied_dirty = 0
237 237 self.series_dirty = 0
238 238 self.guards_dirty = False
239 239 self.active_guards = None
240 240
241 241 def diffopts(self):
242 242 if self._diffopts is None:
243 243 self._diffopts = patch.diffopts(self.ui)
244 244 return self._diffopts
245 245
246 246 def join(self, *p):
247 247 return os.path.join(self.path, *p)
248 248
249 249 def find_series(self, patch):
250 250 pre = re.compile("(\s*)([^#]+)")
251 251 index = 0
252 252 for l in self.full_series:
253 253 m = pre.match(l)
254 254 if m:
255 255 s = m.group(2)
256 256 s = s.rstrip()
257 257 if s == patch:
258 258 return index
259 259 index += 1
260 260 return None
261 261
262 262 guard_re = re.compile(r'\s?#([-+][^-+# \t\r\n\f][^# \t\r\n\f]*)')
263 263
264 264 def parse_series(self):
265 265 self.series = []
266 266 self.series_guards = []
267 267 for l in self.full_series:
268 268 h = l.find('#')
269 269 if h == -1:
270 270 patch = l
271 271 comment = ''
272 272 elif h == 0:
273 273 continue
274 274 else:
275 275 patch = l[:h]
276 276 comment = l[h:]
277 277 patch = patch.strip()
278 278 if patch:
279 279 if patch in self.series:
280 280 raise util.Abort(_('%s appears more than once in %s') %
281 281 (patch, self.join(self.series_path)))
282 282 self.series.append(patch)
283 283 self.series_guards.append(self.guard_re.findall(comment))
284 284
285 285 def check_guard(self, guard):
286 286 if not guard:
287 287 return _('guard cannot be an empty string')
288 288 bad_chars = '# \t\r\n\f'
289 289 first = guard[0]
290 290 if first in '-+':
291 291 return (_('guard %r starts with invalid character: %r') %
292 292 (guard, first))
293 293 for c in bad_chars:
294 294 if c in guard:
295 295 return _('invalid character in guard %r: %r') % (guard, c)
296 296
297 297 def set_active(self, guards):
298 298 for guard in guards:
299 299 bad = self.check_guard(guard)
300 300 if bad:
301 301 raise util.Abort(bad)
302 302 guards = sorted(set(guards))
303 303 self.ui.debug(_('active guards: %s\n') % ' '.join(guards))
304 304 self.active_guards = guards
305 305 self.guards_dirty = True
306 306
307 307 def active(self):
308 308 if self.active_guards is None:
309 309 self.active_guards = []
310 310 try:
311 311 guards = self.opener(self.guards_path).read().split()
312 312 except IOError, err:
313 313 if err.errno != errno.ENOENT: raise
314 314 guards = []
315 315 for i, guard in enumerate(guards):
316 316 bad = self.check_guard(guard)
317 317 if bad:
318 318 self.ui.warn('%s:%d: %s\n' %
319 319 (self.join(self.guards_path), i + 1, bad))
320 320 else:
321 321 self.active_guards.append(guard)
322 322 return self.active_guards
323 323
324 324 def set_guards(self, idx, guards):
325 325 for g in guards:
326 326 if len(g) < 2:
327 327 raise util.Abort(_('guard %r too short') % g)
328 328 if g[0] not in '-+':
329 329 raise util.Abort(_('guard %r starts with invalid char') % g)
330 330 bad = self.check_guard(g[1:])
331 331 if bad:
332 332 raise util.Abort(bad)
333 333 drop = self.guard_re.sub('', self.full_series[idx])
334 334 self.full_series[idx] = drop + ''.join([' #' + g for g in guards])
335 335 self.parse_series()
336 336 self.series_dirty = True
337 337
338 338 def pushable(self, idx):
339 339 if isinstance(idx, str):
340 340 idx = self.series.index(idx)
341 341 patchguards = self.series_guards[idx]
342 342 if not patchguards:
343 343 return True, None
344 344 guards = self.active()
345 345 exactneg = [g for g in patchguards if g[0] == '-' and g[1:] in guards]
346 346 if exactneg:
347 347 return False, exactneg[0]
348 348 pos = [g for g in patchguards if g[0] == '+']
349 349 exactpos = [g for g in pos if g[1:] in guards]
350 350 if pos:
351 351 if exactpos:
352 352 return True, exactpos[0]
353 353 return False, pos
354 354 return True, ''
355 355
356 356 def explain_pushable(self, idx, all_patches=False):
357 357 write = all_patches and self.ui.write or self.ui.warn
358 358 if all_patches or self.ui.verbose:
359 359 if isinstance(idx, str):
360 360 idx = self.series.index(idx)
361 361 pushable, why = self.pushable(idx)
362 362 if all_patches and pushable:
363 363 if why is None:
364 364 write(_('allowing %s - no guards in effect\n') %
365 365 self.series[idx])
366 366 else:
367 367 if not why:
368 368 write(_('allowing %s - no matching negative guards\n') %
369 369 self.series[idx])
370 370 else:
371 371 write(_('allowing %s - guarded by %r\n') %
372 372 (self.series[idx], why))
373 373 if not pushable:
374 374 if why:
375 375 write(_('skipping %s - guarded by %r\n') %
376 376 (self.series[idx], why))
377 377 else:
378 378 write(_('skipping %s - no matching guards\n') %
379 379 self.series[idx])
380 380
381 381 def save_dirty(self):
382 382 def write_list(items, path):
383 383 fp = self.opener(path, 'w')
384 384 for i in items:
385 385 fp.write("%s\n" % i)
386 386 fp.close()
387 387 if self.applied_dirty: write_list(map(str, self.applied), self.status_path)
388 388 if self.series_dirty: write_list(self.full_series, self.series_path)
389 389 if self.guards_dirty: write_list(self.active_guards, self.guards_path)
390 390
391 391 def removeundo(self, repo):
392 392 undo = repo.sjoin('undo')
393 393 if not os.path.exists(undo):
394 394 return
395 395 try:
396 396 os.unlink(undo)
397 397 except OSError, inst:
398 398 self.ui.warn(_('error removing undo: %s\n') % str(inst))
399 399
400 400 def printdiff(self, repo, node1, node2=None, files=None,
401 401 fp=None, changes=None, opts={}):
402 402 m = cmdutil.match(repo, files, opts)
403 403 chunks = patch.diff(repo, node1, node2, m, changes, self.diffopts())
404 404 write = fp is None and repo.ui.write or fp.write
405 405 for chunk in chunks:
406 406 write(chunk)
407 407
408 408 def mergeone(self, repo, mergeq, head, patch, rev):
409 409 # first try just applying the patch
410 410 (err, n) = self.apply(repo, [ patch ], update_status=False,
411 411 strict=True, merge=rev)
412 412
413 413 if err == 0:
414 414 return (err, n)
415 415
416 416 if n is None:
417 417 raise util.Abort(_("apply failed for patch %s") % patch)
418 418
419 419 self.ui.warn(_("patch didn't work out, merging %s\n") % patch)
420 420
421 421 # apply failed, strip away that rev and merge.
422 422 hg.clean(repo, head)
423 423 self.strip(repo, n, update=False, backup='strip')
424 424
425 425 ctx = repo[rev]
426 426 ret = hg.merge(repo, rev)
427 427 if ret:
428 428 raise util.Abort(_("update returned %d") % ret)
429 429 n = repo.commit(ctx.description(), ctx.user(), force=True)
430 430 if n is None:
431 431 raise util.Abort(_("repo commit failed"))
432 432 try:
433 433 ph = patchheader(mergeq.join(patch))
434 434 except:
435 435 raise util.Abort(_("unable to read %s") % patch)
436 436
437 437 patchf = self.opener(patch, "w")
438 438 comments = str(ph)
439 439 if comments:
440 440 patchf.write(comments)
441 441 self.printdiff(repo, head, n, fp=patchf)
442 442 patchf.close()
443 443 self.removeundo(repo)
444 444 return (0, n)
445 445
446 446 def qparents(self, repo, rev=None):
447 447 if rev is None:
448 448 (p1, p2) = repo.dirstate.parents()
449 449 if p2 == nullid:
450 450 return p1
451 451 if len(self.applied) == 0:
452 452 return None
453 453 return bin(self.applied[-1].rev)
454 454 pp = repo.changelog.parents(rev)
455 455 if pp[1] != nullid:
456 456 arevs = [ x.rev for x in self.applied ]
457 457 p0 = hex(pp[0])
458 458 p1 = hex(pp[1])
459 459 if p0 in arevs:
460 460 return pp[0]
461 461 if p1 in arevs:
462 462 return pp[1]
463 463 return pp[0]
464 464
465 465 def mergepatch(self, repo, mergeq, series):
466 466 if len(self.applied) == 0:
467 467 # each of the patches merged in will have two parents. This
468 468 # can confuse the qrefresh, qdiff, and strip code because it
469 469 # needs to know which parent is actually in the patch queue.
470 470 # so, we insert a merge marker with only one parent. This way
471 471 # the first patch in the queue is never a merge patch
472 472 #
473 473 pname = ".hg.patches.merge.marker"
474 474 n = repo.commit('[mq]: merge marker', force=True)
475 475 self.removeundo(repo)
476 476 self.applied.append(statusentry(hex(n), pname))
477 477 self.applied_dirty = 1
478 478
479 479 head = self.qparents(repo)
480 480
481 481 for patch in series:
482 482 patch = mergeq.lookup(patch, strict=True)
483 483 if not patch:
484 484 self.ui.warn(_("patch %s does not exist\n") % patch)
485 485 return (1, None)
486 486 pushable, reason = self.pushable(patch)
487 487 if not pushable:
488 488 self.explain_pushable(patch, all_patches=True)
489 489 continue
490 490 info = mergeq.isapplied(patch)
491 491 if not info:
492 492 self.ui.warn(_("patch %s is not applied\n") % patch)
493 493 return (1, None)
494 494 rev = bin(info[1])
495 495 (err, head) = self.mergeone(repo, mergeq, head, patch, rev)
496 496 if head:
497 497 self.applied.append(statusentry(hex(head), patch))
498 498 self.applied_dirty = 1
499 499 if err:
500 500 return (err, head)
501 501 self.save_dirty()
502 502 return (0, head)
503 503
504 504 def patch(self, repo, patchfile):
505 505 '''Apply patchfile to the working directory.
506 506 patchfile: name of patch file'''
507 507 files = {}
508 508 try:
509 509 fuzz = patch.patch(patchfile, self.ui, strip=1, cwd=repo.root,
510 510 files=files, eolmode=None)
511 511 except Exception, inst:
512 512 self.ui.note(str(inst) + '\n')
513 513 if not self.ui.verbose:
514 514 self.ui.warn(_("patch failed, unable to continue (try -v)\n"))
515 515 return (False, files, False)
516 516
517 517 return (True, files, fuzz)
518 518
519 519 def apply(self, repo, series, list=False, update_status=True,
520 520 strict=False, patchdir=None, merge=None, all_files={}):
521 521 wlock = lock = tr = None
522 522 try:
523 523 wlock = repo.wlock()
524 524 lock = repo.lock()
525 525 tr = repo.transaction()
526 526 try:
527 527 ret = self._apply(repo, series, list, update_status,
528 528 strict, patchdir, merge, all_files=all_files)
529 529 tr.close()
530 530 self.save_dirty()
531 531 return ret
532 532 except:
533 533 try:
534 534 tr.abort()
535 535 finally:
536 536 repo.invalidate()
537 537 repo.dirstate.invalidate()
538 538 raise
539 539 finally:
540 540 del tr
541 541 release(lock, wlock)
542 542 self.removeundo(repo)
543 543
544 544 def _apply(self, repo, series, list=False, update_status=True,
545 545 strict=False, patchdir=None, merge=None, all_files={}):
546 546 '''returns (error, hash)
547 547 error = 1 for unable to read, 2 for patch failed, 3 for patch fuzz'''
548 548 # TODO unify with commands.py
549 549 if not patchdir:
550 550 patchdir = self.path
551 551 err = 0
552 552 n = None
553 553 for patchname in series:
554 554 pushable, reason = self.pushable(patchname)
555 555 if not pushable:
556 556 self.explain_pushable(patchname, all_patches=True)
557 557 continue
558 558 self.ui.warn(_("applying %s\n") % patchname)
559 559 pf = os.path.join(patchdir, patchname)
560 560
561 561 try:
562 562 ph = patchheader(self.join(patchname))
563 563 except:
564 564 self.ui.warn(_("unable to read %s\n") % patchname)
565 565 err = 1
566 566 break
567 567
568 568 message = ph.message
569 569 if not message:
570 570 message = _("imported patch %s\n") % patchname
571 571 else:
572 572 if list:
573 573 message.append(_("\nimported patch %s") % patchname)
574 574 message = '\n'.join(message)
575 575
576 576 if ph.haspatch:
577 577 (patcherr, files, fuzz) = self.patch(repo, pf)
578 578 all_files.update(files)
579 579 patcherr = not patcherr
580 580 else:
581 581 self.ui.warn(_("patch %s is empty\n") % patchname)
582 582 patcherr, files, fuzz = 0, [], 0
583 583
584 584 if merge and files:
585 585 # Mark as removed/merged and update dirstate parent info
586 586 removed = []
587 587 merged = []
588 588 for f in files:
589 589 if os.path.exists(repo.wjoin(f)):
590 590 merged.append(f)
591 591 else:
592 592 removed.append(f)
593 593 for f in removed:
594 594 repo.dirstate.remove(f)
595 595 for f in merged:
596 596 repo.dirstate.merge(f)
597 597 p1, p2 = repo.dirstate.parents()
598 598 repo.dirstate.setparents(p1, merge)
599 599
600 600 files = patch.updatedir(self.ui, repo, files)
601 601 match = cmdutil.matchfiles(repo, files or [])
602 602 n = repo.commit(message, ph.user, ph.date, match=match, force=True)
603 603
604 604 if n is None:
605 605 raise util.Abort(_("repo commit failed"))
606 606
607 607 if update_status:
608 608 self.applied.append(statusentry(hex(n), patchname))
609 609
610 610 if patcherr:
611 611 self.ui.warn(_("patch failed, rejects left in working dir\n"))
612 612 err = 2
613 613 break
614 614
615 615 if fuzz and strict:
616 616 self.ui.warn(_("fuzz found when applying patch, stopping\n"))
617 617 err = 3
618 618 break
619 619 return (err, n)
620 620
621 621 def _cleanup(self, patches, numrevs, keep=False):
622 622 if not keep:
623 623 r = self.qrepo()
624 624 if r:
625 625 r.remove(patches, True)
626 626 else:
627 627 for p in patches:
628 628 os.unlink(self.join(p))
629 629
630 630 if numrevs:
631 631 del self.applied[:numrevs]
632 632 self.applied_dirty = 1
633 633
634 634 for i in sorted([self.find_series(p) for p in patches], reverse=True):
635 635 del self.full_series[i]
636 636 self.parse_series()
637 637 self.series_dirty = 1
638 638
639 639 def _revpatches(self, repo, revs):
640 640 firstrev = repo[self.applied[0].rev].rev()
641 641 patches = []
642 642 for i, rev in enumerate(revs):
643 643
644 644 if rev < firstrev:
645 645 raise util.Abort(_('revision %d is not managed') % rev)
646 646
647 647 ctx = repo[rev]
648 648 base = bin(self.applied[i].rev)
649 649 if ctx.node() != base:
650 650 msg = _('cannot delete revision %d above applied patches')
651 651 raise util.Abort(msg % rev)
652 652
653 653 patch = self.applied[i].name
654 654 for fmt in ('[mq]: %s', 'imported patch %s'):
655 655 if ctx.description() == fmt % patch:
656 656 msg = _('patch %s finalized without changeset message\n')
657 657 repo.ui.status(msg % patch)
658 658 break
659 659
660 660 patches.append(patch)
661 661 return patches
662 662
663 663 def finish(self, repo, revs):
664 664 patches = self._revpatches(repo, sorted(revs))
665 665 self._cleanup(patches, len(patches))
666 666
667 667 def delete(self, repo, patches, opts):
668 668 if not patches and not opts.get('rev'):
669 669 raise util.Abort(_('qdelete requires at least one revision or '
670 670 'patch name'))
671 671
672 672 realpatches = []
673 673 for patch in patches:
674 674 patch = self.lookup(patch, strict=True)
675 675 info = self.isapplied(patch)
676 676 if info:
677 677 raise util.Abort(_("cannot delete applied patch %s") % patch)
678 678 if patch not in self.series:
679 679 raise util.Abort(_("patch %s not in series file") % patch)
680 680 realpatches.append(patch)
681 681
682 682 numrevs = 0
683 683 if opts.get('rev'):
684 684 if not self.applied:
685 685 raise util.Abort(_('no patches applied'))
686 686 revs = cmdutil.revrange(repo, opts['rev'])
687 687 if len(revs) > 1 and revs[0] > revs[1]:
688 688 revs.reverse()
689 689 revpatches = self._revpatches(repo, revs)
690 690 realpatches += revpatches
691 691 numrevs = len(revpatches)
692 692
693 693 self._cleanup(realpatches, numrevs, opts.get('keep'))
694 694
695 695 def check_toppatch(self, repo):
696 696 if len(self.applied) > 0:
697 697 top = bin(self.applied[-1].rev)
698 698 pp = repo.dirstate.parents()
699 699 if top not in pp:
700 700 raise util.Abort(_("working directory revision is not qtip"))
701 701 return top
702 702 return None
703 703 def check_localchanges(self, repo, force=False, refresh=True):
704 704 m, a, r, d = repo.status()[:4]
705 705 if m or a or r or d:
706 706 if not force:
707 707 if refresh:
708 708 raise util.Abort(_("local changes found, refresh first"))
709 709 else:
710 710 raise util.Abort(_("local changes found"))
711 711 return m, a, r, d
712 712
713 713 _reserved = ('series', 'status', 'guards')
714 714 def check_reserved_name(self, name):
715 715 if (name in self._reserved or name.startswith('.hg')
716 716 or name.startswith('.mq')):
717 717 raise util.Abort(_('"%s" cannot be used as the name of a patch')
718 718 % name)
719 719
720 720 def new(self, repo, patchfn, *pats, **opts):
721 721 """options:
722 722 msg: a string or a no-argument function returning a string
723 723 """
724 724 msg = opts.get('msg')
725 725 force = opts.get('force')
726 726 user = opts.get('user')
727 727 date = opts.get('date')
728 728 if date:
729 729 date = util.parsedate(date)
730 730 self.check_reserved_name(patchfn)
731 731 if os.path.exists(self.join(patchfn)):
732 732 raise util.Abort(_('patch "%s" already exists') % patchfn)
733 733 if opts.get('include') or opts.get('exclude') or pats:
734 734 match = cmdutil.match(repo, pats, opts)
735 735 # detect missing files in pats
736 736 def badfn(f, msg):
737 737 raise util.Abort('%s: %s' % (f, msg))
738 738 match.bad = badfn
739 739 m, a, r, d = repo.status(match=match)[:4]
740 740 else:
741 741 m, a, r, d = self.check_localchanges(repo, force)
742 742 match = cmdutil.matchfiles(repo, m + a + r)
743 743 commitfiles = m + a + r
744 744 self.check_toppatch(repo)
745 745 insert = self.full_series_end()
746 746 wlock = repo.wlock()
747 747 try:
748 748 # if patch file write fails, abort early
749 749 p = self.opener(patchfn, "w")
750 750 try:
751 751 if date:
752 752 p.write("# HG changeset patch\n")
753 753 if user:
754 754 p.write("# User " + user + "\n")
755 755 p.write("# Date %d %d\n\n" % date)
756 756 elif user:
757 757 p.write("From: " + user + "\n\n")
758 758
759 759 if hasattr(msg, '__call__'):
760 760 msg = msg()
761 761 commitmsg = msg and msg or ("[mq]: %s" % patchfn)
762 762 n = repo.commit(commitmsg, user, date, match=match, force=True)
763 763 if n is None:
764 764 raise util.Abort(_("repo commit failed"))
765 765 try:
766 766 self.full_series[insert:insert] = [patchfn]
767 767 self.applied.append(statusentry(hex(n), patchfn))
768 768 self.parse_series()
769 769 self.series_dirty = 1
770 770 self.applied_dirty = 1
771 771 if msg:
772 772 msg = msg + "\n\n"
773 773 p.write(msg)
774 774 if commitfiles:
775 775 diffopts = self.diffopts()
776 776 if opts.get('git'): diffopts.git = True
777 777 parent = self.qparents(repo, n)
778 778 chunks = patch.diff(repo, node1=parent, node2=n,
779 779 match=match, opts=diffopts)
780 780 for chunk in chunks:
781 781 p.write(chunk)
782 782 p.close()
783 783 wlock.release()
784 784 wlock = None
785 785 r = self.qrepo()
786 786 if r: r.add([patchfn])
787 787 except:
788 788 repo.rollback()
789 789 raise
790 790 except Exception:
791 791 patchpath = self.join(patchfn)
792 792 try:
793 793 os.unlink(patchpath)
794 794 except:
795 795 self.ui.warn(_('error unlinking %s\n') % patchpath)
796 796 raise
797 797 self.removeundo(repo)
798 798 finally:
799 799 release(wlock)
800 800
801 801 def strip(self, repo, rev, update=True, backup="all", force=None):
802 802 wlock = lock = None
803 803 try:
804 804 wlock = repo.wlock()
805 805 lock = repo.lock()
806 806
807 807 if update:
808 808 self.check_localchanges(repo, force=force, refresh=False)
809 809 urev = self.qparents(repo, rev)
810 810 hg.clean(repo, urev)
811 811 repo.dirstate.write()
812 812
813 813 self.removeundo(repo)
814 814 repair.strip(self.ui, repo, rev, backup)
815 815 # strip may have unbundled a set of backed up revisions after
816 816 # the actual strip
817 817 self.removeundo(repo)
818 818 finally:
819 819 release(lock, wlock)
820 820
821 821 def isapplied(self, patch):
822 822 """returns (index, rev, patch)"""
823 823 for i, a in enumerate(self.applied):
824 824 if a.name == patch:
825 825 return (i, a.rev, a.name)
826 826 return None
827 827
828 828 # if the exact patch name does not exist, we try a few
829 829 # variations. If strict is passed, we try only #1
830 830 #
831 831 # 1) a number to indicate an offset in the series file
832 832 # 2) a unique substring of the patch name was given
833 833 # 3) patchname[-+]num to indicate an offset in the series file
834 834 def lookup(self, patch, strict=False):
835 835 patch = patch and str(patch)
836 836
837 837 def partial_name(s):
838 838 if s in self.series:
839 839 return s
840 840 matches = [x for x in self.series if s in x]
841 841 if len(matches) > 1:
842 842 self.ui.warn(_('patch name "%s" is ambiguous:\n') % s)
843 843 for m in matches:
844 844 self.ui.warn(' %s\n' % m)
845 845 return None
846 846 if matches:
847 847 return matches[0]
848 848 if len(self.series) > 0 and len(self.applied) > 0:
849 849 if s == 'qtip':
850 850 return self.series[self.series_end(True)-1]
851 851 if s == 'qbase':
852 852 return self.series[0]
853 853 return None
854 854
855 855 if patch is None:
856 856 return None
857 857 if patch in self.series:
858 858 return patch
859 859
860 860 if not os.path.isfile(self.join(patch)):
861 861 try:
862 862 sno = int(patch)
863 863 except(ValueError, OverflowError):
864 864 pass
865 865 else:
866 866 if -len(self.series) <= sno < len(self.series):
867 867 return self.series[sno]
868 868
869 869 if not strict:
870 870 res = partial_name(patch)
871 871 if res:
872 872 return res
873 873 minus = patch.rfind('-')
874 874 if minus >= 0:
875 875 res = partial_name(patch[:minus])
876 876 if res:
877 877 i = self.series.index(res)
878 878 try:
879 879 off = int(patch[minus+1:] or 1)
880 880 except(ValueError, OverflowError):
881 881 pass
882 882 else:
883 883 if i - off >= 0:
884 884 return self.series[i - off]
885 885 plus = patch.rfind('+')
886 886 if plus >= 0:
887 887 res = partial_name(patch[:plus])
888 888 if res:
889 889 i = self.series.index(res)
890 890 try:
891 891 off = int(patch[plus+1:] or 1)
892 892 except(ValueError, OverflowError):
893 893 pass
894 894 else:
895 895 if i + off < len(self.series):
896 896 return self.series[i + off]
897 897 raise util.Abort(_("patch %s not in series") % patch)
898 898
899 899 def push(self, repo, patch=None, force=False, list=False,
900 900 mergeq=None, all=False):
901 901 wlock = repo.wlock()
902 902 try:
903 903 if repo.dirstate.parents()[0] not in repo.heads():
904 904 self.ui.status(_("(working directory not at a head)\n"))
905 905
906 906 if not self.series:
907 907 self.ui.warn(_('no patches in series\n'))
908 908 return 0
909 909
910 910 patch = self.lookup(patch)
911 911 # Suppose our series file is: A B C and the current 'top'
912 912 # patch is B. qpush C should be performed (moving forward)
913 913 # qpush B is a NOP (no change) qpush A is an error (can't
914 914 # go backwards with qpush)
915 915 if patch:
916 916 info = self.isapplied(patch)
917 917 if info:
918 918 if info[0] < len(self.applied) - 1:
919 919 raise util.Abort(
920 920 _("cannot push to a previous patch: %s") % patch)
921 921 self.ui.warn(
922 922 _('qpush: %s is already at the top\n') % patch)
923 923 return
924 924 pushable, reason = self.pushable(patch)
925 925 if not pushable:
926 926 if reason:
927 927 reason = _('guarded by %r') % reason
928 928 else:
929 929 reason = _('no matching guards')
930 930 self.ui.warn(_("cannot push '%s' - %s\n") % (patch, reason))
931 931 return 1
932 932 elif all:
933 933 patch = self.series[-1]
934 934 if self.isapplied(patch):
935 935 self.ui.warn(_('all patches are currently applied\n'))
936 936 return 0
937 937
938 938 # Following the above example, starting at 'top' of B:
939 939 # qpush should be performed (pushes C), but a subsequent
940 940 # qpush without an argument is an error (nothing to
941 941 # apply). This allows a loop of "...while hg qpush..." to
942 942 # work as it detects an error when done
943 943 start = self.series_end()
944 944 if start == len(self.series):
945 945 self.ui.warn(_('patch series already fully applied\n'))
946 946 return 1
947 947 if not force:
948 948 self.check_localchanges(repo)
949 949
950 950 self.applied_dirty = 1
951 951 if start > 0:
952 952 self.check_toppatch(repo)
953 953 if not patch:
954 954 patch = self.series[start]
955 955 end = start + 1
956 956 else:
957 957 end = self.series.index(patch, start) + 1
958 958
959 959 s = self.series[start:end]
960 960 all_files = {}
961 961 try:
962 962 if mergeq:
963 963 ret = self.mergepatch(repo, mergeq, s)
964 964 else:
965 965 ret = self.apply(repo, s, list, all_files=all_files)
966 966 except:
967 967 self.ui.warn(_('cleaning up working directory...'))
968 968 node = repo.dirstate.parents()[0]
969 969 hg.revert(repo, node, None)
970 970 unknown = repo.status(unknown=True)[4]
971 971 # only remove unknown files that we know we touched or
972 972 # created while patching
973 973 for f in unknown:
974 974 if f in all_files:
975 975 util.unlink(repo.wjoin(f))
976 976 self.ui.warn(_('done\n'))
977 977 raise
978 978
979 979 top = self.applied[-1].name
980 980 if ret[0] and ret[0] > 1:
981 981 msg = _("errors during apply, please fix and refresh %s\n")
982 982 self.ui.write(msg % top)
983 983 else:
984 984 self.ui.write(_("now at: %s\n") % top)
985 985 return ret[0]
986 986
987 987 finally:
988 988 wlock.release()
989 989
990 990 def pop(self, repo, patch=None, force=False, update=True, all=False):
991 991 def getfile(f, rev, flags):
992 992 t = repo.file(f).read(rev)
993 993 repo.wwrite(f, t, flags)
994 994
995 995 wlock = repo.wlock()
996 996 try:
997 997 if patch:
998 998 # index, rev, patch
999 999 info = self.isapplied(patch)
1000 1000 if not info:
1001 1001 patch = self.lookup(patch)
1002 1002 info = self.isapplied(patch)
1003 1003 if not info:
1004 1004 raise util.Abort(_("patch %s is not applied") % patch)
1005 1005
1006 1006 if len(self.applied) == 0:
1007 1007 # Allow qpop -a to work repeatedly,
1008 1008 # but not qpop without an argument
1009 1009 self.ui.warn(_("no patches applied\n"))
1010 1010 return not all
1011 1011
1012 1012 if all:
1013 1013 start = 0
1014 1014 elif patch:
1015 1015 start = info[0] + 1
1016 1016 else:
1017 1017 start = len(self.applied) - 1
1018 1018
1019 1019 if start >= len(self.applied):
1020 1020 self.ui.warn(_("qpop: %s is already at the top\n") % patch)
1021 1021 return
1022 1022
1023 1023 if not update:
1024 1024 parents = repo.dirstate.parents()
1025 1025 rr = [ bin(x.rev) for x in self.applied ]
1026 1026 for p in parents:
1027 1027 if p in rr:
1028 1028 self.ui.warn(_("qpop: forcing dirstate update\n"))
1029 1029 update = True
1030 1030 else:
1031 1031 parents = [p.hex() for p in repo[None].parents()]
1032 1032 needupdate = False
1033 1033 for entry in self.applied[start:]:
1034 1034 if entry.rev in parents:
1035 1035 needupdate = True
1036 1036 break
1037 1037 update = needupdate
1038 1038
1039 1039 if not force and update:
1040 1040 self.check_localchanges(repo)
1041 1041
1042 1042 self.applied_dirty = 1
1043 1043 end = len(self.applied)
1044 1044 rev = bin(self.applied[start].rev)
1045 1045 if update:
1046 1046 top = self.check_toppatch(repo)
1047 1047
1048 1048 try:
1049 1049 heads = repo.changelog.heads(rev)
1050 1050 except error.LookupError:
1051 1051 node = short(rev)
1052 1052 raise util.Abort(_('trying to pop unknown node %s') % node)
1053 1053
1054 1054 if heads != [bin(self.applied[-1].rev)]:
1055 1055 raise util.Abort(_("popping would remove a revision not "
1056 1056 "managed by this patch queue"))
1057 1057
1058 1058 # we know there are no local changes, so we can make a simplified
1059 1059 # form of hg.update.
1060 1060 if update:
1061 1061 qp = self.qparents(repo, rev)
1062 1062 changes = repo.changelog.read(qp)
1063 1063 mmap = repo.manifest.read(changes[0])
1064 1064 m, a, r, d = repo.status(qp, top)[:4]
1065 1065 if d:
1066 1066 raise util.Abort(_("deletions found between repo revs"))
1067 1067 for f in m:
1068 1068 getfile(f, mmap[f], mmap.flags(f))
1069 1069 for f in r:
1070 1070 getfile(f, mmap[f], mmap.flags(f))
1071 1071 for f in m + r:
1072 1072 repo.dirstate.normal(f)
1073 1073 for f in a:
1074 1074 try:
1075 1075 os.unlink(repo.wjoin(f))
1076 1076 except OSError, e:
1077 1077 if e.errno != errno.ENOENT:
1078 1078 raise
1079 1079 try: os.removedirs(os.path.dirname(repo.wjoin(f)))
1080 1080 except: pass
1081 1081 repo.dirstate.forget(f)
1082 1082 repo.dirstate.setparents(qp, nullid)
1083 1083 del self.applied[start:end]
1084 1084 self.strip(repo, rev, update=False, backup='strip')
1085 1085 if len(self.applied):
1086 1086 self.ui.write(_("now at: %s\n") % self.applied[-1].name)
1087 1087 else:
1088 1088 self.ui.write(_("patch queue now empty\n"))
1089 1089 finally:
1090 1090 wlock.release()
1091 1091
1092 1092 def diff(self, repo, pats, opts):
1093 1093 top = self.check_toppatch(repo)
1094 1094 if not top:
1095 1095 self.ui.write(_("no patches applied\n"))
1096 1096 return
1097 1097 qp = self.qparents(repo, top)
1098 1098 self._diffopts = patch.diffopts(self.ui, opts)
1099 1099 self.printdiff(repo, qp, files=pats, opts=opts)
1100 1100
1101 1101 def refresh(self, repo, pats=None, **opts):
1102 1102 if len(self.applied) == 0:
1103 1103 self.ui.write(_("no patches applied\n"))
1104 1104 return 1
1105 1105 msg = opts.get('msg', '').rstrip()
1106 1106 newuser = opts.get('user')
1107 1107 newdate = opts.get('date')
1108 1108 if newdate:
1109 1109 newdate = '%d %d' % util.parsedate(newdate)
1110 1110 wlock = repo.wlock()
1111 1111 try:
1112 1112 self.check_toppatch(repo)
1113 1113 (top, patchfn) = (self.applied[-1].rev, self.applied[-1].name)
1114 1114 top = bin(top)
1115 1115 if repo.changelog.heads(top) != [top]:
1116 1116 raise util.Abort(_("cannot refresh a revision with children"))
1117 1117 cparents = repo.changelog.parents(top)
1118 1118 patchparent = self.qparents(repo, top)
1119 1119 ph = patchheader(self.join(patchfn))
1120 1120
1121 1121 patchf = self.opener(patchfn, 'r')
1122 1122
1123 1123 # if the patch was a git patch, refresh it as a git patch
1124 1124 for line in patchf:
1125 1125 if line.startswith('diff --git'):
1126 1126 self.diffopts().git = True
1127 1127 break
1128 1128
1129 1129 if msg:
1130 1130 ph.setmessage(msg)
1131 1131 if newuser:
1132 1132 ph.setuser(newuser)
1133 1133 if newdate:
1134 1134 ph.setdate(newdate)
1135 1135
1136 1136 # only commit new patch when write is complete
1137 1137 patchf = self.opener(patchfn, 'w', atomictemp=True)
1138 1138
1139 1139 patchf.seek(0)
1140 1140 patchf.truncate()
1141 1141
1142 1142 comments = str(ph)
1143 1143 if comments:
1144 1144 patchf.write(comments)
1145 1145
1146 1146 if opts.get('git'):
1147 1147 self.diffopts().git = True
1148 1148 tip = repo.changelog.tip()
1149 1149 if top == tip:
1150 1150 # if the top of our patch queue is also the tip, there is an
1151 1151 # optimization here. We update the dirstate in place and strip
1152 1152 # off the tip commit. Then just commit the current directory
1153 1153 # tree. We can also send repo.commit the list of files
1154 1154 # changed to speed up the diff
1155 1155 #
1156 1156 # in short mode, we only diff the files included in the
1157 1157 # patch already plus specified files
1158 1158 #
1159 1159 # this should really read:
1160 1160 # mm, dd, aa, aa2 = repo.status(tip, patchparent)[:4]
1161 1161 # but we do it backwards to take advantage of manifest/chlog
1162 1162 # caching against the next repo.status call
1163 1163 #
1164 1164 mm, aa, dd, aa2 = repo.status(patchparent, tip)[:4]
1165 1165 changes = repo.changelog.read(tip)
1166 1166 man = repo.manifest.read(changes[0])
1167 1167 aaa = aa[:]
1168 1168 matchfn = cmdutil.match(repo, pats, opts)
1169 1169 if opts.get('short'):
1170 1170 # if amending a patch, we start with existing
1171 1171 # files plus specified files - unfiltered
1172 1172 match = cmdutil.matchfiles(repo, mm + aa + dd + matchfn.files())
1173 1173 # filter with inc/exl options
1174 1174 matchfn = cmdutil.match(repo, opts=opts)
1175 1175 else:
1176 1176 match = cmdutil.matchall(repo)
1177 1177 m, a, r, d = repo.status(match=match)[:4]
1178 1178
1179 1179 # we might end up with files that were added between
1180 1180 # tip and the dirstate parent, but then changed in the
1181 1181 # local dirstate. in this case, we want them to only
1182 1182 # show up in the added section
1183 1183 for x in m:
1184 1184 if x not in aa:
1185 1185 mm.append(x)
1186 1186 # we might end up with files added by the local dirstate that
1187 1187 # were deleted by the patch. In this case, they should only
1188 1188 # show up in the changed section.
1189 1189 for x in a:
1190 1190 if x in dd:
1191 1191 del dd[dd.index(x)]
1192 1192 mm.append(x)
1193 1193 else:
1194 1194 aa.append(x)
1195 1195 # make sure any files deleted in the local dirstate
1196 1196 # are not in the add or change column of the patch
1197 1197 forget = []
1198 1198 for x in d + r:
1199 1199 if x in aa:
1200 1200 del aa[aa.index(x)]
1201 1201 forget.append(x)
1202 1202 continue
1203 1203 elif x in mm:
1204 1204 del mm[mm.index(x)]
1205 1205 dd.append(x)
1206 1206
1207 1207 m = list(set(mm))
1208 1208 r = list(set(dd))
1209 1209 a = list(set(aa))
1210 1210 c = [filter(matchfn, l) for l in (m, a, r)]
1211 1211 match = cmdutil.matchfiles(repo, set(c[0] + c[1] + c[2]))
1212 1212 chunks = patch.diff(repo, patchparent, match=match,
1213 1213 changes=c, opts=self.diffopts())
1214 1214 for chunk in chunks:
1215 1215 patchf.write(chunk)
1216 1216
1217 1217 try:
1218 1218 if self.diffopts().git:
1219 1219 copies = {}
1220 1220 for dst in a:
1221 1221 src = repo.dirstate.copied(dst)
1222 1222 # during qfold, the source file for copies may
1223 1223 # be removed. Treat this as a simple add.
1224 1224 if src is not None and src in repo.dirstate:
1225 1225 copies.setdefault(src, []).append(dst)
1226 1226 repo.dirstate.add(dst)
1227 1227 # remember the copies between patchparent and tip
1228 1228 for dst in aaa:
1229 1229 f = repo.file(dst)
1230 1230 src = f.renamed(man[dst])
1231 1231 if src:
1232 1232 copies.setdefault(src[0], []).extend(copies.get(dst, []))
1233 1233 if dst in a:
1234 1234 copies[src[0]].append(dst)
1235 1235 # we can't copy a file created by the patch itself
1236 1236 if dst in copies:
1237 1237 del copies[dst]
1238 1238 for src, dsts in copies.iteritems():
1239 1239 for dst in dsts:
1240 1240 repo.dirstate.copy(src, dst)
1241 1241 else:
1242 1242 for dst in a:
1243 1243 repo.dirstate.add(dst)
1244 1244 # Drop useless copy information
1245 1245 for f in list(repo.dirstate.copies()):
1246 1246 repo.dirstate.copy(None, f)
1247 1247 for f in r:
1248 1248 repo.dirstate.remove(f)
1249 1249 # if the patch excludes a modified file, mark that
1250 1250 # file with mtime=0 so status can see it.
1251 1251 mm = []
1252 1252 for i in xrange(len(m)-1, -1, -1):
1253 1253 if not matchfn(m[i]):
1254 1254 mm.append(m[i])
1255 1255 del m[i]
1256 1256 for f in m:
1257 1257 repo.dirstate.normal(f)
1258 1258 for f in mm:
1259 1259 repo.dirstate.normallookup(f)
1260 1260 for f in forget:
1261 1261 repo.dirstate.forget(f)
1262 1262
1263 1263 if not msg:
1264 1264 if not ph.message:
1265 1265 message = "[mq]: %s\n" % patchfn
1266 1266 else:
1267 1267 message = "\n".join(ph.message)
1268 1268 else:
1269 1269 message = msg
1270 1270
1271 1271 user = ph.user or changes[1]
1272 1272
1273 1273 # assumes strip can roll itself back if interrupted
1274 1274 repo.dirstate.setparents(*cparents)
1275 1275 self.applied.pop()
1276 1276 self.applied_dirty = 1
1277 1277 self.strip(repo, top, update=False,
1278 1278 backup='strip')
1279 1279 except:
1280 1280 repo.dirstate.invalidate()
1281 1281 raise
1282 1282
1283 1283 try:
1284 1284 # might be nice to attempt to roll back strip after this
1285 1285 patchf.rename()
1286 1286 n = repo.commit(message, user, ph.date, match=match,
1287 1287 force=True)
1288 1288 self.applied.append(statusentry(hex(n), patchfn))
1289 1289 except:
1290 1290 ctx = repo[cparents[0]]
1291 1291 repo.dirstate.rebuild(ctx.node(), ctx.manifest())
1292 1292 self.save_dirty()
1293 1293 self.ui.warn(_('refresh interrupted while patch was popped! '
1294 1294 '(revert --all, qpush to recover)\n'))
1295 1295 raise
1296 1296 else:
1297 1297 self.printdiff(repo, patchparent, fp=patchf)
1298 1298 patchf.rename()
1299 1299 added = repo.status()[1]
1300 1300 for a in added:
1301 1301 f = repo.wjoin(a)
1302 1302 try:
1303 1303 os.unlink(f)
1304 1304 except OSError, e:
1305 1305 if e.errno != errno.ENOENT:
1306 1306 raise
1307 1307 try: os.removedirs(os.path.dirname(f))
1308 1308 except: pass
1309 1309 # forget the file copies in the dirstate
1310 1310 # push should readd the files later on
1311 1311 repo.dirstate.forget(a)
1312 1312 self.pop(repo, force=True)
1313 1313 self.push(repo, force=True)
1314 1314 finally:
1315 1315 wlock.release()
1316 1316 self.removeundo(repo)
1317 1317
1318 1318 def init(self, repo, create=False):
1319 1319 if not create and os.path.isdir(self.path):
1320 1320 raise util.Abort(_("patch queue directory already exists"))
1321 1321 try:
1322 1322 os.mkdir(self.path)
1323 1323 except OSError, inst:
1324 1324 if inst.errno != errno.EEXIST or not create:
1325 1325 raise
1326 1326 if create:
1327 1327 return self.qrepo(create=True)
1328 1328
1329 1329 def unapplied(self, repo, patch=None):
1330 1330 if patch and patch not in self.series:
1331 1331 raise util.Abort(_("patch %s is not in series file") % patch)
1332 1332 if not patch:
1333 1333 start = self.series_end()
1334 1334 else:
1335 1335 start = self.series.index(patch) + 1
1336 1336 unapplied = []
1337 1337 for i in xrange(start, len(self.series)):
1338 1338 pushable, reason = self.pushable(i)
1339 1339 if pushable:
1340 1340 unapplied.append((i, self.series[i]))
1341 1341 self.explain_pushable(i)
1342 1342 return unapplied
1343 1343
1344 1344 def qseries(self, repo, missing=None, start=0, length=None, status=None,
1345 1345 summary=False):
1346 def displayname(patchname):
1346 def displayname(pfx, patchname):
1347 1347 if summary:
1348 1348 ph = patchheader(self.join(patchname))
1349 1349 msg = ph.message
1350 1350 msg = msg and ': ' + msg[0] or ': '
1351 1351 else:
1352 1352 msg = ''
1353 return '%s%s' % (patchname, msg)
1353 msg = "%s%s%s" % (pfx, patchname, msg)
1354 if self.ui.interactive():
1355 msg = util.ellipsis(msg, util.termwidth())
1356 self.ui.write(msg + '\n')
1354 1357
1355 1358 applied = set([p.name for p in self.applied])
1356 1359 if length is None:
1357 1360 length = len(self.series) - start
1358 1361 if not missing:
1362 if self.ui.verbose:
1363 idxwidth = len(str(start+length - 1))
1359 1364 for i in xrange(start, start+length):
1360 1365 patch = self.series[i]
1361 1366 if patch in applied:
1362 1367 stat = 'A'
1363 1368 elif self.pushable(i)[0]:
1364 1369 stat = 'U'
1365 1370 else:
1366 1371 stat = 'G'
1367 1372 pfx = ''
1368 1373 if self.ui.verbose:
1369 pfx = '%d %s ' % (i, stat)
1374 pfx = '%*d %s ' % (idxwidth, i, stat)
1370 1375 elif status and status != stat:
1371 1376 continue
1372 self.ui.write('%s%s\n' % (pfx, displayname(patch)))
1377 displayname(pfx, patch)
1373 1378 else:
1374 1379 msng_list = []
1375 1380 for root, dirs, files in os.walk(self.path):
1376 1381 d = root[len(self.path) + 1:]
1377 1382 for f in files:
1378 1383 fl = os.path.join(d, f)
1379 1384 if (fl not in self.series and
1380 1385 fl not in (self.status_path, self.series_path,
1381 1386 self.guards_path)
1382 1387 and not fl.startswith('.')):
1383 1388 msng_list.append(fl)
1384 1389 for x in sorted(msng_list):
1385 1390 pfx = self.ui.verbose and ('D ') or ''
1386 self.ui.write("%s%s\n" % (pfx, displayname(x)))
1391 displayname(pfx, x)
1387 1392
1388 1393 def issaveline(self, l):
1389 1394 if l.name == '.hg.patches.save.line':
1390 1395 return True
1391 1396
1392 1397 def qrepo(self, create=False):
1393 1398 if create or os.path.isdir(self.join(".hg")):
1394 1399 return hg.repository(self.ui, path=self.path, create=create)
1395 1400
1396 1401 def restore(self, repo, rev, delete=None, qupdate=None):
1397 1402 c = repo.changelog.read(rev)
1398 1403 desc = c[4].strip()
1399 1404 lines = desc.splitlines()
1400 1405 i = 0
1401 1406 datastart = None
1402 1407 series = []
1403 1408 applied = []
1404 1409 qpp = None
1405 1410 for i, line in enumerate(lines):
1406 1411 if line == 'Patch Data:':
1407 1412 datastart = i + 1
1408 1413 elif line.startswith('Dirstate:'):
1409 1414 l = line.rstrip()
1410 1415 l = l[10:].split(' ')
1411 1416 qpp = [ bin(x) for x in l ]
1412 1417 elif datastart != None:
1413 1418 l = line.rstrip()
1414 1419 se = statusentry(l)
1415 1420 file_ = se.name
1416 1421 if se.rev:
1417 1422 applied.append(se)
1418 1423 else:
1419 1424 series.append(file_)
1420 1425 if datastart is None:
1421 1426 self.ui.warn(_("No saved patch data found\n"))
1422 1427 return 1
1423 1428 self.ui.warn(_("restoring status: %s\n") % lines[0])
1424 1429 self.full_series = series
1425 1430 self.applied = applied
1426 1431 self.parse_series()
1427 1432 self.series_dirty = 1
1428 1433 self.applied_dirty = 1
1429 1434 heads = repo.changelog.heads()
1430 1435 if delete:
1431 1436 if rev not in heads:
1432 1437 self.ui.warn(_("save entry has children, leaving it alone\n"))
1433 1438 else:
1434 1439 self.ui.warn(_("removing save entry %s\n") % short(rev))
1435 1440 pp = repo.dirstate.parents()
1436 1441 if rev in pp:
1437 1442 update = True
1438 1443 else:
1439 1444 update = False
1440 1445 self.strip(repo, rev, update=update, backup='strip')
1441 1446 if qpp:
1442 1447 self.ui.warn(_("saved queue repository parents: %s %s\n") %
1443 1448 (short(qpp[0]), short(qpp[1])))
1444 1449 if qupdate:
1445 1450 self.ui.status(_("queue directory updating\n"))
1446 1451 r = self.qrepo()
1447 1452 if not r:
1448 1453 self.ui.warn(_("Unable to load queue repository\n"))
1449 1454 return 1
1450 1455 hg.clean(r, qpp[0])
1451 1456
1452 1457 def save(self, repo, msg=None):
1453 1458 if len(self.applied) == 0:
1454 1459 self.ui.warn(_("save: no patches applied, exiting\n"))
1455 1460 return 1
1456 1461 if self.issaveline(self.applied[-1]):
1457 1462 self.ui.warn(_("status is already saved\n"))
1458 1463 return 1
1459 1464
1460 1465 ar = [ ':' + x for x in self.full_series ]
1461 1466 if not msg:
1462 1467 msg = _("hg patches saved state")
1463 1468 else:
1464 1469 msg = "hg patches: " + msg.rstrip('\r\n')
1465 1470 r = self.qrepo()
1466 1471 if r:
1467 1472 pp = r.dirstate.parents()
1468 1473 msg += "\nDirstate: %s %s" % (hex(pp[0]), hex(pp[1]))
1469 1474 msg += "\n\nPatch Data:\n"
1470 1475 text = msg + "\n".join([str(x) for x in self.applied]) + '\n' + (ar and
1471 1476 "\n".join(ar) + '\n' or "")
1472 1477 n = repo.commit(text, force=True)
1473 1478 if not n:
1474 1479 self.ui.warn(_("repo commit failed\n"))
1475 1480 return 1
1476 1481 self.applied.append(statusentry(hex(n),'.hg.patches.save.line'))
1477 1482 self.applied_dirty = 1
1478 1483 self.removeundo(repo)
1479 1484
1480 1485 def full_series_end(self):
1481 1486 if len(self.applied) > 0:
1482 1487 p = self.applied[-1].name
1483 1488 end = self.find_series(p)
1484 1489 if end is None:
1485 1490 return len(self.full_series)
1486 1491 return end + 1
1487 1492 return 0
1488 1493
1489 1494 def series_end(self, all_patches=False):
1490 1495 """If all_patches is False, return the index of the next pushable patch
1491 1496 in the series, or the series length. If all_patches is True, return the
1492 1497 index of the first patch past the last applied one.
1493 1498 """
1494 1499 end = 0
1495 1500 def next(start):
1496 1501 if all_patches:
1497 1502 return start
1498 1503 i = start
1499 1504 while i < len(self.series):
1500 1505 p, reason = self.pushable(i)
1501 1506 if p:
1502 1507 break
1503 1508 self.explain_pushable(i)
1504 1509 i += 1
1505 1510 return i
1506 1511 if len(self.applied) > 0:
1507 1512 p = self.applied[-1].name
1508 1513 try:
1509 1514 end = self.series.index(p)
1510 1515 except ValueError:
1511 1516 return 0
1512 1517 return next(end + 1)
1513 1518 return next(end)
1514 1519
1515 1520 def appliedname(self, index):
1516 1521 pname = self.applied[index].name
1517 1522 if not self.ui.verbose:
1518 1523 p = pname
1519 1524 else:
1520 1525 p = str(self.series.index(pname)) + " " + pname
1521 1526 return p
1522 1527
1523 1528 def qimport(self, repo, files, patchname=None, rev=None, existing=None,
1524 1529 force=None, git=False):
1525 1530 def checkseries(patchname):
1526 1531 if patchname in self.series:
1527 1532 raise util.Abort(_('patch %s is already in the series file')
1528 1533 % patchname)
1529 1534 def checkfile(patchname):
1530 1535 if not force and os.path.exists(self.join(patchname)):
1531 1536 raise util.Abort(_('patch "%s" already exists')
1532 1537 % patchname)
1533 1538
1534 1539 if rev:
1535 1540 if files:
1536 1541 raise util.Abort(_('option "-r" not valid when importing '
1537 1542 'files'))
1538 1543 rev = cmdutil.revrange(repo, rev)
1539 1544 rev.sort(lambda x, y: cmp(y, x))
1540 1545 if (len(files) > 1 or len(rev) > 1) and patchname:
1541 1546 raise util.Abort(_('option "-n" not valid when importing multiple '
1542 1547 'patches'))
1543 1548 i = 0
1544 1549 added = []
1545 1550 if rev:
1546 1551 # If mq patches are applied, we can only import revisions
1547 1552 # that form a linear path to qbase.
1548 1553 # Otherwise, they should form a linear path to a head.
1549 1554 heads = repo.changelog.heads(repo.changelog.node(rev[-1]))
1550 1555 if len(heads) > 1:
1551 1556 raise util.Abort(_('revision %d is the root of more than one '
1552 1557 'branch') % rev[-1])
1553 1558 if self.applied:
1554 1559 base = hex(repo.changelog.node(rev[0]))
1555 1560 if base in [n.rev for n in self.applied]:
1556 1561 raise util.Abort(_('revision %d is already managed')
1557 1562 % rev[0])
1558 1563 if heads != [bin(self.applied[-1].rev)]:
1559 1564 raise util.Abort(_('revision %d is not the parent of '
1560 1565 'the queue') % rev[0])
1561 1566 base = repo.changelog.rev(bin(self.applied[0].rev))
1562 1567 lastparent = repo.changelog.parentrevs(base)[0]
1563 1568 else:
1564 1569 if heads != [repo.changelog.node(rev[0])]:
1565 1570 raise util.Abort(_('revision %d has unmanaged children')
1566 1571 % rev[0])
1567 1572 lastparent = None
1568 1573
1569 1574 if git:
1570 1575 self.diffopts().git = True
1571 1576
1572 1577 for r in rev:
1573 1578 p1, p2 = repo.changelog.parentrevs(r)
1574 1579 n = repo.changelog.node(r)
1575 1580 if p2 != nullrev:
1576 1581 raise util.Abort(_('cannot import merge revision %d') % r)
1577 1582 if lastparent and lastparent != r:
1578 1583 raise util.Abort(_('revision %d is not the parent of %d')
1579 1584 % (r, lastparent))
1580 1585 lastparent = p1
1581 1586
1582 1587 if not patchname:
1583 1588 patchname = normname('%d.diff' % r)
1584 1589 self.check_reserved_name(patchname)
1585 1590 checkseries(patchname)
1586 1591 checkfile(patchname)
1587 1592 self.full_series.insert(0, patchname)
1588 1593
1589 1594 patchf = self.opener(patchname, "w")
1590 1595 patch.export(repo, [n], fp=patchf, opts=self.diffopts())
1591 1596 patchf.close()
1592 1597
1593 1598 se = statusentry(hex(n), patchname)
1594 1599 self.applied.insert(0, se)
1595 1600
1596 1601 added.append(patchname)
1597 1602 patchname = None
1598 1603 self.parse_series()
1599 1604 self.applied_dirty = 1
1600 1605
1601 1606 for filename in files:
1602 1607 if existing:
1603 1608 if filename == '-':
1604 1609 raise util.Abort(_('-e is incompatible with import from -'))
1605 1610 if not patchname:
1606 1611 patchname = normname(filename)
1607 1612 self.check_reserved_name(patchname)
1608 1613 if not os.path.isfile(self.join(patchname)):
1609 1614 raise util.Abort(_("patch %s does not exist") % patchname)
1610 1615 else:
1611 1616 try:
1612 1617 if filename == '-':
1613 1618 if not patchname:
1614 1619 raise util.Abort(_('need --name to import a patch from -'))
1615 1620 text = sys.stdin.read()
1616 1621 else:
1617 1622 text = url.open(self.ui, filename).read()
1618 1623 except (OSError, IOError):
1619 1624 raise util.Abort(_("unable to read %s") % filename)
1620 1625 if not patchname:
1621 1626 patchname = normname(os.path.basename(filename))
1622 1627 self.check_reserved_name(patchname)
1623 1628 checkfile(patchname)
1624 1629 patchf = self.opener(patchname, "w")
1625 1630 patchf.write(text)
1626 1631 if not force:
1627 1632 checkseries(patchname)
1628 1633 if patchname not in self.series:
1629 1634 index = self.full_series_end() + i
1630 1635 self.full_series[index:index] = [patchname]
1631 1636 self.parse_series()
1632 1637 self.ui.warn(_("adding %s to series file\n") % patchname)
1633 1638 i += 1
1634 1639 added.append(patchname)
1635 1640 patchname = None
1636 1641 self.series_dirty = 1
1637 1642 qrepo = self.qrepo()
1638 1643 if qrepo:
1639 1644 qrepo.add(added)
1640 1645
1641 1646 def delete(ui, repo, *patches, **opts):
1642 1647 """remove patches from queue
1643 1648
1644 1649 The patches must not be applied, and at least one patch is required. With
1645 1650 -k/--keep, the patch files are preserved in the patch directory.
1646 1651
1647 1652 To stop managing a patch and move it into permanent history,
1648 1653 use the qfinish command."""
1649 1654 q = repo.mq
1650 1655 q.delete(repo, patches, opts)
1651 1656 q.save_dirty()
1652 1657 return 0
1653 1658
1654 1659 def applied(ui, repo, patch=None, **opts):
1655 1660 """print the patches already applied"""
1656 1661 q = repo.mq
1657 1662 if patch:
1658 1663 if patch not in q.series:
1659 1664 raise util.Abort(_("patch %s is not in series file") % patch)
1660 1665 end = q.series.index(patch) + 1
1661 1666 else:
1662 1667 end = q.series_end(True)
1663 1668 return q.qseries(repo, length=end, status='A', summary=opts.get('summary'))
1664 1669
1665 1670 def unapplied(ui, repo, patch=None, **opts):
1666 1671 """print the patches not yet applied"""
1667 1672 q = repo.mq
1668 1673 if patch:
1669 1674 if patch not in q.series:
1670 1675 raise util.Abort(_("patch %s is not in series file") % patch)
1671 1676 start = q.series.index(patch) + 1
1672 1677 else:
1673 1678 start = q.series_end(True)
1674 1679 q.qseries(repo, start=start, status='U', summary=opts.get('summary'))
1675 1680
1676 1681 def qimport(ui, repo, *filename, **opts):
1677 1682 """import a patch
1678 1683
1679 1684 The patch is inserted into the series after the last applied
1680 1685 patch. If no patches have been applied, qimport prepends the patch
1681 1686 to the series.
1682 1687
1683 1688 The patch will have the same name as its source file unless you
1684 1689 give it a new one with -n/--name.
1685 1690
1686 1691 You can register an existing patch inside the patch directory with
1687 1692 the -e/--existing flag.
1688 1693
1689 1694 With -f/--force, an existing patch of the same name will be
1690 1695 overwritten.
1691 1696
1692 1697 An existing changeset may be placed under mq control with -r/--rev
1693 1698 (e.g. qimport --rev tip -n patch will place tip under mq control).
1694 1699 With -g/--git, patches imported with --rev will use the git diff
1695 1700 format. See the diffs help topic for information on why this is
1696 1701 important for preserving rename/copy information and permission
1697 1702 changes.
1698 1703
1699 1704 To import a patch from standard input, pass - as the patch file.
1700 1705 When importing from standard input, a patch name must be specified
1701 1706 using the --name flag.
1702 1707 """
1703 1708 q = repo.mq
1704 1709 q.qimport(repo, filename, patchname=opts['name'],
1705 1710 existing=opts['existing'], force=opts['force'], rev=opts['rev'],
1706 1711 git=opts['git'])
1707 1712 q.save_dirty()
1708 1713
1709 1714 if opts.get('push') and not opts.get('rev'):
1710 1715 return q.push(repo, None)
1711 1716 return 0
1712 1717
1713 1718 def init(ui, repo, **opts):
1714 1719 """init a new queue repository
1715 1720
1716 1721 The queue repository is unversioned by default. If
1717 1722 -c/--create-repo is specified, qinit will create a separate nested
1718 1723 repository for patches (qinit -c may also be run later to convert
1719 1724 an unversioned patch repository into a versioned one). You can use
1720 1725 qcommit to commit changes to this queue repository."""
1721 1726 q = repo.mq
1722 1727 r = q.init(repo, create=opts['create_repo'])
1723 1728 q.save_dirty()
1724 1729 if r:
1725 1730 if not os.path.exists(r.wjoin('.hgignore')):
1726 1731 fp = r.wopener('.hgignore', 'w')
1727 1732 fp.write('^\\.hg\n')
1728 1733 fp.write('^\\.mq\n')
1729 1734 fp.write('syntax: glob\n')
1730 1735 fp.write('status\n')
1731 1736 fp.write('guards\n')
1732 1737 fp.close()
1733 1738 if not os.path.exists(r.wjoin('series')):
1734 1739 r.wopener('series', 'w').close()
1735 1740 r.add(['.hgignore', 'series'])
1736 1741 commands.add(ui, r)
1737 1742 return 0
1738 1743
1739 1744 def clone(ui, source, dest=None, **opts):
1740 1745 '''clone main and patch repository at same time
1741 1746
1742 1747 If source is local, destination will have no patches applied. If
1743 1748 source is remote, this command can not check if patches are
1744 1749 applied in source, so cannot guarantee that patches are not
1745 1750 applied in destination. If you clone remote repository, be sure
1746 1751 before that it has no patches applied.
1747 1752
1748 1753 Source patch repository is looked for in <src>/.hg/patches by
1749 1754 default. Use -p <url> to change.
1750 1755
1751 1756 The patch directory must be a nested Mercurial repository, as
1752 1757 would be created by qinit -c.
1753 1758 '''
1754 1759 def patchdir(repo):
1755 1760 url = repo.url()
1756 1761 if url.endswith('/'):
1757 1762 url = url[:-1]
1758 1763 return url + '/.hg/patches'
1759 1764 if dest is None:
1760 1765 dest = hg.defaultdest(source)
1761 1766 sr = hg.repository(cmdutil.remoteui(ui, opts), ui.expandpath(source))
1762 1767 if opts['patches']:
1763 1768 patchespath = ui.expandpath(opts['patches'])
1764 1769 else:
1765 1770 patchespath = patchdir(sr)
1766 1771 try:
1767 1772 hg.repository(ui, patchespath)
1768 1773 except error.RepoError:
1769 1774 raise util.Abort(_('versioned patch repository not found'
1770 1775 ' (see qinit -c)'))
1771 1776 qbase, destrev = None, None
1772 1777 if sr.local():
1773 1778 if sr.mq.applied:
1774 1779 qbase = bin(sr.mq.applied[0].rev)
1775 1780 if not hg.islocal(dest):
1776 1781 heads = set(sr.heads())
1777 1782 destrev = list(heads.difference(sr.heads(qbase)))
1778 1783 destrev.append(sr.changelog.parents(qbase)[0])
1779 1784 elif sr.capable('lookup'):
1780 1785 try:
1781 1786 qbase = sr.lookup('qbase')
1782 1787 except error.RepoError:
1783 1788 pass
1784 1789 ui.note(_('cloning main repository\n'))
1785 1790 sr, dr = hg.clone(ui, sr.url(), dest,
1786 1791 pull=opts['pull'],
1787 1792 rev=destrev,
1788 1793 update=False,
1789 1794 stream=opts['uncompressed'])
1790 1795 ui.note(_('cloning patch repository\n'))
1791 1796 hg.clone(ui, opts['patches'] or patchdir(sr), patchdir(dr),
1792 1797 pull=opts['pull'], update=not opts['noupdate'],
1793 1798 stream=opts['uncompressed'])
1794 1799 if dr.local():
1795 1800 if qbase:
1796 1801 ui.note(_('stripping applied patches from destination '
1797 1802 'repository\n'))
1798 1803 dr.mq.strip(dr, qbase, update=False, backup=None)
1799 1804 if not opts['noupdate']:
1800 1805 ui.note(_('updating destination repository\n'))
1801 1806 hg.update(dr, dr.changelog.tip())
1802 1807
1803 1808 def commit(ui, repo, *pats, **opts):
1804 1809 """commit changes in the queue repository"""
1805 1810 q = repo.mq
1806 1811 r = q.qrepo()
1807 1812 if not r: raise util.Abort('no queue repository')
1808 1813 commands.commit(r.ui, r, *pats, **opts)
1809 1814
1810 1815 def series(ui, repo, **opts):
1811 1816 """print the entire series file"""
1812 1817 repo.mq.qseries(repo, missing=opts['missing'], summary=opts['summary'])
1813 1818 return 0
1814 1819
1815 1820 def top(ui, repo, **opts):
1816 1821 """print the name of the current patch"""
1817 1822 q = repo.mq
1818 1823 t = q.applied and q.series_end(True) or 0
1819 1824 if t:
1820 1825 return q.qseries(repo, start=t-1, length=1, status='A',
1821 1826 summary=opts.get('summary'))
1822 1827 else:
1823 1828 ui.write(_("no patches applied\n"))
1824 1829 return 1
1825 1830
1826 1831 def next(ui, repo, **opts):
1827 1832 """print the name of the next patch"""
1828 1833 q = repo.mq
1829 1834 end = q.series_end()
1830 1835 if end == len(q.series):
1831 1836 ui.write(_("all patches applied\n"))
1832 1837 return 1
1833 1838 return q.qseries(repo, start=end, length=1, summary=opts.get('summary'))
1834 1839
1835 1840 def prev(ui, repo, **opts):
1836 1841 """print the name of the previous patch"""
1837 1842 q = repo.mq
1838 1843 l = len(q.applied)
1839 1844 if l == 1:
1840 1845 ui.write(_("only one patch applied\n"))
1841 1846 return 1
1842 1847 if not l:
1843 1848 ui.write(_("no patches applied\n"))
1844 1849 return 1
1845 1850 return q.qseries(repo, start=l-2, length=1, status='A',
1846 1851 summary=opts.get('summary'))
1847 1852
1848 1853 def setupheaderopts(ui, opts):
1849 1854 def do(opt,val):
1850 1855 if not opts[opt] and opts['current' + opt]:
1851 1856 opts[opt] = val
1852 1857 do('user', ui.username())
1853 1858 do('date', "%d %d" % util.makedate())
1854 1859
1855 1860 def new(ui, repo, patch, *args, **opts):
1856 1861 """create a new patch
1857 1862
1858 1863 qnew creates a new patch on top of the currently-applied patch (if
1859 1864 any). It will refuse to run if there are any outstanding changes
1860 1865 unless -f/--force is specified, in which case the patch will be
1861 1866 initialized with them. You may also use -I/--include,
1862 1867 -X/--exclude, and/or a list of files after the patch name to add
1863 1868 only changes to matching files to the new patch, leaving the rest
1864 1869 as uncommitted modifications.
1865 1870
1866 1871 -u/--user and -d/--date can be used to set the (given) user and
1867 1872 date, respectively. -U/--currentuser and -D/--currentdate set user
1868 1873 to current user and date to current date.
1869 1874
1870 1875 -e/--edit, -m/--message or -l/--logfile set the patch header as
1871 1876 well as the commit message. If none is specified, the header is
1872 1877 empty and the commit message is '[mq]: PATCH'.
1873 1878
1874 1879 Use the -g/--git option to keep the patch in the git extended diff
1875 1880 format. Read the diffs help topic for more information on why this
1876 1881 is important for preserving permission changes and copy/rename
1877 1882 information.
1878 1883 """
1879 1884 msg = cmdutil.logmessage(opts)
1880 1885 def getmsg(): return ui.edit(msg, ui.username())
1881 1886 q = repo.mq
1882 1887 opts['msg'] = msg
1883 1888 if opts.get('edit'):
1884 1889 opts['msg'] = getmsg
1885 1890 else:
1886 1891 opts['msg'] = msg
1887 1892 setupheaderopts(ui, opts)
1888 1893 q.new(repo, patch, *args, **opts)
1889 1894 q.save_dirty()
1890 1895 return 0
1891 1896
1892 1897 def refresh(ui, repo, *pats, **opts):
1893 1898 """update the current patch
1894 1899
1895 1900 If any file patterns are provided, the refreshed patch will
1896 1901 contain only the modifications that match those patterns; the
1897 1902 remaining modifications will remain in the working directory.
1898 1903
1899 1904 If -s/--short is specified, files currently included in the patch
1900 1905 will be refreshed just like matched files and remain in the patch.
1901 1906
1902 1907 hg add/remove/copy/rename work as usual, though you might want to
1903 1908 use git-style patches (-g/--git or [diff] git=1) to track copies
1904 1909 and renames. See the diffs help topic for more information on the
1905 1910 git diff format.
1906 1911 """
1907 1912 q = repo.mq
1908 1913 message = cmdutil.logmessage(opts)
1909 1914 if opts['edit']:
1910 1915 if not q.applied:
1911 1916 ui.write(_("no patches applied\n"))
1912 1917 return 1
1913 1918 if message:
1914 1919 raise util.Abort(_('option "-e" incompatible with "-m" or "-l"'))
1915 1920 patch = q.applied[-1].name
1916 1921 ph = patchheader(q.join(patch))
1917 1922 message = ui.edit('\n'.join(ph.message), ph.user or ui.username())
1918 1923 setupheaderopts(ui, opts)
1919 1924 ret = q.refresh(repo, pats, msg=message, **opts)
1920 1925 q.save_dirty()
1921 1926 return ret
1922 1927
1923 1928 def diff(ui, repo, *pats, **opts):
1924 1929 """diff of the current patch and subsequent modifications
1925 1930
1926 1931 Shows a diff which includes the current patch as well as any
1927 1932 changes which have been made in the working directory since the
1928 1933 last refresh (thus showing what the current patch would become
1929 1934 after a qrefresh).
1930 1935
1931 1936 Use 'hg diff' if you only want to see the changes made since the
1932 1937 last qrefresh, or 'hg export qtip' if you want to see changes made
1933 1938 by the current patch without including changes made since the
1934 1939 qrefresh.
1935 1940 """
1936 1941 repo.mq.diff(repo, pats, opts)
1937 1942 return 0
1938 1943
1939 1944 def fold(ui, repo, *files, **opts):
1940 1945 """fold the named patches into the current patch
1941 1946
1942 1947 Patches must not yet be applied. Each patch will be successively
1943 1948 applied to the current patch in the order given. If all the
1944 1949 patches apply successfully, the current patch will be refreshed
1945 1950 with the new cumulative patch, and the folded patches will be
1946 1951 deleted. With -k/--keep, the folded patch files will not be
1947 1952 removed afterwards.
1948 1953
1949 1954 The header for each folded patch will be concatenated with the
1950 1955 current patch header, separated by a line of '* * *'."""
1951 1956
1952 1957 q = repo.mq
1953 1958
1954 1959 if not files:
1955 1960 raise util.Abort(_('qfold requires at least one patch name'))
1956 1961 if not q.check_toppatch(repo):
1957 1962 raise util.Abort(_('No patches applied'))
1958 1963 q.check_localchanges(repo)
1959 1964
1960 1965 message = cmdutil.logmessage(opts)
1961 1966 if opts['edit']:
1962 1967 if message:
1963 1968 raise util.Abort(_('option "-e" incompatible with "-m" or "-l"'))
1964 1969
1965 1970 parent = q.lookup('qtip')
1966 1971 patches = []
1967 1972 messages = []
1968 1973 for f in files:
1969 1974 p = q.lookup(f)
1970 1975 if p in patches or p == parent:
1971 1976 ui.warn(_('Skipping already folded patch %s') % p)
1972 1977 if q.isapplied(p):
1973 1978 raise util.Abort(_('qfold cannot fold already applied patch %s') % p)
1974 1979 patches.append(p)
1975 1980
1976 1981 for p in patches:
1977 1982 if not message:
1978 1983 ph = patchheader(q.join(p))
1979 1984 if ph.message:
1980 1985 messages.append(ph.message)
1981 1986 pf = q.join(p)
1982 1987 (patchsuccess, files, fuzz) = q.patch(repo, pf)
1983 1988 if not patchsuccess:
1984 1989 raise util.Abort(_('Error folding patch %s') % p)
1985 1990 patch.updatedir(ui, repo, files)
1986 1991
1987 1992 if not message:
1988 1993 ph = patchheader(q.join(parent))
1989 1994 message, user = ph.message, ph.user
1990 1995 for msg in messages:
1991 1996 message.append('* * *')
1992 1997 message.extend(msg)
1993 1998 message = '\n'.join(message)
1994 1999
1995 2000 if opts['edit']:
1996 2001 message = ui.edit(message, user or ui.username())
1997 2002
1998 2003 q.refresh(repo, msg=message)
1999 2004 q.delete(repo, patches, opts)
2000 2005 q.save_dirty()
2001 2006
2002 2007 def goto(ui, repo, patch, **opts):
2003 2008 '''push or pop patches until named patch is at top of stack'''
2004 2009 q = repo.mq
2005 2010 patch = q.lookup(patch)
2006 2011 if q.isapplied(patch):
2007 2012 ret = q.pop(repo, patch, force=opts['force'])
2008 2013 else:
2009 2014 ret = q.push(repo, patch, force=opts['force'])
2010 2015 q.save_dirty()
2011 2016 return ret
2012 2017
2013 2018 def guard(ui, repo, *args, **opts):
2014 2019 '''set or print guards for a patch
2015 2020
2016 2021 Guards control whether a patch can be pushed. A patch with no
2017 2022 guards is always pushed. A patch with a positive guard ("+foo") is
2018 2023 pushed only if the qselect command has activated it. A patch with
2019 2024 a negative guard ("-foo") is never pushed if the qselect command
2020 2025 has activated it.
2021 2026
2022 2027 With no arguments, print the currently active guards.
2023 2028 With arguments, set guards for the named patch.
2024 2029 NOTE: Specifying negative guards now requires '--'.
2025 2030
2026 2031 To set guards on another patch:
2027 2032 hg qguard -- other.patch +2.6.17 -stable
2028 2033 '''
2029 2034 def status(idx):
2030 2035 guards = q.series_guards[idx] or ['unguarded']
2031 2036 ui.write('%s: %s\n' % (q.series[idx], ' '.join(guards)))
2032 2037 q = repo.mq
2033 2038 patch = None
2034 2039 args = list(args)
2035 2040 if opts['list']:
2036 2041 if args or opts['none']:
2037 2042 raise util.Abort(_('cannot mix -l/--list with options or arguments'))
2038 2043 for i in xrange(len(q.series)):
2039 2044 status(i)
2040 2045 return
2041 2046 if not args or args[0][0:1] in '-+':
2042 2047 if not q.applied:
2043 2048 raise util.Abort(_('no patches applied'))
2044 2049 patch = q.applied[-1].name
2045 2050 if patch is None and args[0][0:1] not in '-+':
2046 2051 patch = args.pop(0)
2047 2052 if patch is None:
2048 2053 raise util.Abort(_('no patch to work with'))
2049 2054 if args or opts['none']:
2050 2055 idx = q.find_series(patch)
2051 2056 if idx is None:
2052 2057 raise util.Abort(_('no patch named %s') % patch)
2053 2058 q.set_guards(idx, args)
2054 2059 q.save_dirty()
2055 2060 else:
2056 2061 status(q.series.index(q.lookup(patch)))
2057 2062
2058 2063 def header(ui, repo, patch=None):
2059 2064 """print the header of the topmost or specified patch"""
2060 2065 q = repo.mq
2061 2066
2062 2067 if patch:
2063 2068 patch = q.lookup(patch)
2064 2069 else:
2065 2070 if not q.applied:
2066 2071 ui.write('no patches applied\n')
2067 2072 return 1
2068 2073 patch = q.lookup('qtip')
2069 2074 ph = patchheader(repo.mq.join(patch))
2070 2075
2071 2076 ui.write('\n'.join(ph.message) + '\n')
2072 2077
2073 2078 def lastsavename(path):
2074 2079 (directory, base) = os.path.split(path)
2075 2080 names = os.listdir(directory)
2076 2081 namere = re.compile("%s.([0-9]+)" % base)
2077 2082 maxindex = None
2078 2083 maxname = None
2079 2084 for f in names:
2080 2085 m = namere.match(f)
2081 2086 if m:
2082 2087 index = int(m.group(1))
2083 2088 if maxindex is None or index > maxindex:
2084 2089 maxindex = index
2085 2090 maxname = f
2086 2091 if maxname:
2087 2092 return (os.path.join(directory, maxname), maxindex)
2088 2093 return (None, None)
2089 2094
2090 2095 def savename(path):
2091 2096 (last, index) = lastsavename(path)
2092 2097 if last is None:
2093 2098 index = 0
2094 2099 newpath = path + ".%d" % (index + 1)
2095 2100 return newpath
2096 2101
2097 2102 def push(ui, repo, patch=None, **opts):
2098 2103 """push the next patch onto the stack
2099 2104
2100 2105 When -f/--force is applied, all local changes in patched files
2101 2106 will be lost.
2102 2107 """
2103 2108 q = repo.mq
2104 2109 mergeq = None
2105 2110
2106 2111 if opts['merge']:
2107 2112 if opts['name']:
2108 2113 newpath = repo.join(opts['name'])
2109 2114 else:
2110 2115 newpath, i = lastsavename(q.path)
2111 2116 if not newpath:
2112 2117 ui.warn(_("no saved queues found, please use -n\n"))
2113 2118 return 1
2114 2119 mergeq = queue(ui, repo.join(""), newpath)
2115 2120 ui.warn(_("merging with queue at: %s\n") % mergeq.path)
2116 2121 ret = q.push(repo, patch, force=opts['force'], list=opts['list'],
2117 2122 mergeq=mergeq, all=opts.get('all'))
2118 2123 return ret
2119 2124
2120 2125 def pop(ui, repo, patch=None, **opts):
2121 2126 """pop the current patch off the stack
2122 2127
2123 2128 By default, pops off the top of the patch stack. If given a patch
2124 2129 name, keeps popping off patches until the named patch is at the
2125 2130 top of the stack.
2126 2131 """
2127 2132 localupdate = True
2128 2133 if opts['name']:
2129 2134 q = queue(ui, repo.join(""), repo.join(opts['name']))
2130 2135 ui.warn(_('using patch queue: %s\n') % q.path)
2131 2136 localupdate = False
2132 2137 else:
2133 2138 q = repo.mq
2134 2139 ret = q.pop(repo, patch, force=opts['force'], update=localupdate,
2135 2140 all=opts['all'])
2136 2141 q.save_dirty()
2137 2142 return ret
2138 2143
2139 2144 def rename(ui, repo, patch, name=None, **opts):
2140 2145 """rename a patch
2141 2146
2142 2147 With one argument, renames the current patch to PATCH1.
2143 2148 With two arguments, renames PATCH1 to PATCH2."""
2144 2149
2145 2150 q = repo.mq
2146 2151
2147 2152 if not name:
2148 2153 name = patch
2149 2154 patch = None
2150 2155
2151 2156 if patch:
2152 2157 patch = q.lookup(patch)
2153 2158 else:
2154 2159 if not q.applied:
2155 2160 ui.write(_('no patches applied\n'))
2156 2161 return
2157 2162 patch = q.lookup('qtip')
2158 2163 absdest = q.join(name)
2159 2164 if os.path.isdir(absdest):
2160 2165 name = normname(os.path.join(name, os.path.basename(patch)))
2161 2166 absdest = q.join(name)
2162 2167 if os.path.exists(absdest):
2163 2168 raise util.Abort(_('%s already exists') % absdest)
2164 2169
2165 2170 if name in q.series:
2166 2171 raise util.Abort(_('A patch named %s already exists in the series file') % name)
2167 2172
2168 2173 if ui.verbose:
2169 2174 ui.write('renaming %s to %s\n' % (patch, name))
2170 2175 i = q.find_series(patch)
2171 2176 guards = q.guard_re.findall(q.full_series[i])
2172 2177 q.full_series[i] = name + ''.join([' #' + g for g in guards])
2173 2178 q.parse_series()
2174 2179 q.series_dirty = 1
2175 2180
2176 2181 info = q.isapplied(patch)
2177 2182 if info:
2178 2183 q.applied[info[0]] = statusentry(info[1], name)
2179 2184 q.applied_dirty = 1
2180 2185
2181 2186 util.rename(q.join(patch), absdest)
2182 2187 r = q.qrepo()
2183 2188 if r:
2184 2189 wlock = r.wlock()
2185 2190 try:
2186 2191 if r.dirstate[patch] == 'a':
2187 2192 r.dirstate.forget(patch)
2188 2193 r.dirstate.add(name)
2189 2194 else:
2190 2195 if r.dirstate[name] == 'r':
2191 2196 r.undelete([name])
2192 2197 r.copy(patch, name)
2193 2198 r.remove([patch], False)
2194 2199 finally:
2195 2200 wlock.release()
2196 2201
2197 2202 q.save_dirty()
2198 2203
2199 2204 def restore(ui, repo, rev, **opts):
2200 2205 """restore the queue state saved by a revision"""
2201 2206 rev = repo.lookup(rev)
2202 2207 q = repo.mq
2203 2208 q.restore(repo, rev, delete=opts['delete'],
2204 2209 qupdate=opts['update'])
2205 2210 q.save_dirty()
2206 2211 return 0
2207 2212
2208 2213 def save(ui, repo, **opts):
2209 2214 """save current queue state"""
2210 2215 q = repo.mq
2211 2216 message = cmdutil.logmessage(opts)
2212 2217 ret = q.save(repo, msg=message)
2213 2218 if ret:
2214 2219 return ret
2215 2220 q.save_dirty()
2216 2221 if opts['copy']:
2217 2222 path = q.path
2218 2223 if opts['name']:
2219 2224 newpath = os.path.join(q.basepath, opts['name'])
2220 2225 if os.path.exists(newpath):
2221 2226 if not os.path.isdir(newpath):
2222 2227 raise util.Abort(_('destination %s exists and is not '
2223 2228 'a directory') % newpath)
2224 2229 if not opts['force']:
2225 2230 raise util.Abort(_('destination %s exists, '
2226 2231 'use -f to force') % newpath)
2227 2232 else:
2228 2233 newpath = savename(path)
2229 2234 ui.warn(_("copy %s to %s\n") % (path, newpath))
2230 2235 util.copyfiles(path, newpath)
2231 2236 if opts['empty']:
2232 2237 try:
2233 2238 os.unlink(q.join(q.status_path))
2234 2239 except:
2235 2240 pass
2236 2241 return 0
2237 2242
2238 2243 def strip(ui, repo, rev, **opts):
2239 2244 """strip a revision and all its descendants from the repository
2240 2245
2241 2246 If one of the working directory's parent revisions is stripped, the
2242 2247 working directory will be updated to the parent of the stripped
2243 2248 revision.
2244 2249 """
2245 2250 backup = 'all'
2246 2251 if opts['backup']:
2247 2252 backup = 'strip'
2248 2253 elif opts['nobackup']:
2249 2254 backup = 'none'
2250 2255
2251 2256 rev = repo.lookup(rev)
2252 2257 p = repo.dirstate.parents()
2253 2258 cl = repo.changelog
2254 2259 update = True
2255 2260 if p[0] == nullid:
2256 2261 update = False
2257 2262 elif p[1] == nullid and rev != cl.ancestor(p[0], rev):
2258 2263 update = False
2259 2264 elif rev not in (cl.ancestor(p[0], rev), cl.ancestor(p[1], rev)):
2260 2265 update = False
2261 2266
2262 2267 repo.mq.strip(repo, rev, backup=backup, update=update, force=opts['force'])
2263 2268 return 0
2264 2269
2265 2270 def select(ui, repo, *args, **opts):
2266 2271 '''set or print guarded patches to push
2267 2272
2268 2273 Use the qguard command to set or print guards on patch, then use
2269 2274 qselect to tell mq which guards to use. A patch will be pushed if
2270 2275 it has no guards or any positive guards match the currently
2271 2276 selected guard, but will not be pushed if any negative guards
2272 2277 match the current guard. For example:
2273 2278
2274 2279 qguard foo.patch -stable (negative guard)
2275 2280 qguard bar.patch +stable (positive guard)
2276 2281 qselect stable
2277 2282
2278 2283 This activates the "stable" guard. mq will skip foo.patch (because
2279 2284 it has a negative match) but push bar.patch (because it has a
2280 2285 positive match).
2281 2286
2282 2287 With no arguments, prints the currently active guards.
2283 2288 With one argument, sets the active guard.
2284 2289
2285 2290 Use -n/--none to deactivate guards (no other arguments needed).
2286 2291 When no guards are active, patches with positive guards are
2287 2292 skipped and patches with negative guards are pushed.
2288 2293
2289 2294 qselect can change the guards on applied patches. It does not pop
2290 2295 guarded patches by default. Use --pop to pop back to the last
2291 2296 applied patch that is not guarded. Use --reapply (which implies
2292 2297 --pop) to push back to the current patch afterwards, but skip
2293 2298 guarded patches.
2294 2299
2295 2300 Use -s/--series to print a list of all guards in the series file
2296 2301 (no other arguments needed). Use -v for more information.'''
2297 2302
2298 2303 q = repo.mq
2299 2304 guards = q.active()
2300 2305 if args or opts['none']:
2301 2306 old_unapplied = q.unapplied(repo)
2302 2307 old_guarded = [i for i in xrange(len(q.applied)) if
2303 2308 not q.pushable(i)[0]]
2304 2309 q.set_active(args)
2305 2310 q.save_dirty()
2306 2311 if not args:
2307 2312 ui.status(_('guards deactivated\n'))
2308 2313 if not opts['pop'] and not opts['reapply']:
2309 2314 unapplied = q.unapplied(repo)
2310 2315 guarded = [i for i in xrange(len(q.applied))
2311 2316 if not q.pushable(i)[0]]
2312 2317 if len(unapplied) != len(old_unapplied):
2313 2318 ui.status(_('number of unguarded, unapplied patches has '
2314 2319 'changed from %d to %d\n') %
2315 2320 (len(old_unapplied), len(unapplied)))
2316 2321 if len(guarded) != len(old_guarded):
2317 2322 ui.status(_('number of guarded, applied patches has changed '
2318 2323 'from %d to %d\n') %
2319 2324 (len(old_guarded), len(guarded)))
2320 2325 elif opts['series']:
2321 2326 guards = {}
2322 2327 noguards = 0
2323 2328 for gs in q.series_guards:
2324 2329 if not gs:
2325 2330 noguards += 1
2326 2331 for g in gs:
2327 2332 guards.setdefault(g, 0)
2328 2333 guards[g] += 1
2329 2334 if ui.verbose:
2330 2335 guards['NONE'] = noguards
2331 2336 guards = guards.items()
2332 2337 guards.sort(lambda a, b: cmp(a[0][1:], b[0][1:]))
2333 2338 if guards:
2334 2339 ui.note(_('guards in series file:\n'))
2335 2340 for guard, count in guards:
2336 2341 ui.note('%2d ' % count)
2337 2342 ui.write(guard, '\n')
2338 2343 else:
2339 2344 ui.note(_('no guards in series file\n'))
2340 2345 else:
2341 2346 if guards:
2342 2347 ui.note(_('active guards:\n'))
2343 2348 for g in guards:
2344 2349 ui.write(g, '\n')
2345 2350 else:
2346 2351 ui.write(_('no active guards\n'))
2347 2352 reapply = opts['reapply'] and q.applied and q.appliedname(-1)
2348 2353 popped = False
2349 2354 if opts['pop'] or opts['reapply']:
2350 2355 for i in xrange(len(q.applied)):
2351 2356 pushable, reason = q.pushable(i)
2352 2357 if not pushable:
2353 2358 ui.status(_('popping guarded patches\n'))
2354 2359 popped = True
2355 2360 if i == 0:
2356 2361 q.pop(repo, all=True)
2357 2362 else:
2358 2363 q.pop(repo, i-1)
2359 2364 break
2360 2365 if popped:
2361 2366 try:
2362 2367 if reapply:
2363 2368 ui.status(_('reapplying unguarded patches\n'))
2364 2369 q.push(repo, reapply)
2365 2370 finally:
2366 2371 q.save_dirty()
2367 2372
2368 2373 def finish(ui, repo, *revrange, **opts):
2369 2374 """move applied patches into repository history
2370 2375
2371 2376 Finishes the specified revisions (corresponding to applied
2372 2377 patches) by moving them out of mq control into regular repository
2373 2378 history.
2374 2379
2375 2380 Accepts a revision range or the -a/--applied option. If --applied
2376 2381 is specified, all applied mq revisions are removed from mq
2377 2382 control. Otherwise, the given revisions must be at the base of the
2378 2383 stack of applied patches.
2379 2384
2380 2385 This can be especially useful if your changes have been applied to
2381 2386 an upstream repository, or if you are about to push your changes
2382 2387 to upstream.
2383 2388 """
2384 2389 if not opts['applied'] and not revrange:
2385 2390 raise util.Abort(_('no revisions specified'))
2386 2391 elif opts['applied']:
2387 2392 revrange = ('qbase:qtip',) + revrange
2388 2393
2389 2394 q = repo.mq
2390 2395 if not q.applied:
2391 2396 ui.status(_('no patches applied\n'))
2392 2397 return 0
2393 2398
2394 2399 revs = cmdutil.revrange(repo, revrange)
2395 2400 q.finish(repo, revs)
2396 2401 q.save_dirty()
2397 2402 return 0
2398 2403
2399 2404 def reposetup(ui, repo):
2400 2405 class mqrepo(repo.__class__):
2401 2406 @util.propertycache
2402 2407 def mq(self):
2403 2408 return queue(self.ui, self.join(""))
2404 2409
2405 2410 def abort_if_wdir_patched(self, errmsg, force=False):
2406 2411 if self.mq.applied and not force:
2407 2412 parent = hex(self.dirstate.parents()[0])
2408 2413 if parent in [s.rev for s in self.mq.applied]:
2409 2414 raise util.Abort(errmsg)
2410 2415
2411 2416 def commit(self, text="", user=None, date=None, match=None,
2412 2417 force=False, editor=False, extra={}):
2413 2418 self.abort_if_wdir_patched(
2414 2419 _('cannot commit over an applied mq patch'),
2415 2420 force)
2416 2421
2417 2422 return super(mqrepo, self).commit(text, user, date, match, force,
2418 2423 editor, extra)
2419 2424
2420 2425 def push(self, remote, force=False, revs=None):
2421 2426 if self.mq.applied and not force and not revs:
2422 2427 raise util.Abort(_('source has mq patches applied'))
2423 2428 return super(mqrepo, self).push(remote, force, revs)
2424 2429
2425 2430 def tags(self):
2426 2431 if self.tagscache:
2427 2432 return self.tagscache
2428 2433
2429 2434 tagscache = super(mqrepo, self).tags()
2430 2435
2431 2436 q = self.mq
2432 2437 if not q.applied:
2433 2438 return tagscache
2434 2439
2435 2440 mqtags = [(bin(patch.rev), patch.name) for patch in q.applied]
2436 2441
2437 2442 if mqtags[-1][0] not in self.changelog.nodemap:
2438 2443 self.ui.warn(_('mq status file refers to unknown node %s\n')
2439 2444 % short(mqtags[-1][0]))
2440 2445 return tagscache
2441 2446
2442 2447 mqtags.append((mqtags[-1][0], 'qtip'))
2443 2448 mqtags.append((mqtags[0][0], 'qbase'))
2444 2449 mqtags.append((self.changelog.parents(mqtags[0][0])[0], 'qparent'))
2445 2450 for patch in mqtags:
2446 2451 if patch[1] in tagscache:
2447 2452 self.ui.warn(_('Tag %s overrides mq patch of the same name\n')
2448 2453 % patch[1])
2449 2454 else:
2450 2455 tagscache[patch[1]] = patch[0]
2451 2456
2452 2457 return tagscache
2453 2458
2454 2459 def _branchtags(self, partial, lrev):
2455 2460 q = self.mq
2456 2461 if not q.applied:
2457 2462 return super(mqrepo, self)._branchtags(partial, lrev)
2458 2463
2459 2464 cl = self.changelog
2460 2465 qbasenode = bin(q.applied[0].rev)
2461 2466 if qbasenode not in cl.nodemap:
2462 2467 self.ui.warn(_('mq status file refers to unknown node %s\n')
2463 2468 % short(qbasenode))
2464 2469 return super(mqrepo, self)._branchtags(partial, lrev)
2465 2470
2466 2471 qbase = cl.rev(qbasenode)
2467 2472 start = lrev + 1
2468 2473 if start < qbase:
2469 2474 # update the cache (excluding the patches) and save it
2470 2475 self._updatebranchcache(partial, lrev+1, qbase)
2471 2476 self._writebranchcache(partial, cl.node(qbase-1), qbase-1)
2472 2477 start = qbase
2473 2478 # if start = qbase, the cache is as updated as it should be.
2474 2479 # if start > qbase, the cache includes (part of) the patches.
2475 2480 # we might as well use it, but we won't save it.
2476 2481
2477 2482 # update the cache up to the tip
2478 2483 self._updatebranchcache(partial, start, len(cl))
2479 2484
2480 2485 return partial
2481 2486
2482 2487 if repo.local():
2483 2488 repo.__class__ = mqrepo
2484 2489
2485 2490 def mqimport(orig, ui, repo, *args, **kwargs):
2486 2491 if hasattr(repo, 'abort_if_wdir_patched'):
2487 2492 repo.abort_if_wdir_patched(_('cannot import over an applied patch'),
2488 2493 kwargs.get('force'))
2489 2494 return orig(ui, repo, *args, **kwargs)
2490 2495
2491 2496 def uisetup(ui):
2492 2497 extensions.wrapcommand(commands.table, 'import', mqimport)
2493 2498
2494 2499 seriesopts = [('s', 'summary', None, _('print first line of patch header'))]
2495 2500
2496 2501 cmdtable = {
2497 2502 "qapplied": (applied, [] + seriesopts, _('hg qapplied [-s] [PATCH]')),
2498 2503 "qclone":
2499 2504 (clone,
2500 2505 [('', 'pull', None, _('use pull protocol to copy metadata')),
2501 2506 ('U', 'noupdate', None, _('do not update the new working directories')),
2502 2507 ('', 'uncompressed', None,
2503 2508 _('use uncompressed transfer (fast over LAN)')),
2504 2509 ('p', 'patches', '', _('location of source patch repository')),
2505 2510 ] + commands.remoteopts,
2506 2511 _('hg qclone [OPTION]... SOURCE [DEST]')),
2507 2512 "qcommit|qci":
2508 2513 (commit,
2509 2514 commands.table["^commit|ci"][1],
2510 2515 _('hg qcommit [OPTION]... [FILE]...')),
2511 2516 "^qdiff":
2512 2517 (diff,
2513 2518 commands.diffopts + commands.diffopts2 + commands.walkopts,
2514 2519 _('hg qdiff [OPTION]... [FILE]...')),
2515 2520 "qdelete|qremove|qrm":
2516 2521 (delete,
2517 2522 [('k', 'keep', None, _('keep patch file')),
2518 2523 ('r', 'rev', [], _('stop managing a revision (DEPRECATED)'))],
2519 2524 _('hg qdelete [-k] [-r REV]... [PATCH]...')),
2520 2525 'qfold':
2521 2526 (fold,
2522 2527 [('e', 'edit', None, _('edit patch header')),
2523 2528 ('k', 'keep', None, _('keep folded patch files')),
2524 2529 ] + commands.commitopts,
2525 2530 _('hg qfold [-e] [-k] [-m TEXT] [-l FILE] PATCH...')),
2526 2531 'qgoto':
2527 2532 (goto,
2528 2533 [('f', 'force', None, _('overwrite any local changes'))],
2529 2534 _('hg qgoto [OPTION]... PATCH')),
2530 2535 'qguard':
2531 2536 (guard,
2532 2537 [('l', 'list', None, _('list all patches and guards')),
2533 2538 ('n', 'none', None, _('drop all guards'))],
2534 2539 _('hg qguard [-l] [-n] -- [PATCH] [+GUARD]... [-GUARD]...')),
2535 2540 'qheader': (header, [], _('hg qheader [PATCH]')),
2536 2541 "^qimport":
2537 2542 (qimport,
2538 2543 [('e', 'existing', None, _('import file in patch directory')),
2539 2544 ('n', 'name', '', _('name of patch file')),
2540 2545 ('f', 'force', None, _('overwrite existing files')),
2541 2546 ('r', 'rev', [], _('place existing revisions under mq control')),
2542 2547 ('g', 'git', None, _('use git extended diff format')),
2543 2548 ('P', 'push', None, _('qpush after importing'))],
2544 2549 _('hg qimport [-e] [-n NAME] [-f] [-g] [-P] [-r REV]... FILE...')),
2545 2550 "^qinit":
2546 2551 (init,
2547 2552 [('c', 'create-repo', None, _('create queue repository'))],
2548 2553 _('hg qinit [-c]')),
2549 2554 "qnew":
2550 2555 (new,
2551 2556 [('e', 'edit', None, _('edit commit message')),
2552 2557 ('f', 'force', None, _('import uncommitted changes into patch')),
2553 2558 ('g', 'git', None, _('use git extended diff format')),
2554 2559 ('U', 'currentuser', None, _('add "From: <current user>" to patch')),
2555 2560 ('u', 'user', '', _('add "From: <given user>" to patch')),
2556 2561 ('D', 'currentdate', None, _('add "Date: <current date>" to patch')),
2557 2562 ('d', 'date', '', _('add "Date: <given date>" to patch'))
2558 2563 ] + commands.walkopts + commands.commitopts,
2559 2564 _('hg qnew [-e] [-m TEXT] [-l FILE] [-f] PATCH [FILE]...')),
2560 2565 "qnext": (next, [] + seriesopts, _('hg qnext [-s]')),
2561 2566 "qprev": (prev, [] + seriesopts, _('hg qprev [-s]')),
2562 2567 "^qpop":
2563 2568 (pop,
2564 2569 [('a', 'all', None, _('pop all patches')),
2565 2570 ('n', 'name', '', _('queue name to pop')),
2566 2571 ('f', 'force', None, _('forget any local changes'))],
2567 2572 _('hg qpop [-a] [-n NAME] [-f] [PATCH | INDEX]')),
2568 2573 "^qpush":
2569 2574 (push,
2570 2575 [('f', 'force', None, _('apply if the patch has rejects')),
2571 2576 ('l', 'list', None, _('list patch name in commit text')),
2572 2577 ('a', 'all', None, _('apply all patches')),
2573 2578 ('m', 'merge', None, _('merge from another queue')),
2574 2579 ('n', 'name', '', _('merge queue name'))],
2575 2580 _('hg qpush [-f] [-l] [-a] [-m] [-n NAME] [PATCH | INDEX]')),
2576 2581 "^qrefresh":
2577 2582 (refresh,
2578 2583 [('e', 'edit', None, _('edit commit message')),
2579 2584 ('g', 'git', None, _('use git extended diff format')),
2580 2585 ('s', 'short', None, _('refresh only files already in the patch and specified files')),
2581 2586 ('U', 'currentuser', None, _('add/update "From: <current user>" in patch')),
2582 2587 ('u', 'user', '', _('add/update "From: <given user>" in patch')),
2583 2588 ('D', 'currentdate', None, _('update "Date: <current date>" in patch (if present)')),
2584 2589 ('d', 'date', '', _('update "Date: <given date>" in patch (if present)'))
2585 2590 ] + commands.walkopts + commands.commitopts,
2586 2591 _('hg qrefresh [-I] [-X] [-e] [-m TEXT] [-l FILE] [-s] [FILE]...')),
2587 2592 'qrename|qmv':
2588 2593 (rename, [], _('hg qrename PATCH1 [PATCH2]')),
2589 2594 "qrestore":
2590 2595 (restore,
2591 2596 [('d', 'delete', None, _('delete save entry')),
2592 2597 ('u', 'update', None, _('update queue working directory'))],
2593 2598 _('hg qrestore [-d] [-u] REV')),
2594 2599 "qsave":
2595 2600 (save,
2596 2601 [('c', 'copy', None, _('copy patch directory')),
2597 2602 ('n', 'name', '', _('copy directory name')),
2598 2603 ('e', 'empty', None, _('clear queue status file')),
2599 2604 ('f', 'force', None, _('force copy'))] + commands.commitopts,
2600 2605 _('hg qsave [-m TEXT] [-l FILE] [-c] [-n NAME] [-e] [-f]')),
2601 2606 "qselect":
2602 2607 (select,
2603 2608 [('n', 'none', None, _('disable all guards')),
2604 2609 ('s', 'series', None, _('list all guards in series file')),
2605 2610 ('', 'pop', None, _('pop to before first guarded applied patch')),
2606 2611 ('', 'reapply', None, _('pop, then reapply patches'))],
2607 2612 _('hg qselect [OPTION]... [GUARD]...')),
2608 2613 "qseries":
2609 2614 (series,
2610 2615 [('m', 'missing', None, _('print patches not in series')),
2611 2616 ] + seriesopts,
2612 2617 _('hg qseries [-ms]')),
2613 2618 "^strip":
2614 2619 (strip,
2615 2620 [('f', 'force', None, _('force removal with local changes')),
2616 2621 ('b', 'backup', None, _('bundle unrelated changesets')),
2617 2622 ('n', 'nobackup', None, _('no backups'))],
2618 2623 _('hg strip [-f] [-b] [-n] REV')),
2619 2624 "qtop": (top, [] + seriesopts, _('hg qtop [-s]')),
2620 2625 "qunapplied": (unapplied, [] + seriesopts, _('hg qunapplied [-s] [PATCH]')),
2621 2626 "qfinish":
2622 2627 (finish,
2623 2628 [('a', 'applied', None, _('finish all applied changesets'))],
2624 2629 _('hg qfinish [-a] [REV]...')),
2625 2630 }
General Comments 0
You need to be logged in to leave comments. Login now