##// END OF EJS Templates
mq: abort cleanly when invalid patch name is given to qguard
Christian Ebert -
r4133:a9ee6c53 default
parent child Browse files
Show More
@@ -1,2191 +1,2194
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 filelist = filter(matchfn, util.unique(m + r + a))
1019 1019 patch.diff(repo, patchparent, files=filelist, match=matchfn,
1020 1020 fp=patchf, changes=(m, a, r, [], u),
1021 1021 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], force=1, wlock=wlock)
1070 1070 self.applied[-1] = statusentry(revlog.hex(n), patchfn)
1071 1071 self.applied_dirty = 1
1072 1072 else:
1073 1073 self.printdiff(repo, patchparent, fp=patchf)
1074 1074 patchf.close()
1075 1075 added = repo.status()[1]
1076 1076 for a in added:
1077 1077 f = repo.wjoin(a)
1078 1078 try:
1079 1079 os.unlink(f)
1080 1080 except OSError, e:
1081 1081 if e.errno != errno.ENOENT:
1082 1082 raise
1083 1083 try: os.removedirs(os.path.dirname(f))
1084 1084 except: pass
1085 1085 # forget the file copies in the dirstate
1086 1086 # push should readd the files later on
1087 1087 repo.dirstate.forget(added)
1088 1088 self.pop(repo, force=True, wlock=wlock)
1089 1089 self.push(repo, force=True, wlock=wlock)
1090 1090
1091 1091 def init(self, repo, create=False):
1092 1092 if os.path.isdir(self.path):
1093 1093 raise util.Abort(_("patch queue directory already exists"))
1094 1094 os.mkdir(self.path)
1095 1095 if create:
1096 1096 return self.qrepo(create=True)
1097 1097
1098 1098 def unapplied(self, repo, patch=None):
1099 1099 if patch and patch not in self.series:
1100 1100 raise util.Abort(_("patch %s is not in series file") % patch)
1101 1101 if not patch:
1102 1102 start = self.series_end()
1103 1103 else:
1104 1104 start = self.series.index(patch) + 1
1105 1105 unapplied = []
1106 1106 for i in xrange(start, len(self.series)):
1107 1107 pushable, reason = self.pushable(i)
1108 1108 if pushable:
1109 1109 unapplied.append((i, self.series[i]))
1110 1110 self.explain_pushable(i)
1111 1111 return unapplied
1112 1112
1113 1113 def qseries(self, repo, missing=None, start=0, length=0, status=None,
1114 1114 summary=False):
1115 1115 def displayname(patchname):
1116 1116 if summary:
1117 1117 msg = self.readheaders(patchname)[0]
1118 1118 msg = msg and ': ' + msg[0] or ': '
1119 1119 else:
1120 1120 msg = ''
1121 1121 return '%s%s' % (patchname, msg)
1122 1122
1123 1123 def pname(i):
1124 1124 if status == 'A':
1125 1125 return self.applied[i].name
1126 1126 else:
1127 1127 return self.series[i]
1128 1128
1129 1129 applied = dict.fromkeys([p.name for p in self.applied])
1130 1130 if not length:
1131 1131 length = len(self.series) - start
1132 1132 if not missing:
1133 1133 for i in xrange(start, start+length):
1134 1134 pfx = ''
1135 1135 patch = pname(i)
1136 1136 if self.ui.verbose:
1137 1137 if patch in applied:
1138 1138 stat = 'A'
1139 1139 elif self.pushable(i)[0]:
1140 1140 stat = 'U'
1141 1141 else:
1142 1142 stat = 'G'
1143 1143 pfx = '%d %s ' % (i, stat)
1144 1144 self.ui.write('%s%s\n' % (pfx, displayname(patch)))
1145 1145 else:
1146 1146 msng_list = []
1147 1147 for root, dirs, files in os.walk(self.path):
1148 1148 d = root[len(self.path) + 1:]
1149 1149 for f in files:
1150 1150 fl = os.path.join(d, f)
1151 1151 if (fl not in self.series and
1152 1152 fl not in (self.status_path, self.series_path)
1153 1153 and not fl.startswith('.')):
1154 1154 msng_list.append(fl)
1155 1155 msng_list.sort()
1156 1156 for x in msng_list:
1157 1157 pfx = self.ui.verbose and ('D ') or ''
1158 1158 self.ui.write("%s%s\n" % (pfx, displayname(x)))
1159 1159
1160 1160 def issaveline(self, l):
1161 1161 if l.name == '.hg.patches.save.line':
1162 1162 return True
1163 1163
1164 1164 def qrepo(self, create=False):
1165 1165 if create or os.path.isdir(self.join(".hg")):
1166 1166 return hg.repository(self.ui, path=self.path, create=create)
1167 1167
1168 1168 def restore(self, repo, rev, delete=None, qupdate=None):
1169 1169 c = repo.changelog.read(rev)
1170 1170 desc = c[4].strip()
1171 1171 lines = desc.splitlines()
1172 1172 i = 0
1173 1173 datastart = None
1174 1174 series = []
1175 1175 applied = []
1176 1176 qpp = None
1177 1177 for i in xrange(0, len(lines)):
1178 1178 if lines[i] == 'Patch Data:':
1179 1179 datastart = i + 1
1180 1180 elif lines[i].startswith('Dirstate:'):
1181 1181 l = lines[i].rstrip()
1182 1182 l = l[10:].split(' ')
1183 1183 qpp = [ hg.bin(x) for x in l ]
1184 1184 elif datastart != None:
1185 1185 l = lines[i].rstrip()
1186 1186 se = statusentry(l)
1187 1187 file_ = se.name
1188 1188 if se.rev:
1189 1189 applied.append(se)
1190 1190 else:
1191 1191 series.append(file_)
1192 1192 if datastart == None:
1193 1193 self.ui.warn("No saved patch data found\n")
1194 1194 return 1
1195 1195 self.ui.warn("restoring status: %s\n" % lines[0])
1196 1196 self.full_series = series
1197 1197 self.applied = applied
1198 1198 self.parse_series()
1199 1199 self.series_dirty = 1
1200 1200 self.applied_dirty = 1
1201 1201 heads = repo.changelog.heads()
1202 1202 if delete:
1203 1203 if rev not in heads:
1204 1204 self.ui.warn("save entry has children, leaving it alone\n")
1205 1205 else:
1206 1206 self.ui.warn("removing save entry %s\n" % hg.short(rev))
1207 1207 pp = repo.dirstate.parents()
1208 1208 if rev in pp:
1209 1209 update = True
1210 1210 else:
1211 1211 update = False
1212 1212 self.strip(repo, rev, update=update, backup='strip')
1213 1213 if qpp:
1214 1214 self.ui.warn("saved queue repository parents: %s %s\n" %
1215 1215 (hg.short(qpp[0]), hg.short(qpp[1])))
1216 1216 if qupdate:
1217 1217 print "queue directory updating"
1218 1218 r = self.qrepo()
1219 1219 if not r:
1220 1220 self.ui.warn("Unable to load queue repository\n")
1221 1221 return 1
1222 1222 hg.clean(r, qpp[0])
1223 1223
1224 1224 def save(self, repo, msg=None):
1225 1225 if len(self.applied) == 0:
1226 1226 self.ui.warn("save: no patches applied, exiting\n")
1227 1227 return 1
1228 1228 if self.issaveline(self.applied[-1]):
1229 1229 self.ui.warn("status is already saved\n")
1230 1230 return 1
1231 1231
1232 1232 ar = [ ':' + x for x in self.full_series ]
1233 1233 if not msg:
1234 1234 msg = "hg patches saved state"
1235 1235 else:
1236 1236 msg = "hg patches: " + msg.rstrip('\r\n')
1237 1237 r = self.qrepo()
1238 1238 if r:
1239 1239 pp = r.dirstate.parents()
1240 1240 msg += "\nDirstate: %s %s" % (hg.hex(pp[0]), hg.hex(pp[1]))
1241 1241 msg += "\n\nPatch Data:\n"
1242 1242 text = msg + "\n".join([str(x) for x in self.applied]) + '\n' + (ar and
1243 1243 "\n".join(ar) + '\n' or "")
1244 1244 n = repo.commit(None, text, user=None, force=1)
1245 1245 if not n:
1246 1246 self.ui.warn("repo commit failed\n")
1247 1247 return 1
1248 1248 self.applied.append(statusentry(revlog.hex(n),'.hg.patches.save.line'))
1249 1249 self.applied_dirty = 1
1250 1250
1251 1251 def full_series_end(self):
1252 1252 if len(self.applied) > 0:
1253 1253 p = self.applied[-1].name
1254 1254 end = self.find_series(p)
1255 1255 if end == None:
1256 1256 return len(self.full_series)
1257 1257 return end + 1
1258 1258 return 0
1259 1259
1260 1260 def series_end(self, all_patches=False):
1261 1261 end = 0
1262 1262 def next(start):
1263 1263 if all_patches:
1264 1264 return start
1265 1265 i = start
1266 1266 while i < len(self.series):
1267 1267 p, reason = self.pushable(i)
1268 1268 if p:
1269 1269 break
1270 1270 self.explain_pushable(i)
1271 1271 i += 1
1272 1272 return i
1273 1273 if len(self.applied) > 0:
1274 1274 p = self.applied[-1].name
1275 1275 try:
1276 1276 end = self.series.index(p)
1277 1277 except ValueError:
1278 1278 return 0
1279 1279 return next(end + 1)
1280 1280 return next(end)
1281 1281
1282 1282 def appliedname(self, index):
1283 1283 pname = self.applied[index].name
1284 1284 if not self.ui.verbose:
1285 1285 p = pname
1286 1286 else:
1287 1287 p = str(self.series.index(pname)) + " " + pname
1288 1288 return p
1289 1289
1290 1290 def qimport(self, repo, files, patchname=None, rev=None, existing=None,
1291 1291 force=None, git=False):
1292 1292 def checkseries(patchname):
1293 1293 if patchname in self.series:
1294 1294 raise util.Abort(_('patch %s is already in the series file')
1295 1295 % patchname)
1296 1296 def checkfile(patchname):
1297 1297 if not force and os.path.exists(self.join(patchname)):
1298 1298 raise util.Abort(_('patch "%s" already exists')
1299 1299 % patchname)
1300 1300
1301 1301 if rev:
1302 1302 if files:
1303 1303 raise util.Abort(_('option "-r" not valid when importing '
1304 1304 'files'))
1305 1305 rev = cmdutil.revrange(repo, rev)
1306 1306 rev.sort(lambda x, y: cmp(y, x))
1307 1307 if (len(files) > 1 or len(rev) > 1) and patchname:
1308 1308 raise util.Abort(_('option "-n" not valid when importing multiple '
1309 1309 'patches'))
1310 1310 i = 0
1311 1311 added = []
1312 1312 if rev:
1313 1313 # If mq patches are applied, we can only import revisions
1314 1314 # that form a linear path to qbase.
1315 1315 # Otherwise, they should form a linear path to a head.
1316 1316 heads = repo.changelog.heads(repo.changelog.node(rev[-1]))
1317 1317 if len(heads) > 1:
1318 1318 raise util.Abort(_('revision %d is the root of more than one '
1319 1319 'branch') % rev[-1])
1320 1320 if self.applied:
1321 1321 base = revlog.hex(repo.changelog.node(rev[0]))
1322 1322 if base in [n.rev for n in self.applied]:
1323 1323 raise util.Abort(_('revision %d is already managed')
1324 1324 % rev[0])
1325 1325 if heads != [revlog.bin(self.applied[-1].rev)]:
1326 1326 raise util.Abort(_('revision %d is not the parent of '
1327 1327 'the queue') % rev[0])
1328 1328 base = repo.changelog.rev(revlog.bin(self.applied[0].rev))
1329 1329 lastparent = repo.changelog.parentrevs(base)[0]
1330 1330 else:
1331 1331 if heads != [repo.changelog.node(rev[0])]:
1332 1332 raise util.Abort(_('revision %d has unmanaged children')
1333 1333 % rev[0])
1334 1334 lastparent = None
1335 1335
1336 1336 if git:
1337 1337 self.diffopts().git = True
1338 1338
1339 1339 for r in rev:
1340 1340 p1, p2 = repo.changelog.parentrevs(r)
1341 1341 n = repo.changelog.node(r)
1342 1342 if p2 != revlog.nullrev:
1343 1343 raise util.Abort(_('cannot import merge revision %d') % r)
1344 1344 if lastparent and lastparent != r:
1345 1345 raise util.Abort(_('revision %d is not the parent of %d')
1346 1346 % (r, lastparent))
1347 1347 lastparent = p1
1348 1348
1349 1349 if not patchname:
1350 1350 patchname = '%d.diff' % r
1351 1351 checkseries(patchname)
1352 1352 checkfile(patchname)
1353 1353 self.full_series.insert(0, patchname)
1354 1354
1355 1355 patchf = self.opener(patchname, "w")
1356 1356 patch.export(repo, [n], fp=patchf, opts=self.diffopts())
1357 1357 patchf.close()
1358 1358
1359 1359 se = statusentry(revlog.hex(n), patchname)
1360 1360 self.applied.insert(0, se)
1361 1361
1362 1362 added.append(patchname)
1363 1363 patchname = None
1364 1364 self.parse_series()
1365 1365 self.applied_dirty = 1
1366 1366
1367 1367 for filename in files:
1368 1368 if existing:
1369 1369 if filename == '-':
1370 1370 raise util.Abort(_('-e is incompatible with import from -'))
1371 1371 if not patchname:
1372 1372 patchname = filename
1373 1373 if not os.path.isfile(self.join(patchname)):
1374 1374 raise util.Abort(_("patch %s does not exist") % patchname)
1375 1375 else:
1376 1376 try:
1377 1377 if filename == '-':
1378 1378 if not patchname:
1379 1379 raise util.Abort(_('need --name to import a patch from -'))
1380 1380 text = sys.stdin.read()
1381 1381 else:
1382 1382 text = file(filename).read()
1383 1383 except IOError:
1384 1384 raise util.Abort(_("unable to read %s") % patchname)
1385 1385 if not patchname:
1386 1386 patchname = os.path.basename(filename)
1387 1387 checkfile(patchname)
1388 1388 patchf = self.opener(patchname, "w")
1389 1389 patchf.write(text)
1390 1390 checkseries(patchname)
1391 1391 index = self.full_series_end() + i
1392 1392 self.full_series[index:index] = [patchname]
1393 1393 self.parse_series()
1394 1394 self.ui.warn("adding %s to series file\n" % patchname)
1395 1395 i += 1
1396 1396 added.append(patchname)
1397 1397 patchname = None
1398 1398 self.series_dirty = 1
1399 1399 qrepo = self.qrepo()
1400 1400 if qrepo:
1401 1401 qrepo.add(added)
1402 1402
1403 1403 def delete(ui, repo, *patches, **opts):
1404 1404 """remove patches from queue
1405 1405
1406 1406 With --rev, mq will stop managing the named revisions. The
1407 1407 patches must be applied and at the base of the stack. This option
1408 1408 is useful when the patches have been applied upstream.
1409 1409
1410 1410 Otherwise, the patches must not be applied.
1411 1411
1412 1412 With --keep, the patch files are preserved in the patch directory."""
1413 1413 q = repo.mq
1414 1414 q.delete(repo, patches, opts)
1415 1415 q.save_dirty()
1416 1416 return 0
1417 1417
1418 1418 def applied(ui, repo, patch=None, **opts):
1419 1419 """print the patches already applied"""
1420 1420 q = repo.mq
1421 1421 if patch:
1422 1422 if patch not in q.series:
1423 1423 raise util.Abort(_("patch %s is not in series file") % patch)
1424 1424 end = q.series.index(patch) + 1
1425 1425 else:
1426 1426 end = len(q.applied)
1427 1427 if not end:
1428 1428 return
1429 1429
1430 1430 return q.qseries(repo, length=end, status='A', summary=opts.get('summary'))
1431 1431
1432 1432 def unapplied(ui, repo, patch=None, **opts):
1433 1433 """print the patches not yet applied"""
1434 1434 q = repo.mq
1435 1435 if patch:
1436 1436 if patch not in q.series:
1437 1437 raise util.Abort(_("patch %s is not in series file") % patch)
1438 1438 start = q.series.index(patch) + 1
1439 1439 else:
1440 1440 start = q.series_end()
1441 1441 q.qseries(repo, start=start, summary=opts.get('summary'))
1442 1442
1443 1443 def qimport(ui, repo, *filename, **opts):
1444 1444 """import a patch
1445 1445
1446 1446 The patch will have the same name as its source file unless you
1447 1447 give it a new one with --name.
1448 1448
1449 1449 You can register an existing patch inside the patch directory
1450 1450 with the --existing flag.
1451 1451
1452 1452 With --force, an existing patch of the same name will be overwritten.
1453 1453
1454 1454 An existing changeset may be placed under mq control with --rev
1455 1455 (e.g. qimport --rev tip -n patch will place tip under mq control).
1456 1456 With --git, patches imported with --rev will use the git diff
1457 1457 format.
1458 1458 """
1459 1459 q = repo.mq
1460 1460 q.qimport(repo, filename, patchname=opts['name'],
1461 1461 existing=opts['existing'], force=opts['force'], rev=opts['rev'],
1462 1462 git=opts['git'])
1463 1463 q.save_dirty()
1464 1464 return 0
1465 1465
1466 1466 def init(ui, repo, **opts):
1467 1467 """init a new queue repository
1468 1468
1469 1469 The queue repository is unversioned by default. If -c is
1470 1470 specified, qinit will create a separate nested repository
1471 1471 for patches. Use qcommit to commit changes to this queue
1472 1472 repository."""
1473 1473 q = repo.mq
1474 1474 r = q.init(repo, create=opts['create_repo'])
1475 1475 q.save_dirty()
1476 1476 if r:
1477 1477 fp = r.wopener('.hgignore', 'w')
1478 1478 print >> fp, 'syntax: glob'
1479 1479 print >> fp, 'status'
1480 1480 print >> fp, 'guards'
1481 1481 fp.close()
1482 1482 r.wopener('series', 'w').close()
1483 1483 r.add(['.hgignore', 'series'])
1484 1484 return 0
1485 1485
1486 1486 def clone(ui, source, dest=None, **opts):
1487 1487 '''clone main and patch repository at same time
1488 1488
1489 1489 If source is local, destination will have no patches applied. If
1490 1490 source is remote, this command can not check if patches are
1491 1491 applied in source, so cannot guarantee that patches are not
1492 1492 applied in destination. If you clone remote repository, be sure
1493 1493 before that it has no patches applied.
1494 1494
1495 1495 Source patch repository is looked for in <src>/.hg/patches by
1496 1496 default. Use -p <url> to change.
1497 1497 '''
1498 1498 commands.setremoteconfig(ui, opts)
1499 1499 if dest is None:
1500 1500 dest = hg.defaultdest(source)
1501 1501 sr = hg.repository(ui, ui.expandpath(source))
1502 1502 qbase, destrev = None, None
1503 1503 if sr.local():
1504 1504 if sr.mq.applied:
1505 1505 qbase = revlog.bin(sr.mq.applied[0].rev)
1506 1506 if not hg.islocal(dest):
1507 1507 destrev = sr.parents(qbase)[0]
1508 1508 ui.note(_('cloning main repo\n'))
1509 1509 sr, dr = hg.clone(ui, sr, dest,
1510 1510 pull=opts['pull'],
1511 1511 rev=destrev,
1512 1512 update=False,
1513 1513 stream=opts['uncompressed'])
1514 1514 ui.note(_('cloning patch repo\n'))
1515 1515 spr, dpr = hg.clone(ui, opts['patches'] or (sr.url() + '/.hg/patches'),
1516 1516 dr.url() + '/.hg/patches',
1517 1517 pull=opts['pull'],
1518 1518 update=not opts['noupdate'],
1519 1519 stream=opts['uncompressed'])
1520 1520 if dr.local():
1521 1521 if qbase:
1522 1522 ui.note(_('stripping applied patches from destination repo\n'))
1523 1523 dr.mq.strip(dr, qbase, update=False, backup=None)
1524 1524 if not opts['noupdate']:
1525 1525 ui.note(_('updating destination repo\n'))
1526 1526 hg.update(dr, dr.changelog.tip())
1527 1527
1528 1528 def commit(ui, repo, *pats, **opts):
1529 1529 """commit changes in the queue repository"""
1530 1530 q = repo.mq
1531 1531 r = q.qrepo()
1532 1532 if not r: raise util.Abort('no queue repository')
1533 1533 commands.commit(r.ui, r, *pats, **opts)
1534 1534
1535 1535 def series(ui, repo, **opts):
1536 1536 """print the entire series file"""
1537 1537 repo.mq.qseries(repo, missing=opts['missing'], summary=opts['summary'])
1538 1538 return 0
1539 1539
1540 1540 def top(ui, repo, **opts):
1541 1541 """print the name of the current patch"""
1542 1542 q = repo.mq
1543 1543 t = len(q.applied)
1544 1544 if t:
1545 1545 return q.qseries(repo, start=t-1, length=1, status='A',
1546 1546 summary=opts.get('summary'))
1547 1547 else:
1548 1548 ui.write("No patches applied\n")
1549 1549 return 1
1550 1550
1551 1551 def next(ui, repo, **opts):
1552 1552 """print the name of the next patch"""
1553 1553 q = repo.mq
1554 1554 end = q.series_end()
1555 1555 if end == len(q.series):
1556 1556 ui.write("All patches applied\n")
1557 1557 return 1
1558 1558 return q.qseries(repo, start=end, length=1, summary=opts.get('summary'))
1559 1559
1560 1560 def prev(ui, repo, **opts):
1561 1561 """print the name of the previous patch"""
1562 1562 q = repo.mq
1563 1563 l = len(q.applied)
1564 1564 if l == 1:
1565 1565 ui.write("Only one patch applied\n")
1566 1566 return 1
1567 1567 if not l:
1568 1568 ui.write("No patches applied\n")
1569 1569 return 1
1570 1570 return q.qseries(repo, start=l-2, length=1, status='A',
1571 1571 summary=opts.get('summary'))
1572 1572
1573 1573 def new(ui, repo, patch, **opts):
1574 1574 """create a new patch
1575 1575
1576 1576 qnew creates a new patch on top of the currently-applied patch
1577 1577 (if any). It will refuse to run if there are any outstanding
1578 1578 changes unless -f is specified, in which case the patch will
1579 1579 be initialised with them.
1580 1580
1581 1581 -e, -m or -l set the patch header as well as the commit message.
1582 1582 If none is specified, the patch header is empty and the
1583 1583 commit message is 'New patch: PATCH'"""
1584 1584 q = repo.mq
1585 1585 message = commands.logmessage(opts)
1586 1586 if opts['edit']:
1587 1587 message = ui.edit(message, ui.username())
1588 1588 q.new(repo, patch, msg=message, force=opts['force'])
1589 1589 q.save_dirty()
1590 1590 return 0
1591 1591
1592 1592 def refresh(ui, repo, *pats, **opts):
1593 1593 """update the current patch
1594 1594
1595 1595 If any file patterns are provided, the refreshed patch will contain only
1596 1596 the modifications that match those patterns; the remaining modifications
1597 1597 will remain in the working directory.
1598 1598 """
1599 1599 q = repo.mq
1600 1600 message = commands.logmessage(opts)
1601 1601 if opts['edit']:
1602 1602 if message:
1603 1603 raise util.Abort(_('option "-e" incompatible with "-m" or "-l"'))
1604 1604 patch = q.applied[-1].name
1605 1605 (message, comment, user, date, hasdiff) = q.readheaders(patch)
1606 1606 message = ui.edit('\n'.join(message), user or ui.username())
1607 1607 ret = q.refresh(repo, pats, msg=message, **opts)
1608 1608 q.save_dirty()
1609 1609 return ret
1610 1610
1611 1611 def diff(ui, repo, *pats, **opts):
1612 1612 """diff of the current patch"""
1613 1613 repo.mq.diff(repo, pats, opts)
1614 1614 return 0
1615 1615
1616 1616 def fold(ui, repo, *files, **opts):
1617 1617 """fold the named patches into the current patch
1618 1618
1619 1619 Patches must not yet be applied. Each patch will be successively
1620 1620 applied to the current patch in the order given. If all the
1621 1621 patches apply successfully, the current patch will be refreshed
1622 1622 with the new cumulative patch, and the folded patches will
1623 1623 be deleted. With -k/--keep, the folded patch files will not
1624 1624 be removed afterwards.
1625 1625
1626 1626 The header for each folded patch will be concatenated with
1627 1627 the current patch header, separated by a line of '* * *'."""
1628 1628
1629 1629 q = repo.mq
1630 1630
1631 1631 if not files:
1632 1632 raise util.Abort(_('qfold requires at least one patch name'))
1633 1633 if not q.check_toppatch(repo):
1634 1634 raise util.Abort(_('No patches applied'))
1635 1635
1636 1636 message = commands.logmessage(opts)
1637 1637 if opts['edit']:
1638 1638 if message:
1639 1639 raise util.Abort(_('option "-e" incompatible with "-m" or "-l"'))
1640 1640
1641 1641 parent = q.lookup('qtip')
1642 1642 patches = []
1643 1643 messages = []
1644 1644 for f in files:
1645 1645 p = q.lookup(f)
1646 1646 if p in patches or p == parent:
1647 1647 ui.warn(_('Skipping already folded patch %s') % p)
1648 1648 if q.isapplied(p):
1649 1649 raise util.Abort(_('qfold cannot fold already applied patch %s') % p)
1650 1650 patches.append(p)
1651 1651
1652 1652 for p in patches:
1653 1653 if not message:
1654 1654 messages.append(q.readheaders(p)[0])
1655 1655 pf = q.join(p)
1656 1656 (patchsuccess, files, fuzz) = q.patch(repo, pf)
1657 1657 if not patchsuccess:
1658 1658 raise util.Abort(_('Error folding patch %s') % p)
1659 1659 patch.updatedir(ui, repo, files)
1660 1660
1661 1661 if not message:
1662 1662 message, comments, user = q.readheaders(parent)[0:3]
1663 1663 for msg in messages:
1664 1664 message.append('* * *')
1665 1665 message.extend(msg)
1666 1666 message = '\n'.join(message)
1667 1667
1668 1668 if opts['edit']:
1669 1669 message = ui.edit(message, user or ui.username())
1670 1670
1671 1671 q.refresh(repo, msg=message)
1672 1672 q.delete(repo, patches, opts)
1673 1673 q.save_dirty()
1674 1674
1675 1675 def guard(ui, repo, *args, **opts):
1676 1676 '''set or print guards for a patch
1677 1677
1678 1678 Guards control whether a patch can be pushed. A patch with no
1679 1679 guards is always pushed. A patch with a positive guard ("+foo") is
1680 1680 pushed only if the qselect command has activated it. A patch with
1681 1681 a negative guard ("-foo") is never pushed if the qselect command
1682 1682 has activated it.
1683 1683
1684 1684 With no arguments, print the currently active guards.
1685 1685 With arguments, set guards for the named patch.
1686 1686
1687 1687 To set a negative guard "-foo" on topmost patch ("--" is needed so
1688 1688 hg will not interpret "-foo" as an option):
1689 1689 hg qguard -- -foo
1690 1690
1691 1691 To set guards on another patch:
1692 1692 hg qguard other.patch +2.6.17 -stable
1693 1693 '''
1694 1694 def status(idx):
1695 1695 guards = q.series_guards[idx] or ['unguarded']
1696 1696 ui.write('%s: %s\n' % (q.series[idx], ' '.join(guards)))
1697 1697 q = repo.mq
1698 1698 patch = None
1699 1699 args = list(args)
1700 1700 if opts['list']:
1701 1701 if args or opts['none']:
1702 1702 raise util.Abort(_('cannot mix -l/--list with options or arguments'))
1703 1703 for i in xrange(len(q.series)):
1704 1704 status(i)
1705 1705 return
1706 1706 if not args or args[0][0:1] in '-+':
1707 1707 if not q.applied:
1708 1708 raise util.Abort(_('no patches applied'))
1709 1709 patch = q.applied[-1].name
1710 1710 if patch is None and args[0][0:1] not in '-+':
1711 1711 patch = args.pop(0)
1712 1712 if patch is None:
1713 1713 raise util.Abort(_('no patch to work with'))
1714 1714 if args or opts['none']:
1715 q.set_guards(q.find_series(patch), args)
1715 idx = q.find_series(patch)
1716 if idx is None:
1717 raise util.Abort(_('no patch named %s') % patch)
1718 q.set_guards(idx, args)
1716 1719 q.save_dirty()
1717 1720 else:
1718 1721 status(q.series.index(q.lookup(patch)))
1719 1722
1720 1723 def header(ui, repo, patch=None):
1721 1724 """Print the header of the topmost or specified patch"""
1722 1725 q = repo.mq
1723 1726
1724 1727 if patch:
1725 1728 patch = q.lookup(patch)
1726 1729 else:
1727 1730 if not q.applied:
1728 1731 ui.write('No patches applied\n')
1729 1732 return 1
1730 1733 patch = q.lookup('qtip')
1731 1734 message = repo.mq.readheaders(patch)[0]
1732 1735
1733 1736 ui.write('\n'.join(message) + '\n')
1734 1737
1735 1738 def lastsavename(path):
1736 1739 (directory, base) = os.path.split(path)
1737 1740 names = os.listdir(directory)
1738 1741 namere = re.compile("%s.([0-9]+)" % base)
1739 1742 maxindex = None
1740 1743 maxname = None
1741 1744 for f in names:
1742 1745 m = namere.match(f)
1743 1746 if m:
1744 1747 index = int(m.group(1))
1745 1748 if maxindex == None or index > maxindex:
1746 1749 maxindex = index
1747 1750 maxname = f
1748 1751 if maxname:
1749 1752 return (os.path.join(directory, maxname), maxindex)
1750 1753 return (None, None)
1751 1754
1752 1755 def savename(path):
1753 1756 (last, index) = lastsavename(path)
1754 1757 if last is None:
1755 1758 index = 0
1756 1759 newpath = path + ".%d" % (index + 1)
1757 1760 return newpath
1758 1761
1759 1762 def push(ui, repo, patch=None, **opts):
1760 1763 """push the next patch onto the stack"""
1761 1764 q = repo.mq
1762 1765 mergeq = None
1763 1766
1764 1767 if opts['all']:
1765 1768 if not q.series:
1766 1769 raise util.Abort(_('no patches in series'))
1767 1770 patch = q.series[-1]
1768 1771 if opts['merge']:
1769 1772 if opts['name']:
1770 1773 newpath = opts['name']
1771 1774 else:
1772 1775 newpath, i = lastsavename(q.path)
1773 1776 if not newpath:
1774 1777 ui.warn("no saved queues found, please use -n\n")
1775 1778 return 1
1776 1779 mergeq = queue(ui, repo.join(""), newpath)
1777 1780 ui.warn("merging with queue at: %s\n" % mergeq.path)
1778 1781 ret = q.push(repo, patch, force=opts['force'], list=opts['list'],
1779 1782 mergeq=mergeq)
1780 1783 q.save_dirty()
1781 1784 return ret
1782 1785
1783 1786 def pop(ui, repo, patch=None, **opts):
1784 1787 """pop the current patch off the stack"""
1785 1788 localupdate = True
1786 1789 if opts['name']:
1787 1790 q = queue(ui, repo.join(""), repo.join(opts['name']))
1788 1791 ui.warn('using patch queue: %s\n' % q.path)
1789 1792 localupdate = False
1790 1793 else:
1791 1794 q = repo.mq
1792 1795 q.pop(repo, patch, force=opts['force'], update=localupdate, all=opts['all'])
1793 1796 q.save_dirty()
1794 1797 return 0
1795 1798
1796 1799 def rename(ui, repo, patch, name=None, **opts):
1797 1800 """rename a patch
1798 1801
1799 1802 With one argument, renames the current patch to PATCH1.
1800 1803 With two arguments, renames PATCH1 to PATCH2."""
1801 1804
1802 1805 q = repo.mq
1803 1806
1804 1807 if not name:
1805 1808 name = patch
1806 1809 patch = None
1807 1810
1808 1811 if patch:
1809 1812 patch = q.lookup(patch)
1810 1813 else:
1811 1814 if not q.applied:
1812 1815 ui.write(_('No patches applied\n'))
1813 1816 return
1814 1817 patch = q.lookup('qtip')
1815 1818 absdest = q.join(name)
1816 1819 if os.path.isdir(absdest):
1817 1820 name = os.path.join(name, os.path.basename(patch))
1818 1821 absdest = q.join(name)
1819 1822 if os.path.exists(absdest):
1820 1823 raise util.Abort(_('%s already exists') % absdest)
1821 1824
1822 1825 if name in q.series:
1823 1826 raise util.Abort(_('A patch named %s already exists in the series file') % name)
1824 1827
1825 1828 if ui.verbose:
1826 1829 ui.write('Renaming %s to %s\n' % (patch, name))
1827 1830 i = q.find_series(patch)
1828 1831 guards = q.guard_re.findall(q.full_series[i])
1829 1832 q.full_series[i] = name + ''.join([' #' + g for g in guards])
1830 1833 q.parse_series()
1831 1834 q.series_dirty = 1
1832 1835
1833 1836 info = q.isapplied(patch)
1834 1837 if info:
1835 1838 q.applied[info[0]] = statusentry(info[1], name)
1836 1839 q.applied_dirty = 1
1837 1840
1838 1841 util.rename(q.join(patch), absdest)
1839 1842 r = q.qrepo()
1840 1843 if r:
1841 1844 wlock = r.wlock()
1842 1845 if r.dirstate.state(name) == 'r':
1843 1846 r.undelete([name], wlock)
1844 1847 r.copy(patch, name, wlock)
1845 1848 r.remove([patch], False, wlock)
1846 1849
1847 1850 q.save_dirty()
1848 1851
1849 1852 def restore(ui, repo, rev, **opts):
1850 1853 """restore the queue state saved by a rev"""
1851 1854 rev = repo.lookup(rev)
1852 1855 q = repo.mq
1853 1856 q.restore(repo, rev, delete=opts['delete'],
1854 1857 qupdate=opts['update'])
1855 1858 q.save_dirty()
1856 1859 return 0
1857 1860
1858 1861 def save(ui, repo, **opts):
1859 1862 """save current queue state"""
1860 1863 q = repo.mq
1861 1864 message = commands.logmessage(opts)
1862 1865 ret = q.save(repo, msg=message)
1863 1866 if ret:
1864 1867 return ret
1865 1868 q.save_dirty()
1866 1869 if opts['copy']:
1867 1870 path = q.path
1868 1871 if opts['name']:
1869 1872 newpath = os.path.join(q.basepath, opts['name'])
1870 1873 if os.path.exists(newpath):
1871 1874 if not os.path.isdir(newpath):
1872 1875 raise util.Abort(_('destination %s exists and is not '
1873 1876 'a directory') % newpath)
1874 1877 if not opts['force']:
1875 1878 raise util.Abort(_('destination %s exists, '
1876 1879 'use -f to force') % newpath)
1877 1880 else:
1878 1881 newpath = savename(path)
1879 1882 ui.warn("copy %s to %s\n" % (path, newpath))
1880 1883 util.copyfiles(path, newpath)
1881 1884 if opts['empty']:
1882 1885 try:
1883 1886 os.unlink(q.join(q.status_path))
1884 1887 except:
1885 1888 pass
1886 1889 return 0
1887 1890
1888 1891 def strip(ui, repo, rev, **opts):
1889 1892 """strip a revision and all later revs on the same branch"""
1890 1893 rev = repo.lookup(rev)
1891 1894 backup = 'all'
1892 1895 if opts['backup']:
1893 1896 backup = 'strip'
1894 1897 elif opts['nobackup']:
1895 1898 backup = 'none'
1896 1899 update = repo.dirstate.parents()[0] != revlog.nullid
1897 1900 repo.mq.strip(repo, rev, backup=backup, update=update)
1898 1901 return 0
1899 1902
1900 1903 def select(ui, repo, *args, **opts):
1901 1904 '''set or print guarded patches to push
1902 1905
1903 1906 Use the qguard command to set or print guards on patch, then use
1904 1907 qselect to tell mq which guards to use. A patch will be pushed if it
1905 1908 has no guards or any positive guards match the currently selected guard,
1906 1909 but will not be pushed if any negative guards match the current guard.
1907 1910 For example:
1908 1911
1909 1912 qguard foo.patch -stable (negative guard)
1910 1913 qguard bar.patch +stable (positive guard)
1911 1914 qselect stable
1912 1915
1913 1916 This activates the "stable" guard. mq will skip foo.patch (because
1914 1917 it has a negative match) but push bar.patch (because it
1915 1918 has a positive match).
1916 1919
1917 1920 With no arguments, prints the currently active guards.
1918 1921 With one argument, sets the active guard.
1919 1922
1920 1923 Use -n/--none to deactivate guards (no other arguments needed).
1921 1924 When no guards are active, patches with positive guards are skipped
1922 1925 and patches with negative guards are pushed.
1923 1926
1924 1927 qselect can change the guards on applied patches. It does not pop
1925 1928 guarded patches by default. Use --pop to pop back to the last applied
1926 1929 patch that is not guarded. Use --reapply (which implies --pop) to push
1927 1930 back to the current patch afterwards, but skip guarded patches.
1928 1931
1929 1932 Use -s/--series to print a list of all guards in the series file (no
1930 1933 other arguments needed). Use -v for more information.'''
1931 1934
1932 1935 q = repo.mq
1933 1936 guards = q.active()
1934 1937 if args or opts['none']:
1935 1938 old_unapplied = q.unapplied(repo)
1936 1939 old_guarded = [i for i in xrange(len(q.applied)) if
1937 1940 not q.pushable(i)[0]]
1938 1941 q.set_active(args)
1939 1942 q.save_dirty()
1940 1943 if not args:
1941 1944 ui.status(_('guards deactivated\n'))
1942 1945 if not opts['pop'] and not opts['reapply']:
1943 1946 unapplied = q.unapplied(repo)
1944 1947 guarded = [i for i in xrange(len(q.applied))
1945 1948 if not q.pushable(i)[0]]
1946 1949 if len(unapplied) != len(old_unapplied):
1947 1950 ui.status(_('number of unguarded, unapplied patches has '
1948 1951 'changed from %d to %d\n') %
1949 1952 (len(old_unapplied), len(unapplied)))
1950 1953 if len(guarded) != len(old_guarded):
1951 1954 ui.status(_('number of guarded, applied patches has changed '
1952 1955 'from %d to %d\n') %
1953 1956 (len(old_guarded), len(guarded)))
1954 1957 elif opts['series']:
1955 1958 guards = {}
1956 1959 noguards = 0
1957 1960 for gs in q.series_guards:
1958 1961 if not gs:
1959 1962 noguards += 1
1960 1963 for g in gs:
1961 1964 guards.setdefault(g, 0)
1962 1965 guards[g] += 1
1963 1966 if ui.verbose:
1964 1967 guards['NONE'] = noguards
1965 1968 guards = guards.items()
1966 1969 guards.sort(lambda a, b: cmp(a[0][1:], b[0][1:]))
1967 1970 if guards:
1968 1971 ui.note(_('guards in series file:\n'))
1969 1972 for guard, count in guards:
1970 1973 ui.note('%2d ' % count)
1971 1974 ui.write(guard, '\n')
1972 1975 else:
1973 1976 ui.note(_('no guards in series file\n'))
1974 1977 else:
1975 1978 if guards:
1976 1979 ui.note(_('active guards:\n'))
1977 1980 for g in guards:
1978 1981 ui.write(g, '\n')
1979 1982 else:
1980 1983 ui.write(_('no active guards\n'))
1981 1984 reapply = opts['reapply'] and q.applied and q.appliedname(-1)
1982 1985 popped = False
1983 1986 if opts['pop'] or opts['reapply']:
1984 1987 for i in xrange(len(q.applied)):
1985 1988 pushable, reason = q.pushable(i)
1986 1989 if not pushable:
1987 1990 ui.status(_('popping guarded patches\n'))
1988 1991 popped = True
1989 1992 if i == 0:
1990 1993 q.pop(repo, all=True)
1991 1994 else:
1992 1995 q.pop(repo, i-1)
1993 1996 break
1994 1997 if popped:
1995 1998 try:
1996 1999 if reapply:
1997 2000 ui.status(_('reapplying unguarded patches\n'))
1998 2001 q.push(repo, reapply)
1999 2002 finally:
2000 2003 q.save_dirty()
2001 2004
2002 2005 def reposetup(ui, repo):
2003 2006 class mqrepo(repo.__class__):
2004 2007 def abort_if_wdir_patched(self, errmsg, force=False):
2005 2008 if self.mq.applied and not force:
2006 2009 parent = revlog.hex(self.dirstate.parents()[0])
2007 2010 if parent in [s.rev for s in self.mq.applied]:
2008 2011 raise util.Abort(errmsg)
2009 2012
2010 2013 def commit(self, *args, **opts):
2011 2014 if len(args) >= 6:
2012 2015 force = args[5]
2013 2016 else:
2014 2017 force = opts.get('force')
2015 2018 self.abort_if_wdir_patched(
2016 2019 _('cannot commit over an applied mq patch'),
2017 2020 force)
2018 2021
2019 2022 return super(mqrepo, self).commit(*args, **opts)
2020 2023
2021 2024 def push(self, remote, force=False, revs=None):
2022 2025 if self.mq.applied and not force:
2023 2026 raise util.Abort(_('source has mq patches applied'))
2024 2027 return super(mqrepo, self).push(remote, force, revs)
2025 2028
2026 2029 def tags(self):
2027 2030 if self.tagscache:
2028 2031 return self.tagscache
2029 2032
2030 2033 tagscache = super(mqrepo, self).tags()
2031 2034
2032 2035 q = self.mq
2033 2036 if not q.applied:
2034 2037 return tagscache
2035 2038
2036 2039 mqtags = [(patch.rev, patch.name) for patch in q.applied]
2037 2040 mqtags.append((mqtags[-1][0], 'qtip'))
2038 2041 mqtags.append((mqtags[0][0], 'qbase'))
2039 2042 for patch in mqtags:
2040 2043 if patch[1] in tagscache:
2041 2044 self.ui.warn('Tag %s overrides mq patch of the same name\n' % patch[1])
2042 2045 else:
2043 2046 tagscache[patch[1]] = revlog.bin(patch[0])
2044 2047
2045 2048 return tagscache
2046 2049
2047 2050 def _branchtags(self):
2048 2051 q = self.mq
2049 2052 if not q.applied:
2050 2053 return super(mqrepo, self)._branchtags()
2051 2054
2052 2055 self.branchcache = {} # avoid recursion in changectx
2053 2056 cl = self.changelog
2054 2057 partial, last, lrev = self._readbranchcache()
2055 2058
2056 2059 qbase = cl.rev(revlog.bin(q.applied[0].rev))
2057 2060 start = lrev + 1
2058 2061 if start < qbase:
2059 2062 # update the cache (excluding the patches) and save it
2060 2063 self._updatebranchcache(partial, lrev+1, qbase)
2061 2064 self._writebranchcache(partial, cl.node(qbase-1), qbase-1)
2062 2065 start = qbase
2063 2066 # if start = qbase, the cache is as updated as it should be.
2064 2067 # if start > qbase, the cache includes (part of) the patches.
2065 2068 # we might as well use it, but we won't save it.
2066 2069
2067 2070 # update the cache up to the tip
2068 2071 self._updatebranchcache(partial, start, cl.count())
2069 2072
2070 2073 return partial
2071 2074
2072 2075 if repo.local():
2073 2076 repo.__class__ = mqrepo
2074 2077 repo.mq = queue(ui, repo.join(""))
2075 2078
2076 2079 seriesopts = [('s', 'summary', None, _('print first line of patch header'))]
2077 2080
2078 2081 cmdtable = {
2079 2082 "qapplied": (applied, [] + seriesopts, 'hg qapplied [-s] [PATCH]'),
2080 2083 "qclone": (clone,
2081 2084 [('', 'pull', None, _('use pull protocol to copy metadata')),
2082 2085 ('U', 'noupdate', None, _('do not update the new working directories')),
2083 2086 ('', 'uncompressed', None,
2084 2087 _('use uncompressed transfer (fast over LAN)')),
2085 2088 ('e', 'ssh', '', _('specify ssh command to use')),
2086 2089 ('p', 'patches', '', _('location of source patch repo')),
2087 2090 ('', 'remotecmd', '',
2088 2091 _('specify hg command to run on the remote side'))],
2089 2092 'hg qclone [OPTION]... SOURCE [DEST]'),
2090 2093 "qcommit|qci":
2091 2094 (commit,
2092 2095 commands.table["^commit|ci"][1],
2093 2096 'hg qcommit [OPTION]... [FILE]...'),
2094 2097 "^qdiff": (diff,
2095 2098 [('g', 'git', None, _('use git extended diff format')),
2096 2099 ('I', 'include', [], _('include names matching the given patterns')),
2097 2100 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2098 2101 'hg qdiff [-I] [-X] [FILE]...'),
2099 2102 "qdelete|qremove|qrm":
2100 2103 (delete,
2101 2104 [('k', 'keep', None, _('keep patch file')),
2102 2105 ('r', 'rev', [], _('stop managing a revision'))],
2103 2106 'hg qdelete [-k] [-r REV]... PATCH...'),
2104 2107 'qfold':
2105 2108 (fold,
2106 2109 [('e', 'edit', None, _('edit patch header')),
2107 2110 ('k', 'keep', None, _('keep folded patch files'))
2108 2111 ] + commands.commitopts,
2109 2112 'hg qfold [-e] [-m <text>] [-l <file] PATCH...'),
2110 2113 'qguard': (guard, [('l', 'list', None, _('list all patches and guards')),
2111 2114 ('n', 'none', None, _('drop all guards'))],
2112 2115 'hg qguard [PATCH] [+GUARD...] [-GUARD...]'),
2113 2116 'qheader': (header, [],
2114 2117 _('hg qheader [PATCH]')),
2115 2118 "^qimport":
2116 2119 (qimport,
2117 2120 [('e', 'existing', None, 'import file in patch dir'),
2118 2121 ('n', 'name', '', 'patch file name'),
2119 2122 ('f', 'force', None, 'overwrite existing files'),
2120 2123 ('r', 'rev', [], 'place existing revisions under mq control'),
2121 2124 ('g', 'git', None, _('use git extended diff format'))],
2122 2125 'hg qimport [-e] [-n NAME] [-f] [-g] [-r REV]... FILE...'),
2123 2126 "^qinit":
2124 2127 (init,
2125 2128 [('c', 'create-repo', None, 'create queue repository')],
2126 2129 'hg qinit [-c]'),
2127 2130 "qnew":
2128 2131 (new,
2129 2132 [('e', 'edit', None, _('edit commit message')),
2130 2133 ('f', 'force', None, _('import uncommitted changes into patch'))
2131 2134 ] + commands.commitopts,
2132 2135 'hg qnew [-e] [-m TEXT] [-l FILE] [-f] PATCH'),
2133 2136 "qnext": (next, [] + seriesopts, 'hg qnext [-s]'),
2134 2137 "qprev": (prev, [] + seriesopts, 'hg qprev [-s]'),
2135 2138 "^qpop":
2136 2139 (pop,
2137 2140 [('a', 'all', None, 'pop all patches'),
2138 2141 ('n', 'name', '', 'queue name to pop'),
2139 2142 ('f', 'force', None, 'forget any local changes')],
2140 2143 'hg qpop [-a] [-n NAME] [-f] [PATCH | INDEX]'),
2141 2144 "^qpush":
2142 2145 (push,
2143 2146 [('f', 'force', None, 'apply if the patch has rejects'),
2144 2147 ('l', 'list', None, 'list patch name in commit text'),
2145 2148 ('a', 'all', None, 'apply all patches'),
2146 2149 ('m', 'merge', None, 'merge from another queue'),
2147 2150 ('n', 'name', '', 'merge queue name')],
2148 2151 'hg qpush [-f] [-l] [-a] [-m] [-n NAME] [PATCH | INDEX]'),
2149 2152 "^qrefresh":
2150 2153 (refresh,
2151 2154 [('e', 'edit', None, _('edit commit message')),
2152 2155 ('g', 'git', None, _('use git extended diff format')),
2153 2156 ('s', 'short', None, 'refresh only files already in the patch'),
2154 2157 ('I', 'include', [], _('include names matching the given patterns')),
2155 2158 ('X', 'exclude', [], _('exclude names matching the given patterns'))
2156 2159 ] + commands.commitopts,
2157 2160 'hg qrefresh [-I] [-X] [-e] [-m TEXT] [-l FILE] [-s] FILES...'),
2158 2161 'qrename|qmv':
2159 2162 (rename, [], 'hg qrename PATCH1 [PATCH2]'),
2160 2163 "qrestore":
2161 2164 (restore,
2162 2165 [('d', 'delete', None, 'delete save entry'),
2163 2166 ('u', 'update', None, 'update queue working dir')],
2164 2167 'hg qrestore [-d] [-u] REV'),
2165 2168 "qsave":
2166 2169 (save,
2167 2170 [('c', 'copy', None, 'copy patch directory'),
2168 2171 ('n', 'name', '', 'copy directory name'),
2169 2172 ('e', 'empty', None, 'clear queue status file'),
2170 2173 ('f', 'force', None, 'force copy')] + commands.commitopts,
2171 2174 'hg qsave [-m TEXT] [-l FILE] [-c] [-n NAME] [-e] [-f]'),
2172 2175 "qselect": (select,
2173 2176 [('n', 'none', None, _('disable all guards')),
2174 2177 ('s', 'series', None, _('list all guards in series file')),
2175 2178 ('', 'pop', None,
2176 2179 _('pop to before first guarded applied patch')),
2177 2180 ('', 'reapply', None, _('pop, then reapply patches'))],
2178 2181 'hg qselect [OPTION...] [GUARD...]'),
2179 2182 "qseries":
2180 2183 (series,
2181 2184 [('m', 'missing', None, 'print patches not in series')] + seriesopts,
2182 2185 'hg qseries [-ms]'),
2183 2186 "^strip":
2184 2187 (strip,
2185 2188 [('f', 'force', None, 'force multi-head removal'),
2186 2189 ('b', 'backup', None, 'bundle unrelated changesets'),
2187 2190 ('n', 'nobackup', None, 'no backups')],
2188 2191 'hg strip [-f] [-b] [-n] REV'),
2189 2192 "qtop": (top, [] + seriesopts, 'hg qtop [-s]'),
2190 2193 "qunapplied": (unapplied, [] + seriesopts, 'hg qunapplied [-s] [PATCH]'),
2191 2194 }
@@ -1,117 +1,120
1 1 #!/bin/sh
2 2
3 3 echo "[extensions]" >> $HGRCPATH
4 4 echo "mq=" >> $HGRCPATH
5 5
6 6 hg init
7 7 hg qinit
8 8
9 9 echo x > x
10 10 hg ci -Ama
11 11
12 12 hg qnew a.patch
13 13 echo a > a
14 14 hg add a
15 15 hg qrefresh
16 16
17 17 hg qnew b.patch
18 18 echo b > b
19 19 hg add b
20 20 hg qrefresh
21 21
22 22 hg qnew c.patch
23 23 echo c > c
24 24 hg add c
25 25 hg qrefresh
26 26
27 27 hg qpop -a
28 28
29 29 echo % should fail
30 hg qguard does-not-exist.patch +bleh
31
32 echo % should fail
30 33 hg qguard +fail
31 34
32 35 hg qpush
33 36 echo % should guard a.patch
34 37 hg qguard +a
35 38 echo % should print +a
36 39 hg qguard
37 40 hg qpop
38 41
39 42 hg qguard a.patch
40 43 echo % should push b.patch
41 44 hg qpush
42 45
43 46 hg qpop
44 47 hg qselect a
45 48 echo % should push a.patch
46 49 hg qpush
47 50
48 51 hg qguard c.patch -a
49 52 echo % should print -a
50 53 hg qguard c.patch
51 54
52 55 echo % should skip c.patch
53 56 hg qpush -a
54 57
55 58 hg qguard -n c.patch
56 59 echo % should push c.patch
57 60 hg qpush -a
58 61
59 62 hg qpop -a
60 63 hg qselect -n
61 64 echo % should push all
62 65 hg qpush -a
63 66
64 67 hg qpop -a
65 68 hg qguard a.patch +1
66 69 hg qguard b.patch +2
67 70 hg qselect 1
68 71 echo % should push a.patch, not b.patch
69 72 hg qpush
70 73 hg qpush
71 74 hg qpop -a
72 75
73 76 hg qselect 2
74 77 echo % should push b.patch
75 78 hg qpush
76 79 hg qpop -a
77 80
78 81 hg qselect 1 2
79 82 echo % should push a.patch, b.patch
80 83 hg qpush
81 84 hg qpush
82 85 hg qpop -a
83 86
84 87 hg qguard a.patch +1 +2 -3
85 88 hg qselect 1 2 3
86 89 echo % list patches and guards
87 90 hg qguard -l
88 91 echo % list series
89 92 hg qseries -v
90 93 echo % list guards
91 94 hg qselect
92 95 echo % should push b.patch
93 96 hg qpush
94 97
95 98 hg qpush -a
96 99 hg qselect -n --reapply
97 100 echo % guards in series file: +1 +2 -3
98 101 hg qselect -s
99 102 echo % should show c.patch
100 103 hg qapplied
101 104
102 105 hg qrename a.patch new.patch
103 106 echo % should show :
104 107 echo % new.patch: +1 +2 -3
105 108 echo % b.patch: +2
106 109 echo % c.patch: unguarded
107 110 hg qguard -l
108 111
109 112 hg qnew d.patch
110 113 hg qpop
111 114 echo % should show new.patch and b.patch as Guarded, c.patch as Applied
112 115 echo % and d.patch as Unapplied
113 116 hg qseries -v
114 117
115 118 hg qguard d.patch +2
116 119 echo % new.patch, b.patch: Guarded. c.patch: Applied. d.patch: Guarded.
117 120 hg qseries -v
@@ -1,103 +1,105
1 1 adding x
2 2 Patch queue now empty
3 3 % should fail
4 abort: no patch named does-not-exist.patch
5 % should fail
4 6 abort: no patches applied
5 7 applying a.patch
6 8 Now at: a.patch
7 9 % should guard a.patch
8 10 % should print +a
9 11 a.patch: +a
10 12 Patch queue now empty
11 13 a.patch: +a
12 14 % should push b.patch
13 15 applying b.patch
14 16 Now at: b.patch
15 17 Patch queue now empty
16 18 number of unguarded, unapplied patches has changed from 2 to 3
17 19 % should push a.patch
18 20 applying a.patch
19 21 Now at: a.patch
20 22 % should print -a
21 23 c.patch: -a
22 24 % should skip c.patch
23 25 applying b.patch
24 26 skipping c.patch - guarded by '-a'
25 27 Now at: b.patch
26 28 % should push c.patch
27 29 applying c.patch
28 30 Now at: c.patch
29 31 Patch queue now empty
30 32 guards deactivated
31 33 number of unguarded, unapplied patches has changed from 3 to 2
32 34 % should push all
33 35 applying b.patch
34 36 applying c.patch
35 37 Now at: c.patch
36 38 Patch queue now empty
37 39 number of unguarded, unapplied patches has changed from 1 to 2
38 40 % should push a.patch, not b.patch
39 41 applying a.patch
40 42 Now at: a.patch
41 43 applying c.patch
42 44 Now at: c.patch
43 45 Patch queue now empty
44 46 % should push b.patch
45 47 applying b.patch
46 48 Now at: b.patch
47 49 Patch queue now empty
48 50 number of unguarded, unapplied patches has changed from 2 to 3
49 51 % should push a.patch, b.patch
50 52 applying a.patch
51 53 Now at: a.patch
52 54 applying b.patch
53 55 Now at: b.patch
54 56 Patch queue now empty
55 57 number of unguarded, unapplied patches has changed from 3 to 2
56 58 % list patches and guards
57 59 a.patch: +1 +2 -3
58 60 b.patch: +2
59 61 c.patch: unguarded
60 62 % list series
61 63 0 G a.patch
62 64 1 U b.patch
63 65 2 U c.patch
64 66 % list guards
65 67 1
66 68 2
67 69 3
68 70 % should push b.patch
69 71 applying b.patch
70 72 Now at: b.patch
71 73 applying c.patch
72 74 Now at: c.patch
73 75 guards deactivated
74 76 popping guarded patches
75 77 Patch queue now empty
76 78 reapplying unguarded patches
77 79 applying c.patch
78 80 Now at: c.patch
79 81 % guards in series file: +1 +2 -3
80 82 +1
81 83 +2
82 84 -3
83 85 % should show c.patch
84 86 c.patch
85 87 % should show :
86 88 % new.patch: +1 +2 -3
87 89 % b.patch: +2
88 90 % c.patch: unguarded
89 91 new.patch: +1 +2 -3
90 92 b.patch: +2
91 93 c.patch: unguarded
92 94 Now at: c.patch
93 95 % should show new.patch and b.patch as Guarded, c.patch as Applied
94 96 % and d.patch as Unapplied
95 97 0 G new.patch
96 98 1 G b.patch
97 99 2 A c.patch
98 100 3 U d.patch
99 101 % new.patch, b.patch: Guarded. c.patch: Applied. d.patch: Guarded.
100 102 0 G new.patch
101 103 1 G b.patch
102 104 2 A c.patch
103 105 3 G d.patch
General Comments 0
You need to be logged in to leave comments. Login now