##// END OF EJS Templates
Never apply string formatting to generated errors with util.Abort....
Thomas Arendsen Hein -
r3072:bc3fe3b5 default
parent child Browse files
Show More
@@ -1,2007 +1,2007 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,ui,util")
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(':')
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 self.series.append(patch)
118 118 self.series_guards.append(self.guard_re.findall(comment))
119 119
120 120 def check_guard(self, guard):
121 121 bad_chars = '# \t\r\n\f'
122 122 first = guard[0]
123 123 for c in '-+':
124 124 if first == c:
125 125 return (_('guard %r starts with invalid character: %r') %
126 126 (guard, c))
127 127 for c in bad_chars:
128 128 if c in guard:
129 129 return _('invalid character in guard %r: %r') % (guard, c)
130 130
131 131 def set_active(self, guards):
132 132 for guard in guards:
133 133 bad = self.check_guard(guard)
134 134 if bad:
135 135 raise util.Abort(bad)
136 136 guards = dict.fromkeys(guards).keys()
137 137 guards.sort()
138 138 self.ui.debug('active guards: %s\n' % ' '.join(guards))
139 139 self.active_guards = guards
140 140 self.guards_dirty = True
141 141
142 142 def active(self):
143 143 if self.active_guards is None:
144 144 self.active_guards = []
145 145 try:
146 146 guards = self.opener(self.guards_path).read().split()
147 147 except IOError, err:
148 148 if err.errno != errno.ENOENT: raise
149 149 guards = []
150 150 for i, guard in enumerate(guards):
151 151 bad = self.check_guard(guard)
152 152 if bad:
153 153 self.ui.warn('%s:%d: %s\n' %
154 154 (self.join(self.guards_path), i + 1, bad))
155 155 else:
156 156 self.active_guards.append(guard)
157 157 return self.active_guards
158 158
159 159 def set_guards(self, idx, guards):
160 160 for g in guards:
161 161 if len(g) < 2:
162 162 raise util.Abort(_('guard %r too short') % g)
163 163 if g[0] not in '-+':
164 164 raise util.Abort(_('guard %r starts with invalid char') % g)
165 165 bad = self.check_guard(g[1:])
166 166 if bad:
167 167 raise util.Abort(bad)
168 168 drop = self.guard_re.sub('', self.full_series[idx])
169 169 self.full_series[idx] = drop + ''.join([' #' + g for g in guards])
170 170 self.parse_series()
171 171 self.series_dirty = True
172 172
173 173 def pushable(self, idx):
174 174 if isinstance(idx, str):
175 175 idx = self.series.index(idx)
176 176 patchguards = self.series_guards[idx]
177 177 if not patchguards:
178 178 return True, None
179 179 default = False
180 180 guards = self.active()
181 181 exactneg = [g for g in patchguards if g[0] == '-' and g[1:] in guards]
182 182 if exactneg:
183 183 return False, exactneg[0]
184 184 pos = [g for g in patchguards if g[0] == '+']
185 185 exactpos = [g for g in pos if g[1:] in guards]
186 186 if pos:
187 187 if exactpos:
188 188 return True, exactpos[0]
189 189 return False, pos
190 190 return True, ''
191 191
192 192 def explain_pushable(self, idx, all_patches=False):
193 193 write = all_patches and self.ui.write or self.ui.warn
194 194 if all_patches or self.ui.verbose:
195 195 if isinstance(idx, str):
196 196 idx = self.series.index(idx)
197 197 pushable, why = self.pushable(idx)
198 198 if all_patches and pushable:
199 199 if why is None:
200 200 write(_('allowing %s - no guards in effect\n') %
201 201 self.series[idx])
202 202 else:
203 203 if not why:
204 204 write(_('allowing %s - no matching negative guards\n') %
205 205 self.series[idx])
206 206 else:
207 207 write(_('allowing %s - guarded by %r\n') %
208 208 (self.series[idx], why))
209 209 if not pushable:
210 210 if why:
211 211 write(_('skipping %s - guarded by %r\n') %
212 212 (self.series[idx], ' '.join(why)))
213 213 else:
214 214 write(_('skipping %s - no matching guards\n') %
215 215 self.series[idx])
216 216
217 217 def save_dirty(self):
218 218 def write_list(items, path):
219 219 fp = self.opener(path, 'w')
220 220 for i in items:
221 221 print >> fp, i
222 222 fp.close()
223 223 if self.applied_dirty: write_list(map(str, self.applied), self.status_path)
224 224 if self.series_dirty: write_list(self.full_series, self.series_path)
225 225 if self.guards_dirty: write_list(self.active_guards, self.guards_path)
226 226
227 227 def readheaders(self, patch):
228 228 def eatdiff(lines):
229 229 while lines:
230 230 l = lines[-1]
231 231 if (l.startswith("diff -") or
232 232 l.startswith("Index:") or
233 233 l.startswith("===========")):
234 234 del lines[-1]
235 235 else:
236 236 break
237 237 def eatempty(lines):
238 238 while lines:
239 239 l = lines[-1]
240 240 if re.match('\s*$', l):
241 241 del lines[-1]
242 242 else:
243 243 break
244 244
245 245 pf = self.join(patch)
246 246 message = []
247 247 comments = []
248 248 user = None
249 249 date = None
250 250 format = None
251 251 subject = None
252 252 diffstart = 0
253 253
254 254 for line in file(pf):
255 255 line = line.rstrip()
256 256 if line.startswith('diff --git'):
257 257 diffstart = 2
258 258 break
259 259 if diffstart:
260 260 if line.startswith('+++ '):
261 261 diffstart = 2
262 262 break
263 263 if line.startswith("--- "):
264 264 diffstart = 1
265 265 continue
266 266 elif format == "hgpatch":
267 267 # parse values when importing the result of an hg export
268 268 if line.startswith("# User "):
269 269 user = line[7:]
270 270 elif line.startswith("# Date "):
271 271 date = line[7:]
272 272 elif not line.startswith("# ") and line:
273 273 message.append(line)
274 274 format = None
275 275 elif line == '# HG changeset patch':
276 276 format = "hgpatch"
277 277 elif (format != "tagdone" and (line.startswith("Subject: ") or
278 278 line.startswith("subject: "))):
279 279 subject = line[9:]
280 280 format = "tag"
281 281 elif (format != "tagdone" and (line.startswith("From: ") or
282 282 line.startswith("from: "))):
283 283 user = line[6:]
284 284 format = "tag"
285 285 elif format == "tag" and line == "":
286 286 # when looking for tags (subject: from: etc) they
287 287 # end once you find a blank line in the source
288 288 format = "tagdone"
289 289 elif message or line:
290 290 message.append(line)
291 291 comments.append(line)
292 292
293 293 eatdiff(message)
294 294 eatdiff(comments)
295 295 eatempty(message)
296 296 eatempty(comments)
297 297
298 298 # make sure message isn't empty
299 299 if format and format.startswith("tag") and subject:
300 300 message.insert(0, "")
301 301 message.insert(0, subject)
302 302 return (message, comments, user, date, diffstart > 1)
303 303
304 304 def printdiff(self, repo, node1, node2=None, files=None,
305 305 fp=None, changes=None, opts={}):
306 306 fns, matchfn, anypats = cmdutil.matchpats(repo, files, opts)
307 307
308 308 patch.diff(repo, node1, node2, fns, match=matchfn,
309 309 fp=fp, changes=changes, opts=self.diffopts())
310 310
311 311 def mergeone(self, repo, mergeq, head, patch, rev, wlock):
312 312 # first try just applying the patch
313 313 (err, n) = self.apply(repo, [ patch ], update_status=False,
314 314 strict=True, merge=rev, wlock=wlock)
315 315
316 316 if err == 0:
317 317 return (err, n)
318 318
319 319 if n is None:
320 320 raise util.Abort(_("apply failed for patch %s") % patch)
321 321
322 322 self.ui.warn("patch didn't work out, merging %s\n" % patch)
323 323
324 324 # apply failed, strip away that rev and merge.
325 325 hg.clean(repo, head, wlock=wlock)
326 326 self.strip(repo, n, update=False, backup='strip', wlock=wlock)
327 327
328 328 c = repo.changelog.read(rev)
329 329 ret = hg.merge(repo, rev, wlock=wlock)
330 330 if ret:
331 331 raise util.Abort(_("update returned %d") % ret)
332 332 n = repo.commit(None, c[4], c[1], force=1, wlock=wlock)
333 333 if n == None:
334 334 raise util.Abort(_("repo commit failed"))
335 335 try:
336 336 message, comments, user, date, patchfound = mergeq.readheaders(patch)
337 337 except:
338 338 raise util.Abort(_("unable to read %s") % patch)
339 339
340 340 patchf = self.opener(patch, "w")
341 341 if comments:
342 342 comments = "\n".join(comments) + '\n\n'
343 343 patchf.write(comments)
344 344 self.printdiff(repo, head, n, fp=patchf)
345 345 patchf.close()
346 346 return (0, n)
347 347
348 348 def qparents(self, repo, rev=None):
349 349 if rev is None:
350 350 (p1, p2) = repo.dirstate.parents()
351 351 if p2 == revlog.nullid:
352 352 return p1
353 353 if len(self.applied) == 0:
354 354 return None
355 355 return revlog.bin(self.applied[-1].rev)
356 356 pp = repo.changelog.parents(rev)
357 357 if pp[1] != revlog.nullid:
358 358 arevs = [ x.rev for x in self.applied ]
359 359 p0 = revlog.hex(pp[0])
360 360 p1 = revlog.hex(pp[1])
361 361 if p0 in arevs:
362 362 return pp[0]
363 363 if p1 in arevs:
364 364 return pp[1]
365 365 return pp[0]
366 366
367 367 def mergepatch(self, repo, mergeq, series, wlock):
368 368 if len(self.applied) == 0:
369 369 # each of the patches merged in will have two parents. This
370 370 # can confuse the qrefresh, qdiff, and strip code because it
371 371 # needs to know which parent is actually in the patch queue.
372 372 # so, we insert a merge marker with only one parent. This way
373 373 # the first patch in the queue is never a merge patch
374 374 #
375 375 pname = ".hg.patches.merge.marker"
376 376 n = repo.commit(None, '[mq]: merge marker', user=None, force=1,
377 377 wlock=wlock)
378 378 self.applied.append(statusentry(revlog.hex(n), pname))
379 379 self.applied_dirty = 1
380 380
381 381 head = self.qparents(repo)
382 382
383 383 for patch in series:
384 384 patch = mergeq.lookup(patch, strict=True)
385 385 if not patch:
386 386 self.ui.warn("patch %s does not exist\n" % patch)
387 387 return (1, None)
388 388 pushable, reason = self.pushable(patch)
389 389 if not pushable:
390 390 self.explain_pushable(patch, all_patches=True)
391 391 continue
392 392 info = mergeq.isapplied(patch)
393 393 if not info:
394 394 self.ui.warn("patch %s is not applied\n" % patch)
395 395 return (1, None)
396 396 rev = revlog.bin(info[1])
397 397 (err, head) = self.mergeone(repo, mergeq, head, patch, rev, wlock)
398 398 if head:
399 399 self.applied.append(statusentry(revlog.hex(head), patch))
400 400 self.applied_dirty = 1
401 401 if err:
402 402 return (err, head)
403 403 return (0, head)
404 404
405 405 def patch(self, repo, patchfile):
406 406 '''Apply patchfile to the working directory.
407 407 patchfile: file name of patch'''
408 408 try:
409 409 (files, fuzz) = patch.patch(patchfile, self.ui, strip=1,
410 410 cwd=repo.root)
411 411 except Exception, inst:
412 412 self.ui.note(str(inst) + '\n')
413 413 if not self.ui.verbose:
414 414 self.ui.warn("patch failed, unable to continue (try -v)\n")
415 415 return (False, [], False)
416 416
417 417 return (True, files, fuzz)
418 418
419 419 def apply(self, repo, series, list=False, update_status=True,
420 420 strict=False, patchdir=None, merge=None, wlock=None):
421 421 # TODO unify with commands.py
422 422 if not patchdir:
423 423 patchdir = self.path
424 424 err = 0
425 425 if not wlock:
426 426 wlock = repo.wlock()
427 427 lock = repo.lock()
428 428 tr = repo.transaction()
429 429 n = None
430 430 for patchname in series:
431 431 pushable, reason = self.pushable(patchname)
432 432 if not pushable:
433 433 self.explain_pushable(patchname, all_patches=True)
434 434 continue
435 435 self.ui.warn("applying %s\n" % patchname)
436 436 pf = os.path.join(patchdir, patchname)
437 437
438 438 try:
439 439 message, comments, user, date, patchfound = self.readheaders(patchname)
440 440 except:
441 441 self.ui.warn("Unable to read %s\n" % patchname)
442 442 err = 1
443 443 break
444 444
445 445 if not message:
446 446 message = "imported patch %s\n" % patchname
447 447 else:
448 448 if list:
449 449 message.append("\nimported patch %s" % patchname)
450 450 message = '\n'.join(message)
451 451
452 452 (patcherr, files, fuzz) = self.patch(repo, pf)
453 453 patcherr = not patcherr
454 454
455 455 if merge and files:
456 456 # Mark as merged and update dirstate parent info
457 457 repo.dirstate.update(repo.dirstate.filterfiles(files.keys()), 'm')
458 458 p1, p2 = repo.dirstate.parents()
459 459 repo.dirstate.setparents(p1, merge)
460 460 files = patch.updatedir(self.ui, repo, files, wlock=wlock)
461 461 n = repo.commit(files, message, user, date, force=1, lock=lock,
462 462 wlock=wlock)
463 463
464 464 if n == None:
465 465 raise util.Abort(_("repo commit failed"))
466 466
467 467 if update_status:
468 468 self.applied.append(statusentry(revlog.hex(n), patchname))
469 469
470 470 if patcherr:
471 471 if not patchfound:
472 472 self.ui.warn("patch %s is empty\n" % patchname)
473 473 err = 0
474 474 else:
475 475 self.ui.warn("patch failed, rejects left in working dir\n")
476 476 err = 1
477 477 break
478 478
479 479 if fuzz and strict:
480 480 self.ui.warn("fuzz found when applying patch, stopping\n")
481 481 err = 1
482 482 break
483 483 tr.close()
484 484 return (err, n)
485 485
486 486 def delete(self, repo, patches, keep=False):
487 487 realpatches = []
488 488 for patch in patches:
489 489 patch = self.lookup(patch, strict=True)
490 490 info = self.isapplied(patch)
491 491 if info:
492 492 raise util.Abort(_("cannot delete applied patch %s") % patch)
493 493 if patch not in self.series:
494 494 raise util.Abort(_("patch %s not in series file") % patch)
495 495 realpatches.append(patch)
496 496
497 497 if not keep:
498 498 r = self.qrepo()
499 499 if r:
500 500 r.remove(realpatches, True)
501 501 else:
502 502 os.unlink(self.join(patch))
503 503
504 504 indices = [self.find_series(p) for p in realpatches]
505 505 indices.sort()
506 506 for i in indices[-1::-1]:
507 507 del self.full_series[i]
508 508 self.parse_series()
509 509 self.series_dirty = 1
510 510
511 511 def check_toppatch(self, repo):
512 512 if len(self.applied) > 0:
513 513 top = revlog.bin(self.applied[-1].rev)
514 514 pp = repo.dirstate.parents()
515 515 if top not in pp:
516 516 raise util.Abort(_("queue top not at same revision as working directory"))
517 517 return top
518 518 return None
519 519 def check_localchanges(self, repo, force=False, refresh=True):
520 520 m, a, r, d = repo.status()[:4]
521 521 if m or a or r or d:
522 522 if not force:
523 523 if refresh:
524 524 raise util.Abort(_("local changes found, refresh first"))
525 525 else:
526 526 raise util.Abort(_("local changes found"))
527 527 return m, a, r, d
528 528 def new(self, repo, patch, msg=None, force=None):
529 529 if os.path.exists(self.join(patch)):
530 530 raise util.Abort(_('patch "%s" already exists') % patch)
531 531 m, a, r, d = self.check_localchanges(repo, force)
532 532 commitfiles = m + a + r
533 533 self.check_toppatch(repo)
534 534 wlock = repo.wlock()
535 535 insert = self.full_series_end()
536 536 if msg:
537 537 n = repo.commit(commitfiles, "[mq]: %s" % msg, force=True,
538 538 wlock=wlock)
539 539 else:
540 540 n = repo.commit(commitfiles,
541 541 "New patch: %s" % patch, force=True, wlock=wlock)
542 542 if n == None:
543 543 raise util.Abort(_("repo commit failed"))
544 544 self.full_series[insert:insert] = [patch]
545 545 self.applied.append(statusentry(revlog.hex(n), patch))
546 546 self.parse_series()
547 547 self.series_dirty = 1
548 548 self.applied_dirty = 1
549 549 p = self.opener(patch, "w")
550 550 if msg:
551 551 msg = msg + "\n"
552 552 p.write(msg)
553 553 p.close()
554 554 wlock = None
555 555 r = self.qrepo()
556 556 if r: r.add([patch])
557 557 if commitfiles:
558 558 self.refresh(repo, short=True)
559 559
560 560 def strip(self, repo, rev, update=True, backup="all", wlock=None):
561 561 def limitheads(chlog, stop):
562 562 """return the list of all nodes that have no children"""
563 563 p = {}
564 564 h = []
565 565 stoprev = 0
566 566 if stop in chlog.nodemap:
567 567 stoprev = chlog.rev(stop)
568 568
569 569 for r in range(chlog.count() - 1, -1, -1):
570 570 n = chlog.node(r)
571 571 if n not in p:
572 572 h.append(n)
573 573 if n == stop:
574 574 break
575 575 if r < stoprev:
576 576 break
577 577 for pn in chlog.parents(n):
578 578 p[pn] = 1
579 579 return h
580 580
581 581 def bundle(cg):
582 582 backupdir = repo.join("strip-backup")
583 583 if not os.path.isdir(backupdir):
584 584 os.mkdir(backupdir)
585 585 name = os.path.join(backupdir, "%s" % revlog.short(rev))
586 586 name = savename(name)
587 587 self.ui.warn("saving bundle to %s\n" % name)
588 588 # TODO, exclusive open
589 589 f = open(name, "wb")
590 590 try:
591 591 f.write("HG10")
592 592 z = bz2.BZ2Compressor(9)
593 593 while 1:
594 594 chunk = cg.read(4096)
595 595 if not chunk:
596 596 break
597 597 f.write(z.compress(chunk))
598 598 f.write(z.flush())
599 599 except:
600 600 os.unlink(name)
601 601 raise
602 602 f.close()
603 603 return name
604 604
605 605 def stripall(rev, revnum):
606 606 cl = repo.changelog
607 607 c = cl.read(rev)
608 608 mm = repo.manifest.read(c[0])
609 609 seen = {}
610 610
611 611 for x in xrange(revnum, cl.count()):
612 612 c = cl.read(cl.node(x))
613 613 for f in c[3]:
614 614 if f in seen:
615 615 continue
616 616 seen[f] = 1
617 617 if f in mm:
618 618 filerev = mm[f]
619 619 else:
620 620 filerev = 0
621 621 seen[f] = filerev
622 622 # we go in two steps here so the strip loop happens in a
623 623 # sensible order. When stripping many files, this helps keep
624 624 # our disk access patterns under control.
625 625 seen_list = seen.keys()
626 626 seen_list.sort()
627 627 for f in seen_list:
628 628 ff = repo.file(f)
629 629 filerev = seen[f]
630 630 if filerev != 0:
631 631 if filerev in ff.nodemap:
632 632 filerev = ff.rev(filerev)
633 633 else:
634 634 filerev = 0
635 635 ff.strip(filerev, revnum)
636 636
637 637 if not wlock:
638 638 wlock = repo.wlock()
639 639 lock = repo.lock()
640 640 chlog = repo.changelog
641 641 # TODO delete the undo files, and handle undo of merge sets
642 642 pp = chlog.parents(rev)
643 643 revnum = chlog.rev(rev)
644 644
645 645 if update:
646 646 self.check_localchanges(repo, refresh=False)
647 647 urev = self.qparents(repo, rev)
648 648 hg.clean(repo, urev, wlock=wlock)
649 649 repo.dirstate.write()
650 650
651 651 # save is a list of all the branches we are truncating away
652 652 # that we actually want to keep. changegroup will be used
653 653 # to preserve them and add them back after the truncate
654 654 saveheads = []
655 655 savebases = {}
656 656
657 657 heads = limitheads(chlog, rev)
658 658 seen = {}
659 659
660 660 # search through all the heads, finding those where the revision
661 661 # we want to strip away is an ancestor. Also look for merges
662 662 # that might be turned into new heads by the strip.
663 663 while heads:
664 664 h = heads.pop()
665 665 n = h
666 666 while True:
667 667 seen[n] = 1
668 668 pp = chlog.parents(n)
669 669 if pp[1] != revlog.nullid and chlog.rev(pp[1]) > revnum:
670 670 if pp[1] not in seen:
671 671 heads.append(pp[1])
672 672 if pp[0] == revlog.nullid:
673 673 break
674 674 if chlog.rev(pp[0]) < revnum:
675 675 break
676 676 n = pp[0]
677 677 if n == rev:
678 678 break
679 679 r = chlog.reachable(h, rev)
680 680 if rev not in r:
681 681 saveheads.append(h)
682 682 for x in r:
683 683 if chlog.rev(x) > revnum:
684 684 savebases[x] = 1
685 685
686 686 # create a changegroup for all the branches we need to keep
687 687 if backup == "all":
688 688 backupch = repo.changegroupsubset([rev], chlog.heads(), 'strip')
689 689 bundle(backupch)
690 690 if saveheads:
691 691 backupch = repo.changegroupsubset(savebases.keys(), saveheads, 'strip')
692 692 chgrpfile = bundle(backupch)
693 693
694 694 stripall(rev, revnum)
695 695
696 696 change = chlog.read(rev)
697 697 repo.manifest.strip(repo.manifest.rev(change[0]), revnum)
698 698 chlog.strip(revnum, revnum)
699 699 if saveheads:
700 700 self.ui.status("adding branch\n")
701 701 commands.unbundle(self.ui, repo, chgrpfile, update=False)
702 702 if backup != "strip":
703 703 os.unlink(chgrpfile)
704 704
705 705 def isapplied(self, patch):
706 706 """returns (index, rev, patch)"""
707 707 for i in xrange(len(self.applied)):
708 708 a = self.applied[i]
709 709 if a.name == patch:
710 710 return (i, a.rev, a.name)
711 711 return None
712 712
713 713 # if the exact patch name does not exist, we try a few
714 714 # variations. If strict is passed, we try only #1
715 715 #
716 716 # 1) a number to indicate an offset in the series file
717 717 # 2) a unique substring of the patch name was given
718 718 # 3) patchname[-+]num to indicate an offset in the series file
719 719 def lookup(self, patch, strict=False):
720 720 patch = patch and str(patch)
721 721
722 722 def partial_name(s):
723 723 if s in self.series:
724 724 return s
725 725 matches = [x for x in self.series if s in x]
726 726 if len(matches) > 1:
727 727 self.ui.warn(_('patch name "%s" is ambiguous:\n') % s)
728 728 for m in matches:
729 729 self.ui.warn(' %s\n' % m)
730 730 return None
731 731 if matches:
732 732 return matches[0]
733 733 if len(self.series) > 0 and len(self.applied) > 0:
734 734 if s == 'qtip':
735 735 return self.series[self.series_end()-1]
736 736 if s == 'qbase':
737 737 return self.series[0]
738 738 return None
739 739 if patch == None:
740 740 return None
741 741
742 742 # we don't want to return a partial match until we make
743 743 # sure the file name passed in does not exist (checked below)
744 744 res = partial_name(patch)
745 745 if res and res == patch:
746 746 return res
747 747
748 748 if not os.path.isfile(self.join(patch)):
749 749 try:
750 750 sno = int(patch)
751 751 except(ValueError, OverflowError):
752 752 pass
753 753 else:
754 754 if sno < len(self.series):
755 755 return self.series[sno]
756 756 if not strict:
757 757 # return any partial match made above
758 758 if res:
759 759 return res
760 760 minus = patch.rsplit('-', 1)
761 761 if len(minus) > 1:
762 762 res = partial_name(minus[0])
763 763 if res:
764 764 i = self.series.index(res)
765 765 try:
766 766 off = int(minus[1] or 1)
767 767 except(ValueError, OverflowError):
768 768 pass
769 769 else:
770 770 if i - off >= 0:
771 771 return self.series[i - off]
772 772 plus = patch.rsplit('+', 1)
773 773 if len(plus) > 1:
774 774 res = partial_name(plus[0])
775 775 if res:
776 776 i = self.series.index(res)
777 777 try:
778 778 off = int(plus[1] or 1)
779 779 except(ValueError, OverflowError):
780 780 pass
781 781 else:
782 782 if i + off < len(self.series):
783 783 return self.series[i + off]
784 784 raise util.Abort(_("patch %s not in series") % patch)
785 785
786 786 def push(self, repo, patch=None, force=False, list=False,
787 787 mergeq=None, wlock=None):
788 788 if not wlock:
789 789 wlock = repo.wlock()
790 790 patch = self.lookup(patch)
791 791 if patch and self.isapplied(patch):
792 792 self.ui.warn(_("patch %s is already applied\n") % patch)
793 793 sys.exit(1)
794 794 if self.series_end() == len(self.series):
795 795 self.ui.warn(_("patch series fully applied\n"))
796 796 sys.exit(1)
797 797 if not force:
798 798 self.check_localchanges(repo)
799 799
800 800 self.applied_dirty = 1;
801 801 start = self.series_end()
802 802 if start > 0:
803 803 self.check_toppatch(repo)
804 804 if not patch:
805 805 patch = self.series[start]
806 806 end = start + 1
807 807 else:
808 808 end = self.series.index(patch, start) + 1
809 809 s = self.series[start:end]
810 810 if mergeq:
811 811 ret = self.mergepatch(repo, mergeq, s, wlock)
812 812 else:
813 813 ret = self.apply(repo, s, list, wlock=wlock)
814 814 top = self.applied[-1].name
815 815 if ret[0]:
816 816 self.ui.write("Errors during apply, please fix and refresh %s\n" %
817 817 top)
818 818 else:
819 819 self.ui.write("Now at: %s\n" % top)
820 820 return ret[0]
821 821
822 822 def pop(self, repo, patch=None, force=False, update=True, all=False,
823 823 wlock=None):
824 824 def getfile(f, rev):
825 825 t = repo.file(f).read(rev)
826 826 try:
827 827 repo.wfile(f, "w").write(t)
828 828 except IOError:
829 829 try:
830 830 os.makedirs(os.path.dirname(repo.wjoin(f)))
831 831 except OSError, err:
832 832 if err.errno != errno.EEXIST: raise
833 833 repo.wfile(f, "w").write(t)
834 834
835 835 if not wlock:
836 836 wlock = repo.wlock()
837 837 if patch:
838 838 # index, rev, patch
839 839 info = self.isapplied(patch)
840 840 if not info:
841 841 patch = self.lookup(patch)
842 842 info = self.isapplied(patch)
843 843 if not info:
844 844 raise util.Abort(_("patch %s is not applied") % patch)
845 845 if len(self.applied) == 0:
846 846 self.ui.warn(_("no patches applied\n"))
847 847 sys.exit(1)
848 848
849 849 if not update:
850 850 parents = repo.dirstate.parents()
851 851 rr = [ revlog.bin(x.rev) for x in self.applied ]
852 852 for p in parents:
853 853 if p in rr:
854 854 self.ui.warn("qpop: forcing dirstate update\n")
855 855 update = True
856 856
857 857 if not force and update:
858 858 self.check_localchanges(repo)
859 859
860 860 self.applied_dirty = 1;
861 861 end = len(self.applied)
862 862 if not patch:
863 863 if all:
864 864 popi = 0
865 865 else:
866 866 popi = len(self.applied) - 1
867 867 else:
868 868 popi = info[0] + 1
869 869 if popi >= end:
870 870 self.ui.warn("qpop: %s is already at the top\n" % patch)
871 871 return
872 872 info = [ popi ] + [self.applied[popi].rev, self.applied[popi].name]
873 873
874 874 start = info[0]
875 875 rev = revlog.bin(info[1])
876 876
877 877 # we know there are no local changes, so we can make a simplified
878 878 # form of hg.update.
879 879 if update:
880 880 top = self.check_toppatch(repo)
881 881 qp = self.qparents(repo, rev)
882 882 changes = repo.changelog.read(qp)
883 883 mmap = repo.manifest.read(changes[0])
884 884 m, a, r, d, u = repo.status(qp, top)[:5]
885 885 if d:
886 886 raise util.Abort("deletions found between repo revs")
887 887 for f in m:
888 888 getfile(f, mmap[f])
889 889 for f in r:
890 890 getfile(f, mmap[f])
891 891 util.set_exec(repo.wjoin(f), mmap.execf(f))
892 892 repo.dirstate.update(m + r, 'n')
893 893 for f in a:
894 894 try: os.unlink(repo.wjoin(f))
895 895 except: raise
896 896 try: os.removedirs(os.path.dirname(repo.wjoin(f)))
897 897 except: pass
898 898 if a:
899 899 repo.dirstate.forget(a)
900 900 repo.dirstate.setparents(qp, revlog.nullid)
901 901 self.strip(repo, rev, update=False, backup='strip', wlock=wlock)
902 902 del self.applied[start:end]
903 903 if len(self.applied):
904 904 self.ui.write("Now at: %s\n" % self.applied[-1].name)
905 905 else:
906 906 self.ui.write("Patch queue now empty\n")
907 907
908 908 def diff(self, repo, pats, opts):
909 909 top = self.check_toppatch(repo)
910 910 if not top:
911 911 self.ui.write("No patches applied\n")
912 912 return
913 913 qp = self.qparents(repo, top)
914 914 self.printdiff(repo, qp, files=pats, opts=opts)
915 915
916 916 def refresh(self, repo, pats=None, **opts):
917 917 if len(self.applied) == 0:
918 918 self.ui.write("No patches applied\n")
919 919 return 1
920 920 wlock = repo.wlock()
921 921 self.check_toppatch(repo)
922 922 (top, patchfn) = (self.applied[-1].rev, self.applied[-1].name)
923 923 top = revlog.bin(top)
924 924 cparents = repo.changelog.parents(top)
925 925 patchparent = self.qparents(repo, top)
926 926 message, comments, user, date, patchfound = self.readheaders(patchfn)
927 927
928 928 patchf = self.opener(patchfn, "w")
929 929 msg = opts.get('msg', '').rstrip()
930 930 if msg:
931 931 if comments:
932 932 # Remove existing message.
933 933 ci = 0
934 934 for mi in range(len(message)):
935 935 while message[mi] != comments[ci]:
936 936 ci += 1
937 937 del comments[ci]
938 938 comments.append(msg)
939 939 if comments:
940 940 comments = "\n".join(comments) + '\n\n'
941 941 patchf.write(comments)
942 942
943 943 fns, matchfn, anypats = cmdutil.matchpats(repo, pats, opts)
944 944 tip = repo.changelog.tip()
945 945 if top == tip:
946 946 # if the top of our patch queue is also the tip, there is an
947 947 # optimization here. We update the dirstate in place and strip
948 948 # off the tip commit. Then just commit the current directory
949 949 # tree. We can also send repo.commit the list of files
950 950 # changed to speed up the diff
951 951 #
952 952 # in short mode, we only diff the files included in the
953 953 # patch already
954 954 #
955 955 # this should really read:
956 956 # mm, dd, aa, aa2, uu = repo.status(tip, patchparent)[:5]
957 957 # but we do it backwards to take advantage of manifest/chlog
958 958 # caching against the next repo.status call
959 959 #
960 960 mm, aa, dd, aa2, uu = repo.status(patchparent, tip)[:5]
961 961 if opts.get('short'):
962 962 filelist = mm + aa + dd
963 963 else:
964 964 filelist = None
965 965 m, a, r, d, u = repo.status(files=filelist)[:5]
966 966
967 967 # we might end up with files that were added between tip and
968 968 # the dirstate parent, but then changed in the local dirstate.
969 969 # in this case, we want them to only show up in the added section
970 970 for x in m:
971 971 if x not in aa:
972 972 mm.append(x)
973 973 # we might end up with files added by the local dirstate that
974 974 # were deleted by the patch. In this case, they should only
975 975 # show up in the changed section.
976 976 for x in a:
977 977 if x in dd:
978 978 del dd[dd.index(x)]
979 979 mm.append(x)
980 980 else:
981 981 aa.append(x)
982 982 # make sure any files deleted in the local dirstate
983 983 # are not in the add or change column of the patch
984 984 forget = []
985 985 for x in d + r:
986 986 if x in aa:
987 987 del aa[aa.index(x)]
988 988 forget.append(x)
989 989 continue
990 990 elif x in mm:
991 991 del mm[mm.index(x)]
992 992 dd.append(x)
993 993
994 994 m = list(util.unique(mm))
995 995 r = list(util.unique(dd))
996 996 a = list(util.unique(aa))
997 997 filelist = filter(matchfn, util.unique(m + r + a))
998 998 patch.diff(repo, patchparent, files=filelist, match=matchfn,
999 999 fp=patchf, changes=(m, a, r, [], u),
1000 1000 opts=self.diffopts())
1001 1001 patchf.close()
1002 1002
1003 1003 changes = repo.changelog.read(tip)
1004 1004 repo.dirstate.setparents(*cparents)
1005 1005 copies = [(f, repo.dirstate.copied(f)) for f in a]
1006 1006 repo.dirstate.update(a, 'a')
1007 1007 for dst, src in copies:
1008 1008 repo.dirstate.copy(src, dst)
1009 1009 repo.dirstate.update(r, 'r')
1010 1010 # if the patch excludes a modified file, mark that file with mtime=0
1011 1011 # so status can see it.
1012 1012 mm = []
1013 1013 for i in range(len(m)-1, -1, -1):
1014 1014 if not matchfn(m[i]):
1015 1015 mm.append(m[i])
1016 1016 del m[i]
1017 1017 repo.dirstate.update(m, 'n')
1018 1018 repo.dirstate.update(mm, 'n', st_mtime=0)
1019 1019 repo.dirstate.forget(forget)
1020 1020
1021 1021 if not msg:
1022 1022 if not message:
1023 1023 message = "patch queue: %s\n" % patchfn
1024 1024 else:
1025 1025 message = "\n".join(message)
1026 1026 else:
1027 1027 message = msg
1028 1028
1029 1029 self.strip(repo, top, update=False, backup='strip', wlock=wlock)
1030 1030 n = repo.commit(filelist, message, changes[1], force=1, wlock=wlock)
1031 1031 self.applied[-1] = statusentry(revlog.hex(n), patchfn)
1032 1032 self.applied_dirty = 1
1033 1033 else:
1034 1034 self.printdiff(repo, patchparent, fp=patchf)
1035 1035 patchf.close()
1036 1036 self.pop(repo, force=True, wlock=wlock)
1037 1037 self.push(repo, force=True, wlock=wlock)
1038 1038
1039 1039 def init(self, repo, create=False):
1040 1040 if os.path.isdir(self.path):
1041 1041 raise util.Abort(_("patch queue directory already exists"))
1042 1042 os.mkdir(self.path)
1043 1043 if create:
1044 1044 return self.qrepo(create=True)
1045 1045
1046 1046 def unapplied(self, repo, patch=None):
1047 1047 if patch and patch not in self.series:
1048 1048 raise util.Abort(_("patch %s is not in series file") % patch)
1049 1049 if not patch:
1050 1050 start = self.series_end()
1051 1051 else:
1052 1052 start = self.series.index(patch) + 1
1053 1053 unapplied = []
1054 1054 for i in xrange(start, len(self.series)):
1055 1055 pushable, reason = self.pushable(i)
1056 1056 if pushable:
1057 1057 unapplied.append((i, self.series[i]))
1058 1058 self.explain_pushable(i)
1059 1059 return unapplied
1060 1060
1061 1061 def qseries(self, repo, missing=None, summary=False):
1062 1062 start = self.series_end(all_patches=True)
1063 1063 if not missing:
1064 1064 for i in range(len(self.series)):
1065 1065 patch = self.series[i]
1066 1066 if self.ui.verbose:
1067 1067 if i < start:
1068 1068 status = 'A'
1069 1069 elif self.pushable(i)[0]:
1070 1070 status = 'U'
1071 1071 else:
1072 1072 status = 'G'
1073 1073 self.ui.write('%d %s ' % (i, status))
1074 1074 if summary:
1075 1075 msg = self.readheaders(patch)[0]
1076 1076 msg = msg and ': ' + msg[0] or ': '
1077 1077 else:
1078 1078 msg = ''
1079 1079 self.ui.write('%s%s\n' % (patch, msg))
1080 1080 else:
1081 1081 msng_list = []
1082 1082 for root, dirs, files in os.walk(self.path):
1083 1083 d = root[len(self.path) + 1:]
1084 1084 for f in files:
1085 1085 fl = os.path.join(d, f)
1086 1086 if (fl not in self.series and
1087 1087 fl not in (self.status_path, self.series_path)
1088 1088 and not fl.startswith('.')):
1089 1089 msng_list.append(fl)
1090 1090 msng_list.sort()
1091 1091 for x in msng_list:
1092 1092 if self.ui.verbose:
1093 1093 self.ui.write("D ")
1094 1094 self.ui.write("%s\n" % x)
1095 1095
1096 1096 def issaveline(self, l):
1097 1097 if l.name == '.hg.patches.save.line':
1098 1098 return True
1099 1099
1100 1100 def qrepo(self, create=False):
1101 1101 if create or os.path.isdir(self.join(".hg")):
1102 1102 return hg.repository(self.ui, path=self.path, create=create)
1103 1103
1104 1104 def restore(self, repo, rev, delete=None, qupdate=None):
1105 1105 c = repo.changelog.read(rev)
1106 1106 desc = c[4].strip()
1107 1107 lines = desc.splitlines()
1108 1108 i = 0
1109 1109 datastart = None
1110 1110 series = []
1111 1111 applied = []
1112 1112 qpp = None
1113 1113 for i in xrange(0, len(lines)):
1114 1114 if lines[i] == 'Patch Data:':
1115 1115 datastart = i + 1
1116 1116 elif lines[i].startswith('Dirstate:'):
1117 1117 l = lines[i].rstrip()
1118 1118 l = l[10:].split(' ')
1119 1119 qpp = [ hg.bin(x) for x in l ]
1120 1120 elif datastart != None:
1121 1121 l = lines[i].rstrip()
1122 1122 se = statusentry(l)
1123 1123 file_ = se.name
1124 1124 if se.rev:
1125 1125 applied.append(se)
1126 1126 series.append(file_)
1127 1127 if datastart == None:
1128 1128 self.ui.warn("No saved patch data found\n")
1129 1129 return 1
1130 1130 self.ui.warn("restoring status: %s\n" % lines[0])
1131 1131 self.full_series = series
1132 1132 self.applied = applied
1133 1133 self.parse_series()
1134 1134 self.series_dirty = 1
1135 1135 self.applied_dirty = 1
1136 1136 heads = repo.changelog.heads()
1137 1137 if delete:
1138 1138 if rev not in heads:
1139 1139 self.ui.warn("save entry has children, leaving it alone\n")
1140 1140 else:
1141 1141 self.ui.warn("removing save entry %s\n" % hg.short(rev))
1142 1142 pp = repo.dirstate.parents()
1143 1143 if rev in pp:
1144 1144 update = True
1145 1145 else:
1146 1146 update = False
1147 1147 self.strip(repo, rev, update=update, backup='strip')
1148 1148 if qpp:
1149 1149 self.ui.warn("saved queue repository parents: %s %s\n" %
1150 1150 (hg.short(qpp[0]), hg.short(qpp[1])))
1151 1151 if qupdate:
1152 1152 print "queue directory updating"
1153 1153 r = self.qrepo()
1154 1154 if not r:
1155 1155 self.ui.warn("Unable to load queue repository\n")
1156 1156 return 1
1157 1157 hg.clean(r, qpp[0])
1158 1158
1159 1159 def save(self, repo, msg=None):
1160 1160 if len(self.applied) == 0:
1161 1161 self.ui.warn("save: no patches applied, exiting\n")
1162 1162 return 1
1163 1163 if self.issaveline(self.applied[-1]):
1164 1164 self.ui.warn("status is already saved\n")
1165 1165 return 1
1166 1166
1167 1167 ar = [ ':' + x for x in self.full_series ]
1168 1168 if not msg:
1169 1169 msg = "hg patches saved state"
1170 1170 else:
1171 1171 msg = "hg patches: " + msg.rstrip('\r\n')
1172 1172 r = self.qrepo()
1173 1173 if r:
1174 1174 pp = r.dirstate.parents()
1175 1175 msg += "\nDirstate: %s %s" % (hg.hex(pp[0]), hg.hex(pp[1]))
1176 1176 msg += "\n\nPatch Data:\n"
1177 1177 text = msg + "\n".join([str(x) for x in self.applied]) + '\n' + (ar and
1178 1178 "\n".join(ar) + '\n' or "")
1179 1179 n = repo.commit(None, text, user=None, force=1)
1180 1180 if not n:
1181 1181 self.ui.warn("repo commit failed\n")
1182 1182 return 1
1183 1183 self.applied.append(statusentry(revlog.hex(n),'.hg.patches.save.line'))
1184 1184 self.applied_dirty = 1
1185 1185
1186 1186 def full_series_end(self):
1187 1187 if len(self.applied) > 0:
1188 1188 p = self.applied[-1].name
1189 1189 end = self.find_series(p)
1190 1190 if end == None:
1191 1191 return len(self.full_series)
1192 1192 return end + 1
1193 1193 return 0
1194 1194
1195 1195 def series_end(self, all_patches=False):
1196 1196 end = 0
1197 1197 def next(start):
1198 1198 if all_patches:
1199 1199 return start
1200 1200 i = start
1201 1201 while i < len(self.series):
1202 1202 p, reason = self.pushable(i)
1203 1203 if p:
1204 1204 break
1205 1205 self.explain_pushable(i)
1206 1206 i += 1
1207 1207 return i
1208 1208 if len(self.applied) > 0:
1209 1209 p = self.applied[-1].name
1210 1210 try:
1211 1211 end = self.series.index(p)
1212 1212 except ValueError:
1213 1213 return 0
1214 1214 return next(end + 1)
1215 1215 return next(end)
1216 1216
1217 1217 def qapplied(self, repo, patch=None):
1218 1218 if patch and patch not in self.series:
1219 1219 raise util.Abort(_("patch %s is not in series file") % patch)
1220 1220 if not patch:
1221 1221 end = len(self.applied)
1222 1222 else:
1223 1223 end = self.series.index(patch) + 1
1224 1224 for x in xrange(end):
1225 1225 p = self.appliedname(x)
1226 1226 self.ui.write("%s\n" % p)
1227 1227
1228 1228 def appliedname(self, index):
1229 1229 pname = self.applied[index].name
1230 1230 if not self.ui.verbose:
1231 1231 p = pname
1232 1232 else:
1233 1233 p = str(self.series.index(pname)) + " " + pname
1234 1234 return p
1235 1235
1236 1236 def top(self, repo):
1237 1237 if len(self.applied):
1238 1238 p = self.appliedname(-1)
1239 1239 self.ui.write(p + '\n')
1240 1240 else:
1241 1241 self.ui.write("No patches applied\n")
1242 1242 return 1
1243 1243
1244 1244 def next(self, repo):
1245 1245 end = self.series_end()
1246 1246 if end == len(self.series):
1247 1247 self.ui.write("All patches applied\n")
1248 1248 return 1
1249 1249 else:
1250 1250 p = self.series[end]
1251 1251 if self.ui.verbose:
1252 1252 self.ui.write("%d " % self.series.index(p))
1253 1253 self.ui.write(p + '\n')
1254 1254
1255 1255 def prev(self, repo):
1256 1256 if len(self.applied) > 1:
1257 1257 p = self.appliedname(-2)
1258 1258 self.ui.write(p + '\n')
1259 1259 elif len(self.applied) == 1:
1260 1260 self.ui.write("Only one patch applied\n")
1261 1261 return 1
1262 1262 else:
1263 1263 self.ui.write("No patches applied\n")
1264 1264 return 1
1265 1265
1266 1266 def qimport(self, repo, files, patch=None, existing=None, force=None):
1267 1267 if len(files) > 1 and patch:
1268 1268 raise util.Abort(_('option "-n" not valid when importing multiple '
1269 1269 'files'))
1270 1270 i = 0
1271 1271 added = []
1272 1272 for filename in files:
1273 1273 if existing:
1274 1274 if not patch:
1275 1275 patch = filename
1276 1276 if not os.path.isfile(self.join(patch)):
1277 1277 raise util.Abort(_("patch %s does not exist") % patch)
1278 1278 else:
1279 1279 try:
1280 1280 text = file(filename).read()
1281 1281 except IOError:
1282 1282 raise util.Abort(_("unable to read %s") % patch)
1283 1283 if not patch:
1284 1284 patch = os.path.split(filename)[1]
1285 1285 if not force and os.path.exists(self.join(patch)):
1286 1286 raise util.Abort(_('patch "%s" already exists') % patch)
1287 1287 patchf = self.opener(patch, "w")
1288 1288 patchf.write(text)
1289 1289 if patch in self.series:
1290 1290 raise util.Abort(_('patch %s is already in the series file')
1291 1291 % patch)
1292 1292 index = self.full_series_end() + i
1293 1293 self.full_series[index:index] = [patch]
1294 1294 self.parse_series()
1295 1295 self.ui.warn("adding %s to series file\n" % patch)
1296 1296 i += 1
1297 1297 added.append(patch)
1298 1298 patch = None
1299 1299 self.series_dirty = 1
1300 1300 qrepo = self.qrepo()
1301 1301 if qrepo:
1302 1302 qrepo.add(added)
1303 1303
1304 1304 def delete(ui, repo, patch, *patches, **opts):
1305 1305 """remove patches from queue
1306 1306
1307 1307 The patches must not be applied.
1308 1308 With -k, the patch files are preserved in the patch directory."""
1309 1309 q = repo.mq
1310 1310 q.delete(repo, (patch,) + patches, keep=opts.get('keep'))
1311 1311 q.save_dirty()
1312 1312 return 0
1313 1313
1314 1314 def applied(ui, repo, patch=None, **opts):
1315 1315 """print the patches already applied"""
1316 1316 repo.mq.qapplied(repo, patch)
1317 1317 return 0
1318 1318
1319 1319 def unapplied(ui, repo, patch=None, **opts):
1320 1320 """print the patches not yet applied"""
1321 1321 for i, p in repo.mq.unapplied(repo, patch):
1322 1322 if ui.verbose:
1323 1323 ui.write("%d " % i)
1324 1324 ui.write("%s\n" % p)
1325 1325
1326 1326 def qimport(ui, repo, *filename, **opts):
1327 1327 """import a patch"""
1328 1328 q = repo.mq
1329 1329 q.qimport(repo, filename, patch=opts['name'],
1330 1330 existing=opts['existing'], force=opts['force'])
1331 1331 q.save_dirty()
1332 1332 return 0
1333 1333
1334 1334 def init(ui, repo, **opts):
1335 1335 """init a new queue repository
1336 1336
1337 1337 The queue repository is unversioned by default. If -c is
1338 1338 specified, qinit will create a separate nested repository
1339 1339 for patches. Use qcommit to commit changes to this queue
1340 1340 repository."""
1341 1341 q = repo.mq
1342 1342 r = q.init(repo, create=opts['create_repo'])
1343 1343 q.save_dirty()
1344 1344 if r:
1345 1345 fp = r.wopener('.hgignore', 'w')
1346 1346 print >> fp, 'syntax: glob'
1347 1347 print >> fp, 'status'
1348 1348 fp.close()
1349 1349 r.wopener('series', 'w').close()
1350 1350 r.add(['.hgignore', 'series'])
1351 1351 return 0
1352 1352
1353 1353 def clone(ui, source, dest=None, **opts):
1354 1354 '''clone main and patch repository at same time
1355 1355
1356 1356 If source is local, destination will have no patches applied. If
1357 1357 source is remote, this command can not check if patches are
1358 1358 applied in source, so cannot guarantee that patches are not
1359 1359 applied in destination. If you clone remote repository, be sure
1360 1360 before that it has no patches applied.
1361 1361
1362 1362 Source patch repository is looked for in <src>/.hg/patches by
1363 1363 default. Use -p <url> to change.
1364 1364 '''
1365 1365 commands.setremoteconfig(ui, opts)
1366 1366 if dest is None:
1367 1367 dest = hg.defaultdest(source)
1368 1368 sr = hg.repository(ui, ui.expandpath(source))
1369 1369 qbase, destrev = None, None
1370 1370 if sr.local():
1371 1371 reposetup(ui, sr)
1372 1372 if sr.mq.applied:
1373 1373 qbase = revlog.bin(sr.mq.applied[0].rev)
1374 1374 if not hg.islocal(dest):
1375 1375 destrev = sr.parents(qbase)[0]
1376 1376 ui.note(_('cloning main repo\n'))
1377 1377 sr, dr = hg.clone(ui, sr, dest,
1378 1378 pull=opts['pull'],
1379 1379 rev=destrev,
1380 1380 update=False,
1381 1381 stream=opts['uncompressed'])
1382 1382 ui.note(_('cloning patch repo\n'))
1383 1383 spr, dpr = hg.clone(ui, opts['patches'] or (sr.url() + '/.hg/patches'),
1384 1384 dr.url() + '/.hg/patches',
1385 1385 pull=opts['pull'],
1386 1386 update=not opts['noupdate'],
1387 1387 stream=opts['uncompressed'])
1388 1388 if dr.local():
1389 1389 if qbase:
1390 1390 ui.note(_('stripping applied patches from destination repo\n'))
1391 1391 reposetup(ui, dr)
1392 1392 dr.mq.strip(dr, qbase, update=False, backup=None)
1393 1393 if not opts['noupdate']:
1394 1394 ui.note(_('updating destination repo\n'))
1395 1395 hg.update(dr, dr.changelog.tip())
1396 1396
1397 1397 def commit(ui, repo, *pats, **opts):
1398 1398 """commit changes in the queue repository"""
1399 1399 q = repo.mq
1400 1400 r = q.qrepo()
1401 1401 if not r: raise util.Abort('no queue repository')
1402 1402 commands.commit(r.ui, r, *pats, **opts)
1403 1403
1404 1404 def series(ui, repo, **opts):
1405 1405 """print the entire series file"""
1406 1406 repo.mq.qseries(repo, missing=opts['missing'], summary=opts['summary'])
1407 1407 return 0
1408 1408
1409 1409 def top(ui, repo, **opts):
1410 1410 """print the name of the current patch"""
1411 1411 return repo.mq.top(repo)
1412 1412
1413 1413 def next(ui, repo, **opts):
1414 1414 """print the name of the next patch"""
1415 1415 return repo.mq.next(repo)
1416 1416
1417 1417 def prev(ui, repo, **opts):
1418 1418 """print the name of the previous patch"""
1419 1419 return repo.mq.prev(repo)
1420 1420
1421 1421 def new(ui, repo, patch, **opts):
1422 1422 """create a new patch
1423 1423
1424 1424 qnew creates a new patch on top of the currently-applied patch
1425 1425 (if any). It will refuse to run if there are any outstanding
1426 1426 changes unless -f is specified, in which case the patch will
1427 1427 be initialised with them.
1428 1428
1429 1429 -e, -m or -l set the patch header as well as the commit message.
1430 1430 If none is specified, the patch header is empty and the
1431 1431 commit message is 'New patch: PATCH'"""
1432 1432 q = repo.mq
1433 1433 message = commands.logmessage(opts)
1434 1434 if opts['edit']:
1435 1435 message = ui.edit(message, ui.username())
1436 1436 q.new(repo, patch, msg=message, force=opts['force'])
1437 1437 q.save_dirty()
1438 1438 return 0
1439 1439
1440 1440 def refresh(ui, repo, *pats, **opts):
1441 1441 """update the current patch
1442 1442
1443 1443 If any file patterns are provided, the refreshed patch will contain only
1444 1444 the modifications that match those patterns; the remaining modifications
1445 1445 will remain in the working directory.
1446 1446 """
1447 1447 q = repo.mq
1448 1448 message = commands.logmessage(opts)
1449 1449 if opts['edit']:
1450 1450 if message:
1451 1451 raise util.Abort(_('option "-e" incompatible with "-m" or "-l"'))
1452 1452 patch = q.applied[-1].name
1453 1453 (message, comment, user, date, hasdiff) = q.readheaders(patch)
1454 1454 message = ui.edit('\n'.join(message), user or ui.username())
1455 1455 ret = q.refresh(repo, pats, msg=message, **opts)
1456 1456 q.save_dirty()
1457 1457 return ret
1458 1458
1459 1459 def diff(ui, repo, *pats, **opts):
1460 1460 """diff of the current patch"""
1461 1461 repo.mq.diff(repo, pats, opts)
1462 1462 return 0
1463 1463
1464 1464 def fold(ui, repo, *files, **opts):
1465 1465 """fold the named patches into the current patch
1466 1466
1467 1467 Patches must not yet be applied. Each patch will be successively
1468 1468 applied to the current patch in the order given. If all the
1469 1469 patches apply successfully, the current patch will be refreshed
1470 1470 with the new cumulative patch, and the folded patches will
1471 1471 be deleted. With -k/--keep, the folded patch files will not
1472 1472 be removed afterwards.
1473 1473
1474 1474 The header for each folded patch will be concatenated with
1475 1475 the current patch header, separated by a line of '* * *'."""
1476 1476
1477 1477 q = repo.mq
1478 1478
1479 1479 if not files:
1480 1480 raise util.Abort(_('qfold requires at least one patch name'))
1481 1481 if not q.check_toppatch(repo):
1482 raise util.Abort(_('No patches applied\n'))
1482 raise util.Abort(_('No patches applied'))
1483 1483
1484 1484 message = commands.logmessage(opts)
1485 1485 if opts['edit']:
1486 1486 if message:
1487 1487 raise util.Abort(_('option "-e" incompatible with "-m" or "-l"'))
1488 1488
1489 1489 parent = q.lookup('qtip')
1490 1490 patches = []
1491 1491 messages = []
1492 1492 for f in files:
1493 1493 p = q.lookup(f)
1494 1494 if p in patches or p == parent:
1495 1495 ui.warn(_('Skipping already folded patch %s') % p)
1496 1496 if q.isapplied(p):
1497 1497 raise util.Abort(_('qfold cannot fold already applied patch %s') % p)
1498 1498 patches.append(p)
1499 1499
1500 1500 for p in patches:
1501 1501 if not message:
1502 1502 messages.append(q.readheaders(p)[0])
1503 1503 pf = q.join(p)
1504 1504 (patchsuccess, files, fuzz) = q.patch(repo, pf)
1505 1505 if not patchsuccess:
1506 1506 raise util.Abort(_('Error folding patch %s') % p)
1507 1507 patch.updatedir(ui, repo, files)
1508 1508
1509 1509 if not message:
1510 1510 message, comments, user = q.readheaders(parent)[0:3]
1511 1511 for msg in messages:
1512 1512 message.append('* * *')
1513 1513 message.extend(msg)
1514 1514 message = '\n'.join(message)
1515 1515
1516 1516 if opts['edit']:
1517 1517 message = ui.edit(message, user or ui.username())
1518 1518
1519 1519 q.refresh(repo, msg=message)
1520 1520 q.delete(repo, patches, keep=opts['keep'])
1521 1521 q.save_dirty()
1522 1522
1523 1523 def guard(ui, repo, *args, **opts):
1524 1524 '''set or print guards for a patch
1525 1525
1526 1526 Guards control whether a patch can be pushed. A patch with no
1527 1527 guards is always pushed. A patch with a positive guard ("+foo") is
1528 1528 pushed only if the qselect command has activated it. A patch with
1529 1529 a negative guard ("-foo") is never pushed if the qselect command
1530 1530 has activated it.
1531 1531
1532 1532 With no arguments, print the currently active guards.
1533 1533 With arguments, set guards for the named patch.
1534 1534
1535 1535 To set a negative guard "-foo" on topmost patch ("--" is needed so
1536 1536 hg will not interpret "-foo" as an option):
1537 1537 hg qguard -- -foo
1538 1538
1539 1539 To set guards on another patch:
1540 1540 hg qguard other.patch +2.6.17 -stable
1541 1541 '''
1542 1542 def status(idx):
1543 1543 guards = q.series_guards[idx] or ['unguarded']
1544 1544 ui.write('%s: %s\n' % (q.series[idx], ' '.join(guards)))
1545 1545 q = repo.mq
1546 1546 patch = None
1547 1547 args = list(args)
1548 1548 if opts['list']:
1549 1549 if args or opts['none']:
1550 1550 raise util.Abort(_('cannot mix -l/--list with options or arguments'))
1551 1551 for i in xrange(len(q.series)):
1552 1552 status(i)
1553 1553 return
1554 1554 if not args or args[0][0:1] in '-+':
1555 1555 if not q.applied:
1556 1556 raise util.Abort(_('no patches applied'))
1557 1557 patch = q.applied[-1].name
1558 1558 if patch is None and args[0][0:1] not in '-+':
1559 1559 patch = args.pop(0)
1560 1560 if patch is None:
1561 1561 raise util.Abort(_('no patch to work with'))
1562 1562 if args or opts['none']:
1563 1563 q.set_guards(q.find_series(patch), args)
1564 1564 q.save_dirty()
1565 1565 else:
1566 1566 status(q.series.index(q.lookup(patch)))
1567 1567
1568 1568 def header(ui, repo, patch=None):
1569 1569 """Print the header of the topmost or specified patch"""
1570 1570 q = repo.mq
1571 1571
1572 1572 if patch:
1573 1573 patch = q.lookup(patch)
1574 1574 else:
1575 1575 if not q.applied:
1576 1576 ui.write('No patches applied\n')
1577 1577 return 1
1578 1578 patch = q.lookup('qtip')
1579 1579 message = repo.mq.readheaders(patch)[0]
1580 1580
1581 1581 ui.write('\n'.join(message) + '\n')
1582 1582
1583 1583 def lastsavename(path):
1584 1584 (directory, base) = os.path.split(path)
1585 1585 names = os.listdir(directory)
1586 1586 namere = re.compile("%s.([0-9]+)" % base)
1587 1587 maxindex = None
1588 1588 maxname = None
1589 1589 for f in names:
1590 1590 m = namere.match(f)
1591 1591 if m:
1592 1592 index = int(m.group(1))
1593 1593 if maxindex == None or index > maxindex:
1594 1594 maxindex = index
1595 1595 maxname = f
1596 1596 if maxname:
1597 1597 return (os.path.join(directory, maxname), maxindex)
1598 1598 return (None, None)
1599 1599
1600 1600 def savename(path):
1601 1601 (last, index) = lastsavename(path)
1602 1602 if last is None:
1603 1603 index = 0
1604 1604 newpath = path + ".%d" % (index + 1)
1605 1605 return newpath
1606 1606
1607 1607 def push(ui, repo, patch=None, **opts):
1608 1608 """push the next patch onto the stack"""
1609 1609 q = repo.mq
1610 1610 mergeq = None
1611 1611
1612 1612 if opts['all']:
1613 1613 patch = q.series[-1]
1614 1614 if opts['merge']:
1615 1615 if opts['name']:
1616 1616 newpath = opts['name']
1617 1617 else:
1618 1618 newpath, i = lastsavename(q.path)
1619 1619 if not newpath:
1620 1620 ui.warn("no saved queues found, please use -n\n")
1621 1621 return 1
1622 1622 mergeq = queue(ui, repo.join(""), newpath)
1623 1623 ui.warn("merging with queue at: %s\n" % mergeq.path)
1624 1624 ret = q.push(repo, patch, force=opts['force'], list=opts['list'],
1625 1625 mergeq=mergeq)
1626 1626 q.save_dirty()
1627 1627 return ret
1628 1628
1629 1629 def pop(ui, repo, patch=None, **opts):
1630 1630 """pop the current patch off the stack"""
1631 1631 localupdate = True
1632 1632 if opts['name']:
1633 1633 q = queue(ui, repo.join(""), repo.join(opts['name']))
1634 1634 ui.warn('using patch queue: %s\n' % q.path)
1635 1635 localupdate = False
1636 1636 else:
1637 1637 q = repo.mq
1638 1638 q.pop(repo, patch, force=opts['force'], update=localupdate, all=opts['all'])
1639 1639 q.save_dirty()
1640 1640 return 0
1641 1641
1642 1642 def rename(ui, repo, patch, name=None, **opts):
1643 1643 """rename a patch
1644 1644
1645 1645 With one argument, renames the current patch to PATCH1.
1646 1646 With two arguments, renames PATCH1 to PATCH2."""
1647 1647
1648 1648 q = repo.mq
1649 1649
1650 1650 if not name:
1651 1651 name = patch
1652 1652 patch = None
1653 1653
1654 1654 if name in q.series:
1655 1655 raise util.Abort(_('A patch named %s already exists in the series file') % name)
1656 1656
1657 1657 absdest = q.join(name)
1658 1658 if os.path.exists(absdest):
1659 1659 raise util.Abort(_('%s already exists') % absdest)
1660 1660
1661 1661 if patch:
1662 1662 patch = q.lookup(patch)
1663 1663 else:
1664 1664 if not q.applied:
1665 1665 ui.write(_('No patches applied\n'))
1666 1666 return
1667 1667 patch = q.lookup('qtip')
1668 1668
1669 1669 if ui.verbose:
1670 1670 ui.write('Renaming %s to %s\n' % (patch, name))
1671 1671 i = q.find_series(patch)
1672 1672 q.full_series[i] = name
1673 1673 q.parse_series()
1674 1674 q.series_dirty = 1
1675 1675
1676 1676 info = q.isapplied(patch)
1677 1677 if info:
1678 1678 q.applied[info[0]] = statusentry(info[1], name)
1679 1679 q.applied_dirty = 1
1680 1680
1681 1681 util.rename(q.join(patch), absdest)
1682 1682 r = q.qrepo()
1683 1683 if r:
1684 1684 wlock = r.wlock()
1685 1685 if r.dirstate.state(name) == 'r':
1686 1686 r.undelete([name], wlock)
1687 1687 r.copy(patch, name, wlock)
1688 1688 r.remove([patch], False, wlock)
1689 1689
1690 1690 q.save_dirty()
1691 1691
1692 1692 def restore(ui, repo, rev, **opts):
1693 1693 """restore the queue state saved by a rev"""
1694 1694 rev = repo.lookup(rev)
1695 1695 q = repo.mq
1696 1696 q.restore(repo, rev, delete=opts['delete'],
1697 1697 qupdate=opts['update'])
1698 1698 q.save_dirty()
1699 1699 return 0
1700 1700
1701 1701 def save(ui, repo, **opts):
1702 1702 """save current queue state"""
1703 1703 q = repo.mq
1704 1704 message = commands.logmessage(opts)
1705 1705 ret = q.save(repo, msg=message)
1706 1706 if ret:
1707 1707 return ret
1708 1708 q.save_dirty()
1709 1709 if opts['copy']:
1710 1710 path = q.path
1711 1711 if opts['name']:
1712 1712 newpath = os.path.join(q.basepath, opts['name'])
1713 1713 if os.path.exists(newpath):
1714 1714 if not os.path.isdir(newpath):
1715 1715 raise util.Abort(_('destination %s exists and is not '
1716 1716 'a directory') % newpath)
1717 1717 if not opts['force']:
1718 1718 raise util.Abort(_('destination %s exists, '
1719 1719 'use -f to force') % newpath)
1720 1720 else:
1721 1721 newpath = savename(path)
1722 1722 ui.warn("copy %s to %s\n" % (path, newpath))
1723 1723 util.copyfiles(path, newpath)
1724 1724 if opts['empty']:
1725 1725 try:
1726 1726 os.unlink(q.join(q.status_path))
1727 1727 except:
1728 1728 pass
1729 1729 return 0
1730 1730
1731 1731 def strip(ui, repo, rev, **opts):
1732 1732 """strip a revision and all later revs on the same branch"""
1733 1733 rev = repo.lookup(rev)
1734 1734 backup = 'all'
1735 1735 if opts['backup']:
1736 1736 backup = 'strip'
1737 1737 elif opts['nobackup']:
1738 1738 backup = 'none'
1739 1739 repo.mq.strip(repo, rev, backup=backup)
1740 1740 return 0
1741 1741
1742 1742 def select(ui, repo, *args, **opts):
1743 1743 '''set or print guarded patches to push
1744 1744
1745 1745 Use the qguard command to set or print guards on patch, then use
1746 1746 qselect to tell mq which guards to use. A patch will be pushed if it
1747 1747 has no guards or any positive guards match the currently selected guard,
1748 1748 but will not be pushed if any negative guards match the current guard.
1749 1749 For example:
1750 1750
1751 1751 qguard foo.patch -stable (negative guard)
1752 1752 qguard bar.patch +stable (positive guard)
1753 1753 qselect stable
1754 1754
1755 1755 This activates the "stable" guard. mq will skip foo.patch (because
1756 1756 it has a negative match) but push bar.patch (because it
1757 1757 has a positive match).
1758 1758
1759 1759 With no arguments, prints the currently active guards.
1760 1760 With one argument, sets the active guard.
1761 1761
1762 1762 Use -n/--none to deactivate guards (no other arguments needed).
1763 1763 When no guards are active, patches with positive guards are skipped
1764 1764 and patches with negative guards are pushed.
1765 1765
1766 1766 qselect can change the guards on applied patches. It does not pop
1767 1767 guarded patches by default. Use --pop to pop back to the last applied
1768 1768 patch that is not guarded. Use --reapply (which implies --pop) to push
1769 1769 back to the current patch afterwards, but skip guarded patches.
1770 1770
1771 1771 Use -s/--series to print a list of all guards in the series file (no
1772 1772 other arguments needed). Use -v for more information.'''
1773 1773
1774 1774 q = repo.mq
1775 1775 guards = q.active()
1776 1776 if args or opts['none']:
1777 1777 old_unapplied = q.unapplied(repo)
1778 1778 old_guarded = [i for i in xrange(len(q.applied)) if
1779 1779 not q.pushable(i)[0]]
1780 1780 q.set_active(args)
1781 1781 q.save_dirty()
1782 1782 if not args:
1783 1783 ui.status(_('guards deactivated\n'))
1784 1784 if not opts['pop'] and not opts['reapply']:
1785 1785 unapplied = q.unapplied(repo)
1786 1786 guarded = [i for i in xrange(len(q.applied))
1787 1787 if not q.pushable(i)[0]]
1788 1788 if len(unapplied) != len(old_unapplied):
1789 1789 ui.status(_('number of unguarded, unapplied patches has '
1790 1790 'changed from %d to %d\n') %
1791 1791 (len(old_unapplied), len(unapplied)))
1792 1792 if len(guarded) != len(old_guarded):
1793 1793 ui.status(_('number of guarded, applied patches has changed '
1794 1794 'from %d to %d\n') %
1795 1795 (len(old_guarded), len(guarded)))
1796 1796 elif opts['series']:
1797 1797 guards = {}
1798 1798 noguards = 0
1799 1799 for gs in q.series_guards:
1800 1800 if not gs:
1801 1801 noguards += 1
1802 1802 for g in gs:
1803 1803 guards.setdefault(g, 0)
1804 1804 guards[g] += 1
1805 1805 if ui.verbose:
1806 1806 guards['NONE'] = noguards
1807 1807 guards = guards.items()
1808 1808 guards.sort(lambda a, b: cmp(a[0][1:], b[0][1:]))
1809 1809 if guards:
1810 1810 ui.note(_('guards in series file:\n'))
1811 1811 for guard, count in guards:
1812 1812 ui.note('%2d ' % count)
1813 1813 ui.write(guard, '\n')
1814 1814 else:
1815 1815 ui.note(_('no guards in series file\n'))
1816 1816 else:
1817 1817 if guards:
1818 1818 ui.note(_('active guards:\n'))
1819 1819 for g in guards:
1820 1820 ui.write(g, '\n')
1821 1821 else:
1822 1822 ui.write(_('no active guards\n'))
1823 1823 reapply = opts['reapply'] and q.applied and q.appliedname(-1)
1824 1824 popped = False
1825 1825 if opts['pop'] or opts['reapply']:
1826 1826 for i in xrange(len(q.applied)):
1827 1827 pushable, reason = q.pushable(i)
1828 1828 if not pushable:
1829 1829 ui.status(_('popping guarded patches\n'))
1830 1830 popped = True
1831 1831 if i == 0:
1832 1832 q.pop(repo, all=True)
1833 1833 else:
1834 1834 q.pop(repo, i-1)
1835 1835 break
1836 1836 if popped:
1837 1837 try:
1838 1838 if reapply:
1839 1839 ui.status(_('reapplying unguarded patches\n'))
1840 1840 q.push(repo, reapply)
1841 1841 finally:
1842 1842 q.save_dirty()
1843 1843
1844 1844 def reposetup(ui, repo):
1845 1845 class mqrepo(repo.__class__):
1846 1846 def abort_if_wdir_patched(self, errmsg, force=False):
1847 1847 if self.mq.applied and not force:
1848 1848 parent = revlog.hex(self.dirstate.parents()[0])
1849 1849 if parent in [s.rev for s in self.mq.applied]:
1850 1850 raise util.Abort(errmsg)
1851 1851
1852 1852 def commit(self, *args, **opts):
1853 1853 if len(args) >= 6:
1854 1854 force = args[5]
1855 1855 else:
1856 1856 force = opts.get('force')
1857 1857 self.abort_if_wdir_patched(
1858 1858 _('cannot commit over an applied mq patch'),
1859 1859 force)
1860 1860
1861 1861 return super(mqrepo, self).commit(*args, **opts)
1862 1862
1863 1863 def push(self, remote, force=False, revs=None):
1864 1864 if self.mq.applied and not force:
1865 1865 raise util.Abort(_('source has mq patches applied'))
1866 1866 return super(mqrepo, self).push(remote, force, revs)
1867 1867
1868 1868 def tags(self):
1869 1869 if self.tagscache:
1870 1870 return self.tagscache
1871 1871
1872 1872 tagscache = super(mqrepo, self).tags()
1873 1873
1874 1874 q = self.mq
1875 1875 if not q.applied:
1876 1876 return tagscache
1877 1877
1878 1878 mqtags = [(patch.rev, patch.name) for patch in q.applied]
1879 1879 mqtags.append((mqtags[-1][0], 'qtip'))
1880 1880 mqtags.append((mqtags[0][0], 'qbase'))
1881 1881 for patch in mqtags:
1882 1882 if patch[1] in tagscache:
1883 1883 self.ui.warn('Tag %s overrides mq patch of the same name\n' % patch[1])
1884 1884 else:
1885 1885 tagscache[patch[1]] = revlog.bin(patch[0])
1886 1886
1887 1887 return tagscache
1888 1888
1889 1889 if repo.local():
1890 1890 repo.__class__ = mqrepo
1891 1891 repo.mq = queue(ui, repo.join(""))
1892 1892
1893 1893 cmdtable = {
1894 1894 "qapplied": (applied, [], 'hg qapplied [PATCH]'),
1895 1895 "qclone": (clone,
1896 1896 [('', 'pull', None, _('use pull protocol to copy metadata')),
1897 1897 ('U', 'noupdate', None, _('do not update the new working directories')),
1898 1898 ('', 'uncompressed', None,
1899 1899 _('use uncompressed transfer (fast over LAN)')),
1900 1900 ('e', 'ssh', '', _('specify ssh command to use')),
1901 1901 ('p', 'patches', '', _('location of source patch repo')),
1902 1902 ('', 'remotecmd', '',
1903 1903 _('specify hg command to run on the remote side'))],
1904 1904 'hg qclone [OPTION]... SOURCE [DEST]'),
1905 1905 "qcommit|qci":
1906 1906 (commit,
1907 1907 commands.table["^commit|ci"][1],
1908 1908 'hg qcommit [OPTION]... [FILE]...'),
1909 1909 "^qdiff": (diff,
1910 1910 [('I', 'include', [], _('include names matching the given patterns')),
1911 1911 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
1912 1912 'hg qdiff [-I] [-X] [FILE]...'),
1913 1913 "qdelete|qremove|qrm":
1914 1914 (delete,
1915 1915 [('k', 'keep', None, _('keep patch file'))],
1916 1916 'hg qdelete [-k] PATCH'),
1917 1917 'qfold':
1918 1918 (fold,
1919 1919 [('e', 'edit', None, _('edit patch header')),
1920 1920 ('k', 'keep', None, _('keep folded patch files')),
1921 1921 ('m', 'message', '', _('set patch header to <text>')),
1922 1922 ('l', 'logfile', '', _('set patch header to contents of <file>'))],
1923 1923 'hg qfold [-e] [-m <text>] [-l <file] PATCH...'),
1924 1924 'qguard': (guard, [('l', 'list', None, _('list all patches and guards')),
1925 1925 ('n', 'none', None, _('drop all guards'))],
1926 1926 'hg qguard [PATCH] [+GUARD...] [-GUARD...]'),
1927 1927 'qheader': (header, [],
1928 1928 _('hg qheader [PATCH]')),
1929 1929 "^qimport":
1930 1930 (qimport,
1931 1931 [('e', 'existing', None, 'import file in patch dir'),
1932 1932 ('n', 'name', '', 'patch file name'),
1933 1933 ('f', 'force', None, 'overwrite existing files')],
1934 1934 'hg qimport [-e] [-n NAME] [-f] FILE...'),
1935 1935 "^qinit":
1936 1936 (init,
1937 1937 [('c', 'create-repo', None, 'create queue repository')],
1938 1938 'hg qinit [-c]'),
1939 1939 "qnew":
1940 1940 (new,
1941 1941 [('e', 'edit', None, _('edit commit message')),
1942 1942 ('m', 'message', '', _('use <text> as commit message')),
1943 1943 ('l', 'logfile', '', _('read the commit message from <file>')),
1944 1944 ('f', 'force', None, _('import uncommitted changes into patch'))],
1945 1945 'hg qnew [-e] [-m TEXT] [-l FILE] [-f] PATCH'),
1946 1946 "qnext": (next, [], 'hg qnext'),
1947 1947 "qprev": (prev, [], 'hg qprev'),
1948 1948 "^qpop":
1949 1949 (pop,
1950 1950 [('a', 'all', None, 'pop all patches'),
1951 1951 ('n', 'name', '', 'queue name to pop'),
1952 1952 ('f', 'force', None, 'forget any local changes')],
1953 1953 'hg qpop [-a] [-n NAME] [-f] [PATCH | INDEX]'),
1954 1954 "^qpush":
1955 1955 (push,
1956 1956 [('f', 'force', None, 'apply if the patch has rejects'),
1957 1957 ('l', 'list', None, 'list patch name in commit text'),
1958 1958 ('a', 'all', None, 'apply all patches'),
1959 1959 ('m', 'merge', None, 'merge from another queue'),
1960 1960 ('n', 'name', '', 'merge queue name')],
1961 1961 'hg qpush [-f] [-l] [-a] [-m] [-n NAME] [PATCH | INDEX]'),
1962 1962 "^qrefresh":
1963 1963 (refresh,
1964 1964 [('e', 'edit', None, _('edit commit message')),
1965 1965 ('m', 'message', '', _('change commit message with <text>')),
1966 1966 ('l', 'logfile', '', _('change commit message with <file> content')),
1967 1967 ('s', 'short', None, 'short refresh'),
1968 1968 ('I', 'include', [], _('include names matching the given patterns')),
1969 1969 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
1970 1970 'hg qrefresh [-I] [-X] [-e] [-m TEXT] [-l FILE] [-s] FILES...'),
1971 1971 'qrename|qmv':
1972 1972 (rename, [], 'hg qrename PATCH1 [PATCH2]'),
1973 1973 "qrestore":
1974 1974 (restore,
1975 1975 [('d', 'delete', None, 'delete save entry'),
1976 1976 ('u', 'update', None, 'update queue working dir')],
1977 1977 'hg qrestore [-d] [-u] REV'),
1978 1978 "qsave":
1979 1979 (save,
1980 1980 [('m', 'message', '', _('use <text> as commit message')),
1981 1981 ('l', 'logfile', '', _('read the commit message from <file>')),
1982 1982 ('c', 'copy', None, 'copy patch directory'),
1983 1983 ('n', 'name', '', 'copy directory name'),
1984 1984 ('e', 'empty', None, 'clear queue status file'),
1985 1985 ('f', 'force', None, 'force copy')],
1986 1986 'hg qsave [-m TEXT] [-l FILE] [-c] [-n NAME] [-e] [-f]'),
1987 1987 "qselect": (select,
1988 1988 [('n', 'none', None, _('disable all guards')),
1989 1989 ('s', 'series', None, _('list all guards in series file')),
1990 1990 ('', 'pop', None,
1991 1991 _('pop to before first guarded applied patch')),
1992 1992 ('', 'reapply', None, _('pop, then reapply patches'))],
1993 1993 'hg qselect [OPTION...] [GUARD...]'),
1994 1994 "qseries":
1995 1995 (series,
1996 1996 [('m', 'missing', None, 'print patches not in series'),
1997 1997 ('s', 'summary', None, _('print first line of patch header'))],
1998 1998 'hg qseries [-m]'),
1999 1999 "^strip":
2000 2000 (strip,
2001 2001 [('f', 'force', None, 'force multi-head removal'),
2002 2002 ('b', 'backup', None, 'bundle unrelated changesets'),
2003 2003 ('n', 'nobackup', None, 'no backups')],
2004 2004 'hg strip [-f] [-b] [-n] REV'),
2005 2005 "qtop": (top, [], 'hg qtop'),
2006 2006 "qunapplied": (unapplied, [], 'hg qunapplied [PATCH]'),
2007 2007 }
@@ -1,145 +1,145 b''
1 1 # cmdutil.py - help for command processing in mercurial
2 2 #
3 3 # Copyright 2005, 2006 Matt Mackall <mpm@selenic.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 from demandload import demandload
9 9 from node import *
10 10 from i18n import gettext as _
11 11 demandload(globals(), 'mdiff util')
12 12 demandload(globals(), 'os sys')
13 13
14 14 def make_filename(repo, pat, node,
15 15 total=None, seqno=None, revwidth=None, pathname=None):
16 16 node_expander = {
17 17 'H': lambda: hex(node),
18 18 'R': lambda: str(repo.changelog.rev(node)),
19 19 'h': lambda: short(node),
20 20 }
21 21 expander = {
22 22 '%': lambda: '%',
23 23 'b': lambda: os.path.basename(repo.root),
24 24 }
25 25
26 26 try:
27 27 if node:
28 28 expander.update(node_expander)
29 29 if node and revwidth is not None:
30 30 expander['r'] = (lambda:
31 31 str(repo.changelog.rev(node)).zfill(revwidth))
32 32 if total is not None:
33 33 expander['N'] = lambda: str(total)
34 34 if seqno is not None:
35 35 expander['n'] = lambda: str(seqno)
36 36 if total is not None and seqno is not None:
37 37 expander['n'] = lambda:str(seqno).zfill(len(str(total)))
38 38 if pathname is not None:
39 39 expander['s'] = lambda: os.path.basename(pathname)
40 40 expander['d'] = lambda: os.path.dirname(pathname) or '.'
41 41 expander['p'] = lambda: pathname
42 42
43 43 newname = []
44 44 patlen = len(pat)
45 45 i = 0
46 46 while i < patlen:
47 47 c = pat[i]
48 48 if c == '%':
49 49 i += 1
50 50 c = pat[i]
51 51 c = expander[c]()
52 52 newname.append(c)
53 53 i += 1
54 54 return ''.join(newname)
55 55 except KeyError, inst:
56 raise util.Abort(_("invalid format spec '%%%s' in output file name"),
57 inst.args[0])
56 raise util.Abort(_("invalid format spec '%%%s' in output file name") %
57 inst.args[0])
58 58
59 59 def make_file(repo, pat, node=None,
60 60 total=None, seqno=None, revwidth=None, mode='wb', pathname=None):
61 61 if not pat or pat == '-':
62 62 return 'w' in mode and sys.stdout or sys.stdin
63 63 if hasattr(pat, 'write') and 'w' in mode:
64 64 return pat
65 65 if hasattr(pat, 'read') and 'r' in mode:
66 66 return pat
67 67 return open(make_filename(repo, pat, node, total, seqno, revwidth,
68 68 pathname),
69 69 mode)
70 70
71 71 def matchpats(repo, pats=[], opts={}, head=''):
72 72 cwd = repo.getcwd()
73 73 if not pats and cwd:
74 74 opts['include'] = [os.path.join(cwd, i)
75 75 for i in opts.get('include', [])]
76 76 opts['exclude'] = [os.path.join(cwd, x)
77 77 for x in opts.get('exclude', [])]
78 78 cwd = ''
79 79 return util.cmdmatcher(repo.root, cwd, pats or ['.'], opts.get('include'),
80 80 opts.get('exclude'), head)
81 81
82 82 def makewalk(repo, pats=[], opts={}, node=None, head='', badmatch=None):
83 83 files, matchfn, anypats = matchpats(repo, pats, opts, head)
84 84 exact = dict(zip(files, files))
85 85 def walk():
86 86 for src, fn in repo.walk(node=node, files=files, match=matchfn,
87 87 badmatch=badmatch):
88 88 yield src, fn, util.pathto(repo.getcwd(), fn), fn in exact
89 89 return files, matchfn, walk()
90 90
91 91 def walk(repo, pats=[], opts={}, node=None, head='', badmatch=None):
92 92 files, matchfn, results = makewalk(repo, pats, opts, node, head, badmatch)
93 93 for r in results:
94 94 yield r
95 95
96 96 def findrenames(repo, added=None, removed=None, threshold=0.5):
97 97 if added is None or removed is None:
98 98 added, removed = repo.status()[1:3]
99 99 changes = repo.changelog.read(repo.dirstate.parents()[0])
100 100 mf = repo.manifest.read(changes[0])
101 101 for a in added:
102 102 aa = repo.wread(a)
103 103 bestscore, bestname = None, None
104 104 for r in removed:
105 105 rr = repo.file(r).read(mf[r])
106 106 delta = mdiff.textdiff(aa, rr)
107 107 if len(delta) < len(aa):
108 108 myscore = 1.0 - (float(len(delta)) / len(aa))
109 109 if bestscore is None or myscore > bestscore:
110 110 bestscore, bestname = myscore, r
111 111 if bestname and bestscore >= threshold:
112 112 yield bestname, a, bestscore
113 113
114 114 def addremove(repo, pats=[], opts={}, wlock=None, dry_run=None,
115 115 similarity=None):
116 116 if dry_run is None:
117 117 dry_run = opts.get('dry_run')
118 118 if similarity is None:
119 119 similarity = float(opts.get('similarity') or 0)
120 120 add, remove = [], []
121 121 mapping = {}
122 122 for src, abs, rel, exact in walk(repo, pats, opts):
123 123 if src == 'f' and repo.dirstate.state(abs) == '?':
124 124 add.append(abs)
125 125 mapping[abs] = rel, exact
126 126 if repo.ui.verbose or not exact:
127 127 repo.ui.status(_('adding %s\n') % ((pats and rel) or abs))
128 128 if repo.dirstate.state(abs) != 'r' and not os.path.exists(rel):
129 129 remove.append(abs)
130 130 mapping[abs] = rel, exact
131 131 if repo.ui.verbose or not exact:
132 132 repo.ui.status(_('removing %s\n') % ((pats and rel) or abs))
133 133 if not dry_run:
134 134 repo.add(add, wlock=wlock)
135 135 repo.remove(remove, wlock=wlock)
136 136 if similarity > 0:
137 137 for old, new, score in findrenames(repo, add, remove, similarity):
138 138 oldrel, oldexact = mapping[old]
139 139 newrel, newexact = mapping[new]
140 140 if repo.ui.verbose or not oldexact or not newexact:
141 141 repo.ui.status(_('recording removal of %s as rename to %s '
142 142 '(%d%% similar)\n') %
143 143 (oldrel, newrel, score * 100))
144 144 if not dry_run:
145 145 repo.copy(old, new, wlock=wlock)
@@ -1,3507 +1,3507 b''
1 1 # commands.py - command processing for mercurial
2 2 #
3 3 # Copyright 2005, 2006 Matt Mackall <mpm@selenic.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 from demandload import demandload
9 9 from node import *
10 10 from i18n import gettext as _
11 11 demandload(globals(), "os re sys signal shutil imp urllib pdb shlex")
12 12 demandload(globals(), "fancyopts ui hg util lock revlog templater bundlerepo")
13 13 demandload(globals(), "fnmatch difflib patch random signal tempfile time")
14 14 demandload(globals(), "traceback errno socket version struct atexit sets bz2")
15 15 demandload(globals(), "archival cStringIO changegroup")
16 16 demandload(globals(), "cmdutil hgweb.server sshserver")
17 17
18 18 class UnknownCommand(Exception):
19 19 """Exception raised if command is not in the command table."""
20 20 class AmbiguousCommand(Exception):
21 21 """Exception raised if command shortcut matches more than one command."""
22 22
23 23 def bail_if_changed(repo):
24 24 modified, added, removed, deleted = repo.status()[:4]
25 25 if modified or added or removed or deleted:
26 26 raise util.Abort(_("outstanding uncommitted changes"))
27 27
28 28 def relpath(repo, args):
29 29 cwd = repo.getcwd()
30 30 if cwd:
31 31 return [util.normpath(os.path.join(cwd, x)) for x in args]
32 32 return args
33 33
34 34 def logmessage(opts):
35 35 """ get the log message according to -m and -l option """
36 36 message = opts['message']
37 37 logfile = opts['logfile']
38 38
39 39 if message and logfile:
40 40 raise util.Abort(_('options --message and --logfile are mutually '
41 41 'exclusive'))
42 42 if not message and logfile:
43 43 try:
44 44 if logfile == '-':
45 45 message = sys.stdin.read()
46 46 else:
47 47 message = open(logfile).read()
48 48 except IOError, inst:
49 49 raise util.Abort(_("can't read commit message '%s': %s") %
50 50 (logfile, inst.strerror))
51 51 return message
52 52
53 53 def walkchangerevs(ui, repo, pats, opts):
54 54 '''Iterate over files and the revs they changed in.
55 55
56 56 Callers most commonly need to iterate backwards over the history
57 57 it is interested in. Doing so has awful (quadratic-looking)
58 58 performance, so we use iterators in a "windowed" way.
59 59
60 60 We walk a window of revisions in the desired order. Within the
61 61 window, we first walk forwards to gather data, then in the desired
62 62 order (usually backwards) to display it.
63 63
64 64 This function returns an (iterator, getchange, matchfn) tuple. The
65 65 getchange function returns the changelog entry for a numeric
66 66 revision. The iterator yields 3-tuples. They will be of one of
67 67 the following forms:
68 68
69 69 "window", incrementing, lastrev: stepping through a window,
70 70 positive if walking forwards through revs, last rev in the
71 71 sequence iterated over - use to reset state for the current window
72 72
73 73 "add", rev, fns: out-of-order traversal of the given file names
74 74 fns, which changed during revision rev - use to gather data for
75 75 possible display
76 76
77 77 "iter", rev, None: in-order traversal of the revs earlier iterated
78 78 over with "add" - use to display data'''
79 79
80 80 def increasing_windows(start, end, windowsize=8, sizelimit=512):
81 81 if start < end:
82 82 while start < end:
83 83 yield start, min(windowsize, end-start)
84 84 start += windowsize
85 85 if windowsize < sizelimit:
86 86 windowsize *= 2
87 87 else:
88 88 while start > end:
89 89 yield start, min(windowsize, start-end-1)
90 90 start -= windowsize
91 91 if windowsize < sizelimit:
92 92 windowsize *= 2
93 93
94 94
95 95 files, matchfn, anypats = cmdutil.matchpats(repo, pats, opts)
96 96 follow = opts.get('follow') or opts.get('follow_first')
97 97
98 98 if repo.changelog.count() == 0:
99 99 return [], False, matchfn
100 100
101 101 if follow:
102 102 p = repo.dirstate.parents()[0]
103 103 if p == nullid:
104 104 ui.warn(_('No working directory revision; defaulting to tip\n'))
105 105 start = 'tip'
106 106 else:
107 107 start = repo.changelog.rev(p)
108 108 defrange = '%s:0' % start
109 109 else:
110 110 defrange = 'tip:0'
111 111 revs = map(int, revrange(ui, repo, opts['rev'] or [defrange]))
112 112 wanted = {}
113 113 slowpath = anypats
114 114 fncache = {}
115 115
116 116 chcache = {}
117 117 def getchange(rev):
118 118 ch = chcache.get(rev)
119 119 if ch is None:
120 120 chcache[rev] = ch = repo.changelog.read(repo.lookup(str(rev)))
121 121 return ch
122 122
123 123 if not slowpath and not files:
124 124 # No files, no patterns. Display all revs.
125 125 wanted = dict(zip(revs, revs))
126 126 copies = []
127 127 if not slowpath:
128 128 # Only files, no patterns. Check the history of each file.
129 129 def filerevgen(filelog, node):
130 130 cl_count = repo.changelog.count()
131 131 if node is None:
132 132 last = filelog.count() - 1
133 133 else:
134 134 last = filelog.rev(node)
135 135 for i, window in increasing_windows(last, -1):
136 136 revs = []
137 137 for j in xrange(i - window, i + 1):
138 138 n = filelog.node(j)
139 139 revs.append((filelog.linkrev(n),
140 140 follow and filelog.renamed(n)))
141 141 revs.reverse()
142 142 for rev in revs:
143 143 # only yield rev for which we have the changelog, it can
144 144 # happen while doing "hg log" during a pull or commit
145 145 if rev[0] < cl_count:
146 146 yield rev
147 147 def iterfiles():
148 148 for filename in files:
149 149 yield filename, None
150 150 for filename_node in copies:
151 151 yield filename_node
152 152 minrev, maxrev = min(revs), max(revs)
153 153 for file_, node in iterfiles():
154 154 filelog = repo.file(file_)
155 155 # A zero count may be a directory or deleted file, so
156 156 # try to find matching entries on the slow path.
157 157 if filelog.count() == 0:
158 158 slowpath = True
159 159 break
160 160 for rev, copied in filerevgen(filelog, node):
161 161 if rev <= maxrev:
162 162 if rev < minrev:
163 163 break
164 164 fncache.setdefault(rev, [])
165 165 fncache[rev].append(file_)
166 166 wanted[rev] = 1
167 167 if follow and copied:
168 168 copies.append(copied)
169 169 if slowpath:
170 170 if follow:
171 171 raise util.Abort(_('can only follow copies/renames for explicit '
172 172 'file names'))
173 173
174 174 # The slow path checks files modified in every changeset.
175 175 def changerevgen():
176 176 for i, window in increasing_windows(repo.changelog.count()-1, -1):
177 177 for j in xrange(i - window, i + 1):
178 178 yield j, getchange(j)[3]
179 179
180 180 for rev, changefiles in changerevgen():
181 181 matches = filter(matchfn, changefiles)
182 182 if matches:
183 183 fncache[rev] = matches
184 184 wanted[rev] = 1
185 185
186 186 class followfilter:
187 187 def __init__(self, onlyfirst=False):
188 188 self.startrev = -1
189 189 self.roots = []
190 190 self.onlyfirst = onlyfirst
191 191
192 192 def match(self, rev):
193 193 def realparents(rev):
194 194 if self.onlyfirst:
195 195 return repo.changelog.parentrevs(rev)[0:1]
196 196 else:
197 197 return filter(lambda x: x != -1, repo.changelog.parentrevs(rev))
198 198
199 199 if self.startrev == -1:
200 200 self.startrev = rev
201 201 return True
202 202
203 203 if rev > self.startrev:
204 204 # forward: all descendants
205 205 if not self.roots:
206 206 self.roots.append(self.startrev)
207 207 for parent in realparents(rev):
208 208 if parent in self.roots:
209 209 self.roots.append(rev)
210 210 return True
211 211 else:
212 212 # backwards: all parents
213 213 if not self.roots:
214 214 self.roots.extend(realparents(self.startrev))
215 215 if rev in self.roots:
216 216 self.roots.remove(rev)
217 217 self.roots.extend(realparents(rev))
218 218 return True
219 219
220 220 return False
221 221
222 222 # it might be worthwhile to do this in the iterator if the rev range
223 223 # is descending and the prune args are all within that range
224 224 for rev in opts.get('prune', ()):
225 225 rev = repo.changelog.rev(repo.lookup(rev))
226 226 ff = followfilter()
227 227 stop = min(revs[0], revs[-1])
228 228 for x in range(rev, stop-1, -1):
229 229 if ff.match(x) and wanted.has_key(x):
230 230 del wanted[x]
231 231
232 232 def iterate():
233 233 if follow and not files:
234 234 ff = followfilter(onlyfirst=opts.get('follow_first'))
235 235 def want(rev):
236 236 if ff.match(rev) and rev in wanted:
237 237 return True
238 238 return False
239 239 else:
240 240 def want(rev):
241 241 return rev in wanted
242 242
243 243 for i, window in increasing_windows(0, len(revs)):
244 244 yield 'window', revs[0] < revs[-1], revs[-1]
245 245 nrevs = [rev for rev in revs[i:i+window] if want(rev)]
246 246 srevs = list(nrevs)
247 247 srevs.sort()
248 248 for rev in srevs:
249 249 fns = fncache.get(rev) or filter(matchfn, getchange(rev)[3])
250 250 yield 'add', rev, fns
251 251 for rev in nrevs:
252 252 yield 'iter', rev, None
253 253 return iterate(), getchange, matchfn
254 254
255 255 revrangesep = ':'
256 256
257 257 def revfix(repo, val, defval):
258 258 '''turn user-level id of changeset into rev number.
259 259 user-level id can be tag, changeset, rev number, or negative rev
260 260 number relative to number of revs (-1 is tip, etc).'''
261 261 if not val:
262 262 return defval
263 263 try:
264 264 num = int(val)
265 265 if str(num) != val:
266 266 raise ValueError
267 267 if num < 0:
268 268 num += repo.changelog.count()
269 269 if num < 0:
270 270 num = 0
271 271 elif num >= repo.changelog.count():
272 272 raise ValueError
273 273 except ValueError:
274 274 try:
275 275 num = repo.changelog.rev(repo.lookup(val))
276 276 except KeyError:
277 raise util.Abort(_('invalid revision identifier %s'), val)
277 raise util.Abort(_('invalid revision identifier %s') % val)
278 278 return num
279 279
280 280 def revpair(ui, repo, revs):
281 281 '''return pair of nodes, given list of revisions. second item can
282 282 be None, meaning use working dir.'''
283 283 if not revs:
284 284 return repo.dirstate.parents()[0], None
285 285 end = None
286 286 if len(revs) == 1:
287 287 start = revs[0]
288 288 if revrangesep in start:
289 289 start, end = start.split(revrangesep, 1)
290 290 start = revfix(repo, start, 0)
291 291 end = revfix(repo, end, repo.changelog.count() - 1)
292 292 else:
293 293 start = revfix(repo, start, None)
294 294 elif len(revs) == 2:
295 295 if revrangesep in revs[0] or revrangesep in revs[1]:
296 296 raise util.Abort(_('too many revisions specified'))
297 297 start = revfix(repo, revs[0], None)
298 298 end = revfix(repo, revs[1], None)
299 299 else:
300 300 raise util.Abort(_('too many revisions specified'))
301 301 if end is not None: end = repo.lookup(str(end))
302 302 return repo.lookup(str(start)), end
303 303
304 304 def revrange(ui, repo, revs):
305 305 """Yield revision as strings from a list of revision specifications."""
306 306 seen = {}
307 307 for spec in revs:
308 308 if revrangesep in spec:
309 309 start, end = spec.split(revrangesep, 1)
310 310 start = revfix(repo, start, 0)
311 311 end = revfix(repo, end, repo.changelog.count() - 1)
312 312 step = start > end and -1 or 1
313 313 for rev in xrange(start, end+step, step):
314 314 if rev in seen:
315 315 continue
316 316 seen[rev] = 1
317 317 yield str(rev)
318 318 else:
319 319 rev = revfix(repo, spec, None)
320 320 if rev in seen:
321 321 continue
322 322 seen[rev] = 1
323 323 yield str(rev)
324 324
325 325 def write_bundle(cg, filename=None, compress=True):
326 326 """Write a bundle file and return its filename.
327 327
328 328 Existing files will not be overwritten.
329 329 If no filename is specified, a temporary file is created.
330 330 bz2 compression can be turned off.
331 331 The bundle file will be deleted in case of errors.
332 332 """
333 333 class nocompress(object):
334 334 def compress(self, x):
335 335 return x
336 336 def flush(self):
337 337 return ""
338 338
339 339 fh = None
340 340 cleanup = None
341 341 try:
342 342 if filename:
343 343 if os.path.exists(filename):
344 raise util.Abort(_("file '%s' already exists"), filename)
344 raise util.Abort(_("file '%s' already exists") % filename)
345 345 fh = open(filename, "wb")
346 346 else:
347 347 fd, filename = tempfile.mkstemp(prefix="hg-bundle-", suffix=".hg")
348 348 fh = os.fdopen(fd, "wb")
349 349 cleanup = filename
350 350
351 351 if compress:
352 352 fh.write("HG10")
353 353 z = bz2.BZ2Compressor(9)
354 354 else:
355 355 fh.write("HG10UN")
356 356 z = nocompress()
357 357 # parse the changegroup data, otherwise we will block
358 358 # in case of sshrepo because we don't know the end of the stream
359 359
360 360 # an empty chunkiter is the end of the changegroup
361 361 empty = False
362 362 while not empty:
363 363 empty = True
364 364 for chunk in changegroup.chunkiter(cg):
365 365 empty = False
366 366 fh.write(z.compress(changegroup.genchunk(chunk)))
367 367 fh.write(z.compress(changegroup.closechunk()))
368 368 fh.write(z.flush())
369 369 cleanup = None
370 370 return filename
371 371 finally:
372 372 if fh is not None:
373 373 fh.close()
374 374 if cleanup is not None:
375 375 os.unlink(cleanup)
376 376
377 377 def trimuser(ui, name, rev, revcache):
378 378 """trim the name of the user who committed a change"""
379 379 user = revcache.get(rev)
380 380 if user is None:
381 381 user = revcache[rev] = ui.shortuser(name)
382 382 return user
383 383
384 384 class changeset_printer(object):
385 385 '''show changeset information when templating not requested.'''
386 386
387 387 def __init__(self, ui, repo):
388 388 self.ui = ui
389 389 self.repo = repo
390 390
391 391 def show(self, rev=0, changenode=None, brinfo=None):
392 392 '''show a single changeset or file revision'''
393 393 log = self.repo.changelog
394 394 if changenode is None:
395 395 changenode = log.node(rev)
396 396 elif not rev:
397 397 rev = log.rev(changenode)
398 398
399 399 if self.ui.quiet:
400 400 self.ui.write("%d:%s\n" % (rev, short(changenode)))
401 401 return
402 402
403 403 changes = log.read(changenode)
404 404 date = util.datestr(changes[2])
405 405
406 406 hexfunc = self.ui.debugflag and hex or short
407 407
408 408 parents = [(log.rev(p), hexfunc(p)) for p in log.parents(changenode)
409 409 if self.ui.debugflag or p != nullid]
410 410 if (not self.ui.debugflag and len(parents) == 1 and
411 411 parents[0][0] == rev-1):
412 412 parents = []
413 413
414 414 self.ui.write(_("changeset: %d:%s\n") % (rev, hexfunc(changenode)))
415 415
416 416 for tag in self.repo.nodetags(changenode):
417 417 self.ui.status(_("tag: %s\n") % tag)
418 418 for parent in parents:
419 419 self.ui.write(_("parent: %d:%s\n") % parent)
420 420
421 421 if brinfo and changenode in brinfo:
422 422 br = brinfo[changenode]
423 423 self.ui.write(_("branch: %s\n") % " ".join(br))
424 424
425 425 self.ui.debug(_("manifest: %d:%s\n") %
426 426 (self.repo.manifest.rev(changes[0]), hex(changes[0])))
427 427 self.ui.status(_("user: %s\n") % changes[1])
428 428 self.ui.status(_("date: %s\n") % date)
429 429
430 430 if self.ui.debugflag:
431 431 files = self.repo.status(log.parents(changenode)[0], changenode)[:3]
432 432 for key, value in zip([_("files:"), _("files+:"), _("files-:")],
433 433 files):
434 434 if value:
435 435 self.ui.note("%-12s %s\n" % (key, " ".join(value)))
436 436 else:
437 437 self.ui.note(_("files: %s\n") % " ".join(changes[3]))
438 438
439 439 description = changes[4].strip()
440 440 if description:
441 441 if self.ui.verbose:
442 442 self.ui.status(_("description:\n"))
443 443 self.ui.status(description)
444 444 self.ui.status("\n\n")
445 445 else:
446 446 self.ui.status(_("summary: %s\n") %
447 447 description.splitlines()[0])
448 448 self.ui.status("\n")
449 449
450 450 def show_changeset(ui, repo, opts):
451 451 '''show one changeset. uses template or regular display. caller
452 452 can pass in 'style' and 'template' options in opts.'''
453 453
454 454 tmpl = opts.get('template')
455 455 if tmpl:
456 456 tmpl = templater.parsestring(tmpl, quoted=False)
457 457 else:
458 458 tmpl = ui.config('ui', 'logtemplate')
459 459 if tmpl: tmpl = templater.parsestring(tmpl)
460 460 mapfile = opts.get('style') or ui.config('ui', 'style')
461 461 if tmpl or mapfile:
462 462 if mapfile:
463 463 if not os.path.isfile(mapfile):
464 464 mapname = templater.templatepath('map-cmdline.' + mapfile)
465 465 if not mapname: mapname = templater.templatepath(mapfile)
466 466 if mapname: mapfile = mapname
467 467 try:
468 468 t = templater.changeset_templater(ui, repo, mapfile)
469 469 except SyntaxError, inst:
470 470 raise util.Abort(inst.args[0])
471 471 if tmpl: t.use_template(tmpl)
472 472 return t
473 473 return changeset_printer(ui, repo)
474 474
475 475 def setremoteconfig(ui, opts):
476 476 "copy remote options to ui tree"
477 477 if opts.get('ssh'):
478 478 ui.setconfig("ui", "ssh", opts['ssh'])
479 479 if opts.get('remotecmd'):
480 480 ui.setconfig("ui", "remotecmd", opts['remotecmd'])
481 481
482 482 def show_version(ui):
483 483 """output version and copyright information"""
484 484 ui.write(_("Mercurial Distributed SCM (version %s)\n")
485 485 % version.get_version())
486 486 ui.status(_(
487 487 "\nCopyright (C) 2005, 2006 Matt Mackall <mpm@selenic.com>\n"
488 488 "This is free software; see the source for copying conditions. "
489 489 "There is NO\nwarranty; "
490 490 "not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n"
491 491 ))
492 492
493 493 def help_(ui, name=None, with_version=False):
494 494 """show help for a command, extension, or list of commands
495 495
496 496 With no arguments, print a list of commands and short help.
497 497
498 498 Given a command name, print help for that command.
499 499
500 500 Given an extension name, print help for that extension, and the
501 501 commands it provides."""
502 502 option_lists = []
503 503
504 504 def helpcmd(name):
505 505 if with_version:
506 506 show_version(ui)
507 507 ui.write('\n')
508 508 aliases, i = findcmd(ui, name)
509 509 # synopsis
510 510 ui.write("%s\n\n" % i[2])
511 511
512 512 # description
513 513 doc = i[0].__doc__
514 514 if not doc:
515 515 doc = _("(No help text available)")
516 516 if ui.quiet:
517 517 doc = doc.splitlines(0)[0]
518 518 ui.write("%s\n" % doc.rstrip())
519 519
520 520 if not ui.quiet:
521 521 # aliases
522 522 if len(aliases) > 1:
523 523 ui.write(_("\naliases: %s\n") % ', '.join(aliases[1:]))
524 524
525 525 # options
526 526 if i[1]:
527 527 option_lists.append(("options", i[1]))
528 528
529 529 def helplist(select=None):
530 530 h = {}
531 531 cmds = {}
532 532 for c, e in table.items():
533 533 f = c.split("|", 1)[0]
534 534 if select and not select(f):
535 535 continue
536 536 if name == "shortlist" and not f.startswith("^"):
537 537 continue
538 538 f = f.lstrip("^")
539 539 if not ui.debugflag and f.startswith("debug"):
540 540 continue
541 541 doc = e[0].__doc__
542 542 if not doc:
543 543 doc = _("(No help text available)")
544 544 h[f] = doc.splitlines(0)[0].rstrip()
545 545 cmds[f] = c.lstrip("^")
546 546
547 547 fns = h.keys()
548 548 fns.sort()
549 549 m = max(map(len, fns))
550 550 for f in fns:
551 551 if ui.verbose:
552 552 commands = cmds[f].replace("|",", ")
553 553 ui.write(" %s:\n %s\n"%(commands, h[f]))
554 554 else:
555 555 ui.write(' %-*s %s\n' % (m, f, h[f]))
556 556
557 557 def helpext(name):
558 558 try:
559 559 mod = findext(name)
560 560 except KeyError:
561 561 raise UnknownCommand(name)
562 562
563 563 doc = (mod.__doc__ or _('No help text available')).splitlines(0)
564 564 ui.write(_('%s extension - %s\n') % (name.split('.')[-1], doc[0]))
565 565 for d in doc[1:]:
566 566 ui.write(d, '\n')
567 567
568 568 ui.status('\n')
569 569 if ui.verbose:
570 570 ui.status(_('list of commands:\n\n'))
571 571 else:
572 572 ui.status(_('list of commands (use "hg help -v %s" '
573 573 'to show aliases and global options):\n\n') % name)
574 574
575 575 modcmds = dict.fromkeys([c.split('|', 1)[0] for c in mod.cmdtable])
576 576 helplist(modcmds.has_key)
577 577
578 578 if name and name != 'shortlist':
579 579 try:
580 580 helpcmd(name)
581 581 except UnknownCommand:
582 582 helpext(name)
583 583
584 584 else:
585 585 # program name
586 586 if ui.verbose or with_version:
587 587 show_version(ui)
588 588 else:
589 589 ui.status(_("Mercurial Distributed SCM\n"))
590 590 ui.status('\n')
591 591
592 592 # list of commands
593 593 if name == "shortlist":
594 594 ui.status(_('basic commands (use "hg help" '
595 595 'for the full list or option "-v" for details):\n\n'))
596 596 elif ui.verbose:
597 597 ui.status(_('list of commands:\n\n'))
598 598 else:
599 599 ui.status(_('list of commands (use "hg help -v" '
600 600 'to show aliases and global options):\n\n'))
601 601
602 602 helplist()
603 603
604 604 # global options
605 605 if ui.verbose:
606 606 option_lists.append(("global options", globalopts))
607 607
608 608 # list all option lists
609 609 opt_output = []
610 610 for title, options in option_lists:
611 611 opt_output.append(("\n%s:\n" % title, None))
612 612 for shortopt, longopt, default, desc in options:
613 613 opt_output.append(("%2s%s" % (shortopt and "-%s" % shortopt,
614 614 longopt and " --%s" % longopt),
615 615 "%s%s" % (desc,
616 616 default
617 617 and _(" (default: %s)") % default
618 618 or "")))
619 619
620 620 if opt_output:
621 621 opts_len = max([len(line[0]) for line in opt_output if line[1]])
622 622 for first, second in opt_output:
623 623 if second:
624 624 ui.write(" %-*s %s\n" % (opts_len, first, second))
625 625 else:
626 626 ui.write("%s\n" % first)
627 627
628 628 # Commands start here, listed alphabetically
629 629
630 630 def add(ui, repo, *pats, **opts):
631 631 """add the specified files on the next commit
632 632
633 633 Schedule files to be version controlled and added to the repository.
634 634
635 635 The files will be added to the repository at the next commit.
636 636
637 637 If no names are given, add all files in the repository.
638 638 """
639 639
640 640 names = []
641 641 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts):
642 642 if exact:
643 643 if ui.verbose:
644 644 ui.status(_('adding %s\n') % rel)
645 645 names.append(abs)
646 646 elif repo.dirstate.state(abs) == '?':
647 647 ui.status(_('adding %s\n') % rel)
648 648 names.append(abs)
649 649 if not opts.get('dry_run'):
650 650 repo.add(names)
651 651
652 652 def addremove(ui, repo, *pats, **opts):
653 653 """add all new files, delete all missing files (DEPRECATED)
654 654
655 655 Add all new files and remove all missing files from the repository.
656 656
657 657 New files are ignored if they match any of the patterns in .hgignore. As
658 658 with add, these changes take effect at the next commit.
659 659
660 660 Use the -s option to detect renamed files. With a parameter > 0,
661 661 this compares every removed file with every added file and records
662 662 those similar enough as renames. This option takes a percentage
663 663 between 0 (disabled) and 100 (files must be identical) as its
664 664 parameter. Detecting renamed files this way can be expensive.
665 665 """
666 666 sim = float(opts.get('similarity') or 0)
667 667 if sim < 0 or sim > 100:
668 668 raise util.Abort(_('similarity must be between 0 and 100'))
669 669 return cmdutil.addremove(repo, pats, opts, similarity=sim/100.)
670 670
671 671 def annotate(ui, repo, *pats, **opts):
672 672 """show changeset information per file line
673 673
674 674 List changes in files, showing the revision id responsible for each line
675 675
676 676 This command is useful to discover who did a change or when a change took
677 677 place.
678 678
679 679 Without the -a option, annotate will avoid processing files it
680 680 detects as binary. With -a, annotate will generate an annotation
681 681 anyway, probably with undesirable results.
682 682 """
683 683 def getnode(rev):
684 684 return short(repo.changelog.node(rev))
685 685
686 686 ucache = {}
687 687 def getname(rev):
688 688 try:
689 689 return ucache[rev]
690 690 except:
691 691 u = trimuser(ui, repo.changectx(rev).user(), rev, ucache)
692 692 ucache[rev] = u
693 693 return u
694 694
695 695 dcache = {}
696 696 def getdate(rev):
697 697 datestr = dcache.get(rev)
698 698 if datestr is None:
699 699 datestr = dcache[rev] = util.datestr(repo.changectx(rev).date())
700 700 return datestr
701 701
702 702 if not pats:
703 703 raise util.Abort(_('at least one file name or pattern required'))
704 704
705 705 opmap = [['user', getname], ['number', str], ['changeset', getnode],
706 706 ['date', getdate]]
707 707 if not opts['user'] and not opts['changeset'] and not opts['date']:
708 708 opts['number'] = 1
709 709
710 710 ctx = repo.changectx(opts['rev'] or repo.dirstate.parents()[0])
711 711
712 712 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts,
713 713 node=ctx.node()):
714 714 fctx = ctx.filectx(abs)
715 715 if not opts['text'] and util.binary(fctx.data()):
716 716 ui.write(_("%s: binary file\n") % ((pats and rel) or abs))
717 717 continue
718 718
719 719 lines = fctx.annotate()
720 720 pieces = []
721 721
722 722 for o, f in opmap:
723 723 if opts[o]:
724 724 l = [f(n) for n, dummy in lines]
725 725 if l:
726 726 m = max(map(len, l))
727 727 pieces.append(["%*s" % (m, x) for x in l])
728 728
729 729 if pieces:
730 730 for p, l in zip(zip(*pieces), lines):
731 731 ui.write("%s: %s" % (" ".join(p), l[1]))
732 732
733 733 def archive(ui, repo, dest, **opts):
734 734 '''create unversioned archive of a repository revision
735 735
736 736 By default, the revision used is the parent of the working
737 737 directory; use "-r" to specify a different revision.
738 738
739 739 To specify the type of archive to create, use "-t". Valid
740 740 types are:
741 741
742 742 "files" (default): a directory full of files
743 743 "tar": tar archive, uncompressed
744 744 "tbz2": tar archive, compressed using bzip2
745 745 "tgz": tar archive, compressed using gzip
746 746 "uzip": zip archive, uncompressed
747 747 "zip": zip archive, compressed using deflate
748 748
749 749 The exact name of the destination archive or directory is given
750 750 using a format string; see "hg help export" for details.
751 751
752 752 Each member added to an archive file has a directory prefix
753 753 prepended. Use "-p" to specify a format string for the prefix.
754 754 The default is the basename of the archive, with suffixes removed.
755 755 '''
756 756
757 757 if opts['rev']:
758 758 node = repo.lookup(opts['rev'])
759 759 else:
760 760 node, p2 = repo.dirstate.parents()
761 761 if p2 != nullid:
762 762 raise util.Abort(_('uncommitted merge - please provide a '
763 763 'specific revision'))
764 764
765 765 dest = cmdutil.make_filename(repo, dest, node)
766 766 if os.path.realpath(dest) == repo.root:
767 767 raise util.Abort(_('repository root cannot be destination'))
768 768 dummy, matchfn, dummy = cmdutil.matchpats(repo, [], opts)
769 769 kind = opts.get('type') or 'files'
770 770 prefix = opts['prefix']
771 771 if dest == '-':
772 772 if kind == 'files':
773 773 raise util.Abort(_('cannot archive plain files to stdout'))
774 774 dest = sys.stdout
775 775 if not prefix: prefix = os.path.basename(repo.root) + '-%h'
776 776 prefix = cmdutil.make_filename(repo, prefix, node)
777 777 archival.archive(repo, dest, node, kind, not opts['no_decode'],
778 778 matchfn, prefix)
779 779
780 780 def backout(ui, repo, rev, **opts):
781 781 '''reverse effect of earlier changeset
782 782
783 783 Commit the backed out changes as a new changeset. The new
784 784 changeset is a child of the backed out changeset.
785 785
786 786 If you back out a changeset other than the tip, a new head is
787 787 created. This head is the parent of the working directory. If
788 788 you back out an old changeset, your working directory will appear
789 789 old after the backout. You should merge the backout changeset
790 790 with another head.
791 791
792 792 The --merge option remembers the parent of the working directory
793 793 before starting the backout, then merges the new head with that
794 794 changeset afterwards. This saves you from doing the merge by
795 795 hand. The result of this merge is not committed, as for a normal
796 796 merge.'''
797 797
798 798 bail_if_changed(repo)
799 799 op1, op2 = repo.dirstate.parents()
800 800 if op2 != nullid:
801 801 raise util.Abort(_('outstanding uncommitted merge'))
802 802 node = repo.lookup(rev)
803 803 p1, p2 = repo.changelog.parents(node)
804 804 if p1 == nullid:
805 805 raise util.Abort(_('cannot back out a change with no parents'))
806 806 if p2 != nullid:
807 807 if not opts['parent']:
808 808 raise util.Abort(_('cannot back out a merge changeset without '
809 809 '--parent'))
810 810 p = repo.lookup(opts['parent'])
811 811 if p not in (p1, p2):
812 812 raise util.Abort(_('%s is not a parent of %s' %
813 813 (short(p), short(node))))
814 814 parent = p
815 815 else:
816 816 if opts['parent']:
817 817 raise util.Abort(_('cannot use --parent on non-merge changeset'))
818 818 parent = p1
819 819 hg.clean(repo, node, show_stats=False)
820 820 revert_opts = opts.copy()
821 821 revert_opts['all'] = True
822 822 revert_opts['rev'] = hex(parent)
823 823 revert(ui, repo, **revert_opts)
824 824 commit_opts = opts.copy()
825 825 commit_opts['addremove'] = False
826 826 if not commit_opts['message'] and not commit_opts['logfile']:
827 827 commit_opts['message'] = _("Backed out changeset %s") % (hex(node))
828 828 commit_opts['force_editor'] = True
829 829 commit(ui, repo, **commit_opts)
830 830 def nice(node):
831 831 return '%d:%s' % (repo.changelog.rev(node), short(node))
832 832 ui.status(_('changeset %s backs out changeset %s\n') %
833 833 (nice(repo.changelog.tip()), nice(node)))
834 834 if op1 != node:
835 835 if opts['merge']:
836 836 ui.status(_('merging with changeset %s\n') % nice(op1))
837 837 n = _lookup(repo, hex(op1))
838 838 hg.merge(repo, n)
839 839 else:
840 840 ui.status(_('the backout changeset is a new head - '
841 841 'do not forget to merge\n'))
842 842 ui.status(_('(use "backout --merge" '
843 843 'if you want to auto-merge)\n'))
844 844
845 845 def bundle(ui, repo, fname, dest=None, **opts):
846 846 """create a changegroup file
847 847
848 848 Generate a compressed changegroup file collecting all changesets
849 849 not found in the other repository.
850 850
851 851 This file can then be transferred using conventional means and
852 852 applied to another repository with the unbundle command. This is
853 853 useful when native push and pull are not available or when
854 854 exporting an entire repository is undesirable. The standard file
855 855 extension is ".hg".
856 856
857 857 Unlike import/export, this exactly preserves all changeset
858 858 contents including permissions, rename data, and revision history.
859 859 """
860 860 dest = ui.expandpath(dest or 'default-push', dest or 'default')
861 861 other = hg.repository(ui, dest)
862 862 o = repo.findoutgoing(other, force=opts['force'])
863 863 cg = repo.changegroup(o, 'bundle')
864 864 write_bundle(cg, fname)
865 865
866 866 def cat(ui, repo, file1, *pats, **opts):
867 867 """output the latest or given revisions of files
868 868
869 869 Print the specified files as they were at the given revision.
870 870 If no revision is given then the tip is used.
871 871
872 872 Output may be to a file, in which case the name of the file is
873 873 given using a format string. The formatting rules are the same as
874 874 for the export command, with the following additions:
875 875
876 876 %s basename of file being printed
877 877 %d dirname of file being printed, or '.' if in repo root
878 878 %p root-relative path name of file being printed
879 879 """
880 880 ctx = repo.changectx(opts['rev'] or "-1")
881 881 for src, abs, rel, exact in cmdutil.walk(repo, (file1,) + pats, opts,
882 882 ctx.node()):
883 883 fp = cmdutil.make_file(repo, opts['output'], ctx.node(), pathname=abs)
884 884 fp.write(ctx.filectx(abs).data())
885 885
886 886 def clone(ui, source, dest=None, **opts):
887 887 """make a copy of an existing repository
888 888
889 889 Create a copy of an existing repository in a new directory.
890 890
891 891 If no destination directory name is specified, it defaults to the
892 892 basename of the source.
893 893
894 894 The location of the source is added to the new repository's
895 895 .hg/hgrc file, as the default to be used for future pulls.
896 896
897 897 For efficiency, hardlinks are used for cloning whenever the source
898 898 and destination are on the same filesystem (note this applies only
899 899 to the repository data, not to the checked out files). Some
900 900 filesystems, such as AFS, implement hardlinking incorrectly, but
901 901 do not report errors. In these cases, use the --pull option to
902 902 avoid hardlinking.
903 903
904 904 You can safely clone repositories and checked out files using full
905 905 hardlinks with
906 906
907 907 $ cp -al REPO REPOCLONE
908 908
909 909 which is the fastest way to clone. However, the operation is not
910 910 atomic (making sure REPO is not modified during the operation is
911 911 up to you) and you have to make sure your editor breaks hardlinks
912 912 (Emacs and most Linux Kernel tools do so).
913 913
914 914 If you use the -r option to clone up to a specific revision, no
915 915 subsequent revisions will be present in the cloned repository.
916 916 This option implies --pull, even on local repositories.
917 917
918 918 See pull for valid source format details.
919 919
920 920 It is possible to specify an ssh:// URL as the destination, but no
921 921 .hg/hgrc will be created on the remote side. Look at the help text
922 922 for the pull command for important details about ssh:// URLs.
923 923 """
924 924 setremoteconfig(ui, opts)
925 925 hg.clone(ui, ui.expandpath(source), dest,
926 926 pull=opts['pull'],
927 927 stream=opts['uncompressed'],
928 928 rev=opts['rev'],
929 929 update=not opts['noupdate'])
930 930
931 931 def commit(ui, repo, *pats, **opts):
932 932 """commit the specified files or all outstanding changes
933 933
934 934 Commit changes to the given files into the repository.
935 935
936 936 If a list of files is omitted, all changes reported by "hg status"
937 937 will be committed.
938 938
939 939 If no commit message is specified, the editor configured in your hgrc
940 940 or in the EDITOR environment variable is started to enter a message.
941 941 """
942 942 message = logmessage(opts)
943 943
944 944 if opts['addremove']:
945 945 cmdutil.addremove(repo, pats, opts)
946 946 fns, match, anypats = cmdutil.matchpats(repo, pats, opts)
947 947 if pats:
948 948 modified, added, removed = repo.status(files=fns, match=match)[:3]
949 949 files = modified + added + removed
950 950 else:
951 951 files = []
952 952 try:
953 953 repo.commit(files, message, opts['user'], opts['date'], match,
954 954 force_editor=opts.get('force_editor'))
955 955 except ValueError, inst:
956 956 raise util.Abort(str(inst))
957 957
958 958 def docopy(ui, repo, pats, opts, wlock):
959 959 # called with the repo lock held
960 960 cwd = repo.getcwd()
961 961 errors = 0
962 962 copied = []
963 963 targets = {}
964 964
965 965 def okaytocopy(abs, rel, exact):
966 966 reasons = {'?': _('is not managed'),
967 967 'a': _('has been marked for add'),
968 968 'r': _('has been marked for remove')}
969 969 state = repo.dirstate.state(abs)
970 970 reason = reasons.get(state)
971 971 if reason:
972 972 if state == 'a':
973 973 origsrc = repo.dirstate.copied(abs)
974 974 if origsrc is not None:
975 975 return origsrc
976 976 if exact:
977 977 ui.warn(_('%s: not copying - file %s\n') % (rel, reason))
978 978 else:
979 979 return abs
980 980
981 981 def copy(origsrc, abssrc, relsrc, target, exact):
982 982 abstarget = util.canonpath(repo.root, cwd, target)
983 983 reltarget = util.pathto(cwd, abstarget)
984 984 prevsrc = targets.get(abstarget)
985 985 if prevsrc is not None:
986 986 ui.warn(_('%s: not overwriting - %s collides with %s\n') %
987 987 (reltarget, abssrc, prevsrc))
988 988 return
989 989 if (not opts['after'] and os.path.exists(reltarget) or
990 990 opts['after'] and repo.dirstate.state(abstarget) not in '?r'):
991 991 if not opts['force']:
992 992 ui.warn(_('%s: not overwriting - file exists\n') %
993 993 reltarget)
994 994 return
995 995 if not opts['after'] and not opts.get('dry_run'):
996 996 os.unlink(reltarget)
997 997 if opts['after']:
998 998 if not os.path.exists(reltarget):
999 999 return
1000 1000 else:
1001 1001 targetdir = os.path.dirname(reltarget) or '.'
1002 1002 if not os.path.isdir(targetdir) and not opts.get('dry_run'):
1003 1003 os.makedirs(targetdir)
1004 1004 try:
1005 1005 restore = repo.dirstate.state(abstarget) == 'r'
1006 1006 if restore and not opts.get('dry_run'):
1007 1007 repo.undelete([abstarget], wlock)
1008 1008 try:
1009 1009 if not opts.get('dry_run'):
1010 1010 shutil.copyfile(relsrc, reltarget)
1011 1011 shutil.copymode(relsrc, reltarget)
1012 1012 restore = False
1013 1013 finally:
1014 1014 if restore:
1015 1015 repo.remove([abstarget], wlock)
1016 1016 except shutil.Error, inst:
1017 1017 raise util.Abort(str(inst))
1018 1018 except IOError, inst:
1019 1019 if inst.errno == errno.ENOENT:
1020 1020 ui.warn(_('%s: deleted in working copy\n') % relsrc)
1021 1021 else:
1022 1022 ui.warn(_('%s: cannot copy - %s\n') %
1023 1023 (relsrc, inst.strerror))
1024 1024 errors += 1
1025 1025 return
1026 1026 if ui.verbose or not exact:
1027 1027 ui.status(_('copying %s to %s\n') % (relsrc, reltarget))
1028 1028 targets[abstarget] = abssrc
1029 1029 if abstarget != origsrc and not opts.get('dry_run'):
1030 1030 repo.copy(origsrc, abstarget, wlock)
1031 1031 copied.append((abssrc, relsrc, exact))
1032 1032
1033 1033 def targetpathfn(pat, dest, srcs):
1034 1034 if os.path.isdir(pat):
1035 1035 abspfx = util.canonpath(repo.root, cwd, pat)
1036 1036 if destdirexists:
1037 1037 striplen = len(os.path.split(abspfx)[0])
1038 1038 else:
1039 1039 striplen = len(abspfx)
1040 1040 if striplen:
1041 1041 striplen += len(os.sep)
1042 1042 res = lambda p: os.path.join(dest, p[striplen:])
1043 1043 elif destdirexists:
1044 1044 res = lambda p: os.path.join(dest, os.path.basename(p))
1045 1045 else:
1046 1046 res = lambda p: dest
1047 1047 return res
1048 1048
1049 1049 def targetpathafterfn(pat, dest, srcs):
1050 1050 if util.patkind(pat, None)[0]:
1051 1051 # a mercurial pattern
1052 1052 res = lambda p: os.path.join(dest, os.path.basename(p))
1053 1053 else:
1054 1054 abspfx = util.canonpath(repo.root, cwd, pat)
1055 1055 if len(abspfx) < len(srcs[0][0]):
1056 1056 # A directory. Either the target path contains the last
1057 1057 # component of the source path or it does not.
1058 1058 def evalpath(striplen):
1059 1059 score = 0
1060 1060 for s in srcs:
1061 1061 t = os.path.join(dest, s[0][striplen:])
1062 1062 if os.path.exists(t):
1063 1063 score += 1
1064 1064 return score
1065 1065
1066 1066 striplen = len(abspfx)
1067 1067 if striplen:
1068 1068 striplen += len(os.sep)
1069 1069 if os.path.isdir(os.path.join(dest, os.path.split(abspfx)[1])):
1070 1070 score = evalpath(striplen)
1071 1071 striplen1 = len(os.path.split(abspfx)[0])
1072 1072 if striplen1:
1073 1073 striplen1 += len(os.sep)
1074 1074 if evalpath(striplen1) > score:
1075 1075 striplen = striplen1
1076 1076 res = lambda p: os.path.join(dest, p[striplen:])
1077 1077 else:
1078 1078 # a file
1079 1079 if destdirexists:
1080 1080 res = lambda p: os.path.join(dest, os.path.basename(p))
1081 1081 else:
1082 1082 res = lambda p: dest
1083 1083 return res
1084 1084
1085 1085
1086 1086 pats = list(pats)
1087 1087 if not pats:
1088 1088 raise util.Abort(_('no source or destination specified'))
1089 1089 if len(pats) == 1:
1090 1090 raise util.Abort(_('no destination specified'))
1091 1091 dest = pats.pop()
1092 1092 destdirexists = os.path.isdir(dest)
1093 1093 if (len(pats) > 1 or util.patkind(pats[0], None)[0]) and not destdirexists:
1094 1094 raise util.Abort(_('with multiple sources, destination must be an '
1095 1095 'existing directory'))
1096 1096 if opts['after']:
1097 1097 tfn = targetpathafterfn
1098 1098 else:
1099 1099 tfn = targetpathfn
1100 1100 copylist = []
1101 1101 for pat in pats:
1102 1102 srcs = []
1103 1103 for tag, abssrc, relsrc, exact in cmdutil.walk(repo, [pat], opts):
1104 1104 origsrc = okaytocopy(abssrc, relsrc, exact)
1105 1105 if origsrc:
1106 1106 srcs.append((origsrc, abssrc, relsrc, exact))
1107 1107 if not srcs:
1108 1108 continue
1109 1109 copylist.append((tfn(pat, dest, srcs), srcs))
1110 1110 if not copylist:
1111 1111 raise util.Abort(_('no files to copy'))
1112 1112
1113 1113 for targetpath, srcs in copylist:
1114 1114 for origsrc, abssrc, relsrc, exact in srcs:
1115 1115 copy(origsrc, abssrc, relsrc, targetpath(abssrc), exact)
1116 1116
1117 1117 if errors:
1118 1118 ui.warn(_('(consider using --after)\n'))
1119 1119 return errors, copied
1120 1120
1121 1121 def copy(ui, repo, *pats, **opts):
1122 1122 """mark files as copied for the next commit
1123 1123
1124 1124 Mark dest as having copies of source files. If dest is a
1125 1125 directory, copies are put in that directory. If dest is a file,
1126 1126 there can only be one source.
1127 1127
1128 1128 By default, this command copies the contents of files as they
1129 1129 stand in the working directory. If invoked with --after, the
1130 1130 operation is recorded, but no copying is performed.
1131 1131
1132 1132 This command takes effect in the next commit.
1133 1133
1134 1134 NOTE: This command should be treated as experimental. While it
1135 1135 should properly record copied files, this information is not yet
1136 1136 fully used by merge, nor fully reported by log.
1137 1137 """
1138 1138 wlock = repo.wlock(0)
1139 1139 errs, copied = docopy(ui, repo, pats, opts, wlock)
1140 1140 return errs
1141 1141
1142 1142 def debugancestor(ui, index, rev1, rev2):
1143 1143 """find the ancestor revision of two revisions in a given index"""
1144 1144 r = revlog.revlog(util.opener(os.getcwd(), audit=False), index, "", 0)
1145 1145 a = r.ancestor(r.lookup(rev1), r.lookup(rev2))
1146 1146 ui.write("%d:%s\n" % (r.rev(a), hex(a)))
1147 1147
1148 1148 def debugcomplete(ui, cmd='', **opts):
1149 1149 """returns the completion list associated with the given command"""
1150 1150
1151 1151 if opts['options']:
1152 1152 options = []
1153 1153 otables = [globalopts]
1154 1154 if cmd:
1155 1155 aliases, entry = findcmd(ui, cmd)
1156 1156 otables.append(entry[1])
1157 1157 for t in otables:
1158 1158 for o in t:
1159 1159 if o[0]:
1160 1160 options.append('-%s' % o[0])
1161 1161 options.append('--%s' % o[1])
1162 1162 ui.write("%s\n" % "\n".join(options))
1163 1163 return
1164 1164
1165 1165 clist = findpossible(ui, cmd).keys()
1166 1166 clist.sort()
1167 1167 ui.write("%s\n" % "\n".join(clist))
1168 1168
1169 1169 def debugrebuildstate(ui, repo, rev=None):
1170 1170 """rebuild the dirstate as it would look like for the given revision"""
1171 1171 if not rev:
1172 1172 rev = repo.changelog.tip()
1173 1173 else:
1174 1174 rev = repo.lookup(rev)
1175 1175 change = repo.changelog.read(rev)
1176 1176 n = change[0]
1177 1177 files = repo.manifest.read(n)
1178 1178 wlock = repo.wlock()
1179 1179 repo.dirstate.rebuild(rev, files)
1180 1180
1181 1181 def debugcheckstate(ui, repo):
1182 1182 """validate the correctness of the current dirstate"""
1183 1183 parent1, parent2 = repo.dirstate.parents()
1184 1184 repo.dirstate.read()
1185 1185 dc = repo.dirstate.map
1186 1186 keys = dc.keys()
1187 1187 keys.sort()
1188 1188 m1n = repo.changelog.read(parent1)[0]
1189 1189 m2n = repo.changelog.read(parent2)[0]
1190 1190 m1 = repo.manifest.read(m1n)
1191 1191 m2 = repo.manifest.read(m2n)
1192 1192 errors = 0
1193 1193 for f in dc:
1194 1194 state = repo.dirstate.state(f)
1195 1195 if state in "nr" and f not in m1:
1196 1196 ui.warn(_("%s in state %s, but not in manifest1\n") % (f, state))
1197 1197 errors += 1
1198 1198 if state in "a" and f in m1:
1199 1199 ui.warn(_("%s in state %s, but also in manifest1\n") % (f, state))
1200 1200 errors += 1
1201 1201 if state in "m" and f not in m1 and f not in m2:
1202 1202 ui.warn(_("%s in state %s, but not in either manifest\n") %
1203 1203 (f, state))
1204 1204 errors += 1
1205 1205 for f in m1:
1206 1206 state = repo.dirstate.state(f)
1207 1207 if state not in "nrm":
1208 1208 ui.warn(_("%s in manifest1, but listed as state %s") % (f, state))
1209 1209 errors += 1
1210 1210 if errors:
1211 1211 error = _(".hg/dirstate inconsistent with current parent's manifest")
1212 1212 raise util.Abort(error)
1213 1213
1214 1214 def debugconfig(ui, repo, *values):
1215 1215 """show combined config settings from all hgrc files
1216 1216
1217 1217 With no args, print names and values of all config items.
1218 1218
1219 1219 With one arg of the form section.name, print just the value of
1220 1220 that config item.
1221 1221
1222 1222 With multiple args, print names and values of all config items
1223 1223 with matching section names."""
1224 1224
1225 1225 if values:
1226 1226 if len([v for v in values if '.' in v]) > 1:
1227 1227 raise util.Abort(_('only one config item permitted'))
1228 1228 for section, name, value in ui.walkconfig():
1229 1229 sectname = section + '.' + name
1230 1230 if values:
1231 1231 for v in values:
1232 1232 if v == section:
1233 1233 ui.write('%s=%s\n' % (sectname, value))
1234 1234 elif v == sectname:
1235 1235 ui.write(value, '\n')
1236 1236 else:
1237 1237 ui.write('%s=%s\n' % (sectname, value))
1238 1238
1239 1239 def debugsetparents(ui, repo, rev1, rev2=None):
1240 1240 """manually set the parents of the current working directory
1241 1241
1242 1242 This is useful for writing repository conversion tools, but should
1243 1243 be used with care.
1244 1244 """
1245 1245
1246 1246 if not rev2:
1247 1247 rev2 = hex(nullid)
1248 1248
1249 1249 repo.dirstate.setparents(repo.lookup(rev1), repo.lookup(rev2))
1250 1250
1251 1251 def debugstate(ui, repo):
1252 1252 """show the contents of the current dirstate"""
1253 1253 repo.dirstate.read()
1254 1254 dc = repo.dirstate.map
1255 1255 keys = dc.keys()
1256 1256 keys.sort()
1257 1257 for file_ in keys:
1258 1258 ui.write("%c %3o %10d %s %s\n"
1259 1259 % (dc[file_][0], dc[file_][1] & 0777, dc[file_][2],
1260 1260 time.strftime("%x %X",
1261 1261 time.localtime(dc[file_][3])), file_))
1262 1262 for f in repo.dirstate.copies:
1263 1263 ui.write(_("copy: %s -> %s\n") % (repo.dirstate.copies[f], f))
1264 1264
1265 1265 def debugdata(ui, file_, rev):
1266 1266 """dump the contents of an data file revision"""
1267 1267 r = revlog.revlog(util.opener(os.getcwd(), audit=False),
1268 1268 file_[:-2] + ".i", file_, 0)
1269 1269 try:
1270 1270 ui.write(r.revision(r.lookup(rev)))
1271 1271 except KeyError:
1272 raise util.Abort(_('invalid revision identifier %s'), rev)
1272 raise util.Abort(_('invalid revision identifier %s') % rev)
1273 1273
1274 1274 def debugindex(ui, file_):
1275 1275 """dump the contents of an index file"""
1276 1276 r = revlog.revlog(util.opener(os.getcwd(), audit=False), file_, "", 0)
1277 1277 ui.write(" rev offset length base linkrev" +
1278 1278 " nodeid p1 p2\n")
1279 1279 for i in range(r.count()):
1280 1280 node = r.node(i)
1281 1281 pp = r.parents(node)
1282 1282 ui.write("% 6d % 9d % 7d % 6d % 7d %s %s %s\n" % (
1283 1283 i, r.start(i), r.length(i), r.base(i), r.linkrev(node),
1284 1284 short(node), short(pp[0]), short(pp[1])))
1285 1285
1286 1286 def debugindexdot(ui, file_):
1287 1287 """dump an index DAG as a .dot file"""
1288 1288 r = revlog.revlog(util.opener(os.getcwd(), audit=False), file_, "", 0)
1289 1289 ui.write("digraph G {\n")
1290 1290 for i in range(r.count()):
1291 1291 node = r.node(i)
1292 1292 pp = r.parents(node)
1293 1293 ui.write("\t%d -> %d\n" % (r.rev(pp[0]), i))
1294 1294 if pp[1] != nullid:
1295 1295 ui.write("\t%d -> %d\n" % (r.rev(pp[1]), i))
1296 1296 ui.write("}\n")
1297 1297
1298 1298 def debugrename(ui, repo, file, rev=None):
1299 1299 """dump rename information"""
1300 1300 r = repo.file(relpath(repo, [file])[0])
1301 1301 if rev:
1302 1302 try:
1303 1303 # assume all revision numbers are for changesets
1304 1304 n = repo.lookup(rev)
1305 1305 change = repo.changelog.read(n)
1306 1306 m = repo.manifest.read(change[0])
1307 1307 n = m[relpath(repo, [file])[0]]
1308 1308 except (hg.RepoError, KeyError):
1309 1309 n = r.lookup(rev)
1310 1310 else:
1311 1311 n = r.tip()
1312 1312 m = r.renamed(n)
1313 1313 if m:
1314 1314 ui.write(_("renamed from %s:%s\n") % (m[0], hex(m[1])))
1315 1315 else:
1316 1316 ui.write(_("not renamed\n"))
1317 1317
1318 1318 def debugwalk(ui, repo, *pats, **opts):
1319 1319 """show how files match on given patterns"""
1320 1320 items = list(cmdutil.walk(repo, pats, opts))
1321 1321 if not items:
1322 1322 return
1323 1323 fmt = '%%s %%-%ds %%-%ds %%s' % (
1324 1324 max([len(abs) for (src, abs, rel, exact) in items]),
1325 1325 max([len(rel) for (src, abs, rel, exact) in items]))
1326 1326 for src, abs, rel, exact in items:
1327 1327 line = fmt % (src, abs, rel, exact and 'exact' or '')
1328 1328 ui.write("%s\n" % line.rstrip())
1329 1329
1330 1330 def diff(ui, repo, *pats, **opts):
1331 1331 """diff repository (or selected files)
1332 1332
1333 1333 Show differences between revisions for the specified files.
1334 1334
1335 1335 Differences between files are shown using the unified diff format.
1336 1336
1337 1337 When two revision arguments are given, then changes are shown
1338 1338 between those revisions. If only one revision is specified then
1339 1339 that revision is compared to the working directory, and, when no
1340 1340 revisions are specified, the working directory files are compared
1341 1341 to its parent.
1342 1342
1343 1343 Without the -a option, diff will avoid generating diffs of files
1344 1344 it detects as binary. With -a, diff will generate a diff anyway,
1345 1345 probably with undesirable results.
1346 1346 """
1347 1347 node1, node2 = revpair(ui, repo, opts['rev'])
1348 1348
1349 1349 fns, matchfn, anypats = cmdutil.matchpats(repo, pats, opts)
1350 1350
1351 1351 patch.diff(repo, node1, node2, fns, match=matchfn,
1352 1352 opts=patch.diffopts(ui, opts))
1353 1353
1354 1354 def export(ui, repo, *changesets, **opts):
1355 1355 """dump the header and diffs for one or more changesets
1356 1356
1357 1357 Print the changeset header and diffs for one or more revisions.
1358 1358
1359 1359 The information shown in the changeset header is: author,
1360 1360 changeset hash, parent and commit comment.
1361 1361
1362 1362 Output may be to a file, in which case the name of the file is
1363 1363 given using a format string. The formatting rules are as follows:
1364 1364
1365 1365 %% literal "%" character
1366 1366 %H changeset hash (40 bytes of hexadecimal)
1367 1367 %N number of patches being generated
1368 1368 %R changeset revision number
1369 1369 %b basename of the exporting repository
1370 1370 %h short-form changeset hash (12 bytes of hexadecimal)
1371 1371 %n zero-padded sequence number, starting at 1
1372 1372 %r zero-padded changeset revision number
1373 1373
1374 1374 Without the -a option, export will avoid generating diffs of files
1375 1375 it detects as binary. With -a, export will generate a diff anyway,
1376 1376 probably with undesirable results.
1377 1377
1378 1378 With the --switch-parent option, the diff will be against the second
1379 1379 parent. It can be useful to review a merge.
1380 1380 """
1381 1381 if not changesets:
1382 1382 raise util.Abort(_("export requires at least one changeset"))
1383 1383 revs = list(revrange(ui, repo, changesets))
1384 1384 if len(revs) > 1:
1385 1385 ui.note(_('exporting patches:\n'))
1386 1386 else:
1387 1387 ui.note(_('exporting patch:\n'))
1388 1388 patch.export(repo, map(repo.lookup, revs), template=opts['output'],
1389 1389 switch_parent=opts['switch_parent'],
1390 1390 opts=patch.diffopts(ui, opts))
1391 1391
1392 1392 def forget(ui, repo, *pats, **opts):
1393 1393 """don't add the specified files on the next commit (DEPRECATED)
1394 1394
1395 1395 (DEPRECATED)
1396 1396 Undo an 'hg add' scheduled for the next commit.
1397 1397
1398 1398 This command is now deprecated and will be removed in a future
1399 1399 release. Please use revert instead.
1400 1400 """
1401 1401 ui.warn(_("(the forget command is deprecated; use revert instead)\n"))
1402 1402 forget = []
1403 1403 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts):
1404 1404 if repo.dirstate.state(abs) == 'a':
1405 1405 forget.append(abs)
1406 1406 if ui.verbose or not exact:
1407 1407 ui.status(_('forgetting %s\n') % ((pats and rel) or abs))
1408 1408 repo.forget(forget)
1409 1409
1410 1410 def grep(ui, repo, pattern, *pats, **opts):
1411 1411 """search for a pattern in specified files and revisions
1412 1412
1413 1413 Search revisions of files for a regular expression.
1414 1414
1415 1415 This command behaves differently than Unix grep. It only accepts
1416 1416 Python/Perl regexps. It searches repository history, not the
1417 1417 working directory. It always prints the revision number in which
1418 1418 a match appears.
1419 1419
1420 1420 By default, grep only prints output for the first revision of a
1421 1421 file in which it finds a match. To get it to print every revision
1422 1422 that contains a change in match status ("-" for a match that
1423 1423 becomes a non-match, or "+" for a non-match that becomes a match),
1424 1424 use the --all flag.
1425 1425 """
1426 1426 reflags = 0
1427 1427 if opts['ignore_case']:
1428 1428 reflags |= re.I
1429 1429 regexp = re.compile(pattern, reflags)
1430 1430 sep, eol = ':', '\n'
1431 1431 if opts['print0']:
1432 1432 sep = eol = '\0'
1433 1433
1434 1434 fcache = {}
1435 1435 def getfile(fn):
1436 1436 if fn not in fcache:
1437 1437 fcache[fn] = repo.file(fn)
1438 1438 return fcache[fn]
1439 1439
1440 1440 def matchlines(body):
1441 1441 begin = 0
1442 1442 linenum = 0
1443 1443 while True:
1444 1444 match = regexp.search(body, begin)
1445 1445 if not match:
1446 1446 break
1447 1447 mstart, mend = match.span()
1448 1448 linenum += body.count('\n', begin, mstart) + 1
1449 1449 lstart = body.rfind('\n', begin, mstart) + 1 or begin
1450 1450 lend = body.find('\n', mend)
1451 1451 yield linenum, mstart - lstart, mend - lstart, body[lstart:lend]
1452 1452 begin = lend + 1
1453 1453
1454 1454 class linestate(object):
1455 1455 def __init__(self, line, linenum, colstart, colend):
1456 1456 self.line = line
1457 1457 self.linenum = linenum
1458 1458 self.colstart = colstart
1459 1459 self.colend = colend
1460 1460
1461 1461 def __eq__(self, other):
1462 1462 return self.line == other.line
1463 1463
1464 1464 matches = {}
1465 1465 copies = {}
1466 1466 def grepbody(fn, rev, body):
1467 1467 matches[rev].setdefault(fn, [])
1468 1468 m = matches[rev][fn]
1469 1469 for lnum, cstart, cend, line in matchlines(body):
1470 1470 s = linestate(line, lnum, cstart, cend)
1471 1471 m.append(s)
1472 1472
1473 1473 def difflinestates(a, b):
1474 1474 sm = difflib.SequenceMatcher(None, a, b)
1475 1475 for tag, alo, ahi, blo, bhi in sm.get_opcodes():
1476 1476 if tag == 'insert':
1477 1477 for i in range(blo, bhi):
1478 1478 yield ('+', b[i])
1479 1479 elif tag == 'delete':
1480 1480 for i in range(alo, ahi):
1481 1481 yield ('-', a[i])
1482 1482 elif tag == 'replace':
1483 1483 for i in range(alo, ahi):
1484 1484 yield ('-', a[i])
1485 1485 for i in range(blo, bhi):
1486 1486 yield ('+', b[i])
1487 1487
1488 1488 prev = {}
1489 1489 ucache = {}
1490 1490 def display(fn, rev, states, prevstates):
1491 1491 counts = {'-': 0, '+': 0}
1492 1492 filerevmatches = {}
1493 1493 if incrementing or not opts['all']:
1494 1494 a, b = prevstates, states
1495 1495 else:
1496 1496 a, b = states, prevstates
1497 1497 for change, l in difflinestates(a, b):
1498 1498 if incrementing or not opts['all']:
1499 1499 r = rev
1500 1500 else:
1501 1501 r = prev[fn]
1502 1502 cols = [fn, str(r)]
1503 1503 if opts['line_number']:
1504 1504 cols.append(str(l.linenum))
1505 1505 if opts['all']:
1506 1506 cols.append(change)
1507 1507 if opts['user']:
1508 1508 cols.append(trimuser(ui, getchange(r)[1], rev,
1509 1509 ucache))
1510 1510 if opts['files_with_matches']:
1511 1511 c = (fn, rev)
1512 1512 if c in filerevmatches:
1513 1513 continue
1514 1514 filerevmatches[c] = 1
1515 1515 else:
1516 1516 cols.append(l.line)
1517 1517 ui.write(sep.join(cols), eol)
1518 1518 counts[change] += 1
1519 1519 return counts['+'], counts['-']
1520 1520
1521 1521 fstate = {}
1522 1522 skip = {}
1523 1523 changeiter, getchange, matchfn = walkchangerevs(ui, repo, pats, opts)
1524 1524 count = 0
1525 1525 incrementing = False
1526 1526 follow = opts.get('follow')
1527 1527 for st, rev, fns in changeiter:
1528 1528 if st == 'window':
1529 1529 incrementing = rev
1530 1530 matches.clear()
1531 1531 elif st == 'add':
1532 1532 change = repo.changelog.read(repo.lookup(str(rev)))
1533 1533 mf = repo.manifest.read(change[0])
1534 1534 matches[rev] = {}
1535 1535 for fn in fns:
1536 1536 if fn in skip:
1537 1537 continue
1538 1538 fstate.setdefault(fn, {})
1539 1539 try:
1540 1540 grepbody(fn, rev, getfile(fn).read(mf[fn]))
1541 1541 if follow:
1542 1542 copied = getfile(fn).renamed(mf[fn])
1543 1543 if copied:
1544 1544 copies.setdefault(rev, {})[fn] = copied[0]
1545 1545 except KeyError:
1546 1546 pass
1547 1547 elif st == 'iter':
1548 1548 states = matches[rev].items()
1549 1549 states.sort()
1550 1550 for fn, m in states:
1551 1551 copy = copies.get(rev, {}).get(fn)
1552 1552 if fn in skip:
1553 1553 if copy:
1554 1554 skip[copy] = True
1555 1555 continue
1556 1556 if incrementing or not opts['all'] or fstate[fn]:
1557 1557 pos, neg = display(fn, rev, m, fstate[fn])
1558 1558 count += pos + neg
1559 1559 if pos and not opts['all']:
1560 1560 skip[fn] = True
1561 1561 if copy:
1562 1562 skip[copy] = True
1563 1563 fstate[fn] = m
1564 1564 if copy:
1565 1565 fstate[copy] = m
1566 1566 prev[fn] = rev
1567 1567
1568 1568 if not incrementing:
1569 1569 fstate = fstate.items()
1570 1570 fstate.sort()
1571 1571 for fn, state in fstate:
1572 1572 if fn in skip:
1573 1573 continue
1574 1574 if fn not in copies.get(prev[fn], {}):
1575 1575 display(fn, rev, {}, state)
1576 1576 return (count == 0 and 1) or 0
1577 1577
1578 1578 def heads(ui, repo, **opts):
1579 1579 """show current repository heads
1580 1580
1581 1581 Show all repository head changesets.
1582 1582
1583 1583 Repository "heads" are changesets that don't have children
1584 1584 changesets. They are where development generally takes place and
1585 1585 are the usual targets for update and merge operations.
1586 1586 """
1587 1587 if opts['rev']:
1588 1588 heads = repo.heads(repo.lookup(opts['rev']))
1589 1589 else:
1590 1590 heads = repo.heads()
1591 1591 br = None
1592 1592 if opts['branches']:
1593 1593 br = repo.branchlookup(heads)
1594 1594 displayer = show_changeset(ui, repo, opts)
1595 1595 for n in heads:
1596 1596 displayer.show(changenode=n, brinfo=br)
1597 1597
1598 1598 def identify(ui, repo):
1599 1599 """print information about the working copy
1600 1600
1601 1601 Print a short summary of the current state of the repo.
1602 1602
1603 1603 This summary identifies the repository state using one or two parent
1604 1604 hash identifiers, followed by a "+" if there are uncommitted changes
1605 1605 in the working directory, followed by a list of tags for this revision.
1606 1606 """
1607 1607 parents = [p for p in repo.dirstate.parents() if p != nullid]
1608 1608 if not parents:
1609 1609 ui.write(_("unknown\n"))
1610 1610 return
1611 1611
1612 1612 hexfunc = ui.debugflag and hex or short
1613 1613 modified, added, removed, deleted = repo.status()[:4]
1614 1614 output = ["%s%s" %
1615 1615 ('+'.join([hexfunc(parent) for parent in parents]),
1616 1616 (modified or added or removed or deleted) and "+" or "")]
1617 1617
1618 1618 if not ui.quiet:
1619 1619 # multiple tags for a single parent separated by '/'
1620 1620 parenttags = ['/'.join(tags)
1621 1621 for tags in map(repo.nodetags, parents) if tags]
1622 1622 # tags for multiple parents separated by ' + '
1623 1623 if parenttags:
1624 1624 output.append(' + '.join(parenttags))
1625 1625
1626 1626 ui.write("%s\n" % ' '.join(output))
1627 1627
1628 1628 def import_(ui, repo, patch1, *patches, **opts):
1629 1629 """import an ordered set of patches
1630 1630
1631 1631 Import a list of patches and commit them individually.
1632 1632
1633 1633 If there are outstanding changes in the working directory, import
1634 1634 will abort unless given the -f flag.
1635 1635
1636 1636 You can import a patch straight from a mail message. Even patches
1637 1637 as attachments work (body part must be type text/plain or
1638 1638 text/x-patch to be used). From and Subject headers of email
1639 1639 message are used as default committer and commit message. All
1640 1640 text/plain body parts before first diff are added to commit
1641 1641 message.
1642 1642
1643 1643 If imported patch was generated by hg export, user and description
1644 1644 from patch override values from message headers and body. Values
1645 1645 given on command line with -m and -u override these.
1646 1646
1647 1647 To read a patch from standard input, use patch name "-".
1648 1648 """
1649 1649 patches = (patch1,) + patches
1650 1650
1651 1651 if not opts['force']:
1652 1652 bail_if_changed(repo)
1653 1653
1654 1654 d = opts["base"]
1655 1655 strip = opts["strip"]
1656 1656
1657 1657 wlock = repo.wlock()
1658 1658 lock = repo.lock()
1659 1659
1660 1660 for p in patches:
1661 1661 pf = os.path.join(d, p)
1662 1662
1663 1663 if pf == '-':
1664 1664 ui.status(_("applying patch from stdin\n"))
1665 1665 tmpname, message, user, date = patch.extract(ui, sys.stdin)
1666 1666 else:
1667 1667 ui.status(_("applying %s\n") % p)
1668 1668 tmpname, message, user, date = patch.extract(ui, file(pf))
1669 1669
1670 1670 if tmpname is None:
1671 1671 raise util.Abort(_('no diffs found'))
1672 1672
1673 1673 try:
1674 1674 if opts['message']:
1675 1675 # pickup the cmdline msg
1676 1676 message = opts['message']
1677 1677 elif message:
1678 1678 # pickup the patch msg
1679 1679 message = message.strip()
1680 1680 else:
1681 1681 # launch the editor
1682 1682 message = None
1683 1683 ui.debug(_('message:\n%s\n') % message)
1684 1684
1685 1685 files, fuzz = patch.patch(tmpname, ui, strip=strip, cwd=repo.root)
1686 1686 files = patch.updatedir(ui, repo, files, wlock=wlock)
1687 1687 repo.commit(files, message, user, date, wlock=wlock, lock=lock)
1688 1688 finally:
1689 1689 os.unlink(tmpname)
1690 1690
1691 1691 def incoming(ui, repo, source="default", **opts):
1692 1692 """show new changesets found in source
1693 1693
1694 1694 Show new changesets found in the specified path/URL or the default
1695 1695 pull location. These are the changesets that would be pulled if a pull
1696 1696 was requested.
1697 1697
1698 1698 For remote repository, using --bundle avoids downloading the changesets
1699 1699 twice if the incoming is followed by a pull.
1700 1700
1701 1701 See pull for valid source format details.
1702 1702 """
1703 1703 source = ui.expandpath(source)
1704 1704 setremoteconfig(ui, opts)
1705 1705
1706 1706 other = hg.repository(ui, source)
1707 1707 incoming = repo.findincoming(other, force=opts["force"])
1708 1708 if not incoming:
1709 1709 ui.status(_("no changes found\n"))
1710 1710 return
1711 1711
1712 1712 cleanup = None
1713 1713 try:
1714 1714 fname = opts["bundle"]
1715 1715 if fname or not other.local():
1716 1716 # create a bundle (uncompressed if other repo is not local)
1717 1717 cg = other.changegroup(incoming, "incoming")
1718 1718 fname = cleanup = write_bundle(cg, fname, compress=other.local())
1719 1719 # keep written bundle?
1720 1720 if opts["bundle"]:
1721 1721 cleanup = None
1722 1722 if not other.local():
1723 1723 # use the created uncompressed bundlerepo
1724 1724 other = bundlerepo.bundlerepository(ui, repo.root, fname)
1725 1725
1726 1726 revs = None
1727 1727 if opts['rev']:
1728 1728 revs = [other.lookup(rev) for rev in opts['rev']]
1729 1729 o = other.changelog.nodesbetween(incoming, revs)[0]
1730 1730 if opts['newest_first']:
1731 1731 o.reverse()
1732 1732 displayer = show_changeset(ui, other, opts)
1733 1733 for n in o:
1734 1734 parents = [p for p in other.changelog.parents(n) if p != nullid]
1735 1735 if opts['no_merges'] and len(parents) == 2:
1736 1736 continue
1737 1737 displayer.show(changenode=n)
1738 1738 if opts['patch']:
1739 1739 prev = (parents and parents[0]) or nullid
1740 1740 patch.diff(other, prev, n, fp=repo.ui)
1741 1741 ui.write("\n")
1742 1742 finally:
1743 1743 if hasattr(other, 'close'):
1744 1744 other.close()
1745 1745 if cleanup:
1746 1746 os.unlink(cleanup)
1747 1747
1748 1748 def init(ui, dest=".", **opts):
1749 1749 """create a new repository in the given directory
1750 1750
1751 1751 Initialize a new repository in the given directory. If the given
1752 1752 directory does not exist, it is created.
1753 1753
1754 1754 If no directory is given, the current directory is used.
1755 1755
1756 1756 It is possible to specify an ssh:// URL as the destination.
1757 1757 Look at the help text for the pull command for important details
1758 1758 about ssh:// URLs.
1759 1759 """
1760 1760 setremoteconfig(ui, opts)
1761 1761 hg.repository(ui, dest, create=1)
1762 1762
1763 1763 def locate(ui, repo, *pats, **opts):
1764 1764 """locate files matching specific patterns
1765 1765
1766 1766 Print all files under Mercurial control whose names match the
1767 1767 given patterns.
1768 1768
1769 1769 This command searches the current directory and its
1770 1770 subdirectories. To search an entire repository, move to the root
1771 1771 of the repository.
1772 1772
1773 1773 If no patterns are given to match, this command prints all file
1774 1774 names.
1775 1775
1776 1776 If you want to feed the output of this command into the "xargs"
1777 1777 command, use the "-0" option to both this command and "xargs".
1778 1778 This will avoid the problem of "xargs" treating single filenames
1779 1779 that contain white space as multiple filenames.
1780 1780 """
1781 1781 end = opts['print0'] and '\0' or '\n'
1782 1782 rev = opts['rev']
1783 1783 if rev:
1784 1784 node = repo.lookup(rev)
1785 1785 else:
1786 1786 node = None
1787 1787
1788 1788 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts, node=node,
1789 1789 head='(?:.*/|)'):
1790 1790 if not node and repo.dirstate.state(abs) == '?':
1791 1791 continue
1792 1792 if opts['fullpath']:
1793 1793 ui.write(os.path.join(repo.root, abs), end)
1794 1794 else:
1795 1795 ui.write(((pats and rel) or abs), end)
1796 1796
1797 1797 def log(ui, repo, *pats, **opts):
1798 1798 """show revision history of entire repository or files
1799 1799
1800 1800 Print the revision history of the specified files or the entire
1801 1801 project.
1802 1802
1803 1803 File history is shown without following rename or copy history of
1804 1804 files. Use -f/--follow with a file name to follow history across
1805 1805 renames and copies. --follow without a file name will only show
1806 1806 ancestors or descendants of the starting revision. --follow-first
1807 1807 only follows the first parent of merge revisions.
1808 1808
1809 1809 If no revision range is specified, the default is tip:0 unless
1810 1810 --follow is set, in which case the working directory parent is
1811 1811 used as the starting revision.
1812 1812
1813 1813 By default this command outputs: changeset id and hash, tags,
1814 1814 non-trivial parents, user, date and time, and a summary for each
1815 1815 commit. When the -v/--verbose switch is used, the list of changed
1816 1816 files and full commit message is shown.
1817 1817 """
1818 1818 class dui(object):
1819 1819 # Implement and delegate some ui protocol. Save hunks of
1820 1820 # output for later display in the desired order.
1821 1821 def __init__(self, ui):
1822 1822 self.ui = ui
1823 1823 self.hunk = {}
1824 1824 self.header = {}
1825 1825 def bump(self, rev):
1826 1826 self.rev = rev
1827 1827 self.hunk[rev] = []
1828 1828 self.header[rev] = []
1829 1829 def note(self, *args):
1830 1830 if self.verbose:
1831 1831 self.write(*args)
1832 1832 def status(self, *args):
1833 1833 if not self.quiet:
1834 1834 self.write(*args)
1835 1835 def write(self, *args):
1836 1836 self.hunk[self.rev].append(args)
1837 1837 def write_header(self, *args):
1838 1838 self.header[self.rev].append(args)
1839 1839 def debug(self, *args):
1840 1840 if self.debugflag:
1841 1841 self.write(*args)
1842 1842 def __getattr__(self, key):
1843 1843 return getattr(self.ui, key)
1844 1844
1845 1845 changeiter, getchange, matchfn = walkchangerevs(ui, repo, pats, opts)
1846 1846
1847 1847 if opts['limit']:
1848 1848 try:
1849 1849 limit = int(opts['limit'])
1850 1850 except ValueError:
1851 1851 raise util.Abort(_('limit must be a positive integer'))
1852 1852 if limit <= 0: raise util.Abort(_('limit must be positive'))
1853 1853 else:
1854 1854 limit = sys.maxint
1855 1855 count = 0
1856 1856
1857 1857 displayer = show_changeset(ui, repo, opts)
1858 1858 for st, rev, fns in changeiter:
1859 1859 if st == 'window':
1860 1860 du = dui(ui)
1861 1861 displayer.ui = du
1862 1862 elif st == 'add':
1863 1863 du.bump(rev)
1864 1864 changenode = repo.changelog.node(rev)
1865 1865 parents = [p for p in repo.changelog.parents(changenode)
1866 1866 if p != nullid]
1867 1867 if opts['no_merges'] and len(parents) == 2:
1868 1868 continue
1869 1869 if opts['only_merges'] and len(parents) != 2:
1870 1870 continue
1871 1871
1872 1872 if opts['keyword']:
1873 1873 changes = getchange(rev)
1874 1874 miss = 0
1875 1875 for k in [kw.lower() for kw in opts['keyword']]:
1876 1876 if not (k in changes[1].lower() or
1877 1877 k in changes[4].lower() or
1878 1878 k in " ".join(changes[3][:20]).lower()):
1879 1879 miss = 1
1880 1880 break
1881 1881 if miss:
1882 1882 continue
1883 1883
1884 1884 br = None
1885 1885 if opts['branches']:
1886 1886 br = repo.branchlookup([repo.changelog.node(rev)])
1887 1887
1888 1888 displayer.show(rev, brinfo=br)
1889 1889 if opts['patch']:
1890 1890 prev = (parents and parents[0]) or nullid
1891 1891 patch.diff(repo, prev, changenode, match=matchfn, fp=du)
1892 1892 du.write("\n\n")
1893 1893 elif st == 'iter':
1894 1894 if count == limit: break
1895 1895 if du.header[rev]:
1896 1896 for args in du.header[rev]:
1897 1897 ui.write_header(*args)
1898 1898 if du.hunk[rev]:
1899 1899 count += 1
1900 1900 for args in du.hunk[rev]:
1901 1901 ui.write(*args)
1902 1902
1903 1903 def manifest(ui, repo, rev=None):
1904 1904 """output the latest or given revision of the project manifest
1905 1905
1906 1906 Print a list of version controlled files for the given revision.
1907 1907
1908 1908 The manifest is the list of files being version controlled. If no revision
1909 1909 is given then the tip is used.
1910 1910 """
1911 1911 if rev:
1912 1912 try:
1913 1913 # assume all revision numbers are for changesets
1914 1914 n = repo.lookup(rev)
1915 1915 change = repo.changelog.read(n)
1916 1916 n = change[0]
1917 1917 except hg.RepoError:
1918 1918 n = repo.manifest.lookup(rev)
1919 1919 else:
1920 1920 n = repo.manifest.tip()
1921 1921 m = repo.manifest.read(n)
1922 1922 files = m.keys()
1923 1923 files.sort()
1924 1924
1925 1925 for f in files:
1926 1926 ui.write("%40s %3s %s\n" % (hex(m[f]),
1927 1927 m.execf(f) and "755" or "644", f))
1928 1928
1929 1929 def merge(ui, repo, node=None, force=None, branch=None):
1930 1930 """Merge working directory with another revision
1931 1931
1932 1932 Merge the contents of the current working directory and the
1933 1933 requested revision. Files that changed between either parent are
1934 1934 marked as changed for the next commit and a commit must be
1935 1935 performed before any further updates are allowed.
1936 1936
1937 1937 If no revision is specified, the working directory's parent is a
1938 1938 head revision, and the repository contains exactly one other head,
1939 1939 the other head is merged with by default. Otherwise, an explicit
1940 1940 revision to merge with must be provided.
1941 1941 """
1942 1942
1943 1943 if node or branch:
1944 1944 node = _lookup(repo, node, branch)
1945 1945 else:
1946 1946 heads = repo.heads()
1947 1947 if len(heads) > 2:
1948 1948 raise util.Abort(_('repo has %d heads - '
1949 1949 'please merge with an explicit rev') %
1950 1950 len(heads))
1951 1951 if len(heads) == 1:
1952 1952 raise util.Abort(_('there is nothing to merge - '
1953 1953 'use "hg update" instead'))
1954 1954 parent = repo.dirstate.parents()[0]
1955 1955 if parent not in heads:
1956 1956 raise util.Abort(_('working dir not at a head rev - '
1957 1957 'use "hg update" or merge with an explicit rev'))
1958 1958 node = parent == heads[0] and heads[-1] or heads[0]
1959 1959 return hg.merge(repo, node, force=force)
1960 1960
1961 1961 def outgoing(ui, repo, dest=None, **opts):
1962 1962 """show changesets not found in destination
1963 1963
1964 1964 Show changesets not found in the specified destination repository or
1965 1965 the default push location. These are the changesets that would be pushed
1966 1966 if a push was requested.
1967 1967
1968 1968 See pull for valid destination format details.
1969 1969 """
1970 1970 dest = ui.expandpath(dest or 'default-push', dest or 'default')
1971 1971 setremoteconfig(ui, opts)
1972 1972 revs = None
1973 1973 if opts['rev']:
1974 1974 revs = [repo.lookup(rev) for rev in opts['rev']]
1975 1975
1976 1976 other = hg.repository(ui, dest)
1977 1977 o = repo.findoutgoing(other, force=opts['force'])
1978 1978 if not o:
1979 1979 ui.status(_("no changes found\n"))
1980 1980 return
1981 1981 o = repo.changelog.nodesbetween(o, revs)[0]
1982 1982 if opts['newest_first']:
1983 1983 o.reverse()
1984 1984 displayer = show_changeset(ui, repo, opts)
1985 1985 for n in o:
1986 1986 parents = [p for p in repo.changelog.parents(n) if p != nullid]
1987 1987 if opts['no_merges'] and len(parents) == 2:
1988 1988 continue
1989 1989 displayer.show(changenode=n)
1990 1990 if opts['patch']:
1991 1991 prev = (parents and parents[0]) or nullid
1992 1992 patch.diff(repo, prev, n)
1993 1993 ui.write("\n")
1994 1994
1995 1995 def parents(ui, repo, file_=None, rev=None, branches=None, **opts):
1996 1996 """show the parents of the working dir or revision
1997 1997
1998 1998 Print the working directory's parent revisions.
1999 1999 """
2000 2000 # legacy
2001 2001 if file_ and not rev:
2002 2002 try:
2003 2003 rev = repo.lookup(file_)
2004 2004 file_ = None
2005 2005 except hg.RepoError:
2006 2006 pass
2007 2007 else:
2008 2008 ui.warn(_("'hg parent REV' is deprecated, "
2009 2009 "please use 'hg parents -r REV instead\n"))
2010 2010
2011 2011 if rev:
2012 2012 if file_:
2013 2013 ctx = repo.filectx(file_, changeid=rev)
2014 2014 else:
2015 2015 ctx = repo.changectx(rev)
2016 2016 p = [cp.node() for cp in ctx.parents()]
2017 2017 else:
2018 2018 p = repo.dirstate.parents()
2019 2019
2020 2020 br = None
2021 2021 if branches is not None:
2022 2022 br = repo.branchlookup(p)
2023 2023 displayer = show_changeset(ui, repo, opts)
2024 2024 for n in p:
2025 2025 if n != nullid:
2026 2026 displayer.show(changenode=n, brinfo=br)
2027 2027
2028 2028 def paths(ui, repo, search=None):
2029 2029 """show definition of symbolic path names
2030 2030
2031 2031 Show definition of symbolic path name NAME. If no name is given, show
2032 2032 definition of available names.
2033 2033
2034 2034 Path names are defined in the [paths] section of /etc/mercurial/hgrc
2035 2035 and $HOME/.hgrc. If run inside a repository, .hg/hgrc is used, too.
2036 2036 """
2037 2037 if search:
2038 2038 for name, path in ui.configitems("paths"):
2039 2039 if name == search:
2040 2040 ui.write("%s\n" % path)
2041 2041 return
2042 2042 ui.warn(_("not found!\n"))
2043 2043 return 1
2044 2044 else:
2045 2045 for name, path in ui.configitems("paths"):
2046 2046 ui.write("%s = %s\n" % (name, path))
2047 2047
2048 2048 def postincoming(ui, repo, modheads, optupdate):
2049 2049 if modheads == 0:
2050 2050 return
2051 2051 if optupdate:
2052 2052 if modheads == 1:
2053 2053 return hg.update(repo, repo.changelog.tip()) # update
2054 2054 else:
2055 2055 ui.status(_("not updating, since new heads added\n"))
2056 2056 if modheads > 1:
2057 2057 ui.status(_("(run 'hg heads' to see heads, 'hg merge' to merge)\n"))
2058 2058 else:
2059 2059 ui.status(_("(run 'hg update' to get a working copy)\n"))
2060 2060
2061 2061 def pull(ui, repo, source="default", **opts):
2062 2062 """pull changes from the specified source
2063 2063
2064 2064 Pull changes from a remote repository to a local one.
2065 2065
2066 2066 This finds all changes from the repository at the specified path
2067 2067 or URL and adds them to the local repository. By default, this
2068 2068 does not update the copy of the project in the working directory.
2069 2069
2070 2070 Valid URLs are of the form:
2071 2071
2072 2072 local/filesystem/path
2073 2073 http://[user@]host[:port]/[path]
2074 2074 https://[user@]host[:port]/[path]
2075 2075 ssh://[user@]host[:port]/[path]
2076 2076
2077 2077 Some notes about using SSH with Mercurial:
2078 2078 - SSH requires an accessible shell account on the destination machine
2079 2079 and a copy of hg in the remote path or specified with as remotecmd.
2080 2080 - path is relative to the remote user's home directory by default.
2081 2081 Use an extra slash at the start of a path to specify an absolute path:
2082 2082 ssh://example.com//tmp/repository
2083 2083 - Mercurial doesn't use its own compression via SSH; the right thing
2084 2084 to do is to configure it in your ~/.ssh/ssh_config, e.g.:
2085 2085 Host *.mylocalnetwork.example.com
2086 2086 Compression off
2087 2087 Host *
2088 2088 Compression on
2089 2089 Alternatively specify "ssh -C" as your ssh command in your hgrc or
2090 2090 with the --ssh command line option.
2091 2091 """
2092 2092 source = ui.expandpath(source)
2093 2093 setremoteconfig(ui, opts)
2094 2094
2095 2095 other = hg.repository(ui, source)
2096 2096 ui.status(_('pulling from %s\n') % (source))
2097 2097 revs = None
2098 2098 if opts['rev'] and not other.local():
2099 2099 raise util.Abort(_("pull -r doesn't work for remote repositories yet"))
2100 2100 elif opts['rev']:
2101 2101 revs = [other.lookup(rev) for rev in opts['rev']]
2102 2102 modheads = repo.pull(other, heads=revs, force=opts['force'])
2103 2103 return postincoming(ui, repo, modheads, opts['update'])
2104 2104
2105 2105 def push(ui, repo, dest=None, **opts):
2106 2106 """push changes to the specified destination
2107 2107
2108 2108 Push changes from the local repository to the given destination.
2109 2109
2110 2110 This is the symmetrical operation for pull. It helps to move
2111 2111 changes from the current repository to a different one. If the
2112 2112 destination is local this is identical to a pull in that directory
2113 2113 from the current one.
2114 2114
2115 2115 By default, push will refuse to run if it detects the result would
2116 2116 increase the number of remote heads. This generally indicates the
2117 2117 the client has forgotten to sync and merge before pushing.
2118 2118
2119 2119 Valid URLs are of the form:
2120 2120
2121 2121 local/filesystem/path
2122 2122 ssh://[user@]host[:port]/[path]
2123 2123
2124 2124 Look at the help text for the pull command for important details
2125 2125 about ssh:// URLs.
2126 2126
2127 2127 Pushing to http:// and https:// URLs is possible, too, if this
2128 2128 feature is enabled on the remote Mercurial server.
2129 2129 """
2130 2130 dest = ui.expandpath(dest or 'default-push', dest or 'default')
2131 2131 setremoteconfig(ui, opts)
2132 2132
2133 2133 other = hg.repository(ui, dest)
2134 2134 ui.status('pushing to %s\n' % (dest))
2135 2135 revs = None
2136 2136 if opts['rev']:
2137 2137 revs = [repo.lookup(rev) for rev in opts['rev']]
2138 2138 r = repo.push(other, opts['force'], revs=revs)
2139 2139 return r == 0
2140 2140
2141 2141 def rawcommit(ui, repo, *flist, **rc):
2142 2142 """raw commit interface (DEPRECATED)
2143 2143
2144 2144 (DEPRECATED)
2145 2145 Lowlevel commit, for use in helper scripts.
2146 2146
2147 2147 This command is not intended to be used by normal users, as it is
2148 2148 primarily useful for importing from other SCMs.
2149 2149
2150 2150 This command is now deprecated and will be removed in a future
2151 2151 release, please use debugsetparents and commit instead.
2152 2152 """
2153 2153
2154 2154 ui.warn(_("(the rawcommit command is deprecated)\n"))
2155 2155
2156 2156 message = rc['message']
2157 2157 if not message and rc['logfile']:
2158 2158 try:
2159 2159 message = open(rc['logfile']).read()
2160 2160 except IOError:
2161 2161 pass
2162 2162 if not message and not rc['logfile']:
2163 2163 raise util.Abort(_("missing commit message"))
2164 2164
2165 2165 files = relpath(repo, list(flist))
2166 2166 if rc['files']:
2167 2167 files += open(rc['files']).read().splitlines()
2168 2168
2169 2169 rc['parent'] = map(repo.lookup, rc['parent'])
2170 2170
2171 2171 try:
2172 2172 repo.rawcommit(files, message, rc['user'], rc['date'], *rc['parent'])
2173 2173 except ValueError, inst:
2174 2174 raise util.Abort(str(inst))
2175 2175
2176 2176 def recover(ui, repo):
2177 2177 """roll back an interrupted transaction
2178 2178
2179 2179 Recover from an interrupted commit or pull.
2180 2180
2181 2181 This command tries to fix the repository status after an interrupted
2182 2182 operation. It should only be necessary when Mercurial suggests it.
2183 2183 """
2184 2184 if repo.recover():
2185 2185 return hg.verify(repo)
2186 2186 return 1
2187 2187
2188 2188 def remove(ui, repo, *pats, **opts):
2189 2189 """remove the specified files on the next commit
2190 2190
2191 2191 Schedule the indicated files for removal from the repository.
2192 2192
2193 2193 This command schedules the files to be removed at the next commit.
2194 2194 This only removes files from the current branch, not from the
2195 2195 entire project history. If the files still exist in the working
2196 2196 directory, they will be deleted from it. If invoked with --after,
2197 2197 files that have been manually deleted are marked as removed.
2198 2198
2199 2199 Modified files and added files are not removed by default. To
2200 2200 remove them, use the -f/--force option.
2201 2201 """
2202 2202 names = []
2203 2203 if not opts['after'] and not pats:
2204 2204 raise util.Abort(_('no files specified'))
2205 2205 files, matchfn, anypats = cmdutil.matchpats(repo, pats, opts)
2206 2206 exact = dict.fromkeys(files)
2207 2207 mardu = map(dict.fromkeys, repo.status(files=files, match=matchfn))[:5]
2208 2208 modified, added, removed, deleted, unknown = mardu
2209 2209 remove, forget = [], []
2210 2210 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts):
2211 2211 reason = None
2212 2212 if abs not in deleted and opts['after']:
2213 2213 reason = _('is still present')
2214 2214 elif abs in modified and not opts['force']:
2215 2215 reason = _('is modified (use -f to force removal)')
2216 2216 elif abs in added:
2217 2217 if opts['force']:
2218 2218 forget.append(abs)
2219 2219 continue
2220 2220 reason = _('has been marked for add (use -f to force removal)')
2221 2221 elif abs in unknown:
2222 2222 reason = _('is not managed')
2223 2223 elif abs in removed:
2224 2224 continue
2225 2225 if reason:
2226 2226 if exact:
2227 2227 ui.warn(_('not removing %s: file %s\n') % (rel, reason))
2228 2228 else:
2229 2229 if ui.verbose or not exact:
2230 2230 ui.status(_('removing %s\n') % rel)
2231 2231 remove.append(abs)
2232 2232 repo.forget(forget)
2233 2233 repo.remove(remove, unlink=not opts['after'])
2234 2234
2235 2235 def rename(ui, repo, *pats, **opts):
2236 2236 """rename files; equivalent of copy + remove
2237 2237
2238 2238 Mark dest as copies of sources; mark sources for deletion. If
2239 2239 dest is a directory, copies are put in that directory. If dest is
2240 2240 a file, there can only be one source.
2241 2241
2242 2242 By default, this command copies the contents of files as they
2243 2243 stand in the working directory. If invoked with --after, the
2244 2244 operation is recorded, but no copying is performed.
2245 2245
2246 2246 This command takes effect in the next commit.
2247 2247
2248 2248 NOTE: This command should be treated as experimental. While it
2249 2249 should properly record rename files, this information is not yet
2250 2250 fully used by merge, nor fully reported by log.
2251 2251 """
2252 2252 wlock = repo.wlock(0)
2253 2253 errs, copied = docopy(ui, repo, pats, opts, wlock)
2254 2254 names = []
2255 2255 for abs, rel, exact in copied:
2256 2256 if ui.verbose or not exact:
2257 2257 ui.status(_('removing %s\n') % rel)
2258 2258 names.append(abs)
2259 2259 if not opts.get('dry_run'):
2260 2260 repo.remove(names, True, wlock)
2261 2261 return errs
2262 2262
2263 2263 def revert(ui, repo, *pats, **opts):
2264 2264 """revert files or dirs to their states as of some revision
2265 2265
2266 2266 With no revision specified, revert the named files or directories
2267 2267 to the contents they had in the parent of the working directory.
2268 2268 This restores the contents of the affected files to an unmodified
2269 2269 state. If the working directory has two parents, you must
2270 2270 explicitly specify the revision to revert to.
2271 2271
2272 2272 Modified files are saved with a .orig suffix before reverting.
2273 2273 To disable these backups, use --no-backup.
2274 2274
2275 2275 Using the -r option, revert the given files or directories to their
2276 2276 contents as of a specific revision. This can be helpful to "roll
2277 2277 back" some or all of a change that should not have been committed.
2278 2278
2279 2279 Revert modifies the working directory. It does not commit any
2280 2280 changes, or change the parent of the working directory. If you
2281 2281 revert to a revision other than the parent of the working
2282 2282 directory, the reverted files will thus appear modified
2283 2283 afterwards.
2284 2284
2285 2285 If a file has been deleted, it is recreated. If the executable
2286 2286 mode of a file was changed, it is reset.
2287 2287
2288 2288 If names are given, all files matching the names are reverted.
2289 2289
2290 2290 If no arguments are given, no files are reverted.
2291 2291 """
2292 2292
2293 2293 if not pats and not opts['all']:
2294 2294 raise util.Abort(_('no files or directories specified; '
2295 2295 'use --all to revert the whole repo'))
2296 2296
2297 2297 parent, p2 = repo.dirstate.parents()
2298 2298 if opts['rev']:
2299 2299 node = repo.lookup(opts['rev'])
2300 2300 elif p2 != nullid:
2301 2301 raise util.Abort(_('working dir has two parents; '
2302 2302 'you must specify the revision to revert to'))
2303 2303 else:
2304 2304 node = parent
2305 2305 mf = repo.manifest.read(repo.changelog.read(node)[0])
2306 2306 if node == parent:
2307 2307 pmf = mf
2308 2308 else:
2309 2309 pmf = None
2310 2310
2311 2311 wlock = repo.wlock()
2312 2312
2313 2313 # need all matching names in dirstate and manifest of target rev,
2314 2314 # so have to walk both. do not print errors if files exist in one
2315 2315 # but not other.
2316 2316
2317 2317 names = {}
2318 2318 target_only = {}
2319 2319
2320 2320 # walk dirstate.
2321 2321
2322 2322 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts,
2323 2323 badmatch=mf.has_key):
2324 2324 names[abs] = (rel, exact)
2325 2325 if src == 'b':
2326 2326 target_only[abs] = True
2327 2327
2328 2328 # walk target manifest.
2329 2329
2330 2330 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts, node=node,
2331 2331 badmatch=names.has_key):
2332 2332 if abs in names: continue
2333 2333 names[abs] = (rel, exact)
2334 2334 target_only[abs] = True
2335 2335
2336 2336 changes = repo.status(match=names.has_key, wlock=wlock)[:5]
2337 2337 modified, added, removed, deleted, unknown = map(dict.fromkeys, changes)
2338 2338
2339 2339 revert = ([], _('reverting %s\n'))
2340 2340 add = ([], _('adding %s\n'))
2341 2341 remove = ([], _('removing %s\n'))
2342 2342 forget = ([], _('forgetting %s\n'))
2343 2343 undelete = ([], _('undeleting %s\n'))
2344 2344 update = {}
2345 2345
2346 2346 disptable = (
2347 2347 # dispatch table:
2348 2348 # file state
2349 2349 # action if in target manifest
2350 2350 # action if not in target manifest
2351 2351 # make backup if in target manifest
2352 2352 # make backup if not in target manifest
2353 2353 (modified, revert, remove, True, True),
2354 2354 (added, revert, forget, True, False),
2355 2355 (removed, undelete, None, False, False),
2356 2356 (deleted, revert, remove, False, False),
2357 2357 (unknown, add, None, True, False),
2358 2358 (target_only, add, None, False, False),
2359 2359 )
2360 2360
2361 2361 entries = names.items()
2362 2362 entries.sort()
2363 2363
2364 2364 for abs, (rel, exact) in entries:
2365 2365 mfentry = mf.get(abs)
2366 2366 def handle(xlist, dobackup):
2367 2367 xlist[0].append(abs)
2368 2368 update[abs] = 1
2369 2369 if dobackup and not opts['no_backup'] and os.path.exists(rel):
2370 2370 bakname = "%s.orig" % rel
2371 2371 ui.note(_('saving current version of %s as %s\n') %
2372 2372 (rel, bakname))
2373 2373 if not opts.get('dry_run'):
2374 2374 shutil.copyfile(rel, bakname)
2375 2375 shutil.copymode(rel, bakname)
2376 2376 if ui.verbose or not exact:
2377 2377 ui.status(xlist[1] % rel)
2378 2378 for table, hitlist, misslist, backuphit, backupmiss in disptable:
2379 2379 if abs not in table: continue
2380 2380 # file has changed in dirstate
2381 2381 if mfentry:
2382 2382 handle(hitlist, backuphit)
2383 2383 elif misslist is not None:
2384 2384 handle(misslist, backupmiss)
2385 2385 else:
2386 2386 if exact: ui.warn(_('file not managed: %s\n' % rel))
2387 2387 break
2388 2388 else:
2389 2389 # file has not changed in dirstate
2390 2390 if node == parent:
2391 2391 if exact: ui.warn(_('no changes needed to %s\n' % rel))
2392 2392 continue
2393 2393 if pmf is None:
2394 2394 # only need parent manifest in this unlikely case,
2395 2395 # so do not read by default
2396 2396 pmf = repo.manifest.read(repo.changelog.read(parent)[0])
2397 2397 if abs in pmf:
2398 2398 if mfentry:
2399 2399 # if version of file is same in parent and target
2400 2400 # manifests, do nothing
2401 2401 if pmf[abs] != mfentry:
2402 2402 handle(revert, False)
2403 2403 else:
2404 2404 handle(remove, False)
2405 2405
2406 2406 if not opts.get('dry_run'):
2407 2407 repo.dirstate.forget(forget[0])
2408 2408 r = hg.revert(repo, node, update.has_key, wlock)
2409 2409 repo.dirstate.update(add[0], 'a')
2410 2410 repo.dirstate.update(undelete[0], 'n')
2411 2411 repo.dirstate.update(remove[0], 'r')
2412 2412 return r
2413 2413
2414 2414 def rollback(ui, repo):
2415 2415 """roll back the last transaction in this repository
2416 2416
2417 2417 Roll back the last transaction in this repository, restoring the
2418 2418 project to its state prior to the transaction.
2419 2419
2420 2420 Transactions are used to encapsulate the effects of all commands
2421 2421 that create new changesets or propagate existing changesets into a
2422 2422 repository. For example, the following commands are transactional,
2423 2423 and their effects can be rolled back:
2424 2424
2425 2425 commit
2426 2426 import
2427 2427 pull
2428 2428 push (with this repository as destination)
2429 2429 unbundle
2430 2430
2431 2431 This command should be used with care. There is only one level of
2432 2432 rollback, and there is no way to undo a rollback.
2433 2433
2434 2434 This command is not intended for use on public repositories. Once
2435 2435 changes are visible for pull by other users, rolling a transaction
2436 2436 back locally is ineffective (someone else may already have pulled
2437 2437 the changes). Furthermore, a race is possible with readers of the
2438 2438 repository; for example an in-progress pull from the repository
2439 2439 may fail if a rollback is performed.
2440 2440 """
2441 2441 repo.rollback()
2442 2442
2443 2443 def root(ui, repo):
2444 2444 """print the root (top) of the current working dir
2445 2445
2446 2446 Print the root directory of the current repository.
2447 2447 """
2448 2448 ui.write(repo.root + "\n")
2449 2449
2450 2450 def serve(ui, repo, **opts):
2451 2451 """export the repository via HTTP
2452 2452
2453 2453 Start a local HTTP repository browser and pull server.
2454 2454
2455 2455 By default, the server logs accesses to stdout and errors to
2456 2456 stderr. Use the "-A" and "-E" options to log to files.
2457 2457 """
2458 2458
2459 2459 if opts["stdio"]:
2460 2460 if repo is None:
2461 2461 raise hg.RepoError(_('no repo found'))
2462 2462 s = sshserver.sshserver(ui, repo)
2463 2463 s.serve_forever()
2464 2464
2465 2465 optlist = ("name templates style address port ipv6"
2466 2466 " accesslog errorlog webdir_conf")
2467 2467 for o in optlist.split():
2468 2468 if opts[o]:
2469 2469 ui.setconfig("web", o, opts[o])
2470 2470
2471 2471 if repo is None and not ui.config("web", "webdir_conf"):
2472 2472 raise hg.RepoError(_('no repo found'))
2473 2473
2474 2474 if opts['daemon'] and not opts['daemon_pipefds']:
2475 2475 rfd, wfd = os.pipe()
2476 2476 args = sys.argv[:]
2477 2477 args.append('--daemon-pipefds=%d,%d' % (rfd, wfd))
2478 2478 pid = os.spawnvp(os.P_NOWAIT | getattr(os, 'P_DETACH', 0),
2479 2479 args[0], args)
2480 2480 os.close(wfd)
2481 2481 os.read(rfd, 1)
2482 2482 os._exit(0)
2483 2483
2484 2484 try:
2485 2485 httpd = hgweb.server.create_server(ui, repo)
2486 2486 except socket.error, inst:
2487 raise util.Abort(_('cannot start server: ') + inst.args[1])
2487 raise util.Abort(_('cannot start server: %s') % inst.args[1])
2488 2488
2489 2489 if ui.verbose:
2490 2490 addr, port = httpd.socket.getsockname()
2491 2491 if addr == '0.0.0.0':
2492 2492 addr = socket.gethostname()
2493 2493 else:
2494 2494 try:
2495 2495 addr = socket.gethostbyaddr(addr)[0]
2496 2496 except socket.error:
2497 2497 pass
2498 2498 if port != 80:
2499 2499 ui.status(_('listening at http://%s:%d/\n') % (addr, port))
2500 2500 else:
2501 2501 ui.status(_('listening at http://%s/\n') % addr)
2502 2502
2503 2503 if opts['pid_file']:
2504 2504 fp = open(opts['pid_file'], 'w')
2505 2505 fp.write(str(os.getpid()) + '\n')
2506 2506 fp.close()
2507 2507
2508 2508 if opts['daemon_pipefds']:
2509 2509 rfd, wfd = [int(x) for x in opts['daemon_pipefds'].split(',')]
2510 2510 os.close(rfd)
2511 2511 os.write(wfd, 'y')
2512 2512 os.close(wfd)
2513 2513 sys.stdout.flush()
2514 2514 sys.stderr.flush()
2515 2515 fd = os.open(util.nulldev, os.O_RDWR)
2516 2516 if fd != 0: os.dup2(fd, 0)
2517 2517 if fd != 1: os.dup2(fd, 1)
2518 2518 if fd != 2: os.dup2(fd, 2)
2519 2519 if fd not in (0, 1, 2): os.close(fd)
2520 2520
2521 2521 httpd.serve_forever()
2522 2522
2523 2523 def status(ui, repo, *pats, **opts):
2524 2524 """show changed files in the working directory
2525 2525
2526 2526 Show status of files in the repository. If names are given, only
2527 2527 files that match are shown. Files that are clean or ignored, are
2528 2528 not listed unless -c (clean), -i (ignored) or -A is given.
2529 2529
2530 2530 The codes used to show the status of files are:
2531 2531 M = modified
2532 2532 A = added
2533 2533 R = removed
2534 2534 C = clean
2535 2535 ! = deleted, but still tracked
2536 2536 ? = not tracked
2537 2537 I = ignored (not shown by default)
2538 2538 = the previous added file was copied from here
2539 2539 """
2540 2540
2541 2541 all = opts['all']
2542 2542
2543 2543 files, matchfn, anypats = cmdutil.matchpats(repo, pats, opts)
2544 2544 cwd = (pats and repo.getcwd()) or ''
2545 2545 modified, added, removed, deleted, unknown, ignored, clean = [
2546 2546 [util.pathto(cwd, x) for x in n]
2547 2547 for n in repo.status(files=files, match=matchfn,
2548 2548 list_ignored=all or opts['ignored'],
2549 2549 list_clean=all or opts['clean'])]
2550 2550
2551 2551 changetypes = (('modified', 'M', modified),
2552 2552 ('added', 'A', added),
2553 2553 ('removed', 'R', removed),
2554 2554 ('deleted', '!', deleted),
2555 2555 ('unknown', '?', unknown),
2556 2556 ('ignored', 'I', ignored))
2557 2557
2558 2558 explicit_changetypes = changetypes + (('clean', 'C', clean),)
2559 2559
2560 2560 end = opts['print0'] and '\0' or '\n'
2561 2561
2562 2562 for opt, char, changes in ([ct for ct in explicit_changetypes
2563 2563 if all or opts[ct[0]]]
2564 2564 or changetypes):
2565 2565 if opts['no_status']:
2566 2566 format = "%%s%s" % end
2567 2567 else:
2568 2568 format = "%s %%s%s" % (char, end)
2569 2569
2570 2570 for f in changes:
2571 2571 ui.write(format % f)
2572 2572 if ((all or opts.get('copies')) and not opts.get('no_status')
2573 2573 and opt == 'added' and repo.dirstate.copies.has_key(f)):
2574 2574 ui.write(' %s%s' % (repo.dirstate.copies[f], end))
2575 2575
2576 2576 def tag(ui, repo, name, rev_=None, **opts):
2577 2577 """add a tag for the current tip or a given revision
2578 2578
2579 2579 Name a particular revision using <name>.
2580 2580
2581 2581 Tags are used to name particular revisions of the repository and are
2582 2582 very useful to compare different revision, to go back to significant
2583 2583 earlier versions or to mark branch points as releases, etc.
2584 2584
2585 2585 If no revision is given, the parent of the working directory is used.
2586 2586
2587 2587 To facilitate version control, distribution, and merging of tags,
2588 2588 they are stored as a file named ".hgtags" which is managed
2589 2589 similarly to other project files and can be hand-edited if
2590 2590 necessary. The file '.hg/localtags' is used for local tags (not
2591 2591 shared among repositories).
2592 2592 """
2593 2593 if name in ['tip', '.']:
2594 2594 raise util.Abort(_("the name '%s' is reserved") % name)
2595 2595 if rev_ is not None:
2596 2596 ui.warn(_("use of 'hg tag NAME [REV]' is deprecated, "
2597 2597 "please use 'hg tag [-r REV] NAME' instead\n"))
2598 2598 if opts['rev']:
2599 2599 raise util.Abort(_("use only one form to specify the revision"))
2600 2600 if opts['rev']:
2601 2601 rev_ = opts['rev']
2602 2602 if rev_:
2603 2603 r = repo.lookup(rev_)
2604 2604 else:
2605 2605 p1, p2 = repo.dirstate.parents()
2606 2606 if p1 == nullid:
2607 2607 raise util.Abort(_('no revision to tag'))
2608 2608 if p2 != nullid:
2609 2609 raise util.Abort(_('outstanding uncommitted merges'))
2610 2610 r = p1
2611 2611
2612 2612 message = opts['message']
2613 2613 if not message:
2614 2614 message = _('Added tag %s for changeset %s') % (name, short(r))
2615 2615
2616 2616 repo.tag(name, r, message, opts['local'], opts['user'], opts['date'])
2617 2617
2618 2618 def tags(ui, repo):
2619 2619 """list repository tags
2620 2620
2621 2621 List the repository tags.
2622 2622
2623 2623 This lists both regular and local tags.
2624 2624 """
2625 2625
2626 2626 l = repo.tagslist()
2627 2627 l.reverse()
2628 2628 hexfunc = ui.debugflag and hex or short
2629 2629 for t, n in l:
2630 2630 try:
2631 2631 r = "%5d:%s" % (repo.changelog.rev(n), hexfunc(n))
2632 2632 except KeyError:
2633 2633 r = " ?:?"
2634 2634 if ui.quiet:
2635 2635 ui.write("%s\n" % t)
2636 2636 else:
2637 2637 ui.write("%-30s %s\n" % (t, r))
2638 2638
2639 2639 def tip(ui, repo, **opts):
2640 2640 """show the tip revision
2641 2641
2642 2642 Show the tip revision.
2643 2643 """
2644 2644 n = repo.changelog.tip()
2645 2645 br = None
2646 2646 if opts['branches']:
2647 2647 br = repo.branchlookup([n])
2648 2648 show_changeset(ui, repo, opts).show(changenode=n, brinfo=br)
2649 2649 if opts['patch']:
2650 2650 patch.diff(repo, repo.changelog.parents(n)[0], n)
2651 2651
2652 2652 def unbundle(ui, repo, fname, **opts):
2653 2653 """apply a changegroup file
2654 2654
2655 2655 Apply a compressed changegroup file generated by the bundle
2656 2656 command.
2657 2657 """
2658 2658 f = urllib.urlopen(fname)
2659 2659
2660 2660 header = f.read(6)
2661 2661 if not header.startswith("HG"):
2662 2662 raise util.Abort(_("%s: not a Mercurial bundle file") % fname)
2663 2663 elif not header.startswith("HG10"):
2664 2664 raise util.Abort(_("%s: unknown bundle version") % fname)
2665 2665 elif header == "HG10BZ":
2666 2666 def generator(f):
2667 2667 zd = bz2.BZ2Decompressor()
2668 2668 zd.decompress("BZ")
2669 2669 for chunk in f:
2670 2670 yield zd.decompress(chunk)
2671 2671 elif header == "HG10UN":
2672 2672 def generator(f):
2673 2673 for chunk in f:
2674 2674 yield chunk
2675 2675 else:
2676 2676 raise util.Abort(_("%s: unknown bundle compression type")
2677 2677 % fname)
2678 2678 gen = generator(util.filechunkiter(f, 4096))
2679 2679 modheads = repo.addchangegroup(util.chunkbuffer(gen), 'unbundle',
2680 2680 'bundle:' + fname)
2681 2681 return postincoming(ui, repo, modheads, opts['update'])
2682 2682
2683 2683 def undo(ui, repo):
2684 2684 """undo the last commit or pull (DEPRECATED)
2685 2685
2686 2686 (DEPRECATED)
2687 2687 This command is now deprecated and will be removed in a future
2688 2688 release. Please use the rollback command instead. For usage
2689 2689 instructions, see the rollback command.
2690 2690 """
2691 2691 ui.warn(_('(the undo command is deprecated; use rollback instead)\n'))
2692 2692 repo.rollback()
2693 2693
2694 2694 def update(ui, repo, node=None, merge=False, clean=False, force=None,
2695 2695 branch=None):
2696 2696 """update or merge working directory
2697 2697
2698 2698 Update the working directory to the specified revision.
2699 2699
2700 2700 If there are no outstanding changes in the working directory and
2701 2701 there is a linear relationship between the current version and the
2702 2702 requested version, the result is the requested version.
2703 2703
2704 2704 To merge the working directory with another revision, use the
2705 2705 merge command.
2706 2706
2707 2707 By default, update will refuse to run if doing so would require
2708 2708 merging or discarding local changes.
2709 2709 """
2710 2710 node = _lookup(repo, node, branch)
2711 2711 if merge:
2712 2712 ui.warn(_('(the -m/--merge option is deprecated; '
2713 2713 'use the merge command instead)\n'))
2714 2714 return hg.merge(repo, node, force=force)
2715 2715 elif clean:
2716 2716 return hg.clean(repo, node)
2717 2717 else:
2718 2718 return hg.update(repo, node)
2719 2719
2720 2720 def _lookup(repo, node, branch=None):
2721 2721 if branch:
2722 2722 br = repo.branchlookup(branch=branch)
2723 2723 found = []
2724 2724 for x in br:
2725 2725 if branch in br[x]:
2726 2726 found.append(x)
2727 2727 if len(found) > 1:
2728 2728 repo.ui.warn(_("Found multiple heads for %s\n") % branch)
2729 2729 for x in found:
2730 2730 show_changeset(ui, repo, {}).show(changenode=x, brinfo=br)
2731 2731 raise util.Abort("")
2732 2732 if len(found) == 1:
2733 2733 node = found[0]
2734 2734 repo.ui.warn(_("Using head %s for branch %s\n")
2735 2735 % (short(node), branch))
2736 2736 else:
2737 raise util.Abort(_("branch %s not found\n") % (branch))
2737 raise util.Abort(_("branch %s not found") % branch)
2738 2738 else:
2739 2739 node = node and repo.lookup(node) or repo.changelog.tip()
2740 2740 return node
2741 2741
2742 2742 def verify(ui, repo):
2743 2743 """verify the integrity of the repository
2744 2744
2745 2745 Verify the integrity of the current repository.
2746 2746
2747 2747 This will perform an extensive check of the repository's
2748 2748 integrity, validating the hashes and checksums of each entry in
2749 2749 the changelog, manifest, and tracked files, as well as the
2750 2750 integrity of their crosslinks and indices.
2751 2751 """
2752 2752 return hg.verify(repo)
2753 2753
2754 2754 # Command options and aliases are listed here, alphabetically
2755 2755
2756 2756 table = {
2757 2757 "^add":
2758 2758 (add,
2759 2759 [('I', 'include', [], _('include names matching the given patterns')),
2760 2760 ('X', 'exclude', [], _('exclude names matching the given patterns')),
2761 2761 ('n', 'dry-run', None, _('do not perform actions, just print output'))],
2762 2762 _('hg add [OPTION]... [FILE]...')),
2763 2763 "addremove":
2764 2764 (addremove,
2765 2765 [('I', 'include', [], _('include names matching the given patterns')),
2766 2766 ('X', 'exclude', [], _('exclude names matching the given patterns')),
2767 2767 ('n', 'dry-run', None,
2768 2768 _('do not perform actions, just print output')),
2769 2769 ('s', 'similarity', '',
2770 2770 _('guess renamed files by similarity (0<=s<=1)'))],
2771 2771 _('hg addremove [OPTION]... [FILE]...')),
2772 2772 "^annotate":
2773 2773 (annotate,
2774 2774 [('r', 'rev', '', _('annotate the specified revision')),
2775 2775 ('a', 'text', None, _('treat all files as text')),
2776 2776 ('u', 'user', None, _('list the author')),
2777 2777 ('d', 'date', None, _('list the date')),
2778 2778 ('n', 'number', None, _('list the revision number (default)')),
2779 2779 ('c', 'changeset', None, _('list the changeset')),
2780 2780 ('I', 'include', [], _('include names matching the given patterns')),
2781 2781 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2782 2782 _('hg annotate [-r REV] [-a] [-u] [-d] [-n] [-c] FILE...')),
2783 2783 "archive":
2784 2784 (archive,
2785 2785 [('', 'no-decode', None, _('do not pass files through decoders')),
2786 2786 ('p', 'prefix', '', _('directory prefix for files in archive')),
2787 2787 ('r', 'rev', '', _('revision to distribute')),
2788 2788 ('t', 'type', '', _('type of distribution to create')),
2789 2789 ('I', 'include', [], _('include names matching the given patterns')),
2790 2790 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2791 2791 _('hg archive [OPTION]... DEST')),
2792 2792 "backout":
2793 2793 (backout,
2794 2794 [('', 'merge', None,
2795 2795 _('merge with old dirstate parent after backout')),
2796 2796 ('m', 'message', '', _('use <text> as commit message')),
2797 2797 ('l', 'logfile', '', _('read commit message from <file>')),
2798 2798 ('d', 'date', '', _('record datecode as commit date')),
2799 2799 ('', 'parent', '', _('parent to choose when backing out merge')),
2800 2800 ('u', 'user', '', _('record user as committer')),
2801 2801 ('I', 'include', [], _('include names matching the given patterns')),
2802 2802 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2803 2803 _('hg backout [OPTION]... REV')),
2804 2804 "bundle":
2805 2805 (bundle,
2806 2806 [('f', 'force', None,
2807 2807 _('run even when remote repository is unrelated'))],
2808 2808 _('hg bundle FILE DEST')),
2809 2809 "cat":
2810 2810 (cat,
2811 2811 [('o', 'output', '', _('print output to file with formatted name')),
2812 2812 ('r', 'rev', '', _('print the given revision')),
2813 2813 ('I', 'include', [], _('include names matching the given patterns')),
2814 2814 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2815 2815 _('hg cat [OPTION]... FILE...')),
2816 2816 "^clone":
2817 2817 (clone,
2818 2818 [('U', 'noupdate', None, _('do not update the new working directory')),
2819 2819 ('r', 'rev', [],
2820 2820 _('a changeset you would like to have after cloning')),
2821 2821 ('', 'pull', None, _('use pull protocol to copy metadata')),
2822 2822 ('', 'uncompressed', None,
2823 2823 _('use uncompressed transfer (fast over LAN)')),
2824 2824 ('e', 'ssh', '', _('specify ssh command to use')),
2825 2825 ('', 'remotecmd', '',
2826 2826 _('specify hg command to run on the remote side'))],
2827 2827 _('hg clone [OPTION]... SOURCE [DEST]')),
2828 2828 "^commit|ci":
2829 2829 (commit,
2830 2830 [('A', 'addremove', None,
2831 2831 _('mark new/missing files as added/removed before committing')),
2832 2832 ('m', 'message', '', _('use <text> as commit message')),
2833 2833 ('l', 'logfile', '', _('read the commit message from <file>')),
2834 2834 ('d', 'date', '', _('record datecode as commit date')),
2835 2835 ('u', 'user', '', _('record user as commiter')),
2836 2836 ('I', 'include', [], _('include names matching the given patterns')),
2837 2837 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2838 2838 _('hg commit [OPTION]... [FILE]...')),
2839 2839 "copy|cp":
2840 2840 (copy,
2841 2841 [('A', 'after', None, _('record a copy that has already occurred')),
2842 2842 ('f', 'force', None,
2843 2843 _('forcibly copy over an existing managed file')),
2844 2844 ('I', 'include', [], _('include names matching the given patterns')),
2845 2845 ('X', 'exclude', [], _('exclude names matching the given patterns')),
2846 2846 ('n', 'dry-run', None, _('do not perform actions, just print output'))],
2847 2847 _('hg copy [OPTION]... [SOURCE]... DEST')),
2848 2848 "debugancestor": (debugancestor, [], _('debugancestor INDEX REV1 REV2')),
2849 2849 "debugcomplete":
2850 2850 (debugcomplete,
2851 2851 [('o', 'options', None, _('show the command options'))],
2852 2852 _('debugcomplete [-o] CMD')),
2853 2853 "debugrebuildstate":
2854 2854 (debugrebuildstate,
2855 2855 [('r', 'rev', '', _('revision to rebuild to'))],
2856 2856 _('debugrebuildstate [-r REV] [REV]')),
2857 2857 "debugcheckstate": (debugcheckstate, [], _('debugcheckstate')),
2858 2858 "debugconfig": (debugconfig, [], _('debugconfig [NAME]...')),
2859 2859 "debugsetparents": (debugsetparents, [], _('debugsetparents REV1 [REV2]')),
2860 2860 "debugstate": (debugstate, [], _('debugstate')),
2861 2861 "debugdata": (debugdata, [], _('debugdata FILE REV')),
2862 2862 "debugindex": (debugindex, [], _('debugindex FILE')),
2863 2863 "debugindexdot": (debugindexdot, [], _('debugindexdot FILE')),
2864 2864 "debugrename": (debugrename, [], _('debugrename FILE [REV]')),
2865 2865 "debugwalk":
2866 2866 (debugwalk,
2867 2867 [('I', 'include', [], _('include names matching the given patterns')),
2868 2868 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2869 2869 _('debugwalk [OPTION]... [FILE]...')),
2870 2870 "^diff":
2871 2871 (diff,
2872 2872 [('r', 'rev', [], _('revision')),
2873 2873 ('a', 'text', None, _('treat all files as text')),
2874 2874 ('p', 'show-function', None,
2875 2875 _('show which function each change is in')),
2876 2876 ('g', 'git', None, _('use git extended diff format')),
2877 2877 ('w', 'ignore-all-space', None,
2878 2878 _('ignore white space when comparing lines')),
2879 2879 ('b', 'ignore-space-change', None,
2880 2880 _('ignore changes in the amount of white space')),
2881 2881 ('B', 'ignore-blank-lines', None,
2882 2882 _('ignore changes whose lines are all blank')),
2883 2883 ('I', 'include', [], _('include names matching the given patterns')),
2884 2884 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2885 2885 _('hg diff [-a] [-I] [-X] [-r REV1 [-r REV2]] [FILE]...')),
2886 2886 "^export":
2887 2887 (export,
2888 2888 [('o', 'output', '', _('print output to file with formatted name')),
2889 2889 ('a', 'text', None, _('treat all files as text')),
2890 2890 ('g', 'git', None, _('use git extended diff format')),
2891 2891 ('', 'switch-parent', None, _('diff against the second parent'))],
2892 2892 _('hg export [-a] [-o OUTFILESPEC] REV...')),
2893 2893 "debugforget|forget":
2894 2894 (forget,
2895 2895 [('I', 'include', [], _('include names matching the given patterns')),
2896 2896 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2897 2897 _('hg forget [OPTION]... FILE...')),
2898 2898 "grep":
2899 2899 (grep,
2900 2900 [('0', 'print0', None, _('end fields with NUL')),
2901 2901 ('', 'all', None, _('print all revisions that match')),
2902 2902 ('f', 'follow', None,
2903 2903 _('follow changeset history, or file history across copies and renames')),
2904 2904 ('i', 'ignore-case', None, _('ignore case when matching')),
2905 2905 ('l', 'files-with-matches', None,
2906 2906 _('print only filenames and revs that match')),
2907 2907 ('n', 'line-number', None, _('print matching line numbers')),
2908 2908 ('r', 'rev', [], _('search in given revision range')),
2909 2909 ('u', 'user', None, _('print user who committed change')),
2910 2910 ('I', 'include', [], _('include names matching the given patterns')),
2911 2911 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2912 2912 _('hg grep [OPTION]... PATTERN [FILE]...')),
2913 2913 "heads":
2914 2914 (heads,
2915 2915 [('b', 'branches', None, _('show branches')),
2916 2916 ('', 'style', '', _('display using template map file')),
2917 2917 ('r', 'rev', '', _('show only heads which are descendants of rev')),
2918 2918 ('', 'template', '', _('display with template'))],
2919 2919 _('hg heads [-b] [-r <rev>]')),
2920 2920 "help": (help_, [], _('hg help [COMMAND]')),
2921 2921 "identify|id": (identify, [], _('hg identify')),
2922 2922 "import|patch":
2923 2923 (import_,
2924 2924 [('p', 'strip', 1,
2925 2925 _('directory strip option for patch. This has the same\n'
2926 2926 'meaning as the corresponding patch option')),
2927 2927 ('m', 'message', '', _('use <text> as commit message')),
2928 2928 ('b', 'base', '', _('base path')),
2929 2929 ('f', 'force', None,
2930 2930 _('skip check for outstanding uncommitted changes'))],
2931 2931 _('hg import [-p NUM] [-b BASE] [-m MESSAGE] [-f] PATCH...')),
2932 2932 "incoming|in": (incoming,
2933 2933 [('M', 'no-merges', None, _('do not show merges')),
2934 2934 ('f', 'force', None,
2935 2935 _('run even when remote repository is unrelated')),
2936 2936 ('', 'style', '', _('display using template map file')),
2937 2937 ('n', 'newest-first', None, _('show newest record first')),
2938 2938 ('', 'bundle', '', _('file to store the bundles into')),
2939 2939 ('p', 'patch', None, _('show patch')),
2940 2940 ('r', 'rev', [], _('a specific revision up to which you would like to pull')),
2941 2941 ('', 'template', '', _('display with template')),
2942 2942 ('e', 'ssh', '', _('specify ssh command to use')),
2943 2943 ('', 'remotecmd', '',
2944 2944 _('specify hg command to run on the remote side'))],
2945 2945 _('hg incoming [-p] [-n] [-M] [-r REV]...'
2946 2946 ' [--bundle FILENAME] [SOURCE]')),
2947 2947 "^init":
2948 2948 (init,
2949 2949 [('e', 'ssh', '', _('specify ssh command to use')),
2950 2950 ('', 'remotecmd', '',
2951 2951 _('specify hg command to run on the remote side'))],
2952 2952 _('hg init [-e FILE] [--remotecmd FILE] [DEST]')),
2953 2953 "locate":
2954 2954 (locate,
2955 2955 [('r', 'rev', '', _('search the repository as it stood at rev')),
2956 2956 ('0', 'print0', None,
2957 2957 _('end filenames with NUL, for use with xargs')),
2958 2958 ('f', 'fullpath', None,
2959 2959 _('print complete paths from the filesystem root')),
2960 2960 ('I', 'include', [], _('include names matching the given patterns')),
2961 2961 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2962 2962 _('hg locate [OPTION]... [PATTERN]...')),
2963 2963 "^log|history":
2964 2964 (log,
2965 2965 [('b', 'branches', None, _('show branches')),
2966 2966 ('f', 'follow', None,
2967 2967 _('follow changeset history, or file history across copies and renames')),
2968 2968 ('', 'follow-first', None,
2969 2969 _('only follow the first parent of merge changesets')),
2970 2970 ('k', 'keyword', [], _('search for a keyword')),
2971 2971 ('l', 'limit', '', _('limit number of changes displayed')),
2972 2972 ('r', 'rev', [], _('show the specified revision or range')),
2973 2973 ('M', 'no-merges', None, _('do not show merges')),
2974 2974 ('', 'style', '', _('display using template map file')),
2975 2975 ('m', 'only-merges', None, _('show only merges')),
2976 2976 ('p', 'patch', None, _('show patch')),
2977 2977 ('P', 'prune', [], _('do not display revision or any of its ancestors')),
2978 2978 ('', 'template', '', _('display with template')),
2979 2979 ('I', 'include', [], _('include names matching the given patterns')),
2980 2980 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
2981 2981 _('hg log [OPTION]... [FILE]')),
2982 2982 "manifest": (manifest, [], _('hg manifest [REV]')),
2983 2983 "merge":
2984 2984 (merge,
2985 2985 [('b', 'branch', '', _('merge with head of a specific branch')),
2986 2986 ('f', 'force', None, _('force a merge with outstanding changes'))],
2987 2987 _('hg merge [-b TAG] [-f] [REV]')),
2988 2988 "outgoing|out": (outgoing,
2989 2989 [('M', 'no-merges', None, _('do not show merges')),
2990 2990 ('f', 'force', None,
2991 2991 _('run even when remote repository is unrelated')),
2992 2992 ('p', 'patch', None, _('show patch')),
2993 2993 ('', 'style', '', _('display using template map file')),
2994 2994 ('r', 'rev', [], _('a specific revision you would like to push')),
2995 2995 ('n', 'newest-first', None, _('show newest record first')),
2996 2996 ('', 'template', '', _('display with template')),
2997 2997 ('e', 'ssh', '', _('specify ssh command to use')),
2998 2998 ('', 'remotecmd', '',
2999 2999 _('specify hg command to run on the remote side'))],
3000 3000 _('hg outgoing [-M] [-p] [-n] [-r REV]... [DEST]')),
3001 3001 "^parents":
3002 3002 (parents,
3003 3003 [('b', 'branches', None, _('show branches')),
3004 3004 ('r', 'rev', '', _('show parents from the specified rev')),
3005 3005 ('', 'style', '', _('display using template map file')),
3006 3006 ('', 'template', '', _('display with template'))],
3007 3007 _('hg parents [-b] [-r REV] [FILE]')),
3008 3008 "paths": (paths, [], _('hg paths [NAME]')),
3009 3009 "^pull":
3010 3010 (pull,
3011 3011 [('u', 'update', None,
3012 3012 _('update the working directory to tip after pull')),
3013 3013 ('e', 'ssh', '', _('specify ssh command to use')),
3014 3014 ('f', 'force', None,
3015 3015 _('run even when remote repository is unrelated')),
3016 3016 ('r', 'rev', [], _('a specific revision up to which you would like to pull')),
3017 3017 ('', 'remotecmd', '',
3018 3018 _('specify hg command to run on the remote side'))],
3019 3019 _('hg pull [-u] [-r REV]... [-e FILE] [--remotecmd FILE] [SOURCE]')),
3020 3020 "^push":
3021 3021 (push,
3022 3022 [('f', 'force', None, _('force push')),
3023 3023 ('e', 'ssh', '', _('specify ssh command to use')),
3024 3024 ('r', 'rev', [], _('a specific revision you would like to push')),
3025 3025 ('', 'remotecmd', '',
3026 3026 _('specify hg command to run on the remote side'))],
3027 3027 _('hg push [-f] [-r REV]... [-e FILE] [--remotecmd FILE] [DEST]')),
3028 3028 "debugrawcommit|rawcommit":
3029 3029 (rawcommit,
3030 3030 [('p', 'parent', [], _('parent')),
3031 3031 ('d', 'date', '', _('date code')),
3032 3032 ('u', 'user', '', _('user')),
3033 3033 ('F', 'files', '', _('file list')),
3034 3034 ('m', 'message', '', _('commit message')),
3035 3035 ('l', 'logfile', '', _('commit message file'))],
3036 3036 _('hg debugrawcommit [OPTION]... [FILE]...')),
3037 3037 "recover": (recover, [], _('hg recover')),
3038 3038 "^remove|rm":
3039 3039 (remove,
3040 3040 [('A', 'after', None, _('record remove that has already occurred')),
3041 3041 ('f', 'force', None, _('remove file even if modified')),
3042 3042 ('I', 'include', [], _('include names matching the given patterns')),
3043 3043 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
3044 3044 _('hg remove [OPTION]... FILE...')),
3045 3045 "rename|mv":
3046 3046 (rename,
3047 3047 [('A', 'after', None, _('record a rename that has already occurred')),
3048 3048 ('f', 'force', None,
3049 3049 _('forcibly copy over an existing managed file')),
3050 3050 ('I', 'include', [], _('include names matching the given patterns')),
3051 3051 ('X', 'exclude', [], _('exclude names matching the given patterns')),
3052 3052 ('n', 'dry-run', None, _('do not perform actions, just print output'))],
3053 3053 _('hg rename [OPTION]... SOURCE... DEST')),
3054 3054 "^revert":
3055 3055 (revert,
3056 3056 [('a', 'all', None, _('revert all changes when no arguments given')),
3057 3057 ('r', 'rev', '', _('revision to revert to')),
3058 3058 ('', 'no-backup', None, _('do not save backup copies of files')),
3059 3059 ('I', 'include', [], _('include names matching given patterns')),
3060 3060 ('X', 'exclude', [], _('exclude names matching given patterns')),
3061 3061 ('n', 'dry-run', None, _('do not perform actions, just print output'))],
3062 3062 _('hg revert [-r REV] [NAME]...')),
3063 3063 "rollback": (rollback, [], _('hg rollback')),
3064 3064 "root": (root, [], _('hg root')),
3065 3065 "^serve":
3066 3066 (serve,
3067 3067 [('A', 'accesslog', '', _('name of access log file to write to')),
3068 3068 ('d', 'daemon', None, _('run server in background')),
3069 3069 ('', 'daemon-pipefds', '', _('used internally by daemon mode')),
3070 3070 ('E', 'errorlog', '', _('name of error log file to write to')),
3071 3071 ('p', 'port', 0, _('port to use (default: 8000)')),
3072 3072 ('a', 'address', '', _('address to use')),
3073 3073 ('n', 'name', '',
3074 3074 _('name to show in web pages (default: working dir)')),
3075 3075 ('', 'webdir-conf', '', _('name of the webdir config file'
3076 3076 ' (serve more than one repo)')),
3077 3077 ('', 'pid-file', '', _('name of file to write process ID to')),
3078 3078 ('', 'stdio', None, _('for remote clients')),
3079 3079 ('t', 'templates', '', _('web templates to use')),
3080 3080 ('', 'style', '', _('template style to use')),
3081 3081 ('6', 'ipv6', None, _('use IPv6 in addition to IPv4'))],
3082 3082 _('hg serve [OPTION]...')),
3083 3083 "^status|st":
3084 3084 (status,
3085 3085 [('A', 'all', None, _('show status of all files')),
3086 3086 ('m', 'modified', None, _('show only modified files')),
3087 3087 ('a', 'added', None, _('show only added files')),
3088 3088 ('r', 'removed', None, _('show only removed files')),
3089 3089 ('d', 'deleted', None, _('show only deleted (but tracked) files')),
3090 3090 ('c', 'clean', None, _('show only files without changes')),
3091 3091 ('u', 'unknown', None, _('show only unknown (not tracked) files')),
3092 3092 ('i', 'ignored', None, _('show ignored files')),
3093 3093 ('n', 'no-status', None, _('hide status prefix')),
3094 3094 ('C', 'copies', None, _('show source of copied files')),
3095 3095 ('0', 'print0', None,
3096 3096 _('end filenames with NUL, for use with xargs')),
3097 3097 ('I', 'include', [], _('include names matching the given patterns')),
3098 3098 ('X', 'exclude', [], _('exclude names matching the given patterns'))],
3099 3099 _('hg status [OPTION]... [FILE]...')),
3100 3100 "tag":
3101 3101 (tag,
3102 3102 [('l', 'local', None, _('make the tag local')),
3103 3103 ('m', 'message', '', _('message for tag commit log entry')),
3104 3104 ('d', 'date', '', _('record datecode as commit date')),
3105 3105 ('u', 'user', '', _('record user as commiter')),
3106 3106 ('r', 'rev', '', _('revision to tag'))],
3107 3107 _('hg tag [-l] [-m TEXT] [-d DATE] [-u USER] [-r REV] NAME')),
3108 3108 "tags": (tags, [], _('hg tags')),
3109 3109 "tip":
3110 3110 (tip,
3111 3111 [('b', 'branches', None, _('show branches')),
3112 3112 ('', 'style', '', _('display using template map file')),
3113 3113 ('p', 'patch', None, _('show patch')),
3114 3114 ('', 'template', '', _('display with template'))],
3115 3115 _('hg tip [-b] [-p]')),
3116 3116 "unbundle":
3117 3117 (unbundle,
3118 3118 [('u', 'update', None,
3119 3119 _('update the working directory to tip after unbundle'))],
3120 3120 _('hg unbundle [-u] FILE')),
3121 3121 "debugundo|undo": (undo, [], _('hg undo')),
3122 3122 "^update|up|checkout|co":
3123 3123 (update,
3124 3124 [('b', 'branch', '', _('checkout the head of a specific branch')),
3125 3125 ('m', 'merge', None, _('allow merging of branches (DEPRECATED)')),
3126 3126 ('C', 'clean', None, _('overwrite locally modified files')),
3127 3127 ('f', 'force', None, _('force a merge with outstanding changes'))],
3128 3128 _('hg update [-b TAG] [-m] [-C] [-f] [REV]')),
3129 3129 "verify": (verify, [], _('hg verify')),
3130 3130 "version": (show_version, [], _('hg version')),
3131 3131 }
3132 3132
3133 3133 globalopts = [
3134 3134 ('R', 'repository', '',
3135 3135 _('repository root directory or symbolic path name')),
3136 3136 ('', 'cwd', '', _('change working directory')),
3137 3137 ('y', 'noninteractive', None,
3138 3138 _('do not prompt, assume \'yes\' for any required answers')),
3139 3139 ('q', 'quiet', None, _('suppress output')),
3140 3140 ('v', 'verbose', None, _('enable additional output')),
3141 3141 ('', 'config', [], _('set/override config option')),
3142 3142 ('', 'debug', None, _('enable debugging output')),
3143 3143 ('', 'debugger', None, _('start debugger')),
3144 3144 ('', 'lsprof', None, _('print improved command execution profile')),
3145 3145 ('', 'traceback', None, _('print traceback on exception')),
3146 3146 ('', 'time', None, _('time how long the command takes')),
3147 3147 ('', 'profile', None, _('print command execution profile')),
3148 3148 ('', 'version', None, _('output version information and exit')),
3149 3149 ('h', 'help', None, _('display help and exit')),
3150 3150 ]
3151 3151
3152 3152 norepo = ("clone init version help debugancestor debugcomplete debugdata"
3153 3153 " debugindex debugindexdot")
3154 3154 optionalrepo = ("paths serve debugconfig")
3155 3155
3156 3156 def findpossible(ui, cmd):
3157 3157 """
3158 3158 Return cmd -> (aliases, command table entry)
3159 3159 for each matching command.
3160 3160 Return debug commands (or their aliases) only if no normal command matches.
3161 3161 """
3162 3162 choice = {}
3163 3163 debugchoice = {}
3164 3164 for e in table.keys():
3165 3165 aliases = e.lstrip("^").split("|")
3166 3166 found = None
3167 3167 if cmd in aliases:
3168 3168 found = cmd
3169 3169 elif not ui.config("ui", "strict"):
3170 3170 for a in aliases:
3171 3171 if a.startswith(cmd):
3172 3172 found = a
3173 3173 break
3174 3174 if found is not None:
3175 3175 if aliases[0].startswith("debug"):
3176 3176 debugchoice[found] = (aliases, table[e])
3177 3177 else:
3178 3178 choice[found] = (aliases, table[e])
3179 3179
3180 3180 if not choice and debugchoice:
3181 3181 choice = debugchoice
3182 3182
3183 3183 return choice
3184 3184
3185 3185 def findcmd(ui, cmd):
3186 3186 """Return (aliases, command table entry) for command string."""
3187 3187 choice = findpossible(ui, cmd)
3188 3188
3189 3189 if choice.has_key(cmd):
3190 3190 return choice[cmd]
3191 3191
3192 3192 if len(choice) > 1:
3193 3193 clist = choice.keys()
3194 3194 clist.sort()
3195 3195 raise AmbiguousCommand(cmd, clist)
3196 3196
3197 3197 if choice:
3198 3198 return choice.values()[0]
3199 3199
3200 3200 raise UnknownCommand(cmd)
3201 3201
3202 3202 def catchterm(*args):
3203 3203 raise util.SignalInterrupt
3204 3204
3205 3205 def run():
3206 3206 sys.exit(dispatch(sys.argv[1:]))
3207 3207
3208 3208 class ParseError(Exception):
3209 3209 """Exception raised on errors in parsing the command line."""
3210 3210
3211 3211 def parse(ui, args):
3212 3212 options = {}
3213 3213 cmdoptions = {}
3214 3214
3215 3215 try:
3216 3216 args = fancyopts.fancyopts(args, globalopts, options)
3217 3217 except fancyopts.getopt.GetoptError, inst:
3218 3218 raise ParseError(None, inst)
3219 3219
3220 3220 if args:
3221 3221 cmd, args = args[0], args[1:]
3222 3222 aliases, i = findcmd(ui, cmd)
3223 3223 cmd = aliases[0]
3224 3224 defaults = ui.config("defaults", cmd)
3225 3225 if defaults:
3226 3226 args = shlex.split(defaults) + args
3227 3227 c = list(i[1])
3228 3228 else:
3229 3229 cmd = None
3230 3230 c = []
3231 3231
3232 3232 # combine global options into local
3233 3233 for o in globalopts:
3234 3234 c.append((o[0], o[1], options[o[1]], o[3]))
3235 3235
3236 3236 try:
3237 3237 args = fancyopts.fancyopts(args, c, cmdoptions)
3238 3238 except fancyopts.getopt.GetoptError, inst:
3239 3239 raise ParseError(cmd, inst)
3240 3240
3241 3241 # separate global options back out
3242 3242 for o in globalopts:
3243 3243 n = o[1]
3244 3244 options[n] = cmdoptions[n]
3245 3245 del cmdoptions[n]
3246 3246
3247 3247 return (cmd, cmd and i[0] or None, args, options, cmdoptions)
3248 3248
3249 3249 external = {}
3250 3250
3251 3251 def findext(name):
3252 3252 '''return module with given extension name'''
3253 3253 try:
3254 3254 return sys.modules[external[name]]
3255 3255 except KeyError:
3256 3256 for k, v in external.iteritems():
3257 3257 if k.endswith('.' + name) or k.endswith('/' + name) or v == name:
3258 3258 return sys.modules[v]
3259 3259 raise KeyError(name)
3260 3260
3261 3261 def load_extensions(ui):
3262 3262 added = []
3263 3263 for ext_name, load_from_name in ui.extensions():
3264 3264 if ext_name in external:
3265 3265 continue
3266 3266 try:
3267 3267 if load_from_name:
3268 3268 # the module will be loaded in sys.modules
3269 3269 # choose an unique name so that it doesn't
3270 3270 # conflicts with other modules
3271 3271 module_name = "hgext_%s" % ext_name.replace('.', '_')
3272 3272 mod = imp.load_source(module_name, load_from_name)
3273 3273 else:
3274 3274 def importh(name):
3275 3275 mod = __import__(name)
3276 3276 components = name.split('.')
3277 3277 for comp in components[1:]:
3278 3278 mod = getattr(mod, comp)
3279 3279 return mod
3280 3280 try:
3281 3281 mod = importh("hgext.%s" % ext_name)
3282 3282 except ImportError:
3283 3283 mod = importh(ext_name)
3284 3284 external[ext_name] = mod.__name__
3285 3285 added.append((mod, ext_name))
3286 3286 except (util.SignalInterrupt, KeyboardInterrupt):
3287 3287 raise
3288 3288 except Exception, inst:
3289 3289 ui.warn(_("*** failed to import extension %s: %s\n") %
3290 3290 (ext_name, inst))
3291 3291 if ui.print_exc():
3292 3292 return 1
3293 3293
3294 3294 for mod, name in added:
3295 3295 uisetup = getattr(mod, 'uisetup', None)
3296 3296 if uisetup:
3297 3297 uisetup(ui)
3298 3298 cmdtable = getattr(mod, 'cmdtable', {})
3299 3299 for t in cmdtable:
3300 3300 if t in table:
3301 3301 ui.warn(_("module %s overrides %s\n") % (name, t))
3302 3302 table.update(cmdtable)
3303 3303
3304 3304 def dispatch(args):
3305 3305 for name in 'SIGBREAK', 'SIGHUP', 'SIGTERM':
3306 3306 num = getattr(signal, name, None)
3307 3307 if num: signal.signal(num, catchterm)
3308 3308
3309 3309 try:
3310 3310 u = ui.ui(traceback='--traceback' in sys.argv[1:])
3311 3311 except util.Abort, inst:
3312 3312 sys.stderr.write(_("abort: %s\n") % inst)
3313 3313 return -1
3314 3314
3315 3315 load_extensions(u)
3316 3316 u.addreadhook(load_extensions)
3317 3317
3318 3318 try:
3319 3319 cmd, func, args, options, cmdoptions = parse(u, args)
3320 3320 if options["time"]:
3321 3321 def get_times():
3322 3322 t = os.times()
3323 3323 if t[4] == 0.0: # Windows leaves this as zero, so use time.clock()
3324 3324 t = (t[0], t[1], t[2], t[3], time.clock())
3325 3325 return t
3326 3326 s = get_times()
3327 3327 def print_time():
3328 3328 t = get_times()
3329 3329 u.warn(_("Time: real %.3f secs (user %.3f+%.3f sys %.3f+%.3f)\n") %
3330 3330 (t[4]-s[4], t[0]-s[0], t[2]-s[2], t[1]-s[1], t[3]-s[3]))
3331 3331 atexit.register(print_time)
3332 3332
3333 3333 u.updateopts(options["verbose"], options["debug"], options["quiet"],
3334 3334 not options["noninteractive"], options["traceback"],
3335 3335 options["config"])
3336 3336
3337 3337 # enter the debugger before command execution
3338 3338 if options['debugger']:
3339 3339 pdb.set_trace()
3340 3340
3341 3341 try:
3342 3342 if options['cwd']:
3343 3343 try:
3344 3344 os.chdir(options['cwd'])
3345 3345 except OSError, inst:
3346 3346 raise util.Abort('%s: %s' %
3347 3347 (options['cwd'], inst.strerror))
3348 3348
3349 3349 path = u.expandpath(options["repository"]) or ""
3350 3350 repo = path and hg.repository(u, path=path) or None
3351 3351
3352 3352 if options['help']:
3353 3353 return help_(u, cmd, options['version'])
3354 3354 elif options['version']:
3355 3355 return show_version(u)
3356 3356 elif not cmd:
3357 3357 return help_(u, 'shortlist')
3358 3358
3359 3359 if cmd not in norepo.split():
3360 3360 try:
3361 3361 if not repo:
3362 3362 repo = hg.repository(u, path=path)
3363 3363 u = repo.ui
3364 3364 for name in external.itervalues():
3365 3365 mod = sys.modules[name]
3366 3366 if hasattr(mod, 'reposetup'):
3367 3367 mod.reposetup(u, repo)
3368 3368 hg.repo_setup_hooks.append(mod.reposetup)
3369 3369 except hg.RepoError:
3370 3370 if cmd not in optionalrepo.split():
3371 3371 raise
3372 3372 d = lambda: func(u, repo, *args, **cmdoptions)
3373 3373 else:
3374 3374 d = lambda: func(u, *args, **cmdoptions)
3375 3375
3376 3376 # reupdate the options, repo/.hg/hgrc may have changed them
3377 3377 u.updateopts(options["verbose"], options["debug"], options["quiet"],
3378 3378 not options["noninteractive"], options["traceback"],
3379 3379 options["config"])
3380 3380
3381 3381 try:
3382 3382 if options['profile']:
3383 3383 import hotshot, hotshot.stats
3384 3384 prof = hotshot.Profile("hg.prof")
3385 3385 try:
3386 3386 try:
3387 3387 return prof.runcall(d)
3388 3388 except:
3389 3389 try:
3390 3390 u.warn(_('exception raised - generating '
3391 3391 'profile anyway\n'))
3392 3392 except:
3393 3393 pass
3394 3394 raise
3395 3395 finally:
3396 3396 prof.close()
3397 3397 stats = hotshot.stats.load("hg.prof")
3398 3398 stats.strip_dirs()
3399 3399 stats.sort_stats('time', 'calls')
3400 3400 stats.print_stats(40)
3401 3401 elif options['lsprof']:
3402 3402 try:
3403 3403 from mercurial import lsprof
3404 3404 except ImportError:
3405 3405 raise util.Abort(_(
3406 3406 'lsprof not available - install from '
3407 3407 'http://codespeak.net/svn/user/arigo/hack/misc/lsprof/'))
3408 3408 p = lsprof.Profiler()
3409 3409 p.enable(subcalls=True)
3410 3410 try:
3411 3411 return d()
3412 3412 finally:
3413 3413 p.disable()
3414 3414 stats = lsprof.Stats(p.getstats())
3415 3415 stats.sort()
3416 3416 stats.pprint(top=10, file=sys.stderr, climit=5)
3417 3417 else:
3418 3418 return d()
3419 3419 finally:
3420 3420 u.flush()
3421 3421 except:
3422 3422 # enter the debugger when we hit an exception
3423 3423 if options['debugger']:
3424 3424 pdb.post_mortem(sys.exc_info()[2])
3425 3425 u.print_exc()
3426 3426 raise
3427 3427 except ParseError, inst:
3428 3428 if inst.args[0]:
3429 3429 u.warn(_("hg %s: %s\n") % (inst.args[0], inst.args[1]))
3430 3430 help_(u, inst.args[0])
3431 3431 else:
3432 3432 u.warn(_("hg: %s\n") % inst.args[1])
3433 3433 help_(u, 'shortlist')
3434 3434 except AmbiguousCommand, inst:
3435 3435 u.warn(_("hg: command '%s' is ambiguous:\n %s\n") %
3436 3436 (inst.args[0], " ".join(inst.args[1])))
3437 3437 except UnknownCommand, inst:
3438 3438 u.warn(_("hg: unknown command '%s'\n") % inst.args[0])
3439 3439 help_(u, 'shortlist')
3440 3440 except hg.RepoError, inst:
3441 3441 u.warn(_("abort: %s!\n") % inst)
3442 3442 except lock.LockHeld, inst:
3443 3443 if inst.errno == errno.ETIMEDOUT:
3444 3444 reason = _('timed out waiting for lock held by %s') % inst.locker
3445 3445 else:
3446 3446 reason = _('lock held by %s') % inst.locker
3447 3447 u.warn(_("abort: %s: %s\n") % (inst.desc or inst.filename, reason))
3448 3448 except lock.LockUnavailable, inst:
3449 3449 u.warn(_("abort: could not lock %s: %s\n") %
3450 3450 (inst.desc or inst.filename, inst.strerror))
3451 3451 except revlog.RevlogError, inst:
3452 u.warn(_("abort: "), inst, "!\n")
3452 u.warn(_("abort: %s!\n") % inst)
3453 3453 except util.SignalInterrupt:
3454 3454 u.warn(_("killed!\n"))
3455 3455 except KeyboardInterrupt:
3456 3456 try:
3457 3457 u.warn(_("interrupted!\n"))
3458 3458 except IOError, inst:
3459 3459 if inst.errno == errno.EPIPE:
3460 3460 if u.debugflag:
3461 3461 u.warn(_("\nbroken pipe\n"))
3462 3462 else:
3463 3463 raise
3464 3464 except IOError, inst:
3465 3465 if hasattr(inst, "code"):
3466 3466 u.warn(_("abort: %s\n") % inst)
3467 3467 elif hasattr(inst, "reason"):
3468 3468 u.warn(_("abort: error: %s\n") % inst.reason[1])
3469 3469 elif hasattr(inst, "args") and inst[0] == errno.EPIPE:
3470 3470 if u.debugflag:
3471 3471 u.warn(_("broken pipe\n"))
3472 3472 elif getattr(inst, "strerror", None):
3473 3473 if getattr(inst, "filename", None):
3474 3474 u.warn(_("abort: %s - %s\n") % (inst.strerror, inst.filename))
3475 3475 else:
3476 3476 u.warn(_("abort: %s\n") % inst.strerror)
3477 3477 else:
3478 3478 raise
3479 3479 except OSError, inst:
3480 3480 if hasattr(inst, "filename"):
3481 3481 u.warn(_("abort: %s: %s\n") % (inst.strerror, inst.filename))
3482 3482 else:
3483 3483 u.warn(_("abort: %s\n") % inst.strerror)
3484 3484 except util.Abort, inst:
3485 u.warn(_('abort: '), inst.args[0] % inst.args[1:], '\n')
3485 u.warn(_("abort: %s\n") % inst)
3486 3486 except TypeError, inst:
3487 3487 # was this an argument error?
3488 3488 tb = traceback.extract_tb(sys.exc_info()[2])
3489 3489 if len(tb) > 2: # no
3490 3490 raise
3491 3491 u.debug(inst, "\n")
3492 3492 u.warn(_("%s: invalid arguments\n") % cmd)
3493 3493 help_(u, cmd)
3494 3494 except SystemExit, inst:
3495 3495 # Commands shouldn't sys.exit directly, but give a return code.
3496 3496 # Just in case catch this and and pass exit code to caller.
3497 3497 return inst.code
3498 3498 except:
3499 3499 u.warn(_("** unknown exception encountered, details follow\n"))
3500 3500 u.warn(_("** report bug details to "
3501 3501 "http://www.selenic.com/mercurial/bts\n"))
3502 3502 u.warn(_("** or mercurial@selenic.com\n"))
3503 3503 u.warn(_("** Mercurial Distributed SCM (version %s)\n")
3504 3504 % version.get_version())
3505 3505 raise
3506 3506
3507 3507 return -1
@@ -1,231 +1,231 b''
1 1 # hg.py - repository classes for mercurial
2 2 #
3 3 # Copyright 2005, 2006 Matt Mackall <mpm@selenic.com>
4 4 # Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
5 5 #
6 6 # This software may be used and distributed according to the terms
7 7 # of the GNU General Public License, incorporated herein by reference.
8 8
9 9 from node import *
10 10 from repo import *
11 11 from demandload import *
12 12 from i18n import gettext as _
13 13 demandload(globals(), "localrepo bundlerepo httprepo sshrepo statichttprepo")
14 14 demandload(globals(), "errno lock os shutil util merge@_merge verify@_verify")
15 15
16 16 def _local(path):
17 17 return (os.path.isfile(path and util.drop_scheme('file', path)) and
18 18 bundlerepo or localrepo)
19 19
20 20 schemes = {
21 21 'bundle': bundlerepo,
22 22 'file': _local,
23 23 'hg': httprepo,
24 24 'http': httprepo,
25 25 'https': httprepo,
26 26 'old-http': statichttprepo,
27 27 'ssh': sshrepo,
28 28 'static-http': statichttprepo,
29 29 }
30 30
31 31 def _lookup(path):
32 32 scheme = 'file'
33 33 if path:
34 34 c = path.find(':')
35 35 if c > 0:
36 36 scheme = path[:c]
37 37 thing = schemes.get(scheme) or schemes['file']
38 38 try:
39 39 return thing(path)
40 40 except TypeError:
41 41 return thing
42 42
43 43 def islocal(repo):
44 44 '''return true if repo or path is local'''
45 45 if isinstance(repo, str):
46 46 try:
47 47 return _lookup(repo).islocal(repo)
48 48 except AttributeError:
49 49 return False
50 50 return repo.local()
51 51
52 52 repo_setup_hooks = []
53 53
54 54 def repository(ui, path=None, create=False):
55 55 """return a repository object for the specified path"""
56 56 repo = _lookup(path).instance(ui, path, create)
57 57 for hook in repo_setup_hooks:
58 58 hook(ui, repo)
59 59 return repo
60 60
61 61 def defaultdest(source):
62 62 '''return default destination of clone if none is given'''
63 63 return os.path.basename(os.path.normpath(source))
64 64
65 65 def clone(ui, source, dest=None, pull=False, rev=None, update=True,
66 66 stream=False):
67 67 """Make a copy of an existing repository.
68 68
69 69 Create a copy of an existing repository in a new directory. The
70 70 source and destination are URLs, as passed to the repository
71 71 function. Returns a pair of repository objects, the source and
72 72 newly created destination.
73 73
74 74 The location of the source is added to the new repository's
75 75 .hg/hgrc file, as the default to be used for future pulls and
76 76 pushes.
77 77
78 78 If an exception is raised, the partly cloned/updated destination
79 79 repository will be deleted.
80 80
81 81 Arguments:
82 82
83 83 source: repository object or URL
84 84
85 85 dest: URL of destination repository to create (defaults to base
86 86 name of source repository)
87 87
88 88 pull: always pull from source repository, even in local case
89 89
90 90 stream: stream raw data uncompressed from repository (fast over
91 91 LAN, slow over WAN)
92 92
93 93 rev: revision to clone up to (implies pull=True)
94 94
95 95 update: update working directory after clone completes, if
96 96 destination is local repository
97 97 """
98 98 if isinstance(source, str):
99 99 src_repo = repository(ui, source)
100 100 else:
101 101 src_repo = source
102 102 source = src_repo.url()
103 103
104 104 if dest is None:
105 105 dest = defaultdest(source)
106 106
107 107 def localpath(path):
108 108 if path.startswith('file://'):
109 109 return path[7:]
110 110 if path.startswith('file:'):
111 111 return path[5:]
112 112 return path
113 113
114 114 dest = localpath(dest)
115 115 source = localpath(source)
116 116
117 117 if os.path.exists(dest):
118 raise util.Abort(_("destination '%s' already exists"), dest)
118 raise util.Abort(_("destination '%s' already exists") % dest)
119 119
120 120 class DirCleanup(object):
121 121 def __init__(self, dir_):
122 122 self.rmtree = shutil.rmtree
123 123 self.dir_ = dir_
124 124 def close(self):
125 125 self.dir_ = None
126 126 def __del__(self):
127 127 if self.dir_:
128 128 self.rmtree(self.dir_, True)
129 129
130 130 dest_repo = repository(ui, dest, create=True)
131 131
132 132 dest_path = None
133 133 dir_cleanup = None
134 134 if dest_repo.local():
135 135 dest_path = os.path.realpath(dest_repo.root)
136 136 dir_cleanup = DirCleanup(dest_path)
137 137
138 138 abspath = source
139 139 copy = False
140 140 if src_repo.local() and dest_repo.local():
141 141 abspath = os.path.abspath(source)
142 142 copy = not pull and not rev
143 143
144 144 src_lock, dest_lock = None, None
145 145 if copy:
146 146 try:
147 147 # we use a lock here because if we race with commit, we
148 148 # can end up with extra data in the cloned revlogs that's
149 149 # not pointed to by changesets, thus causing verify to
150 150 # fail
151 151 src_lock = src_repo.lock()
152 152 except lock.LockException:
153 153 copy = False
154 154
155 155 if copy:
156 156 # we lock here to avoid premature writing to the target
157 157 dest_lock = lock.lock(os.path.join(dest_path, ".hg", "lock"))
158 158
159 159 # we need to remove the (empty) data dir in dest so copyfiles
160 160 # can do its work
161 161 os.rmdir(os.path.join(dest_path, ".hg", "data"))
162 162 files = "data 00manifest.d 00manifest.i 00changelog.d 00changelog.i"
163 163 for f in files.split():
164 164 src = os.path.join(source, ".hg", f)
165 165 dst = os.path.join(dest_path, ".hg", f)
166 166 try:
167 167 util.copyfiles(src, dst)
168 168 except OSError, inst:
169 169 if inst.errno != errno.ENOENT:
170 170 raise
171 171
172 172 # we need to re-init the repo after manually copying the data
173 173 # into it
174 174 dest_repo = repository(ui, dest)
175 175
176 176 else:
177 177 revs = None
178 178 if rev:
179 179 if not src_repo.local():
180 180 raise util.Abort(_("clone by revision not supported yet "
181 181 "for remote repositories"))
182 182 revs = [src_repo.lookup(r) for r in rev]
183 183
184 184 if dest_repo.local():
185 185 dest_repo.clone(src_repo, heads=revs, stream=stream)
186 186 elif src_repo.local():
187 187 src_repo.push(dest_repo, revs=revs)
188 188 else:
189 189 raise util.Abort(_("clone from remote to remote not supported"))
190 190
191 191 if src_lock:
192 192 src_lock.release()
193 193
194 194 if dest_repo.local():
195 195 fp = dest_repo.opener("hgrc", "w", text=True)
196 196 fp.write("[paths]\n")
197 197 fp.write("default = %s\n" % abspath)
198 198 fp.close()
199 199
200 200 if dest_lock:
201 201 dest_lock.release()
202 202
203 203 if update:
204 204 _merge.update(dest_repo, dest_repo.changelog.tip())
205 205 if dir_cleanup:
206 206 dir_cleanup.close()
207 207
208 208 return src_repo, dest_repo
209 209
210 210 def update(repo, node):
211 211 """update the working directory to node, merging linear changes"""
212 212 return _merge.update(repo, node)
213 213
214 214 def clean(repo, node, wlock=None, show_stats=True):
215 215 """forcibly switch the working directory to node, clobbering changes"""
216 216 return _merge.update(repo, node, force=True, wlock=wlock,
217 217 show_stats=show_stats)
218 218
219 219 def merge(repo, node, force=None, remind=True, wlock=None):
220 220 """branch merge with node, resolving changes"""
221 221 return _merge.update(repo, node, branchmerge=True, force=force,
222 222 remind=remind, wlock=wlock)
223 223
224 224 def revert(repo, node, choose, wlock):
225 225 """revert changes to revision in node without updating dirstate"""
226 226 return _merge.update(repo, node, force=True, partial=choose,
227 227 show_stats=False, wlock=wlock)
228 228
229 229 def verify(repo):
230 230 """verify the consistency of a repository"""
231 231 return _verify.verify(repo)
@@ -1,352 +1,352 b''
1 1 # httprepo.py - HTTP repository proxy classes for mercurial
2 2 #
3 3 # Copyright 2005, 2006 Matt Mackall <mpm@selenic.com>
4 4 # Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
5 5 #
6 6 # This software may be used and distributed according to the terms
7 7 # of the GNU General Public License, incorporated herein by reference.
8 8
9 9 from node import *
10 10 from remoterepo import *
11 11 from i18n import gettext as _
12 12 from demandload import *
13 13 demandload(globals(), "hg os urllib urllib2 urlparse zlib util httplib")
14 14 demandload(globals(), "errno keepalive tempfile socket")
15 15
16 16 class passwordmgr(urllib2.HTTPPasswordMgrWithDefaultRealm):
17 17 def __init__(self, ui):
18 18 urllib2.HTTPPasswordMgrWithDefaultRealm.__init__(self)
19 19 self.ui = ui
20 20
21 21 def find_user_password(self, realm, authuri):
22 22 authinfo = urllib2.HTTPPasswordMgrWithDefaultRealm.find_user_password(
23 23 self, realm, authuri)
24 24 user, passwd = authinfo
25 25 if user and passwd:
26 26 return (user, passwd)
27 27
28 28 if not self.ui.interactive:
29 29 raise util.Abort(_('http authorization required'))
30 30
31 31 self.ui.write(_("http authorization required\n"))
32 32 self.ui.status(_("realm: %s\n") % realm)
33 33 if user:
34 34 self.ui.status(_("user: %s\n") % user)
35 35 else:
36 36 user = self.ui.prompt(_("user:"), default=None)
37 37
38 38 if not passwd:
39 39 passwd = self.ui.getpass()
40 40
41 41 self.add_password(realm, authuri, user, passwd)
42 42 return (user, passwd)
43 43
44 44 def netlocsplit(netloc):
45 45 '''split [user[:passwd]@]host[:port] into 4-tuple.'''
46 46
47 47 a = netloc.find('@')
48 48 if a == -1:
49 49 user, passwd = None, None
50 50 else:
51 51 userpass, netloc = netloc[:a], netloc[a+1:]
52 52 c = userpass.find(':')
53 53 if c == -1:
54 54 user, passwd = urllib.unquote(userpass), None
55 55 else:
56 56 user = urllib.unquote(userpass[:c])
57 57 passwd = urllib.unquote(userpass[c+1:])
58 58 c = netloc.find(':')
59 59 if c == -1:
60 60 host, port = netloc, None
61 61 else:
62 62 host, port = netloc[:c], netloc[c+1:]
63 63 return host, port, user, passwd
64 64
65 65 def netlocunsplit(host, port, user=None, passwd=None):
66 66 '''turn host, port, user, passwd into [user[:passwd]@]host[:port].'''
67 67 if port:
68 68 hostport = host + ':' + port
69 69 else:
70 70 hostport = host
71 71 if user:
72 72 if passwd:
73 73 userpass = urllib.quote(user) + ':' + urllib.quote(passwd)
74 74 else:
75 75 userpass = urllib.quote(user)
76 76 return userpass + '@' + hostport
77 77 return hostport
78 78
79 79 class httpconnection(keepalive.HTTPConnection):
80 80 # must be able to send big bundle as stream.
81 81
82 82 def send(self, data):
83 83 if isinstance(data, str):
84 84 keepalive.HTTPConnection.send(self, data)
85 85 else:
86 86 # if auth required, some data sent twice, so rewind here
87 87 data.seek(0)
88 88 for chunk in util.filechunkiter(data):
89 89 keepalive.HTTPConnection.send(self, chunk)
90 90
91 91 class basehttphandler(keepalive.HTTPHandler):
92 92 def http_open(self, req):
93 93 return self.do_open(httpconnection, req)
94 94
95 95 has_https = hasattr(urllib2, 'HTTPSHandler')
96 96 if has_https:
97 97 class httpsconnection(httplib.HTTPSConnection):
98 98 response_class = keepalive.HTTPResponse
99 99 # must be able to send big bundle as stream.
100 100
101 101 def send(self, data):
102 102 if isinstance(data, str):
103 103 httplib.HTTPSConnection.send(self, data)
104 104 else:
105 105 # if auth required, some data sent twice, so rewind here
106 106 data.seek(0)
107 107 for chunk in util.filechunkiter(data):
108 108 httplib.HTTPSConnection.send(self, chunk)
109 109
110 110 class httphandler(basehttphandler, urllib2.HTTPSHandler):
111 111 def https_open(self, req):
112 112 return self.do_open(httpsconnection, req)
113 113 else:
114 114 class httphandler(basehttphandler):
115 115 pass
116 116
117 117 class httprepository(remoterepository):
118 118 def __init__(self, ui, path):
119 119 self.path = path
120 120 self.caps = None
121 121 scheme, netloc, urlpath, query, frag = urlparse.urlsplit(path)
122 122 if query or frag:
123 123 raise util.Abort(_('unsupported URL component: "%s"') %
124 124 (query or frag))
125 125 if not urlpath: urlpath = '/'
126 126 host, port, user, passwd = netlocsplit(netloc)
127 127
128 128 # urllib cannot handle URLs with embedded user or passwd
129 129 self._url = urlparse.urlunsplit((scheme, netlocunsplit(host, port),
130 130 urlpath, '', ''))
131 131 self.ui = ui
132 132
133 133 proxyurl = ui.config("http_proxy", "host") or os.getenv('http_proxy')
134 134 proxyauthinfo = None
135 135 handler = httphandler()
136 136
137 137 if proxyurl:
138 138 # proxy can be proper url or host[:port]
139 139 if not (proxyurl.startswith('http:') or
140 140 proxyurl.startswith('https:')):
141 141 proxyurl = 'http://' + proxyurl + '/'
142 142 snpqf = urlparse.urlsplit(proxyurl)
143 143 proxyscheme, proxynetloc, proxypath, proxyquery, proxyfrag = snpqf
144 144 hpup = netlocsplit(proxynetloc)
145 145
146 146 proxyhost, proxyport, proxyuser, proxypasswd = hpup
147 147 if not proxyuser:
148 148 proxyuser = ui.config("http_proxy", "user")
149 149 proxypasswd = ui.config("http_proxy", "passwd")
150 150
151 151 # see if we should use a proxy for this url
152 152 no_list = [ "localhost", "127.0.0.1" ]
153 153 no_list.extend([p.lower() for
154 154 p in ui.configlist("http_proxy", "no")])
155 155 no_list.extend([p.strip().lower() for
156 156 p in os.getenv("no_proxy", '').split(',')
157 157 if p.strip()])
158 158 # "http_proxy.always" config is for running tests on localhost
159 159 if (not ui.configbool("http_proxy", "always") and
160 160 host.lower() in no_list):
161 161 ui.debug(_('disabling proxy for %s\n') % host)
162 162 else:
163 163 proxyurl = urlparse.urlunsplit((
164 164 proxyscheme, netlocunsplit(proxyhost, proxyport,
165 165 proxyuser, proxypasswd or ''),
166 166 proxypath, proxyquery, proxyfrag))
167 167 handler = urllib2.ProxyHandler({scheme: proxyurl})
168 168 ui.debug(_('proxying through %s\n') % proxyurl)
169 169
170 170 # urllib2 takes proxy values from the environment and those
171 171 # will take precedence if found, so drop them
172 172 for env in ["HTTP_PROXY", "http_proxy", "no_proxy"]:
173 173 try:
174 174 if os.environ.has_key(env):
175 175 del os.environ[env]
176 176 except OSError:
177 177 pass
178 178
179 179 passmgr = passwordmgr(ui)
180 180 if user:
181 181 ui.debug(_('http auth: user %s, password %s\n') %
182 182 (user, passwd and '*' * len(passwd) or 'not set'))
183 183 passmgr.add_password(None, host, user, passwd or '')
184 184
185 185 opener = urllib2.build_opener(
186 186 handler,
187 187 urllib2.HTTPBasicAuthHandler(passmgr),
188 188 urllib2.HTTPDigestAuthHandler(passmgr))
189 189
190 190 # 1.0 here is the _protocol_ version
191 191 opener.addheaders = [('User-agent', 'mercurial/proto-1.0')]
192 192 urllib2.install_opener(opener)
193 193
194 194 def url(self):
195 195 return self.path
196 196
197 197 # look up capabilities only when needed
198 198
199 199 def get_caps(self):
200 200 if self.caps is None:
201 201 try:
202 202 self.caps = self.do_read('capabilities').split()
203 203 except hg.RepoError:
204 204 self.caps = ()
205 205 self.ui.debug(_('capabilities: %s\n') %
206 206 (' '.join(self.caps or ['none'])))
207 207 return self.caps
208 208
209 209 capabilities = property(get_caps)
210 210
211 211 def lock(self):
212 212 raise util.Abort(_('operation not supported over http'))
213 213
214 214 def do_cmd(self, cmd, **args):
215 215 data = args.pop('data', None)
216 216 headers = args.pop('headers', {})
217 217 self.ui.debug(_("sending %s command\n") % cmd)
218 218 q = {"cmd": cmd}
219 219 q.update(args)
220 220 qs = urllib.urlencode(q)
221 221 cu = "%s?%s" % (self._url, qs)
222 222 try:
223 223 resp = urllib2.urlopen(urllib2.Request(cu, data, headers))
224 224 except urllib2.HTTPError, inst:
225 225 if inst.code == 401:
226 226 raise util.Abort(_('authorization failed'))
227 227 raise
228 228 except httplib.HTTPException, inst:
229 229 self.ui.debug(_('http error while sending %s command\n') % cmd)
230 230 self.ui.print_exc()
231 231 raise IOError(None, inst)
232 232 try:
233 233 proto = resp.getheader('content-type')
234 234 except AttributeError:
235 235 proto = resp.headers['content-type']
236 236
237 237 # accept old "text/plain" and "application/hg-changegroup" for now
238 238 if not proto.startswith('application/mercurial') and \
239 239 not proto.startswith('text/plain') and \
240 240 not proto.startswith('application/hg-changegroup'):
241 241 raise hg.RepoError(_("'%s' does not appear to be an hg repository") %
242 242 self._url)
243 243
244 244 if proto.startswith('application/mercurial'):
245 245 version = proto[22:]
246 246 if float(version) > 0.1:
247 247 raise hg.RepoError(_("'%s' uses newer protocol %s") %
248 248 (self._url, version))
249 249
250 250 return resp
251 251
252 252 def do_read(self, cmd, **args):
253 253 fp = self.do_cmd(cmd, **args)
254 254 try:
255 255 return fp.read()
256 256 finally:
257 257 # if using keepalive, allow connection to be reused
258 258 fp.close()
259 259
260 260 def heads(self):
261 261 d = self.do_read("heads")
262 262 try:
263 263 return map(bin, d[:-1].split(" "))
264 264 except:
265 265 self.ui.warn(_("unexpected response:\n") + d[:400] + "\n...\n")
266 266 raise
267 267
268 268 def branches(self, nodes):
269 269 n = " ".join(map(hex, nodes))
270 270 d = self.do_read("branches", nodes=n)
271 271 try:
272 272 br = [ tuple(map(bin, b.split(" "))) for b in d.splitlines() ]
273 273 return br
274 274 except:
275 275 self.ui.warn(_("unexpected response:\n") + d[:400] + "\n...\n")
276 276 raise
277 277
278 278 def between(self, pairs):
279 279 n = "\n".join(["-".join(map(hex, p)) for p in pairs])
280 280 d = self.do_read("between", pairs=n)
281 281 try:
282 282 p = [ l and map(bin, l.split(" ")) or [] for l in d.splitlines() ]
283 283 return p
284 284 except:
285 285 self.ui.warn(_("unexpected response:\n") + d[:400] + "\n...\n")
286 286 raise
287 287
288 288 def changegroup(self, nodes, kind):
289 289 n = " ".join(map(hex, nodes))
290 290 f = self.do_cmd("changegroup", roots=n)
291 291 bytes = 0
292 292
293 293 def zgenerator(f):
294 294 zd = zlib.decompressobj()
295 295 try:
296 296 for chnk in f:
297 297 yield zd.decompress(chnk)
298 298 except httplib.HTTPException, inst:
299 299 raise IOError(None, _('connection ended unexpectedly'))
300 300 yield zd.flush()
301 301
302 302 return util.chunkbuffer(zgenerator(util.filechunkiter(f)))
303 303
304 304 def unbundle(self, cg, heads, source):
305 305 # have to stream bundle to a temp file because we do not have
306 306 # http 1.1 chunked transfer.
307 307
308 308 fd, tempname = tempfile.mkstemp(prefix='hg-unbundle-')
309 309 fp = os.fdopen(fd, 'wb+')
310 310 try:
311 311 for chunk in util.filechunkiter(cg):
312 312 fp.write(chunk)
313 313 length = fp.tell()
314 314 try:
315 315 rfp = self.do_cmd(
316 316 'unbundle', data=fp,
317 317 headers={'content-length': length,
318 318 'content-type': 'application/octet-stream'},
319 319 heads=' '.join(map(hex, heads)))
320 320 try:
321 321 ret = int(rfp.readline())
322 322 self.ui.write(rfp.read())
323 323 return ret
324 324 finally:
325 325 rfp.close()
326 326 except socket.error, err:
327 327 if err[0] in (errno.ECONNRESET, errno.EPIPE):
328 raise util.Abort(_('push failed: %s'), err[1])
328 raise util.Abort(_('push failed: %s') % err[1])
329 329 raise util.Abort(err[1])
330 330 finally:
331 331 fp.close()
332 332 os.unlink(tempname)
333 333
334 334 def stream_out(self):
335 335 return self.do_cmd('stream_out')
336 336
337 337 class httpsrepository(httprepository):
338 338 def __init__(self, ui, path):
339 339 if not has_https:
340 340 raise util.Abort(_('Python support for SSL and HTTPS '
341 341 'is not installed'))
342 342 httprepository.__init__(self, ui, path)
343 343
344 344 def instance(ui, path, create):
345 345 if create:
346 346 raise util.Abort(_('cannot create new http repository'))
347 347 if path.startswith('hg:'):
348 348 ui.warn(_("hg:// syntax is deprecated, please use http:// instead\n"))
349 349 path = 'http:' + path[3:]
350 350 if path.startswith('https:'):
351 351 return httpsrepository(ui, path)
352 352 return httprepository(ui, path)
@@ -1,1751 +1,1751 b''
1 1 # localrepo.py - read/write repository class for mercurial
2 2 #
3 3 # Copyright 2005, 2006 Matt Mackall <mpm@selenic.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 from node import *
9 9 from i18n import gettext as _
10 10 from demandload import *
11 11 import repo
12 12 demandload(globals(), "appendfile changegroup")
13 13 demandload(globals(), "changelog dirstate filelog manifest context")
14 14 demandload(globals(), "re lock transaction tempfile stat mdiff errno ui")
15 15 demandload(globals(), "os revlog time util")
16 16
17 17 class localrepository(repo.repository):
18 18 capabilities = ()
19 19
20 20 def __del__(self):
21 21 self.transhandle = None
22 22 def __init__(self, parentui, path=None, create=0):
23 23 repo.repository.__init__(self)
24 24 if not path:
25 25 p = os.getcwd()
26 26 while not os.path.isdir(os.path.join(p, ".hg")):
27 27 oldp = p
28 28 p = os.path.dirname(p)
29 29 if p == oldp:
30 30 raise repo.RepoError(_("no repo found"))
31 31 path = p
32 32 self.path = os.path.join(path, ".hg")
33 33
34 34 if not os.path.isdir(self.path):
35 35 if create:
36 36 if not os.path.exists(path):
37 37 os.mkdir(path)
38 38 os.mkdir(self.path)
39 39 os.mkdir(self.join("data"))
40 40 else:
41 41 raise repo.RepoError(_("repository %s not found") % path)
42 42 elif create:
43 43 raise repo.RepoError(_("repository %s already exists") % path)
44 44
45 45 self.root = os.path.abspath(path)
46 46 self.origroot = path
47 47 self.ui = ui.ui(parentui=parentui)
48 48 self.opener = util.opener(self.path)
49 49 self.wopener = util.opener(self.root)
50 50
51 51 try:
52 52 self.ui.readconfig(self.join("hgrc"), self.root)
53 53 except IOError:
54 54 pass
55 55
56 56 v = self.ui.revlogopts
57 57 self.revlogversion = int(v.get('format', revlog.REVLOG_DEFAULT_FORMAT))
58 58 self.revlogv1 = self.revlogversion != revlog.REVLOGV0
59 59 fl = v.get('flags', None)
60 60 flags = 0
61 61 if fl != None:
62 62 for x in fl.split():
63 63 flags |= revlog.flagstr(x)
64 64 elif self.revlogv1:
65 65 flags = revlog.REVLOG_DEFAULT_FLAGS
66 66
67 67 v = self.revlogversion | flags
68 68 self.manifest = manifest.manifest(self.opener, v)
69 69 self.changelog = changelog.changelog(self.opener, v)
70 70
71 71 # the changelog might not have the inline index flag
72 72 # on. If the format of the changelog is the same as found in
73 73 # .hgrc, apply any flags found in the .hgrc as well.
74 74 # Otherwise, just version from the changelog
75 75 v = self.changelog.version
76 76 if v == self.revlogversion:
77 77 v |= flags
78 78 self.revlogversion = v
79 79
80 80 self.tagscache = None
81 81 self.nodetagscache = None
82 82 self.encodepats = None
83 83 self.decodepats = None
84 84 self.transhandle = None
85 85
86 86 self.dirstate = dirstate.dirstate(self.opener, self.ui, self.root)
87 87
88 88 def url(self):
89 89 return 'file:' + self.root
90 90
91 91 def hook(self, name, throw=False, **args):
92 92 def callhook(hname, funcname):
93 93 '''call python hook. hook is callable object, looked up as
94 94 name in python module. if callable returns "true", hook
95 95 fails, else passes. if hook raises exception, treated as
96 96 hook failure. exception propagates if throw is "true".
97 97
98 98 reason for "true" meaning "hook failed" is so that
99 99 unmodified commands (e.g. mercurial.commands.update) can
100 100 be run as hooks without wrappers to convert return values.'''
101 101
102 102 self.ui.note(_("calling hook %s: %s\n") % (hname, funcname))
103 103 d = funcname.rfind('.')
104 104 if d == -1:
105 105 raise util.Abort(_('%s hook is invalid ("%s" not in a module)')
106 106 % (hname, funcname))
107 107 modname = funcname[:d]
108 108 try:
109 109 obj = __import__(modname)
110 110 except ImportError:
111 111 try:
112 112 # extensions are loaded with hgext_ prefix
113 113 obj = __import__("hgext_%s" % modname)
114 114 except ImportError:
115 115 raise util.Abort(_('%s hook is invalid '
116 116 '(import of "%s" failed)') %
117 117 (hname, modname))
118 118 try:
119 119 for p in funcname.split('.')[1:]:
120 120 obj = getattr(obj, p)
121 121 except AttributeError, err:
122 122 raise util.Abort(_('%s hook is invalid '
123 123 '("%s" is not defined)') %
124 124 (hname, funcname))
125 125 if not callable(obj):
126 126 raise util.Abort(_('%s hook is invalid '
127 127 '("%s" is not callable)') %
128 128 (hname, funcname))
129 129 try:
130 130 r = obj(ui=self.ui, repo=self, hooktype=name, **args)
131 131 except (KeyboardInterrupt, util.SignalInterrupt):
132 132 raise
133 133 except Exception, exc:
134 134 if isinstance(exc, util.Abort):
135 135 self.ui.warn(_('error: %s hook failed: %s\n') %
136 (hname, exc.args[0] % exc.args[1:]))
136 (hname, exc.args[0]))
137 137 else:
138 138 self.ui.warn(_('error: %s hook raised an exception: '
139 139 '%s\n') % (hname, exc))
140 140 if throw:
141 141 raise
142 142 self.ui.print_exc()
143 143 return True
144 144 if r:
145 145 if throw:
146 146 raise util.Abort(_('%s hook failed') % hname)
147 147 self.ui.warn(_('warning: %s hook failed\n') % hname)
148 148 return r
149 149
150 150 def runhook(name, cmd):
151 151 self.ui.note(_("running hook %s: %s\n") % (name, cmd))
152 152 env = dict([('HG_' + k.upper(), v) for k, v in args.iteritems()])
153 153 r = util.system(cmd, environ=env, cwd=self.root)
154 154 if r:
155 155 desc, r = util.explain_exit(r)
156 156 if throw:
157 157 raise util.Abort(_('%s hook %s') % (name, desc))
158 158 self.ui.warn(_('warning: %s hook %s\n') % (name, desc))
159 159 return r
160 160
161 161 r = False
162 162 hooks = [(hname, cmd) for hname, cmd in self.ui.configitems("hooks")
163 163 if hname.split(".", 1)[0] == name and cmd]
164 164 hooks.sort()
165 165 for hname, cmd in hooks:
166 166 if cmd.startswith('python:'):
167 167 r = callhook(hname, cmd[7:].strip()) or r
168 168 else:
169 169 r = runhook(hname, cmd) or r
170 170 return r
171 171
172 172 tag_disallowed = ':\r\n'
173 173
174 174 def tag(self, name, node, message, local, user, date):
175 175 '''tag a revision with a symbolic name.
176 176
177 177 if local is True, the tag is stored in a per-repository file.
178 178 otherwise, it is stored in the .hgtags file, and a new
179 179 changeset is committed with the change.
180 180
181 181 keyword arguments:
182 182
183 183 local: whether to store tag in non-version-controlled file
184 184 (default False)
185 185
186 186 message: commit message to use if committing
187 187
188 188 user: name of user to use if committing
189 189
190 190 date: date tuple to use if committing'''
191 191
192 192 for c in self.tag_disallowed:
193 193 if c in name:
194 194 raise util.Abort(_('%r cannot be used in a tag name') % c)
195 195
196 196 self.hook('pretag', throw=True, node=hex(node), tag=name, local=local)
197 197
198 198 if local:
199 199 self.opener('localtags', 'a').write('%s %s\n' % (hex(node), name))
200 200 self.hook('tag', node=hex(node), tag=name, local=local)
201 201 return
202 202
203 203 for x in self.status()[:5]:
204 204 if '.hgtags' in x:
205 205 raise util.Abort(_('working copy of .hgtags is changed '
206 206 '(please commit .hgtags manually)'))
207 207
208 208 self.wfile('.hgtags', 'ab').write('%s %s\n' % (hex(node), name))
209 209 if self.dirstate.state('.hgtags') == '?':
210 210 self.add(['.hgtags'])
211 211
212 212 self.commit(['.hgtags'], message, user, date)
213 213 self.hook('tag', node=hex(node), tag=name, local=local)
214 214
215 215 def tags(self):
216 216 '''return a mapping of tag to node'''
217 217 if not self.tagscache:
218 218 self.tagscache = {}
219 219
220 220 def parsetag(line, context):
221 221 if not line:
222 222 return
223 223 s = l.split(" ", 1)
224 224 if len(s) != 2:
225 225 self.ui.warn(_("%s: cannot parse entry\n") % context)
226 226 return
227 227 node, key = s
228 228 key = key.strip()
229 229 try:
230 230 bin_n = bin(node)
231 231 except TypeError:
232 232 self.ui.warn(_("%s: node '%s' is not well formed\n") %
233 233 (context, node))
234 234 return
235 235 if bin_n not in self.changelog.nodemap:
236 236 self.ui.warn(_("%s: tag '%s' refers to unknown node\n") %
237 237 (context, key))
238 238 return
239 239 self.tagscache[key] = bin_n
240 240
241 241 # read the tags file from each head, ending with the tip,
242 242 # and add each tag found to the map, with "newer" ones
243 243 # taking precedence
244 244 heads = self.heads()
245 245 heads.reverse()
246 246 fl = self.file(".hgtags")
247 247 for node in heads:
248 248 change = self.changelog.read(node)
249 249 rev = self.changelog.rev(node)
250 250 fn, ff = self.manifest.find(change[0], '.hgtags')
251 251 if fn is None: continue
252 252 count = 0
253 253 for l in fl.read(fn).splitlines():
254 254 count += 1
255 255 parsetag(l, _(".hgtags (rev %d:%s), line %d") %
256 256 (rev, short(node), count))
257 257 try:
258 258 f = self.opener("localtags")
259 259 count = 0
260 260 for l in f:
261 261 count += 1
262 262 parsetag(l, _("localtags, line %d") % count)
263 263 except IOError:
264 264 pass
265 265
266 266 self.tagscache['tip'] = self.changelog.tip()
267 267
268 268 return self.tagscache
269 269
270 270 def tagslist(self):
271 271 '''return a list of tags ordered by revision'''
272 272 l = []
273 273 for t, n in self.tags().items():
274 274 try:
275 275 r = self.changelog.rev(n)
276 276 except:
277 277 r = -2 # sort to the beginning of the list if unknown
278 278 l.append((r, t, n))
279 279 l.sort()
280 280 return [(t, n) for r, t, n in l]
281 281
282 282 def nodetags(self, node):
283 283 '''return the tags associated with a node'''
284 284 if not self.nodetagscache:
285 285 self.nodetagscache = {}
286 286 for t, n in self.tags().items():
287 287 self.nodetagscache.setdefault(n, []).append(t)
288 288 return self.nodetagscache.get(node, [])
289 289
290 290 def lookup(self, key):
291 291 try:
292 292 return self.tags()[key]
293 293 except KeyError:
294 294 if key == '.':
295 295 key = self.dirstate.parents()[0]
296 296 if key == nullid:
297 297 raise repo.RepoError(_("no revision checked out"))
298 298 try:
299 299 return self.changelog.lookup(key)
300 300 except:
301 301 raise repo.RepoError(_("unknown revision '%s'") % key)
302 302
303 303 def dev(self):
304 304 return os.lstat(self.path).st_dev
305 305
306 306 def local(self):
307 307 return True
308 308
309 309 def join(self, f):
310 310 return os.path.join(self.path, f)
311 311
312 312 def wjoin(self, f):
313 313 return os.path.join(self.root, f)
314 314
315 315 def file(self, f):
316 316 if f[0] == '/':
317 317 f = f[1:]
318 318 return filelog.filelog(self.opener, f, self.revlogversion)
319 319
320 320 def changectx(self, changeid):
321 321 return context.changectx(self, changeid)
322 322
323 323 def filectx(self, path, changeid=None, fileid=None):
324 324 """changeid can be a changeset revision, node, or tag.
325 325 fileid can be a file revision or node."""
326 326 return context.filectx(self, path, changeid, fileid)
327 327
328 328 def getcwd(self):
329 329 return self.dirstate.getcwd()
330 330
331 331 def wfile(self, f, mode='r'):
332 332 return self.wopener(f, mode)
333 333
334 334 def wread(self, filename):
335 335 if self.encodepats == None:
336 336 l = []
337 337 for pat, cmd in self.ui.configitems("encode"):
338 338 mf = util.matcher(self.root, "", [pat], [], [])[1]
339 339 l.append((mf, cmd))
340 340 self.encodepats = l
341 341
342 342 data = self.wopener(filename, 'r').read()
343 343
344 344 for mf, cmd in self.encodepats:
345 345 if mf(filename):
346 346 self.ui.debug(_("filtering %s through %s\n") % (filename, cmd))
347 347 data = util.filter(data, cmd)
348 348 break
349 349
350 350 return data
351 351
352 352 def wwrite(self, filename, data, fd=None):
353 353 if self.decodepats == None:
354 354 l = []
355 355 for pat, cmd in self.ui.configitems("decode"):
356 356 mf = util.matcher(self.root, "", [pat], [], [])[1]
357 357 l.append((mf, cmd))
358 358 self.decodepats = l
359 359
360 360 for mf, cmd in self.decodepats:
361 361 if mf(filename):
362 362 self.ui.debug(_("filtering %s through %s\n") % (filename, cmd))
363 363 data = util.filter(data, cmd)
364 364 break
365 365
366 366 if fd:
367 367 return fd.write(data)
368 368 return self.wopener(filename, 'w').write(data)
369 369
370 370 def transaction(self):
371 371 tr = self.transhandle
372 372 if tr != None and tr.running():
373 373 return tr.nest()
374 374
375 375 # save dirstate for rollback
376 376 try:
377 377 ds = self.opener("dirstate").read()
378 378 except IOError:
379 379 ds = ""
380 380 self.opener("journal.dirstate", "w").write(ds)
381 381
382 382 tr = transaction.transaction(self.ui.warn, self.opener,
383 383 self.join("journal"),
384 384 aftertrans(self.path))
385 385 self.transhandle = tr
386 386 return tr
387 387
388 388 def recover(self):
389 389 l = self.lock()
390 390 if os.path.exists(self.join("journal")):
391 391 self.ui.status(_("rolling back interrupted transaction\n"))
392 392 transaction.rollback(self.opener, self.join("journal"))
393 393 self.reload()
394 394 return True
395 395 else:
396 396 self.ui.warn(_("no interrupted transaction available\n"))
397 397 return False
398 398
399 399 def rollback(self, wlock=None):
400 400 if not wlock:
401 401 wlock = self.wlock()
402 402 l = self.lock()
403 403 if os.path.exists(self.join("undo")):
404 404 self.ui.status(_("rolling back last transaction\n"))
405 405 transaction.rollback(self.opener, self.join("undo"))
406 406 util.rename(self.join("undo.dirstate"), self.join("dirstate"))
407 407 self.reload()
408 408 self.wreload()
409 409 else:
410 410 self.ui.warn(_("no rollback information available\n"))
411 411
412 412 def wreload(self):
413 413 self.dirstate.read()
414 414
415 415 def reload(self):
416 416 self.changelog.load()
417 417 self.manifest.load()
418 418 self.tagscache = None
419 419 self.nodetagscache = None
420 420
421 421 def do_lock(self, lockname, wait, releasefn=None, acquirefn=None,
422 422 desc=None):
423 423 try:
424 424 l = lock.lock(self.join(lockname), 0, releasefn, desc=desc)
425 425 except lock.LockHeld, inst:
426 426 if not wait:
427 427 raise
428 428 self.ui.warn(_("waiting for lock on %s held by %s\n") %
429 429 (desc, inst.args[0]))
430 430 # default to 600 seconds timeout
431 431 l = lock.lock(self.join(lockname),
432 432 int(self.ui.config("ui", "timeout") or 600),
433 433 releasefn, desc=desc)
434 434 if acquirefn:
435 435 acquirefn()
436 436 return l
437 437
438 438 def lock(self, wait=1):
439 439 return self.do_lock("lock", wait, acquirefn=self.reload,
440 440 desc=_('repository %s') % self.origroot)
441 441
442 442 def wlock(self, wait=1):
443 443 return self.do_lock("wlock", wait, self.dirstate.write,
444 444 self.wreload,
445 445 desc=_('working directory of %s') % self.origroot)
446 446
447 447 def checkfilemerge(self, filename, text, filelog, manifest1, manifest2):
448 448 "determine whether a new filenode is needed"
449 449 fp1 = manifest1.get(filename, nullid)
450 450 fp2 = manifest2.get(filename, nullid)
451 451
452 452 if fp2 != nullid:
453 453 # is one parent an ancestor of the other?
454 454 fpa = filelog.ancestor(fp1, fp2)
455 455 if fpa == fp1:
456 456 fp1, fp2 = fp2, nullid
457 457 elif fpa == fp2:
458 458 fp2 = nullid
459 459
460 460 # is the file unmodified from the parent? report existing entry
461 461 if fp2 == nullid and text == filelog.read(fp1):
462 462 return (fp1, None, None)
463 463
464 464 return (None, fp1, fp2)
465 465
466 466 def rawcommit(self, files, text, user, date, p1=None, p2=None, wlock=None):
467 467 orig_parent = self.dirstate.parents()[0] or nullid
468 468 p1 = p1 or self.dirstate.parents()[0] or nullid
469 469 p2 = p2 or self.dirstate.parents()[1] or nullid
470 470 c1 = self.changelog.read(p1)
471 471 c2 = self.changelog.read(p2)
472 472 m1 = self.manifest.read(c1[0]).copy()
473 473 m2 = self.manifest.read(c2[0])
474 474 changed = []
475 475
476 476 if orig_parent == p1:
477 477 update_dirstate = 1
478 478 else:
479 479 update_dirstate = 0
480 480
481 481 if not wlock:
482 482 wlock = self.wlock()
483 483 l = self.lock()
484 484 tr = self.transaction()
485 485 linkrev = self.changelog.count()
486 486 for f in files:
487 487 try:
488 488 t = self.wread(f)
489 489 m1.set(f, util.is_exec(self.wjoin(f), m1.execf(f)))
490 490 r = self.file(f)
491 491
492 492 (entry, fp1, fp2) = self.checkfilemerge(f, t, r, m1, m2)
493 493 if entry:
494 494 m1[f] = entry
495 495 continue
496 496
497 497 m1[f] = r.add(t, {}, tr, linkrev, fp1, fp2)
498 498 changed.append(f)
499 499 if update_dirstate:
500 500 self.dirstate.update([f], "n")
501 501 except IOError:
502 502 try:
503 503 del m1[f]
504 504 if update_dirstate:
505 505 self.dirstate.forget([f])
506 506 except:
507 507 # deleted from p2?
508 508 pass
509 509
510 510 mnode = self.manifest.add(m1, tr, linkrev, c1[0], c2[0])
511 511 user = user or self.ui.username()
512 512 n = self.changelog.add(mnode, changed, text, tr, p1, p2, user, date)
513 513 tr.close()
514 514 if update_dirstate:
515 515 self.dirstate.setparents(n, nullid)
516 516
517 517 def commit(self, files=None, text="", user=None, date=None,
518 518 match=util.always, force=False, lock=None, wlock=None,
519 519 force_editor=False):
520 520 commit = []
521 521 remove = []
522 522 changed = []
523 523
524 524 if files:
525 525 for f in files:
526 526 s = self.dirstate.state(f)
527 527 if s in 'nmai':
528 528 commit.append(f)
529 529 elif s == 'r':
530 530 remove.append(f)
531 531 else:
532 532 self.ui.warn(_("%s not tracked!\n") % f)
533 533 else:
534 534 modified, added, removed, deleted, unknown = self.status(match=match)[:5]
535 535 commit = modified + added
536 536 remove = removed
537 537
538 538 p1, p2 = self.dirstate.parents()
539 539 c1 = self.changelog.read(p1)
540 540 c2 = self.changelog.read(p2)
541 541 m1 = self.manifest.read(c1[0]).copy()
542 542 m2 = self.manifest.read(c2[0])
543 543
544 544 if not commit and not remove and not force and p2 == nullid:
545 545 self.ui.status(_("nothing changed\n"))
546 546 return None
547 547
548 548 xp1 = hex(p1)
549 549 if p2 == nullid: xp2 = ''
550 550 else: xp2 = hex(p2)
551 551
552 552 self.hook("precommit", throw=True, parent1=xp1, parent2=xp2)
553 553
554 554 if not wlock:
555 555 wlock = self.wlock()
556 556 if not lock:
557 557 lock = self.lock()
558 558 tr = self.transaction()
559 559
560 560 # check in files
561 561 new = {}
562 562 linkrev = self.changelog.count()
563 563 commit.sort()
564 564 for f in commit:
565 565 self.ui.note(f + "\n")
566 566 try:
567 567 m1.set(f, util.is_exec(self.wjoin(f), m1.execf(f)))
568 568 t = self.wread(f)
569 569 except IOError:
570 570 self.ui.warn(_("trouble committing %s!\n") % f)
571 571 raise
572 572
573 573 r = self.file(f)
574 574
575 575 meta = {}
576 576 cp = self.dirstate.copied(f)
577 577 if cp:
578 578 meta["copy"] = cp
579 579 meta["copyrev"] = hex(m1.get(cp, m2.get(cp, nullid)))
580 580 self.ui.debug(_(" %s: copy %s:%s\n") % (f, cp, meta["copyrev"]))
581 581 fp1, fp2 = nullid, nullid
582 582 else:
583 583 entry, fp1, fp2 = self.checkfilemerge(f, t, r, m1, m2)
584 584 if entry:
585 585 new[f] = entry
586 586 continue
587 587
588 588 new[f] = r.add(t, meta, tr, linkrev, fp1, fp2)
589 589 # remember what we've added so that we can later calculate
590 590 # the files to pull from a set of changesets
591 591 changed.append(f)
592 592
593 593 # update manifest
594 594 m1.update(new)
595 595 for f in remove:
596 596 if f in m1:
597 597 del m1[f]
598 598 mn = self.manifest.add(m1, tr, linkrev, c1[0], c2[0],
599 599 (new, remove))
600 600
601 601 # add changeset
602 602 new = new.keys()
603 603 new.sort()
604 604
605 605 user = user or self.ui.username()
606 606 if not text or force_editor:
607 607 edittext = []
608 608 if text:
609 609 edittext.append(text)
610 610 edittext.append("")
611 611 if p2 != nullid:
612 612 edittext.append("HG: branch merge")
613 613 edittext.extend(["HG: changed %s" % f for f in changed])
614 614 edittext.extend(["HG: removed %s" % f for f in remove])
615 615 if not changed and not remove:
616 616 edittext.append("HG: no files changed")
617 617 edittext.append("")
618 618 # run editor in the repository root
619 619 olddir = os.getcwd()
620 620 os.chdir(self.root)
621 621 text = self.ui.edit("\n".join(edittext), user)
622 622 os.chdir(olddir)
623 623
624 624 lines = [line.rstrip() for line in text.rstrip().splitlines()]
625 625 while lines and not lines[0]:
626 626 del lines[0]
627 627 if not lines:
628 628 return None
629 629 text = '\n'.join(lines)
630 630 n = self.changelog.add(mn, changed + remove, text, tr, p1, p2, user, date)
631 631 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
632 632 parent2=xp2)
633 633 tr.close()
634 634
635 635 self.dirstate.setparents(n)
636 636 self.dirstate.update(new, "n")
637 637 self.dirstate.forget(remove)
638 638
639 639 self.hook("commit", node=hex(n), parent1=xp1, parent2=xp2)
640 640 return n
641 641
642 642 def walk(self, node=None, files=[], match=util.always, badmatch=None):
643 643 if node:
644 644 fdict = dict.fromkeys(files)
645 645 for fn in self.manifest.read(self.changelog.read(node)[0]):
646 646 for ffn in fdict:
647 647 # match if the file is the exact name or a directory
648 648 if ffn == fn or fn.startswith("%s/" % ffn):
649 649 del fdict[ffn]
650 650 break
651 651 if match(fn):
652 652 yield 'm', fn
653 653 for fn in fdict:
654 654 if badmatch and badmatch(fn):
655 655 if match(fn):
656 656 yield 'b', fn
657 657 else:
658 658 self.ui.warn(_('%s: No such file in rev %s\n') % (
659 659 util.pathto(self.getcwd(), fn), short(node)))
660 660 else:
661 661 for src, fn in self.dirstate.walk(files, match, badmatch=badmatch):
662 662 yield src, fn
663 663
664 664 def status(self, node1=None, node2=None, files=[], match=util.always,
665 665 wlock=None, list_ignored=False, list_clean=False):
666 666 """return status of files between two nodes or node and working directory
667 667
668 668 If node1 is None, use the first dirstate parent instead.
669 669 If node2 is None, compare node1 with working directory.
670 670 """
671 671
672 672 def fcmp(fn, mf):
673 673 t1 = self.wread(fn)
674 674 return self.file(fn).cmp(mf.get(fn, nullid), t1)
675 675
676 676 def mfmatches(node):
677 677 change = self.changelog.read(node)
678 678 mf = dict(self.manifest.read(change[0]))
679 679 for fn in mf.keys():
680 680 if not match(fn):
681 681 del mf[fn]
682 682 return mf
683 683
684 684 modified, added, removed, deleted, unknown = [], [], [], [], []
685 685 ignored, clean = [], []
686 686
687 687 compareworking = False
688 688 if not node1 or (not node2 and node1 == self.dirstate.parents()[0]):
689 689 compareworking = True
690 690
691 691 if not compareworking:
692 692 # read the manifest from node1 before the manifest from node2,
693 693 # so that we'll hit the manifest cache if we're going through
694 694 # all the revisions in parent->child order.
695 695 mf1 = mfmatches(node1)
696 696
697 697 # are we comparing the working directory?
698 698 if not node2:
699 699 if not wlock:
700 700 try:
701 701 wlock = self.wlock(wait=0)
702 702 except lock.LockException:
703 703 wlock = None
704 704 (lookup, modified, added, removed, deleted, unknown,
705 705 ignored, clean) = self.dirstate.status(files, match,
706 706 list_ignored, list_clean)
707 707
708 708 # are we comparing working dir against its parent?
709 709 if compareworking:
710 710 if lookup:
711 711 # do a full compare of any files that might have changed
712 712 mf2 = mfmatches(self.dirstate.parents()[0])
713 713 for f in lookup:
714 714 if fcmp(f, mf2):
715 715 modified.append(f)
716 716 else:
717 717 clean.append(f)
718 718 if wlock is not None:
719 719 self.dirstate.update([f], "n")
720 720 else:
721 721 # we are comparing working dir against non-parent
722 722 # generate a pseudo-manifest for the working dir
723 723 mf2 = mfmatches(self.dirstate.parents()[0])
724 724 for f in lookup + modified + added:
725 725 mf2[f] = ""
726 726 for f in removed:
727 727 if f in mf2:
728 728 del mf2[f]
729 729 else:
730 730 # we are comparing two revisions
731 731 mf2 = mfmatches(node2)
732 732
733 733 if not compareworking:
734 734 # flush lists from dirstate before comparing manifests
735 735 modified, added, clean = [], [], []
736 736
737 737 # make sure to sort the files so we talk to the disk in a
738 738 # reasonable order
739 739 mf2keys = mf2.keys()
740 740 mf2keys.sort()
741 741 for fn in mf2keys:
742 742 if mf1.has_key(fn):
743 743 if mf1[fn] != mf2[fn] and (mf2[fn] != "" or fcmp(fn, mf1)):
744 744 modified.append(fn)
745 745 elif list_clean:
746 746 clean.append(fn)
747 747 del mf1[fn]
748 748 else:
749 749 added.append(fn)
750 750
751 751 removed = mf1.keys()
752 752
753 753 # sort and return results:
754 754 for l in modified, added, removed, deleted, unknown, ignored, clean:
755 755 l.sort()
756 756 return (modified, added, removed, deleted, unknown, ignored, clean)
757 757
758 758 def add(self, list, wlock=None):
759 759 if not wlock:
760 760 wlock = self.wlock()
761 761 for f in list:
762 762 p = self.wjoin(f)
763 763 if not os.path.exists(p):
764 764 self.ui.warn(_("%s does not exist!\n") % f)
765 765 elif not os.path.isfile(p):
766 766 self.ui.warn(_("%s not added: only files supported currently\n")
767 767 % f)
768 768 elif self.dirstate.state(f) in 'an':
769 769 self.ui.warn(_("%s already tracked!\n") % f)
770 770 else:
771 771 self.dirstate.update([f], "a")
772 772
773 773 def forget(self, list, wlock=None):
774 774 if not wlock:
775 775 wlock = self.wlock()
776 776 for f in list:
777 777 if self.dirstate.state(f) not in 'ai':
778 778 self.ui.warn(_("%s not added!\n") % f)
779 779 else:
780 780 self.dirstate.forget([f])
781 781
782 782 def remove(self, list, unlink=False, wlock=None):
783 783 if unlink:
784 784 for f in list:
785 785 try:
786 786 util.unlink(self.wjoin(f))
787 787 except OSError, inst:
788 788 if inst.errno != errno.ENOENT:
789 789 raise
790 790 if not wlock:
791 791 wlock = self.wlock()
792 792 for f in list:
793 793 p = self.wjoin(f)
794 794 if os.path.exists(p):
795 795 self.ui.warn(_("%s still exists!\n") % f)
796 796 elif self.dirstate.state(f) == 'a':
797 797 self.dirstate.forget([f])
798 798 elif f not in self.dirstate:
799 799 self.ui.warn(_("%s not tracked!\n") % f)
800 800 else:
801 801 self.dirstate.update([f], "r")
802 802
803 803 def undelete(self, list, wlock=None):
804 804 p = self.dirstate.parents()[0]
805 805 mn = self.changelog.read(p)[0]
806 806 m = self.manifest.read(mn)
807 807 if not wlock:
808 808 wlock = self.wlock()
809 809 for f in list:
810 810 if self.dirstate.state(f) not in "r":
811 811 self.ui.warn("%s not removed!\n" % f)
812 812 else:
813 813 t = self.file(f).read(m[f])
814 814 self.wwrite(f, t)
815 815 util.set_exec(self.wjoin(f), m.execf(f))
816 816 self.dirstate.update([f], "n")
817 817
818 818 def copy(self, source, dest, wlock=None):
819 819 p = self.wjoin(dest)
820 820 if not os.path.exists(p):
821 821 self.ui.warn(_("%s does not exist!\n") % dest)
822 822 elif not os.path.isfile(p):
823 823 self.ui.warn(_("copy failed: %s is not a file\n") % dest)
824 824 else:
825 825 if not wlock:
826 826 wlock = self.wlock()
827 827 if self.dirstate.state(dest) == '?':
828 828 self.dirstate.update([dest], "a")
829 829 self.dirstate.copy(source, dest)
830 830
831 831 def heads(self, start=None):
832 832 heads = self.changelog.heads(start)
833 833 # sort the output in rev descending order
834 834 heads = [(-self.changelog.rev(h), h) for h in heads]
835 835 heads.sort()
836 836 return [n for (r, n) in heads]
837 837
838 838 # branchlookup returns a dict giving a list of branches for
839 839 # each head. A branch is defined as the tag of a node or
840 840 # the branch of the node's parents. If a node has multiple
841 841 # branch tags, tags are eliminated if they are visible from other
842 842 # branch tags.
843 843 #
844 844 # So, for this graph: a->b->c->d->e
845 845 # \ /
846 846 # aa -----/
847 847 # a has tag 2.6.12
848 848 # d has tag 2.6.13
849 849 # e would have branch tags for 2.6.12 and 2.6.13. Because the node
850 850 # for 2.6.12 can be reached from the node 2.6.13, that is eliminated
851 851 # from the list.
852 852 #
853 853 # It is possible that more than one head will have the same branch tag.
854 854 # callers need to check the result for multiple heads under the same
855 855 # branch tag if that is a problem for them (ie checkout of a specific
856 856 # branch).
857 857 #
858 858 # passing in a specific branch will limit the depth of the search
859 859 # through the parents. It won't limit the branches returned in the
860 860 # result though.
861 861 def branchlookup(self, heads=None, branch=None):
862 862 if not heads:
863 863 heads = self.heads()
864 864 headt = [ h for h in heads ]
865 865 chlog = self.changelog
866 866 branches = {}
867 867 merges = []
868 868 seenmerge = {}
869 869
870 870 # traverse the tree once for each head, recording in the branches
871 871 # dict which tags are visible from this head. The branches
872 872 # dict also records which tags are visible from each tag
873 873 # while we traverse.
874 874 while headt or merges:
875 875 if merges:
876 876 n, found = merges.pop()
877 877 visit = [n]
878 878 else:
879 879 h = headt.pop()
880 880 visit = [h]
881 881 found = [h]
882 882 seen = {}
883 883 while visit:
884 884 n = visit.pop()
885 885 if n in seen:
886 886 continue
887 887 pp = chlog.parents(n)
888 888 tags = self.nodetags(n)
889 889 if tags:
890 890 for x in tags:
891 891 if x == 'tip':
892 892 continue
893 893 for f in found:
894 894 branches.setdefault(f, {})[n] = 1
895 895 branches.setdefault(n, {})[n] = 1
896 896 break
897 897 if n not in found:
898 898 found.append(n)
899 899 if branch in tags:
900 900 continue
901 901 seen[n] = 1
902 902 if pp[1] != nullid and n not in seenmerge:
903 903 merges.append((pp[1], [x for x in found]))
904 904 seenmerge[n] = 1
905 905 if pp[0] != nullid:
906 906 visit.append(pp[0])
907 907 # traverse the branches dict, eliminating branch tags from each
908 908 # head that are visible from another branch tag for that head.
909 909 out = {}
910 910 viscache = {}
911 911 for h in heads:
912 912 def visible(node):
913 913 if node in viscache:
914 914 return viscache[node]
915 915 ret = {}
916 916 visit = [node]
917 917 while visit:
918 918 x = visit.pop()
919 919 if x in viscache:
920 920 ret.update(viscache[x])
921 921 elif x not in ret:
922 922 ret[x] = 1
923 923 if x in branches:
924 924 visit[len(visit):] = branches[x].keys()
925 925 viscache[node] = ret
926 926 return ret
927 927 if h not in branches:
928 928 continue
929 929 # O(n^2), but somewhat limited. This only searches the
930 930 # tags visible from a specific head, not all the tags in the
931 931 # whole repo.
932 932 for b in branches[h]:
933 933 vis = False
934 934 for bb in branches[h].keys():
935 935 if b != bb:
936 936 if b in visible(bb):
937 937 vis = True
938 938 break
939 939 if not vis:
940 940 l = out.setdefault(h, [])
941 941 l[len(l):] = self.nodetags(b)
942 942 return out
943 943
944 944 def branches(self, nodes):
945 945 if not nodes:
946 946 nodes = [self.changelog.tip()]
947 947 b = []
948 948 for n in nodes:
949 949 t = n
950 950 while 1:
951 951 p = self.changelog.parents(n)
952 952 if p[1] != nullid or p[0] == nullid:
953 953 b.append((t, n, p[0], p[1]))
954 954 break
955 955 n = p[0]
956 956 return b
957 957
958 958 def between(self, pairs):
959 959 r = []
960 960
961 961 for top, bottom in pairs:
962 962 n, l, i = top, [], 0
963 963 f = 1
964 964
965 965 while n != bottom:
966 966 p = self.changelog.parents(n)[0]
967 967 if i == f:
968 968 l.append(n)
969 969 f = f * 2
970 970 n = p
971 971 i += 1
972 972
973 973 r.append(l)
974 974
975 975 return r
976 976
977 977 def findincoming(self, remote, base=None, heads=None, force=False):
978 978 """Return list of roots of the subsets of missing nodes from remote
979 979
980 980 If base dict is specified, assume that these nodes and their parents
981 981 exist on the remote side and that no child of a node of base exists
982 982 in both remote and self.
983 983 Furthermore base will be updated to include the nodes that exists
984 984 in self and remote but no children exists in self and remote.
985 985 If a list of heads is specified, return only nodes which are heads
986 986 or ancestors of these heads.
987 987
988 988 All the ancestors of base are in self and in remote.
989 989 All the descendants of the list returned are missing in self.
990 990 (and so we know that the rest of the nodes are missing in remote, see
991 991 outgoing)
992 992 """
993 993 m = self.changelog.nodemap
994 994 search = []
995 995 fetch = {}
996 996 seen = {}
997 997 seenbranch = {}
998 998 if base == None:
999 999 base = {}
1000 1000
1001 1001 if not heads:
1002 1002 heads = remote.heads()
1003 1003
1004 1004 if self.changelog.tip() == nullid:
1005 1005 base[nullid] = 1
1006 1006 if heads != [nullid]:
1007 1007 return [nullid]
1008 1008 return []
1009 1009
1010 1010 # assume we're closer to the tip than the root
1011 1011 # and start by examining the heads
1012 1012 self.ui.status(_("searching for changes\n"))
1013 1013
1014 1014 unknown = []
1015 1015 for h in heads:
1016 1016 if h not in m:
1017 1017 unknown.append(h)
1018 1018 else:
1019 1019 base[h] = 1
1020 1020
1021 1021 if not unknown:
1022 1022 return []
1023 1023
1024 1024 req = dict.fromkeys(unknown)
1025 1025 reqcnt = 0
1026 1026
1027 1027 # search through remote branches
1028 1028 # a 'branch' here is a linear segment of history, with four parts:
1029 1029 # head, root, first parent, second parent
1030 1030 # (a branch always has two parents (or none) by definition)
1031 1031 unknown = remote.branches(unknown)
1032 1032 while unknown:
1033 1033 r = []
1034 1034 while unknown:
1035 1035 n = unknown.pop(0)
1036 1036 if n[0] in seen:
1037 1037 continue
1038 1038
1039 1039 self.ui.debug(_("examining %s:%s\n")
1040 1040 % (short(n[0]), short(n[1])))
1041 1041 if n[0] == nullid: # found the end of the branch
1042 1042 pass
1043 1043 elif n in seenbranch:
1044 1044 self.ui.debug(_("branch already found\n"))
1045 1045 continue
1046 1046 elif n[1] and n[1] in m: # do we know the base?
1047 1047 self.ui.debug(_("found incomplete branch %s:%s\n")
1048 1048 % (short(n[0]), short(n[1])))
1049 1049 search.append(n) # schedule branch range for scanning
1050 1050 seenbranch[n] = 1
1051 1051 else:
1052 1052 if n[1] not in seen and n[1] not in fetch:
1053 1053 if n[2] in m and n[3] in m:
1054 1054 self.ui.debug(_("found new changeset %s\n") %
1055 1055 short(n[1]))
1056 1056 fetch[n[1]] = 1 # earliest unknown
1057 1057 for p in n[2:4]:
1058 1058 if p in m:
1059 1059 base[p] = 1 # latest known
1060 1060
1061 1061 for p in n[2:4]:
1062 1062 if p not in req and p not in m:
1063 1063 r.append(p)
1064 1064 req[p] = 1
1065 1065 seen[n[0]] = 1
1066 1066
1067 1067 if r:
1068 1068 reqcnt += 1
1069 1069 self.ui.debug(_("request %d: %s\n") %
1070 1070 (reqcnt, " ".join(map(short, r))))
1071 1071 for p in range(0, len(r), 10):
1072 1072 for b in remote.branches(r[p:p+10]):
1073 1073 self.ui.debug(_("received %s:%s\n") %
1074 1074 (short(b[0]), short(b[1])))
1075 1075 unknown.append(b)
1076 1076
1077 1077 # do binary search on the branches we found
1078 1078 while search:
1079 1079 n = search.pop(0)
1080 1080 reqcnt += 1
1081 1081 l = remote.between([(n[0], n[1])])[0]
1082 1082 l.append(n[1])
1083 1083 p = n[0]
1084 1084 f = 1
1085 1085 for i in l:
1086 1086 self.ui.debug(_("narrowing %d:%d %s\n") % (f, len(l), short(i)))
1087 1087 if i in m:
1088 1088 if f <= 2:
1089 1089 self.ui.debug(_("found new branch changeset %s\n") %
1090 1090 short(p))
1091 1091 fetch[p] = 1
1092 1092 base[i] = 1
1093 1093 else:
1094 1094 self.ui.debug(_("narrowed branch search to %s:%s\n")
1095 1095 % (short(p), short(i)))
1096 1096 search.append((p, i))
1097 1097 break
1098 1098 p, f = i, f * 2
1099 1099
1100 1100 # sanity check our fetch list
1101 1101 for f in fetch.keys():
1102 1102 if f in m:
1103 1103 raise repo.RepoError(_("already have changeset ") + short(f[:4]))
1104 1104
1105 1105 if base.keys() == [nullid]:
1106 1106 if force:
1107 1107 self.ui.warn(_("warning: repository is unrelated\n"))
1108 1108 else:
1109 1109 raise util.Abort(_("repository is unrelated"))
1110 1110
1111 1111 self.ui.debug(_("found new changesets starting at ") +
1112 1112 " ".join([short(f) for f in fetch]) + "\n")
1113 1113
1114 1114 self.ui.debug(_("%d total queries\n") % reqcnt)
1115 1115
1116 1116 return fetch.keys()
1117 1117
1118 1118 def findoutgoing(self, remote, base=None, heads=None, force=False):
1119 1119 """Return list of nodes that are roots of subsets not in remote
1120 1120
1121 1121 If base dict is specified, assume that these nodes and their parents
1122 1122 exist on the remote side.
1123 1123 If a list of heads is specified, return only nodes which are heads
1124 1124 or ancestors of these heads, and return a second element which
1125 1125 contains all remote heads which get new children.
1126 1126 """
1127 1127 if base == None:
1128 1128 base = {}
1129 1129 self.findincoming(remote, base, heads, force=force)
1130 1130
1131 1131 self.ui.debug(_("common changesets up to ")
1132 1132 + " ".join(map(short, base.keys())) + "\n")
1133 1133
1134 1134 remain = dict.fromkeys(self.changelog.nodemap)
1135 1135
1136 1136 # prune everything remote has from the tree
1137 1137 del remain[nullid]
1138 1138 remove = base.keys()
1139 1139 while remove:
1140 1140 n = remove.pop(0)
1141 1141 if n in remain:
1142 1142 del remain[n]
1143 1143 for p in self.changelog.parents(n):
1144 1144 remove.append(p)
1145 1145
1146 1146 # find every node whose parents have been pruned
1147 1147 subset = []
1148 1148 # find every remote head that will get new children
1149 1149 updated_heads = {}
1150 1150 for n in remain:
1151 1151 p1, p2 = self.changelog.parents(n)
1152 1152 if p1 not in remain and p2 not in remain:
1153 1153 subset.append(n)
1154 1154 if heads:
1155 1155 if p1 in heads:
1156 1156 updated_heads[p1] = True
1157 1157 if p2 in heads:
1158 1158 updated_heads[p2] = True
1159 1159
1160 1160 # this is the set of all roots we have to push
1161 1161 if heads:
1162 1162 return subset, updated_heads.keys()
1163 1163 else:
1164 1164 return subset
1165 1165
1166 1166 def pull(self, remote, heads=None, force=False, lock=None):
1167 1167 mylock = False
1168 1168 if not lock:
1169 1169 lock = self.lock()
1170 1170 mylock = True
1171 1171
1172 1172 try:
1173 1173 fetch = self.findincoming(remote, force=force)
1174 1174 if fetch == [nullid]:
1175 1175 self.ui.status(_("requesting all changes\n"))
1176 1176
1177 1177 if not fetch:
1178 1178 self.ui.status(_("no changes found\n"))
1179 1179 return 0
1180 1180
1181 1181 if heads is None:
1182 1182 cg = remote.changegroup(fetch, 'pull')
1183 1183 else:
1184 1184 cg = remote.changegroupsubset(fetch, heads, 'pull')
1185 1185 return self.addchangegroup(cg, 'pull', remote.url())
1186 1186 finally:
1187 1187 if mylock:
1188 1188 lock.release()
1189 1189
1190 1190 def push(self, remote, force=False, revs=None):
1191 1191 # there are two ways to push to remote repo:
1192 1192 #
1193 1193 # addchangegroup assumes local user can lock remote
1194 1194 # repo (local filesystem, old ssh servers).
1195 1195 #
1196 1196 # unbundle assumes local user cannot lock remote repo (new ssh
1197 1197 # servers, http servers).
1198 1198
1199 1199 if remote.capable('unbundle'):
1200 1200 return self.push_unbundle(remote, force, revs)
1201 1201 return self.push_addchangegroup(remote, force, revs)
1202 1202
1203 1203 def prepush(self, remote, force, revs):
1204 1204 base = {}
1205 1205 remote_heads = remote.heads()
1206 1206 inc = self.findincoming(remote, base, remote_heads, force=force)
1207 1207 if not force and inc:
1208 1208 self.ui.warn(_("abort: unsynced remote changes!\n"))
1209 1209 self.ui.status(_("(did you forget to sync?"
1210 1210 " use push -f to force)\n"))
1211 1211 return None, 1
1212 1212
1213 1213 update, updated_heads = self.findoutgoing(remote, base, remote_heads)
1214 1214 if revs is not None:
1215 1215 msng_cl, bases, heads = self.changelog.nodesbetween(update, revs)
1216 1216 else:
1217 1217 bases, heads = update, self.changelog.heads()
1218 1218
1219 1219 if not bases:
1220 1220 self.ui.status(_("no changes found\n"))
1221 1221 return None, 1
1222 1222 elif not force:
1223 1223 # FIXME we don't properly detect creation of new heads
1224 1224 # in the push -r case, assume the user knows what he's doing
1225 1225 if not revs and len(remote_heads) < len(heads) \
1226 1226 and remote_heads != [nullid]:
1227 1227 self.ui.warn(_("abort: push creates new remote branches!\n"))
1228 1228 self.ui.status(_("(did you forget to merge?"
1229 1229 " use push -f to force)\n"))
1230 1230 return None, 1
1231 1231
1232 1232 if revs is None:
1233 1233 cg = self.changegroup(update, 'push')
1234 1234 else:
1235 1235 cg = self.changegroupsubset(update, revs, 'push')
1236 1236 return cg, remote_heads
1237 1237
1238 1238 def push_addchangegroup(self, remote, force, revs):
1239 1239 lock = remote.lock()
1240 1240
1241 1241 ret = self.prepush(remote, force, revs)
1242 1242 if ret[0] is not None:
1243 1243 cg, remote_heads = ret
1244 1244 return remote.addchangegroup(cg, 'push', self.url())
1245 1245 return ret[1]
1246 1246
1247 1247 def push_unbundle(self, remote, force, revs):
1248 1248 # local repo finds heads on server, finds out what revs it
1249 1249 # must push. once revs transferred, if server finds it has
1250 1250 # different heads (someone else won commit/push race), server
1251 1251 # aborts.
1252 1252
1253 1253 ret = self.prepush(remote, force, revs)
1254 1254 if ret[0] is not None:
1255 1255 cg, remote_heads = ret
1256 1256 if force: remote_heads = ['force']
1257 1257 return remote.unbundle(cg, remote_heads, 'push')
1258 1258 return ret[1]
1259 1259
1260 1260 def changegroupsubset(self, bases, heads, source):
1261 1261 """This function generates a changegroup consisting of all the nodes
1262 1262 that are descendents of any of the bases, and ancestors of any of
1263 1263 the heads.
1264 1264
1265 1265 It is fairly complex as determining which filenodes and which
1266 1266 manifest nodes need to be included for the changeset to be complete
1267 1267 is non-trivial.
1268 1268
1269 1269 Another wrinkle is doing the reverse, figuring out which changeset in
1270 1270 the changegroup a particular filenode or manifestnode belongs to."""
1271 1271
1272 1272 self.hook('preoutgoing', throw=True, source=source)
1273 1273
1274 1274 # Set up some initial variables
1275 1275 # Make it easy to refer to self.changelog
1276 1276 cl = self.changelog
1277 1277 # msng is short for missing - compute the list of changesets in this
1278 1278 # changegroup.
1279 1279 msng_cl_lst, bases, heads = cl.nodesbetween(bases, heads)
1280 1280 # Some bases may turn out to be superfluous, and some heads may be
1281 1281 # too. nodesbetween will return the minimal set of bases and heads
1282 1282 # necessary to re-create the changegroup.
1283 1283
1284 1284 # Known heads are the list of heads that it is assumed the recipient
1285 1285 # of this changegroup will know about.
1286 1286 knownheads = {}
1287 1287 # We assume that all parents of bases are known heads.
1288 1288 for n in bases:
1289 1289 for p in cl.parents(n):
1290 1290 if p != nullid:
1291 1291 knownheads[p] = 1
1292 1292 knownheads = knownheads.keys()
1293 1293 if knownheads:
1294 1294 # Now that we know what heads are known, we can compute which
1295 1295 # changesets are known. The recipient must know about all
1296 1296 # changesets required to reach the known heads from the null
1297 1297 # changeset.
1298 1298 has_cl_set, junk, junk = cl.nodesbetween(None, knownheads)
1299 1299 junk = None
1300 1300 # Transform the list into an ersatz set.
1301 1301 has_cl_set = dict.fromkeys(has_cl_set)
1302 1302 else:
1303 1303 # If there were no known heads, the recipient cannot be assumed to
1304 1304 # know about any changesets.
1305 1305 has_cl_set = {}
1306 1306
1307 1307 # Make it easy to refer to self.manifest
1308 1308 mnfst = self.manifest
1309 1309 # We don't know which manifests are missing yet
1310 1310 msng_mnfst_set = {}
1311 1311 # Nor do we know which filenodes are missing.
1312 1312 msng_filenode_set = {}
1313 1313
1314 1314 junk = mnfst.index[mnfst.count() - 1] # Get around a bug in lazyindex
1315 1315 junk = None
1316 1316
1317 1317 # A changeset always belongs to itself, so the changenode lookup
1318 1318 # function for a changenode is identity.
1319 1319 def identity(x):
1320 1320 return x
1321 1321
1322 1322 # A function generating function. Sets up an environment for the
1323 1323 # inner function.
1324 1324 def cmp_by_rev_func(revlog):
1325 1325 # Compare two nodes by their revision number in the environment's
1326 1326 # revision history. Since the revision number both represents the
1327 1327 # most efficient order to read the nodes in, and represents a
1328 1328 # topological sorting of the nodes, this function is often useful.
1329 1329 def cmp_by_rev(a, b):
1330 1330 return cmp(revlog.rev(a), revlog.rev(b))
1331 1331 return cmp_by_rev
1332 1332
1333 1333 # If we determine that a particular file or manifest node must be a
1334 1334 # node that the recipient of the changegroup will already have, we can
1335 1335 # also assume the recipient will have all the parents. This function
1336 1336 # prunes them from the set of missing nodes.
1337 1337 def prune_parents(revlog, hasset, msngset):
1338 1338 haslst = hasset.keys()
1339 1339 haslst.sort(cmp_by_rev_func(revlog))
1340 1340 for node in haslst:
1341 1341 parentlst = [p for p in revlog.parents(node) if p != nullid]
1342 1342 while parentlst:
1343 1343 n = parentlst.pop()
1344 1344 if n not in hasset:
1345 1345 hasset[n] = 1
1346 1346 p = [p for p in revlog.parents(n) if p != nullid]
1347 1347 parentlst.extend(p)
1348 1348 for n in hasset:
1349 1349 msngset.pop(n, None)
1350 1350
1351 1351 # This is a function generating function used to set up an environment
1352 1352 # for the inner function to execute in.
1353 1353 def manifest_and_file_collector(changedfileset):
1354 1354 # This is an information gathering function that gathers
1355 1355 # information from each changeset node that goes out as part of
1356 1356 # the changegroup. The information gathered is a list of which
1357 1357 # manifest nodes are potentially required (the recipient may
1358 1358 # already have them) and total list of all files which were
1359 1359 # changed in any changeset in the changegroup.
1360 1360 #
1361 1361 # We also remember the first changenode we saw any manifest
1362 1362 # referenced by so we can later determine which changenode 'owns'
1363 1363 # the manifest.
1364 1364 def collect_manifests_and_files(clnode):
1365 1365 c = cl.read(clnode)
1366 1366 for f in c[3]:
1367 1367 # This is to make sure we only have one instance of each
1368 1368 # filename string for each filename.
1369 1369 changedfileset.setdefault(f, f)
1370 1370 msng_mnfst_set.setdefault(c[0], clnode)
1371 1371 return collect_manifests_and_files
1372 1372
1373 1373 # Figure out which manifest nodes (of the ones we think might be part
1374 1374 # of the changegroup) the recipient must know about and remove them
1375 1375 # from the changegroup.
1376 1376 def prune_manifests():
1377 1377 has_mnfst_set = {}
1378 1378 for n in msng_mnfst_set:
1379 1379 # If a 'missing' manifest thinks it belongs to a changenode
1380 1380 # the recipient is assumed to have, obviously the recipient
1381 1381 # must have that manifest.
1382 1382 linknode = cl.node(mnfst.linkrev(n))
1383 1383 if linknode in has_cl_set:
1384 1384 has_mnfst_set[n] = 1
1385 1385 prune_parents(mnfst, has_mnfst_set, msng_mnfst_set)
1386 1386
1387 1387 # Use the information collected in collect_manifests_and_files to say
1388 1388 # which changenode any manifestnode belongs to.
1389 1389 def lookup_manifest_link(mnfstnode):
1390 1390 return msng_mnfst_set[mnfstnode]
1391 1391
1392 1392 # A function generating function that sets up the initial environment
1393 1393 # the inner function.
1394 1394 def filenode_collector(changedfiles):
1395 1395 next_rev = [0]
1396 1396 # This gathers information from each manifestnode included in the
1397 1397 # changegroup about which filenodes the manifest node references
1398 1398 # so we can include those in the changegroup too.
1399 1399 #
1400 1400 # It also remembers which changenode each filenode belongs to. It
1401 1401 # does this by assuming the a filenode belongs to the changenode
1402 1402 # the first manifest that references it belongs to.
1403 1403 def collect_msng_filenodes(mnfstnode):
1404 1404 r = mnfst.rev(mnfstnode)
1405 1405 if r == next_rev[0]:
1406 1406 # If the last rev we looked at was the one just previous,
1407 1407 # we only need to see a diff.
1408 1408 delta = mdiff.patchtext(mnfst.delta(mnfstnode))
1409 1409 # For each line in the delta
1410 1410 for dline in delta.splitlines():
1411 1411 # get the filename and filenode for that line
1412 1412 f, fnode = dline.split('\0')
1413 1413 fnode = bin(fnode[:40])
1414 1414 f = changedfiles.get(f, None)
1415 1415 # And if the file is in the list of files we care
1416 1416 # about.
1417 1417 if f is not None:
1418 1418 # Get the changenode this manifest belongs to
1419 1419 clnode = msng_mnfst_set[mnfstnode]
1420 1420 # Create the set of filenodes for the file if
1421 1421 # there isn't one already.
1422 1422 ndset = msng_filenode_set.setdefault(f, {})
1423 1423 # And set the filenode's changelog node to the
1424 1424 # manifest's if it hasn't been set already.
1425 1425 ndset.setdefault(fnode, clnode)
1426 1426 else:
1427 1427 # Otherwise we need a full manifest.
1428 1428 m = mnfst.read(mnfstnode)
1429 1429 # For every file in we care about.
1430 1430 for f in changedfiles:
1431 1431 fnode = m.get(f, None)
1432 1432 # If it's in the manifest
1433 1433 if fnode is not None:
1434 1434 # See comments above.
1435 1435 clnode = msng_mnfst_set[mnfstnode]
1436 1436 ndset = msng_filenode_set.setdefault(f, {})
1437 1437 ndset.setdefault(fnode, clnode)
1438 1438 # Remember the revision we hope to see next.
1439 1439 next_rev[0] = r + 1
1440 1440 return collect_msng_filenodes
1441 1441
1442 1442 # We have a list of filenodes we think we need for a file, lets remove
1443 1443 # all those we now the recipient must have.
1444 1444 def prune_filenodes(f, filerevlog):
1445 1445 msngset = msng_filenode_set[f]
1446 1446 hasset = {}
1447 1447 # If a 'missing' filenode thinks it belongs to a changenode we
1448 1448 # assume the recipient must have, then the recipient must have
1449 1449 # that filenode.
1450 1450 for n in msngset:
1451 1451 clnode = cl.node(filerevlog.linkrev(n))
1452 1452 if clnode in has_cl_set:
1453 1453 hasset[n] = 1
1454 1454 prune_parents(filerevlog, hasset, msngset)
1455 1455
1456 1456 # A function generator function that sets up the a context for the
1457 1457 # inner function.
1458 1458 def lookup_filenode_link_func(fname):
1459 1459 msngset = msng_filenode_set[fname]
1460 1460 # Lookup the changenode the filenode belongs to.
1461 1461 def lookup_filenode_link(fnode):
1462 1462 return msngset[fnode]
1463 1463 return lookup_filenode_link
1464 1464
1465 1465 # Now that we have all theses utility functions to help out and
1466 1466 # logically divide up the task, generate the group.
1467 1467 def gengroup():
1468 1468 # The set of changed files starts empty.
1469 1469 changedfiles = {}
1470 1470 # Create a changenode group generator that will call our functions
1471 1471 # back to lookup the owning changenode and collect information.
1472 1472 group = cl.group(msng_cl_lst, identity,
1473 1473 manifest_and_file_collector(changedfiles))
1474 1474 for chnk in group:
1475 1475 yield chnk
1476 1476
1477 1477 # The list of manifests has been collected by the generator
1478 1478 # calling our functions back.
1479 1479 prune_manifests()
1480 1480 msng_mnfst_lst = msng_mnfst_set.keys()
1481 1481 # Sort the manifestnodes by revision number.
1482 1482 msng_mnfst_lst.sort(cmp_by_rev_func(mnfst))
1483 1483 # Create a generator for the manifestnodes that calls our lookup
1484 1484 # and data collection functions back.
1485 1485 group = mnfst.group(msng_mnfst_lst, lookup_manifest_link,
1486 1486 filenode_collector(changedfiles))
1487 1487 for chnk in group:
1488 1488 yield chnk
1489 1489
1490 1490 # These are no longer needed, dereference and toss the memory for
1491 1491 # them.
1492 1492 msng_mnfst_lst = None
1493 1493 msng_mnfst_set.clear()
1494 1494
1495 1495 changedfiles = changedfiles.keys()
1496 1496 changedfiles.sort()
1497 1497 # Go through all our files in order sorted by name.
1498 1498 for fname in changedfiles:
1499 1499 filerevlog = self.file(fname)
1500 1500 # Toss out the filenodes that the recipient isn't really
1501 1501 # missing.
1502 1502 if msng_filenode_set.has_key(fname):
1503 1503 prune_filenodes(fname, filerevlog)
1504 1504 msng_filenode_lst = msng_filenode_set[fname].keys()
1505 1505 else:
1506 1506 msng_filenode_lst = []
1507 1507 # If any filenodes are left, generate the group for them,
1508 1508 # otherwise don't bother.
1509 1509 if len(msng_filenode_lst) > 0:
1510 1510 yield changegroup.genchunk(fname)
1511 1511 # Sort the filenodes by their revision #
1512 1512 msng_filenode_lst.sort(cmp_by_rev_func(filerevlog))
1513 1513 # Create a group generator and only pass in a changenode
1514 1514 # lookup function as we need to collect no information
1515 1515 # from filenodes.
1516 1516 group = filerevlog.group(msng_filenode_lst,
1517 1517 lookup_filenode_link_func(fname))
1518 1518 for chnk in group:
1519 1519 yield chnk
1520 1520 if msng_filenode_set.has_key(fname):
1521 1521 # Don't need this anymore, toss it to free memory.
1522 1522 del msng_filenode_set[fname]
1523 1523 # Signal that no more groups are left.
1524 1524 yield changegroup.closechunk()
1525 1525
1526 1526 if msng_cl_lst:
1527 1527 self.hook('outgoing', node=hex(msng_cl_lst[0]), source=source)
1528 1528
1529 1529 return util.chunkbuffer(gengroup())
1530 1530
1531 1531 def changegroup(self, basenodes, source):
1532 1532 """Generate a changegroup of all nodes that we have that a recipient
1533 1533 doesn't.
1534 1534
1535 1535 This is much easier than the previous function as we can assume that
1536 1536 the recipient has any changenode we aren't sending them."""
1537 1537
1538 1538 self.hook('preoutgoing', throw=True, source=source)
1539 1539
1540 1540 cl = self.changelog
1541 1541 nodes = cl.nodesbetween(basenodes, None)[0]
1542 1542 revset = dict.fromkeys([cl.rev(n) for n in nodes])
1543 1543
1544 1544 def identity(x):
1545 1545 return x
1546 1546
1547 1547 def gennodelst(revlog):
1548 1548 for r in xrange(0, revlog.count()):
1549 1549 n = revlog.node(r)
1550 1550 if revlog.linkrev(n) in revset:
1551 1551 yield n
1552 1552
1553 1553 def changed_file_collector(changedfileset):
1554 1554 def collect_changed_files(clnode):
1555 1555 c = cl.read(clnode)
1556 1556 for fname in c[3]:
1557 1557 changedfileset[fname] = 1
1558 1558 return collect_changed_files
1559 1559
1560 1560 def lookuprevlink_func(revlog):
1561 1561 def lookuprevlink(n):
1562 1562 return cl.node(revlog.linkrev(n))
1563 1563 return lookuprevlink
1564 1564
1565 1565 def gengroup():
1566 1566 # construct a list of all changed files
1567 1567 changedfiles = {}
1568 1568
1569 1569 for chnk in cl.group(nodes, identity,
1570 1570 changed_file_collector(changedfiles)):
1571 1571 yield chnk
1572 1572 changedfiles = changedfiles.keys()
1573 1573 changedfiles.sort()
1574 1574
1575 1575 mnfst = self.manifest
1576 1576 nodeiter = gennodelst(mnfst)
1577 1577 for chnk in mnfst.group(nodeiter, lookuprevlink_func(mnfst)):
1578 1578 yield chnk
1579 1579
1580 1580 for fname in changedfiles:
1581 1581 filerevlog = self.file(fname)
1582 1582 nodeiter = gennodelst(filerevlog)
1583 1583 nodeiter = list(nodeiter)
1584 1584 if nodeiter:
1585 1585 yield changegroup.genchunk(fname)
1586 1586 lookup = lookuprevlink_func(filerevlog)
1587 1587 for chnk in filerevlog.group(nodeiter, lookup):
1588 1588 yield chnk
1589 1589
1590 1590 yield changegroup.closechunk()
1591 1591
1592 1592 if nodes:
1593 1593 self.hook('outgoing', node=hex(nodes[0]), source=source)
1594 1594
1595 1595 return util.chunkbuffer(gengroup())
1596 1596
1597 1597 def addchangegroup(self, source, srctype, url):
1598 1598 """add changegroup to repo.
1599 1599 returns number of heads modified or added + 1."""
1600 1600
1601 1601 def csmap(x):
1602 1602 self.ui.debug(_("add changeset %s\n") % short(x))
1603 1603 return cl.count()
1604 1604
1605 1605 def revmap(x):
1606 1606 return cl.rev(x)
1607 1607
1608 1608 if not source:
1609 1609 return 0
1610 1610
1611 1611 self.hook('prechangegroup', throw=True, source=srctype, url=url)
1612 1612
1613 1613 changesets = files = revisions = 0
1614 1614
1615 1615 tr = self.transaction()
1616 1616
1617 1617 # write changelog data to temp files so concurrent readers will not see
1618 1618 # inconsistent view
1619 1619 cl = None
1620 1620 try:
1621 1621 cl = appendfile.appendchangelog(self.opener, self.changelog.version)
1622 1622
1623 1623 oldheads = len(cl.heads())
1624 1624
1625 1625 # pull off the changeset group
1626 1626 self.ui.status(_("adding changesets\n"))
1627 1627 cor = cl.count() - 1
1628 1628 chunkiter = changegroup.chunkiter(source)
1629 1629 if cl.addgroup(chunkiter, csmap, tr, 1) is None:
1630 1630 raise util.Abort(_("received changelog group is empty"))
1631 1631 cnr = cl.count() - 1
1632 1632 changesets = cnr - cor
1633 1633
1634 1634 # pull off the manifest group
1635 1635 self.ui.status(_("adding manifests\n"))
1636 1636 chunkiter = changegroup.chunkiter(source)
1637 1637 # no need to check for empty manifest group here:
1638 1638 # if the result of the merge of 1 and 2 is the same in 3 and 4,
1639 1639 # no new manifest will be created and the manifest group will
1640 1640 # be empty during the pull
1641 1641 self.manifest.addgroup(chunkiter, revmap, tr)
1642 1642
1643 1643 # process the files
1644 1644 self.ui.status(_("adding file changes\n"))
1645 1645 while 1:
1646 1646 f = changegroup.getchunk(source)
1647 1647 if not f:
1648 1648 break
1649 1649 self.ui.debug(_("adding %s revisions\n") % f)
1650 1650 fl = self.file(f)
1651 1651 o = fl.count()
1652 1652 chunkiter = changegroup.chunkiter(source)
1653 1653 if fl.addgroup(chunkiter, revmap, tr) is None:
1654 1654 raise util.Abort(_("received file revlog group is empty"))
1655 1655 revisions += fl.count() - o
1656 1656 files += 1
1657 1657
1658 1658 cl.writedata()
1659 1659 finally:
1660 1660 if cl:
1661 1661 cl.cleanup()
1662 1662
1663 1663 # make changelog see real files again
1664 1664 self.changelog = changelog.changelog(self.opener, self.changelog.version)
1665 1665 self.changelog.checkinlinesize(tr)
1666 1666
1667 1667 newheads = len(self.changelog.heads())
1668 1668 heads = ""
1669 1669 if oldheads and newheads != oldheads:
1670 1670 heads = _(" (%+d heads)") % (newheads - oldheads)
1671 1671
1672 1672 self.ui.status(_("added %d changesets"
1673 1673 " with %d changes to %d files%s\n")
1674 1674 % (changesets, revisions, files, heads))
1675 1675
1676 1676 if changesets > 0:
1677 1677 self.hook('pretxnchangegroup', throw=True,
1678 1678 node=hex(self.changelog.node(cor+1)), source=srctype,
1679 1679 url=url)
1680 1680
1681 1681 tr.close()
1682 1682
1683 1683 if changesets > 0:
1684 1684 self.hook("changegroup", node=hex(self.changelog.node(cor+1)),
1685 1685 source=srctype, url=url)
1686 1686
1687 1687 for i in range(cor + 1, cnr + 1):
1688 1688 self.hook("incoming", node=hex(self.changelog.node(i)),
1689 1689 source=srctype, url=url)
1690 1690
1691 1691 return newheads - oldheads + 1
1692 1692
1693 1693
1694 1694 def stream_in(self, remote):
1695 1695 fp = remote.stream_out()
1696 1696 resp = int(fp.readline())
1697 1697 if resp != 0:
1698 1698 raise util.Abort(_('operation forbidden by server'))
1699 1699 self.ui.status(_('streaming all changes\n'))
1700 1700 total_files, total_bytes = map(int, fp.readline().split(' ', 1))
1701 1701 self.ui.status(_('%d files to transfer, %s of data\n') %
1702 1702 (total_files, util.bytecount(total_bytes)))
1703 1703 start = time.time()
1704 1704 for i in xrange(total_files):
1705 1705 name, size = fp.readline().split('\0', 1)
1706 1706 size = int(size)
1707 1707 self.ui.debug('adding %s (%s)\n' % (name, util.bytecount(size)))
1708 1708 ofp = self.opener(name, 'w')
1709 1709 for chunk in util.filechunkiter(fp, limit=size):
1710 1710 ofp.write(chunk)
1711 1711 ofp.close()
1712 1712 elapsed = time.time() - start
1713 1713 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
1714 1714 (util.bytecount(total_bytes), elapsed,
1715 1715 util.bytecount(total_bytes / elapsed)))
1716 1716 self.reload()
1717 1717 return len(self.heads()) + 1
1718 1718
1719 1719 def clone(self, remote, heads=[], stream=False):
1720 1720 '''clone remote repository.
1721 1721
1722 1722 keyword arguments:
1723 1723 heads: list of revs to clone (forces use of pull)
1724 1724 stream: use streaming clone if possible'''
1725 1725
1726 1726 # now, all clients that can request uncompressed clones can
1727 1727 # read repo formats supported by all servers that can serve
1728 1728 # them.
1729 1729
1730 1730 # if revlog format changes, client will have to check version
1731 1731 # and format flags on "stream" capability, and use
1732 1732 # uncompressed only if compatible.
1733 1733
1734 1734 if stream and not heads and remote.capable('stream'):
1735 1735 return self.stream_in(remote)
1736 1736 return self.pull(remote, heads)
1737 1737
1738 1738 # used to avoid circular references so destructors work
1739 1739 def aftertrans(base):
1740 1740 p = base
1741 1741 def a():
1742 1742 util.rename(os.path.join(p, "journal"), os.path.join(p, "undo"))
1743 1743 util.rename(os.path.join(p, "journal.dirstate"),
1744 1744 os.path.join(p, "undo.dirstate"))
1745 1745 return a
1746 1746
1747 1747 def instance(ui, path, create):
1748 1748 return localrepository(ui, util.drop_scheme('file', path), create)
1749 1749
1750 1750 def islocal(path):
1751 1751 return True
General Comments 0
You need to be logged in to leave comments. Login now