##// END OF EJS Templates
fix errors spotted by pychecker
Benoit Boissinot -
r3859:8c24b6fd default
parent child Browse files
Show More
@@ -1,120 +1,120 b''
1 1 """
2 2 changegroup.py - Mercurial changegroup manipulation functions
3 3
4 4 Copyright 2006 Matt Mackall <mpm@selenic.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 i18n import gettext as _
10 10 from demandload import *
11 11 demandload(globals(), "struct os bz2 zlib util tempfile")
12 12
13 13 def getchunk(source):
14 14 """get a chunk from a changegroup"""
15 15 d = source.read(4)
16 16 if not d:
17 17 return ""
18 18 l = struct.unpack(">l", d)[0]
19 19 if l <= 4:
20 20 return ""
21 21 d = source.read(l - 4)
22 22 if len(d) < l - 4:
23 23 raise util.Abort(_("premature EOF reading chunk"
24 24 " (got %d bytes, expected %d)")
25 25 % (len(d), l - 4))
26 26 return d
27 27
28 28 def chunkiter(source):
29 29 """iterate through the chunks in source"""
30 30 while 1:
31 31 c = getchunk(source)
32 32 if not c:
33 33 break
34 34 yield c
35 35
36 36 def genchunk(data):
37 37 """build a changegroup chunk"""
38 38 header = struct.pack(">l", len(data)+ 4)
39 39 return "%s%s" % (header, data)
40 40
41 41 def closechunk():
42 42 return struct.pack(">l", 0)
43 43
44 44 class nocompress(object):
45 45 def compress(self, x):
46 46 return x
47 47 def flush(self):
48 48 return ""
49 49
50 50 bundletypes = {
51 51 "": ("", nocompress),
52 52 "HG10UN": ("HG10UN", nocompress),
53 53 "HG10BZ": ("HG10", lambda: bz2.BZ2Compressor()),
54 54 "HG10GZ": ("HG10GZ", lambda: zlib.compressobj()),
55 55 }
56 56
57 57 def writebundle(cg, filename, bundletype):
58 58 """Write a bundle file and return its filename.
59 59
60 60 Existing files will not be overwritten.
61 61 If no filename is specified, a temporary file is created.
62 62 bz2 compression can be turned off.
63 63 The bundle file will be deleted in case of errors.
64 64 """
65 65
66 66 fh = None
67 67 cleanup = None
68 68 try:
69 69 if filename:
70 70 if os.path.exists(filename):
71 71 raise util.Abort(_("file '%s' already exists") % filename)
72 72 fh = open(filename, "wb")
73 73 else:
74 74 fd, filename = tempfile.mkstemp(prefix="hg-bundle-", suffix=".hg")
75 75 fh = os.fdopen(fd, "wb")
76 76 cleanup = filename
77 77
78 78 header, compressor = bundletypes[bundletype]
79 79 fh.write(header)
80 80 z = compressor()
81 81
82 82 # parse the changegroup data, otherwise we will block
83 83 # in case of sshrepo because we don't know the end of the stream
84 84
85 85 # an empty chunkiter is the end of the changegroup
86 86 empty = False
87 87 while not empty:
88 88 empty = True
89 89 for chunk in chunkiter(cg):
90 90 empty = False
91 91 fh.write(z.compress(genchunk(chunk)))
92 92 fh.write(z.compress(closechunk()))
93 93 fh.write(z.flush())
94 94 cleanup = None
95 95 return filename
96 96 finally:
97 97 if fh is not None:
98 98 fh.close()
99 99 if cleanup is not None:
100 100 os.unlink(cleanup)
101 101
102 def readbundle(fh):
102 def readbundle(fh, fname):
103 103 header = fh.read(6)
104 104 if not header.startswith("HG"):
105 105 raise util.Abort(_("%s: not a Mercurial bundle file") % fname)
106 106 elif not header.startswith("HG10"):
107 107 raise util.Abort(_("%s: unknown bundle version") % fname)
108 108
109 109 if header == "HG10BZ":
110 110 def generator(f):
111 111 zd = bz2.BZ2Decompressor()
112 112 zd.decompress("BZ")
113 113 for chunk in util.filechunkiter(f, 4096):
114 114 yield zd.decompress(chunk)
115 115 return util.chunkbuffer(generator(fh))
116 116 elif header == "HG10UN":
117 117 return fh
118 118
119 119 raise util.Abort(_("%s: unknown bundle compression type")
120 120 % fname)
@@ -1,3314 +1,3314 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(), "bisect os re sys signal imp urllib pdb shlex stat")
12 12 demandload(globals(), "fancyopts ui hg util lock revlog bundlerepo")
13 13 demandload(globals(), "difflib patch time help mdiff tempfile")
14 14 demandload(globals(), "traceback errno version atexit")
15 15 demandload(globals(), "archival changegroup cmdutil hgweb.server sshserver")
16 16
17 17 class UnknownCommand(Exception):
18 18 """Exception raised if command is not in the command table."""
19 19 class AmbiguousCommand(Exception):
20 20 """Exception raised if command shortcut matches more than one command."""
21 21
22 22 def bail_if_changed(repo):
23 23 modified, added, removed, deleted = repo.status()[:4]
24 24 if modified or added or removed or deleted:
25 25 raise util.Abort(_("outstanding uncommitted changes"))
26 26
27 27 def logmessage(opts):
28 28 """ get the log message according to -m and -l option """
29 29 message = opts['message']
30 30 logfile = opts['logfile']
31 31
32 32 if message and logfile:
33 33 raise util.Abort(_('options --message and --logfile are mutually '
34 34 'exclusive'))
35 35 if not message and logfile:
36 36 try:
37 37 if logfile == '-':
38 38 message = sys.stdin.read()
39 39 else:
40 40 message = open(logfile).read()
41 41 except IOError, inst:
42 42 raise util.Abort(_("can't read commit message '%s': %s") %
43 43 (logfile, inst.strerror))
44 44 return message
45 45
46 46 def setremoteconfig(ui, opts):
47 47 "copy remote options to ui tree"
48 48 if opts.get('ssh'):
49 49 ui.setconfig("ui", "ssh", opts['ssh'])
50 50 if opts.get('remotecmd'):
51 51 ui.setconfig("ui", "remotecmd", opts['remotecmd'])
52 52
53 53 # Commands start here, listed alphabetically
54 54
55 55 def add(ui, repo, *pats, **opts):
56 56 """add the specified files on the next commit
57 57
58 58 Schedule files to be version controlled and added to the repository.
59 59
60 60 The files will be added to the repository at the next commit. To
61 61 undo an add before that, see hg revert.
62 62
63 63 If no names are given, add all files in the repository.
64 64 """
65 65
66 66 names = []
67 67 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts):
68 68 if exact:
69 69 if ui.verbose:
70 70 ui.status(_('adding %s\n') % rel)
71 71 names.append(abs)
72 72 elif repo.dirstate.state(abs) == '?':
73 73 ui.status(_('adding %s\n') % rel)
74 74 names.append(abs)
75 75 if not opts.get('dry_run'):
76 76 repo.add(names)
77 77
78 78 def addremove(ui, repo, *pats, **opts):
79 79 """add all new files, delete all missing files
80 80
81 81 Add all new files and remove all missing files from the repository.
82 82
83 83 New files are ignored if they match any of the patterns in .hgignore. As
84 84 with add, these changes take effect at the next commit.
85 85
86 86 Use the -s option to detect renamed files. With a parameter > 0,
87 87 this compares every removed file with every added file and records
88 88 those similar enough as renames. This option takes a percentage
89 89 between 0 (disabled) and 100 (files must be identical) as its
90 90 parameter. Detecting renamed files this way can be expensive.
91 91 """
92 92 sim = float(opts.get('similarity') or 0)
93 93 if sim < 0 or sim > 100:
94 94 raise util.Abort(_('similarity must be between 0 and 100'))
95 95 return cmdutil.addremove(repo, pats, opts, similarity=sim/100.)
96 96
97 97 def annotate(ui, repo, *pats, **opts):
98 98 """show changeset information per file line
99 99
100 100 List changes in files, showing the revision id responsible for each line
101 101
102 102 This command is useful to discover who did a change or when a change took
103 103 place.
104 104
105 105 Without the -a option, annotate will avoid processing files it
106 106 detects as binary. With -a, annotate will generate an annotation
107 107 anyway, probably with undesirable results.
108 108 """
109 109 getdate = util.cachefunc(lambda x: util.datestr(x.date()))
110 110
111 111 if not pats:
112 112 raise util.Abort(_('at least one file name or pattern required'))
113 113
114 114 opmap = [['user', lambda x: ui.shortuser(x.user())],
115 115 ['number', lambda x: str(x.rev())],
116 116 ['changeset', lambda x: short(x.node())],
117 117 ['date', getdate], ['follow', lambda x: x.path()]]
118 118 if (not opts['user'] and not opts['changeset'] and not opts['date']
119 119 and not opts['follow']):
120 120 opts['number'] = 1
121 121
122 122 ctx = repo.changectx(opts['rev'])
123 123
124 124 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts,
125 125 node=ctx.node()):
126 126 fctx = ctx.filectx(abs)
127 127 if not opts['text'] and util.binary(fctx.data()):
128 128 ui.write(_("%s: binary file\n") % ((pats and rel) or abs))
129 129 continue
130 130
131 131 lines = fctx.annotate(follow=opts.get('follow'))
132 132 pieces = []
133 133
134 134 for o, f in opmap:
135 135 if opts[o]:
136 136 l = [f(n) for n, dummy in lines]
137 137 if l:
138 138 m = max(map(len, l))
139 139 pieces.append(["%*s" % (m, x) for x in l])
140 140
141 141 if pieces:
142 142 for p, l in zip(zip(*pieces), lines):
143 143 ui.write("%s: %s" % (" ".join(p), l[1]))
144 144
145 145 def archive(ui, repo, dest, **opts):
146 146 '''create unversioned archive of a repository revision
147 147
148 148 By default, the revision used is the parent of the working
149 149 directory; use "-r" to specify a different revision.
150 150
151 151 To specify the type of archive to create, use "-t". Valid
152 152 types are:
153 153
154 154 "files" (default): a directory full of files
155 155 "tar": tar archive, uncompressed
156 156 "tbz2": tar archive, compressed using bzip2
157 157 "tgz": tar archive, compressed using gzip
158 158 "uzip": zip archive, uncompressed
159 159 "zip": zip archive, compressed using deflate
160 160
161 161 The exact name of the destination archive or directory is given
162 162 using a format string; see "hg help export" for details.
163 163
164 164 Each member added to an archive file has a directory prefix
165 165 prepended. Use "-p" to specify a format string for the prefix.
166 166 The default is the basename of the archive, with suffixes removed.
167 167 '''
168 168
169 169 node = repo.changectx(opts['rev']).node()
170 170 dest = cmdutil.make_filename(repo, dest, node)
171 171 if os.path.realpath(dest) == repo.root:
172 172 raise util.Abort(_('repository root cannot be destination'))
173 173 dummy, matchfn, dummy = cmdutil.matchpats(repo, [], opts)
174 174 kind = opts.get('type') or 'files'
175 175 prefix = opts['prefix']
176 176 if dest == '-':
177 177 if kind == 'files':
178 178 raise util.Abort(_('cannot archive plain files to stdout'))
179 179 dest = sys.stdout
180 180 if not prefix: prefix = os.path.basename(repo.root) + '-%h'
181 181 prefix = cmdutil.make_filename(repo, prefix, node)
182 182 archival.archive(repo, dest, node, kind, not opts['no_decode'],
183 183 matchfn, prefix)
184 184
185 185 def backout(ui, repo, rev, **opts):
186 186 '''reverse effect of earlier changeset
187 187
188 188 Commit the backed out changes as a new changeset. The new
189 189 changeset is a child of the backed out changeset.
190 190
191 191 If you back out a changeset other than the tip, a new head is
192 192 created. This head is the parent of the working directory. If
193 193 you back out an old changeset, your working directory will appear
194 194 old after the backout. You should merge the backout changeset
195 195 with another head.
196 196
197 197 The --merge option remembers the parent of the working directory
198 198 before starting the backout, then merges the new head with that
199 199 changeset afterwards. This saves you from doing the merge by
200 200 hand. The result of this merge is not committed, as for a normal
201 201 merge.'''
202 202
203 203 bail_if_changed(repo)
204 204 op1, op2 = repo.dirstate.parents()
205 205 if op2 != nullid:
206 206 raise util.Abort(_('outstanding uncommitted merge'))
207 207 node = repo.lookup(rev)
208 208 p1, p2 = repo.changelog.parents(node)
209 209 if p1 == nullid:
210 210 raise util.Abort(_('cannot back out a change with no parents'))
211 211 if p2 != nullid:
212 212 if not opts['parent']:
213 213 raise util.Abort(_('cannot back out a merge changeset without '
214 214 '--parent'))
215 215 p = repo.lookup(opts['parent'])
216 216 if p not in (p1, p2):
217 217 raise util.Abort(_('%s is not a parent of %s') %
218 218 (short(p), short(node)))
219 219 parent = p
220 220 else:
221 221 if opts['parent']:
222 222 raise util.Abort(_('cannot use --parent on non-merge changeset'))
223 223 parent = p1
224 224 hg.clean(repo, node, show_stats=False)
225 225 revert_opts = opts.copy()
226 226 revert_opts['date'] = None
227 227 revert_opts['all'] = True
228 228 revert_opts['rev'] = hex(parent)
229 229 revert(ui, repo, **revert_opts)
230 230 commit_opts = opts.copy()
231 231 commit_opts['addremove'] = False
232 232 if not commit_opts['message'] and not commit_opts['logfile']:
233 233 commit_opts['message'] = _("Backed out changeset %s") % (hex(node))
234 234 commit_opts['force_editor'] = True
235 235 commit(ui, repo, **commit_opts)
236 236 def nice(node):
237 237 return '%d:%s' % (repo.changelog.rev(node), short(node))
238 238 ui.status(_('changeset %s backs out changeset %s\n') %
239 239 (nice(repo.changelog.tip()), nice(node)))
240 240 if op1 != node:
241 241 if opts['merge']:
242 242 ui.status(_('merging with changeset %s\n') % nice(op1))
243 243 n = _lookup(repo, hex(op1))
244 244 hg.merge(repo, n)
245 245 else:
246 246 ui.status(_('the backout changeset is a new head - '
247 247 'do not forget to merge\n'))
248 248 ui.status(_('(use "backout --merge" '
249 249 'if you want to auto-merge)\n'))
250 250
251 251 def branch(ui, repo, label=None):
252 252 """set or show the current branch name
253 253
254 254 With <name>, set the current branch name. Otherwise, show the
255 255 current branch name.
256 256 """
257 257
258 258 if label is not None:
259 259 repo.opener("branch", "w").write(label)
260 260 else:
261 261 b = repo.workingctx().branch()
262 262 if b:
263 263 ui.write("%s\n" % b)
264 264
265 265 def branches(ui, repo):
266 266 """list repository named branches
267 267
268 268 List the repository's named branches.
269 269 """
270 270 b = repo.branchtags()
271 271 l = [(-repo.changelog.rev(n), n, t) for t, n in b.items()]
272 272 l.sort()
273 273 for r, n, t in l:
274 274 hexfunc = ui.debugflag and hex or short
275 275 if ui.quiet:
276 276 ui.write("%s\n" % t)
277 277 else:
278 278 t = util.localsub(t, 30)
279 279 t += " " * (30 - util.locallen(t))
280 280 ui.write("%s %s:%s\n" % (t, -r, hexfunc(n)))
281 281
282 282 def bundle(ui, repo, fname, dest=None, **opts):
283 283 """create a changegroup file
284 284
285 285 Generate a compressed changegroup file collecting changesets not
286 286 found in the other repository.
287 287
288 288 If no destination repository is specified the destination is assumed
289 289 to have all the nodes specified by one or more --base parameters.
290 290
291 291 The bundle file can then be transferred using conventional means and
292 292 applied to another repository with the unbundle or pull command.
293 293 This is useful when direct push and pull are not available or when
294 294 exporting an entire repository is undesirable.
295 295
296 296 Applying bundles preserves all changeset contents including
297 297 permissions, copy/rename information, and revision history.
298 298 """
299 299 revs = opts.get('rev') or None
300 300 if revs:
301 301 revs = [repo.lookup(rev) for rev in revs]
302 302 base = opts.get('base')
303 303 if base:
304 304 if dest:
305 305 raise util.Abort(_("--base is incompatible with specifiying "
306 306 "a destination"))
307 307 base = [repo.lookup(rev) for rev in base]
308 308 # create the right base
309 309 # XXX: nodesbetween / changegroup* should be "fixed" instead
310 310 o = []
311 311 has = {nullid: None}
312 312 for n in base:
313 313 has.update(repo.changelog.reachable(n))
314 314 if revs:
315 315 visit = list(revs)
316 316 else:
317 317 visit = repo.changelog.heads()
318 318 seen = {}
319 319 while visit:
320 320 n = visit.pop(0)
321 321 parents = [p for p in repo.changelog.parents(n) if p not in has]
322 322 if len(parents) == 0:
323 323 o.insert(0, n)
324 324 else:
325 325 for p in parents:
326 326 if p not in seen:
327 327 seen[p] = 1
328 328 visit.append(p)
329 329 else:
330 330 setremoteconfig(ui, opts)
331 331 dest = ui.expandpath(dest or 'default-push', dest or 'default')
332 332 other = hg.repository(ui, dest)
333 333 o = repo.findoutgoing(other, force=opts['force'])
334 334
335 335 if revs:
336 336 cg = repo.changegroupsubset(o, revs, 'bundle')
337 337 else:
338 338 cg = repo.changegroup(o, 'bundle')
339 339 changegroup.writebundle(cg, fname, "HG10BZ")
340 340
341 341 def cat(ui, repo, file1, *pats, **opts):
342 342 """output the latest or given revisions of files
343 343
344 344 Print the specified files as they were at the given revision.
345 345 If no revision is given then working dir parent is used, or tip
346 346 if no revision is checked out.
347 347
348 348 Output may be to a file, in which case the name of the file is
349 349 given using a format string. The formatting rules are the same as
350 350 for the export command, with the following additions:
351 351
352 352 %s basename of file being printed
353 353 %d dirname of file being printed, or '.' if in repo root
354 354 %p root-relative path name of file being printed
355 355 """
356 356 ctx = repo.changectx(opts['rev'])
357 357 for src, abs, rel, exact in cmdutil.walk(repo, (file1,) + pats, opts,
358 358 ctx.node()):
359 359 fp = cmdutil.make_file(repo, opts['output'], ctx.node(), pathname=abs)
360 360 fp.write(ctx.filectx(abs).data())
361 361
362 362 def clone(ui, source, dest=None, **opts):
363 363 """make a copy of an existing repository
364 364
365 365 Create a copy of an existing repository in a new directory.
366 366
367 367 If no destination directory name is specified, it defaults to the
368 368 basename of the source.
369 369
370 370 The location of the source is added to the new repository's
371 371 .hg/hgrc file, as the default to be used for future pulls.
372 372
373 373 For efficiency, hardlinks are used for cloning whenever the source
374 374 and destination are on the same filesystem (note this applies only
375 375 to the repository data, not to the checked out files). Some
376 376 filesystems, such as AFS, implement hardlinking incorrectly, but
377 377 do not report errors. In these cases, use the --pull option to
378 378 avoid hardlinking.
379 379
380 380 You can safely clone repositories and checked out files using full
381 381 hardlinks with
382 382
383 383 $ cp -al REPO REPOCLONE
384 384
385 385 which is the fastest way to clone. However, the operation is not
386 386 atomic (making sure REPO is not modified during the operation is
387 387 up to you) and you have to make sure your editor breaks hardlinks
388 388 (Emacs and most Linux Kernel tools do so).
389 389
390 390 If you use the -r option to clone up to a specific revision, no
391 391 subsequent revisions will be present in the cloned repository.
392 392 This option implies --pull, even on local repositories.
393 393
394 394 See pull for valid source format details.
395 395
396 396 It is possible to specify an ssh:// URL as the destination, but no
397 397 .hg/hgrc and working directory will be created on the remote side.
398 398 Look at the help text for the pull command for important details
399 399 about ssh:// URLs.
400 400 """
401 401 setremoteconfig(ui, opts)
402 402 hg.clone(ui, ui.expandpath(source), dest,
403 403 pull=opts['pull'],
404 404 stream=opts['uncompressed'],
405 405 rev=opts['rev'],
406 406 update=not opts['noupdate'])
407 407
408 408 def commit(ui, repo, *pats, **opts):
409 409 """commit the specified files or all outstanding changes
410 410
411 411 Commit changes to the given files into the repository.
412 412
413 413 If a list of files is omitted, all changes reported by "hg status"
414 414 will be committed.
415 415
416 416 If no commit message is specified, the editor configured in your hgrc
417 417 or in the EDITOR environment variable is started to enter a message.
418 418 """
419 419 message = logmessage(opts)
420 420
421 421 if opts['addremove']:
422 422 cmdutil.addremove(repo, pats, opts)
423 423 fns, match, anypats = cmdutil.matchpats(repo, pats, opts)
424 424 if pats:
425 425 status = repo.status(files=fns, match=match)
426 426 modified, added, removed, deleted, unknown = status[:5]
427 427 files = modified + added + removed
428 428 slist = None
429 429 for f in fns:
430 430 if f not in files:
431 431 rf = repo.wjoin(f)
432 432 if f in unknown:
433 433 raise util.Abort(_("file %s not tracked!") % rf)
434 434 try:
435 435 mode = os.lstat(rf)[stat.ST_MODE]
436 436 except OSError:
437 437 raise util.Abort(_("file %s not found!") % rf)
438 438 if stat.S_ISDIR(mode):
439 439 name = f + '/'
440 440 if slist is None:
441 441 slist = list(files)
442 442 slist.sort()
443 443 i = bisect.bisect(slist, name)
444 444 if i >= len(slist) or not slist[i].startswith(name):
445 445 raise util.Abort(_("no match under directory %s!")
446 446 % rf)
447 447 elif not stat.S_ISREG(mode):
448 448 raise util.Abort(_("can't commit %s: "
449 449 "unsupported file type!") % rf)
450 450 else:
451 451 files = []
452 452 try:
453 453 repo.commit(files, message, opts['user'], opts['date'], match,
454 454 force_editor=opts.get('force_editor'))
455 455 except ValueError, inst:
456 456 raise util.Abort(str(inst))
457 457
458 458 def docopy(ui, repo, pats, opts, wlock):
459 459 # called with the repo lock held
460 460 #
461 461 # hgsep => pathname that uses "/" to separate directories
462 462 # ossep => pathname that uses os.sep to separate directories
463 463 cwd = repo.getcwd()
464 464 errors = 0
465 465 copied = []
466 466 targets = {}
467 467
468 468 # abs: hgsep
469 469 # rel: ossep
470 470 # return: hgsep
471 471 def okaytocopy(abs, rel, exact):
472 472 reasons = {'?': _('is not managed'),
473 473 'a': _('has been marked for add'),
474 474 'r': _('has been marked for remove')}
475 475 state = repo.dirstate.state(abs)
476 476 reason = reasons.get(state)
477 477 if reason:
478 478 if state == 'a':
479 479 origsrc = repo.dirstate.copied(abs)
480 480 if origsrc is not None:
481 481 return origsrc
482 482 if exact:
483 483 ui.warn(_('%s: not copying - file %s\n') % (rel, reason))
484 484 else:
485 485 return abs
486 486
487 487 # origsrc: hgsep
488 488 # abssrc: hgsep
489 489 # relsrc: ossep
490 490 # target: ossep
491 491 def copy(origsrc, abssrc, relsrc, target, exact):
492 492 abstarget = util.canonpath(repo.root, cwd, target)
493 493 reltarget = util.pathto(cwd, abstarget)
494 494 prevsrc = targets.get(abstarget)
495 495 if prevsrc is not None:
496 496 ui.warn(_('%s: not overwriting - %s collides with %s\n') %
497 497 (reltarget, util.localpath(abssrc),
498 498 util.localpath(prevsrc)))
499 499 return
500 500 if (not opts['after'] and os.path.exists(reltarget) or
501 501 opts['after'] and repo.dirstate.state(abstarget) not in '?r'):
502 502 if not opts['force']:
503 503 ui.warn(_('%s: not overwriting - file exists\n') %
504 504 reltarget)
505 505 return
506 506 if not opts['after'] and not opts.get('dry_run'):
507 507 os.unlink(reltarget)
508 508 if opts['after']:
509 509 if not os.path.exists(reltarget):
510 510 return
511 511 else:
512 512 targetdir = os.path.dirname(reltarget) or '.'
513 513 if not os.path.isdir(targetdir) and not opts.get('dry_run'):
514 514 os.makedirs(targetdir)
515 515 try:
516 516 restore = repo.dirstate.state(abstarget) == 'r'
517 517 if restore and not opts.get('dry_run'):
518 518 repo.undelete([abstarget], wlock)
519 519 try:
520 520 if not opts.get('dry_run'):
521 521 util.copyfile(relsrc, reltarget)
522 522 restore = False
523 523 finally:
524 524 if restore:
525 525 repo.remove([abstarget], wlock)
526 526 except IOError, inst:
527 527 if inst.errno == errno.ENOENT:
528 528 ui.warn(_('%s: deleted in working copy\n') % relsrc)
529 529 else:
530 530 ui.warn(_('%s: cannot copy - %s\n') %
531 531 (relsrc, inst.strerror))
532 532 errors += 1
533 533 return
534 534 if ui.verbose or not exact:
535 535 ui.status(_('copying %s to %s\n') % (relsrc, reltarget))
536 536 targets[abstarget] = abssrc
537 537 if abstarget != origsrc and not opts.get('dry_run'):
538 538 repo.copy(origsrc, abstarget, wlock)
539 539 copied.append((abssrc, relsrc, exact))
540 540
541 541 # pat: ossep
542 542 # dest ossep
543 543 # srcs: list of (hgsep, hgsep, ossep, bool)
544 544 # return: function that takes hgsep and returns ossep
545 545 def targetpathfn(pat, dest, srcs):
546 546 if os.path.isdir(pat):
547 547 abspfx = util.canonpath(repo.root, cwd, pat)
548 548 abspfx = util.localpath(abspfx)
549 549 if destdirexists:
550 550 striplen = len(os.path.split(abspfx)[0])
551 551 else:
552 552 striplen = len(abspfx)
553 553 if striplen:
554 554 striplen += len(os.sep)
555 555 res = lambda p: os.path.join(dest, util.localpath(p)[striplen:])
556 556 elif destdirexists:
557 557 res = lambda p: os.path.join(dest,
558 558 os.path.basename(util.localpath(p)))
559 559 else:
560 560 res = lambda p: dest
561 561 return res
562 562
563 563 # pat: ossep
564 564 # dest ossep
565 565 # srcs: list of (hgsep, hgsep, ossep, bool)
566 566 # return: function that takes hgsep and returns ossep
567 567 def targetpathafterfn(pat, dest, srcs):
568 568 if util.patkind(pat, None)[0]:
569 569 # a mercurial pattern
570 570 res = lambda p: os.path.join(dest,
571 571 os.path.basename(util.localpath(p)))
572 572 else:
573 573 abspfx = util.canonpath(repo.root, cwd, pat)
574 574 if len(abspfx) < len(srcs[0][0]):
575 575 # A directory. Either the target path contains the last
576 576 # component of the source path or it does not.
577 577 def evalpath(striplen):
578 578 score = 0
579 579 for s in srcs:
580 580 t = os.path.join(dest, util.localpath(s[0])[striplen:])
581 581 if os.path.exists(t):
582 582 score += 1
583 583 return score
584 584
585 585 abspfx = util.localpath(abspfx)
586 586 striplen = len(abspfx)
587 587 if striplen:
588 588 striplen += len(os.sep)
589 589 if os.path.isdir(os.path.join(dest, os.path.split(abspfx)[1])):
590 590 score = evalpath(striplen)
591 591 striplen1 = len(os.path.split(abspfx)[0])
592 592 if striplen1:
593 593 striplen1 += len(os.sep)
594 594 if evalpath(striplen1) > score:
595 595 striplen = striplen1
596 596 res = lambda p: os.path.join(dest,
597 597 util.localpath(p)[striplen:])
598 598 else:
599 599 # a file
600 600 if destdirexists:
601 601 res = lambda p: os.path.join(dest,
602 602 os.path.basename(util.localpath(p)))
603 603 else:
604 604 res = lambda p: dest
605 605 return res
606 606
607 607
608 608 pats = list(pats)
609 609 if not pats:
610 610 raise util.Abort(_('no source or destination specified'))
611 611 if len(pats) == 1:
612 612 raise util.Abort(_('no destination specified'))
613 613 dest = pats.pop()
614 614 destdirexists = os.path.isdir(dest)
615 615 if (len(pats) > 1 or util.patkind(pats[0], None)[0]) and not destdirexists:
616 616 raise util.Abort(_('with multiple sources, destination must be an '
617 617 'existing directory'))
618 618 if opts['after']:
619 619 tfn = targetpathafterfn
620 620 else:
621 621 tfn = targetpathfn
622 622 copylist = []
623 623 for pat in pats:
624 624 srcs = []
625 625 for tag, abssrc, relsrc, exact in cmdutil.walk(repo, [pat], opts):
626 626 origsrc = okaytocopy(abssrc, relsrc, exact)
627 627 if origsrc:
628 628 srcs.append((origsrc, abssrc, relsrc, exact))
629 629 if not srcs:
630 630 continue
631 631 copylist.append((tfn(pat, dest, srcs), srcs))
632 632 if not copylist:
633 633 raise util.Abort(_('no files to copy'))
634 634
635 635 for targetpath, srcs in copylist:
636 636 for origsrc, abssrc, relsrc, exact in srcs:
637 637 copy(origsrc, abssrc, relsrc, targetpath(abssrc), exact)
638 638
639 639 if errors:
640 640 ui.warn(_('(consider using --after)\n'))
641 641 return errors, copied
642 642
643 643 def copy(ui, repo, *pats, **opts):
644 644 """mark files as copied for the next commit
645 645
646 646 Mark dest as having copies of source files. If dest is a
647 647 directory, copies are put in that directory. If dest is a file,
648 648 there can only be one source.
649 649
650 650 By default, this command copies the contents of files as they
651 651 stand in the working directory. If invoked with --after, the
652 652 operation is recorded, but no copying is performed.
653 653
654 654 This command takes effect in the next commit. To undo a copy
655 655 before that, see hg revert.
656 656 """
657 657 wlock = repo.wlock(0)
658 658 errs, copied = docopy(ui, repo, pats, opts, wlock)
659 659 return errs
660 660
661 661 def debugancestor(ui, index, rev1, rev2):
662 662 """find the ancestor revision of two revisions in a given index"""
663 663 r = revlog.revlog(util.opener(os.getcwd(), audit=False), index, "", 0)
664 664 a = r.ancestor(r.lookup(rev1), r.lookup(rev2))
665 665 ui.write("%d:%s\n" % (r.rev(a), hex(a)))
666 666
667 667 def debugcomplete(ui, cmd='', **opts):
668 668 """returns the completion list associated with the given command"""
669 669
670 670 if opts['options']:
671 671 options = []
672 672 otables = [globalopts]
673 673 if cmd:
674 674 aliases, entry = findcmd(ui, cmd)
675 675 otables.append(entry[1])
676 676 for t in otables:
677 677 for o in t:
678 678 if o[0]:
679 679 options.append('-%s' % o[0])
680 680 options.append('--%s' % o[1])
681 681 ui.write("%s\n" % "\n".join(options))
682 682 return
683 683
684 684 clist = findpossible(ui, cmd).keys()
685 685 clist.sort()
686 686 ui.write("%s\n" % "\n".join(clist))
687 687
688 688 def debugrebuildstate(ui, repo, rev=None):
689 689 """rebuild the dirstate as it would look like for the given revision"""
690 690 if not rev:
691 691 rev = repo.changelog.tip()
692 692 else:
693 693 rev = repo.lookup(rev)
694 694 change = repo.changelog.read(rev)
695 695 n = change[0]
696 696 files = repo.manifest.read(n)
697 697 wlock = repo.wlock()
698 698 repo.dirstate.rebuild(rev, files)
699 699
700 700 def debugcheckstate(ui, repo):
701 701 """validate the correctness of the current dirstate"""
702 702 parent1, parent2 = repo.dirstate.parents()
703 703 repo.dirstate.read()
704 704 dc = repo.dirstate.map
705 705 keys = dc.keys()
706 706 keys.sort()
707 707 m1n = repo.changelog.read(parent1)[0]
708 708 m2n = repo.changelog.read(parent2)[0]
709 709 m1 = repo.manifest.read(m1n)
710 710 m2 = repo.manifest.read(m2n)
711 711 errors = 0
712 712 for f in dc:
713 713 state = repo.dirstate.state(f)
714 714 if state in "nr" and f not in m1:
715 715 ui.warn(_("%s in state %s, but not in manifest1\n") % (f, state))
716 716 errors += 1
717 717 if state in "a" and f in m1:
718 718 ui.warn(_("%s in state %s, but also in manifest1\n") % (f, state))
719 719 errors += 1
720 720 if state in "m" and f not in m1 and f not in m2:
721 721 ui.warn(_("%s in state %s, but not in either manifest\n") %
722 722 (f, state))
723 723 errors += 1
724 724 for f in m1:
725 725 state = repo.dirstate.state(f)
726 726 if state not in "nrm":
727 727 ui.warn(_("%s in manifest1, but listed as state %s") % (f, state))
728 728 errors += 1
729 729 if errors:
730 730 error = _(".hg/dirstate inconsistent with current parent's manifest")
731 731 raise util.Abort(error)
732 732
733 733 def showconfig(ui, repo, *values, **opts):
734 734 """show combined config settings from all hgrc files
735 735
736 736 With no args, print names and values of all config items.
737 737
738 738 With one arg of the form section.name, print just the value of
739 739 that config item.
740 740
741 741 With multiple args, print names and values of all config items
742 742 with matching section names."""
743 743
744 744 untrusted = bool(opts.get('untrusted'))
745 745 if values:
746 746 if len([v for v in values if '.' in v]) > 1:
747 747 raise util.Abort(_('only one config item permitted'))
748 748 for section, name, value in ui.walkconfig(untrusted=untrusted):
749 749 sectname = section + '.' + name
750 750 if values:
751 751 for v in values:
752 752 if v == section:
753 753 ui.write('%s=%s\n' % (sectname, value))
754 754 elif v == sectname:
755 755 ui.write(value, '\n')
756 756 else:
757 757 ui.write('%s=%s\n' % (sectname, value))
758 758
759 759 def debugsetparents(ui, repo, rev1, rev2=None):
760 760 """manually set the parents of the current working directory
761 761
762 762 This is useful for writing repository conversion tools, but should
763 763 be used with care.
764 764 """
765 765
766 766 if not rev2:
767 767 rev2 = hex(nullid)
768 768
769 769 repo.dirstate.setparents(repo.lookup(rev1), repo.lookup(rev2))
770 770
771 771 def debugstate(ui, repo):
772 772 """show the contents of the current dirstate"""
773 773 repo.dirstate.read()
774 774 dc = repo.dirstate.map
775 775 keys = dc.keys()
776 776 keys.sort()
777 777 for file_ in keys:
778 778 ui.write("%c %3o %10d %s %s\n"
779 779 % (dc[file_][0], dc[file_][1] & 0777, dc[file_][2],
780 780 time.strftime("%x %X",
781 781 time.localtime(dc[file_][3])), file_))
782 782 for f in repo.dirstate.copies():
783 783 ui.write(_("copy: %s -> %s\n") % (repo.dirstate.copied(f), f))
784 784
785 785 def debugdata(ui, file_, rev):
786 786 """dump the contents of an data file revision"""
787 787 r = revlog.revlog(util.opener(os.getcwd(), audit=False),
788 788 file_[:-2] + ".i", file_, 0)
789 789 try:
790 790 ui.write(r.revision(r.lookup(rev)))
791 791 except KeyError:
792 792 raise util.Abort(_('invalid revision identifier %s') % rev)
793 793
794 794 def debugdate(ui, date, range=None, **opts):
795 795 """parse and display a date"""
796 796 if opts["extended"]:
797 797 d = util.parsedate(date, util.extendeddateformats)
798 798 else:
799 799 d = util.parsedate(date)
800 800 ui.write("internal: %s %s\n" % d)
801 801 ui.write("standard: %s\n" % util.datestr(d))
802 802 if range:
803 803 m = util.matchdate(range)
804 804 ui.write("match: %s\n" % m(d[0]))
805 805
806 806 def debugindex(ui, file_):
807 807 """dump the contents of an index file"""
808 808 r = revlog.revlog(util.opener(os.getcwd(), audit=False), file_, "", 0)
809 809 ui.write(" rev offset length base linkrev" +
810 810 " nodeid p1 p2\n")
811 811 for i in xrange(r.count()):
812 812 node = r.node(i)
813 813 pp = r.parents(node)
814 814 ui.write("% 6d % 9d % 7d % 6d % 7d %s %s %s\n" % (
815 815 i, r.start(i), r.length(i), r.base(i), r.linkrev(node),
816 816 short(node), short(pp[0]), short(pp[1])))
817 817
818 818 def debugindexdot(ui, file_):
819 819 """dump an index DAG as a .dot file"""
820 820 r = revlog.revlog(util.opener(os.getcwd(), audit=False), file_, "", 0)
821 821 ui.write("digraph G {\n")
822 822 for i in xrange(r.count()):
823 823 node = r.node(i)
824 824 pp = r.parents(node)
825 825 ui.write("\t%d -> %d\n" % (r.rev(pp[0]), i))
826 826 if pp[1] != nullid:
827 827 ui.write("\t%d -> %d\n" % (r.rev(pp[1]), i))
828 828 ui.write("}\n")
829 829
830 830 def debuginstall(ui):
831 831 '''test Mercurial installation'''
832 832
833 833 def writetemp(contents):
834 834 (fd, name) = tempfile.mkstemp()
835 835 f = os.fdopen(fd, "wb")
836 836 f.write(contents)
837 837 f.close()
838 838 return name
839 839
840 840 problems = 0
841 841
842 842 # encoding
843 843 ui.status(_("Checking encoding (%s)...\n") % util._encoding)
844 844 try:
845 845 util.fromlocal("test")
846 846 except util.Abort, inst:
847 847 ui.write(" %s\n" % inst)
848 848 ui.write(_(" (check that your locale is properly set)\n"))
849 849 problems += 1
850 850
851 851 # compiled modules
852 852 ui.status(_("Checking extensions...\n"))
853 853 try:
854 854 import bdiff, mpatch, base85
855 855 except Exception, inst:
856 856 ui.write(" %s\n" % inst)
857 857 ui.write(_(" One or more extensions could not be found"))
858 858 ui.write(_(" (check that you compiled the extensions)\n"))
859 859 problems += 1
860 860
861 861 # templates
862 862 ui.status(_("Checking templates...\n"))
863 863 try:
864 864 import templater
865 865 t = templater.templater(templater.templatepath("map-cmdline.default"))
866 866 except Exception, inst:
867 867 ui.write(" %s\n" % inst)
868 868 ui.write(_(" (templates seem to have been installed incorrectly)\n"))
869 869 problems += 1
870 870
871 871 # patch
872 872 ui.status(_("Checking patch...\n"))
873 873 path = os.environ.get('PATH', '')
874 874 patcher = util.find_in_path('gpatch', path,
875 875 util.find_in_path('patch', path, None))
876 876 if not patcher:
877 877 ui.write(_(" Can't find patch or gpatch in PATH\n"))
878 878 ui.write(_(" (specify a patch utility in your .hgrc file)\n"))
879 879 problems += 1
880 880 else:
881 881 # actually attempt a patch here
882 882 a = "1\n2\n3\n4\n"
883 883 b = "1\n2\n3\ninsert\n4\n"
884 884 d = mdiff.unidiff(a, None, b, None, "a")
885 885 fa = writetemp(a)
886 886 fd = writetemp(d)
887 887 fp = os.popen('%s %s %s' % (patcher, fa, fd))
888 888 files = []
889 889 output = ""
890 890 for line in fp:
891 891 output += line
892 892 if line.startswith('patching file '):
893 893 pf = util.parse_patch_output(line.rstrip())
894 894 files.append(pf)
895 895 if files != [fa]:
896 896 ui.write(_(" unexpected patch output!"))
897 897 ui.write(_(" (you may have an incompatible version of patch)\n"))
898 898 ui.write(output)
899 899 problems += 1
900 900 a = file(fa).read()
901 901 if a != b:
902 902 ui.write(_(" patch test failed!"))
903 903 ui.write(_(" (you may have an incompatible version of patch)\n"))
904 904 problems += 1
905 905 os.unlink(fa)
906 906 os.unlink(fd)
907 907
908 908 # merge helper
909 909 ui.status(_("Checking merge helper...\n"))
910 910 cmd = (os.environ.get("HGMERGE") or ui.config("ui", "merge")
911 911 or "hgmerge")
912 912 cmdpath = util.find_in_path(cmd, path)
913 913 if not cmdpath:
914 914 cmdpath = util.find_in_path(cmd.split()[0], path)
915 915 if not cmdpath:
916 916 if cmd == 'hgmerge':
917 917 ui.write(_(" No merge helper set and can't find default"
918 918 " hgmerge script in PATH\n"))
919 919 ui.write(_(" (specify a merge helper in your .hgrc file)\n"))
920 920 else:
921 921 ui.write(_(" Can't find merge helper '%s' in PATH\n") % cmd)
922 922 ui.write(_(" (specify a merge helper in your .hgrc file)\n"))
923 923 problems += 1
924 924 else:
925 925 # actually attempt a patch here
926 926 fa = writetemp("1\n2\n3\n4\n")
927 927 fl = writetemp("1\n2\n3\ninsert\n4\n")
928 928 fr = writetemp("begin\n1\n2\n3\n4\n")
929 929 r = os.system('%s %s %s %s' % (cmd, fl, fa, fr))
930 930 if r:
931 931 ui.write(_(" got unexpected merge error %d!") % r)
932 932 problems += 1
933 933 m = file(fl).read()
934 934 if m != "begin\n1\n2\n3\ninsert\n4\n":
935 935 ui.write(_(" got unexpected merge results!") % r)
936 936 ui.write(_(" (your merge helper may have the"
937 937 " wrong argument order)\n"))
938 938 ui.write(m)
939 939 os.unlink(fa)
940 940 os.unlink(fl)
941 941 os.unlink(fr)
942 942
943 943 # editor
944 944 ui.status(_("Checking commit editor...\n"))
945 945 editor = (os.environ.get("HGEDITOR") or
946 946 ui.config("ui", "editor") or
947 947 os.environ.get("EDITOR", "vi"))
948 948 cmdpath = util.find_in_path(editor, path)
949 949 if not cmdpath:
950 950 cmdpath = util.find_in_path(editor.split()[0], path)
951 951 if not cmdpath:
952 952 if editor == 'vi':
953 953 ui.write(_(" No commit editor set and can't find vi in PATH\n"))
954 954 ui.write(_(" (specify a commit editor in your .hgrc file)\n"))
955 955 else:
956 956 ui.write(_(" Can't find editor '%s' in PATH\n") % editor)
957 957 ui.write(_(" (specify a commit editor in your .hgrc file)\n"))
958 958 problems += 1
959 959
960 960 # check username
961 961 ui.status(_("Checking username...\n"))
962 962 user = os.environ.get("HGUSER")
963 963 if user is None:
964 964 user = ui.config("ui", "username")
965 965 if user is None:
966 966 user = os.environ.get("EMAIL")
967 967 if not user:
968 968 ui.warn(" ")
969 969 ui.username()
970 970 ui.write(_(" (specify a username in your .hgrc file)\n"))
971 971
972 972 if not problems:
973 973 ui.status(_("No problems detected\n"))
974 974 else:
975 975 ui.write(_("%s problems detected,"
976 976 " please check your install!\n") % problems)
977 977
978 978 return problems
979 979
980 980 def debugrename(ui, repo, file1, *pats, **opts):
981 981 """dump rename information"""
982 982
983 983 ctx = repo.changectx(opts.get('rev', 'tip'))
984 984 for src, abs, rel, exact in cmdutil.walk(repo, (file1,) + pats, opts,
985 985 ctx.node()):
986 986 m = ctx.filectx(abs).renamed()
987 987 if m:
988 988 ui.write(_("%s renamed from %s:%s\n") % (rel, m[0], hex(m[1])))
989 989 else:
990 990 ui.write(_("%s not renamed\n") % rel)
991 991
992 992 def debugwalk(ui, repo, *pats, **opts):
993 993 """show how files match on given patterns"""
994 994 items = list(cmdutil.walk(repo, pats, opts))
995 995 if not items:
996 996 return
997 997 fmt = '%%s %%-%ds %%-%ds %%s' % (
998 998 max([len(abs) for (src, abs, rel, exact) in items]),
999 999 max([len(rel) for (src, abs, rel, exact) in items]))
1000 1000 for src, abs, rel, exact in items:
1001 1001 line = fmt % (src, abs, rel, exact and 'exact' or '')
1002 1002 ui.write("%s\n" % line.rstrip())
1003 1003
1004 1004 def diff(ui, repo, *pats, **opts):
1005 1005 """diff repository (or selected files)
1006 1006
1007 1007 Show differences between revisions for the specified files.
1008 1008
1009 1009 Differences between files are shown using the unified diff format.
1010 1010
1011 1011 NOTE: diff may generate unexpected results for merges, as it will
1012 1012 default to comparing against the working directory's first parent
1013 1013 changeset if no revisions are specified.
1014 1014
1015 1015 When two revision arguments are given, then changes are shown
1016 1016 between those revisions. If only one revision is specified then
1017 1017 that revision is compared to the working directory, and, when no
1018 1018 revisions are specified, the working directory files are compared
1019 1019 to its parent.
1020 1020
1021 1021 Without the -a option, diff will avoid generating diffs of files
1022 1022 it detects as binary. With -a, diff will generate a diff anyway,
1023 1023 probably with undesirable results.
1024 1024 """
1025 1025 node1, node2 = cmdutil.revpair(repo, opts['rev'])
1026 1026
1027 1027 fns, matchfn, anypats = cmdutil.matchpats(repo, pats, opts)
1028 1028
1029 1029 patch.diff(repo, node1, node2, fns, match=matchfn,
1030 1030 opts=patch.diffopts(ui, opts))
1031 1031
1032 1032 def export(ui, repo, *changesets, **opts):
1033 1033 """dump the header and diffs for one or more changesets
1034 1034
1035 1035 Print the changeset header and diffs for one or more revisions.
1036 1036
1037 1037 The information shown in the changeset header is: author,
1038 1038 changeset hash, parent(s) and commit comment.
1039 1039
1040 1040 NOTE: export may generate unexpected diff output for merge changesets,
1041 1041 as it will compare the merge changeset against its first parent only.
1042 1042
1043 1043 Output may be to a file, in which case the name of the file is
1044 1044 given using a format string. The formatting rules are as follows:
1045 1045
1046 1046 %% literal "%" character
1047 1047 %H changeset hash (40 bytes of hexadecimal)
1048 1048 %N number of patches being generated
1049 1049 %R changeset revision number
1050 1050 %b basename of the exporting repository
1051 1051 %h short-form changeset hash (12 bytes of hexadecimal)
1052 1052 %n zero-padded sequence number, starting at 1
1053 1053 %r zero-padded changeset revision number
1054 1054
1055 1055 Without the -a option, export will avoid generating diffs of files
1056 1056 it detects as binary. With -a, export will generate a diff anyway,
1057 1057 probably with undesirable results.
1058 1058
1059 1059 With the --switch-parent option, the diff will be against the second
1060 1060 parent. It can be useful to review a merge.
1061 1061 """
1062 1062 if not changesets:
1063 1063 raise util.Abort(_("export requires at least one changeset"))
1064 1064 revs = cmdutil.revrange(repo, changesets)
1065 1065 if len(revs) > 1:
1066 1066 ui.note(_('exporting patches:\n'))
1067 1067 else:
1068 1068 ui.note(_('exporting patch:\n'))
1069 1069 patch.export(repo, map(repo.lookup, revs), template=opts['output'],
1070 1070 switch_parent=opts['switch_parent'],
1071 1071 opts=patch.diffopts(ui, opts))
1072 1072
1073 1073 def grep(ui, repo, pattern, *pats, **opts):
1074 1074 """search for a pattern in specified files and revisions
1075 1075
1076 1076 Search revisions of files for a regular expression.
1077 1077
1078 1078 This command behaves differently than Unix grep. It only accepts
1079 1079 Python/Perl regexps. It searches repository history, not the
1080 1080 working directory. It always prints the revision number in which
1081 1081 a match appears.
1082 1082
1083 1083 By default, grep only prints output for the first revision of a
1084 1084 file in which it finds a match. To get it to print every revision
1085 1085 that contains a change in match status ("-" for a match that
1086 1086 becomes a non-match, or "+" for a non-match that becomes a match),
1087 1087 use the --all flag.
1088 1088 """
1089 1089 reflags = 0
1090 1090 if opts['ignore_case']:
1091 1091 reflags |= re.I
1092 1092 regexp = re.compile(pattern, reflags)
1093 1093 sep, eol = ':', '\n'
1094 1094 if opts['print0']:
1095 1095 sep = eol = '\0'
1096 1096
1097 1097 fcache = {}
1098 1098 def getfile(fn):
1099 1099 if fn not in fcache:
1100 1100 fcache[fn] = repo.file(fn)
1101 1101 return fcache[fn]
1102 1102
1103 1103 def matchlines(body):
1104 1104 begin = 0
1105 1105 linenum = 0
1106 1106 while True:
1107 1107 match = regexp.search(body, begin)
1108 1108 if not match:
1109 1109 break
1110 1110 mstart, mend = match.span()
1111 1111 linenum += body.count('\n', begin, mstart) + 1
1112 1112 lstart = body.rfind('\n', begin, mstart) + 1 or begin
1113 1113 lend = body.find('\n', mend)
1114 1114 yield linenum, mstart - lstart, mend - lstart, body[lstart:lend]
1115 1115 begin = lend + 1
1116 1116
1117 1117 class linestate(object):
1118 1118 def __init__(self, line, linenum, colstart, colend):
1119 1119 self.line = line
1120 1120 self.linenum = linenum
1121 1121 self.colstart = colstart
1122 1122 self.colend = colend
1123 1123
1124 1124 def __eq__(self, other):
1125 1125 return self.line == other.line
1126 1126
1127 1127 matches = {}
1128 1128 copies = {}
1129 1129 def grepbody(fn, rev, body):
1130 1130 matches[rev].setdefault(fn, [])
1131 1131 m = matches[rev][fn]
1132 1132 for lnum, cstart, cend, line in matchlines(body):
1133 1133 s = linestate(line, lnum, cstart, cend)
1134 1134 m.append(s)
1135 1135
1136 1136 def difflinestates(a, b):
1137 1137 sm = difflib.SequenceMatcher(None, a, b)
1138 1138 for tag, alo, ahi, blo, bhi in sm.get_opcodes():
1139 1139 if tag == 'insert':
1140 1140 for i in xrange(blo, bhi):
1141 1141 yield ('+', b[i])
1142 1142 elif tag == 'delete':
1143 1143 for i in xrange(alo, ahi):
1144 1144 yield ('-', a[i])
1145 1145 elif tag == 'replace':
1146 1146 for i in xrange(alo, ahi):
1147 1147 yield ('-', a[i])
1148 1148 for i in xrange(blo, bhi):
1149 1149 yield ('+', b[i])
1150 1150
1151 1151 prev = {}
1152 1152 def display(fn, rev, states, prevstates):
1153 1153 counts = {'-': 0, '+': 0}
1154 1154 filerevmatches = {}
1155 1155 if incrementing or not opts['all']:
1156 1156 a, b, r = prevstates, states, rev
1157 1157 else:
1158 1158 a, b, r = states, prevstates, prev.get(fn, -1)
1159 1159 for change, l in difflinestates(a, b):
1160 1160 cols = [fn, str(r)]
1161 1161 if opts['line_number']:
1162 1162 cols.append(str(l.linenum))
1163 1163 if opts['all']:
1164 1164 cols.append(change)
1165 1165 if opts['user']:
1166 1166 cols.append(ui.shortuser(get(r)[1]))
1167 1167 if opts['files_with_matches']:
1168 1168 c = (fn, r)
1169 1169 if c in filerevmatches:
1170 1170 continue
1171 1171 filerevmatches[c] = 1
1172 1172 else:
1173 1173 cols.append(l.line)
1174 1174 ui.write(sep.join(cols), eol)
1175 1175 counts[change] += 1
1176 1176 return counts['+'], counts['-']
1177 1177
1178 1178 fstate = {}
1179 1179 skip = {}
1180 1180 get = util.cachefunc(lambda r: repo.changectx(r).changeset())
1181 1181 changeiter, matchfn = cmdutil.walkchangerevs(ui, repo, pats, get, opts)
1182 1182 count = 0
1183 1183 incrementing = False
1184 1184 follow = opts.get('follow')
1185 1185 for st, rev, fns in changeiter:
1186 1186 if st == 'window':
1187 1187 incrementing = rev
1188 1188 matches.clear()
1189 1189 elif st == 'add':
1190 1190 mf = repo.changectx(rev).manifest()
1191 1191 matches[rev] = {}
1192 1192 for fn in fns:
1193 1193 if fn in skip:
1194 1194 continue
1195 1195 fstate.setdefault(fn, {})
1196 1196 try:
1197 1197 grepbody(fn, rev, getfile(fn).read(mf[fn]))
1198 1198 if follow:
1199 1199 copied = getfile(fn).renamed(mf[fn])
1200 1200 if copied:
1201 1201 copies.setdefault(rev, {})[fn] = copied[0]
1202 1202 except KeyError:
1203 1203 pass
1204 1204 elif st == 'iter':
1205 1205 states = matches[rev].items()
1206 1206 states.sort()
1207 1207 for fn, m in states:
1208 1208 copy = copies.get(rev, {}).get(fn)
1209 1209 if fn in skip:
1210 1210 if copy:
1211 1211 skip[copy] = True
1212 1212 continue
1213 1213 if incrementing or not opts['all'] or fstate[fn]:
1214 1214 pos, neg = display(fn, rev, m, fstate[fn])
1215 1215 count += pos + neg
1216 1216 if pos and not opts['all']:
1217 1217 skip[fn] = True
1218 1218 if copy:
1219 1219 skip[copy] = True
1220 1220 fstate[fn] = m
1221 1221 if copy:
1222 1222 fstate[copy] = m
1223 1223 prev[fn] = rev
1224 1224
1225 1225 if not incrementing:
1226 1226 fstate = fstate.items()
1227 1227 fstate.sort()
1228 1228 for fn, state in fstate:
1229 1229 if fn in skip:
1230 1230 continue
1231 1231 if fn not in copies.get(prev[fn], {}):
1232 1232 display(fn, rev, {}, state)
1233 1233 return (count == 0 and 1) or 0
1234 1234
1235 1235 def heads(ui, repo, **opts):
1236 1236 """show current repository heads
1237 1237
1238 1238 Show all repository head changesets.
1239 1239
1240 1240 Repository "heads" are changesets that don't have children
1241 1241 changesets. They are where development generally takes place and
1242 1242 are the usual targets for update and merge operations.
1243 1243 """
1244 1244 if opts['rev']:
1245 1245 heads = repo.heads(repo.lookup(opts['rev']))
1246 1246 else:
1247 1247 heads = repo.heads()
1248 1248 displayer = cmdutil.show_changeset(ui, repo, opts)
1249 1249 for n in heads:
1250 1250 displayer.show(changenode=n)
1251 1251
1252 1252 def help_(ui, name=None, with_version=False):
1253 1253 """show help for a command, extension, or list of commands
1254 1254
1255 1255 With no arguments, print a list of commands and short help.
1256 1256
1257 1257 Given a command name, print help for that command.
1258 1258
1259 1259 Given an extension name, print help for that extension, and the
1260 1260 commands it provides."""
1261 1261 option_lists = []
1262 1262
1263 1263 def helpcmd(name):
1264 1264 if with_version:
1265 1265 version_(ui)
1266 1266 ui.write('\n')
1267 1267 aliases, i = findcmd(ui, name)
1268 1268 # synopsis
1269 1269 ui.write("%s\n\n" % i[2])
1270 1270
1271 1271 # description
1272 1272 doc = i[0].__doc__
1273 1273 if not doc:
1274 1274 doc = _("(No help text available)")
1275 1275 if ui.quiet:
1276 1276 doc = doc.splitlines(0)[0]
1277 1277 ui.write("%s\n" % doc.rstrip())
1278 1278
1279 1279 if not ui.quiet:
1280 1280 # aliases
1281 1281 if len(aliases) > 1:
1282 1282 ui.write(_("\naliases: %s\n") % ', '.join(aliases[1:]))
1283 1283
1284 1284 # options
1285 1285 if i[1]:
1286 1286 option_lists.append(("options", i[1]))
1287 1287
1288 1288 def helplist(select=None):
1289 1289 h = {}
1290 1290 cmds = {}
1291 1291 for c, e in table.items():
1292 1292 f = c.split("|", 1)[0]
1293 1293 if select and not select(f):
1294 1294 continue
1295 1295 if name == "shortlist" and not f.startswith("^"):
1296 1296 continue
1297 1297 f = f.lstrip("^")
1298 1298 if not ui.debugflag and f.startswith("debug"):
1299 1299 continue
1300 1300 doc = e[0].__doc__
1301 1301 if not doc:
1302 1302 doc = _("(No help text available)")
1303 1303 h[f] = doc.splitlines(0)[0].rstrip()
1304 1304 cmds[f] = c.lstrip("^")
1305 1305
1306 1306 fns = h.keys()
1307 1307 fns.sort()
1308 1308 m = max(map(len, fns))
1309 1309 for f in fns:
1310 1310 if ui.verbose:
1311 1311 commands = cmds[f].replace("|",", ")
1312 1312 ui.write(" %s:\n %s\n"%(commands, h[f]))
1313 1313 else:
1314 1314 ui.write(' %-*s %s\n' % (m, f, h[f]))
1315 1315
1316 1316 def helptopic(name):
1317 1317 v = None
1318 1318 for i in help.helptable:
1319 1319 l = i.split('|')
1320 1320 if name in l:
1321 1321 v = i
1322 1322 header = l[-1]
1323 1323 if not v:
1324 1324 raise UnknownCommand(name)
1325 1325
1326 1326 # description
1327 1327 doc = help.helptable[v]
1328 1328 if not doc:
1329 1329 doc = _("(No help text available)")
1330 1330 if callable(doc):
1331 1331 doc = doc()
1332 1332
1333 1333 ui.write("%s\n" % header)
1334 1334 ui.write("%s\n" % doc.rstrip())
1335 1335
1336 1336 def helpext(name):
1337 1337 try:
1338 1338 mod = findext(name)
1339 1339 except KeyError:
1340 1340 raise UnknownCommand(name)
1341 1341
1342 1342 doc = (mod.__doc__ or _('No help text available')).splitlines(0)
1343 1343 ui.write(_('%s extension - %s\n') % (name.split('.')[-1], doc[0]))
1344 1344 for d in doc[1:]:
1345 1345 ui.write(d, '\n')
1346 1346
1347 1347 ui.status('\n')
1348 1348 if ui.verbose:
1349 1349 ui.status(_('list of commands:\n\n'))
1350 1350 else:
1351 1351 ui.status(_('list of commands (use "hg help -v %s" '
1352 1352 'to show aliases and global options):\n\n') % name)
1353 1353
1354 1354 modcmds = dict.fromkeys([c.split('|', 1)[0] for c in mod.cmdtable])
1355 1355 helplist(modcmds.has_key)
1356 1356
1357 1357 if name and name != 'shortlist':
1358 1358 i = None
1359 1359 for f in (helpcmd, helptopic, helpext):
1360 1360 try:
1361 1361 f(name)
1362 1362 i = None
1363 1363 break
1364 1364 except UnknownCommand, inst:
1365 1365 i = inst
1366 1366 if i:
1367 1367 raise i
1368 1368
1369 1369 else:
1370 1370 # program name
1371 1371 if ui.verbose or with_version:
1372 1372 version_(ui)
1373 1373 else:
1374 1374 ui.status(_("Mercurial Distributed SCM\n"))
1375 1375 ui.status('\n')
1376 1376
1377 1377 # list of commands
1378 1378 if name == "shortlist":
1379 1379 ui.status(_('basic commands (use "hg help" '
1380 1380 'for the full list or option "-v" for details):\n\n'))
1381 1381 elif ui.verbose:
1382 1382 ui.status(_('list of commands:\n\n'))
1383 1383 else:
1384 1384 ui.status(_('list of commands (use "hg help -v" '
1385 1385 'to show aliases and global options):\n\n'))
1386 1386
1387 1387 helplist()
1388 1388
1389 1389 # global options
1390 1390 if ui.verbose:
1391 1391 option_lists.append(("global options", globalopts))
1392 1392
1393 1393 # list all option lists
1394 1394 opt_output = []
1395 1395 for title, options in option_lists:
1396 1396 opt_output.append(("\n%s:\n" % title, None))
1397 1397 for shortopt, longopt, default, desc in options:
1398 1398 if "DEPRECATED" in desc and not ui.verbose: continue
1399 1399 opt_output.append(("%2s%s" % (shortopt and "-%s" % shortopt,
1400 1400 longopt and " --%s" % longopt),
1401 1401 "%s%s" % (desc,
1402 1402 default
1403 1403 and _(" (default: %s)") % default
1404 1404 or "")))
1405 1405
1406 1406 if opt_output:
1407 1407 opts_len = max([len(line[0]) for line in opt_output if line[1]])
1408 1408 for first, second in opt_output:
1409 1409 if second:
1410 1410 ui.write(" %-*s %s\n" % (opts_len, first, second))
1411 1411 else:
1412 1412 ui.write("%s\n" % first)
1413 1413
1414 1414 def identify(ui, repo):
1415 1415 """print information about the working copy
1416 1416
1417 1417 Print a short summary of the current state of the repo.
1418 1418
1419 1419 This summary identifies the repository state using one or two parent
1420 1420 hash identifiers, followed by a "+" if there are uncommitted changes
1421 1421 in the working directory, followed by a list of tags for this revision.
1422 1422 """
1423 1423 parents = [p for p in repo.dirstate.parents() if p != nullid]
1424 1424 if not parents:
1425 1425 ui.write(_("unknown\n"))
1426 1426 return
1427 1427
1428 1428 hexfunc = ui.debugflag and hex or short
1429 1429 modified, added, removed, deleted = repo.status()[:4]
1430 1430 output = ["%s%s" %
1431 1431 ('+'.join([hexfunc(parent) for parent in parents]),
1432 1432 (modified or added or removed or deleted) and "+" or "")]
1433 1433
1434 1434 if not ui.quiet:
1435 1435
1436 1436 branch = repo.workingctx().branch()
1437 1437 if branch:
1438 1438 output.append("(%s)" % branch)
1439 1439
1440 1440 # multiple tags for a single parent separated by '/'
1441 1441 parenttags = ['/'.join(tags)
1442 1442 for tags in map(repo.nodetags, parents) if tags]
1443 1443 # tags for multiple parents separated by ' + '
1444 1444 if parenttags:
1445 1445 output.append(' + '.join(parenttags))
1446 1446
1447 1447 ui.write("%s\n" % ' '.join(output))
1448 1448
1449 1449 def import_(ui, repo, patch1, *patches, **opts):
1450 1450 """import an ordered set of patches
1451 1451
1452 1452 Import a list of patches and commit them individually.
1453 1453
1454 1454 If there are outstanding changes in the working directory, import
1455 1455 will abort unless given the -f flag.
1456 1456
1457 1457 You can import a patch straight from a mail message. Even patches
1458 1458 as attachments work (body part must be type text/plain or
1459 1459 text/x-patch to be used). From and Subject headers of email
1460 1460 message are used as default committer and commit message. All
1461 1461 text/plain body parts before first diff are added to commit
1462 1462 message.
1463 1463
1464 1464 If imported patch was generated by hg export, user and description
1465 1465 from patch override values from message headers and body. Values
1466 1466 given on command line with -m and -u override these.
1467 1467
1468 1468 To read a patch from standard input, use patch name "-".
1469 1469 """
1470 1470 patches = (patch1,) + patches
1471 1471
1472 1472 if not opts['force']:
1473 1473 bail_if_changed(repo)
1474 1474
1475 1475 d = opts["base"]
1476 1476 strip = opts["strip"]
1477 1477
1478 1478 wlock = repo.wlock()
1479 1479 lock = repo.lock()
1480 1480
1481 1481 for p in patches:
1482 1482 pf = os.path.join(d, p)
1483 1483
1484 1484 if pf == '-':
1485 1485 ui.status(_("applying patch from stdin\n"))
1486 1486 tmpname, message, user, date = patch.extract(ui, sys.stdin)
1487 1487 else:
1488 1488 ui.status(_("applying %s\n") % p)
1489 1489 tmpname, message, user, date = patch.extract(ui, file(pf))
1490 1490
1491 1491 if tmpname is None:
1492 1492 raise util.Abort(_('no diffs found'))
1493 1493
1494 1494 try:
1495 1495 cmdline_message = logmessage(opts)
1496 1496 if cmdline_message:
1497 1497 # pickup the cmdline msg
1498 1498 message = cmdline_message
1499 1499 elif message:
1500 1500 # pickup the patch msg
1501 1501 message = message.strip()
1502 1502 else:
1503 1503 # launch the editor
1504 1504 message = None
1505 1505 ui.debug(_('message:\n%s\n') % message)
1506 1506
1507 1507 files = {}
1508 1508 try:
1509 1509 fuzz = patch.patch(tmpname, ui, strip=strip, cwd=repo.root,
1510 1510 files=files)
1511 1511 finally:
1512 1512 files = patch.updatedir(ui, repo, files, wlock=wlock)
1513 1513 repo.commit(files, message, user, date, wlock=wlock, lock=lock)
1514 1514 finally:
1515 1515 os.unlink(tmpname)
1516 1516
1517 1517 def incoming(ui, repo, source="default", **opts):
1518 1518 """show new changesets found in source
1519 1519
1520 1520 Show new changesets found in the specified path/URL or the default
1521 1521 pull location. These are the changesets that would be pulled if a pull
1522 1522 was requested.
1523 1523
1524 1524 For remote repository, using --bundle avoids downloading the changesets
1525 1525 twice if the incoming is followed by a pull.
1526 1526
1527 1527 See pull for valid source format details.
1528 1528 """
1529 1529 source = ui.expandpath(source)
1530 1530 setremoteconfig(ui, opts)
1531 1531
1532 1532 other = hg.repository(ui, source)
1533 1533 incoming = repo.findincoming(other, force=opts["force"])
1534 1534 if not incoming:
1535 1535 ui.status(_("no changes found\n"))
1536 1536 return
1537 1537
1538 1538 cleanup = None
1539 1539 try:
1540 1540 fname = opts["bundle"]
1541 1541 if fname or not other.local():
1542 1542 # create a bundle (uncompressed if other repo is not local)
1543 1543 cg = other.changegroup(incoming, "incoming")
1544 1544 bundletype = other.local() and "HG10BZ" or "HG10UN"
1545 1545 fname = cleanup = changegroup.writebundle(cg, fname, bundletype)
1546 1546 # keep written bundle?
1547 1547 if opts["bundle"]:
1548 1548 cleanup = None
1549 1549 if not other.local():
1550 1550 # use the created uncompressed bundlerepo
1551 1551 other = bundlerepo.bundlerepository(ui, repo.root, fname)
1552 1552
1553 1553 revs = None
1554 1554 if opts['rev']:
1555 1555 revs = [other.lookup(rev) for rev in opts['rev']]
1556 1556 o = other.changelog.nodesbetween(incoming, revs)[0]
1557 1557 if opts['newest_first']:
1558 1558 o.reverse()
1559 1559 displayer = cmdutil.show_changeset(ui, other, opts)
1560 1560 for n in o:
1561 1561 parents = [p for p in other.changelog.parents(n) if p != nullid]
1562 1562 if opts['no_merges'] and len(parents) == 2:
1563 1563 continue
1564 1564 displayer.show(changenode=n)
1565 1565 finally:
1566 1566 if hasattr(other, 'close'):
1567 1567 other.close()
1568 1568 if cleanup:
1569 1569 os.unlink(cleanup)
1570 1570
1571 1571 def init(ui, dest=".", **opts):
1572 1572 """create a new repository in the given directory
1573 1573
1574 1574 Initialize a new repository in the given directory. If the given
1575 1575 directory does not exist, it is created.
1576 1576
1577 1577 If no directory is given, the current directory is used.
1578 1578
1579 1579 It is possible to specify an ssh:// URL as the destination.
1580 1580 Look at the help text for the pull command for important details
1581 1581 about ssh:// URLs.
1582 1582 """
1583 1583 setremoteconfig(ui, opts)
1584 1584 hg.repository(ui, dest, create=1)
1585 1585
1586 1586 def locate(ui, repo, *pats, **opts):
1587 1587 """locate files matching specific patterns
1588 1588
1589 1589 Print all files under Mercurial control whose names match the
1590 1590 given patterns.
1591 1591
1592 1592 This command searches the current directory and its
1593 1593 subdirectories. To search an entire repository, move to the root
1594 1594 of the repository.
1595 1595
1596 1596 If no patterns are given to match, this command prints all file
1597 1597 names.
1598 1598
1599 1599 If you want to feed the output of this command into the "xargs"
1600 1600 command, use the "-0" option to both this command and "xargs".
1601 1601 This will avoid the problem of "xargs" treating single filenames
1602 1602 that contain white space as multiple filenames.
1603 1603 """
1604 1604 end = opts['print0'] and '\0' or '\n'
1605 1605 rev = opts['rev']
1606 1606 if rev:
1607 1607 node = repo.lookup(rev)
1608 1608 else:
1609 1609 node = None
1610 1610
1611 1611 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts, node=node,
1612 1612 head='(?:.*/|)'):
1613 1613 if not node and repo.dirstate.state(abs) == '?':
1614 1614 continue
1615 1615 if opts['fullpath']:
1616 1616 ui.write(os.path.join(repo.root, abs), end)
1617 1617 else:
1618 1618 ui.write(((pats and rel) or abs), end)
1619 1619
1620 1620 def log(ui, repo, *pats, **opts):
1621 1621 """show revision history of entire repository or files
1622 1622
1623 1623 Print the revision history of the specified files or the entire
1624 1624 project.
1625 1625
1626 1626 File history is shown without following rename or copy history of
1627 1627 files. Use -f/--follow with a file name to follow history across
1628 1628 renames and copies. --follow without a file name will only show
1629 1629 ancestors or descendants of the starting revision. --follow-first
1630 1630 only follows the first parent of merge revisions.
1631 1631
1632 1632 If no revision range is specified, the default is tip:0 unless
1633 1633 --follow is set, in which case the working directory parent is
1634 1634 used as the starting revision.
1635 1635
1636 1636 By default this command outputs: changeset id and hash, tags,
1637 1637 non-trivial parents, user, date and time, and a summary for each
1638 1638 commit. When the -v/--verbose switch is used, the list of changed
1639 1639 files and full commit message is shown.
1640 1640
1641 1641 NOTE: log -p may generate unexpected diff output for merge
1642 1642 changesets, as it will compare the merge changeset against its
1643 1643 first parent only. Also, the files: list will only reflect files
1644 1644 that are different from BOTH parents.
1645 1645
1646 1646 """
1647 1647
1648 1648 get = util.cachefunc(lambda r: repo.changectx(r).changeset())
1649 1649 changeiter, matchfn = cmdutil.walkchangerevs(ui, repo, pats, get, opts)
1650 1650
1651 1651 if opts['limit']:
1652 1652 try:
1653 1653 limit = int(opts['limit'])
1654 1654 except ValueError:
1655 1655 raise util.Abort(_('limit must be a positive integer'))
1656 1656 if limit <= 0: raise util.Abort(_('limit must be positive'))
1657 1657 else:
1658 1658 limit = sys.maxint
1659 1659 count = 0
1660 1660
1661 1661 if opts['copies'] and opts['rev']:
1662 1662 endrev = max(cmdutil.revrange(repo, opts['rev'])) + 1
1663 1663 else:
1664 1664 endrev = repo.changelog.count()
1665 1665 rcache = {}
1666 1666 ncache = {}
1667 1667 dcache = []
1668 1668 def getrenamed(fn, rev, man):
1669 1669 '''looks up all renames for a file (up to endrev) the first
1670 1670 time the file is given. It indexes on the changerev and only
1671 1671 parses the manifest if linkrev != changerev.
1672 1672 Returns rename info for fn at changerev rev.'''
1673 1673 if fn not in rcache:
1674 1674 rcache[fn] = {}
1675 1675 ncache[fn] = {}
1676 1676 fl = repo.file(fn)
1677 1677 for i in xrange(fl.count()):
1678 1678 node = fl.node(i)
1679 1679 lr = fl.linkrev(node)
1680 1680 renamed = fl.renamed(node)
1681 1681 rcache[fn][lr] = renamed
1682 1682 if renamed:
1683 1683 ncache[fn][node] = renamed
1684 1684 if lr >= endrev:
1685 1685 break
1686 1686 if rev in rcache[fn]:
1687 1687 return rcache[fn][rev]
1688 1688 mr = repo.manifest.rev(man)
1689 1689 if repo.manifest.parentrevs(mr) != (mr - 1, nullrev):
1690 1690 return ncache[fn].get(repo.manifest.find(man, fn)[0])
1691 1691 if not dcache or dcache[0] != man:
1692 1692 dcache[:] = [man, repo.manifest.readdelta(man)]
1693 1693 if fn in dcache[1]:
1694 1694 return ncache[fn].get(dcache[1][fn])
1695 1695 return None
1696 1696
1697 1697 df = False
1698 1698 if opts["date"]:
1699 1699 df = util.matchdate(opts["date"])
1700 1700
1701 1701
1702 1702 displayer = cmdutil.show_changeset(ui, repo, opts, True, matchfn)
1703 1703 for st, rev, fns in changeiter:
1704 1704 if st == 'add':
1705 1705 changenode = repo.changelog.node(rev)
1706 1706 parents = [p for p in repo.changelog.parentrevs(rev)
1707 1707 if p != nullrev]
1708 1708 if opts['no_merges'] and len(parents) == 2:
1709 1709 continue
1710 1710 if opts['only_merges'] and len(parents) != 2:
1711 1711 continue
1712 1712
1713 1713 if df:
1714 1714 changes = get(rev)
1715 1715 if not df(changes[2][0]):
1716 1716 continue
1717 1717
1718 1718 if opts['keyword']:
1719 1719 changes = get(rev)
1720 1720 miss = 0
1721 1721 for k in [kw.lower() for kw in opts['keyword']]:
1722 1722 if not (k in changes[1].lower() or
1723 1723 k in changes[4].lower() or
1724 1724 k in " ".join(changes[3][:20]).lower()):
1725 1725 miss = 1
1726 1726 break
1727 1727 if miss:
1728 1728 continue
1729 1729
1730 1730 copies = []
1731 1731 if opts.get('copies') and rev:
1732 1732 mf = get(rev)[0]
1733 1733 for fn in get(rev)[3]:
1734 1734 rename = getrenamed(fn, rev, mf)
1735 1735 if rename:
1736 1736 copies.append((fn, rename[0]))
1737 1737 displayer.show(rev, changenode, copies=copies)
1738 1738 elif st == 'iter':
1739 1739 if count == limit: break
1740 1740 if displayer.flush(rev):
1741 1741 count += 1
1742 1742
1743 1743 def manifest(ui, repo, rev=None):
1744 1744 """output the latest or given revision of the project manifest
1745 1745
1746 1746 Print a list of version controlled files for the given revision.
1747 1747
1748 1748 The manifest is the list of files being version controlled. If no revision
1749 1749 is given then the first parent of the working directory is used.
1750 1750
1751 1751 With -v flag, print file permissions. With --debug flag, print
1752 1752 file revision hashes.
1753 1753 """
1754 1754
1755 1755 m = repo.changectx(rev).manifest()
1756 1756 files = m.keys()
1757 1757 files.sort()
1758 1758
1759 1759 for f in files:
1760 1760 if ui.debugflag:
1761 1761 ui.write("%40s " % hex(m[f]))
1762 1762 if ui.verbose:
1763 1763 ui.write("%3s " % (m.execf(f) and "755" or "644"))
1764 1764 ui.write("%s\n" % f)
1765 1765
1766 1766 def merge(ui, repo, node=None, force=None, branch=None):
1767 1767 """Merge working directory with another revision
1768 1768
1769 1769 Merge the contents of the current working directory and the
1770 1770 requested revision. Files that changed between either parent are
1771 1771 marked as changed for the next commit and a commit must be
1772 1772 performed before any further updates are allowed.
1773 1773
1774 1774 If no revision is specified, the working directory's parent is a
1775 1775 head revision, and the repository contains exactly one other head,
1776 1776 the other head is merged with by default. Otherwise, an explicit
1777 1777 revision to merge with must be provided.
1778 1778 """
1779 1779
1780 1780 if node or branch:
1781 1781 node = _lookup(repo, node, branch)
1782 1782 else:
1783 1783 heads = repo.heads()
1784 1784 if len(heads) > 2:
1785 1785 raise util.Abort(_('repo has %d heads - '
1786 1786 'please merge with an explicit rev') %
1787 1787 len(heads))
1788 1788 if len(heads) == 1:
1789 1789 raise util.Abort(_('there is nothing to merge - '
1790 1790 'use "hg update" instead'))
1791 1791 parent = repo.dirstate.parents()[0]
1792 1792 if parent not in heads:
1793 1793 raise util.Abort(_('working dir not at a head rev - '
1794 1794 'use "hg update" or merge with an explicit rev'))
1795 1795 node = parent == heads[0] and heads[-1] or heads[0]
1796 1796 return hg.merge(repo, node, force=force)
1797 1797
1798 1798 def outgoing(ui, repo, dest=None, **opts):
1799 1799 """show changesets not found in destination
1800 1800
1801 1801 Show changesets not found in the specified destination repository or
1802 1802 the default push location. These are the changesets that would be pushed
1803 1803 if a push was requested.
1804 1804
1805 1805 See pull for valid destination format details.
1806 1806 """
1807 1807 dest = ui.expandpath(dest or 'default-push', dest or 'default')
1808 1808 setremoteconfig(ui, opts)
1809 1809 revs = None
1810 1810 if opts['rev']:
1811 1811 revs = [repo.lookup(rev) for rev in opts['rev']]
1812 1812
1813 1813 other = hg.repository(ui, dest)
1814 1814 o = repo.findoutgoing(other, force=opts['force'])
1815 1815 if not o:
1816 1816 ui.status(_("no changes found\n"))
1817 1817 return
1818 1818 o = repo.changelog.nodesbetween(o, revs)[0]
1819 1819 if opts['newest_first']:
1820 1820 o.reverse()
1821 1821 displayer = cmdutil.show_changeset(ui, repo, opts)
1822 1822 for n in o:
1823 1823 parents = [p for p in repo.changelog.parents(n) if p != nullid]
1824 1824 if opts['no_merges'] and len(parents) == 2:
1825 1825 continue
1826 1826 displayer.show(changenode=n)
1827 1827
1828 1828 def parents(ui, repo, file_=None, **opts):
1829 1829 """show the parents of the working dir or revision
1830 1830
1831 1831 Print the working directory's parent revisions.
1832 1832 """
1833 1833 rev = opts.get('rev')
1834 1834 if rev:
1835 1835 if file_:
1836 1836 ctx = repo.filectx(file_, changeid=rev)
1837 1837 else:
1838 1838 ctx = repo.changectx(rev)
1839 1839 p = [cp.node() for cp in ctx.parents()]
1840 1840 else:
1841 1841 p = repo.dirstate.parents()
1842 1842
1843 1843 displayer = cmdutil.show_changeset(ui, repo, opts)
1844 1844 for n in p:
1845 1845 if n != nullid:
1846 1846 displayer.show(changenode=n)
1847 1847
1848 1848 def paths(ui, repo, search=None):
1849 1849 """show definition of symbolic path names
1850 1850
1851 1851 Show definition of symbolic path name NAME. If no name is given, show
1852 1852 definition of available names.
1853 1853
1854 1854 Path names are defined in the [paths] section of /etc/mercurial/hgrc
1855 1855 and $HOME/.hgrc. If run inside a repository, .hg/hgrc is used, too.
1856 1856 """
1857 1857 if search:
1858 1858 for name, path in ui.configitems("paths"):
1859 1859 if name == search:
1860 1860 ui.write("%s\n" % path)
1861 1861 return
1862 1862 ui.warn(_("not found!\n"))
1863 1863 return 1
1864 1864 else:
1865 1865 for name, path in ui.configitems("paths"):
1866 1866 ui.write("%s = %s\n" % (name, path))
1867 1867
1868 1868 def postincoming(ui, repo, modheads, optupdate):
1869 1869 if modheads == 0:
1870 1870 return
1871 1871 if optupdate:
1872 1872 if modheads == 1:
1873 1873 return hg.update(repo, repo.changelog.tip()) # update
1874 1874 else:
1875 1875 ui.status(_("not updating, since new heads added\n"))
1876 1876 if modheads > 1:
1877 1877 ui.status(_("(run 'hg heads' to see heads, 'hg merge' to merge)\n"))
1878 1878 else:
1879 1879 ui.status(_("(run 'hg update' to get a working copy)\n"))
1880 1880
1881 1881 def pull(ui, repo, source="default", **opts):
1882 1882 """pull changes from the specified source
1883 1883
1884 1884 Pull changes from a remote repository to a local one.
1885 1885
1886 1886 This finds all changes from the repository at the specified path
1887 1887 or URL and adds them to the local repository. By default, this
1888 1888 does not update the copy of the project in the working directory.
1889 1889
1890 1890 Valid URLs are of the form:
1891 1891
1892 1892 local/filesystem/path (or file://local/filesystem/path)
1893 1893 http://[user@]host[:port]/[path]
1894 1894 https://[user@]host[:port]/[path]
1895 1895 ssh://[user@]host[:port]/[path]
1896 1896 static-http://host[:port]/[path]
1897 1897
1898 1898 Paths in the local filesystem can either point to Mercurial
1899 1899 repositories or to bundle files (as created by 'hg bundle' or
1900 1900 'hg incoming --bundle'). The static-http:// protocol, albeit slow,
1901 1901 allows access to a Mercurial repository where you simply use a web
1902 1902 server to publish the .hg directory as static content.
1903 1903
1904 1904 Some notes about using SSH with Mercurial:
1905 1905 - SSH requires an accessible shell account on the destination machine
1906 1906 and a copy of hg in the remote path or specified with as remotecmd.
1907 1907 - path is relative to the remote user's home directory by default.
1908 1908 Use an extra slash at the start of a path to specify an absolute path:
1909 1909 ssh://example.com//tmp/repository
1910 1910 - Mercurial doesn't use its own compression via SSH; the right thing
1911 1911 to do is to configure it in your ~/.ssh/config, e.g.:
1912 1912 Host *.mylocalnetwork.example.com
1913 1913 Compression no
1914 1914 Host *
1915 1915 Compression yes
1916 1916 Alternatively specify "ssh -C" as your ssh command in your hgrc or
1917 1917 with the --ssh command line option.
1918 1918 """
1919 1919 source = ui.expandpath(source)
1920 1920 setremoteconfig(ui, opts)
1921 1921
1922 1922 other = hg.repository(ui, source)
1923 1923 ui.status(_('pulling from %s\n') % (source))
1924 1924 revs = None
1925 1925 if opts['rev']:
1926 1926 if 'lookup' in other.capabilities:
1927 1927 revs = [other.lookup(rev) for rev in opts['rev']]
1928 1928 else:
1929 1929 error = _("Other repository doesn't support revision lookup, so a rev cannot be specified.")
1930 1930 raise util.Abort(error)
1931 1931 modheads = repo.pull(other, heads=revs, force=opts['force'])
1932 1932 return postincoming(ui, repo, modheads, opts['update'])
1933 1933
1934 1934 def push(ui, repo, dest=None, **opts):
1935 1935 """push changes to the specified destination
1936 1936
1937 1937 Push changes from the local repository to the given destination.
1938 1938
1939 1939 This is the symmetrical operation for pull. It helps to move
1940 1940 changes from the current repository to a different one. If the
1941 1941 destination is local this is identical to a pull in that directory
1942 1942 from the current one.
1943 1943
1944 1944 By default, push will refuse to run if it detects the result would
1945 1945 increase the number of remote heads. This generally indicates the
1946 1946 the client has forgotten to sync and merge before pushing.
1947 1947
1948 1948 Valid URLs are of the form:
1949 1949
1950 1950 local/filesystem/path (or file://local/filesystem/path)
1951 1951 ssh://[user@]host[:port]/[path]
1952 1952 http://[user@]host[:port]/[path]
1953 1953 https://[user@]host[:port]/[path]
1954 1954
1955 1955 Look at the help text for the pull command for important details
1956 1956 about ssh:// URLs.
1957 1957
1958 1958 Pushing to http:// and https:// URLs is only possible, if this
1959 1959 feature is explicitly enabled on the remote Mercurial server.
1960 1960 """
1961 1961 dest = ui.expandpath(dest or 'default-push', dest or 'default')
1962 1962 setremoteconfig(ui, opts)
1963 1963
1964 1964 other = hg.repository(ui, dest)
1965 1965 ui.status('pushing to %s\n' % (dest))
1966 1966 revs = None
1967 1967 if opts['rev']:
1968 1968 revs = [repo.lookup(rev) for rev in opts['rev']]
1969 1969 r = repo.push(other, opts['force'], revs=revs)
1970 1970 return r == 0
1971 1971
1972 1972 def rawcommit(ui, repo, *pats, **opts):
1973 1973 """raw commit interface (DEPRECATED)
1974 1974
1975 1975 (DEPRECATED)
1976 1976 Lowlevel commit, for use in helper scripts.
1977 1977
1978 1978 This command is not intended to be used by normal users, as it is
1979 1979 primarily useful for importing from other SCMs.
1980 1980
1981 1981 This command is now deprecated and will be removed in a future
1982 1982 release, please use debugsetparents and commit instead.
1983 1983 """
1984 1984
1985 1985 ui.warn(_("(the rawcommit command is deprecated)\n"))
1986 1986
1987 1987 message = logmessage(opts)
1988 1988
1989 1989 files, match, anypats = cmdutil.matchpats(repo, pats, opts)
1990 1990 if opts['files']:
1991 1991 files += open(opts['files']).read().splitlines()
1992 1992
1993 1993 parents = [repo.lookup(p) for p in opts['parent']]
1994 1994
1995 1995 try:
1996 1996 repo.rawcommit(files, message, opts['user'], opts['date'], *parents)
1997 1997 except ValueError, inst:
1998 1998 raise util.Abort(str(inst))
1999 1999
2000 2000 def recover(ui, repo):
2001 2001 """roll back an interrupted transaction
2002 2002
2003 2003 Recover from an interrupted commit or pull.
2004 2004
2005 2005 This command tries to fix the repository status after an interrupted
2006 2006 operation. It should only be necessary when Mercurial suggests it.
2007 2007 """
2008 2008 if repo.recover():
2009 2009 return hg.verify(repo)
2010 2010 return 1
2011 2011
2012 2012 def remove(ui, repo, *pats, **opts):
2013 2013 """remove the specified files on the next commit
2014 2014
2015 2015 Schedule the indicated files for removal from the repository.
2016 2016
2017 2017 This only removes files from the current branch, not from the
2018 2018 entire project history. If the files still exist in the working
2019 2019 directory, they will be deleted from it. If invoked with --after,
2020 2020 files that have been manually deleted are marked as removed.
2021 2021
2022 2022 This command schedules the files to be removed at the next commit.
2023 2023 To undo a remove before that, see hg revert.
2024 2024
2025 2025 Modified files and added files are not removed by default. To
2026 2026 remove them, use the -f/--force option.
2027 2027 """
2028 2028 names = []
2029 2029 if not opts['after'] and not pats:
2030 2030 raise util.Abort(_('no files specified'))
2031 2031 files, matchfn, anypats = cmdutil.matchpats(repo, pats, opts)
2032 2032 exact = dict.fromkeys(files)
2033 2033 mardu = map(dict.fromkeys, repo.status(files=files, match=matchfn))[:5]
2034 2034 modified, added, removed, deleted, unknown = mardu
2035 2035 remove, forget = [], []
2036 2036 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts):
2037 2037 reason = None
2038 2038 if abs not in deleted and opts['after']:
2039 2039 reason = _('is still present')
2040 2040 elif abs in modified and not opts['force']:
2041 2041 reason = _('is modified (use -f to force removal)')
2042 2042 elif abs in added:
2043 2043 if opts['force']:
2044 2044 forget.append(abs)
2045 2045 continue
2046 2046 reason = _('has been marked for add (use -f to force removal)')
2047 2047 elif abs in unknown:
2048 2048 reason = _('is not managed')
2049 2049 elif abs in removed:
2050 2050 continue
2051 2051 if reason:
2052 2052 if exact:
2053 2053 ui.warn(_('not removing %s: file %s\n') % (rel, reason))
2054 2054 else:
2055 2055 if ui.verbose or not exact:
2056 2056 ui.status(_('removing %s\n') % rel)
2057 2057 remove.append(abs)
2058 2058 repo.forget(forget)
2059 2059 repo.remove(remove, unlink=not opts['after'])
2060 2060
2061 2061 def rename(ui, repo, *pats, **opts):
2062 2062 """rename files; equivalent of copy + remove
2063 2063
2064 2064 Mark dest as copies of sources; mark sources for deletion. If
2065 2065 dest is a directory, copies are put in that directory. If dest is
2066 2066 a file, there can only be one source.
2067 2067
2068 2068 By default, this command copies the contents of files as they
2069 2069 stand in the working directory. If invoked with --after, the
2070 2070 operation is recorded, but no copying is performed.
2071 2071
2072 2072 This command takes effect in the next commit. To undo a rename
2073 2073 before that, see hg revert.
2074 2074 """
2075 2075 wlock = repo.wlock(0)
2076 2076 errs, copied = docopy(ui, repo, pats, opts, wlock)
2077 2077 names = []
2078 2078 for abs, rel, exact in copied:
2079 2079 if ui.verbose or not exact:
2080 2080 ui.status(_('removing %s\n') % rel)
2081 2081 names.append(abs)
2082 2082 if not opts.get('dry_run'):
2083 2083 repo.remove(names, True, wlock)
2084 2084 return errs
2085 2085
2086 2086 def revert(ui, repo, *pats, **opts):
2087 2087 """revert files or dirs to their states as of some revision
2088 2088
2089 2089 With no revision specified, revert the named files or directories
2090 2090 to the contents they had in the parent of the working directory.
2091 2091 This restores the contents of the affected files to an unmodified
2092 2092 state and unschedules adds, removes, copies, and renames. If the
2093 2093 working directory has two parents, you must explicitly specify the
2094 2094 revision to revert to.
2095 2095
2096 2096 Modified files are saved with a .orig suffix before reverting.
2097 2097 To disable these backups, use --no-backup.
2098 2098
2099 2099 Using the -r option, revert the given files or directories to their
2100 2100 contents as of a specific revision. This can be helpful to "roll
2101 2101 back" some or all of a change that should not have been committed.
2102 2102
2103 2103 Revert modifies the working directory. It does not commit any
2104 2104 changes, or change the parent of the working directory. If you
2105 2105 revert to a revision other than the parent of the working
2106 2106 directory, the reverted files will thus appear modified
2107 2107 afterwards.
2108 2108
2109 2109 If a file has been deleted, it is recreated. If the executable
2110 2110 mode of a file was changed, it is reset.
2111 2111
2112 2112 If names are given, all files matching the names are reverted.
2113 2113
2114 2114 If no arguments are given, no files are reverted.
2115 2115 """
2116 2116
2117 2117 if opts["date"]:
2118 2118 if opts["rev"]:
2119 2119 raise util.Abort(_("you can't specify a revision and a date"))
2120 2120 opts["rev"] = cmdutil.finddate(ui, repo, opts["date"])
2121 2121
2122 2122 if not pats and not opts['all']:
2123 2123 raise util.Abort(_('no files or directories specified; '
2124 2124 'use --all to revert the whole repo'))
2125 2125
2126 2126 parent, p2 = repo.dirstate.parents()
2127 2127 if not opts['rev'] and p2 != nullid:
2128 2128 raise util.Abort(_('uncommitted merge - please provide a '
2129 2129 'specific revision'))
2130 2130 node = repo.changectx(opts['rev']).node()
2131 2131 mf = repo.manifest.read(repo.changelog.read(node)[0])
2132 2132 if node == parent:
2133 2133 pmf = mf
2134 2134 else:
2135 2135 pmf = None
2136 2136
2137 2137 wlock = repo.wlock()
2138 2138
2139 2139 # need all matching names in dirstate and manifest of target rev,
2140 2140 # so have to walk both. do not print errors if files exist in one
2141 2141 # but not other.
2142 2142
2143 2143 names = {}
2144 2144 target_only = {}
2145 2145
2146 2146 # walk dirstate.
2147 2147
2148 2148 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts,
2149 2149 badmatch=mf.has_key):
2150 2150 names[abs] = (rel, exact)
2151 2151 if src == 'b':
2152 2152 target_only[abs] = True
2153 2153
2154 2154 # walk target manifest.
2155 2155
2156 2156 for src, abs, rel, exact in cmdutil.walk(repo, pats, opts, node=node,
2157 2157 badmatch=names.has_key):
2158 2158 if abs in names: continue
2159 2159 names[abs] = (rel, exact)
2160 2160 target_only[abs] = True
2161 2161
2162 2162 changes = repo.status(match=names.has_key, wlock=wlock)[:5]
2163 2163 modified, added, removed, deleted, unknown = map(dict.fromkeys, changes)
2164 2164
2165 2165 revert = ([], _('reverting %s\n'))
2166 2166 add = ([], _('adding %s\n'))
2167 2167 remove = ([], _('removing %s\n'))
2168 2168 forget = ([], _('forgetting %s\n'))
2169 2169 undelete = ([], _('undeleting %s\n'))
2170 2170 update = {}
2171 2171
2172 2172 disptable = (
2173 2173 # dispatch table:
2174 2174 # file state
2175 2175 # action if in target manifest
2176 2176 # action if not in target manifest
2177 2177 # make backup if in target manifest
2178 2178 # make backup if not in target manifest
2179 2179 (modified, revert, remove, True, True),
2180 2180 (added, revert, forget, True, False),
2181 2181 (removed, undelete, None, False, False),
2182 2182 (deleted, revert, remove, False, False),
2183 2183 (unknown, add, None, True, False),
2184 2184 (target_only, add, None, False, False),
2185 2185 )
2186 2186
2187 2187 entries = names.items()
2188 2188 entries.sort()
2189 2189
2190 2190 for abs, (rel, exact) in entries:
2191 2191 mfentry = mf.get(abs)
2192 2192 def handle(xlist, dobackup):
2193 2193 xlist[0].append(abs)
2194 2194 update[abs] = 1
2195 2195 if dobackup and not opts['no_backup'] and os.path.exists(rel):
2196 2196 bakname = "%s.orig" % rel
2197 2197 ui.note(_('saving current version of %s as %s\n') %
2198 2198 (rel, bakname))
2199 2199 if not opts.get('dry_run'):
2200 2200 util.copyfile(rel, bakname)
2201 2201 if ui.verbose or not exact:
2202 2202 ui.status(xlist[1] % rel)
2203 2203 for table, hitlist, misslist, backuphit, backupmiss in disptable:
2204 2204 if abs not in table: continue
2205 2205 # file has changed in dirstate
2206 2206 if mfentry:
2207 2207 handle(hitlist, backuphit)
2208 2208 elif misslist is not None:
2209 2209 handle(misslist, backupmiss)
2210 2210 else:
2211 2211 if exact: ui.warn(_('file not managed: %s\n') % rel)
2212 2212 break
2213 2213 else:
2214 2214 # file has not changed in dirstate
2215 2215 if node == parent:
2216 2216 if exact: ui.warn(_('no changes needed to %s\n') % rel)
2217 2217 continue
2218 2218 if pmf is None:
2219 2219 # only need parent manifest in this unlikely case,
2220 2220 # so do not read by default
2221 2221 pmf = repo.manifest.read(repo.changelog.read(parent)[0])
2222 2222 if abs in pmf:
2223 2223 if mfentry:
2224 2224 # if version of file is same in parent and target
2225 2225 # manifests, do nothing
2226 2226 if pmf[abs] != mfentry:
2227 2227 handle(revert, False)
2228 2228 else:
2229 2229 handle(remove, False)
2230 2230
2231 2231 if not opts.get('dry_run'):
2232 2232 repo.dirstate.forget(forget[0])
2233 2233 r = hg.revert(repo, node, update.has_key, wlock)
2234 2234 repo.dirstate.update(add[0], 'a')
2235 2235 repo.dirstate.update(undelete[0], 'n')
2236 2236 repo.dirstate.update(remove[0], 'r')
2237 2237 return r
2238 2238
2239 2239 def rollback(ui, repo):
2240 2240 """roll back the last transaction in this repository
2241 2241
2242 2242 Roll back the last transaction in this repository, restoring the
2243 2243 project to its state prior to the transaction.
2244 2244
2245 2245 Transactions are used to encapsulate the effects of all commands
2246 2246 that create new changesets or propagate existing changesets into a
2247 2247 repository. For example, the following commands are transactional,
2248 2248 and their effects can be rolled back:
2249 2249
2250 2250 commit
2251 2251 import
2252 2252 pull
2253 2253 push (with this repository as destination)
2254 2254 unbundle
2255 2255
2256 2256 This command should be used with care. There is only one level of
2257 2257 rollback, and there is no way to undo a rollback.
2258 2258
2259 2259 This command is not intended for use on public repositories. Once
2260 2260 changes are visible for pull by other users, rolling a transaction
2261 2261 back locally is ineffective (someone else may already have pulled
2262 2262 the changes). Furthermore, a race is possible with readers of the
2263 2263 repository; for example an in-progress pull from the repository
2264 2264 may fail if a rollback is performed.
2265 2265 """
2266 2266 repo.rollback()
2267 2267
2268 2268 def root(ui, repo):
2269 2269 """print the root (top) of the current working dir
2270 2270
2271 2271 Print the root directory of the current repository.
2272 2272 """
2273 2273 ui.write(repo.root + "\n")
2274 2274
2275 2275 def serve(ui, repo, **opts):
2276 2276 """export the repository via HTTP
2277 2277
2278 2278 Start a local HTTP repository browser and pull server.
2279 2279
2280 2280 By default, the server logs accesses to stdout and errors to
2281 2281 stderr. Use the "-A" and "-E" options to log to files.
2282 2282 """
2283 2283
2284 2284 if opts["stdio"]:
2285 2285 if repo is None:
2286 2286 raise hg.RepoError(_("There is no Mercurial repository here"
2287 2287 " (.hg not found)"))
2288 2288 s = sshserver.sshserver(ui, repo)
2289 2289 s.serve_forever()
2290 2290
2291 2291 optlist = ("name templates style address port ipv6"
2292 2292 " accesslog errorlog webdir_conf")
2293 2293 for o in optlist.split():
2294 2294 if opts[o]:
2295 2295 ui.setconfig("web", o, str(opts[o]))
2296 2296
2297 2297 if repo is None and not ui.config("web", "webdir_conf"):
2298 2298 raise hg.RepoError(_("There is no Mercurial repository here"
2299 2299 " (.hg not found)"))
2300 2300
2301 2301 if opts['daemon'] and not opts['daemon_pipefds']:
2302 2302 rfd, wfd = os.pipe()
2303 2303 args = sys.argv[:]
2304 2304 args.append('--daemon-pipefds=%d,%d' % (rfd, wfd))
2305 2305 pid = os.spawnvp(os.P_NOWAIT | getattr(os, 'P_DETACH', 0),
2306 2306 args[0], args)
2307 2307 os.close(wfd)
2308 2308 os.read(rfd, 1)
2309 2309 os._exit(0)
2310 2310
2311 2311 httpd = hgweb.server.create_server(ui, repo)
2312 2312
2313 2313 if ui.verbose:
2314 2314 if httpd.port != 80:
2315 2315 ui.status(_('listening at http://%s:%d/\n') %
2316 2316 (httpd.addr, httpd.port))
2317 2317 else:
2318 2318 ui.status(_('listening at http://%s/\n') % httpd.addr)
2319 2319
2320 2320 if opts['pid_file']:
2321 2321 fp = open(opts['pid_file'], 'w')
2322 2322 fp.write(str(os.getpid()) + '\n')
2323 2323 fp.close()
2324 2324
2325 2325 if opts['daemon_pipefds']:
2326 2326 rfd, wfd = [int(x) for x in opts['daemon_pipefds'].split(',')]
2327 2327 os.close(rfd)
2328 2328 os.write(wfd, 'y')
2329 2329 os.close(wfd)
2330 2330 sys.stdout.flush()
2331 2331 sys.stderr.flush()
2332 2332 fd = os.open(util.nulldev, os.O_RDWR)
2333 2333 if fd != 0: os.dup2(fd, 0)
2334 2334 if fd != 1: os.dup2(fd, 1)
2335 2335 if fd != 2: os.dup2(fd, 2)
2336 2336 if fd not in (0, 1, 2): os.close(fd)
2337 2337
2338 2338 httpd.serve_forever()
2339 2339
2340 2340 def status(ui, repo, *pats, **opts):
2341 2341 """show changed files in the working directory
2342 2342
2343 2343 Show status of files in the repository. If names are given, only
2344 2344 files that match are shown. Files that are clean or ignored, are
2345 2345 not listed unless -c (clean), -i (ignored) or -A is given.
2346 2346
2347 2347 NOTE: status may appear to disagree with diff if permissions have
2348 2348 changed or a merge has occurred. The standard diff format does not
2349 2349 report permission changes and diff only reports changes relative
2350 2350 to one merge parent.
2351 2351
2352 2352 If one revision is given, it is used as the base revision.
2353 2353 If two revisions are given, the difference between them is shown.
2354 2354
2355 2355 The codes used to show the status of files are:
2356 2356 M = modified
2357 2357 A = added
2358 2358 R = removed
2359 2359 C = clean
2360 2360 ! = deleted, but still tracked
2361 2361 ? = not tracked
2362 2362 I = ignored (not shown by default)
2363 2363 = the previous added file was copied from here
2364 2364 """
2365 2365
2366 2366 all = opts['all']
2367 2367 node1, node2 = cmdutil.revpair(repo, opts.get('rev'))
2368 2368
2369 2369 files, matchfn, anypats = cmdutil.matchpats(repo, pats, opts)
2370 2370 cwd = (pats and repo.getcwd()) or ''
2371 2371 modified, added, removed, deleted, unknown, ignored, clean = [
2372 2372 [util.pathto(cwd, x) for x in n]
2373 2373 for n in repo.status(node1=node1, node2=node2, files=files,
2374 2374 match=matchfn,
2375 2375 list_ignored=all or opts['ignored'],
2376 2376 list_clean=all or opts['clean'])]
2377 2377
2378 2378 changetypes = (('modified', 'M', modified),
2379 2379 ('added', 'A', added),
2380 2380 ('removed', 'R', removed),
2381 2381 ('deleted', '!', deleted),
2382 2382 ('unknown', '?', unknown),
2383 2383 ('ignored', 'I', ignored))
2384 2384
2385 2385 explicit_changetypes = changetypes + (('clean', 'C', clean),)
2386 2386
2387 2387 end = opts['print0'] and '\0' or '\n'
2388 2388
2389 2389 for opt, char, changes in ([ct for ct in explicit_changetypes
2390 2390 if all or opts[ct[0]]]
2391 2391 or changetypes):
2392 2392 if opts['no_status']:
2393 2393 format = "%%s%s" % end
2394 2394 else:
2395 2395 format = "%s %%s%s" % (char, end)
2396 2396
2397 2397 for f in changes:
2398 2398 ui.write(format % f)
2399 2399 if ((all or opts.get('copies')) and not opts.get('no_status')):
2400 2400 copied = repo.dirstate.copied(f)
2401 2401 if copied:
2402 2402 ui.write(' %s%s' % (copied, end))
2403 2403
2404 2404 def tag(ui, repo, name, rev_=None, **opts):
2405 2405 """add a tag for the current tip or a given revision
2406 2406
2407 2407 Name a particular revision using <name>.
2408 2408
2409 2409 Tags are used to name particular revisions of the repository and are
2410 2410 very useful to compare different revision, to go back to significant
2411 2411 earlier versions or to mark branch points as releases, etc.
2412 2412
2413 2413 If no revision is given, the parent of the working directory is used.
2414 2414
2415 2415 To facilitate version control, distribution, and merging of tags,
2416 2416 they are stored as a file named ".hgtags" which is managed
2417 2417 similarly to other project files and can be hand-edited if
2418 2418 necessary. The file '.hg/localtags' is used for local tags (not
2419 2419 shared among repositories).
2420 2420 """
2421 2421 if name in ['tip', '.', 'null']:
2422 2422 raise util.Abort(_("the name '%s' is reserved") % name)
2423 2423 if rev_ is not None:
2424 2424 ui.warn(_("use of 'hg tag NAME [REV]' is deprecated, "
2425 2425 "please use 'hg tag [-r REV] NAME' instead\n"))
2426 2426 if opts['rev']:
2427 2427 raise util.Abort(_("use only one form to specify the revision"))
2428 2428 if opts['rev']:
2429 2429 rev_ = opts['rev']
2430 2430 if not rev_ and repo.dirstate.parents()[1] != nullid:
2431 2431 raise util.Abort(_('uncommitted merge - please provide a '
2432 2432 'specific revision'))
2433 2433 r = repo.changectx(rev_).node()
2434 2434
2435 2435 message = opts['message']
2436 2436 if not message:
2437 2437 message = _('Added tag %s for changeset %s') % (name, short(r))
2438 2438
2439 2439 repo.tag(name, r, message, opts['local'], opts['user'], opts['date'])
2440 2440
2441 2441 def tags(ui, repo):
2442 2442 """list repository tags
2443 2443
2444 2444 List the repository tags.
2445 2445
2446 2446 This lists both regular and local tags.
2447 2447 """
2448 2448
2449 2449 l = repo.tagslist()
2450 2450 l.reverse()
2451 2451 hexfunc = ui.debugflag and hex or short
2452 2452 for t, n in l:
2453 2453 try:
2454 2454 r = "%5d:%s" % (repo.changelog.rev(n), hexfunc(n))
2455 2455 except KeyError:
2456 2456 r = " ?:?"
2457 2457 if ui.quiet:
2458 2458 ui.write("%s\n" % t)
2459 2459 else:
2460 2460 t = util.localsub(t, 30)
2461 2461 t += " " * (30 - util.locallen(t))
2462 2462 ui.write("%s %s\n" % (t, r))
2463 2463
2464 2464 def tip(ui, repo, **opts):
2465 2465 """show the tip revision
2466 2466
2467 2467 Show the tip revision.
2468 2468 """
2469 2469 cmdutil.show_changeset(ui, repo, opts).show(nullrev+repo.changelog.count())
2470 2470
2471 2471 def unbundle(ui, repo, fname, **opts):
2472 2472 """apply a changegroup file
2473 2473
2474 2474 Apply a compressed changegroup file generated by the bundle
2475 2475 command.
2476 2476 """
2477 gen = changegroup.readbundle(urllib.urlopen(fname))
2477 gen = changegroup.readbundle(urllib.urlopen(fname), fname)
2478 2478 modheads = repo.addchangegroup(gen, 'unbundle', 'bundle:' + fname)
2479 2479 return postincoming(ui, repo, modheads, opts['update'])
2480 2480
2481 2481 def update(ui, repo, node=None, clean=False, branch=None, date=None):
2482 2482 """update or merge working directory
2483 2483
2484 2484 Update the working directory to the specified revision.
2485 2485
2486 2486 If there are no outstanding changes in the working directory and
2487 2487 there is a linear relationship between the current version and the
2488 2488 requested version, the result is the requested version.
2489 2489
2490 2490 To merge the working directory with another revision, use the
2491 2491 merge command.
2492 2492
2493 2493 By default, update will refuse to run if doing so would require
2494 2494 merging or discarding local changes.
2495 2495 """
2496 2496 if date:
2497 2497 if node:
2498 2498 raise util.Abort(_("you can't specify a revision and a date"))
2499 2499 node = cmdutil.finddate(ui, repo, date)
2500 2500
2501 2501 node = _lookup(repo, node, branch)
2502 2502 if clean:
2503 2503 return hg.clean(repo, node)
2504 2504 else:
2505 2505 return hg.update(repo, node)
2506 2506
2507 2507 def _lookup(repo, node, branch=None):
2508 2508 if branch:
2509 2509 repo.ui.warn(_("the --branch option is deprecated, "
2510 2510 "please use 'hg branch' instead\n"))
2511 2511 br = repo.branchlookup(branch=branch)
2512 2512 found = []
2513 2513 for x in br:
2514 2514 if branch in br[x]:
2515 2515 found.append(x)
2516 2516 if len(found) > 1:
2517 2517 repo.ui.warn(_("Found multiple heads for %s\n") % branch)
2518 2518 for x in found:
2519 2519 cmdutil.show_changeset(ui, repo, {}).show(changenode=x)
2520 2520 raise util.Abort("")
2521 2521 if len(found) == 1:
2522 2522 node = found[0]
2523 2523 repo.ui.warn(_("Using head %s for branch %s\n")
2524 2524 % (short(node), branch))
2525 2525 else:
2526 2526 raise util.Abort(_("branch %s not found") % branch)
2527 2527 else:
2528 2528 node = node and repo.lookup(node) or repo.changelog.tip()
2529 2529 return node
2530 2530
2531 2531 def verify(ui, repo):
2532 2532 """verify the integrity of the repository
2533 2533
2534 2534 Verify the integrity of the current repository.
2535 2535
2536 2536 This will perform an extensive check of the repository's
2537 2537 integrity, validating the hashes and checksums of each entry in
2538 2538 the changelog, manifest, and tracked files, as well as the
2539 2539 integrity of their crosslinks and indices.
2540 2540 """
2541 2541 return hg.verify(repo)
2542 2542
2543 2543 def version_(ui):
2544 2544 """output version and copyright information"""
2545 2545 ui.write(_("Mercurial Distributed SCM (version %s)\n")
2546 2546 % version.get_version())
2547 2547 ui.status(_(
2548 2548 "\nCopyright (C) 2005, 2006 Matt Mackall <mpm@selenic.com>\n"
2549 2549 "This is free software; see the source for copying conditions. "
2550 2550 "There is NO\nwarranty; "
2551 2551 "not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n"
2552 2552 ))
2553 2553
2554 2554 # Command options and aliases are listed here, alphabetically
2555 2555
2556 2556 globalopts = [
2557 2557 ('R', 'repository', '',
2558 2558 _('repository root directory or symbolic path name')),
2559 2559 ('', 'cwd', '', _('change working directory')),
2560 2560 ('y', 'noninteractive', None,
2561 2561 _('do not prompt, assume \'yes\' for any required answers')),
2562 2562 ('q', 'quiet', None, _('suppress output')),
2563 2563 ('v', 'verbose', None, _('enable additional output')),
2564 2564 ('', 'config', [], _('set/override config option')),
2565 2565 ('', 'debug', None, _('enable debugging output')),
2566 2566 ('', 'debugger', None, _('start debugger')),
2567 2567 ('', 'encoding', util._encoding, _('set the charset encoding')),
2568 2568 ('', 'encodingmode', util._encodingmode, _('set the charset encoding mode')),
2569 2569 ('', 'lsprof', None, _('print improved command execution profile')),
2570 2570 ('', 'traceback', None, _('print traceback on exception')),
2571 2571 ('', 'time', None, _('time how long the command takes')),
2572 2572 ('', 'profile', None, _('print command execution profile')),
2573 2573 ('', 'version', None, _('output version information and exit')),
2574 2574 ('h', 'help', None, _('display help and exit')),
2575 2575 ]
2576 2576
2577 2577 dryrunopts = [('n', 'dry-run', None,
2578 2578 _('do not perform actions, just print output'))]
2579 2579
2580 2580 remoteopts = [
2581 2581 ('e', 'ssh', '', _('specify ssh command to use')),
2582 2582 ('', 'remotecmd', '', _('specify hg command to run on the remote side')),
2583 2583 ]
2584 2584
2585 2585 walkopts = [
2586 2586 ('I', 'include', [], _('include names matching the given patterns')),
2587 2587 ('X', 'exclude', [], _('exclude names matching the given patterns')),
2588 2588 ]
2589 2589
2590 2590 commitopts = [
2591 2591 ('m', 'message', '', _('use <text> as commit message')),
2592 2592 ('l', 'logfile', '', _('read commit message from <file>')),
2593 2593 ]
2594 2594
2595 2595 table = {
2596 2596 "^add": (add, walkopts + dryrunopts, _('hg add [OPTION]... [FILE]...')),
2597 2597 "addremove":
2598 2598 (addremove,
2599 2599 [('s', 'similarity', '',
2600 2600 _('guess renamed files by similarity (0<=s<=100)')),
2601 2601 ] + walkopts + dryrunopts,
2602 2602 _('hg addremove [OPTION]... [FILE]...')),
2603 2603 "^annotate":
2604 2604 (annotate,
2605 2605 [('r', 'rev', '', _('annotate the specified revision')),
2606 2606 ('f', 'follow', None, _('follow file copies and renames')),
2607 2607 ('a', 'text', None, _('treat all files as text')),
2608 2608 ('u', 'user', None, _('list the author')),
2609 2609 ('d', 'date', None, _('list the date')),
2610 2610 ('n', 'number', None, _('list the revision number (default)')),
2611 2611 ('c', 'changeset', None, _('list the changeset')),
2612 2612 ] + walkopts,
2613 2613 _('hg annotate [-r REV] [-a] [-u] [-d] [-n] [-c] FILE...')),
2614 2614 "archive":
2615 2615 (archive,
2616 2616 [('', 'no-decode', None, _('do not pass files through decoders')),
2617 2617 ('p', 'prefix', '', _('directory prefix for files in archive')),
2618 2618 ('r', 'rev', '', _('revision to distribute')),
2619 2619 ('t', 'type', '', _('type of distribution to create')),
2620 2620 ] + walkopts,
2621 2621 _('hg archive [OPTION]... DEST')),
2622 2622 "backout":
2623 2623 (backout,
2624 2624 [('', 'merge', None,
2625 2625 _('merge with old dirstate parent after backout')),
2626 2626 ('d', 'date', '', _('record datecode as commit date')),
2627 2627 ('', 'parent', '', _('parent to choose when backing out merge')),
2628 2628 ('u', 'user', '', _('record user as committer')),
2629 2629 ] + walkopts + commitopts,
2630 2630 _('hg backout [OPTION]... REV')),
2631 2631 "branch": (branch, [], _('hg branch [NAME]')),
2632 2632 "branches": (branches, [], _('hg branches')),
2633 2633 "bundle":
2634 2634 (bundle,
2635 2635 [('f', 'force', None,
2636 2636 _('run even when remote repository is unrelated')),
2637 2637 ('r', 'rev', [],
2638 2638 _('a changeset you would like to bundle')),
2639 2639 ('', 'base', [],
2640 2640 _('a base changeset to specify instead of a destination')),
2641 2641 ] + remoteopts,
2642 2642 _('hg bundle [--base REV]... [--rev REV]... FILE [DEST]')),
2643 2643 "cat":
2644 2644 (cat,
2645 2645 [('o', 'output', '', _('print output to file with formatted name')),
2646 2646 ('r', 'rev', '', _('print the given revision')),
2647 2647 ] + walkopts,
2648 2648 _('hg cat [OPTION]... FILE...')),
2649 2649 "^clone":
2650 2650 (clone,
2651 2651 [('U', 'noupdate', None, _('do not update the new working directory')),
2652 2652 ('r', 'rev', [],
2653 2653 _('a changeset you would like to have after cloning')),
2654 2654 ('', 'pull', None, _('use pull protocol to copy metadata')),
2655 2655 ('', 'uncompressed', None,
2656 2656 _('use uncompressed transfer (fast over LAN)')),
2657 2657 ] + remoteopts,
2658 2658 _('hg clone [OPTION]... SOURCE [DEST]')),
2659 2659 "^commit|ci":
2660 2660 (commit,
2661 2661 [('A', 'addremove', None,
2662 2662 _('mark new/missing files as added/removed before committing')),
2663 2663 ('d', 'date', '', _('record datecode as commit date')),
2664 2664 ('u', 'user', '', _('record user as commiter')),
2665 2665 ] + walkopts + commitopts,
2666 2666 _('hg commit [OPTION]... [FILE]...')),
2667 2667 "copy|cp":
2668 2668 (copy,
2669 2669 [('A', 'after', None, _('record a copy that has already occurred')),
2670 2670 ('f', 'force', None,
2671 2671 _('forcibly copy over an existing managed file')),
2672 2672 ] + walkopts + dryrunopts,
2673 2673 _('hg copy [OPTION]... [SOURCE]... DEST')),
2674 2674 "debugancestor": (debugancestor, [], _('debugancestor INDEX REV1 REV2')),
2675 2675 "debugcomplete":
2676 2676 (debugcomplete,
2677 2677 [('o', 'options', None, _('show the command options'))],
2678 2678 _('debugcomplete [-o] CMD')),
2679 2679 "debuginstall": (debuginstall, [], _('debuginstall')),
2680 2680 "debugrebuildstate":
2681 2681 (debugrebuildstate,
2682 2682 [('r', 'rev', '', _('revision to rebuild to'))],
2683 2683 _('debugrebuildstate [-r REV] [REV]')),
2684 2684 "debugcheckstate": (debugcheckstate, [], _('debugcheckstate')),
2685 2685 "debugsetparents": (debugsetparents, [], _('debugsetparents REV1 [REV2]')),
2686 2686 "debugstate": (debugstate, [], _('debugstate')),
2687 2687 "debugdate":
2688 2688 (debugdate,
2689 2689 [('e', 'extended', None, _('try extended date formats'))],
2690 2690 _('debugdate [-e] DATE [RANGE]')),
2691 2691 "debugdata": (debugdata, [], _('debugdata FILE REV')),
2692 2692 "debugindex": (debugindex, [], _('debugindex FILE')),
2693 2693 "debugindexdot": (debugindexdot, [], _('debugindexdot FILE')),
2694 2694 "debugrename": (debugrename, [], _('debugrename FILE [REV]')),
2695 2695 "debugwalk": (debugwalk, walkopts, _('debugwalk [OPTION]... [FILE]...')),
2696 2696 "^diff":
2697 2697 (diff,
2698 2698 [('r', 'rev', [], _('revision')),
2699 2699 ('a', 'text', None, _('treat all files as text')),
2700 2700 ('p', 'show-function', None,
2701 2701 _('show which function each change is in')),
2702 2702 ('g', 'git', None, _('use git extended diff format')),
2703 2703 ('', 'nodates', None, _("don't include dates in diff headers")),
2704 2704 ('w', 'ignore-all-space', None,
2705 2705 _('ignore white space when comparing lines')),
2706 2706 ('b', 'ignore-space-change', None,
2707 2707 _('ignore changes in the amount of white space')),
2708 2708 ('B', 'ignore-blank-lines', None,
2709 2709 _('ignore changes whose lines are all blank')),
2710 2710 ] + walkopts,
2711 2711 _('hg diff [-a] [-I] [-X] [-r REV1 [-r REV2]] [FILE]...')),
2712 2712 "^export":
2713 2713 (export,
2714 2714 [('o', 'output', '', _('print output to file with formatted name')),
2715 2715 ('a', 'text', None, _('treat all files as text')),
2716 2716 ('g', 'git', None, _('use git extended diff format')),
2717 2717 ('', 'nodates', None, _("don't include dates in diff headers")),
2718 2718 ('', 'switch-parent', None, _('diff against the second parent'))],
2719 2719 _('hg export [-a] [-o OUTFILESPEC] REV...')),
2720 2720 "grep":
2721 2721 (grep,
2722 2722 [('0', 'print0', None, _('end fields with NUL')),
2723 2723 ('', 'all', None, _('print all revisions that match')),
2724 2724 ('f', 'follow', None,
2725 2725 _('follow changeset history, or file history across copies and renames')),
2726 2726 ('i', 'ignore-case', None, _('ignore case when matching')),
2727 2727 ('l', 'files-with-matches', None,
2728 2728 _('print only filenames and revs that match')),
2729 2729 ('n', 'line-number', None, _('print matching line numbers')),
2730 2730 ('r', 'rev', [], _('search in given revision range')),
2731 2731 ('u', 'user', None, _('print user who committed change')),
2732 2732 ] + walkopts,
2733 2733 _('hg grep [OPTION]... PATTERN [FILE]...')),
2734 2734 "heads":
2735 2735 (heads,
2736 2736 [('b', 'branches', None, _('show branches (DEPRECATED)')),
2737 2737 ('', 'style', '', _('display using template map file')),
2738 2738 ('r', 'rev', '', _('show only heads which are descendants of rev')),
2739 2739 ('', 'template', '', _('display with template'))],
2740 2740 _('hg heads [-r REV]')),
2741 2741 "help": (help_, [], _('hg help [COMMAND]')),
2742 2742 "identify|id": (identify, [], _('hg identify')),
2743 2743 "import|patch":
2744 2744 (import_,
2745 2745 [('p', 'strip', 1,
2746 2746 _('directory strip option for patch. This has the same\n'
2747 2747 'meaning as the corresponding patch option')),
2748 2748 ('b', 'base', '', _('base path (DEPRECATED)')),
2749 2749 ('f', 'force', None,
2750 2750 _('skip check for outstanding uncommitted changes'))] + commitopts,
2751 2751 _('hg import [-p NUM] [-m MESSAGE] [-f] PATCH...')),
2752 2752 "incoming|in": (incoming,
2753 2753 [('M', 'no-merges', None, _('do not show merges')),
2754 2754 ('f', 'force', None,
2755 2755 _('run even when remote repository is unrelated')),
2756 2756 ('', 'style', '', _('display using template map file')),
2757 2757 ('n', 'newest-first', None, _('show newest record first')),
2758 2758 ('', 'bundle', '', _('file to store the bundles into')),
2759 2759 ('p', 'patch', None, _('show patch')),
2760 2760 ('r', 'rev', [], _('a specific revision up to which you would like to pull')),
2761 2761 ('', 'template', '', _('display with template')),
2762 2762 ] + remoteopts,
2763 2763 _('hg incoming [-p] [-n] [-M] [-r REV]...'
2764 2764 ' [--bundle FILENAME] [SOURCE]')),
2765 2765 "^init":
2766 2766 (init, remoteopts, _('hg init [-e FILE] [--remotecmd FILE] [DEST]')),
2767 2767 "locate":
2768 2768 (locate,
2769 2769 [('r', 'rev', '', _('search the repository as it stood at rev')),
2770 2770 ('0', 'print0', None,
2771 2771 _('end filenames with NUL, for use with xargs')),
2772 2772 ('f', 'fullpath', None,
2773 2773 _('print complete paths from the filesystem root')),
2774 2774 ] + walkopts,
2775 2775 _('hg locate [OPTION]... [PATTERN]...')),
2776 2776 "^log|history":
2777 2777 (log,
2778 2778 [('b', 'branches', None, _('show branches (DEPRECATED)')),
2779 2779 ('f', 'follow', None,
2780 2780 _('follow changeset history, or file history across copies and renames')),
2781 2781 ('', 'follow-first', None,
2782 2782 _('only follow the first parent of merge changesets')),
2783 2783 ('d', 'date', '', _('show revs matching date spec')),
2784 2784 ('C', 'copies', None, _('show copied files')),
2785 2785 ('k', 'keyword', [], _('search for a keyword')),
2786 2786 ('l', 'limit', '', _('limit number of changes displayed')),
2787 2787 ('r', 'rev', [], _('show the specified revision or range')),
2788 2788 ('', 'removed', None, _('include revs where files were removed')),
2789 2789 ('M', 'no-merges', None, _('do not show merges')),
2790 2790 ('', 'style', '', _('display using template map file')),
2791 2791 ('m', 'only-merges', None, _('show only merges')),
2792 2792 ('p', 'patch', None, _('show patch')),
2793 2793 ('P', 'prune', [], _('do not display revision or any of its ancestors')),
2794 2794 ('', 'template', '', _('display with template')),
2795 2795 ] + walkopts,
2796 2796 _('hg log [OPTION]... [FILE]')),
2797 2797 "manifest": (manifest, [], _('hg manifest [REV]')),
2798 2798 "merge":
2799 2799 (merge,
2800 2800 [('b', 'branch', '', _('merge with head of a specific branch (DEPRECATED)')),
2801 2801 ('f', 'force', None, _('force a merge with outstanding changes'))],
2802 2802 _('hg merge [-f] [REV]')),
2803 2803 "outgoing|out": (outgoing,
2804 2804 [('M', 'no-merges', None, _('do not show merges')),
2805 2805 ('f', 'force', None,
2806 2806 _('run even when remote repository is unrelated')),
2807 2807 ('p', 'patch', None, _('show patch')),
2808 2808 ('', 'style', '', _('display using template map file')),
2809 2809 ('r', 'rev', [], _('a specific revision you would like to push')),
2810 2810 ('n', 'newest-first', None, _('show newest record first')),
2811 2811 ('', 'template', '', _('display with template')),
2812 2812 ] + remoteopts,
2813 2813 _('hg outgoing [-M] [-p] [-n] [-r REV]... [DEST]')),
2814 2814 "^parents":
2815 2815 (parents,
2816 2816 [('b', 'branches', None, _('show branches (DEPRECATED)')),
2817 2817 ('r', 'rev', '', _('show parents from the specified rev')),
2818 2818 ('', 'style', '', _('display using template map file')),
2819 2819 ('', 'template', '', _('display with template'))],
2820 2820 _('hg parents [-r REV] [FILE]')),
2821 2821 "paths": (paths, [], _('hg paths [NAME]')),
2822 2822 "^pull":
2823 2823 (pull,
2824 2824 [('u', 'update', None,
2825 2825 _('update to new tip if changesets were pulled')),
2826 2826 ('f', 'force', None,
2827 2827 _('run even when remote repository is unrelated')),
2828 2828 ('r', 'rev', [], _('a specific revision up to which you would like to pull')),
2829 2829 ] + remoteopts,
2830 2830 _('hg pull [-u] [-r REV]... [-e FILE] [--remotecmd FILE] [SOURCE]')),
2831 2831 "^push":
2832 2832 (push,
2833 2833 [('f', 'force', None, _('force push')),
2834 2834 ('r', 'rev', [], _('a specific revision you would like to push')),
2835 2835 ] + remoteopts,
2836 2836 _('hg push [-f] [-r REV]... [-e FILE] [--remotecmd FILE] [DEST]')),
2837 2837 "debugrawcommit|rawcommit":
2838 2838 (rawcommit,
2839 2839 [('p', 'parent', [], _('parent')),
2840 2840 ('d', 'date', '', _('date code')),
2841 2841 ('u', 'user', '', _('user')),
2842 2842 ('F', 'files', '', _('file list'))
2843 2843 ] + commitopts,
2844 2844 _('hg debugrawcommit [OPTION]... [FILE]...')),
2845 2845 "recover": (recover, [], _('hg recover')),
2846 2846 "^remove|rm":
2847 2847 (remove,
2848 2848 [('A', 'after', None, _('record remove that has already occurred')),
2849 2849 ('f', 'force', None, _('remove file even if modified')),
2850 2850 ] + walkopts,
2851 2851 _('hg remove [OPTION]... FILE...')),
2852 2852 "rename|mv":
2853 2853 (rename,
2854 2854 [('A', 'after', None, _('record a rename that has already occurred')),
2855 2855 ('f', 'force', None,
2856 2856 _('forcibly copy over an existing managed file')),
2857 2857 ] + walkopts + dryrunopts,
2858 2858 _('hg rename [OPTION]... SOURCE... DEST')),
2859 2859 "^revert":
2860 2860 (revert,
2861 2861 [('a', 'all', None, _('revert all changes when no arguments given')),
2862 2862 ('d', 'date', '', _('tipmost revision matching date')),
2863 2863 ('r', 'rev', '', _('revision to revert to')),
2864 2864 ('', 'no-backup', None, _('do not save backup copies of files')),
2865 2865 ] + walkopts + dryrunopts,
2866 2866 _('hg revert [-r REV] [NAME]...')),
2867 2867 "rollback": (rollback, [], _('hg rollback')),
2868 2868 "root": (root, [], _('hg root')),
2869 2869 "showconfig|debugconfig":
2870 2870 (showconfig,
2871 2871 [('u', 'untrusted', None, _('show untrusted configuration options'))],
2872 2872 _('showconfig [-u] [NAME]...')),
2873 2873 "^serve":
2874 2874 (serve,
2875 2875 [('A', 'accesslog', '', _('name of access log file to write to')),
2876 2876 ('d', 'daemon', None, _('run server in background')),
2877 2877 ('', 'daemon-pipefds', '', _('used internally by daemon mode')),
2878 2878 ('E', 'errorlog', '', _('name of error log file to write to')),
2879 2879 ('p', 'port', 0, _('port to use (default: 8000)')),
2880 2880 ('a', 'address', '', _('address to use')),
2881 2881 ('n', 'name', '',
2882 2882 _('name to show in web pages (default: working dir)')),
2883 2883 ('', 'webdir-conf', '', _('name of the webdir config file'
2884 2884 ' (serve more than one repo)')),
2885 2885 ('', 'pid-file', '', _('name of file to write process ID to')),
2886 2886 ('', 'stdio', None, _('for remote clients')),
2887 2887 ('t', 'templates', '', _('web templates to use')),
2888 2888 ('', 'style', '', _('template style to use')),
2889 2889 ('6', 'ipv6', None, _('use IPv6 in addition to IPv4'))],
2890 2890 _('hg serve [OPTION]...')),
2891 2891 "^status|st":
2892 2892 (status,
2893 2893 [('A', 'all', None, _('show status of all files')),
2894 2894 ('m', 'modified', None, _('show only modified files')),
2895 2895 ('a', 'added', None, _('show only added files')),
2896 2896 ('r', 'removed', None, _('show only removed files')),
2897 2897 ('d', 'deleted', None, _('show only deleted (but tracked) files')),
2898 2898 ('c', 'clean', None, _('show only files without changes')),
2899 2899 ('u', 'unknown', None, _('show only unknown (not tracked) files')),
2900 2900 ('i', 'ignored', None, _('show ignored files')),
2901 2901 ('n', 'no-status', None, _('hide status prefix')),
2902 2902 ('C', 'copies', None, _('show source of copied files')),
2903 2903 ('0', 'print0', None,
2904 2904 _('end filenames with NUL, for use with xargs')),
2905 2905 ('', 'rev', [], _('show difference from revision')),
2906 2906 ] + walkopts,
2907 2907 _('hg status [OPTION]... [FILE]...')),
2908 2908 "tag":
2909 2909 (tag,
2910 2910 [('l', 'local', None, _('make the tag local')),
2911 2911 ('m', 'message', '', _('message for tag commit log entry')),
2912 2912 ('d', 'date', '', _('record datecode as commit date')),
2913 2913 ('u', 'user', '', _('record user as commiter')),
2914 2914 ('r', 'rev', '', _('revision to tag'))],
2915 2915 _('hg tag [-l] [-m TEXT] [-d DATE] [-u USER] [-r REV] NAME')),
2916 2916 "tags": (tags, [], _('hg tags')),
2917 2917 "tip":
2918 2918 (tip,
2919 2919 [('b', 'branches', None, _('show branches (DEPRECATED)')),
2920 2920 ('', 'style', '', _('display using template map file')),
2921 2921 ('p', 'patch', None, _('show patch')),
2922 2922 ('', 'template', '', _('display with template'))],
2923 2923 _('hg tip [-p]')),
2924 2924 "unbundle":
2925 2925 (unbundle,
2926 2926 [('u', 'update', None,
2927 2927 _('update to new tip if changesets were unbundled'))],
2928 2928 _('hg unbundle [-u] FILE')),
2929 2929 "^update|up|checkout|co":
2930 2930 (update,
2931 2931 [('b', 'branch', '',
2932 2932 _('checkout the head of a specific branch (DEPRECATED)')),
2933 2933 ('C', 'clean', None, _('overwrite locally modified files')),
2934 2934 ('d', 'date', '', _('tipmost revision matching date'))],
2935 2935 _('hg update [-C] [REV]')),
2936 2936 "verify": (verify, [], _('hg verify')),
2937 2937 "version": (version_, [], _('hg version')),
2938 2938 }
2939 2939
2940 2940 norepo = ("clone init version help debugancestor debugcomplete debugdata"
2941 2941 " debugindex debugindexdot debugdate debuginstall")
2942 2942 optionalrepo = ("paths serve showconfig")
2943 2943
2944 2944 def findpossible(ui, cmd):
2945 2945 """
2946 2946 Return cmd -> (aliases, command table entry)
2947 2947 for each matching command.
2948 2948 Return debug commands (or their aliases) only if no normal command matches.
2949 2949 """
2950 2950 choice = {}
2951 2951 debugchoice = {}
2952 2952 for e in table.keys():
2953 2953 aliases = e.lstrip("^").split("|")
2954 2954 found = None
2955 2955 if cmd in aliases:
2956 2956 found = cmd
2957 2957 elif not ui.config("ui", "strict"):
2958 2958 for a in aliases:
2959 2959 if a.startswith(cmd):
2960 2960 found = a
2961 2961 break
2962 2962 if found is not None:
2963 2963 if aliases[0].startswith("debug") or found.startswith("debug"):
2964 2964 debugchoice[found] = (aliases, table[e])
2965 2965 else:
2966 2966 choice[found] = (aliases, table[e])
2967 2967
2968 2968 if not choice and debugchoice:
2969 2969 choice = debugchoice
2970 2970
2971 2971 return choice
2972 2972
2973 2973 def findcmd(ui, cmd):
2974 2974 """Return (aliases, command table entry) for command string."""
2975 2975 choice = findpossible(ui, cmd)
2976 2976
2977 2977 if choice.has_key(cmd):
2978 2978 return choice[cmd]
2979 2979
2980 2980 if len(choice) > 1:
2981 2981 clist = choice.keys()
2982 2982 clist.sort()
2983 2983 raise AmbiguousCommand(cmd, clist)
2984 2984
2985 2985 if choice:
2986 2986 return choice.values()[0]
2987 2987
2988 2988 raise UnknownCommand(cmd)
2989 2989
2990 2990 def catchterm(*args):
2991 2991 raise util.SignalInterrupt
2992 2992
2993 2993 def run():
2994 2994 sys.exit(dispatch(sys.argv[1:]))
2995 2995
2996 2996 class ParseError(Exception):
2997 2997 """Exception raised on errors in parsing the command line."""
2998 2998
2999 2999 def parse(ui, args):
3000 3000 options = {}
3001 3001 cmdoptions = {}
3002 3002
3003 3003 try:
3004 3004 args = fancyopts.fancyopts(args, globalopts, options)
3005 3005 except fancyopts.getopt.GetoptError, inst:
3006 3006 raise ParseError(None, inst)
3007 3007
3008 3008 if args:
3009 3009 cmd, args = args[0], args[1:]
3010 3010 aliases, i = findcmd(ui, cmd)
3011 3011 cmd = aliases[0]
3012 3012 defaults = ui.config("defaults", cmd)
3013 3013 if defaults:
3014 3014 args = shlex.split(defaults) + args
3015 3015 c = list(i[1])
3016 3016 else:
3017 3017 cmd = None
3018 3018 c = []
3019 3019
3020 3020 # combine global options into local
3021 3021 for o in globalopts:
3022 3022 c.append((o[0], o[1], options[o[1]], o[3]))
3023 3023
3024 3024 try:
3025 3025 args = fancyopts.fancyopts(args, c, cmdoptions)
3026 3026 except fancyopts.getopt.GetoptError, inst:
3027 3027 raise ParseError(cmd, inst)
3028 3028
3029 3029 # separate global options back out
3030 3030 for o in globalopts:
3031 3031 n = o[1]
3032 3032 options[n] = cmdoptions[n]
3033 3033 del cmdoptions[n]
3034 3034
3035 3035 return (cmd, cmd and i[0] or None, args, options, cmdoptions)
3036 3036
3037 3037 external = {}
3038 3038
3039 3039 def findext(name):
3040 3040 '''return module with given extension name'''
3041 3041 try:
3042 3042 return sys.modules[external[name]]
3043 3043 except KeyError:
3044 3044 for k, v in external.iteritems():
3045 3045 if k.endswith('.' + name) or k.endswith('/' + name) or v == name:
3046 3046 return sys.modules[v]
3047 3047 raise KeyError(name)
3048 3048
3049 3049 def load_extensions(ui):
3050 3050 added = []
3051 3051 for ext_name, load_from_name in ui.extensions():
3052 3052 if ext_name in external:
3053 3053 continue
3054 3054 try:
3055 3055 if load_from_name:
3056 3056 # the module will be loaded in sys.modules
3057 3057 # choose an unique name so that it doesn't
3058 3058 # conflicts with other modules
3059 3059 module_name = "hgext_%s" % ext_name.replace('.', '_')
3060 3060 mod = imp.load_source(module_name, load_from_name)
3061 3061 else:
3062 3062 def importh(name):
3063 3063 mod = __import__(name)
3064 3064 components = name.split('.')
3065 3065 for comp in components[1:]:
3066 3066 mod = getattr(mod, comp)
3067 3067 return mod
3068 3068 try:
3069 3069 mod = importh("hgext.%s" % ext_name)
3070 3070 except ImportError:
3071 3071 mod = importh(ext_name)
3072 3072 external[ext_name] = mod.__name__
3073 3073 added.append((mod, ext_name))
3074 3074 except (util.SignalInterrupt, KeyboardInterrupt):
3075 3075 raise
3076 3076 except Exception, inst:
3077 3077 ui.warn(_("*** failed to import extension %s: %s\n") %
3078 3078 (ext_name, inst))
3079 3079 if ui.print_exc():
3080 3080 return 1
3081 3081
3082 3082 for mod, name in added:
3083 3083 uisetup = getattr(mod, 'uisetup', None)
3084 3084 if uisetup:
3085 3085 uisetup(ui)
3086 3086 cmdtable = getattr(mod, 'cmdtable', {})
3087 3087 for t in cmdtable:
3088 3088 if t in table:
3089 3089 ui.warn(_("module %s overrides %s\n") % (name, t))
3090 3090 table.update(cmdtable)
3091 3091
3092 3092 def parseconfig(config):
3093 3093 """parse the --config options from the command line"""
3094 3094 parsed = []
3095 3095 for cfg in config:
3096 3096 try:
3097 3097 name, value = cfg.split('=', 1)
3098 3098 section, name = name.split('.', 1)
3099 3099 if not section or not name:
3100 3100 raise IndexError
3101 3101 parsed.append((section, name, value))
3102 3102 except (IndexError, ValueError):
3103 3103 raise util.Abort(_('malformed --config option: %s') % cfg)
3104 3104 return parsed
3105 3105
3106 3106 def dispatch(args):
3107 3107 for name in 'SIGBREAK', 'SIGHUP', 'SIGTERM':
3108 3108 num = getattr(signal, name, None)
3109 3109 if num: signal.signal(num, catchterm)
3110 3110
3111 3111 try:
3112 3112 u = ui.ui(traceback='--traceback' in sys.argv[1:])
3113 3113 except util.Abort, inst:
3114 3114 sys.stderr.write(_("abort: %s\n") % inst)
3115 3115 return -1
3116 3116
3117 3117 load_extensions(u)
3118 3118 u.addreadhook(load_extensions)
3119 3119
3120 3120 try:
3121 3121 cmd, func, args, options, cmdoptions = parse(u, args)
3122 3122 if options["encoding"]:
3123 3123 util._encoding = options["encoding"]
3124 3124 if options["encodingmode"]:
3125 3125 util._encodingmode = options["encodingmode"]
3126 3126 if options["time"]:
3127 3127 def get_times():
3128 3128 t = os.times()
3129 3129 if t[4] == 0.0: # Windows leaves this as zero, so use time.clock()
3130 3130 t = (t[0], t[1], t[2], t[3], time.clock())
3131 3131 return t
3132 3132 s = get_times()
3133 3133 def print_time():
3134 3134 t = get_times()
3135 3135 u.warn(_("Time: real %.3f secs (user %.3f+%.3f sys %.3f+%.3f)\n") %
3136 3136 (t[4]-s[4], t[0]-s[0], t[2]-s[2], t[1]-s[1], t[3]-s[3]))
3137 3137 atexit.register(print_time)
3138 3138
3139 3139 # enter the debugger before command execution
3140 3140 if options['debugger']:
3141 3141 pdb.set_trace()
3142 3142
3143 3143 try:
3144 3144 if options['cwd']:
3145 3145 os.chdir(options['cwd'])
3146 3146
3147 3147 u.updateopts(options["verbose"], options["debug"], options["quiet"],
3148 3148 not options["noninteractive"], options["traceback"],
3149 3149 parseconfig(options["config"]))
3150 3150
3151 3151 path = u.expandpath(options["repository"]) or ""
3152 3152 repo = path and hg.repository(u, path=path) or None
3153 3153 if repo and not repo.local():
3154 3154 raise util.Abort(_("repository '%s' is not local") % path)
3155 3155
3156 3156 if options['help']:
3157 3157 return help_(u, cmd, options['version'])
3158 3158 elif options['version']:
3159 3159 return version_(u)
3160 3160 elif not cmd:
3161 3161 return help_(u, 'shortlist')
3162 3162
3163 3163 if cmd not in norepo.split():
3164 3164 try:
3165 3165 if not repo:
3166 3166 repo = hg.repository(u, path=path)
3167 3167 u = repo.ui
3168 3168 for name in external.itervalues():
3169 3169 mod = sys.modules[name]
3170 3170 if hasattr(mod, 'reposetup'):
3171 3171 mod.reposetup(u, repo)
3172 3172 hg.repo_setup_hooks.append(mod.reposetup)
3173 3173 except hg.RepoError:
3174 3174 if cmd not in optionalrepo.split():
3175 3175 raise
3176 3176 d = lambda: func(u, repo, *args, **cmdoptions)
3177 3177 else:
3178 3178 d = lambda: func(u, *args, **cmdoptions)
3179 3179
3180 3180 try:
3181 3181 if options['profile']:
3182 3182 import hotshot, hotshot.stats
3183 3183 prof = hotshot.Profile("hg.prof")
3184 3184 try:
3185 3185 try:
3186 3186 return prof.runcall(d)
3187 3187 except:
3188 3188 try:
3189 3189 u.warn(_('exception raised - generating '
3190 3190 'profile anyway\n'))
3191 3191 except:
3192 3192 pass
3193 3193 raise
3194 3194 finally:
3195 3195 prof.close()
3196 3196 stats = hotshot.stats.load("hg.prof")
3197 3197 stats.strip_dirs()
3198 3198 stats.sort_stats('time', 'calls')
3199 3199 stats.print_stats(40)
3200 3200 elif options['lsprof']:
3201 3201 try:
3202 3202 from mercurial import lsprof
3203 3203 except ImportError:
3204 3204 raise util.Abort(_(
3205 3205 'lsprof not available - install from '
3206 3206 'http://codespeak.net/svn/user/arigo/hack/misc/lsprof/'))
3207 3207 p = lsprof.Profiler()
3208 3208 p.enable(subcalls=True)
3209 3209 try:
3210 3210 return d()
3211 3211 finally:
3212 3212 p.disable()
3213 3213 stats = lsprof.Stats(p.getstats())
3214 3214 stats.sort()
3215 3215 stats.pprint(top=10, file=sys.stderr, climit=5)
3216 3216 else:
3217 3217 return d()
3218 3218 finally:
3219 3219 u.flush()
3220 3220 except:
3221 3221 # enter the debugger when we hit an exception
3222 3222 if options['debugger']:
3223 3223 pdb.post_mortem(sys.exc_info()[2])
3224 3224 u.print_exc()
3225 3225 raise
3226 3226 except ParseError, inst:
3227 3227 if inst.args[0]:
3228 3228 u.warn(_("hg %s: %s\n") % (inst.args[0], inst.args[1]))
3229 3229 help_(u, inst.args[0])
3230 3230 else:
3231 3231 u.warn(_("hg: %s\n") % inst.args[1])
3232 3232 help_(u, 'shortlist')
3233 3233 except AmbiguousCommand, inst:
3234 3234 u.warn(_("hg: command '%s' is ambiguous:\n %s\n") %
3235 3235 (inst.args[0], " ".join(inst.args[1])))
3236 3236 except UnknownCommand, inst:
3237 3237 u.warn(_("hg: unknown command '%s'\n") % inst.args[0])
3238 3238 help_(u, 'shortlist')
3239 3239 except hg.RepoError, inst:
3240 3240 u.warn(_("abort: %s!\n") % inst)
3241 3241 except lock.LockHeld, inst:
3242 3242 if inst.errno == errno.ETIMEDOUT:
3243 3243 reason = _('timed out waiting for lock held by %s') % inst.locker
3244 3244 else:
3245 3245 reason = _('lock held by %s') % inst.locker
3246 3246 u.warn(_("abort: %s: %s\n") % (inst.desc or inst.filename, reason))
3247 3247 except lock.LockUnavailable, inst:
3248 3248 u.warn(_("abort: could not lock %s: %s\n") %
3249 3249 (inst.desc or inst.filename, inst.strerror))
3250 3250 except revlog.RevlogError, inst:
3251 3251 u.warn(_("abort: %s!\n") % inst)
3252 3252 except util.SignalInterrupt:
3253 3253 u.warn(_("killed!\n"))
3254 3254 except KeyboardInterrupt:
3255 3255 try:
3256 3256 u.warn(_("interrupted!\n"))
3257 3257 except IOError, inst:
3258 3258 if inst.errno == errno.EPIPE:
3259 3259 if u.debugflag:
3260 3260 u.warn(_("\nbroken pipe\n"))
3261 3261 else:
3262 3262 raise
3263 3263 except IOError, inst:
3264 3264 if hasattr(inst, "code"):
3265 3265 u.warn(_("abort: %s\n") % inst)
3266 3266 elif hasattr(inst, "reason"):
3267 3267 u.warn(_("abort: error: %s\n") % inst.reason[1])
3268 3268 elif hasattr(inst, "args") and inst[0] == errno.EPIPE:
3269 3269 if u.debugflag:
3270 3270 u.warn(_("broken pipe\n"))
3271 3271 elif getattr(inst, "strerror", None):
3272 3272 if getattr(inst, "filename", None):
3273 3273 u.warn(_("abort: %s: %s\n") % (inst.strerror, inst.filename))
3274 3274 else:
3275 3275 u.warn(_("abort: %s\n") % inst.strerror)
3276 3276 else:
3277 3277 raise
3278 3278 except OSError, inst:
3279 3279 if getattr(inst, "filename", None):
3280 3280 u.warn(_("abort: %s: %s\n") % (inst.strerror, inst.filename))
3281 3281 else:
3282 3282 u.warn(_("abort: %s\n") % inst.strerror)
3283 3283 except util.UnexpectedOutput, inst:
3284 3284 u.warn(_("abort: %s") % inst[0])
3285 3285 if not isinstance(inst[1], basestring):
3286 3286 u.warn(" %r\n" % (inst[1],))
3287 3287 elif not inst[1]:
3288 3288 u.warn(_(" empty string\n"))
3289 3289 else:
3290 3290 u.warn("\n%r\n" % util.ellipsis(inst[1]))
3291 3291 except util.Abort, inst:
3292 3292 u.warn(_("abort: %s\n") % inst)
3293 3293 except TypeError, inst:
3294 3294 # was this an argument error?
3295 3295 tb = traceback.extract_tb(sys.exc_info()[2])
3296 3296 if len(tb) > 2: # no
3297 3297 raise
3298 3298 u.debug(inst, "\n")
3299 3299 u.warn(_("%s: invalid arguments\n") % cmd)
3300 3300 help_(u, cmd)
3301 3301 except SystemExit, inst:
3302 3302 # Commands shouldn't sys.exit directly, but give a return code.
3303 3303 # Just in case catch this and and pass exit code to caller.
3304 3304 return inst.code
3305 3305 except:
3306 3306 u.warn(_("** unknown exception encountered, details follow\n"))
3307 3307 u.warn(_("** report bug details to "
3308 3308 "http://www.selenic.com/mercurial/bts\n"))
3309 3309 u.warn(_("** or mercurial@selenic.com\n"))
3310 3310 u.warn(_("** Mercurial Distributed SCM (version %s)\n")
3311 3311 % version.get_version())
3312 3312 raise
3313 3313
3314 3314 return -1
@@ -1,86 +1,86 b''
1 1 # statichttprepo.py - simple http repository class for mercurial
2 2 #
3 3 # This provides read-only repo access to repositories exported via static http
4 4 #
5 5 # Copyright 2005, 2006 Matt Mackall <mpm@selenic.com>
6 6 #
7 7 # This software may be used and distributed according to the terms
8 8 # of the GNU General Public License, incorporated herein by reference.
9 9
10 10 from demandload import *
11 11 from i18n import gettext as _
12 12 demandload(globals(), "changelog filelog httprangereader")
13 demandload(globals(), "localrepo manifest os urllib urllib2 util")
13 demandload(globals(), "repo localrepo manifest os urllib urllib2 util")
14 14
15 15 class rangereader(httprangereader.httprangereader):
16 16 def read(self, size=None):
17 17 try:
18 18 return httprangereader.httprangereader.read(self, size)
19 19 except urllib2.HTTPError, inst:
20 20 raise IOError(None, inst)
21 21 except urllib2.URLError, inst:
22 22 raise IOError(None, inst.reason[1])
23 23
24 24 def opener(base):
25 25 """return a function that opens files over http"""
26 26 p = base
27 27 def o(path, mode="r"):
28 28 f = "/".join((p, urllib.quote(path)))
29 29 return rangereader(f)
30 30 return o
31 31
32 32 class statichttprepository(localrepo.localrepository):
33 33 def __init__(self, ui, path):
34 34 self._url = path
35 35 self.ui = ui
36 36 self.revlogversion = 0
37 37
38 38 self.path = (path + "/.hg")
39 39 self.opener = opener(self.path)
40 40 # find requirements
41 41 try:
42 42 requirements = self.opener("requires").read().splitlines()
43 43 except IOError:
44 44 requirements = []
45 45 # check them
46 46 for r in requirements:
47 47 if r not in self.supported:
48 48 raise repo.RepoError(_("requirement '%s' not supported") % r)
49 49
50 50 # setup store
51 51 if "store" in requirements:
52 52 self.encodefn = util.encodefilename
53 53 self.decodefn = util.decodefilename
54 54 self.spath = self.path + "/store"
55 55 else:
56 56 self.encodefn = lambda x: x
57 57 self.decodefn = lambda x: x
58 58 self.spath = self.path
59 59 self.sopener = util.encodedopener(opener(self.spath), self.encodefn)
60 60
61 61 self.manifest = manifest.manifest(self.sopener)
62 62 self.changelog = changelog.changelog(self.sopener)
63 63 self.tagscache = None
64 64 self.nodetagscache = None
65 65 self.encodepats = None
66 66 self.decodepats = None
67 67
68 68 def url(self):
69 69 return 'static-' + self._url
70 70
71 71 def dev(self):
72 72 return -1
73 73
74 74 def local(self):
75 75 return False
76 76
77 77 def instance(ui, path, create):
78 78 if create:
79 79 raise util.Abort(_('cannot create new static-http repository'))
80 80 if path.startswith('old-http:'):
81 81 ui.warn(_("old-http:// syntax is deprecated, "
82 82 "please use static-http:// instead\n"))
83 83 path = path[4:]
84 84 else:
85 85 path = path[7:]
86 86 return statichttprepository(ui, path)
@@ -1,1322 +1,1322 b''
1 1 """
2 2 util.py - Mercurial utility functions and platform specfic implementations
3 3
4 4 Copyright 2005 K. Thananchayan <thananck@yahoo.com>
5 5 Copyright 2005, 2006 Matt Mackall <mpm@selenic.com>
6 6 Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
7 7
8 8 This software may be used and distributed according to the terms
9 9 of the GNU General Public License, incorporated herein by reference.
10 10
11 11 This contains helper routines that are independent of the SCM core and hide
12 12 platform-specific details from the core.
13 13 """
14 14
15 15 from i18n import gettext as _
16 16 from demandload import *
17 17 demandload(globals(), "cStringIO errno getpass popen2 re shutil sys tempfile")
18 18 demandload(globals(), "os threading time calendar ConfigParser locale")
19 19
20 20 _encoding = os.environ.get("HGENCODING") or locale.getpreferredencoding() \
21 21 or "ascii"
22 22 _encodingmode = os.environ.get("HGENCODINGMODE", "strict")
23 23 _fallbackencoding = 'ISO-8859-1'
24 24
25 25 def tolocal(s):
26 26 """
27 27 Convert a string from internal UTF-8 to local encoding
28 28
29 29 All internal strings should be UTF-8 but some repos before the
30 30 implementation of locale support may contain latin1 or possibly
31 31 other character sets. We attempt to decode everything strictly
32 32 using UTF-8, then Latin-1, and failing that, we use UTF-8 and
33 33 replace unknown characters.
34 34 """
35 35 for e in ('UTF-8', _fallbackencoding):
36 36 try:
37 37 u = s.decode(e) # attempt strict decoding
38 38 return u.encode(_encoding, "replace")
39 39 except LookupError, k:
40 40 raise Abort(_("%s, please check your locale settings") % k)
41 41 except UnicodeDecodeError:
42 42 pass
43 43 u = s.decode("utf-8", "replace") # last ditch
44 44 return u.encode(_encoding, "replace")
45 45
46 46 def fromlocal(s):
47 47 """
48 48 Convert a string from the local character encoding to UTF-8
49 49
50 50 We attempt to decode strings using the encoding mode set by
51 51 HG_ENCODINGMODE, which defaults to 'strict'. In this mode, unknown
52 52 characters will cause an error message. Other modes include
53 53 'replace', which replaces unknown characters with a special
54 54 Unicode character, and 'ignore', which drops the character.
55 55 """
56 56 try:
57 57 return s.decode(_encoding, _encodingmode).encode("utf-8")
58 58 except UnicodeDecodeError, inst:
59 59 sub = s[max(0, inst.start-10):inst.start+10]
60 60 raise Abort("decoding near '%s': %s!" % (sub, inst))
61 61 except LookupError, k:
62 62 raise Abort(_("%s, please check your locale settings") % k)
63 63
64 64 def locallen(s):
65 65 """Find the length in characters of a local string"""
66 66 return len(s.decode(_encoding, "replace"))
67 67
68 68 def localsub(s, a, b=None):
69 69 try:
70 70 u = s.decode(_encoding, _encodingmode)
71 71 if b is not None:
72 72 u = u[a:b]
73 73 else:
74 74 u = u[:a]
75 75 return u.encode(_encoding, _encodingmode)
76 76 except UnicodeDecodeError, inst:
77 77 sub = s[max(0, inst.start-10), inst.start+10]
78 78 raise Abort(_("decoding near '%s': %s!\n") % (sub, inst))
79 79
80 80 # used by parsedate
81 81 defaultdateformats = (
82 82 '%Y-%m-%d %H:%M:%S',
83 83 '%Y-%m-%d %I:%M:%S%p',
84 84 '%Y-%m-%d %H:%M',
85 85 '%Y-%m-%d %I:%M%p',
86 86 '%Y-%m-%d',
87 87 '%m-%d',
88 88 '%m/%d',
89 89 '%m/%d/%y',
90 90 '%m/%d/%Y',
91 91 '%a %b %d %H:%M:%S %Y',
92 92 '%a %b %d %I:%M:%S%p %Y',
93 93 '%b %d %H:%M:%S %Y',
94 94 '%b %d %I:%M:%S%p %Y',
95 95 '%b %d %H:%M:%S',
96 96 '%b %d %I:%M:%S%p',
97 97 '%b %d %H:%M',
98 98 '%b %d %I:%M%p',
99 99 '%b %d %Y',
100 100 '%b %d',
101 101 '%H:%M:%S',
102 102 '%I:%M:%SP',
103 103 '%H:%M',
104 104 '%I:%M%p',
105 105 )
106 106
107 107 extendeddateformats = defaultdateformats + (
108 108 "%Y",
109 109 "%Y-%m",
110 110 "%b",
111 111 "%b %Y",
112 112 )
113 113
114 114 class SignalInterrupt(Exception):
115 115 """Exception raised on SIGTERM and SIGHUP."""
116 116
117 117 # like SafeConfigParser but with case-sensitive keys
118 118 class configparser(ConfigParser.SafeConfigParser):
119 119 def optionxform(self, optionstr):
120 120 return optionstr
121 121
122 122 def cachefunc(func):
123 123 '''cache the result of function calls'''
124 124 # XXX doesn't handle keywords args
125 125 cache = {}
126 126 if func.func_code.co_argcount == 1:
127 127 # we gain a small amount of time because
128 128 # we don't need to pack/unpack the list
129 129 def f(arg):
130 130 if arg not in cache:
131 131 cache[arg] = func(arg)
132 132 return cache[arg]
133 133 else:
134 134 def f(*args):
135 135 if args not in cache:
136 136 cache[args] = func(*args)
137 137 return cache[args]
138 138
139 139 return f
140 140
141 141 def pipefilter(s, cmd):
142 142 '''filter string S through command CMD, returning its output'''
143 143 (pout, pin) = popen2.popen2(cmd, -1, 'b')
144 144 def writer():
145 145 try:
146 146 pin.write(s)
147 147 pin.close()
148 148 except IOError, inst:
149 149 if inst.errno != errno.EPIPE:
150 150 raise
151 151
152 152 # we should use select instead on UNIX, but this will work on most
153 153 # systems, including Windows
154 154 w = threading.Thread(target=writer)
155 155 w.start()
156 156 f = pout.read()
157 157 pout.close()
158 158 w.join()
159 159 return f
160 160
161 161 def tempfilter(s, cmd):
162 162 '''filter string S through a pair of temporary files with CMD.
163 163 CMD is used as a template to create the real command to be run,
164 164 with the strings INFILE and OUTFILE replaced by the real names of
165 165 the temporary files generated.'''
166 166 inname, outname = None, None
167 167 try:
168 168 infd, inname = tempfile.mkstemp(prefix='hg-filter-in-')
169 169 fp = os.fdopen(infd, 'wb')
170 170 fp.write(s)
171 171 fp.close()
172 172 outfd, outname = tempfile.mkstemp(prefix='hg-filter-out-')
173 173 os.close(outfd)
174 174 cmd = cmd.replace('INFILE', inname)
175 175 cmd = cmd.replace('OUTFILE', outname)
176 176 code = os.system(cmd)
177 177 if code: raise Abort(_("command '%s' failed: %s") %
178 178 (cmd, explain_exit(code)))
179 179 return open(outname, 'rb').read()
180 180 finally:
181 181 try:
182 182 if inname: os.unlink(inname)
183 183 except: pass
184 184 try:
185 185 if outname: os.unlink(outname)
186 186 except: pass
187 187
188 188 filtertable = {
189 189 'tempfile:': tempfilter,
190 190 'pipe:': pipefilter,
191 191 }
192 192
193 193 def filter(s, cmd):
194 194 "filter a string through a command that transforms its input to its output"
195 195 for name, fn in filtertable.iteritems():
196 196 if cmd.startswith(name):
197 197 return fn(s, cmd[len(name):].lstrip())
198 198 return pipefilter(s, cmd)
199 199
200 200 def find_in_path(name, path, default=None):
201 201 '''find name in search path. path can be string (will be split
202 202 with os.pathsep), or iterable thing that returns strings. if name
203 203 found, return path to name. else return default.'''
204 204 if isinstance(path, str):
205 205 path = path.split(os.pathsep)
206 206 for p in path:
207 207 p_name = os.path.join(p, name)
208 208 if os.path.exists(p_name):
209 209 return p_name
210 210 return default
211 211
212 212 def binary(s):
213 213 """return true if a string is binary data using diff's heuristic"""
214 214 if s and '\0' in s[:4096]:
215 215 return True
216 216 return False
217 217
218 218 def unique(g):
219 219 """return the uniq elements of iterable g"""
220 220 seen = {}
221 221 l = []
222 222 for f in g:
223 223 if f not in seen:
224 224 seen[f] = 1
225 225 l.append(f)
226 226 return l
227 227
228 228 class Abort(Exception):
229 229 """Raised if a command needs to print an error and exit."""
230 230
231 231 class UnexpectedOutput(Abort):
232 232 """Raised to print an error with part of output and exit."""
233 233
234 234 def always(fn): return True
235 235 def never(fn): return False
236 236
237 237 def patkind(name, dflt_pat='glob'):
238 238 """Split a string into an optional pattern kind prefix and the
239 239 actual pattern."""
240 240 for prefix in 're', 'glob', 'path', 'relglob', 'relpath', 'relre':
241 241 if name.startswith(prefix + ':'): return name.split(':', 1)
242 242 return dflt_pat, name
243 243
244 244 def globre(pat, head='^', tail='$'):
245 245 "convert a glob pattern into a regexp"
246 246 i, n = 0, len(pat)
247 247 res = ''
248 248 group = False
249 249 def peek(): return i < n and pat[i]
250 250 while i < n:
251 251 c = pat[i]
252 252 i = i+1
253 253 if c == '*':
254 254 if peek() == '*':
255 255 i += 1
256 256 res += '.*'
257 257 else:
258 258 res += '[^/]*'
259 259 elif c == '?':
260 260 res += '.'
261 261 elif c == '[':
262 262 j = i
263 263 if j < n and pat[j] in '!]':
264 264 j += 1
265 265 while j < n and pat[j] != ']':
266 266 j += 1
267 267 if j >= n:
268 268 res += '\\['
269 269 else:
270 270 stuff = pat[i:j].replace('\\','\\\\')
271 271 i = j + 1
272 272 if stuff[0] == '!':
273 273 stuff = '^' + stuff[1:]
274 274 elif stuff[0] == '^':
275 275 stuff = '\\' + stuff
276 276 res = '%s[%s]' % (res, stuff)
277 277 elif c == '{':
278 278 group = True
279 279 res += '(?:'
280 280 elif c == '}' and group:
281 281 res += ')'
282 282 group = False
283 283 elif c == ',' and group:
284 284 res += '|'
285 285 elif c == '\\':
286 286 p = peek()
287 287 if p:
288 288 i += 1
289 289 res += re.escape(p)
290 290 else:
291 291 res += re.escape(c)
292 292 else:
293 293 res += re.escape(c)
294 294 return head + res + tail
295 295
296 296 _globchars = {'[': 1, '{': 1, '*': 1, '?': 1}
297 297
298 298 def pathto(n1, n2):
299 299 '''return the relative path from one place to another.
300 300 n1 should use os.sep to separate directories
301 301 n2 should use "/" to separate directories
302 302 returns an os.sep-separated path.
303 303 '''
304 304 if not n1: return localpath(n2)
305 305 a, b = n1.split(os.sep), n2.split('/')
306 306 a.reverse()
307 307 b.reverse()
308 308 while a and b and a[-1] == b[-1]:
309 309 a.pop()
310 310 b.pop()
311 311 b.reverse()
312 312 return os.sep.join((['..'] * len(a)) + b)
313 313
314 314 def canonpath(root, cwd, myname):
315 315 """return the canonical path of myname, given cwd and root"""
316 316 if root == os.sep:
317 317 rootsep = os.sep
318 318 elif root.endswith(os.sep):
319 319 rootsep = root
320 320 else:
321 321 rootsep = root + os.sep
322 322 name = myname
323 323 if not os.path.isabs(name):
324 324 name = os.path.join(root, cwd, name)
325 325 name = os.path.normpath(name)
326 326 if name != rootsep and name.startswith(rootsep):
327 327 name = name[len(rootsep):]
328 328 audit_path(name)
329 329 return pconvert(name)
330 330 elif name == root:
331 331 return ''
332 332 else:
333 333 # Determine whether `name' is in the hierarchy at or beneath `root',
334 334 # by iterating name=dirname(name) until that causes no change (can't
335 335 # check name == '/', because that doesn't work on windows). For each
336 336 # `name', compare dev/inode numbers. If they match, the list `rel'
337 337 # holds the reversed list of components making up the relative file
338 338 # name we want.
339 339 root_st = os.stat(root)
340 340 rel = []
341 341 while True:
342 342 try:
343 343 name_st = os.stat(name)
344 344 except OSError:
345 345 break
346 346 if samestat(name_st, root_st):
347 347 rel.reverse()
348 348 name = os.path.join(*rel)
349 349 audit_path(name)
350 350 return pconvert(name)
351 351 dirname, basename = os.path.split(name)
352 352 rel.append(basename)
353 353 if dirname == name:
354 354 break
355 355 name = dirname
356 356
357 357 raise Abort('%s not under root' % myname)
358 358
359 359 def matcher(canonroot, cwd='', names=['.'], inc=[], exc=[], head='', src=None):
360 360 return _matcher(canonroot, cwd, names, inc, exc, head, 'glob', src)
361 361
362 362 def cmdmatcher(canonroot, cwd='', names=['.'], inc=[], exc=[], head='', src=None):
363 363 if os.name == 'nt':
364 364 dflt_pat = 'glob'
365 365 else:
366 366 dflt_pat = 'relpath'
367 367 return _matcher(canonroot, cwd, names, inc, exc, head, dflt_pat, src)
368 368
369 369 def _matcher(canonroot, cwd, names, inc, exc, head, dflt_pat, src):
370 370 """build a function to match a set of file patterns
371 371
372 372 arguments:
373 373 canonroot - the canonical root of the tree you're matching against
374 374 cwd - the current working directory, if relevant
375 375 names - patterns to find
376 376 inc - patterns to include
377 377 exc - patterns to exclude
378 378 head - a regex to prepend to patterns to control whether a match is rooted
379 379
380 380 a pattern is one of:
381 381 'glob:<rooted glob>'
382 382 're:<rooted regexp>'
383 383 'path:<rooted path>'
384 384 'relglob:<relative glob>'
385 385 'relpath:<relative path>'
386 386 'relre:<relative regexp>'
387 387 '<rooted path or regexp>'
388 388
389 389 returns:
390 390 a 3-tuple containing
391 391 - list of explicit non-pattern names passed in
392 392 - a bool match(filename) function
393 393 - a bool indicating if any patterns were passed in
394 394
395 395 todo:
396 396 make head regex a rooted bool
397 397 """
398 398
399 399 def contains_glob(name):
400 400 for c in name:
401 401 if c in _globchars: return True
402 402 return False
403 403
404 404 def regex(kind, name, tail):
405 405 '''convert a pattern into a regular expression'''
406 406 if kind == 're':
407 407 return name
408 408 elif kind == 'path':
409 409 return '^' + re.escape(name) + '(?:/|$)'
410 410 elif kind == 'relglob':
411 411 return head + globre(name, '(?:|.*/)', tail)
412 412 elif kind == 'relpath':
413 413 return head + re.escape(name) + tail
414 414 elif kind == 'relre':
415 415 if name.startswith('^'):
416 416 return name
417 417 return '.*' + name
418 418 return head + globre(name, '', tail)
419 419
420 420 def matchfn(pats, tail):
421 421 """build a matching function from a set of patterns"""
422 422 if not pats:
423 423 return
424 424 matches = []
425 425 for k, p in pats:
426 426 try:
427 427 pat = '(?:%s)' % regex(k, p, tail)
428 428 matches.append(re.compile(pat).match)
429 429 except re.error:
430 430 if src: raise Abort("%s: invalid pattern (%s): %s" % (src, k, p))
431 431 else: raise Abort("invalid pattern (%s): %s" % (k, p))
432 432
433 433 def buildfn(text):
434 434 for m in matches:
435 435 r = m(text)
436 436 if r:
437 437 return r
438 438
439 439 return buildfn
440 440
441 441 def globprefix(pat):
442 442 '''return the non-glob prefix of a path, e.g. foo/* -> foo'''
443 443 root = []
444 444 for p in pat.split(os.sep):
445 445 if contains_glob(p): break
446 446 root.append(p)
447 447 return '/'.join(root)
448 448
449 449 pats = []
450 450 files = []
451 451 roots = []
452 452 for kind, name in [patkind(p, dflt_pat) for p in names]:
453 453 if kind in ('glob', 'relpath'):
454 454 name = canonpath(canonroot, cwd, name)
455 455 if name == '':
456 456 kind, name = 'glob', '**'
457 457 if kind in ('glob', 'path', 're'):
458 458 pats.append((kind, name))
459 459 if kind == 'glob':
460 460 root = globprefix(name)
461 461 if root: roots.append(root)
462 462 elif kind == 'relpath':
463 463 files.append((kind, name))
464 464 roots.append(name)
465 465
466 466 patmatch = matchfn(pats, '$') or always
467 467 filematch = matchfn(files, '(?:/|$)') or always
468 468 incmatch = always
469 469 if inc:
470 470 inckinds = [patkind(canonpath(canonroot, cwd, i)) for i in inc]
471 471 incmatch = matchfn(inckinds, '(?:/|$)')
472 472 excmatch = lambda fn: False
473 473 if exc:
474 474 exckinds = [patkind(canonpath(canonroot, cwd, x)) for x in exc]
475 475 excmatch = matchfn(exckinds, '(?:/|$)')
476 476
477 477 return (roots,
478 478 lambda fn: (incmatch(fn) and not excmatch(fn) and
479 479 (fn.endswith('/') or
480 480 (not pats and not files) or
481 481 (pats and patmatch(fn)) or
482 482 (files and filematch(fn)))),
483 483 (inc or exc or (pats and pats != [('glob', '**')])) and True)
484 484
485 485 def system(cmd, environ={}, cwd=None, onerr=None, errprefix=None):
486 486 '''enhanced shell command execution.
487 487 run with environment maybe modified, maybe in different dir.
488 488
489 489 if command fails and onerr is None, return status. if ui object,
490 490 print error message and return status, else raise onerr object as
491 491 exception.'''
492 492 def py2shell(val):
493 493 'convert python object into string that is useful to shell'
494 494 if val in (None, False):
495 495 return '0'
496 496 if val == True:
497 497 return '1'
498 498 return str(val)
499 499 oldenv = {}
500 500 for k in environ:
501 501 oldenv[k] = os.environ.get(k)
502 502 if cwd is not None:
503 503 oldcwd = os.getcwd()
504 504 try:
505 505 for k, v in environ.iteritems():
506 506 os.environ[k] = py2shell(v)
507 507 if cwd is not None and oldcwd != cwd:
508 508 os.chdir(cwd)
509 509 rc = os.system(cmd)
510 510 if rc and onerr:
511 511 errmsg = '%s %s' % (os.path.basename(cmd.split(None, 1)[0]),
512 512 explain_exit(rc)[0])
513 513 if errprefix:
514 514 errmsg = '%s: %s' % (errprefix, errmsg)
515 515 try:
516 516 onerr.warn(errmsg + '\n')
517 517 except AttributeError:
518 518 raise onerr(errmsg)
519 519 return rc
520 520 finally:
521 521 for k, v in oldenv.iteritems():
522 522 if v is None:
523 523 del os.environ[k]
524 524 else:
525 525 os.environ[k] = v
526 526 if cwd is not None and oldcwd != cwd:
527 527 os.chdir(oldcwd)
528 528
529 529 def rename(src, dst):
530 530 """forcibly rename a file"""
531 531 try:
532 532 os.rename(src, dst)
533 533 except OSError, err:
534 534 # on windows, rename to existing file is not allowed, so we
535 535 # must delete destination first. but if file is open, unlink
536 536 # schedules it for delete but does not delete it. rename
537 537 # happens immediately even for open files, so we create
538 538 # temporary file, delete it, rename destination to that name,
539 539 # then delete that. then rename is safe to do.
540 540 fd, temp = tempfile.mkstemp(dir=os.path.dirname(dst) or '.')
541 541 os.close(fd)
542 542 os.unlink(temp)
543 543 os.rename(dst, temp)
544 544 os.unlink(temp)
545 545 os.rename(src, dst)
546 546
547 547 def unlink(f):
548 548 """unlink and remove the directory if it is empty"""
549 549 os.unlink(f)
550 550 # try removing directories that might now be empty
551 551 try:
552 552 os.removedirs(os.path.dirname(f))
553 553 except OSError:
554 554 pass
555 555
556 556 def copyfile(src, dest):
557 557 "copy a file, preserving mode"
558 558 try:
559 559 shutil.copyfile(src, dest)
560 560 shutil.copymode(src, dest)
561 561 except shutil.Error, inst:
562 562 raise util.Abort(str(inst))
563 563
564 564 def copyfiles(src, dst, hardlink=None):
565 565 """Copy a directory tree using hardlinks if possible"""
566 566
567 567 if hardlink is None:
568 568 hardlink = (os.stat(src).st_dev ==
569 569 os.stat(os.path.dirname(dst)).st_dev)
570 570
571 571 if os.path.isdir(src):
572 572 os.mkdir(dst)
573 573 for name in os.listdir(src):
574 574 srcname = os.path.join(src, name)
575 575 dstname = os.path.join(dst, name)
576 576 copyfiles(srcname, dstname, hardlink)
577 577 else:
578 578 if hardlink:
579 579 try:
580 580 os_link(src, dst)
581 581 except (IOError, OSError):
582 582 hardlink = False
583 583 shutil.copy(src, dst)
584 584 else:
585 585 shutil.copy(src, dst)
586 586
587 587 def audit_path(path):
588 588 """Abort if path contains dangerous components"""
589 589 parts = os.path.normcase(path).split(os.sep)
590 590 if (os.path.splitdrive(path)[0] or parts[0] in ('.hg', '')
591 591 or os.pardir in parts):
592 592 raise Abort(_("path contains illegal component: %s\n") % path)
593 593
594 594 def _makelock_file(info, pathname):
595 595 ld = os.open(pathname, os.O_CREAT | os.O_WRONLY | os.O_EXCL)
596 596 os.write(ld, info)
597 597 os.close(ld)
598 598
599 599 def _readlock_file(pathname):
600 600 return posixfile(pathname).read()
601 601
602 602 def nlinks(pathname):
603 603 """Return number of hardlinks for the given file."""
604 604 return os.lstat(pathname).st_nlink
605 605
606 606 if hasattr(os, 'link'):
607 607 os_link = os.link
608 608 else:
609 609 def os_link(src, dst):
610 610 raise OSError(0, _("Hardlinks not supported"))
611 611
612 612 def fstat(fp):
613 613 '''stat file object that may not have fileno method.'''
614 614 try:
615 615 return os.fstat(fp.fileno())
616 616 except AttributeError:
617 617 return os.stat(fp.name)
618 618
619 619 posixfile = file
620 620
621 621 def is_win_9x():
622 622 '''return true if run on windows 95, 98 or me.'''
623 623 try:
624 624 return sys.getwindowsversion()[3] == 1
625 625 except AttributeError:
626 626 return os.name == 'nt' and 'command' in os.environ.get('comspec', '')
627 627
628 628 getuser_fallback = None
629 629
630 630 def getuser():
631 631 '''return name of current user'''
632 632 try:
633 633 return getpass.getuser()
634 634 except ImportError:
635 635 # import of pwd will fail on windows - try fallback
636 636 if getuser_fallback:
637 637 return getuser_fallback()
638 638 # raised if win32api not available
639 639 raise Abort(_('user name not available - set USERNAME '
640 640 'environment variable'))
641 641
642 642 def username(uid=None):
643 643 """Return the name of the user with the given uid.
644 644
645 645 If uid is None, return the name of the current user."""
646 646 try:
647 647 import pwd
648 648 if uid is None:
649 649 uid = os.getuid()
650 650 try:
651 651 return pwd.getpwuid(uid)[0]
652 652 except KeyError:
653 653 return str(uid)
654 654 except ImportError:
655 655 return None
656 656
657 657 def groupname(gid=None):
658 658 """Return the name of the group with the given gid.
659 659
660 660 If gid is None, return the name of the current group."""
661 661 try:
662 662 import grp
663 663 if gid is None:
664 664 gid = os.getgid()
665 665 try:
666 666 return grp.getgrgid(gid)[0]
667 667 except KeyError:
668 668 return str(gid)
669 669 except ImportError:
670 670 return None
671 671
672 672 # File system features
673 673
674 674 def checkfolding(path):
675 675 """
676 676 Check whether the given path is on a case-sensitive filesystem
677 677
678 678 Requires a path (like /foo/.hg) ending with a foldable final
679 679 directory component.
680 680 """
681 681 s1 = os.stat(path)
682 682 d, b = os.path.split(path)
683 683 p2 = os.path.join(d, b.upper())
684 684 if path == p2:
685 685 p2 = os.path.join(d, b.lower())
686 686 try:
687 687 s2 = os.stat(p2)
688 688 if s2 == s1:
689 689 return False
690 690 return True
691 691 except:
692 692 return True
693 693
694 694 # Platform specific variants
695 695 if os.name == 'nt':
696 696 demandload(globals(), "msvcrt")
697 697 nulldev = 'NUL:'
698 698
699 699 class winstdout:
700 700 '''stdout on windows misbehaves if sent through a pipe'''
701 701
702 702 def __init__(self, fp):
703 703 self.fp = fp
704 704
705 705 def __getattr__(self, key):
706 706 return getattr(self.fp, key)
707 707
708 708 def close(self):
709 709 try:
710 710 self.fp.close()
711 711 except: pass
712 712
713 713 def write(self, s):
714 714 try:
715 715 return self.fp.write(s)
716 716 except IOError, inst:
717 717 if inst.errno != 0: raise
718 718 self.close()
719 719 raise IOError(errno.EPIPE, 'Broken pipe')
720 720
721 721 sys.stdout = winstdout(sys.stdout)
722 722
723 723 def system_rcpath():
724 724 try:
725 725 return system_rcpath_win32()
726 726 except:
727 727 return [r'c:\mercurial\mercurial.ini']
728 728
729 729 def os_rcpath():
730 730 '''return default os-specific hgrc search path'''
731 731 path = system_rcpath()
732 732 path.append(user_rcpath())
733 733 userprofile = os.environ.get('USERPROFILE')
734 734 if userprofile:
735 735 path.append(os.path.join(userprofile, 'mercurial.ini'))
736 736 return path
737 737
738 738 def user_rcpath():
739 739 '''return os-specific hgrc search path to the user dir'''
740 740 return os.path.join(os.path.expanduser('~'), 'mercurial.ini')
741 741
742 742 def parse_patch_output(output_line):
743 743 """parses the output produced by patch and returns the file name"""
744 744 pf = output_line[14:]
745 745 if pf[0] == '`':
746 746 pf = pf[1:-1] # Remove the quotes
747 747 return pf
748 748
749 749 def testpid(pid):
750 750 '''return False if pid dead, True if running or not known'''
751 751 return True
752 752
753 753 def is_exec(f, last):
754 754 return last
755 755
756 756 def set_exec(f, mode):
757 757 pass
758 758
759 759 def set_binary(fd):
760 760 msvcrt.setmode(fd.fileno(), os.O_BINARY)
761 761
762 762 def pconvert(path):
763 763 return path.replace("\\", "/")
764 764
765 765 def localpath(path):
766 766 return path.replace('/', '\\')
767 767
768 768 def normpath(path):
769 769 return pconvert(os.path.normpath(path))
770 770
771 771 makelock = _makelock_file
772 772 readlock = _readlock_file
773 773
774 774 def samestat(s1, s2):
775 775 return False
776 776
777 777 def shellquote(s):
778 778 return '"%s"' % s.replace('"', '\\"')
779 779
780 780 def explain_exit(code):
781 781 return _("exited with status %d") % code, code
782 782
783 783 # if you change this stub into a real check, please try to implement the
784 784 # username and groupname functions above, too.
785 785 def isowner(fp, st=None):
786 786 return True
787 787
788 788 try:
789 789 # override functions with win32 versions if possible
790 790 from util_win32 import *
791 791 if not is_win_9x():
792 792 posixfile = posixfile_nt
793 793 except ImportError:
794 794 pass
795 795
796 796 else:
797 797 nulldev = '/dev/null'
798 798
799 799 def rcfiles(path):
800 800 rcs = [os.path.join(path, 'hgrc')]
801 801 rcdir = os.path.join(path, 'hgrc.d')
802 802 try:
803 803 rcs.extend([os.path.join(rcdir, f) for f in os.listdir(rcdir)
804 804 if f.endswith(".rc")])
805 805 except OSError:
806 806 pass
807 807 return rcs
808 808
809 809 def os_rcpath():
810 810 '''return default os-specific hgrc search path'''
811 811 path = []
812 812 # old mod_python does not set sys.argv
813 813 if len(getattr(sys, 'argv', [])) > 0:
814 814 path.extend(rcfiles(os.path.dirname(sys.argv[0]) +
815 815 '/../etc/mercurial'))
816 816 path.extend(rcfiles('/etc/mercurial'))
817 817 path.append(os.path.expanduser('~/.hgrc'))
818 818 path = [os.path.normpath(f) for f in path]
819 819 return path
820 820
821 821 def parse_patch_output(output_line):
822 822 """parses the output produced by patch and returns the file name"""
823 823 pf = output_line[14:]
824 824 if pf.startswith("'") and pf.endswith("'") and " " in pf:
825 825 pf = pf[1:-1] # Remove the quotes
826 826 return pf
827 827
828 828 def is_exec(f, last):
829 829 """check whether a file is executable"""
830 830 return (os.lstat(f).st_mode & 0100 != 0)
831 831
832 832 def set_exec(f, mode):
833 833 s = os.lstat(f).st_mode
834 834 if (s & 0100 != 0) == mode:
835 835 return
836 836 if mode:
837 837 # Turn on +x for every +r bit when making a file executable
838 838 # and obey umask.
839 839 umask = os.umask(0)
840 840 os.umask(umask)
841 841 os.chmod(f, s | (s & 0444) >> 2 & ~umask)
842 842 else:
843 843 os.chmod(f, s & 0666)
844 844
845 845 def set_binary(fd):
846 846 pass
847 847
848 848 def pconvert(path):
849 849 return path
850 850
851 851 def localpath(path):
852 852 return path
853 853
854 854 normpath = os.path.normpath
855 855 samestat = os.path.samestat
856 856
857 857 def makelock(info, pathname):
858 858 try:
859 859 os.symlink(info, pathname)
860 860 except OSError, why:
861 861 if why.errno == errno.EEXIST:
862 862 raise
863 863 else:
864 864 _makelock_file(info, pathname)
865 865
866 866 def readlock(pathname):
867 867 try:
868 868 return os.readlink(pathname)
869 869 except OSError, why:
870 870 if why.errno == errno.EINVAL:
871 871 return _readlock_file(pathname)
872 872 else:
873 873 raise
874 874
875 875 def shellquote(s):
876 876 return "'%s'" % s.replace("'", "'\\''")
877 877
878 878 def testpid(pid):
879 879 '''return False if pid dead, True if running or not sure'''
880 880 try:
881 881 os.kill(pid, 0)
882 882 return True
883 883 except OSError, inst:
884 884 return inst.errno != errno.ESRCH
885 885
886 886 def explain_exit(code):
887 887 """return a 2-tuple (desc, code) describing a process's status"""
888 888 if os.WIFEXITED(code):
889 889 val = os.WEXITSTATUS(code)
890 890 return _("exited with status %d") % val, val
891 891 elif os.WIFSIGNALED(code):
892 892 val = os.WTERMSIG(code)
893 893 return _("killed by signal %d") % val, val
894 894 elif os.WIFSTOPPED(code):
895 895 val = os.WSTOPSIG(code)
896 896 return _("stopped by signal %d") % val, val
897 897 raise ValueError(_("invalid exit code"))
898 898
899 899 def isowner(fp, st=None):
900 900 """Return True if the file object f belongs to the current user.
901 901
902 902 The return value of a util.fstat(f) may be passed as the st argument.
903 903 """
904 904 if st is None:
905 st = fstat(f)
905 st = fstat(fp)
906 906 return st.st_uid == os.getuid()
907 907
908 908 def _buildencodefun():
909 909 e = '_'
910 910 win_reserved = [ord(x) for x in '|\?*<":>+[]']
911 911 cmap = dict([ (chr(x), chr(x)) for x in xrange(127) ])
912 912 for x in (range(32) + range(126, 256) + win_reserved):
913 913 cmap[chr(x)] = "~%02x" % x
914 914 for x in range(ord("A"), ord("Z")+1) + [ord(e)]:
915 915 cmap[chr(x)] = e + chr(x).lower()
916 916 dmap = {}
917 917 for k, v in cmap.iteritems():
918 918 dmap[v] = k
919 919 def decode(s):
920 920 i = 0
921 921 while i < len(s):
922 922 for l in xrange(1, 4):
923 923 try:
924 924 yield dmap[s[i:i+l]]
925 925 i += l
926 926 break
927 927 except KeyError:
928 928 pass
929 929 else:
930 930 raise KeyError
931 931 return (lambda s: "".join([cmap[c] for c in s]),
932 932 lambda s: "".join(list(decode(s))))
933 933
934 934 encodefilename, decodefilename = _buildencodefun()
935 935
936 936 def encodedopener(openerfn, fn):
937 937 def o(path, *args, **kw):
938 938 return openerfn(fn(path), *args, **kw)
939 939 return o
940 940
941 941 def opener(base, audit=True):
942 942 """
943 943 return a function that opens files relative to base
944 944
945 945 this function is used to hide the details of COW semantics and
946 946 remote file access from higher level code.
947 947 """
948 948 p = base
949 949 audit_p = audit
950 950
951 951 def mktempcopy(name):
952 952 d, fn = os.path.split(name)
953 953 fd, temp = tempfile.mkstemp(prefix='.%s-' % fn, dir=d)
954 954 os.close(fd)
955 955 ofp = posixfile(temp, "wb")
956 956 try:
957 957 try:
958 958 ifp = posixfile(name, "rb")
959 959 except IOError, inst:
960 960 if not getattr(inst, 'filename', None):
961 961 inst.filename = name
962 962 raise
963 963 for chunk in filechunkiter(ifp):
964 964 ofp.write(chunk)
965 965 ifp.close()
966 966 ofp.close()
967 967 except:
968 968 try: os.unlink(temp)
969 969 except: pass
970 970 raise
971 971 st = os.lstat(name)
972 972 os.chmod(temp, st.st_mode)
973 973 return temp
974 974
975 975 class atomictempfile(posixfile):
976 976 """the file will only be copied when rename is called"""
977 977 def __init__(self, name, mode):
978 978 self.__name = name
979 979 self.temp = mktempcopy(name)
980 980 posixfile.__init__(self, self.temp, mode)
981 981 def rename(self):
982 982 if not self.closed:
983 983 posixfile.close(self)
984 984 rename(self.temp, localpath(self.__name))
985 985 def __del__(self):
986 986 if not self.closed:
987 987 try:
988 988 os.unlink(self.temp)
989 989 except: pass
990 990 posixfile.close(self)
991 991
992 992 class atomicfile(atomictempfile):
993 993 """the file will only be copied on close"""
994 994 def __init__(self, name, mode):
995 995 atomictempfile.__init__(self, name, mode)
996 996 def close(self):
997 997 self.rename()
998 998 def __del__(self):
999 999 self.rename()
1000 1000
1001 1001 def o(path, mode="r", text=False, atomic=False, atomictemp=False):
1002 1002 if audit_p:
1003 1003 audit_path(path)
1004 1004 f = os.path.join(p, path)
1005 1005
1006 1006 if not text:
1007 1007 mode += "b" # for that other OS
1008 1008
1009 1009 if mode[0] != "r":
1010 1010 try:
1011 1011 nlink = nlinks(f)
1012 1012 except OSError:
1013 1013 d = os.path.dirname(f)
1014 1014 if not os.path.isdir(d):
1015 1015 os.makedirs(d)
1016 1016 else:
1017 1017 if atomic:
1018 1018 return atomicfile(f, mode)
1019 1019 elif atomictemp:
1020 1020 return atomictempfile(f, mode)
1021 1021 if nlink > 1:
1022 1022 rename(mktempcopy(f), f)
1023 1023 return posixfile(f, mode)
1024 1024
1025 1025 return o
1026 1026
1027 1027 class chunkbuffer(object):
1028 1028 """Allow arbitrary sized chunks of data to be efficiently read from an
1029 1029 iterator over chunks of arbitrary size."""
1030 1030
1031 1031 def __init__(self, in_iter, targetsize = 2**16):
1032 1032 """in_iter is the iterator that's iterating over the input chunks.
1033 1033 targetsize is how big a buffer to try to maintain."""
1034 1034 self.in_iter = iter(in_iter)
1035 1035 self.buf = ''
1036 1036 self.targetsize = int(targetsize)
1037 1037 if self.targetsize <= 0:
1038 1038 raise ValueError(_("targetsize must be greater than 0, was %d") %
1039 1039 targetsize)
1040 1040 self.iterempty = False
1041 1041
1042 1042 def fillbuf(self):
1043 1043 """Ignore target size; read every chunk from iterator until empty."""
1044 1044 if not self.iterempty:
1045 1045 collector = cStringIO.StringIO()
1046 1046 collector.write(self.buf)
1047 1047 for ch in self.in_iter:
1048 1048 collector.write(ch)
1049 1049 self.buf = collector.getvalue()
1050 1050 self.iterempty = True
1051 1051
1052 1052 def read(self, l):
1053 1053 """Read L bytes of data from the iterator of chunks of data.
1054 1054 Returns less than L bytes if the iterator runs dry."""
1055 1055 if l > len(self.buf) and not self.iterempty:
1056 1056 # Clamp to a multiple of self.targetsize
1057 1057 targetsize = self.targetsize * ((l // self.targetsize) + 1)
1058 1058 collector = cStringIO.StringIO()
1059 1059 collector.write(self.buf)
1060 1060 collected = len(self.buf)
1061 1061 for chunk in self.in_iter:
1062 1062 collector.write(chunk)
1063 1063 collected += len(chunk)
1064 1064 if collected >= targetsize:
1065 1065 break
1066 1066 if collected < targetsize:
1067 1067 self.iterempty = True
1068 1068 self.buf = collector.getvalue()
1069 1069 s, self.buf = self.buf[:l], buffer(self.buf, l)
1070 1070 return s
1071 1071
1072 1072 def filechunkiter(f, size=65536, limit=None):
1073 1073 """Create a generator that produces the data in the file size
1074 1074 (default 65536) bytes at a time, up to optional limit (default is
1075 1075 to read all data). Chunks may be less than size bytes if the
1076 1076 chunk is the last chunk in the file, or the file is a socket or
1077 1077 some other type of file that sometimes reads less data than is
1078 1078 requested."""
1079 1079 assert size >= 0
1080 1080 assert limit is None or limit >= 0
1081 1081 while True:
1082 1082 if limit is None: nbytes = size
1083 1083 else: nbytes = min(limit, size)
1084 1084 s = nbytes and f.read(nbytes)
1085 1085 if not s: break
1086 1086 if limit: limit -= len(s)
1087 1087 yield s
1088 1088
1089 1089 def makedate():
1090 1090 lt = time.localtime()
1091 1091 if lt[8] == 1 and time.daylight:
1092 1092 tz = time.altzone
1093 1093 else:
1094 1094 tz = time.timezone
1095 1095 return time.mktime(lt), tz
1096 1096
1097 1097 def datestr(date=None, format='%a %b %d %H:%M:%S %Y', timezone=True):
1098 1098 """represent a (unixtime, offset) tuple as a localized time.
1099 1099 unixtime is seconds since the epoch, and offset is the time zone's
1100 1100 number of seconds away from UTC. if timezone is false, do not
1101 1101 append time zone to string."""
1102 1102 t, tz = date or makedate()
1103 1103 s = time.strftime(format, time.gmtime(float(t) - tz))
1104 1104 if timezone:
1105 1105 s += " %+03d%02d" % (-tz / 3600, ((-tz % 3600) / 60))
1106 1106 return s
1107 1107
1108 1108 def strdate(string, format, defaults):
1109 1109 """parse a localized time string and return a (unixtime, offset) tuple.
1110 1110 if the string cannot be parsed, ValueError is raised."""
1111 1111 def timezone(string):
1112 1112 tz = string.split()[-1]
1113 1113 if tz[0] in "+-" and len(tz) == 5 and tz[1:].isdigit():
1114 1114 tz = int(tz)
1115 1115 offset = - 3600 * (tz / 100) - 60 * (tz % 100)
1116 1116 return offset
1117 1117 if tz == "GMT" or tz == "UTC":
1118 1118 return 0
1119 1119 return None
1120 1120
1121 1121 # NOTE: unixtime = localunixtime + offset
1122 1122 offset, date = timezone(string), string
1123 1123 if offset != None:
1124 1124 date = " ".join(string.split()[:-1])
1125 1125
1126 1126 # add missing elements from defaults
1127 1127 for part in defaults:
1128 1128 found = [True for p in part if ("%"+p) in format]
1129 1129 if not found:
1130 1130 date += "@" + defaults[part]
1131 1131 format += "@%" + part[0]
1132 1132
1133 1133 timetuple = time.strptime(date, format)
1134 1134 localunixtime = int(calendar.timegm(timetuple))
1135 1135 if offset is None:
1136 1136 # local timezone
1137 1137 unixtime = int(time.mktime(timetuple))
1138 1138 offset = unixtime - localunixtime
1139 1139 else:
1140 1140 unixtime = localunixtime + offset
1141 1141 return unixtime, offset
1142 1142
1143 1143 def parsedate(string, formats=None, defaults=None):
1144 1144 """parse a localized time string and return a (unixtime, offset) tuple.
1145 1145 The date may be a "unixtime offset" string or in one of the specified
1146 1146 formats."""
1147 1147 if not string:
1148 1148 return 0, 0
1149 1149 if not formats:
1150 1150 formats = defaultdateformats
1151 1151 string = string.strip()
1152 1152 try:
1153 1153 when, offset = map(int, string.split(' '))
1154 1154 except ValueError:
1155 1155 # fill out defaults
1156 1156 if not defaults:
1157 1157 defaults = {}
1158 1158 now = makedate()
1159 1159 for part in "d mb yY HI M S".split():
1160 1160 if part not in defaults:
1161 1161 if part[0] in "HMS":
1162 1162 defaults[part] = "00"
1163 1163 elif part[0] in "dm":
1164 1164 defaults[part] = "1"
1165 1165 else:
1166 1166 defaults[part] = datestr(now, "%" + part[0], False)
1167 1167
1168 1168 for format in formats:
1169 1169 try:
1170 1170 when, offset = strdate(string, format, defaults)
1171 1171 except ValueError:
1172 1172 pass
1173 1173 else:
1174 1174 break
1175 1175 else:
1176 1176 raise Abort(_('invalid date: %r ') % string)
1177 1177 # validate explicit (probably user-specified) date and
1178 1178 # time zone offset. values must fit in signed 32 bits for
1179 1179 # current 32-bit linux runtimes. timezones go from UTC-12
1180 1180 # to UTC+14
1181 1181 if abs(when) > 0x7fffffff:
1182 1182 raise Abort(_('date exceeds 32 bits: %d') % when)
1183 1183 if offset < -50400 or offset > 43200:
1184 1184 raise Abort(_('impossible time zone offset: %d') % offset)
1185 1185 return when, offset
1186 1186
1187 1187 def matchdate(date):
1188 1188 """Return a function that matches a given date match specifier
1189 1189
1190 1190 Formats include:
1191 1191
1192 1192 '{date}' match a given date to the accuracy provided
1193 1193
1194 1194 '<{date}' on or before a given date
1195 1195
1196 1196 '>{date}' on or after a given date
1197 1197
1198 1198 """
1199 1199
1200 1200 def lower(date):
1201 1201 return parsedate(date, extendeddateformats)[0]
1202 1202
1203 1203 def upper(date):
1204 1204 d = dict(mb="12", HI="23", M="59", S="59")
1205 1205 for days in "31 30 29".split():
1206 1206 try:
1207 1207 d["d"] = days
1208 1208 return parsedate(date, extendeddateformats, d)[0]
1209 1209 except:
1210 1210 pass
1211 1211 d["d"] = "28"
1212 1212 return parsedate(date, extendeddateformats, d)[0]
1213 1213
1214 1214 if date[0] == "<":
1215 1215 when = upper(date[1:])
1216 1216 return lambda x: x <= when
1217 1217 elif date[0] == ">":
1218 1218 when = lower(date[1:])
1219 1219 return lambda x: x >= when
1220 1220 elif date[0] == "-":
1221 1221 try:
1222 1222 days = int(date[1:])
1223 1223 except ValueError:
1224 1224 raise Abort(_("invalid day spec: %s") % date[1:])
1225 1225 when = makedate()[0] - days * 3600 * 24
1226 1226 return lambda x: x >= when
1227 1227 elif " to " in date:
1228 1228 a, b = date.split(" to ")
1229 1229 start, stop = lower(a), upper(b)
1230 1230 return lambda x: x >= start and x <= stop
1231 1231 else:
1232 1232 start, stop = lower(date), upper(date)
1233 1233 return lambda x: x >= start and x <= stop
1234 1234
1235 1235 def shortuser(user):
1236 1236 """Return a short representation of a user name or email address."""
1237 1237 f = user.find('@')
1238 1238 if f >= 0:
1239 1239 user = user[:f]
1240 1240 f = user.find('<')
1241 1241 if f >= 0:
1242 1242 user = user[f+1:]
1243 1243 f = user.find(' ')
1244 1244 if f >= 0:
1245 1245 user = user[:f]
1246 1246 f = user.find('.')
1247 1247 if f >= 0:
1248 1248 user = user[:f]
1249 1249 return user
1250 1250
1251 1251 def ellipsis(text, maxlength=400):
1252 1252 """Trim string to at most maxlength (default: 400) characters."""
1253 1253 if len(text) <= maxlength:
1254 1254 return text
1255 1255 else:
1256 1256 return "%s..." % (text[:maxlength-3])
1257 1257
1258 1258 def walkrepos(path):
1259 1259 '''yield every hg repository under path, recursively.'''
1260 1260 def errhandler(err):
1261 1261 if err.filename == path:
1262 1262 raise err
1263 1263
1264 1264 for root, dirs, files in os.walk(path, onerror=errhandler):
1265 1265 for d in dirs:
1266 1266 if d == '.hg':
1267 1267 yield root
1268 1268 dirs[:] = []
1269 1269 break
1270 1270
1271 1271 _rcpath = None
1272 1272
1273 1273 def rcpath():
1274 1274 '''return hgrc search path. if env var HGRCPATH is set, use it.
1275 1275 for each item in path, if directory, use files ending in .rc,
1276 1276 else use item.
1277 1277 make HGRCPATH empty to only look in .hg/hgrc of current repo.
1278 1278 if no HGRCPATH, use default os-specific path.'''
1279 1279 global _rcpath
1280 1280 if _rcpath is None:
1281 1281 if 'HGRCPATH' in os.environ:
1282 1282 _rcpath = []
1283 1283 for p in os.environ['HGRCPATH'].split(os.pathsep):
1284 1284 if not p: continue
1285 1285 if os.path.isdir(p):
1286 1286 for f in os.listdir(p):
1287 1287 if f.endswith('.rc'):
1288 1288 _rcpath.append(os.path.join(p, f))
1289 1289 else:
1290 1290 _rcpath.append(p)
1291 1291 else:
1292 1292 _rcpath = os_rcpath()
1293 1293 return _rcpath
1294 1294
1295 1295 def bytecount(nbytes):
1296 1296 '''return byte count formatted as readable string, with units'''
1297 1297
1298 1298 units = (
1299 1299 (100, 1<<30, _('%.0f GB')),
1300 1300 (10, 1<<30, _('%.1f GB')),
1301 1301 (1, 1<<30, _('%.2f GB')),
1302 1302 (100, 1<<20, _('%.0f MB')),
1303 1303 (10, 1<<20, _('%.1f MB')),
1304 1304 (1, 1<<20, _('%.2f MB')),
1305 1305 (100, 1<<10, _('%.0f KB')),
1306 1306 (10, 1<<10, _('%.1f KB')),
1307 1307 (1, 1<<10, _('%.2f KB')),
1308 1308 (1, 1, _('%.0f bytes')),
1309 1309 )
1310 1310
1311 1311 for multiplier, divisor, format in units:
1312 1312 if nbytes >= divisor * multiplier:
1313 1313 return format % (nbytes / float(divisor))
1314 1314 return units[-1][2] % nbytes
1315 1315
1316 1316 def drop_scheme(scheme, path):
1317 1317 sc = scheme + ':'
1318 1318 if path.startswith(sc):
1319 1319 path = path[len(sc):]
1320 1320 if path.startswith('//'):
1321 1321 path = path[2:]
1322 1322 return path
General Comments 0
You need to be logged in to leave comments. Login now