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