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