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