##// END OF EJS Templates
[PATCH] rawcommit fix again...
mpm@selenic.com -
r403:fda7bb48 default
parent child Browse files
Show More
@@ -1,811 +1,811 b''
1 1 # commands.py - command processing for mercurial
2 2 #
3 3 # Copyright 2005 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 import os, re, sys, signal
9 9 import fancyopts, ui, hg
10 10 from demandload import *
11 11 demandload(globals(), "mdiff time hgweb traceback random signal errno")
12 12
13 13 class UnknownCommand(Exception): pass
14 14
15 15 def filterfiles(filters, files):
16 16 l = [ x for x in files if x in filters ]
17 17
18 18 for t in filters:
19 19 if t and t[-1] != os.sep: t += os.sep
20 20 l += [ x for x in files if x.startswith(t) ]
21 21 return l
22 22
23 23 def relfilter(repo, files):
24 24 if os.getcwd() != repo.root:
25 25 p = os.getcwd()[len(repo.root) + 1: ]
26 26 return filterfiles([p], files)
27 27 return files
28 28
29 29 def relpath(repo, args):
30 30 if os.getcwd() != repo.root:
31 31 p = os.getcwd()[len(repo.root) + 1: ]
32 32 return [ os.path.normpath(os.path.join(p, x)) for x in args ]
33 33 return args
34 34
35 35 def dodiff(ui, repo, path, files = None, node1 = None, node2 = None):
36 36 def date(c):
37 37 return time.asctime(time.gmtime(float(c[2].split(' ')[0])))
38 38
39 39 if node2:
40 40 change = repo.changelog.read(node2)
41 41 mmap2 = repo.manifest.read(change[0])
42 42 (c, a, d) = repo.diffrevs(node1, node2)
43 43 def read(f): return repo.file(f).read(mmap2[f])
44 44 date2 = date(change)
45 45 else:
46 46 date2 = time.asctime()
47 47 (c, a, d, u) = repo.diffdir(path, node1)
48 48 if not node1:
49 49 node1 = repo.dirstate.parents()[0]
50 50 def read(f): return file(os.path.join(repo.root, f)).read()
51 51
52 52 if ui.quiet:
53 53 r = None
54 54 else:
55 55 hexfunc = ui.verbose and hg.hex or hg.short
56 56 r = [hexfunc(node) for node in [node1, node2] if node]
57 57
58 58 change = repo.changelog.read(node1)
59 59 mmap = repo.manifest.read(change[0])
60 60 date1 = date(change)
61 61
62 62 if files:
63 63 c, a, d = map(lambda x: filterfiles(files, x), (c, a, d))
64 64
65 65 for f in c:
66 66 to = None
67 67 if f in mmap:
68 68 to = repo.file(f).read(mmap[f])
69 69 tn = read(f)
70 70 sys.stdout.write(mdiff.unidiff(to, date1, tn, date2, f, r))
71 71 for f in a:
72 72 to = None
73 73 tn = read(f)
74 74 sys.stdout.write(mdiff.unidiff(to, date1, tn, date2, f, r))
75 75 for f in d:
76 76 to = repo.file(f).read(mmap[f])
77 77 tn = None
78 78 sys.stdout.write(mdiff.unidiff(to, date1, tn, date2, f, r))
79 79
80 80 def show_changeset(ui, repo, rev=0, changenode=None, filelog=None):
81 81 """show a single changeset or file revision"""
82 82 changelog = repo.changelog
83 83 if filelog:
84 84 log = filelog
85 85 filerev = rev
86 86 node = filenode = filelog.node(filerev)
87 87 changerev = filelog.linkrev(filenode)
88 88 changenode = changenode or changelog.node(changerev)
89 89 else:
90 90 log = changelog
91 91 changerev = rev
92 92 if changenode is None:
93 93 changenode = changelog.node(changerev)
94 94 elif not changerev:
95 95 rev = changerev = changelog.rev(changenode)
96 96 node = changenode
97 97
98 98 if ui.quiet:
99 99 ui.write("%d:%s\n" % (rev, hg.hex(node)))
100 100 return
101 101
102 102 changes = changelog.read(changenode)
103 103
104 104 parents = [(log.rev(parent), hg.hex(parent))
105 105 for parent in log.parents(node)
106 106 if ui.debugflag or parent != hg.nullid]
107 107 if not ui.debugflag and len(parents) == 1 and parents[0][0] == rev-1:
108 108 parents = []
109 109
110 110 if filelog:
111 111 ui.write("revision: %d:%s\n" % (filerev, hg.hex(filenode)))
112 112 for parent in parents:
113 113 ui.write("parent: %d:%s\n" % parent)
114 114 ui.status("changeset: %d:%s\n" % (changerev, hg.hex(changenode)))
115 115 else:
116 116 ui.write("changeset: %d:%s\n" % (changerev, hg.hex(changenode)))
117 117 for tag in repo.nodetags(changenode):
118 118 ui.status("tag: %s\n" % tag)
119 119 for parent in parents:
120 120 ui.write("parent: %d:%s\n" % parent)
121 121 ui.note("manifest: %d:%s\n" % (repo.manifest.rev(changes[0]),
122 122 hg.hex(changes[0])))
123 123 ui.status("user: %s\n" % changes[1])
124 124 ui.status("date: %s\n" % time.asctime(
125 125 time.localtime(float(changes[2].split(' ')[0]))))
126 126 ui.note("files: %s\n" % " ".join(changes[3]))
127 127 description = changes[4].strip()
128 128 if description:
129 129 if ui.verbose:
130 130 ui.status("description:\n")
131 131 ui.status(description)
132 132 ui.status("\n")
133 133 else:
134 134 ui.status("summary: %s\n" % description.splitlines()[0])
135 135 ui.status("\n")
136 136
137 137 def help(ui, cmd=None):
138 138 '''show help for a given command or all commands'''
139 139 if cmd:
140 140 try:
141 141 i = find(cmd)
142 142 ui.write("%s\n\n" % i[2])
143 143
144 144 if i[1]:
145 145 for s, l, d, c in i[1]:
146 146 opt=' '
147 147 if s: opt = opt + '-' + s + ' '
148 148 if l: opt = opt + '--' + l + ' '
149 149 if d: opt = opt + '(' + str(d) + ')'
150 150 ui.write(opt, "\n")
151 151 if c: ui.write(' %s\n' % c)
152 152 ui.write("\n")
153 153
154 154 ui.write(i[0].__doc__, "\n")
155 155 except UnknownCommand:
156 156 ui.warn("hg: unknown command %s\n" % cmd)
157 157 sys.exit(0)
158 158 else:
159 159 ui.status('hg commands:\n\n')
160 160
161 161 h = {}
162 162 for e in table.values():
163 163 f = e[0]
164 164 if f.__name__.startswith("debug"): continue
165 165 d = ""
166 166 if f.__doc__:
167 167 d = f.__doc__.splitlines(0)[0].rstrip()
168 168 h[f.__name__] = d
169 169
170 170 fns = h.keys()
171 171 fns.sort()
172 172 m = max(map(len, fns))
173 173 for f in fns:
174 174 ui.status(' %-*s %s\n' % (m, f, h[f]))
175 175
176 176 # Commands start here, listed alphabetically
177 177
178 178 def add(ui, repo, file, *files):
179 179 '''add the specified files on the next commit'''
180 180 repo.add(relpath(repo, (file,) + files))
181 181
182 182 def addremove(ui, repo, *files):
183 183 """add all new files, delete all missing files"""
184 184 if files:
185 185 files = relpath(repo, files)
186 186 d = []
187 187 u = []
188 188 for f in files:
189 189 p = repo.wjoin(f)
190 190 s = repo.dirstate.state(f)
191 191 isfile = os.path.isfile(p)
192 192 if s != 'r' and not isfile:
193 193 d.append(f)
194 194 elif s not in 'nmai' and isfile:
195 195 u.append(f)
196 196 else:
197 197 (c, a, d, u) = repo.diffdir(repo.root)
198 198 repo.add(u)
199 199 repo.remove(d)
200 200
201 201 def annotate(u, repo, file, *files, **ops):
202 202 """show changeset information per file line"""
203 203 def getnode(rev):
204 204 return hg.short(repo.changelog.node(rev))
205 205
206 206 def getname(rev):
207 207 try:
208 208 return bcache[rev]
209 209 except KeyError:
210 210 cl = repo.changelog.read(repo.changelog.node(rev))
211 211 name = cl[1]
212 212 f = name.find('@')
213 213 if f >= 0:
214 214 name = name[:f]
215 215 bcache[rev] = name
216 216 return name
217 217
218 218 bcache = {}
219 219 opmap = [['user', getname], ['number', str], ['changeset', getnode]]
220 220 if not ops['user'] and not ops['changeset']:
221 221 ops['number'] = 1
222 222
223 223 node = repo.dirstate.parents()[0]
224 224 if ops['revision']:
225 225 node = repo.changelog.lookup(ops['revision'])
226 226 change = repo.changelog.read(node)
227 227 mmap = repo.manifest.read(change[0])
228 228 maxuserlen = 0
229 229 maxchangelen = 0
230 230 for f in relpath(repo, (file,) + files):
231 231 lines = repo.file(f).annotate(mmap[f])
232 232 pieces = []
233 233
234 234 for o, f in opmap:
235 235 if ops[o]:
236 236 l = [ f(n) for n,t in lines ]
237 237 m = max(map(len, l))
238 238 pieces.append([ "%*s" % (m, x) for x in l])
239 239
240 240 for p,l in zip(zip(*pieces), lines):
241 241 u.write(" ".join(p) + ": " + l[1])
242 242
243 243 def cat(ui, repo, file, rev = []):
244 244 """output the latest or given revision of a file"""
245 245 r = repo.file(relpath(repo, [file])[0])
246 246 n = r.tip()
247 247 if rev: n = r.lookup(rev)
248 248 sys.stdout.write(r.read(n))
249 249
250 250 def commit(ui, repo, *files, **opts):
251 251 """commit the specified files or all outstanding changes"""
252 252 text = opts['text']
253 253 if not text and opts['logfile']:
254 254 try: text = open(opts['logfile']).read()
255 255 except IOError: pass
256 256
257 257 if opts['addremove']:
258 258 addremove(ui, repo, *files)
259 259 repo.commit(relpath(repo, files), text, opts['user'], opts['date'])
260 260
261 261 def copy(ui, repo, source, dest):
262 262 """mark a file as copied or renamed for the next commit"""
263 263 return repo.copy(*relpath(repo, (source, dest)))
264 264
265 265 def debugaddchangegroup(ui, repo):
266 266 data = sys.stdin.read()
267 267 repo.addchangegroup(data)
268 268
269 269 def debugchangegroup(ui, repo, roots):
270 270 newer = repo.newer(map(repo.lookup, roots))
271 271 for chunk in repo.changegroup(newer):
272 272 sys.stdout.write(chunk)
273 273
274 274 def debugindex(ui, file):
275 275 r = hg.revlog(open, file, "")
276 276 print " rev offset length base linkrev"+\
277 277 " p1 p2 nodeid"
278 278 for i in range(r.count()):
279 279 e = r.index[i]
280 280 print "% 6d % 9d % 7d % 6d % 7d %s.. %s.. %s.." % (
281 281 i, e[0], e[1], e[2], e[3],
282 282 hg.hex(e[4][:5]), hg.hex(e[5][:5]), hg.hex(e[6][:5]))
283 283
284 284 def debugindexdot(ui, file):
285 285 r = hg.revlog(open, file, "")
286 286 print "digraph G {"
287 287 for i in range(r.count()):
288 288 e = r.index[i]
289 289 print "\t%d -> %d" % (r.rev(e[4]), i)
290 290 if e[5] != hg.nullid:
291 291 print "\t%d -> %d" % (r.rev(e[5]), i)
292 292 print "}"
293 293
294 294 def diff(ui, repo, *files, **opts):
295 295 """diff working directory (or selected files)"""
296 296 revs = []
297 297 if opts['rev']:
298 298 revs = map(lambda x: repo.lookup(x), opts['rev'])
299 299
300 300 if len(revs) > 2:
301 301 self.ui.warn("too many revisions to diff\n")
302 302 sys.exit(1)
303 303
304 304 if files:
305 305 files = relpath(repo, files)
306 306 else:
307 307 files = relpath(repo, [""])
308 308
309 309 dodiff(ui, repo, os.getcwd(), files, *revs)
310 310
311 311 def export(ui, repo, changeset):
312 312 """dump the changeset header and diffs for a revision"""
313 313 node = repo.lookup(changeset)
314 314 prev, other = repo.changelog.parents(node)
315 315 change = repo.changelog.read(node)
316 316 print "# HG changeset patch"
317 317 print "# User %s" % change[1]
318 318 print "# Node ID %s" % hg.hex(node)
319 319 print "# Parent %s" % hg.hex(prev)
320 320 print
321 321 if other != hg.nullid:
322 322 print "# Parent %s" % hg.hex(other)
323 323 print change[4].rstrip()
324 324 print
325 325
326 326 dodiff(ui, repo, "", None, prev, node)
327 327
328 328 def forget(ui, repo, file, *files):
329 329 """don't add the specified files on the next commit"""
330 330 repo.forget(relpath(repo, (file,) + files))
331 331
332 332 def heads(ui, repo):
333 333 """show current repository heads"""
334 334 for n in repo.changelog.heads():
335 335 show_changeset(ui, repo, changenode=n)
336 336
337 337 def history(ui, repo):
338 338 """show the changelog history"""
339 339 for i in range(repo.changelog.count() - 1, -1, -1):
340 340 show_changeset(ui, repo, rev=i)
341 341
342 342 def identify(ui, repo):
343 343 """print information about the working copy"""
344 344 parents = [p for p in repo.dirstate.parents() if p != hg.nullid]
345 345 if not parents:
346 346 ui.write("unknown\n")
347 347 return
348 348
349 349 hexfunc = ui.verbose and hg.hex or hg.short
350 350 (c, a, d, u) = repo.diffdir(repo.root)
351 351 output = ["%s%s" % ('+'.join([hexfunc(parent) for parent in parents]),
352 352 (c or a or d) and "+" or "")]
353 353
354 354 if not ui.quiet:
355 355 # multiple tags for a single parent separated by '/'
356 356 parenttags = ['/'.join(tags)
357 357 for tags in map(repo.nodetags, parents) if tags]
358 358 # tags for multiple parents separated by ' + '
359 359 output.append(' + '.join(parenttags))
360 360
361 361 ui.write("%s\n" % ' '.join(output))
362 362
363 363 def init(ui, source=None, **opts):
364 364 """create a new repository or copy an existing one"""
365 365
366 366 if source:
367 367 paths = {}
368 368 for name, path in ui.configitems("paths"):
369 369 paths[name] = path
370 370
371 371 if source in paths: source = paths[source]
372 372
373 373 link = 0
374 374 if not source.startswith("http://"):
375 375 d1 = os.stat(os.getcwd()).st_dev
376 376 d2 = os.stat(source).st_dev
377 377 if d1 == d2: link = 1
378 378
379 379 if link:
380 380 ui.debug("copying by hardlink\n")
381 381 os.system("cp -al %s/.hg .hg" % source)
382 382 try:
383 383 os.remove(".hg/dirstate")
384 384 except: pass
385 385
386 386 repo = hg.repository(ui, ".")
387 387
388 388 else:
389 389 repo = hg.repository(ui, ".", create=1)
390 390 other = hg.repository(ui, source)
391 391 cg = repo.getchangegroup(other)
392 392 repo.addchangegroup(cg)
393 393
394 394 f = repo.opener("hgrc", "w")
395 395 f.write("[paths]\n")
396 396 f.write("default = %s\n" % source)
397 397
398 398 if opts['update']:
399 399 update(ui, repo)
400 400 else:
401 401 repo = hg.repository(ui, ".", create=1)
402 402
403 403 def log(ui, repo, f):
404 404 """show the revision history of a single file"""
405 405 f = relpath(repo, [f])[0]
406 406
407 407 r = repo.file(f)
408 408 for i in range(r.count() - 1, -1, -1):
409 409 show_changeset(ui, repo, filelog=r, rev=i)
410 410
411 411 def manifest(ui, repo, rev = []):
412 412 """output the latest or given revision of the project manifest"""
413 413 n = repo.manifest.tip()
414 414 if rev:
415 415 n = repo.manifest.lookup(rev)
416 416 m = repo.manifest.read(n)
417 417 mf = repo.manifest.readflags(n)
418 418 files = m.keys()
419 419 files.sort()
420 420
421 421 for f in files:
422 422 ui.write("%40s %3s %s\n" % (hg.hex(m[f]), mf[f] and "755" or "644", f))
423 423
424 424 def parents(ui, repo, node = None):
425 425 '''show the parents of the current working dir'''
426 426 if node:
427 427 p = repo.changelog.parents(repo.lookup(hg.bin(node)))
428 428 else:
429 429 p = repo.dirstate.parents()
430 430
431 431 for n in p:
432 432 if n != hg.nullid:
433 433 show_changeset(ui, repo, changenode=n)
434 434
435 435 def patch(ui, repo, patch1, *patches, **opts):
436 436 """import an ordered set of patches"""
437 437 try:
438 438 import psyco
439 439 psyco.full()
440 440 except:
441 441 pass
442 442
443 443 patches = (patch1,) + patches
444 444
445 445 d = opts["base"]
446 446 strip = opts["strip"]
447 447 quiet = opts["quiet"] and "> /dev/null" or ""
448 448
449 449 for patch in patches:
450 450 ui.status("applying %s\n" % patch)
451 451 pf = os.path.join(d, patch)
452 452
453 453 text = ""
454 454 for l in file(pf):
455 455 if l[:4] == "--- ": break
456 456 text += l
457 457
458 458 # make sure text isn't empty
459 459 if not text: text = "imported patch %s\n" % patch
460 460
461 461 f = os.popen("patch -p%d < %s" % (strip, pf))
462 462 files = []
463 463 for l in f.read().splitlines():
464 464 l.rstrip('\r\n');
465 465 if not quiet:
466 466 print l
467 467 if l[:14] == 'patching file ':
468 468 files.append(l[14:])
469 469 f.close()
470 470
471 471 if len(files) > 0:
472 472 addremove(ui, repo, *files)
473 473 repo.commit(files, text)
474 474
475 475 def pull(ui, repo, source="default"):
476 476 """pull changes from the specified source"""
477 477 paths = {}
478 478 for name, path in ui.configitems("paths"):
479 479 paths[name] = path
480 480
481 481 if source in paths: source = paths[source]
482 482
483 483 other = hg.repository(ui, source)
484 484 cg = repo.getchangegroup(other)
485 485 repo.addchangegroup(cg)
486 486
487 487 def push(ui, repo, dest="default-push"):
488 488 """push changes to the specified destination"""
489 489 paths = {}
490 490 for name, path in ui.configitems("paths"):
491 491 paths[name] = path
492 492
493 493 if dest in paths: dest = paths[dest]
494 494
495 495 if not dest.startswith("ssh://"):
496 496 ui.warn("abort: can only push to ssh:// destinations currently\n")
497 497 return 1
498 498
499 499 m = re.match(r'ssh://(([^@]+)@)?([^:/]+)(:(\d+))?(/(.*))?', dest)
500 500 if not m:
501 501 ui.warn("abort: couldn't parse destination %s\n" % dest)
502 502 return 1
503 503
504 504 user, host, port, path = map(m.group, (2, 3, 5, 7))
505 505 host = user and ("%s@%s" % (user, host)) or host
506 506 port = port and (" -p %s") % port or ""
507 507 path = path or ""
508 508
509 509 sport = random.randrange(30000, 60000)
510 510 cmd = "ssh %s%s -R %d:localhost:%d 'cd %s; hg pull http://localhost:%d/'"
511 511 cmd = cmd % (host, port, sport+1, sport, path, sport+1)
512 512
513 513 child = os.fork()
514 514 if not child:
515 515 sys.stdout = file("/dev/null", "w")
516 516 sys.stderr = sys.stdout
517 517 hgweb.server(repo.root, "pull", "", "localhost", sport)
518 518 else:
519 519 r = os.system(cmd)
520 520 os.kill(child, signal.SIGTERM)
521 521 return r
522 522
523 def rawcommit(ui, repo, flist, **rc):
523 def rawcommit(ui, repo, *flist, **rc):
524 524 "raw commit interface"
525 525
526 526 text = rc['text']
527 527 if not text and rc['logfile']:
528 528 try: text = open(rc['logfile']).read()
529 529 except IOError: pass
530 530 if not text and not rc['logfile']:
531 531 print "missing commit text"
532 532 return 1
533 533
534 files = relpath(repo, flist)
534 files = relpath(repo, list(flist))
535 535 if rc['files']:
536 536 files += open(rc['files']).read().splitlines()
537 537
538 538 repo.rawcommit(files, text, rc['user'], rc['date'], *rc['parent'])
539 539
540 540 def recover(ui, repo):
541 541 """roll back an interrupted transaction"""
542 542 repo.recover()
543 543
544 544 def remove(ui, repo, file, *files):
545 545 """remove the specified files on the next commit"""
546 546 repo.remove(relpath(repo, (file,) + files))
547 547
548 548 def serve(ui, repo, **opts):
549 549 """export the repository via HTTP"""
550 550 hgweb.server(repo.root, opts["name"], opts["templates"],
551 551 opts["address"], opts["port"])
552 552
553 553 def status(ui, repo):
554 554 '''show changed files in the working directory
555 555
556 556 C = changed
557 557 A = added
558 558 R = removed
559 559 ? = not tracked'''
560 560
561 561 (c, a, d, u) = repo.diffdir(os.getcwd())
562 562 (c, a, d, u) = map(lambda x: relfilter(repo, x), (c, a, d, u))
563 563
564 564 for f in c: print "C", f
565 565 for f in a: print "A", f
566 566 for f in d: print "R", f
567 567 for f in u: print "?", f
568 568
569 569 def tag(ui, repo, name, rev = None, **opts):
570 570 """add a tag for the current tip or a given revision"""
571 571
572 572 if name == "tip":
573 573 ui.warn("abort: 'tip' is a reserved name!\n")
574 574 return -1
575 575
576 576 (c, a, d, u) = repo.diffdir(repo.root)
577 577 for x in (c, a, d, u):
578 578 if ".hgtags" in x:
579 579 ui.warn("abort: working copy of .hgtags is changed!\n")
580 580 ui.status("(please commit .hgtags manually)\n")
581 581 return -1
582 582
583 583 if rev:
584 584 r = hg.hex(repo.lookup(rev))
585 585 else:
586 586 r = hg.hex(repo.changelog.tip())
587 587
588 588 add = 0
589 589 if not os.path.exists(repo.wjoin(".hgtags")): add = 1
590 590 repo.wfile(".hgtags", "a").write("%s %s\n" % (r, name))
591 591 if add: repo.add([".hgtags"])
592 592
593 593 if not opts['text']:
594 594 opts['text'] = "Added tag %s for changeset %s" % (name, r)
595 595
596 596 repo.commit([".hgtags"], opts['text'], opts['user'], opts['date'])
597 597
598 598 def tags(ui, repo):
599 599 """list repository tags"""
600 600
601 601 l = repo.tagslist()
602 602 l.reverse()
603 603 for t,n in l:
604 604 try:
605 605 r = repo.changelog.rev(n)
606 606 except KeyError:
607 607 r = "?"
608 608 print "%-30s %5d:%s" % (t, repo.changelog.rev(n), hg.hex(n))
609 609
610 610 def tip(ui, repo):
611 611 """show the tip revision"""
612 612 n = repo.changelog.tip()
613 613 show_changeset(ui, repo, changenode=n)
614 614
615 615 def undo(ui, repo):
616 616 """undo the last transaction"""
617 617 repo.undo()
618 618
619 619 def update(ui, repo, node=None, merge=False, clean=False):
620 620 '''update or merge working directory
621 621
622 622 If there are no outstanding changes in the working directory and
623 623 there is a linear relationship between the current version and the
624 624 requested version, the result is the requested version.
625 625
626 626 Otherwise the result is a merge between the contents of the
627 627 current working directory and the requested version. Files that
628 628 changed between either parent are marked as changed for the next
629 629 commit and a commit must be performed before any further updates
630 630 are allowed.
631 631 '''
632 632 node = node and repo.lookup(node) or repo.changelog.tip()
633 633 return repo.update(node, allow=merge, force=clean)
634 634
635 635 def verify(ui, repo):
636 636 """verify the integrity of the repository"""
637 637 return repo.verify()
638 638
639 639 # Command options and aliases are listed here, alphabetically
640 640
641 641 table = {
642 642 "add": (add, [], "hg add [files]"),
643 643 "addremove": (addremove, [], "hg addremove [files]"),
644 644 "ann|annotate": (annotate,
645 645 [('r', 'revision', '', 'revision'),
646 646 ('u', 'user', None, 'show user'),
647 647 ('n', 'number', None, 'show revision number'),
648 648 ('c', 'changeset', None, 'show changeset')],
649 649 'hg annotate [-u] [-c] [-n] [-r id] [files]'),
650 650 "cat|dump": (cat, [], 'hg cat <file> [rev]'),
651 651 "commit|ci": (commit,
652 652 [('t', 'text', "", 'commit text'),
653 653 ('A', 'addremove', None, 'run add/remove during commit'),
654 654 ('l', 'logfile', "", 'commit text file'),
655 655 ('d', 'date', "", 'data'),
656 656 ('u', 'user', "", 'user')],
657 657 'hg commit [files]'),
658 658 "copy": (copy, [], 'hg copy <source> <dest>'),
659 659 "debugaddchangegroup": (debugaddchangegroup, [], 'debugaddchangegroup'),
660 660 "debugchangegroup": (debugchangegroup, [], 'debugchangegroup [roots]'),
661 661 "debugindex": (debugindex, [], 'debugindex <file>'),
662 662 "debugindexdot": (debugindexdot, [], 'debugindexdot <file>'),
663 663 "diff": (diff, [('r', 'rev', [], 'revision')],
664 664 'hg diff [-r A] [-r B] [files]'),
665 665 "export": (export, [], "hg export <changeset>"),
666 666 "forget": (forget, [], "hg forget [files]"),
667 667 "heads": (heads, [], 'hg heads'),
668 668 "history": (history, [], 'hg history'),
669 669 "help": (help, [], 'hg help [command]'),
670 670 "identify|id": (identify, [], 'hg identify'),
671 671 "init": (init, [('u', 'update', None, 'update after init')],
672 672 'hg init [options] [url]'),
673 673 "log": (log, [], 'hg log <file>'),
674 674 "manifest|dumpmanifest": (manifest, [], 'hg manifest [rev]'),
675 675 "parents": (parents, [], 'hg parents [node]'),
676 676 "patch|import": (patch,
677 677 [('p', 'strip', 1, 'path strip'),
678 678 ('b', 'base', "", 'base path'),
679 679 ('q', 'quiet', "", 'silence diff')],
680 680 "hg import [options] patches"),
681 681 "pull|merge": (pull, [], 'hg pull [source]'),
682 682 "push": (push, [], 'hg push <destination>'),
683 683 "rawcommit": (rawcommit,
684 684 [('p', 'parent', [], 'parent'),
685 685 ('d', 'date', "", 'data'),
686 686 ('u', 'user', "", 'user'),
687 687 ('F', 'files', "", 'file list'),
688 688 ('t', 'text', "", 'commit text'),
689 689 ('l', 'logfile', "", 'commit text file')],
690 690 'hg rawcommit [options] [files]'),
691 691 "recover": (recover, [], "hg recover"),
692 692 "remove": (remove, [], "hg remove [files]"),
693 693 "serve": (serve, [('p', 'port', 8000, 'listen port'),
694 694 ('a', 'address', '', 'interface address'),
695 695 ('n', 'name', os.getcwd(), 'repository name'),
696 696 ('t', 'templates', "", 'template map')],
697 697 "hg serve [options]"),
698 698 "status": (status, [], 'hg status'),
699 699 "tag": (tag, [('t', 'text', "", 'commit text'),
700 700 ('d', 'date', "", 'date'),
701 701 ('u', 'user', "", 'user')],
702 702 'hg tag [options] <name> [rev]'),
703 703 "tags": (tags, [], 'hg tags'),
704 704 "tip": (tip, [], 'hg tip'),
705 705 "undo": (undo, [], 'hg undo'),
706 706 "update|up|checkout|co|resolve": (update,
707 707 [('m', 'merge', None,
708 708 'allow merging of conflicts'),
709 709 ('C', 'clean', None,
710 710 'overwrite locally modified files')],
711 711 'hg update [options] [node]'),
712 712 "verify": (verify, [], 'hg verify'),
713 713 }
714 714
715 715 norepo = "init branch help debugindex debugindexdot"
716 716
717 717 def find(cmd):
718 718 i = None
719 719 for e in table.keys():
720 720 if re.match("(%s)$" % e, cmd):
721 721 return table[e]
722 722
723 723 raise UnknownCommand(cmd)
724 724
725 725 class SignalInterrupt(Exception): pass
726 726
727 727 def catchterm(*args):
728 728 raise SignalInterrupt
729 729
730 730 def run():
731 731 sys.exit(dispatch(sys.argv[1:]))
732 732
733 733 def dispatch(args):
734 734 options = {}
735 735 opts = [('v', 'verbose', None, 'verbose'),
736 736 ('d', 'debug', None, 'debug'),
737 737 ('q', 'quiet', None, 'quiet'),
738 738 ('p', 'profile', None, 'profile'),
739 739 ('y', 'noninteractive', None, 'run non-interactively'),
740 740 ]
741 741
742 742 args = fancyopts.fancyopts(args, opts, options,
743 743 'hg [options] <command> [options] [files]')
744 744
745 745 if not args:
746 746 cmd = "help"
747 747 else:
748 748 cmd, args = args[0], args[1:]
749 749
750 750 u = ui.ui(options["verbose"], options["debug"], options["quiet"],
751 751 not options["noninteractive"])
752 752
753 753 try:
754 754 i = find(cmd)
755 755 except UnknownCommand:
756 756 u.warn("hg: unknown command '%s'\n" % cmd)
757 757 help(u)
758 758 sys.exit(1)
759 759
760 760 signal.signal(signal.SIGTERM, catchterm)
761 761
762 762 cmdoptions = {}
763 763 try:
764 764 args = fancyopts.fancyopts(args, i[1], cmdoptions, i[2])
765 765 except fancyopts.getopt.GetoptError, inst:
766 766 u.warn("hg %s: %s\n" % (cmd, inst))
767 767 help(u, cmd)
768 768 sys.exit(-1)
769 769
770 770 if cmd not in norepo.split():
771 771 repo = hg.repository(ui = u)
772 772 d = lambda: i[0](u, repo, *args, **cmdoptions)
773 773 else:
774 774 d = lambda: i[0](u, *args, **cmdoptions)
775 775
776 776 try:
777 777 if options['profile']:
778 778 import hotshot, hotshot.stats
779 779 prof = hotshot.Profile("hg.prof")
780 780 r = prof.runcall(d)
781 781 prof.close()
782 782 stats = hotshot.stats.load("hg.prof")
783 783 stats.strip_dirs()
784 784 stats.sort_stats('time', 'calls')
785 785 stats.print_stats(40)
786 786 return r
787 787 else:
788 788 return d()
789 789 except SignalInterrupt:
790 790 u.warn("killed!\n")
791 791 except KeyboardInterrupt:
792 792 u.warn("interrupted!\n")
793 793 except IOError, inst:
794 794 if hasattr(inst, "code"):
795 795 u.warn("abort: %s\n" % inst)
796 796 elif hasattr(inst, "reason"):
797 797 u.warn("abort: error %d: %s\n" % (inst.reason[0], inst.reason[1]))
798 798 elif hasattr(inst, "args") and inst[0] == errno.EPIPE:
799 799 u.warn("broken pipe\n")
800 800 else:
801 801 raise
802 802 except TypeError, inst:
803 803 # was this an argument error?
804 804 tb = traceback.extract_tb(sys.exc_info()[2])
805 805 if len(tb) > 2: # no
806 806 raise
807 807 u.debug(inst, "\n")
808 808 u.warn("%s: invalid arguments\n" % i[0].__name__)
809 809 help(u, cmd)
810 810 sys.exit(-1)
811 811
General Comments 0
You need to be logged in to leave comments. Login now