##// END OF EJS Templates
vfs: split "expand" into "realpath"/"expandpath" to apply each separately...
FUJIWARA Katsunori -
r18945:e75b72ff default
parent child Browse files
Show More
@@ -1,2594 +1,2594 b''
1 1 # localrepo.py - read/write repository class for mercurial
2 2 #
3 3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 4 #
5 5 # This software may be used and distributed according to the terms of the
6 6 # GNU General Public License version 2 or any later version.
7 7 from node import hex, nullid, short
8 8 from i18n import _
9 9 import peer, changegroup, subrepo, discovery, pushkey, obsolete, repoview
10 10 import changelog, dirstate, filelog, manifest, context, bookmarks, phases
11 11 import lock, transaction, store, encoding, base85
12 12 import scmutil, util, extensions, hook, error, revset
13 13 import match as matchmod
14 14 import merge as mergemod
15 15 import tags as tagsmod
16 16 from lock import release
17 17 import weakref, errno, os, time, inspect
18 18 import branchmap
19 19 propertycache = util.propertycache
20 20 filecache = scmutil.filecache
21 21
22 22 class repofilecache(filecache):
23 23 """All filecache usage on repo are done for logic that should be unfiltered
24 24 """
25 25
26 26 def __get__(self, repo, type=None):
27 27 return super(repofilecache, self).__get__(repo.unfiltered(), type)
28 28 def __set__(self, repo, value):
29 29 return super(repofilecache, self).__set__(repo.unfiltered(), value)
30 30 def __delete__(self, repo):
31 31 return super(repofilecache, self).__delete__(repo.unfiltered())
32 32
33 33 class storecache(repofilecache):
34 34 """filecache for files in the store"""
35 35 def join(self, obj, fname):
36 36 return obj.sjoin(fname)
37 37
38 38 class unfilteredpropertycache(propertycache):
39 39 """propertycache that apply to unfiltered repo only"""
40 40
41 41 def __get__(self, repo, type=None):
42 42 return super(unfilteredpropertycache, self).__get__(repo.unfiltered())
43 43
44 44 class filteredpropertycache(propertycache):
45 45 """propertycache that must take filtering in account"""
46 46
47 47 def cachevalue(self, obj, value):
48 48 object.__setattr__(obj, self.name, value)
49 49
50 50
51 51 def hasunfilteredcache(repo, name):
52 52 """check if a repo has an unfilteredpropertycache value for <name>"""
53 53 return name in vars(repo.unfiltered())
54 54
55 55 def unfilteredmethod(orig):
56 56 """decorate method that always need to be run on unfiltered version"""
57 57 def wrapper(repo, *args, **kwargs):
58 58 return orig(repo.unfiltered(), *args, **kwargs)
59 59 return wrapper
60 60
61 61 MODERNCAPS = set(('lookup', 'branchmap', 'pushkey', 'known', 'getbundle'))
62 62 LEGACYCAPS = MODERNCAPS.union(set(['changegroupsubset']))
63 63
64 64 class localpeer(peer.peerrepository):
65 65 '''peer for a local repo; reflects only the most recent API'''
66 66
67 67 def __init__(self, repo, caps=MODERNCAPS):
68 68 peer.peerrepository.__init__(self)
69 69 self._repo = repo.filtered('served')
70 70 self.ui = repo.ui
71 71 self._caps = repo._restrictcapabilities(caps)
72 72 self.requirements = repo.requirements
73 73 self.supportedformats = repo.supportedformats
74 74
75 75 def close(self):
76 76 self._repo.close()
77 77
78 78 def _capabilities(self):
79 79 return self._caps
80 80
81 81 def local(self):
82 82 return self._repo
83 83
84 84 def canpush(self):
85 85 return True
86 86
87 87 def url(self):
88 88 return self._repo.url()
89 89
90 90 def lookup(self, key):
91 91 return self._repo.lookup(key)
92 92
93 93 def branchmap(self):
94 94 return self._repo.branchmap()
95 95
96 96 def heads(self):
97 97 return self._repo.heads()
98 98
99 99 def known(self, nodes):
100 100 return self._repo.known(nodes)
101 101
102 102 def getbundle(self, source, heads=None, common=None):
103 103 return self._repo.getbundle(source, heads=heads, common=common)
104 104
105 105 # TODO We might want to move the next two calls into legacypeer and add
106 106 # unbundle instead.
107 107
108 108 def lock(self):
109 109 return self._repo.lock()
110 110
111 111 def addchangegroup(self, cg, source, url):
112 112 return self._repo.addchangegroup(cg, source, url)
113 113
114 114 def pushkey(self, namespace, key, old, new):
115 115 return self._repo.pushkey(namespace, key, old, new)
116 116
117 117 def listkeys(self, namespace):
118 118 return self._repo.listkeys(namespace)
119 119
120 120 def debugwireargs(self, one, two, three=None, four=None, five=None):
121 121 '''used to test argument passing over the wire'''
122 122 return "%s %s %s %s %s" % (one, two, three, four, five)
123 123
124 124 class locallegacypeer(localpeer):
125 125 '''peer extension which implements legacy methods too; used for tests with
126 126 restricted capabilities'''
127 127
128 128 def __init__(self, repo):
129 129 localpeer.__init__(self, repo, caps=LEGACYCAPS)
130 130
131 131 def branches(self, nodes):
132 132 return self._repo.branches(nodes)
133 133
134 134 def between(self, pairs):
135 135 return self._repo.between(pairs)
136 136
137 137 def changegroup(self, basenodes, source):
138 138 return self._repo.changegroup(basenodes, source)
139 139
140 140 def changegroupsubset(self, bases, heads, source):
141 141 return self._repo.changegroupsubset(bases, heads, source)
142 142
143 143 class localrepository(object):
144 144
145 145 supportedformats = set(('revlogv1', 'generaldelta'))
146 146 supported = supportedformats | set(('store', 'fncache', 'shared',
147 147 'dotencode'))
148 148 openerreqs = set(('revlogv1', 'generaldelta'))
149 149 requirements = ['revlogv1']
150 150 filtername = None
151 151
152 152 def _baserequirements(self, create):
153 153 return self.requirements[:]
154 154
155 155 def __init__(self, baseui, path=None, create=False):
156 self.wvfs = scmutil.vfs(path, expand=True)
156 self.wvfs = scmutil.vfs(path, expandpath=True, realpath=True)
157 157 self.wopener = self.wvfs
158 158 self.root = self.wvfs.base
159 159 self.path = self.wvfs.join(".hg")
160 160 self.origroot = path
161 161 self.auditor = scmutil.pathauditor(self.root, self._checknested)
162 162 self.vfs = scmutil.vfs(self.path)
163 163 self.opener = self.vfs
164 164 self.baseui = baseui
165 165 self.ui = baseui.copy()
166 166 # A list of callback to shape the phase if no data were found.
167 167 # Callback are in the form: func(repo, roots) --> processed root.
168 168 # This list it to be filled by extension during repo setup
169 169 self._phasedefaults = []
170 170 try:
171 171 self.ui.readconfig(self.join("hgrc"), self.root)
172 172 extensions.loadall(self.ui)
173 173 except IOError:
174 174 pass
175 175
176 176 if not self.vfs.isdir():
177 177 if create:
178 178 if not self.wvfs.exists():
179 179 self.wvfs.makedirs()
180 180 self.vfs.makedir(notindexed=True)
181 181 requirements = self._baserequirements(create)
182 182 if self.ui.configbool('format', 'usestore', True):
183 183 self.vfs.mkdir("store")
184 184 requirements.append("store")
185 185 if self.ui.configbool('format', 'usefncache', True):
186 186 requirements.append("fncache")
187 187 if self.ui.configbool('format', 'dotencode', True):
188 188 requirements.append('dotencode')
189 189 # create an invalid changelog
190 190 self.vfs.append(
191 191 "00changelog.i",
192 192 '\0\0\0\2' # represents revlogv2
193 193 ' dummy changelog to prevent using the old repo layout'
194 194 )
195 195 if self.ui.configbool('format', 'generaldelta', False):
196 196 requirements.append("generaldelta")
197 197 requirements = set(requirements)
198 198 else:
199 199 raise error.RepoError(_("repository %s not found") % path)
200 200 elif create:
201 201 raise error.RepoError(_("repository %s already exists") % path)
202 202 else:
203 203 try:
204 204 requirements = scmutil.readrequires(self.vfs, self.supported)
205 205 except IOError, inst:
206 206 if inst.errno != errno.ENOENT:
207 207 raise
208 208 requirements = set()
209 209
210 210 self.sharedpath = self.path
211 211 try:
212 212 s = os.path.realpath(self.opener.read("sharedpath").rstrip('\n'))
213 213 if not os.path.exists(s):
214 214 raise error.RepoError(
215 215 _('.hg/sharedpath points to nonexistent directory %s') % s)
216 216 self.sharedpath = s
217 217 except IOError, inst:
218 218 if inst.errno != errno.ENOENT:
219 219 raise
220 220
221 221 self.store = store.store(requirements, self.sharedpath, scmutil.vfs)
222 222 self.spath = self.store.path
223 223 self.svfs = self.store.vfs
224 224 self.sopener = self.svfs
225 225 self.sjoin = self.store.join
226 226 self.vfs.createmode = self.store.createmode
227 227 self._applyrequirements(requirements)
228 228 if create:
229 229 self._writerequirements()
230 230
231 231
232 232 self._branchcaches = {}
233 233 self.filterpats = {}
234 234 self._datafilters = {}
235 235 self._transref = self._lockref = self._wlockref = None
236 236
237 237 # A cache for various files under .hg/ that tracks file changes,
238 238 # (used by the filecache decorator)
239 239 #
240 240 # Maps a property name to its util.filecacheentry
241 241 self._filecache = {}
242 242
243 243 # hold sets of revision to be filtered
244 244 # should be cleared when something might have changed the filter value:
245 245 # - new changesets,
246 246 # - phase change,
247 247 # - new obsolescence marker,
248 248 # - working directory parent change,
249 249 # - bookmark changes
250 250 self.filteredrevcache = {}
251 251
252 252 def close(self):
253 253 pass
254 254
255 255 def _restrictcapabilities(self, caps):
256 256 return caps
257 257
258 258 def _applyrequirements(self, requirements):
259 259 self.requirements = requirements
260 260 self.sopener.options = dict((r, 1) for r in requirements
261 261 if r in self.openerreqs)
262 262
263 263 def _writerequirements(self):
264 264 reqfile = self.opener("requires", "w")
265 265 for r in sorted(self.requirements):
266 266 reqfile.write("%s\n" % r)
267 267 reqfile.close()
268 268
269 269 def _checknested(self, path):
270 270 """Determine if path is a legal nested repository."""
271 271 if not path.startswith(self.root):
272 272 return False
273 273 subpath = path[len(self.root) + 1:]
274 274 normsubpath = util.pconvert(subpath)
275 275
276 276 # XXX: Checking against the current working copy is wrong in
277 277 # the sense that it can reject things like
278 278 #
279 279 # $ hg cat -r 10 sub/x.txt
280 280 #
281 281 # if sub/ is no longer a subrepository in the working copy
282 282 # parent revision.
283 283 #
284 284 # However, it can of course also allow things that would have
285 285 # been rejected before, such as the above cat command if sub/
286 286 # is a subrepository now, but was a normal directory before.
287 287 # The old path auditor would have rejected by mistake since it
288 288 # panics when it sees sub/.hg/.
289 289 #
290 290 # All in all, checking against the working copy seems sensible
291 291 # since we want to prevent access to nested repositories on
292 292 # the filesystem *now*.
293 293 ctx = self[None]
294 294 parts = util.splitpath(subpath)
295 295 while parts:
296 296 prefix = '/'.join(parts)
297 297 if prefix in ctx.substate:
298 298 if prefix == normsubpath:
299 299 return True
300 300 else:
301 301 sub = ctx.sub(prefix)
302 302 return sub.checknested(subpath[len(prefix) + 1:])
303 303 else:
304 304 parts.pop()
305 305 return False
306 306
307 307 def peer(self):
308 308 return localpeer(self) # not cached to avoid reference cycle
309 309
310 310 def unfiltered(self):
311 311 """Return unfiltered version of the repository
312 312
313 313 Intended to be overwritten by filtered repo."""
314 314 return self
315 315
316 316 def filtered(self, name):
317 317 """Return a filtered version of a repository"""
318 318 # build a new class with the mixin and the current class
319 319 # (possibly subclass of the repo)
320 320 class proxycls(repoview.repoview, self.unfiltered().__class__):
321 321 pass
322 322 return proxycls(self, name)
323 323
324 324 @repofilecache('bookmarks')
325 325 def _bookmarks(self):
326 326 return bookmarks.bmstore(self)
327 327
328 328 @repofilecache('bookmarks.current')
329 329 def _bookmarkcurrent(self):
330 330 return bookmarks.readcurrent(self)
331 331
332 332 def bookmarkheads(self, bookmark):
333 333 name = bookmark.split('@', 1)[0]
334 334 heads = []
335 335 for mark, n in self._bookmarks.iteritems():
336 336 if mark.split('@', 1)[0] == name:
337 337 heads.append(n)
338 338 return heads
339 339
340 340 @storecache('phaseroots')
341 341 def _phasecache(self):
342 342 return phases.phasecache(self, self._phasedefaults)
343 343
344 344 @storecache('obsstore')
345 345 def obsstore(self):
346 346 store = obsolete.obsstore(self.sopener)
347 347 if store and not obsolete._enabled:
348 348 # message is rare enough to not be translated
349 349 msg = 'obsolete feature not enabled but %i markers found!\n'
350 350 self.ui.warn(msg % len(list(store)))
351 351 return store
352 352
353 353 @storecache('00changelog.i')
354 354 def changelog(self):
355 355 c = changelog.changelog(self.sopener)
356 356 if 'HG_PENDING' in os.environ:
357 357 p = os.environ['HG_PENDING']
358 358 if p.startswith(self.root):
359 359 c.readpending('00changelog.i.a')
360 360 return c
361 361
362 362 @storecache('00manifest.i')
363 363 def manifest(self):
364 364 return manifest.manifest(self.sopener)
365 365
366 366 @repofilecache('dirstate')
367 367 def dirstate(self):
368 368 warned = [0]
369 369 def validate(node):
370 370 try:
371 371 self.changelog.rev(node)
372 372 return node
373 373 except error.LookupError:
374 374 if not warned[0]:
375 375 warned[0] = True
376 376 self.ui.warn(_("warning: ignoring unknown"
377 377 " working parent %s!\n") % short(node))
378 378 return nullid
379 379
380 380 return dirstate.dirstate(self.opener, self.ui, self.root, validate)
381 381
382 382 def __getitem__(self, changeid):
383 383 if changeid is None:
384 384 return context.workingctx(self)
385 385 return context.changectx(self, changeid)
386 386
387 387 def __contains__(self, changeid):
388 388 try:
389 389 return bool(self.lookup(changeid))
390 390 except error.RepoLookupError:
391 391 return False
392 392
393 393 def __nonzero__(self):
394 394 return True
395 395
396 396 def __len__(self):
397 397 return len(self.changelog)
398 398
399 399 def __iter__(self):
400 400 return iter(self.changelog)
401 401
402 402 def revs(self, expr, *args):
403 403 '''Return a list of revisions matching the given revset'''
404 404 expr = revset.formatspec(expr, *args)
405 405 m = revset.match(None, expr)
406 406 return [r for r in m(self, list(self))]
407 407
408 408 def set(self, expr, *args):
409 409 '''
410 410 Yield a context for each matching revision, after doing arg
411 411 replacement via revset.formatspec
412 412 '''
413 413 for r in self.revs(expr, *args):
414 414 yield self[r]
415 415
416 416 def url(self):
417 417 return 'file:' + self.root
418 418
419 419 def hook(self, name, throw=False, **args):
420 420 return hook.hook(self.ui, self, name, throw, **args)
421 421
422 422 @unfilteredmethod
423 423 def _tag(self, names, node, message, local, user, date, extra={}):
424 424 if isinstance(names, str):
425 425 names = (names,)
426 426
427 427 branches = self.branchmap()
428 428 for name in names:
429 429 self.hook('pretag', throw=True, node=hex(node), tag=name,
430 430 local=local)
431 431 if name in branches:
432 432 self.ui.warn(_("warning: tag %s conflicts with existing"
433 433 " branch name\n") % name)
434 434
435 435 def writetags(fp, names, munge, prevtags):
436 436 fp.seek(0, 2)
437 437 if prevtags and prevtags[-1] != '\n':
438 438 fp.write('\n')
439 439 for name in names:
440 440 m = munge and munge(name) or name
441 441 if (self._tagscache.tagtypes and
442 442 name in self._tagscache.tagtypes):
443 443 old = self.tags().get(name, nullid)
444 444 fp.write('%s %s\n' % (hex(old), m))
445 445 fp.write('%s %s\n' % (hex(node), m))
446 446 fp.close()
447 447
448 448 prevtags = ''
449 449 if local:
450 450 try:
451 451 fp = self.opener('localtags', 'r+')
452 452 except IOError:
453 453 fp = self.opener('localtags', 'a')
454 454 else:
455 455 prevtags = fp.read()
456 456
457 457 # local tags are stored in the current charset
458 458 writetags(fp, names, None, prevtags)
459 459 for name in names:
460 460 self.hook('tag', node=hex(node), tag=name, local=local)
461 461 return
462 462
463 463 try:
464 464 fp = self.wfile('.hgtags', 'rb+')
465 465 except IOError, e:
466 466 if e.errno != errno.ENOENT:
467 467 raise
468 468 fp = self.wfile('.hgtags', 'ab')
469 469 else:
470 470 prevtags = fp.read()
471 471
472 472 # committed tags are stored in UTF-8
473 473 writetags(fp, names, encoding.fromlocal, prevtags)
474 474
475 475 fp.close()
476 476
477 477 self.invalidatecaches()
478 478
479 479 if '.hgtags' not in self.dirstate:
480 480 self[None].add(['.hgtags'])
481 481
482 482 m = matchmod.exact(self.root, '', ['.hgtags'])
483 483 tagnode = self.commit(message, user, date, extra=extra, match=m)
484 484
485 485 for name in names:
486 486 self.hook('tag', node=hex(node), tag=name, local=local)
487 487
488 488 return tagnode
489 489
490 490 def tag(self, names, node, message, local, user, date):
491 491 '''tag a revision with one or more symbolic names.
492 492
493 493 names is a list of strings or, when adding a single tag, names may be a
494 494 string.
495 495
496 496 if local is True, the tags are stored in a per-repository file.
497 497 otherwise, they are stored in the .hgtags file, and a new
498 498 changeset is committed with the change.
499 499
500 500 keyword arguments:
501 501
502 502 local: whether to store tags in non-version-controlled file
503 503 (default False)
504 504
505 505 message: commit message to use if committing
506 506
507 507 user: name of user to use if committing
508 508
509 509 date: date tuple to use if committing'''
510 510
511 511 if not local:
512 512 for x in self.status()[:5]:
513 513 if '.hgtags' in x:
514 514 raise util.Abort(_('working copy of .hgtags is changed '
515 515 '(please commit .hgtags manually)'))
516 516
517 517 self.tags() # instantiate the cache
518 518 self._tag(names, node, message, local, user, date)
519 519
520 520 @filteredpropertycache
521 521 def _tagscache(self):
522 522 '''Returns a tagscache object that contains various tags related
523 523 caches.'''
524 524
525 525 # This simplifies its cache management by having one decorated
526 526 # function (this one) and the rest simply fetch things from it.
527 527 class tagscache(object):
528 528 def __init__(self):
529 529 # These two define the set of tags for this repository. tags
530 530 # maps tag name to node; tagtypes maps tag name to 'global' or
531 531 # 'local'. (Global tags are defined by .hgtags across all
532 532 # heads, and local tags are defined in .hg/localtags.)
533 533 # They constitute the in-memory cache of tags.
534 534 self.tags = self.tagtypes = None
535 535
536 536 self.nodetagscache = self.tagslist = None
537 537
538 538 cache = tagscache()
539 539 cache.tags, cache.tagtypes = self._findtags()
540 540
541 541 return cache
542 542
543 543 def tags(self):
544 544 '''return a mapping of tag to node'''
545 545 t = {}
546 546 if self.changelog.filteredrevs:
547 547 tags, tt = self._findtags()
548 548 else:
549 549 tags = self._tagscache.tags
550 550 for k, v in tags.iteritems():
551 551 try:
552 552 # ignore tags to unknown nodes
553 553 self.changelog.rev(v)
554 554 t[k] = v
555 555 except (error.LookupError, ValueError):
556 556 pass
557 557 return t
558 558
559 559 def _findtags(self):
560 560 '''Do the hard work of finding tags. Return a pair of dicts
561 561 (tags, tagtypes) where tags maps tag name to node, and tagtypes
562 562 maps tag name to a string like \'global\' or \'local\'.
563 563 Subclasses or extensions are free to add their own tags, but
564 564 should be aware that the returned dicts will be retained for the
565 565 duration of the localrepo object.'''
566 566
567 567 # XXX what tagtype should subclasses/extensions use? Currently
568 568 # mq and bookmarks add tags, but do not set the tagtype at all.
569 569 # Should each extension invent its own tag type? Should there
570 570 # be one tagtype for all such "virtual" tags? Or is the status
571 571 # quo fine?
572 572
573 573 alltags = {} # map tag name to (node, hist)
574 574 tagtypes = {}
575 575
576 576 tagsmod.findglobaltags(self.ui, self, alltags, tagtypes)
577 577 tagsmod.readlocaltags(self.ui, self, alltags, tagtypes)
578 578
579 579 # Build the return dicts. Have to re-encode tag names because
580 580 # the tags module always uses UTF-8 (in order not to lose info
581 581 # writing to the cache), but the rest of Mercurial wants them in
582 582 # local encoding.
583 583 tags = {}
584 584 for (name, (node, hist)) in alltags.iteritems():
585 585 if node != nullid:
586 586 tags[encoding.tolocal(name)] = node
587 587 tags['tip'] = self.changelog.tip()
588 588 tagtypes = dict([(encoding.tolocal(name), value)
589 589 for (name, value) in tagtypes.iteritems()])
590 590 return (tags, tagtypes)
591 591
592 592 def tagtype(self, tagname):
593 593 '''
594 594 return the type of the given tag. result can be:
595 595
596 596 'local' : a local tag
597 597 'global' : a global tag
598 598 None : tag does not exist
599 599 '''
600 600
601 601 return self._tagscache.tagtypes.get(tagname)
602 602
603 603 def tagslist(self):
604 604 '''return a list of tags ordered by revision'''
605 605 if not self._tagscache.tagslist:
606 606 l = []
607 607 for t, n in self.tags().iteritems():
608 608 r = self.changelog.rev(n)
609 609 l.append((r, t, n))
610 610 self._tagscache.tagslist = [(t, n) for r, t, n in sorted(l)]
611 611
612 612 return self._tagscache.tagslist
613 613
614 614 def nodetags(self, node):
615 615 '''return the tags associated with a node'''
616 616 if not self._tagscache.nodetagscache:
617 617 nodetagscache = {}
618 618 for t, n in self._tagscache.tags.iteritems():
619 619 nodetagscache.setdefault(n, []).append(t)
620 620 for tags in nodetagscache.itervalues():
621 621 tags.sort()
622 622 self._tagscache.nodetagscache = nodetagscache
623 623 return self._tagscache.nodetagscache.get(node, [])
624 624
625 625 def nodebookmarks(self, node):
626 626 marks = []
627 627 for bookmark, n in self._bookmarks.iteritems():
628 628 if n == node:
629 629 marks.append(bookmark)
630 630 return sorted(marks)
631 631
632 632 def branchmap(self):
633 633 '''returns a dictionary {branch: [branchheads]}'''
634 634 branchmap.updatecache(self)
635 635 return self._branchcaches[self.filtername]
636 636
637 637
638 638 def _branchtip(self, heads):
639 639 '''return the tipmost branch head in heads'''
640 640 tip = heads[-1]
641 641 for h in reversed(heads):
642 642 if not self[h].closesbranch():
643 643 tip = h
644 644 break
645 645 return tip
646 646
647 647 def branchtip(self, branch):
648 648 '''return the tip node for a given branch'''
649 649 if branch not in self.branchmap():
650 650 raise error.RepoLookupError(_("unknown branch '%s'") % branch)
651 651 return self._branchtip(self.branchmap()[branch])
652 652
653 653 def branchtags(self):
654 654 '''return a dict where branch names map to the tipmost head of
655 655 the branch, open heads come before closed'''
656 656 bt = {}
657 657 for bn, heads in self.branchmap().iteritems():
658 658 bt[bn] = self._branchtip(heads)
659 659 return bt
660 660
661 661 def lookup(self, key):
662 662 return self[key].node()
663 663
664 664 def lookupbranch(self, key, remote=None):
665 665 repo = remote or self
666 666 if key in repo.branchmap():
667 667 return key
668 668
669 669 repo = (remote and remote.local()) and remote or self
670 670 return repo[key].branch()
671 671
672 672 def known(self, nodes):
673 673 nm = self.changelog.nodemap
674 674 pc = self._phasecache
675 675 result = []
676 676 for n in nodes:
677 677 r = nm.get(n)
678 678 resp = not (r is None or pc.phase(self, r) >= phases.secret)
679 679 result.append(resp)
680 680 return result
681 681
682 682 def local(self):
683 683 return self
684 684
685 685 def cancopy(self):
686 686 return self.local() # so statichttprepo's override of local() works
687 687
688 688 def join(self, f):
689 689 return os.path.join(self.path, f)
690 690
691 691 def wjoin(self, f):
692 692 return os.path.join(self.root, f)
693 693
694 694 def file(self, f):
695 695 if f[0] == '/':
696 696 f = f[1:]
697 697 return filelog.filelog(self.sopener, f)
698 698
699 699 def changectx(self, changeid):
700 700 return self[changeid]
701 701
702 702 def parents(self, changeid=None):
703 703 '''get list of changectxs for parents of changeid'''
704 704 return self[changeid].parents()
705 705
706 706 def setparents(self, p1, p2=nullid):
707 707 copies = self.dirstate.setparents(p1, p2)
708 708 pctx = self[p1]
709 709 if copies:
710 710 # Adjust copy records, the dirstate cannot do it, it
711 711 # requires access to parents manifests. Preserve them
712 712 # only for entries added to first parent.
713 713 for f in copies:
714 714 if f not in pctx and copies[f] in pctx:
715 715 self.dirstate.copy(copies[f], f)
716 716 if p2 == nullid:
717 717 for f, s in sorted(self.dirstate.copies().items()):
718 718 if f not in pctx and s not in pctx:
719 719 self.dirstate.copy(None, f)
720 720
721 721 def filectx(self, path, changeid=None, fileid=None):
722 722 """changeid can be a changeset revision, node, or tag.
723 723 fileid can be a file revision or node."""
724 724 return context.filectx(self, path, changeid, fileid)
725 725
726 726 def getcwd(self):
727 727 return self.dirstate.getcwd()
728 728
729 729 def pathto(self, f, cwd=None):
730 730 return self.dirstate.pathto(f, cwd)
731 731
732 732 def wfile(self, f, mode='r'):
733 733 return self.wopener(f, mode)
734 734
735 735 def _link(self, f):
736 736 return os.path.islink(self.wjoin(f))
737 737
738 738 def _loadfilter(self, filter):
739 739 if filter not in self.filterpats:
740 740 l = []
741 741 for pat, cmd in self.ui.configitems(filter):
742 742 if cmd == '!':
743 743 continue
744 744 mf = matchmod.match(self.root, '', [pat])
745 745 fn = None
746 746 params = cmd
747 747 for name, filterfn in self._datafilters.iteritems():
748 748 if cmd.startswith(name):
749 749 fn = filterfn
750 750 params = cmd[len(name):].lstrip()
751 751 break
752 752 if not fn:
753 753 fn = lambda s, c, **kwargs: util.filter(s, c)
754 754 # Wrap old filters not supporting keyword arguments
755 755 if not inspect.getargspec(fn)[2]:
756 756 oldfn = fn
757 757 fn = lambda s, c, **kwargs: oldfn(s, c)
758 758 l.append((mf, fn, params))
759 759 self.filterpats[filter] = l
760 760 return self.filterpats[filter]
761 761
762 762 def _filter(self, filterpats, filename, data):
763 763 for mf, fn, cmd in filterpats:
764 764 if mf(filename):
765 765 self.ui.debug("filtering %s through %s\n" % (filename, cmd))
766 766 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
767 767 break
768 768
769 769 return data
770 770
771 771 @unfilteredpropertycache
772 772 def _encodefilterpats(self):
773 773 return self._loadfilter('encode')
774 774
775 775 @unfilteredpropertycache
776 776 def _decodefilterpats(self):
777 777 return self._loadfilter('decode')
778 778
779 779 def adddatafilter(self, name, filter):
780 780 self._datafilters[name] = filter
781 781
782 782 def wread(self, filename):
783 783 if self._link(filename):
784 784 data = os.readlink(self.wjoin(filename))
785 785 else:
786 786 data = self.wopener.read(filename)
787 787 return self._filter(self._encodefilterpats, filename, data)
788 788
789 789 def wwrite(self, filename, data, flags):
790 790 data = self._filter(self._decodefilterpats, filename, data)
791 791 if 'l' in flags:
792 792 self.wopener.symlink(data, filename)
793 793 else:
794 794 self.wopener.write(filename, data)
795 795 if 'x' in flags:
796 796 util.setflags(self.wjoin(filename), False, True)
797 797
798 798 def wwritedata(self, filename, data):
799 799 return self._filter(self._decodefilterpats, filename, data)
800 800
801 801 def transaction(self, desc):
802 802 tr = self._transref and self._transref() or None
803 803 if tr and tr.running():
804 804 return tr.nest()
805 805
806 806 # abort here if the journal already exists
807 807 if os.path.exists(self.sjoin("journal")):
808 808 raise error.RepoError(
809 809 _("abandoned transaction found - run hg recover"))
810 810
811 811 self._writejournal(desc)
812 812 renames = [(x, undoname(x)) for x in self._journalfiles()]
813 813
814 814 tr = transaction.transaction(self.ui.warn, self.sopener,
815 815 self.sjoin("journal"),
816 816 aftertrans(renames),
817 817 self.store.createmode)
818 818 self._transref = weakref.ref(tr)
819 819 return tr
820 820
821 821 def _journalfiles(self):
822 822 return (self.sjoin('journal'), self.join('journal.dirstate'),
823 823 self.join('journal.branch'), self.join('journal.desc'),
824 824 self.join('journal.bookmarks'),
825 825 self.sjoin('journal.phaseroots'))
826 826
827 827 def undofiles(self):
828 828 return [undoname(x) for x in self._journalfiles()]
829 829
830 830 def _writejournal(self, desc):
831 831 self.opener.write("journal.dirstate",
832 832 self.opener.tryread("dirstate"))
833 833 self.opener.write("journal.branch",
834 834 encoding.fromlocal(self.dirstate.branch()))
835 835 self.opener.write("journal.desc",
836 836 "%d\n%s\n" % (len(self), desc))
837 837 self.opener.write("journal.bookmarks",
838 838 self.opener.tryread("bookmarks"))
839 839 self.sopener.write("journal.phaseroots",
840 840 self.sopener.tryread("phaseroots"))
841 841
842 842 def recover(self):
843 843 lock = self.lock()
844 844 try:
845 845 if os.path.exists(self.sjoin("journal")):
846 846 self.ui.status(_("rolling back interrupted transaction\n"))
847 847 transaction.rollback(self.sopener, self.sjoin("journal"),
848 848 self.ui.warn)
849 849 self.invalidate()
850 850 return True
851 851 else:
852 852 self.ui.warn(_("no interrupted transaction available\n"))
853 853 return False
854 854 finally:
855 855 lock.release()
856 856
857 857 def rollback(self, dryrun=False, force=False):
858 858 wlock = lock = None
859 859 try:
860 860 wlock = self.wlock()
861 861 lock = self.lock()
862 862 if os.path.exists(self.sjoin("undo")):
863 863 return self._rollback(dryrun, force)
864 864 else:
865 865 self.ui.warn(_("no rollback information available\n"))
866 866 return 1
867 867 finally:
868 868 release(lock, wlock)
869 869
870 870 @unfilteredmethod # Until we get smarter cache management
871 871 def _rollback(self, dryrun, force):
872 872 ui = self.ui
873 873 try:
874 874 args = self.opener.read('undo.desc').splitlines()
875 875 (oldlen, desc, detail) = (int(args[0]), args[1], None)
876 876 if len(args) >= 3:
877 877 detail = args[2]
878 878 oldtip = oldlen - 1
879 879
880 880 if detail and ui.verbose:
881 881 msg = (_('repository tip rolled back to revision %s'
882 882 ' (undo %s: %s)\n')
883 883 % (oldtip, desc, detail))
884 884 else:
885 885 msg = (_('repository tip rolled back to revision %s'
886 886 ' (undo %s)\n')
887 887 % (oldtip, desc))
888 888 except IOError:
889 889 msg = _('rolling back unknown transaction\n')
890 890 desc = None
891 891
892 892 if not force and self['.'] != self['tip'] and desc == 'commit':
893 893 raise util.Abort(
894 894 _('rollback of last commit while not checked out '
895 895 'may lose data'), hint=_('use -f to force'))
896 896
897 897 ui.status(msg)
898 898 if dryrun:
899 899 return 0
900 900
901 901 parents = self.dirstate.parents()
902 902 self.destroying()
903 903 transaction.rollback(self.sopener, self.sjoin('undo'), ui.warn)
904 904 if os.path.exists(self.join('undo.bookmarks')):
905 905 util.rename(self.join('undo.bookmarks'),
906 906 self.join('bookmarks'))
907 907 if os.path.exists(self.sjoin('undo.phaseroots')):
908 908 util.rename(self.sjoin('undo.phaseroots'),
909 909 self.sjoin('phaseroots'))
910 910 self.invalidate()
911 911
912 912 parentgone = (parents[0] not in self.changelog.nodemap or
913 913 parents[1] not in self.changelog.nodemap)
914 914 if parentgone:
915 915 util.rename(self.join('undo.dirstate'), self.join('dirstate'))
916 916 try:
917 917 branch = self.opener.read('undo.branch')
918 918 self.dirstate.setbranch(encoding.tolocal(branch))
919 919 except IOError:
920 920 ui.warn(_('named branch could not be reset: '
921 921 'current branch is still \'%s\'\n')
922 922 % self.dirstate.branch())
923 923
924 924 self.dirstate.invalidate()
925 925 parents = tuple([p.rev() for p in self.parents()])
926 926 if len(parents) > 1:
927 927 ui.status(_('working directory now based on '
928 928 'revisions %d and %d\n') % parents)
929 929 else:
930 930 ui.status(_('working directory now based on '
931 931 'revision %d\n') % parents)
932 932 # TODO: if we know which new heads may result from this rollback, pass
933 933 # them to destroy(), which will prevent the branchhead cache from being
934 934 # invalidated.
935 935 self.destroyed()
936 936 return 0
937 937
938 938 def invalidatecaches(self):
939 939
940 940 if '_tagscache' in vars(self):
941 941 # can't use delattr on proxy
942 942 del self.__dict__['_tagscache']
943 943
944 944 self.unfiltered()._branchcaches.clear()
945 945 self.invalidatevolatilesets()
946 946
947 947 def invalidatevolatilesets(self):
948 948 self.filteredrevcache.clear()
949 949 obsolete.clearobscaches(self)
950 950
951 951 def invalidatedirstate(self):
952 952 '''Invalidates the dirstate, causing the next call to dirstate
953 953 to check if it was modified since the last time it was read,
954 954 rereading it if it has.
955 955
956 956 This is different to dirstate.invalidate() that it doesn't always
957 957 rereads the dirstate. Use dirstate.invalidate() if you want to
958 958 explicitly read the dirstate again (i.e. restoring it to a previous
959 959 known good state).'''
960 960 if hasunfilteredcache(self, 'dirstate'):
961 961 for k in self.dirstate._filecache:
962 962 try:
963 963 delattr(self.dirstate, k)
964 964 except AttributeError:
965 965 pass
966 966 delattr(self.unfiltered(), 'dirstate')
967 967
968 968 def invalidate(self):
969 969 unfiltered = self.unfiltered() # all file caches are stored unfiltered
970 970 for k in self._filecache:
971 971 # dirstate is invalidated separately in invalidatedirstate()
972 972 if k == 'dirstate':
973 973 continue
974 974
975 975 try:
976 976 delattr(unfiltered, k)
977 977 except AttributeError:
978 978 pass
979 979 self.invalidatecaches()
980 980
981 981 def _lock(self, lockname, wait, releasefn, acquirefn, desc):
982 982 try:
983 983 l = lock.lock(lockname, 0, releasefn, desc=desc)
984 984 except error.LockHeld, inst:
985 985 if not wait:
986 986 raise
987 987 self.ui.warn(_("waiting for lock on %s held by %r\n") %
988 988 (desc, inst.locker))
989 989 # default to 600 seconds timeout
990 990 l = lock.lock(lockname, int(self.ui.config("ui", "timeout", "600")),
991 991 releasefn, desc=desc)
992 992 if acquirefn:
993 993 acquirefn()
994 994 return l
995 995
996 996 def _afterlock(self, callback):
997 997 """add a callback to the current repository lock.
998 998
999 999 The callback will be executed on lock release."""
1000 1000 l = self._lockref and self._lockref()
1001 1001 if l:
1002 1002 l.postrelease.append(callback)
1003 1003 else:
1004 1004 callback()
1005 1005
1006 1006 def lock(self, wait=True):
1007 1007 '''Lock the repository store (.hg/store) and return a weak reference
1008 1008 to the lock. Use this before modifying the store (e.g. committing or
1009 1009 stripping). If you are opening a transaction, get a lock as well.)'''
1010 1010 l = self._lockref and self._lockref()
1011 1011 if l is not None and l.held:
1012 1012 l.lock()
1013 1013 return l
1014 1014
1015 1015 def unlock():
1016 1016 self.store.write()
1017 1017 if hasunfilteredcache(self, '_phasecache'):
1018 1018 self._phasecache.write()
1019 1019 for k, ce in self._filecache.items():
1020 1020 if k == 'dirstate' or k not in self.__dict__:
1021 1021 continue
1022 1022 ce.refresh()
1023 1023
1024 1024 l = self._lock(self.sjoin("lock"), wait, unlock,
1025 1025 self.invalidate, _('repository %s') % self.origroot)
1026 1026 self._lockref = weakref.ref(l)
1027 1027 return l
1028 1028
1029 1029 def wlock(self, wait=True):
1030 1030 '''Lock the non-store parts of the repository (everything under
1031 1031 .hg except .hg/store) and return a weak reference to the lock.
1032 1032 Use this before modifying files in .hg.'''
1033 1033 l = self._wlockref and self._wlockref()
1034 1034 if l is not None and l.held:
1035 1035 l.lock()
1036 1036 return l
1037 1037
1038 1038 def unlock():
1039 1039 self.dirstate.write()
1040 1040 self._filecache['dirstate'].refresh()
1041 1041
1042 1042 l = self._lock(self.join("wlock"), wait, unlock,
1043 1043 self.invalidatedirstate, _('working directory of %s') %
1044 1044 self.origroot)
1045 1045 self._wlockref = weakref.ref(l)
1046 1046 return l
1047 1047
1048 1048 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
1049 1049 """
1050 1050 commit an individual file as part of a larger transaction
1051 1051 """
1052 1052
1053 1053 fname = fctx.path()
1054 1054 text = fctx.data()
1055 1055 flog = self.file(fname)
1056 1056 fparent1 = manifest1.get(fname, nullid)
1057 1057 fparent2 = fparent2o = manifest2.get(fname, nullid)
1058 1058
1059 1059 meta = {}
1060 1060 copy = fctx.renamed()
1061 1061 if copy and copy[0] != fname:
1062 1062 # Mark the new revision of this file as a copy of another
1063 1063 # file. This copy data will effectively act as a parent
1064 1064 # of this new revision. If this is a merge, the first
1065 1065 # parent will be the nullid (meaning "look up the copy data")
1066 1066 # and the second one will be the other parent. For example:
1067 1067 #
1068 1068 # 0 --- 1 --- 3 rev1 changes file foo
1069 1069 # \ / rev2 renames foo to bar and changes it
1070 1070 # \- 2 -/ rev3 should have bar with all changes and
1071 1071 # should record that bar descends from
1072 1072 # bar in rev2 and foo in rev1
1073 1073 #
1074 1074 # this allows this merge to succeed:
1075 1075 #
1076 1076 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
1077 1077 # \ / merging rev3 and rev4 should use bar@rev2
1078 1078 # \- 2 --- 4 as the merge base
1079 1079 #
1080 1080
1081 1081 cfname = copy[0]
1082 1082 crev = manifest1.get(cfname)
1083 1083 newfparent = fparent2
1084 1084
1085 1085 if manifest2: # branch merge
1086 1086 if fparent2 == nullid or crev is None: # copied on remote side
1087 1087 if cfname in manifest2:
1088 1088 crev = manifest2[cfname]
1089 1089 newfparent = fparent1
1090 1090
1091 1091 # find source in nearest ancestor if we've lost track
1092 1092 if not crev:
1093 1093 self.ui.debug(" %s: searching for copy revision for %s\n" %
1094 1094 (fname, cfname))
1095 1095 for ancestor in self[None].ancestors():
1096 1096 if cfname in ancestor:
1097 1097 crev = ancestor[cfname].filenode()
1098 1098 break
1099 1099
1100 1100 if crev:
1101 1101 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
1102 1102 meta["copy"] = cfname
1103 1103 meta["copyrev"] = hex(crev)
1104 1104 fparent1, fparent2 = nullid, newfparent
1105 1105 else:
1106 1106 self.ui.warn(_("warning: can't find ancestor for '%s' "
1107 1107 "copied from '%s'!\n") % (fname, cfname))
1108 1108
1109 1109 elif fparent2 != nullid:
1110 1110 # is one parent an ancestor of the other?
1111 1111 fparentancestor = flog.ancestor(fparent1, fparent2)
1112 1112 if fparentancestor == fparent1:
1113 1113 fparent1, fparent2 = fparent2, nullid
1114 1114 elif fparentancestor == fparent2:
1115 1115 fparent2 = nullid
1116 1116
1117 1117 # is the file changed?
1118 1118 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
1119 1119 changelist.append(fname)
1120 1120 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
1121 1121
1122 1122 # are just the flags changed during merge?
1123 1123 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
1124 1124 changelist.append(fname)
1125 1125
1126 1126 return fparent1
1127 1127
1128 1128 @unfilteredmethod
1129 1129 def commit(self, text="", user=None, date=None, match=None, force=False,
1130 1130 editor=False, extra={}):
1131 1131 """Add a new revision to current repository.
1132 1132
1133 1133 Revision information is gathered from the working directory,
1134 1134 match can be used to filter the committed files. If editor is
1135 1135 supplied, it is called to get a commit message.
1136 1136 """
1137 1137
1138 1138 def fail(f, msg):
1139 1139 raise util.Abort('%s: %s' % (f, msg))
1140 1140
1141 1141 if not match:
1142 1142 match = matchmod.always(self.root, '')
1143 1143
1144 1144 if not force:
1145 1145 vdirs = []
1146 1146 match.dir = vdirs.append
1147 1147 match.bad = fail
1148 1148
1149 1149 wlock = self.wlock()
1150 1150 try:
1151 1151 wctx = self[None]
1152 1152 merge = len(wctx.parents()) > 1
1153 1153
1154 1154 if (not force and merge and match and
1155 1155 (match.files() or match.anypats())):
1156 1156 raise util.Abort(_('cannot partially commit a merge '
1157 1157 '(do not specify files or patterns)'))
1158 1158
1159 1159 changes = self.status(match=match, clean=force)
1160 1160 if force:
1161 1161 changes[0].extend(changes[6]) # mq may commit unchanged files
1162 1162
1163 1163 # check subrepos
1164 1164 subs = []
1165 1165 commitsubs = set()
1166 1166 newstate = wctx.substate.copy()
1167 1167 # only manage subrepos and .hgsubstate if .hgsub is present
1168 1168 if '.hgsub' in wctx:
1169 1169 # we'll decide whether to track this ourselves, thanks
1170 1170 if '.hgsubstate' in changes[0]:
1171 1171 changes[0].remove('.hgsubstate')
1172 1172 if '.hgsubstate' in changes[2]:
1173 1173 changes[2].remove('.hgsubstate')
1174 1174
1175 1175 # compare current state to last committed state
1176 1176 # build new substate based on last committed state
1177 1177 oldstate = wctx.p1().substate
1178 1178 for s in sorted(newstate.keys()):
1179 1179 if not match(s):
1180 1180 # ignore working copy, use old state if present
1181 1181 if s in oldstate:
1182 1182 newstate[s] = oldstate[s]
1183 1183 continue
1184 1184 if not force:
1185 1185 raise util.Abort(
1186 1186 _("commit with new subrepo %s excluded") % s)
1187 1187 if wctx.sub(s).dirty(True):
1188 1188 if not self.ui.configbool('ui', 'commitsubrepos'):
1189 1189 raise util.Abort(
1190 1190 _("uncommitted changes in subrepo %s") % s,
1191 1191 hint=_("use --subrepos for recursive commit"))
1192 1192 subs.append(s)
1193 1193 commitsubs.add(s)
1194 1194 else:
1195 1195 bs = wctx.sub(s).basestate()
1196 1196 newstate[s] = (newstate[s][0], bs, newstate[s][2])
1197 1197 if oldstate.get(s, (None, None, None))[1] != bs:
1198 1198 subs.append(s)
1199 1199
1200 1200 # check for removed subrepos
1201 1201 for p in wctx.parents():
1202 1202 r = [s for s in p.substate if s not in newstate]
1203 1203 subs += [s for s in r if match(s)]
1204 1204 if subs:
1205 1205 if (not match('.hgsub') and
1206 1206 '.hgsub' in (wctx.modified() + wctx.added())):
1207 1207 raise util.Abort(
1208 1208 _("can't commit subrepos without .hgsub"))
1209 1209 changes[0].insert(0, '.hgsubstate')
1210 1210
1211 1211 elif '.hgsub' in changes[2]:
1212 1212 # clean up .hgsubstate when .hgsub is removed
1213 1213 if ('.hgsubstate' in wctx and
1214 1214 '.hgsubstate' not in changes[0] + changes[1] + changes[2]):
1215 1215 changes[2].insert(0, '.hgsubstate')
1216 1216
1217 1217 # make sure all explicit patterns are matched
1218 1218 if not force and match.files():
1219 1219 matched = set(changes[0] + changes[1] + changes[2])
1220 1220
1221 1221 for f in match.files():
1222 1222 f = self.dirstate.normalize(f)
1223 1223 if f == '.' or f in matched or f in wctx.substate:
1224 1224 continue
1225 1225 if f in changes[3]: # missing
1226 1226 fail(f, _('file not found!'))
1227 1227 if f in vdirs: # visited directory
1228 1228 d = f + '/'
1229 1229 for mf in matched:
1230 1230 if mf.startswith(d):
1231 1231 break
1232 1232 else:
1233 1233 fail(f, _("no match under directory!"))
1234 1234 elif f not in self.dirstate:
1235 1235 fail(f, _("file not tracked!"))
1236 1236
1237 1237 cctx = context.workingctx(self, text, user, date, extra, changes)
1238 1238
1239 1239 if (not force and not extra.get("close") and not merge
1240 1240 and not cctx.files()
1241 1241 and wctx.branch() == wctx.p1().branch()):
1242 1242 return None
1243 1243
1244 1244 if merge and cctx.deleted():
1245 1245 raise util.Abort(_("cannot commit merge with missing files"))
1246 1246
1247 1247 ms = mergemod.mergestate(self)
1248 1248 for f in changes[0]:
1249 1249 if f in ms and ms[f] == 'u':
1250 1250 raise util.Abort(_("unresolved merge conflicts "
1251 1251 "(see hg help resolve)"))
1252 1252
1253 1253 if editor:
1254 1254 cctx._text = editor(self, cctx, subs)
1255 1255 edited = (text != cctx._text)
1256 1256
1257 1257 # commit subs and write new state
1258 1258 if subs:
1259 1259 for s in sorted(commitsubs):
1260 1260 sub = wctx.sub(s)
1261 1261 self.ui.status(_('committing subrepository %s\n') %
1262 1262 subrepo.subrelpath(sub))
1263 1263 sr = sub.commit(cctx._text, user, date)
1264 1264 newstate[s] = (newstate[s][0], sr)
1265 1265 subrepo.writestate(self, newstate)
1266 1266
1267 1267 # Save commit message in case this transaction gets rolled back
1268 1268 # (e.g. by a pretxncommit hook). Leave the content alone on
1269 1269 # the assumption that the user will use the same editor again.
1270 1270 msgfn = self.savecommitmessage(cctx._text)
1271 1271
1272 1272 p1, p2 = self.dirstate.parents()
1273 1273 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1274 1274 try:
1275 1275 self.hook("precommit", throw=True, parent1=hookp1,
1276 1276 parent2=hookp2)
1277 1277 ret = self.commitctx(cctx, True)
1278 1278 except: # re-raises
1279 1279 if edited:
1280 1280 self.ui.write(
1281 1281 _('note: commit message saved in %s\n') % msgfn)
1282 1282 raise
1283 1283
1284 1284 # update bookmarks, dirstate and mergestate
1285 1285 bookmarks.update(self, [p1, p2], ret)
1286 1286 cctx.markcommitted(ret)
1287 1287 ms.reset()
1288 1288 finally:
1289 1289 wlock.release()
1290 1290
1291 1291 def commithook(node=hex(ret), parent1=hookp1, parent2=hookp2):
1292 1292 self.hook("commit", node=node, parent1=parent1, parent2=parent2)
1293 1293 self._afterlock(commithook)
1294 1294 return ret
1295 1295
1296 1296 @unfilteredmethod
1297 1297 def commitctx(self, ctx, error=False):
1298 1298 """Add a new revision to current repository.
1299 1299 Revision information is passed via the context argument.
1300 1300 """
1301 1301
1302 1302 tr = lock = None
1303 1303 removed = list(ctx.removed())
1304 1304 p1, p2 = ctx.p1(), ctx.p2()
1305 1305 user = ctx.user()
1306 1306
1307 1307 lock = self.lock()
1308 1308 try:
1309 1309 tr = self.transaction("commit")
1310 1310 trp = weakref.proxy(tr)
1311 1311
1312 1312 if ctx.files():
1313 1313 m1 = p1.manifest().copy()
1314 1314 m2 = p2.manifest()
1315 1315
1316 1316 # check in files
1317 1317 new = {}
1318 1318 changed = []
1319 1319 linkrev = len(self)
1320 1320 for f in sorted(ctx.modified() + ctx.added()):
1321 1321 self.ui.note(f + "\n")
1322 1322 try:
1323 1323 fctx = ctx[f]
1324 1324 new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
1325 1325 changed)
1326 1326 m1.set(f, fctx.flags())
1327 1327 except OSError, inst:
1328 1328 self.ui.warn(_("trouble committing %s!\n") % f)
1329 1329 raise
1330 1330 except IOError, inst:
1331 1331 errcode = getattr(inst, 'errno', errno.ENOENT)
1332 1332 if error or errcode and errcode != errno.ENOENT:
1333 1333 self.ui.warn(_("trouble committing %s!\n") % f)
1334 1334 raise
1335 1335 else:
1336 1336 removed.append(f)
1337 1337
1338 1338 # update manifest
1339 1339 m1.update(new)
1340 1340 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1341 1341 drop = [f for f in removed if f in m1]
1342 1342 for f in drop:
1343 1343 del m1[f]
1344 1344 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1345 1345 p2.manifestnode(), (new, drop))
1346 1346 files = changed + removed
1347 1347 else:
1348 1348 mn = p1.manifestnode()
1349 1349 files = []
1350 1350
1351 1351 # update changelog
1352 1352 self.changelog.delayupdate()
1353 1353 n = self.changelog.add(mn, files, ctx.description(),
1354 1354 trp, p1.node(), p2.node(),
1355 1355 user, ctx.date(), ctx.extra().copy())
1356 1356 p = lambda: self.changelog.writepending() and self.root or ""
1357 1357 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1358 1358 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1359 1359 parent2=xp2, pending=p)
1360 1360 self.changelog.finalize(trp)
1361 1361 # set the new commit is proper phase
1362 1362 targetphase = phases.newcommitphase(self.ui)
1363 1363 if targetphase:
1364 1364 # retract boundary do not alter parent changeset.
1365 1365 # if a parent have higher the resulting phase will
1366 1366 # be compliant anyway
1367 1367 #
1368 1368 # if minimal phase was 0 we don't need to retract anything
1369 1369 phases.retractboundary(self, targetphase, [n])
1370 1370 tr.close()
1371 1371 branchmap.updatecache(self.filtered('served'))
1372 1372 return n
1373 1373 finally:
1374 1374 if tr:
1375 1375 tr.release()
1376 1376 lock.release()
1377 1377
1378 1378 @unfilteredmethod
1379 1379 def destroying(self):
1380 1380 '''Inform the repository that nodes are about to be destroyed.
1381 1381 Intended for use by strip and rollback, so there's a common
1382 1382 place for anything that has to be done before destroying history.
1383 1383
1384 1384 This is mostly useful for saving state that is in memory and waiting
1385 1385 to be flushed when the current lock is released. Because a call to
1386 1386 destroyed is imminent, the repo will be invalidated causing those
1387 1387 changes to stay in memory (waiting for the next unlock), or vanish
1388 1388 completely.
1389 1389 '''
1390 1390 # When using the same lock to commit and strip, the phasecache is left
1391 1391 # dirty after committing. Then when we strip, the repo is invalidated,
1392 1392 # causing those changes to disappear.
1393 1393 if '_phasecache' in vars(self):
1394 1394 self._phasecache.write()
1395 1395
1396 1396 @unfilteredmethod
1397 1397 def destroyed(self):
1398 1398 '''Inform the repository that nodes have been destroyed.
1399 1399 Intended for use by strip and rollback, so there's a common
1400 1400 place for anything that has to be done after destroying history.
1401 1401 '''
1402 1402 # When one tries to:
1403 1403 # 1) destroy nodes thus calling this method (e.g. strip)
1404 1404 # 2) use phasecache somewhere (e.g. commit)
1405 1405 #
1406 1406 # then 2) will fail because the phasecache contains nodes that were
1407 1407 # removed. We can either remove phasecache from the filecache,
1408 1408 # causing it to reload next time it is accessed, or simply filter
1409 1409 # the removed nodes now and write the updated cache.
1410 1410 self._phasecache.filterunknown(self)
1411 1411 self._phasecache.write()
1412 1412
1413 1413 # update the 'served' branch cache to help read only server process
1414 1414 # Thanks to branchcache collaboration this is done from the nearest
1415 1415 # filtered subset and it is expected to be fast.
1416 1416 branchmap.updatecache(self.filtered('served'))
1417 1417
1418 1418 # Ensure the persistent tag cache is updated. Doing it now
1419 1419 # means that the tag cache only has to worry about destroyed
1420 1420 # heads immediately after a strip/rollback. That in turn
1421 1421 # guarantees that "cachetip == currenttip" (comparing both rev
1422 1422 # and node) always means no nodes have been added or destroyed.
1423 1423
1424 1424 # XXX this is suboptimal when qrefresh'ing: we strip the current
1425 1425 # head, refresh the tag cache, then immediately add a new head.
1426 1426 # But I think doing it this way is necessary for the "instant
1427 1427 # tag cache retrieval" case to work.
1428 1428 self.invalidate()
1429 1429
1430 1430 def walk(self, match, node=None):
1431 1431 '''
1432 1432 walk recursively through the directory tree or a given
1433 1433 changeset, finding all files matched by the match
1434 1434 function
1435 1435 '''
1436 1436 return self[node].walk(match)
1437 1437
1438 1438 def status(self, node1='.', node2=None, match=None,
1439 1439 ignored=False, clean=False, unknown=False,
1440 1440 listsubrepos=False):
1441 1441 """return status of files between two nodes or node and working
1442 1442 directory.
1443 1443
1444 1444 If node1 is None, use the first dirstate parent instead.
1445 1445 If node2 is None, compare node1 with working directory.
1446 1446 """
1447 1447
1448 1448 def mfmatches(ctx):
1449 1449 mf = ctx.manifest().copy()
1450 1450 if match.always():
1451 1451 return mf
1452 1452 for fn in mf.keys():
1453 1453 if not match(fn):
1454 1454 del mf[fn]
1455 1455 return mf
1456 1456
1457 1457 if isinstance(node1, context.changectx):
1458 1458 ctx1 = node1
1459 1459 else:
1460 1460 ctx1 = self[node1]
1461 1461 if isinstance(node2, context.changectx):
1462 1462 ctx2 = node2
1463 1463 else:
1464 1464 ctx2 = self[node2]
1465 1465
1466 1466 working = ctx2.rev() is None
1467 1467 parentworking = working and ctx1 == self['.']
1468 1468 match = match or matchmod.always(self.root, self.getcwd())
1469 1469 listignored, listclean, listunknown = ignored, clean, unknown
1470 1470
1471 1471 # load earliest manifest first for caching reasons
1472 1472 if not working and ctx2.rev() < ctx1.rev():
1473 1473 ctx2.manifest()
1474 1474
1475 1475 if not parentworking:
1476 1476 def bad(f, msg):
1477 1477 # 'f' may be a directory pattern from 'match.files()',
1478 1478 # so 'f not in ctx1' is not enough
1479 1479 if f not in ctx1 and f not in ctx1.dirs():
1480 1480 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1481 1481 match.bad = bad
1482 1482
1483 1483 if working: # we need to scan the working dir
1484 1484 subrepos = []
1485 1485 if '.hgsub' in self.dirstate:
1486 1486 subrepos = sorted(ctx2.substate)
1487 1487 s = self.dirstate.status(match, subrepos, listignored,
1488 1488 listclean, listunknown)
1489 1489 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1490 1490
1491 1491 # check for any possibly clean files
1492 1492 if parentworking and cmp:
1493 1493 fixup = []
1494 1494 # do a full compare of any files that might have changed
1495 1495 for f in sorted(cmp):
1496 1496 if (f not in ctx1 or ctx2.flags(f) != ctx1.flags(f)
1497 1497 or ctx1[f].cmp(ctx2[f])):
1498 1498 modified.append(f)
1499 1499 else:
1500 1500 fixup.append(f)
1501 1501
1502 1502 # update dirstate for files that are actually clean
1503 1503 if fixup:
1504 1504 if listclean:
1505 1505 clean += fixup
1506 1506
1507 1507 try:
1508 1508 # updating the dirstate is optional
1509 1509 # so we don't wait on the lock
1510 1510 wlock = self.wlock(False)
1511 1511 try:
1512 1512 for f in fixup:
1513 1513 self.dirstate.normal(f)
1514 1514 finally:
1515 1515 wlock.release()
1516 1516 except error.LockError:
1517 1517 pass
1518 1518
1519 1519 if not parentworking:
1520 1520 mf1 = mfmatches(ctx1)
1521 1521 if working:
1522 1522 # we are comparing working dir against non-parent
1523 1523 # generate a pseudo-manifest for the working dir
1524 1524 mf2 = mfmatches(self['.'])
1525 1525 for f in cmp + modified + added:
1526 1526 mf2[f] = None
1527 1527 mf2.set(f, ctx2.flags(f))
1528 1528 for f in removed:
1529 1529 if f in mf2:
1530 1530 del mf2[f]
1531 1531 else:
1532 1532 # we are comparing two revisions
1533 1533 deleted, unknown, ignored = [], [], []
1534 1534 mf2 = mfmatches(ctx2)
1535 1535
1536 1536 modified, added, clean = [], [], []
1537 1537 withflags = mf1.withflags() | mf2.withflags()
1538 1538 for fn, mf2node in mf2.iteritems():
1539 1539 if fn in mf1:
1540 1540 if (fn not in deleted and
1541 1541 ((fn in withflags and mf1.flags(fn) != mf2.flags(fn)) or
1542 1542 (mf1[fn] != mf2node and
1543 1543 (mf2node or ctx1[fn].cmp(ctx2[fn]))))):
1544 1544 modified.append(fn)
1545 1545 elif listclean:
1546 1546 clean.append(fn)
1547 1547 del mf1[fn]
1548 1548 elif fn not in deleted:
1549 1549 added.append(fn)
1550 1550 removed = mf1.keys()
1551 1551
1552 1552 if working and modified and not self.dirstate._checklink:
1553 1553 # Symlink placeholders may get non-symlink-like contents
1554 1554 # via user error or dereferencing by NFS or Samba servers,
1555 1555 # so we filter out any placeholders that don't look like a
1556 1556 # symlink
1557 1557 sane = []
1558 1558 for f in modified:
1559 1559 if ctx2.flags(f) == 'l':
1560 1560 d = ctx2[f].data()
1561 1561 if len(d) >= 1024 or '\n' in d or util.binary(d):
1562 1562 self.ui.debug('ignoring suspect symlink placeholder'
1563 1563 ' "%s"\n' % f)
1564 1564 continue
1565 1565 sane.append(f)
1566 1566 modified = sane
1567 1567
1568 1568 r = modified, added, removed, deleted, unknown, ignored, clean
1569 1569
1570 1570 if listsubrepos:
1571 1571 for subpath, sub in subrepo.itersubrepos(ctx1, ctx2):
1572 1572 if working:
1573 1573 rev2 = None
1574 1574 else:
1575 1575 rev2 = ctx2.substate[subpath][1]
1576 1576 try:
1577 1577 submatch = matchmod.narrowmatcher(subpath, match)
1578 1578 s = sub.status(rev2, match=submatch, ignored=listignored,
1579 1579 clean=listclean, unknown=listunknown,
1580 1580 listsubrepos=True)
1581 1581 for rfiles, sfiles in zip(r, s):
1582 1582 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
1583 1583 except error.LookupError:
1584 1584 self.ui.status(_("skipping missing subrepository: %s\n")
1585 1585 % subpath)
1586 1586
1587 1587 for l in r:
1588 1588 l.sort()
1589 1589 return r
1590 1590
1591 1591 def heads(self, start=None):
1592 1592 heads = self.changelog.heads(start)
1593 1593 # sort the output in rev descending order
1594 1594 return sorted(heads, key=self.changelog.rev, reverse=True)
1595 1595
1596 1596 def branchheads(self, branch=None, start=None, closed=False):
1597 1597 '''return a (possibly filtered) list of heads for the given branch
1598 1598
1599 1599 Heads are returned in topological order, from newest to oldest.
1600 1600 If branch is None, use the dirstate branch.
1601 1601 If start is not None, return only heads reachable from start.
1602 1602 If closed is True, return heads that are marked as closed as well.
1603 1603 '''
1604 1604 if branch is None:
1605 1605 branch = self[None].branch()
1606 1606 branches = self.branchmap()
1607 1607 if branch not in branches:
1608 1608 return []
1609 1609 # the cache returns heads ordered lowest to highest
1610 1610 bheads = list(reversed(branches[branch]))
1611 1611 if start is not None:
1612 1612 # filter out the heads that cannot be reached from startrev
1613 1613 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1614 1614 bheads = [h for h in bheads if h in fbheads]
1615 1615 if not closed:
1616 1616 bheads = [h for h in bheads if not self[h].closesbranch()]
1617 1617 return bheads
1618 1618
1619 1619 def branches(self, nodes):
1620 1620 if not nodes:
1621 1621 nodes = [self.changelog.tip()]
1622 1622 b = []
1623 1623 for n in nodes:
1624 1624 t = n
1625 1625 while True:
1626 1626 p = self.changelog.parents(n)
1627 1627 if p[1] != nullid or p[0] == nullid:
1628 1628 b.append((t, n, p[0], p[1]))
1629 1629 break
1630 1630 n = p[0]
1631 1631 return b
1632 1632
1633 1633 def between(self, pairs):
1634 1634 r = []
1635 1635
1636 1636 for top, bottom in pairs:
1637 1637 n, l, i = top, [], 0
1638 1638 f = 1
1639 1639
1640 1640 while n != bottom and n != nullid:
1641 1641 p = self.changelog.parents(n)[0]
1642 1642 if i == f:
1643 1643 l.append(n)
1644 1644 f = f * 2
1645 1645 n = p
1646 1646 i += 1
1647 1647
1648 1648 r.append(l)
1649 1649
1650 1650 return r
1651 1651
1652 1652 def pull(self, remote, heads=None, force=False):
1653 1653 # don't open transaction for nothing or you break future useful
1654 1654 # rollback call
1655 1655 tr = None
1656 1656 trname = 'pull\n' + util.hidepassword(remote.url())
1657 1657 lock = self.lock()
1658 1658 try:
1659 1659 tmp = discovery.findcommonincoming(self, remote, heads=heads,
1660 1660 force=force)
1661 1661 common, fetch, rheads = tmp
1662 1662 if not fetch:
1663 1663 self.ui.status(_("no changes found\n"))
1664 1664 added = []
1665 1665 result = 0
1666 1666 else:
1667 1667 tr = self.transaction(trname)
1668 1668 if heads is None and list(common) == [nullid]:
1669 1669 self.ui.status(_("requesting all changes\n"))
1670 1670 elif heads is None and remote.capable('changegroupsubset'):
1671 1671 # issue1320, avoid a race if remote changed after discovery
1672 1672 heads = rheads
1673 1673
1674 1674 if remote.capable('getbundle'):
1675 1675 cg = remote.getbundle('pull', common=common,
1676 1676 heads=heads or rheads)
1677 1677 elif heads is None:
1678 1678 cg = remote.changegroup(fetch, 'pull')
1679 1679 elif not remote.capable('changegroupsubset'):
1680 1680 raise util.Abort(_("partial pull cannot be done because "
1681 1681 "other repository doesn't support "
1682 1682 "changegroupsubset."))
1683 1683 else:
1684 1684 cg = remote.changegroupsubset(fetch, heads, 'pull')
1685 1685 # we use unfiltered changelog here because hidden revision must
1686 1686 # be taken in account for phase synchronization. They may
1687 1687 # becomes public and becomes visible again.
1688 1688 cl = self.unfiltered().changelog
1689 1689 clstart = len(cl)
1690 1690 result = self.addchangegroup(cg, 'pull', remote.url())
1691 1691 clend = len(cl)
1692 1692 added = [cl.node(r) for r in xrange(clstart, clend)]
1693 1693
1694 1694 # compute target subset
1695 1695 if heads is None:
1696 1696 # We pulled every thing possible
1697 1697 # sync on everything common
1698 1698 subset = common + added
1699 1699 else:
1700 1700 # We pulled a specific subset
1701 1701 # sync on this subset
1702 1702 subset = heads
1703 1703
1704 1704 # Get remote phases data from remote
1705 1705 remotephases = remote.listkeys('phases')
1706 1706 publishing = bool(remotephases.get('publishing', False))
1707 1707 if remotephases and not publishing:
1708 1708 # remote is new and unpublishing
1709 1709 pheads, _dr = phases.analyzeremotephases(self, subset,
1710 1710 remotephases)
1711 1711 phases.advanceboundary(self, phases.public, pheads)
1712 1712 phases.advanceboundary(self, phases.draft, subset)
1713 1713 else:
1714 1714 # Remote is old or publishing all common changesets
1715 1715 # should be seen as public
1716 1716 phases.advanceboundary(self, phases.public, subset)
1717 1717
1718 1718 if obsolete._enabled:
1719 1719 self.ui.debug('fetching remote obsolete markers\n')
1720 1720 remoteobs = remote.listkeys('obsolete')
1721 1721 if 'dump0' in remoteobs:
1722 1722 if tr is None:
1723 1723 tr = self.transaction(trname)
1724 1724 for key in sorted(remoteobs, reverse=True):
1725 1725 if key.startswith('dump'):
1726 1726 data = base85.b85decode(remoteobs[key])
1727 1727 self.obsstore.mergemarkers(tr, data)
1728 1728 self.invalidatevolatilesets()
1729 1729 if tr is not None:
1730 1730 tr.close()
1731 1731 finally:
1732 1732 if tr is not None:
1733 1733 tr.release()
1734 1734 lock.release()
1735 1735
1736 1736 return result
1737 1737
1738 1738 def checkpush(self, force, revs):
1739 1739 """Extensions can override this function if additional checks have
1740 1740 to be performed before pushing, or call it if they override push
1741 1741 command.
1742 1742 """
1743 1743 pass
1744 1744
1745 1745 def push(self, remote, force=False, revs=None, newbranch=False):
1746 1746 '''Push outgoing changesets (limited by revs) from the current
1747 1747 repository to remote. Return an integer:
1748 1748 - None means nothing to push
1749 1749 - 0 means HTTP error
1750 1750 - 1 means we pushed and remote head count is unchanged *or*
1751 1751 we have outgoing changesets but refused to push
1752 1752 - other values as described by addchangegroup()
1753 1753 '''
1754 1754 # there are two ways to push to remote repo:
1755 1755 #
1756 1756 # addchangegroup assumes local user can lock remote
1757 1757 # repo (local filesystem, old ssh servers).
1758 1758 #
1759 1759 # unbundle assumes local user cannot lock remote repo (new ssh
1760 1760 # servers, http servers).
1761 1761
1762 1762 if not remote.canpush():
1763 1763 raise util.Abort(_("destination does not support push"))
1764 1764 unfi = self.unfiltered()
1765 1765 # get local lock as we might write phase data
1766 1766 locallock = self.lock()
1767 1767 try:
1768 1768 self.checkpush(force, revs)
1769 1769 lock = None
1770 1770 unbundle = remote.capable('unbundle')
1771 1771 if not unbundle:
1772 1772 lock = remote.lock()
1773 1773 try:
1774 1774 # discovery
1775 1775 fci = discovery.findcommonincoming
1776 1776 commoninc = fci(unfi, remote, force=force)
1777 1777 common, inc, remoteheads = commoninc
1778 1778 fco = discovery.findcommonoutgoing
1779 1779 outgoing = fco(unfi, remote, onlyheads=revs,
1780 1780 commoninc=commoninc, force=force)
1781 1781
1782 1782
1783 1783 if not outgoing.missing:
1784 1784 # nothing to push
1785 1785 scmutil.nochangesfound(unfi.ui, unfi, outgoing.excluded)
1786 1786 ret = None
1787 1787 else:
1788 1788 # something to push
1789 1789 if not force:
1790 1790 # if self.obsstore == False --> no obsolete
1791 1791 # then, save the iteration
1792 1792 if unfi.obsstore:
1793 1793 # this message are here for 80 char limit reason
1794 1794 mso = _("push includes obsolete changeset: %s!")
1795 1795 mst = "push includes %s changeset: %s!"
1796 1796 # plain versions for i18n tool to detect them
1797 1797 _("push includes unstable changeset: %s!")
1798 1798 _("push includes bumped changeset: %s!")
1799 1799 _("push includes divergent changeset: %s!")
1800 1800 # If we are to push if there is at least one
1801 1801 # obsolete or unstable changeset in missing, at
1802 1802 # least one of the missinghead will be obsolete or
1803 1803 # unstable. So checking heads only is ok
1804 1804 for node in outgoing.missingheads:
1805 1805 ctx = unfi[node]
1806 1806 if ctx.obsolete():
1807 1807 raise util.Abort(mso % ctx)
1808 1808 elif ctx.troubled():
1809 1809 raise util.Abort(_(mst)
1810 1810 % (ctx.troubles()[0],
1811 1811 ctx))
1812 1812 discovery.checkheads(unfi, remote, outgoing,
1813 1813 remoteheads, newbranch,
1814 1814 bool(inc))
1815 1815
1816 1816 # create a changegroup from local
1817 1817 if revs is None and not outgoing.excluded:
1818 1818 # push everything,
1819 1819 # use the fast path, no race possible on push
1820 1820 cg = self._changegroup(outgoing.missing, 'push')
1821 1821 else:
1822 1822 cg = self.getlocalbundle('push', outgoing)
1823 1823
1824 1824 # apply changegroup to remote
1825 1825 if unbundle:
1826 1826 # local repo finds heads on server, finds out what
1827 1827 # revs it must push. once revs transferred, if server
1828 1828 # finds it has different heads (someone else won
1829 1829 # commit/push race), server aborts.
1830 1830 if force:
1831 1831 remoteheads = ['force']
1832 1832 # ssh: return remote's addchangegroup()
1833 1833 # http: return remote's addchangegroup() or 0 for error
1834 1834 ret = remote.unbundle(cg, remoteheads, 'push')
1835 1835 else:
1836 1836 # we return an integer indicating remote head count
1837 1837 # change
1838 1838 ret = remote.addchangegroup(cg, 'push', self.url())
1839 1839
1840 1840 if ret:
1841 1841 # push succeed, synchronize target of the push
1842 1842 cheads = outgoing.missingheads
1843 1843 elif revs is None:
1844 1844 # All out push fails. synchronize all common
1845 1845 cheads = outgoing.commonheads
1846 1846 else:
1847 1847 # I want cheads = heads(::missingheads and ::commonheads)
1848 1848 # (missingheads is revs with secret changeset filtered out)
1849 1849 #
1850 1850 # This can be expressed as:
1851 1851 # cheads = ( (missingheads and ::commonheads)
1852 1852 # + (commonheads and ::missingheads))"
1853 1853 # )
1854 1854 #
1855 1855 # while trying to push we already computed the following:
1856 1856 # common = (::commonheads)
1857 1857 # missing = ((commonheads::missingheads) - commonheads)
1858 1858 #
1859 1859 # We can pick:
1860 1860 # * missingheads part of common (::commonheads)
1861 1861 common = set(outgoing.common)
1862 1862 cheads = [node for node in revs if node in common]
1863 1863 # and
1864 1864 # * commonheads parents on missing
1865 1865 revset = unfi.set('%ln and parents(roots(%ln))',
1866 1866 outgoing.commonheads,
1867 1867 outgoing.missing)
1868 1868 cheads.extend(c.node() for c in revset)
1869 1869 # even when we don't push, exchanging phase data is useful
1870 1870 remotephases = remote.listkeys('phases')
1871 1871 if (self.ui.configbool('ui', '_usedassubrepo', False)
1872 1872 and remotephases # server supports phases
1873 1873 and ret is None # nothing was pushed
1874 1874 and remotephases.get('publishing', False)):
1875 1875 # When:
1876 1876 # - this is a subrepo push
1877 1877 # - and remote support phase
1878 1878 # - and no changeset was pushed
1879 1879 # - and remote is publishing
1880 1880 # We may be in issue 3871 case!
1881 1881 # We drop the possible phase synchronisation done by
1882 1882 # courtesy to publish changesets possibly locally draft
1883 1883 # on the remote.
1884 1884 remotephases = {'publishing': 'True'}
1885 1885 if not remotephases: # old server or public only repo
1886 1886 phases.advanceboundary(self, phases.public, cheads)
1887 1887 # don't push any phase data as there is nothing to push
1888 1888 else:
1889 1889 ana = phases.analyzeremotephases(self, cheads, remotephases)
1890 1890 pheads, droots = ana
1891 1891 ### Apply remote phase on local
1892 1892 if remotephases.get('publishing', False):
1893 1893 phases.advanceboundary(self, phases.public, cheads)
1894 1894 else: # publish = False
1895 1895 phases.advanceboundary(self, phases.public, pheads)
1896 1896 phases.advanceboundary(self, phases.draft, cheads)
1897 1897 ### Apply local phase on remote
1898 1898
1899 1899 # Get the list of all revs draft on remote by public here.
1900 1900 # XXX Beware that revset break if droots is not strictly
1901 1901 # XXX root we may want to ensure it is but it is costly
1902 1902 outdated = unfi.set('heads((%ln::%ln) and public())',
1903 1903 droots, cheads)
1904 1904 for newremotehead in outdated:
1905 1905 r = remote.pushkey('phases',
1906 1906 newremotehead.hex(),
1907 1907 str(phases.draft),
1908 1908 str(phases.public))
1909 1909 if not r:
1910 1910 self.ui.warn(_('updating %s to public failed!\n')
1911 1911 % newremotehead)
1912 1912 self.ui.debug('try to push obsolete markers to remote\n')
1913 1913 if (obsolete._enabled and self.obsstore and
1914 1914 'obsolete' in remote.listkeys('namespaces')):
1915 1915 rslts = []
1916 1916 remotedata = self.listkeys('obsolete')
1917 1917 for key in sorted(remotedata, reverse=True):
1918 1918 # reverse sort to ensure we end with dump0
1919 1919 data = remotedata[key]
1920 1920 rslts.append(remote.pushkey('obsolete', key, '', data))
1921 1921 if [r for r in rslts if not r]:
1922 1922 msg = _('failed to push some obsolete markers!\n')
1923 1923 self.ui.warn(msg)
1924 1924 finally:
1925 1925 if lock is not None:
1926 1926 lock.release()
1927 1927 finally:
1928 1928 locallock.release()
1929 1929
1930 1930 self.ui.debug("checking for updated bookmarks\n")
1931 1931 rb = remote.listkeys('bookmarks')
1932 1932 for k in rb.keys():
1933 1933 if k in unfi._bookmarks:
1934 1934 nr, nl = rb[k], hex(self._bookmarks[k])
1935 1935 if nr in unfi:
1936 1936 cr = unfi[nr]
1937 1937 cl = unfi[nl]
1938 1938 if bookmarks.validdest(unfi, cr, cl):
1939 1939 r = remote.pushkey('bookmarks', k, nr, nl)
1940 1940 if r:
1941 1941 self.ui.status(_("updating bookmark %s\n") % k)
1942 1942 else:
1943 1943 self.ui.warn(_('updating bookmark %s'
1944 1944 ' failed!\n') % k)
1945 1945
1946 1946 return ret
1947 1947
1948 1948 def changegroupinfo(self, nodes, source):
1949 1949 if self.ui.verbose or source == 'bundle':
1950 1950 self.ui.status(_("%d changesets found\n") % len(nodes))
1951 1951 if self.ui.debugflag:
1952 1952 self.ui.debug("list of changesets:\n")
1953 1953 for node in nodes:
1954 1954 self.ui.debug("%s\n" % hex(node))
1955 1955
1956 1956 def changegroupsubset(self, bases, heads, source):
1957 1957 """Compute a changegroup consisting of all the nodes that are
1958 1958 descendants of any of the bases and ancestors of any of the heads.
1959 1959 Return a chunkbuffer object whose read() method will return
1960 1960 successive changegroup chunks.
1961 1961
1962 1962 It is fairly complex as determining which filenodes and which
1963 1963 manifest nodes need to be included for the changeset to be complete
1964 1964 is non-trivial.
1965 1965
1966 1966 Another wrinkle is doing the reverse, figuring out which changeset in
1967 1967 the changegroup a particular filenode or manifestnode belongs to.
1968 1968 """
1969 1969 cl = self.changelog
1970 1970 if not bases:
1971 1971 bases = [nullid]
1972 1972 csets, bases, heads = cl.nodesbetween(bases, heads)
1973 1973 # We assume that all ancestors of bases are known
1974 1974 common = cl.ancestors([cl.rev(n) for n in bases])
1975 1975 return self._changegroupsubset(common, csets, heads, source)
1976 1976
1977 1977 def getlocalbundle(self, source, outgoing):
1978 1978 """Like getbundle, but taking a discovery.outgoing as an argument.
1979 1979
1980 1980 This is only implemented for local repos and reuses potentially
1981 1981 precomputed sets in outgoing."""
1982 1982 if not outgoing.missing:
1983 1983 return None
1984 1984 return self._changegroupsubset(outgoing.common,
1985 1985 outgoing.missing,
1986 1986 outgoing.missingheads,
1987 1987 source)
1988 1988
1989 1989 def getbundle(self, source, heads=None, common=None):
1990 1990 """Like changegroupsubset, but returns the set difference between the
1991 1991 ancestors of heads and the ancestors common.
1992 1992
1993 1993 If heads is None, use the local heads. If common is None, use [nullid].
1994 1994
1995 1995 The nodes in common might not all be known locally due to the way the
1996 1996 current discovery protocol works.
1997 1997 """
1998 1998 cl = self.changelog
1999 1999 if common:
2000 2000 hasnode = cl.hasnode
2001 2001 common = [n for n in common if hasnode(n)]
2002 2002 else:
2003 2003 common = [nullid]
2004 2004 if not heads:
2005 2005 heads = cl.heads()
2006 2006 return self.getlocalbundle(source,
2007 2007 discovery.outgoing(cl, common, heads))
2008 2008
2009 2009 @unfilteredmethod
2010 2010 def _changegroupsubset(self, commonrevs, csets, heads, source):
2011 2011
2012 2012 cl = self.changelog
2013 2013 mf = self.manifest
2014 2014 mfs = {} # needed manifests
2015 2015 fnodes = {} # needed file nodes
2016 2016 changedfiles = set()
2017 2017 fstate = ['', {}]
2018 2018 count = [0, 0]
2019 2019
2020 2020 # can we go through the fast path ?
2021 2021 heads.sort()
2022 2022 if heads == sorted(self.heads()):
2023 2023 return self._changegroup(csets, source)
2024 2024
2025 2025 # slow path
2026 2026 self.hook('preoutgoing', throw=True, source=source)
2027 2027 self.changegroupinfo(csets, source)
2028 2028
2029 2029 # filter any nodes that claim to be part of the known set
2030 2030 def prune(revlog, missing):
2031 2031 rr, rl = revlog.rev, revlog.linkrev
2032 2032 return [n for n in missing
2033 2033 if rl(rr(n)) not in commonrevs]
2034 2034
2035 2035 progress = self.ui.progress
2036 2036 _bundling = _('bundling')
2037 2037 _changesets = _('changesets')
2038 2038 _manifests = _('manifests')
2039 2039 _files = _('files')
2040 2040
2041 2041 def lookup(revlog, x):
2042 2042 if revlog == cl:
2043 2043 c = cl.read(x)
2044 2044 changedfiles.update(c[3])
2045 2045 mfs.setdefault(c[0], x)
2046 2046 count[0] += 1
2047 2047 progress(_bundling, count[0],
2048 2048 unit=_changesets, total=count[1])
2049 2049 return x
2050 2050 elif revlog == mf:
2051 2051 clnode = mfs[x]
2052 2052 mdata = mf.readfast(x)
2053 2053 for f, n in mdata.iteritems():
2054 2054 if f in changedfiles:
2055 2055 fnodes[f].setdefault(n, clnode)
2056 2056 count[0] += 1
2057 2057 progress(_bundling, count[0],
2058 2058 unit=_manifests, total=count[1])
2059 2059 return clnode
2060 2060 else:
2061 2061 progress(_bundling, count[0], item=fstate[0],
2062 2062 unit=_files, total=count[1])
2063 2063 return fstate[1][x]
2064 2064
2065 2065 bundler = changegroup.bundle10(lookup)
2066 2066 reorder = self.ui.config('bundle', 'reorder', 'auto')
2067 2067 if reorder == 'auto':
2068 2068 reorder = None
2069 2069 else:
2070 2070 reorder = util.parsebool(reorder)
2071 2071
2072 2072 def gengroup():
2073 2073 # Create a changenode group generator that will call our functions
2074 2074 # back to lookup the owning changenode and collect information.
2075 2075 count[:] = [0, len(csets)]
2076 2076 for chunk in cl.group(csets, bundler, reorder=reorder):
2077 2077 yield chunk
2078 2078 progress(_bundling, None)
2079 2079
2080 2080 # Create a generator for the manifestnodes that calls our lookup
2081 2081 # and data collection functions back.
2082 2082 for f in changedfiles:
2083 2083 fnodes[f] = {}
2084 2084 count[:] = [0, len(mfs)]
2085 2085 for chunk in mf.group(prune(mf, mfs), bundler, reorder=reorder):
2086 2086 yield chunk
2087 2087 progress(_bundling, None)
2088 2088
2089 2089 mfs.clear()
2090 2090
2091 2091 # Go through all our files in order sorted by name.
2092 2092 count[:] = [0, len(changedfiles)]
2093 2093 for fname in sorted(changedfiles):
2094 2094 filerevlog = self.file(fname)
2095 2095 if not len(filerevlog):
2096 2096 raise util.Abort(_("empty or missing revlog for %s")
2097 2097 % fname)
2098 2098 fstate[0] = fname
2099 2099 fstate[1] = fnodes.pop(fname, {})
2100 2100
2101 2101 nodelist = prune(filerevlog, fstate[1])
2102 2102 if nodelist:
2103 2103 count[0] += 1
2104 2104 yield bundler.fileheader(fname)
2105 2105 for chunk in filerevlog.group(nodelist, bundler, reorder):
2106 2106 yield chunk
2107 2107
2108 2108 # Signal that no more groups are left.
2109 2109 yield bundler.close()
2110 2110 progress(_bundling, None)
2111 2111
2112 2112 if csets:
2113 2113 self.hook('outgoing', node=hex(csets[0]), source=source)
2114 2114
2115 2115 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
2116 2116
2117 2117 def changegroup(self, basenodes, source):
2118 2118 # to avoid a race we use changegroupsubset() (issue1320)
2119 2119 return self.changegroupsubset(basenodes, self.heads(), source)
2120 2120
2121 2121 @unfilteredmethod
2122 2122 def _changegroup(self, nodes, source):
2123 2123 """Compute the changegroup of all nodes that we have that a recipient
2124 2124 doesn't. Return a chunkbuffer object whose read() method will return
2125 2125 successive changegroup chunks.
2126 2126
2127 2127 This is much easier than the previous function as we can assume that
2128 2128 the recipient has any changenode we aren't sending them.
2129 2129
2130 2130 nodes is the set of nodes to send"""
2131 2131
2132 2132 cl = self.changelog
2133 2133 mf = self.manifest
2134 2134 mfs = {}
2135 2135 changedfiles = set()
2136 2136 fstate = ['']
2137 2137 count = [0, 0]
2138 2138
2139 2139 self.hook('preoutgoing', throw=True, source=source)
2140 2140 self.changegroupinfo(nodes, source)
2141 2141
2142 2142 revset = set([cl.rev(n) for n in nodes])
2143 2143
2144 2144 def gennodelst(log):
2145 2145 ln, llr = log.node, log.linkrev
2146 2146 return [ln(r) for r in log if llr(r) in revset]
2147 2147
2148 2148 progress = self.ui.progress
2149 2149 _bundling = _('bundling')
2150 2150 _changesets = _('changesets')
2151 2151 _manifests = _('manifests')
2152 2152 _files = _('files')
2153 2153
2154 2154 def lookup(revlog, x):
2155 2155 if revlog == cl:
2156 2156 c = cl.read(x)
2157 2157 changedfiles.update(c[3])
2158 2158 mfs.setdefault(c[0], x)
2159 2159 count[0] += 1
2160 2160 progress(_bundling, count[0],
2161 2161 unit=_changesets, total=count[1])
2162 2162 return x
2163 2163 elif revlog == mf:
2164 2164 count[0] += 1
2165 2165 progress(_bundling, count[0],
2166 2166 unit=_manifests, total=count[1])
2167 2167 return cl.node(revlog.linkrev(revlog.rev(x)))
2168 2168 else:
2169 2169 progress(_bundling, count[0], item=fstate[0],
2170 2170 total=count[1], unit=_files)
2171 2171 return cl.node(revlog.linkrev(revlog.rev(x)))
2172 2172
2173 2173 bundler = changegroup.bundle10(lookup)
2174 2174 reorder = self.ui.config('bundle', 'reorder', 'auto')
2175 2175 if reorder == 'auto':
2176 2176 reorder = None
2177 2177 else:
2178 2178 reorder = util.parsebool(reorder)
2179 2179
2180 2180 def gengroup():
2181 2181 '''yield a sequence of changegroup chunks (strings)'''
2182 2182 # construct a list of all changed files
2183 2183
2184 2184 count[:] = [0, len(nodes)]
2185 2185 for chunk in cl.group(nodes, bundler, reorder=reorder):
2186 2186 yield chunk
2187 2187 progress(_bundling, None)
2188 2188
2189 2189 count[:] = [0, len(mfs)]
2190 2190 for chunk in mf.group(gennodelst(mf), bundler, reorder=reorder):
2191 2191 yield chunk
2192 2192 progress(_bundling, None)
2193 2193
2194 2194 count[:] = [0, len(changedfiles)]
2195 2195 for fname in sorted(changedfiles):
2196 2196 filerevlog = self.file(fname)
2197 2197 if not len(filerevlog):
2198 2198 raise util.Abort(_("empty or missing revlog for %s")
2199 2199 % fname)
2200 2200 fstate[0] = fname
2201 2201 nodelist = gennodelst(filerevlog)
2202 2202 if nodelist:
2203 2203 count[0] += 1
2204 2204 yield bundler.fileheader(fname)
2205 2205 for chunk in filerevlog.group(nodelist, bundler, reorder):
2206 2206 yield chunk
2207 2207 yield bundler.close()
2208 2208 progress(_bundling, None)
2209 2209
2210 2210 if nodes:
2211 2211 self.hook('outgoing', node=hex(nodes[0]), source=source)
2212 2212
2213 2213 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
2214 2214
2215 2215 @unfilteredmethod
2216 2216 def addchangegroup(self, source, srctype, url, emptyok=False):
2217 2217 """Add the changegroup returned by source.read() to this repo.
2218 2218 srctype is a string like 'push', 'pull', or 'unbundle'. url is
2219 2219 the URL of the repo where this changegroup is coming from.
2220 2220
2221 2221 Return an integer summarizing the change to this repo:
2222 2222 - nothing changed or no source: 0
2223 2223 - more heads than before: 1+added heads (2..n)
2224 2224 - fewer heads than before: -1-removed heads (-2..-n)
2225 2225 - number of heads stays the same: 1
2226 2226 """
2227 2227 def csmap(x):
2228 2228 self.ui.debug("add changeset %s\n" % short(x))
2229 2229 return len(cl)
2230 2230
2231 2231 def revmap(x):
2232 2232 return cl.rev(x)
2233 2233
2234 2234 if not source:
2235 2235 return 0
2236 2236
2237 2237 self.hook('prechangegroup', throw=True, source=srctype, url=url)
2238 2238
2239 2239 changesets = files = revisions = 0
2240 2240 efiles = set()
2241 2241
2242 2242 # write changelog data to temp files so concurrent readers will not see
2243 2243 # inconsistent view
2244 2244 cl = self.changelog
2245 2245 cl.delayupdate()
2246 2246 oldheads = cl.heads()
2247 2247
2248 2248 tr = self.transaction("\n".join([srctype, util.hidepassword(url)]))
2249 2249 try:
2250 2250 trp = weakref.proxy(tr)
2251 2251 # pull off the changeset group
2252 2252 self.ui.status(_("adding changesets\n"))
2253 2253 clstart = len(cl)
2254 2254 class prog(object):
2255 2255 step = _('changesets')
2256 2256 count = 1
2257 2257 ui = self.ui
2258 2258 total = None
2259 2259 def __call__(self):
2260 2260 self.ui.progress(self.step, self.count, unit=_('chunks'),
2261 2261 total=self.total)
2262 2262 self.count += 1
2263 2263 pr = prog()
2264 2264 source.callback = pr
2265 2265
2266 2266 source.changelogheader()
2267 2267 srccontent = cl.addgroup(source, csmap, trp)
2268 2268 if not (srccontent or emptyok):
2269 2269 raise util.Abort(_("received changelog group is empty"))
2270 2270 clend = len(cl)
2271 2271 changesets = clend - clstart
2272 2272 for c in xrange(clstart, clend):
2273 2273 efiles.update(self[c].files())
2274 2274 efiles = len(efiles)
2275 2275 self.ui.progress(_('changesets'), None)
2276 2276
2277 2277 # pull off the manifest group
2278 2278 self.ui.status(_("adding manifests\n"))
2279 2279 pr.step = _('manifests')
2280 2280 pr.count = 1
2281 2281 pr.total = changesets # manifests <= changesets
2282 2282 # no need to check for empty manifest group here:
2283 2283 # if the result of the merge of 1 and 2 is the same in 3 and 4,
2284 2284 # no new manifest will be created and the manifest group will
2285 2285 # be empty during the pull
2286 2286 source.manifestheader()
2287 2287 self.manifest.addgroup(source, revmap, trp)
2288 2288 self.ui.progress(_('manifests'), None)
2289 2289
2290 2290 needfiles = {}
2291 2291 if self.ui.configbool('server', 'validate', default=False):
2292 2292 # validate incoming csets have their manifests
2293 2293 for cset in xrange(clstart, clend):
2294 2294 mfest = self.changelog.read(self.changelog.node(cset))[0]
2295 2295 mfest = self.manifest.readdelta(mfest)
2296 2296 # store file nodes we must see
2297 2297 for f, n in mfest.iteritems():
2298 2298 needfiles.setdefault(f, set()).add(n)
2299 2299
2300 2300 # process the files
2301 2301 self.ui.status(_("adding file changes\n"))
2302 2302 pr.step = _('files')
2303 2303 pr.count = 1
2304 2304 pr.total = efiles
2305 2305 source.callback = None
2306 2306
2307 2307 while True:
2308 2308 chunkdata = source.filelogheader()
2309 2309 if not chunkdata:
2310 2310 break
2311 2311 f = chunkdata["filename"]
2312 2312 self.ui.debug("adding %s revisions\n" % f)
2313 2313 pr()
2314 2314 fl = self.file(f)
2315 2315 o = len(fl)
2316 2316 if not fl.addgroup(source, revmap, trp):
2317 2317 raise util.Abort(_("received file revlog group is empty"))
2318 2318 revisions += len(fl) - o
2319 2319 files += 1
2320 2320 if f in needfiles:
2321 2321 needs = needfiles[f]
2322 2322 for new in xrange(o, len(fl)):
2323 2323 n = fl.node(new)
2324 2324 if n in needs:
2325 2325 needs.remove(n)
2326 2326 else:
2327 2327 raise util.Abort(
2328 2328 _("received spurious file revlog entry"))
2329 2329 if not needs:
2330 2330 del needfiles[f]
2331 2331 self.ui.progress(_('files'), None)
2332 2332
2333 2333 for f, needs in needfiles.iteritems():
2334 2334 fl = self.file(f)
2335 2335 for n in needs:
2336 2336 try:
2337 2337 fl.rev(n)
2338 2338 except error.LookupError:
2339 2339 raise util.Abort(
2340 2340 _('missing file data for %s:%s - run hg verify') %
2341 2341 (f, hex(n)))
2342 2342
2343 2343 dh = 0
2344 2344 if oldheads:
2345 2345 heads = cl.heads()
2346 2346 dh = len(heads) - len(oldheads)
2347 2347 for h in heads:
2348 2348 if h not in oldheads and self[h].closesbranch():
2349 2349 dh -= 1
2350 2350 htext = ""
2351 2351 if dh:
2352 2352 htext = _(" (%+d heads)") % dh
2353 2353
2354 2354 self.ui.status(_("added %d changesets"
2355 2355 " with %d changes to %d files%s\n")
2356 2356 % (changesets, revisions, files, htext))
2357 2357 self.invalidatevolatilesets()
2358 2358
2359 2359 if changesets > 0:
2360 2360 p = lambda: cl.writepending() and self.root or ""
2361 2361 self.hook('pretxnchangegroup', throw=True,
2362 2362 node=hex(cl.node(clstart)), source=srctype,
2363 2363 url=url, pending=p)
2364 2364
2365 2365 added = [cl.node(r) for r in xrange(clstart, clend)]
2366 2366 publishing = self.ui.configbool('phases', 'publish', True)
2367 2367 if srctype == 'push':
2368 2368 # Old server can not push the boundary themself.
2369 2369 # New server won't push the boundary if changeset already
2370 2370 # existed locally as secrete
2371 2371 #
2372 2372 # We should not use added here but the list of all change in
2373 2373 # the bundle
2374 2374 if publishing:
2375 2375 phases.advanceboundary(self, phases.public, srccontent)
2376 2376 else:
2377 2377 phases.advanceboundary(self, phases.draft, srccontent)
2378 2378 phases.retractboundary(self, phases.draft, added)
2379 2379 elif srctype != 'strip':
2380 2380 # publishing only alter behavior during push
2381 2381 #
2382 2382 # strip should not touch boundary at all
2383 2383 phases.retractboundary(self, phases.draft, added)
2384 2384
2385 2385 # make changelog see real files again
2386 2386 cl.finalize(trp)
2387 2387
2388 2388 tr.close()
2389 2389
2390 2390 if changesets > 0:
2391 2391 if srctype != 'strip':
2392 2392 # During strip, branchcache is invalid but coming call to
2393 2393 # `destroyed` will repair it.
2394 2394 # In other case we can safely update cache on disk.
2395 2395 branchmap.updatecache(self.filtered('served'))
2396 2396 def runhooks():
2397 2397 # forcefully update the on-disk branch cache
2398 2398 self.ui.debug("updating the branch cache\n")
2399 2399 self.hook("changegroup", node=hex(cl.node(clstart)),
2400 2400 source=srctype, url=url)
2401 2401
2402 2402 for n in added:
2403 2403 self.hook("incoming", node=hex(n), source=srctype,
2404 2404 url=url)
2405 2405
2406 2406 newheads = [h for h in self.heads() if h not in oldheads]
2407 2407 self.ui.log("incoming",
2408 2408 "%s incoming changes - new heads: %s\n",
2409 2409 len(added),
2410 2410 ', '.join([hex(c[:6]) for c in newheads]))
2411 2411 self._afterlock(runhooks)
2412 2412
2413 2413 finally:
2414 2414 tr.release()
2415 2415 # never return 0 here:
2416 2416 if dh < 0:
2417 2417 return dh - 1
2418 2418 else:
2419 2419 return dh + 1
2420 2420
2421 2421 def stream_in(self, remote, requirements):
2422 2422 lock = self.lock()
2423 2423 try:
2424 2424 # Save remote branchmap. We will use it later
2425 2425 # to speed up branchcache creation
2426 2426 rbranchmap = None
2427 2427 if remote.capable("branchmap"):
2428 2428 rbranchmap = remote.branchmap()
2429 2429
2430 2430 fp = remote.stream_out()
2431 2431 l = fp.readline()
2432 2432 try:
2433 2433 resp = int(l)
2434 2434 except ValueError:
2435 2435 raise error.ResponseError(
2436 2436 _('unexpected response from remote server:'), l)
2437 2437 if resp == 1:
2438 2438 raise util.Abort(_('operation forbidden by server'))
2439 2439 elif resp == 2:
2440 2440 raise util.Abort(_('locking the remote repository failed'))
2441 2441 elif resp != 0:
2442 2442 raise util.Abort(_('the server sent an unknown error code'))
2443 2443 self.ui.status(_('streaming all changes\n'))
2444 2444 l = fp.readline()
2445 2445 try:
2446 2446 total_files, total_bytes = map(int, l.split(' ', 1))
2447 2447 except (ValueError, TypeError):
2448 2448 raise error.ResponseError(
2449 2449 _('unexpected response from remote server:'), l)
2450 2450 self.ui.status(_('%d files to transfer, %s of data\n') %
2451 2451 (total_files, util.bytecount(total_bytes)))
2452 2452 handled_bytes = 0
2453 2453 self.ui.progress(_('clone'), 0, total=total_bytes)
2454 2454 start = time.time()
2455 2455 for i in xrange(total_files):
2456 2456 # XXX doesn't support '\n' or '\r' in filenames
2457 2457 l = fp.readline()
2458 2458 try:
2459 2459 name, size = l.split('\0', 1)
2460 2460 size = int(size)
2461 2461 except (ValueError, TypeError):
2462 2462 raise error.ResponseError(
2463 2463 _('unexpected response from remote server:'), l)
2464 2464 if self.ui.debugflag:
2465 2465 self.ui.debug('adding %s (%s)\n' %
2466 2466 (name, util.bytecount(size)))
2467 2467 # for backwards compat, name was partially encoded
2468 2468 ofp = self.sopener(store.decodedir(name), 'w')
2469 2469 for chunk in util.filechunkiter(fp, limit=size):
2470 2470 handled_bytes += len(chunk)
2471 2471 self.ui.progress(_('clone'), handled_bytes,
2472 2472 total=total_bytes)
2473 2473 ofp.write(chunk)
2474 2474 ofp.close()
2475 2475 elapsed = time.time() - start
2476 2476 if elapsed <= 0:
2477 2477 elapsed = 0.001
2478 2478 self.ui.progress(_('clone'), None)
2479 2479 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
2480 2480 (util.bytecount(total_bytes), elapsed,
2481 2481 util.bytecount(total_bytes / elapsed)))
2482 2482
2483 2483 # new requirements = old non-format requirements +
2484 2484 # new format-related
2485 2485 # requirements from the streamed-in repository
2486 2486 requirements.update(set(self.requirements) - self.supportedformats)
2487 2487 self._applyrequirements(requirements)
2488 2488 self._writerequirements()
2489 2489
2490 2490 if rbranchmap:
2491 2491 rbheads = []
2492 2492 for bheads in rbranchmap.itervalues():
2493 2493 rbheads.extend(bheads)
2494 2494
2495 2495 if rbheads:
2496 2496 rtiprev = max((int(self.changelog.rev(node))
2497 2497 for node in rbheads))
2498 2498 cache = branchmap.branchcache(rbranchmap,
2499 2499 self[rtiprev].node(),
2500 2500 rtiprev)
2501 2501 # Try to stick it as low as possible
2502 2502 # filter above served are unlikely to be fetch from a clone
2503 2503 for candidate in ('base', 'immutable', 'served'):
2504 2504 rview = self.filtered(candidate)
2505 2505 if cache.validfor(rview):
2506 2506 self._branchcaches[candidate] = cache
2507 2507 cache.write(rview)
2508 2508 break
2509 2509 self.invalidate()
2510 2510 return len(self.heads()) + 1
2511 2511 finally:
2512 2512 lock.release()
2513 2513
2514 2514 def clone(self, remote, heads=[], stream=False):
2515 2515 '''clone remote repository.
2516 2516
2517 2517 keyword arguments:
2518 2518 heads: list of revs to clone (forces use of pull)
2519 2519 stream: use streaming clone if possible'''
2520 2520
2521 2521 # now, all clients that can request uncompressed clones can
2522 2522 # read repo formats supported by all servers that can serve
2523 2523 # them.
2524 2524
2525 2525 # if revlog format changes, client will have to check version
2526 2526 # and format flags on "stream" capability, and use
2527 2527 # uncompressed only if compatible.
2528 2528
2529 2529 if not stream:
2530 2530 # if the server explicitly prefers to stream (for fast LANs)
2531 2531 stream = remote.capable('stream-preferred')
2532 2532
2533 2533 if stream and not heads:
2534 2534 # 'stream' means remote revlog format is revlogv1 only
2535 2535 if remote.capable('stream'):
2536 2536 return self.stream_in(remote, set(('revlogv1',)))
2537 2537 # otherwise, 'streamreqs' contains the remote revlog format
2538 2538 streamreqs = remote.capable('streamreqs')
2539 2539 if streamreqs:
2540 2540 streamreqs = set(streamreqs.split(','))
2541 2541 # if we support it, stream in and adjust our requirements
2542 2542 if not streamreqs - self.supportedformats:
2543 2543 return self.stream_in(remote, streamreqs)
2544 2544 return self.pull(remote, heads)
2545 2545
2546 2546 def pushkey(self, namespace, key, old, new):
2547 2547 self.hook('prepushkey', throw=True, namespace=namespace, key=key,
2548 2548 old=old, new=new)
2549 2549 self.ui.debug('pushing key for "%s:%s"\n' % (namespace, key))
2550 2550 ret = pushkey.push(self, namespace, key, old, new)
2551 2551 self.hook('pushkey', namespace=namespace, key=key, old=old, new=new,
2552 2552 ret=ret)
2553 2553 return ret
2554 2554
2555 2555 def listkeys(self, namespace):
2556 2556 self.hook('prelistkeys', throw=True, namespace=namespace)
2557 2557 self.ui.debug('listing keys for "%s"\n' % namespace)
2558 2558 values = pushkey.list(self, namespace)
2559 2559 self.hook('listkeys', namespace=namespace, values=values)
2560 2560 return values
2561 2561
2562 2562 def debugwireargs(self, one, two, three=None, four=None, five=None):
2563 2563 '''used to test argument passing over the wire'''
2564 2564 return "%s %s %s %s %s" % (one, two, three, four, five)
2565 2565
2566 2566 def savecommitmessage(self, text):
2567 2567 fp = self.opener('last-message.txt', 'wb')
2568 2568 try:
2569 2569 fp.write(text)
2570 2570 finally:
2571 2571 fp.close()
2572 2572 return self.pathto(fp.name[len(self.root) + 1:])
2573 2573
2574 2574 # used to avoid circular references so destructors work
2575 2575 def aftertrans(files):
2576 2576 renamefiles = [tuple(t) for t in files]
2577 2577 def a():
2578 2578 for src, dest in renamefiles:
2579 2579 try:
2580 2580 util.rename(src, dest)
2581 2581 except OSError: # journal file does not yet exist
2582 2582 pass
2583 2583 return a
2584 2584
2585 2585 def undoname(fn):
2586 2586 base, name = os.path.split(fn)
2587 2587 assert name.startswith('journal')
2588 2588 return os.path.join(base, name.replace('journal', 'undo', 1))
2589 2589
2590 2590 def instance(ui, path, create):
2591 2591 return localrepository(ui, util.urllocalpath(path), create)
2592 2592
2593 2593 def islocal(path):
2594 2594 return True
@@ -1,937 +1,939 b''
1 1 # scmutil.py - Mercurial core utility functions
2 2 #
3 3 # Copyright Matt Mackall <mpm@selenic.com>
4 4 #
5 5 # This software may be used and distributed according to the terms of the
6 6 # GNU General Public License version 2 or any later version.
7 7
8 8 from i18n import _
9 9 from mercurial.node import nullrev
10 10 import util, error, osutil, revset, similar, encoding, phases, parsers
11 11 import match as matchmod
12 12 import os, errno, re, stat, glob
13 13
14 14 if os.name == 'nt':
15 15 import scmwindows as scmplatform
16 16 else:
17 17 import scmposix as scmplatform
18 18
19 19 systemrcpath = scmplatform.systemrcpath
20 20 userrcpath = scmplatform.userrcpath
21 21
22 22 def nochangesfound(ui, repo, excluded=None):
23 23 '''Report no changes for push/pull, excluded is None or a list of
24 24 nodes excluded from the push/pull.
25 25 '''
26 26 secretlist = []
27 27 if excluded:
28 28 for n in excluded:
29 29 if n not in repo:
30 30 # discovery should not have included the filtered revision,
31 31 # we have to explicitly exclude it until discovery is cleanup.
32 32 continue
33 33 ctx = repo[n]
34 34 if ctx.phase() >= phases.secret and not ctx.extinct():
35 35 secretlist.append(n)
36 36
37 37 if secretlist:
38 38 ui.status(_("no changes found (ignored %d secret changesets)\n")
39 39 % len(secretlist))
40 40 else:
41 41 ui.status(_("no changes found\n"))
42 42
43 43 def checknewlabel(repo, lbl, kind):
44 44 if lbl in ['tip', '.', 'null']:
45 45 raise util.Abort(_("the name '%s' is reserved") % lbl)
46 46 for c in (':', '\0', '\n', '\r'):
47 47 if c in lbl:
48 48 raise util.Abort(_("%r cannot be used in a name") % c)
49 49 try:
50 50 int(lbl)
51 51 raise util.Abort(_("a %s cannot have an integer as its name") % kind)
52 52 except ValueError:
53 53 pass
54 54
55 55 def checkfilename(f):
56 56 '''Check that the filename f is an acceptable filename for a tracked file'''
57 57 if '\r' in f or '\n' in f:
58 58 raise util.Abort(_("'\\n' and '\\r' disallowed in filenames: %r") % f)
59 59
60 60 def checkportable(ui, f):
61 61 '''Check if filename f is portable and warn or abort depending on config'''
62 62 checkfilename(f)
63 63 abort, warn = checkportabilityalert(ui)
64 64 if abort or warn:
65 65 msg = util.checkwinfilename(f)
66 66 if msg:
67 67 msg = "%s: %r" % (msg, f)
68 68 if abort:
69 69 raise util.Abort(msg)
70 70 ui.warn(_("warning: %s\n") % msg)
71 71
72 72 def checkportabilityalert(ui):
73 73 '''check if the user's config requests nothing, a warning, or abort for
74 74 non-portable filenames'''
75 75 val = ui.config('ui', 'portablefilenames', 'warn')
76 76 lval = val.lower()
77 77 bval = util.parsebool(val)
78 78 abort = os.name == 'nt' or lval == 'abort'
79 79 warn = bval or lval == 'warn'
80 80 if bval is None and not (warn or abort or lval == 'ignore'):
81 81 raise error.ConfigError(
82 82 _("ui.portablefilenames value is invalid ('%s')") % val)
83 83 return abort, warn
84 84
85 85 class casecollisionauditor(object):
86 86 def __init__(self, ui, abort, dirstate):
87 87 self._ui = ui
88 88 self._abort = abort
89 89 allfiles = '\0'.join(dirstate._map)
90 90 self._loweredfiles = set(encoding.lower(allfiles).split('\0'))
91 91 self._dirstate = dirstate
92 92 # The purpose of _newfiles is so that we don't complain about
93 93 # case collisions if someone were to call this object with the
94 94 # same filename twice.
95 95 self._newfiles = set()
96 96
97 97 def __call__(self, f):
98 98 fl = encoding.lower(f)
99 99 if (fl in self._loweredfiles and f not in self._dirstate and
100 100 f not in self._newfiles):
101 101 msg = _('possible case-folding collision for %s') % f
102 102 if self._abort:
103 103 raise util.Abort(msg)
104 104 self._ui.warn(_("warning: %s\n") % msg)
105 105 self._loweredfiles.add(fl)
106 106 self._newfiles.add(f)
107 107
108 108 class pathauditor(object):
109 109 '''ensure that a filesystem path contains no banned components.
110 110 the following properties of a path are checked:
111 111
112 112 - ends with a directory separator
113 113 - under top-level .hg
114 114 - starts at the root of a windows drive
115 115 - contains ".."
116 116 - traverses a symlink (e.g. a/symlink_here/b)
117 117 - inside a nested repository (a callback can be used to approve
118 118 some nested repositories, e.g., subrepositories)
119 119 '''
120 120
121 121 def __init__(self, root, callback=None):
122 122 self.audited = set()
123 123 self.auditeddir = set()
124 124 self.root = root
125 125 self.callback = callback
126 126 if os.path.lexists(root) and not util.checkcase(root):
127 127 self.normcase = util.normcase
128 128 else:
129 129 self.normcase = lambda x: x
130 130
131 131 def __call__(self, path):
132 132 '''Check the relative path.
133 133 path may contain a pattern (e.g. foodir/**.txt)'''
134 134
135 135 path = util.localpath(path)
136 136 normpath = self.normcase(path)
137 137 if normpath in self.audited:
138 138 return
139 139 # AIX ignores "/" at end of path, others raise EISDIR.
140 140 if util.endswithsep(path):
141 141 raise util.Abort(_("path ends in directory separator: %s") % path)
142 142 parts = util.splitpath(path)
143 143 if (os.path.splitdrive(path)[0]
144 144 or parts[0].lower() in ('.hg', '.hg.', '')
145 145 or os.pardir in parts):
146 146 raise util.Abort(_("path contains illegal component: %s") % path)
147 147 if '.hg' in path.lower():
148 148 lparts = [p.lower() for p in parts]
149 149 for p in '.hg', '.hg.':
150 150 if p in lparts[1:]:
151 151 pos = lparts.index(p)
152 152 base = os.path.join(*parts[:pos])
153 153 raise util.Abort(_("path '%s' is inside nested repo %r")
154 154 % (path, base))
155 155
156 156 normparts = util.splitpath(normpath)
157 157 assert len(parts) == len(normparts)
158 158
159 159 parts.pop()
160 160 normparts.pop()
161 161 prefixes = []
162 162 while parts:
163 163 prefix = os.sep.join(parts)
164 164 normprefix = os.sep.join(normparts)
165 165 if normprefix in self.auditeddir:
166 166 break
167 167 curpath = os.path.join(self.root, prefix)
168 168 try:
169 169 st = os.lstat(curpath)
170 170 except OSError, err:
171 171 # EINVAL can be raised as invalid path syntax under win32.
172 172 # They must be ignored for patterns can be checked too.
173 173 if err.errno not in (errno.ENOENT, errno.ENOTDIR, errno.EINVAL):
174 174 raise
175 175 else:
176 176 if stat.S_ISLNK(st.st_mode):
177 177 raise util.Abort(
178 178 _('path %r traverses symbolic link %r')
179 179 % (path, prefix))
180 180 elif (stat.S_ISDIR(st.st_mode) and
181 181 os.path.isdir(os.path.join(curpath, '.hg'))):
182 182 if not self.callback or not self.callback(curpath):
183 183 raise util.Abort(_("path '%s' is inside nested "
184 184 "repo %r")
185 185 % (path, prefix))
186 186 prefixes.append(normprefix)
187 187 parts.pop()
188 188 normparts.pop()
189 189
190 190 self.audited.add(normpath)
191 191 # only add prefixes to the cache after checking everything: we don't
192 192 # want to add "foo/bar/baz" before checking if there's a "foo/.hg"
193 193 self.auditeddir.update(prefixes)
194 194
195 195 def check(self, path):
196 196 try:
197 197 self(path)
198 198 return True
199 199 except (OSError, util.Abort):
200 200 return False
201 201
202 202 class abstractvfs(object):
203 203 """Abstract base class; cannot be instantiated"""
204 204
205 205 def __init__(self, *args, **kwargs):
206 206 '''Prevent instantiation; don't call this from subclasses.'''
207 207 raise NotImplementedError('attempted instantiating ' + str(type(self)))
208 208
209 209 def tryread(self, path):
210 210 '''gracefully return an empty string for missing files'''
211 211 try:
212 212 return self.read(path)
213 213 except IOError, inst:
214 214 if inst.errno != errno.ENOENT:
215 215 raise
216 216 return ""
217 217
218 218 def read(self, path):
219 219 fp = self(path, 'rb')
220 220 try:
221 221 return fp.read()
222 222 finally:
223 223 fp.close()
224 224
225 225 def write(self, path, data):
226 226 fp = self(path, 'wb')
227 227 try:
228 228 return fp.write(data)
229 229 finally:
230 230 fp.close()
231 231
232 232 def append(self, path, data):
233 233 fp = self(path, 'ab')
234 234 try:
235 235 return fp.write(data)
236 236 finally:
237 237 fp.close()
238 238
239 239 def exists(self, path=None):
240 240 return os.path.exists(self.join(path))
241 241
242 242 def isdir(self, path=None):
243 243 return os.path.isdir(self.join(path))
244 244
245 245 def makedir(self, path=None, notindexed=True):
246 246 return util.makedir(self.join(path), notindexed)
247 247
248 248 def makedirs(self, path=None, mode=None):
249 249 return util.makedirs(self.join(path), mode)
250 250
251 251 def mkdir(self, path=None):
252 252 return os.mkdir(self.join(path))
253 253
254 254 def readdir(self, path=None, stat=None, skip=None):
255 255 return osutil.listdir(self.join(path), stat, skip)
256 256
257 257 def stat(self, path=None):
258 258 return os.stat(self.join(path))
259 259
260 260 class vfs(abstractvfs):
261 261 '''Operate files relative to a base directory
262 262
263 263 This class is used to hide the details of COW semantics and
264 264 remote file access from higher level code.
265 265 '''
266 def __init__(self, base, audit=True, expand=False):
267 if expand:
268 base = os.path.realpath(util.expandpath(base))
266 def __init__(self, base, audit=True, expandpath=False, realpath=False):
267 if expandpath:
268 base = util.expandpath(base)
269 if realpath:
270 base = os.path.realpath(base)
269 271 self.base = base
270 272 self._setmustaudit(audit)
271 273 self.createmode = None
272 274 self._trustnlink = None
273 275
274 276 def _getmustaudit(self):
275 277 return self._audit
276 278
277 279 def _setmustaudit(self, onoff):
278 280 self._audit = onoff
279 281 if onoff:
280 282 self.audit = pathauditor(self.base)
281 283 else:
282 284 self.audit = util.always
283 285
284 286 mustaudit = property(_getmustaudit, _setmustaudit)
285 287
286 288 @util.propertycache
287 289 def _cansymlink(self):
288 290 return util.checklink(self.base)
289 291
290 292 @util.propertycache
291 293 def _chmod(self):
292 294 return util.checkexec(self.base)
293 295
294 296 def _fixfilemode(self, name):
295 297 if self.createmode is None or not self._chmod:
296 298 return
297 299 os.chmod(name, self.createmode & 0666)
298 300
299 301 def __call__(self, path, mode="r", text=False, atomictemp=False):
300 302 if self._audit:
301 303 r = util.checkosfilename(path)
302 304 if r:
303 305 raise util.Abort("%s: %r" % (r, path))
304 306 self.audit(path)
305 307 f = self.join(path)
306 308
307 309 if not text and "b" not in mode:
308 310 mode += "b" # for that other OS
309 311
310 312 nlink = -1
311 313 if mode not in ('r', 'rb'):
312 314 dirname, basename = util.split(f)
313 315 # If basename is empty, then the path is malformed because it points
314 316 # to a directory. Let the posixfile() call below raise IOError.
315 317 if basename:
316 318 if atomictemp:
317 319 util.ensuredirs(dirname, self.createmode)
318 320 return util.atomictempfile(f, mode, self.createmode)
319 321 try:
320 322 if 'w' in mode:
321 323 util.unlink(f)
322 324 nlink = 0
323 325 else:
324 326 # nlinks() may behave differently for files on Windows
325 327 # shares if the file is open.
326 328 fd = util.posixfile(f)
327 329 nlink = util.nlinks(f)
328 330 if nlink < 1:
329 331 nlink = 2 # force mktempcopy (issue1922)
330 332 fd.close()
331 333 except (OSError, IOError), e:
332 334 if e.errno != errno.ENOENT:
333 335 raise
334 336 nlink = 0
335 337 util.ensuredirs(dirname, self.createmode)
336 338 if nlink > 0:
337 339 if self._trustnlink is None:
338 340 self._trustnlink = nlink > 1 or util.checknlink(f)
339 341 if nlink > 1 or not self._trustnlink:
340 342 util.rename(util.mktempcopy(f), f)
341 343 fp = util.posixfile(f, mode)
342 344 if nlink == 0:
343 345 self._fixfilemode(f)
344 346 return fp
345 347
346 348 def symlink(self, src, dst):
347 349 self.audit(dst)
348 350 linkname = self.join(dst)
349 351 try:
350 352 os.unlink(linkname)
351 353 except OSError:
352 354 pass
353 355
354 356 util.ensuredirs(os.path.dirname(linkname), self.createmode)
355 357
356 358 if self._cansymlink:
357 359 try:
358 360 os.symlink(src, linkname)
359 361 except OSError, err:
360 362 raise OSError(err.errno, _('could not symlink to %r: %s') %
361 363 (src, err.strerror), linkname)
362 364 else:
363 365 self.write(dst, src)
364 366
365 367 def join(self, path):
366 368 if path:
367 369 return os.path.join(self.base, path)
368 370 else:
369 371 return self.base
370 372
371 373 opener = vfs
372 374
373 375 class auditvfs(object):
374 376 def __init__(self, vfs):
375 377 self.vfs = vfs
376 378
377 379 def _getmustaudit(self):
378 380 return self.vfs.mustaudit
379 381
380 382 def _setmustaudit(self, onoff):
381 383 self.vfs.mustaudit = onoff
382 384
383 385 mustaudit = property(_getmustaudit, _setmustaudit)
384 386
385 387 class filtervfs(abstractvfs, auditvfs):
386 388 '''Wrapper vfs for filtering filenames with a function.'''
387 389
388 390 def __init__(self, vfs, filter):
389 391 auditvfs.__init__(self, vfs)
390 392 self._filter = filter
391 393
392 394 def __call__(self, path, *args, **kwargs):
393 395 return self.vfs(self._filter(path), *args, **kwargs)
394 396
395 397 def join(self, path):
396 398 if path:
397 399 return self.vfs.join(self._filter(path))
398 400 else:
399 401 return self.vfs.join(path)
400 402
401 403 filteropener = filtervfs
402 404
403 405 class readonlyvfs(abstractvfs, auditvfs):
404 406 '''Wrapper vfs preventing any writing.'''
405 407
406 408 def __init__(self, vfs):
407 409 auditvfs.__init__(self, vfs)
408 410
409 411 def __call__(self, path, mode='r', *args, **kw):
410 412 if mode not in ('r', 'rb'):
411 413 raise util.Abort('this vfs is read only')
412 414 return self.vfs(path, mode, *args, **kw)
413 415
414 416
415 417 def canonpath(root, cwd, myname, auditor=None):
416 418 '''return the canonical path of myname, given cwd and root'''
417 419 if util.endswithsep(root):
418 420 rootsep = root
419 421 else:
420 422 rootsep = root + os.sep
421 423 name = myname
422 424 if not os.path.isabs(name):
423 425 name = os.path.join(root, cwd, name)
424 426 name = os.path.normpath(name)
425 427 if auditor is None:
426 428 auditor = pathauditor(root)
427 429 if name != rootsep and name.startswith(rootsep):
428 430 name = name[len(rootsep):]
429 431 auditor(name)
430 432 return util.pconvert(name)
431 433 elif name == root:
432 434 return ''
433 435 else:
434 436 # Determine whether `name' is in the hierarchy at or beneath `root',
435 437 # by iterating name=dirname(name) until that causes no change (can't
436 438 # check name == '/', because that doesn't work on windows). The list
437 439 # `rel' holds the reversed list of components making up the relative
438 440 # file name we want.
439 441 rel = []
440 442 while True:
441 443 try:
442 444 s = util.samefile(name, root)
443 445 except OSError:
444 446 s = False
445 447 if s:
446 448 if not rel:
447 449 # name was actually the same as root (maybe a symlink)
448 450 return ''
449 451 rel.reverse()
450 452 name = os.path.join(*rel)
451 453 auditor(name)
452 454 return util.pconvert(name)
453 455 dirname, basename = util.split(name)
454 456 rel.append(basename)
455 457 if dirname == name:
456 458 break
457 459 name = dirname
458 460
459 461 raise util.Abort(_("%s not under root '%s'") % (myname, root))
460 462
461 463 def walkrepos(path, followsym=False, seen_dirs=None, recurse=False):
462 464 '''yield every hg repository under path, always recursively.
463 465 The recurse flag will only control recursion into repo working dirs'''
464 466 def errhandler(err):
465 467 if err.filename == path:
466 468 raise err
467 469 samestat = getattr(os.path, 'samestat', None)
468 470 if followsym and samestat is not None:
469 471 def adddir(dirlst, dirname):
470 472 match = False
471 473 dirstat = os.stat(dirname)
472 474 for lstdirstat in dirlst:
473 475 if samestat(dirstat, lstdirstat):
474 476 match = True
475 477 break
476 478 if not match:
477 479 dirlst.append(dirstat)
478 480 return not match
479 481 else:
480 482 followsym = False
481 483
482 484 if (seen_dirs is None) and followsym:
483 485 seen_dirs = []
484 486 adddir(seen_dirs, path)
485 487 for root, dirs, files in os.walk(path, topdown=True, onerror=errhandler):
486 488 dirs.sort()
487 489 if '.hg' in dirs:
488 490 yield root # found a repository
489 491 qroot = os.path.join(root, '.hg', 'patches')
490 492 if os.path.isdir(os.path.join(qroot, '.hg')):
491 493 yield qroot # we have a patch queue repo here
492 494 if recurse:
493 495 # avoid recursing inside the .hg directory
494 496 dirs.remove('.hg')
495 497 else:
496 498 dirs[:] = [] # don't descend further
497 499 elif followsym:
498 500 newdirs = []
499 501 for d in dirs:
500 502 fname = os.path.join(root, d)
501 503 if adddir(seen_dirs, fname):
502 504 if os.path.islink(fname):
503 505 for hgname in walkrepos(fname, True, seen_dirs):
504 506 yield hgname
505 507 else:
506 508 newdirs.append(d)
507 509 dirs[:] = newdirs
508 510
509 511 def osrcpath():
510 512 '''return default os-specific hgrc search path'''
511 513 path = systemrcpath()
512 514 path.extend(userrcpath())
513 515 path = [os.path.normpath(f) for f in path]
514 516 return path
515 517
516 518 _rcpath = None
517 519
518 520 def rcpath():
519 521 '''return hgrc search path. if env var HGRCPATH is set, use it.
520 522 for each item in path, if directory, use files ending in .rc,
521 523 else use item.
522 524 make HGRCPATH empty to only look in .hg/hgrc of current repo.
523 525 if no HGRCPATH, use default os-specific path.'''
524 526 global _rcpath
525 527 if _rcpath is None:
526 528 if 'HGRCPATH' in os.environ:
527 529 _rcpath = []
528 530 for p in os.environ['HGRCPATH'].split(os.pathsep):
529 531 if not p:
530 532 continue
531 533 p = util.expandpath(p)
532 534 if os.path.isdir(p):
533 535 for f, kind in osutil.listdir(p):
534 536 if f.endswith('.rc'):
535 537 _rcpath.append(os.path.join(p, f))
536 538 else:
537 539 _rcpath.append(p)
538 540 else:
539 541 _rcpath = osrcpath()
540 542 return _rcpath
541 543
542 544 def revsingle(repo, revspec, default='.'):
543 545 if not revspec:
544 546 return repo[default]
545 547
546 548 l = revrange(repo, [revspec])
547 549 if len(l) < 1:
548 550 raise util.Abort(_('empty revision set'))
549 551 return repo[l[-1]]
550 552
551 553 def revpair(repo, revs):
552 554 if not revs:
553 555 return repo.dirstate.p1(), None
554 556
555 557 l = revrange(repo, revs)
556 558
557 559 if len(l) == 0:
558 560 if revs:
559 561 raise util.Abort(_('empty revision range'))
560 562 return repo.dirstate.p1(), None
561 563
562 564 if len(l) == 1 and len(revs) == 1 and _revrangesep not in revs[0]:
563 565 return repo.lookup(l[0]), None
564 566
565 567 return repo.lookup(l[0]), repo.lookup(l[-1])
566 568
567 569 _revrangesep = ':'
568 570
569 571 def revrange(repo, revs):
570 572 """Yield revision as strings from a list of revision specifications."""
571 573
572 574 def revfix(repo, val, defval):
573 575 if not val and val != 0 and defval is not None:
574 576 return defval
575 577 return repo[val].rev()
576 578
577 579 seen, l = set(), []
578 580 for spec in revs:
579 581 if l and not seen:
580 582 seen = set(l)
581 583 # attempt to parse old-style ranges first to deal with
582 584 # things like old-tag which contain query metacharacters
583 585 try:
584 586 if isinstance(spec, int):
585 587 seen.add(spec)
586 588 l.append(spec)
587 589 continue
588 590
589 591 if _revrangesep in spec:
590 592 start, end = spec.split(_revrangesep, 1)
591 593 start = revfix(repo, start, 0)
592 594 end = revfix(repo, end, len(repo) - 1)
593 595 if end == nullrev and start <= 0:
594 596 start = nullrev
595 597 rangeiter = repo.changelog.revs(start, end)
596 598 if not seen and not l:
597 599 # by far the most common case: revs = ["-1:0"]
598 600 l = list(rangeiter)
599 601 # defer syncing seen until next iteration
600 602 continue
601 603 newrevs = set(rangeiter)
602 604 if seen:
603 605 newrevs.difference_update(seen)
604 606 seen.update(newrevs)
605 607 else:
606 608 seen = newrevs
607 609 l.extend(sorted(newrevs, reverse=start > end))
608 610 continue
609 611 elif spec and spec in repo: # single unquoted rev
610 612 rev = revfix(repo, spec, None)
611 613 if rev in seen:
612 614 continue
613 615 seen.add(rev)
614 616 l.append(rev)
615 617 continue
616 618 except error.RepoLookupError:
617 619 pass
618 620
619 621 # fall through to new-style queries if old-style fails
620 622 m = revset.match(repo.ui, spec)
621 623 dl = [r for r in m(repo, list(repo)) if r not in seen]
622 624 l.extend(dl)
623 625 seen.update(dl)
624 626
625 627 return l
626 628
627 629 def expandpats(pats):
628 630 if not util.expandglobs:
629 631 return list(pats)
630 632 ret = []
631 633 for p in pats:
632 634 kind, name = matchmod._patsplit(p, None)
633 635 if kind is None:
634 636 try:
635 637 globbed = glob.glob(name)
636 638 except re.error:
637 639 globbed = [name]
638 640 if globbed:
639 641 ret.extend(globbed)
640 642 continue
641 643 ret.append(p)
642 644 return ret
643 645
644 646 def matchandpats(ctx, pats=[], opts={}, globbed=False, default='relpath'):
645 647 if pats == ("",):
646 648 pats = []
647 649 if not globbed and default == 'relpath':
648 650 pats = expandpats(pats or [])
649 651
650 652 m = ctx.match(pats, opts.get('include'), opts.get('exclude'),
651 653 default)
652 654 def badfn(f, msg):
653 655 ctx._repo.ui.warn("%s: %s\n" % (m.rel(f), msg))
654 656 m.bad = badfn
655 657 return m, pats
656 658
657 659 def match(ctx, pats=[], opts={}, globbed=False, default='relpath'):
658 660 return matchandpats(ctx, pats, opts, globbed, default)[0]
659 661
660 662 def matchall(repo):
661 663 return matchmod.always(repo.root, repo.getcwd())
662 664
663 665 def matchfiles(repo, files):
664 666 return matchmod.exact(repo.root, repo.getcwd(), files)
665 667
666 668 def addremove(repo, pats=[], opts={}, dry_run=None, similarity=None):
667 669 if dry_run is None:
668 670 dry_run = opts.get('dry_run')
669 671 if similarity is None:
670 672 similarity = float(opts.get('similarity') or 0)
671 673 # we'd use status here, except handling of symlinks and ignore is tricky
672 674 added, unknown, deleted, removed = [], [], [], []
673 675 audit_path = pathauditor(repo.root)
674 676 m = match(repo[None], pats, opts)
675 677 rejected = []
676 678 m.bad = lambda x, y: rejected.append(x)
677 679
678 680 ctx = repo[None]
679 681 dirstate = repo.dirstate
680 682 walkresults = dirstate.walk(m, sorted(ctx.substate), True, False)
681 683 for abs, st in walkresults.iteritems():
682 684 dstate = dirstate[abs]
683 685 if dstate == '?' and audit_path.check(abs):
684 686 unknown.append(abs)
685 687 elif dstate != 'r' and not st:
686 688 deleted.append(abs)
687 689 # for finding renames
688 690 elif dstate == 'r':
689 691 removed.append(abs)
690 692 elif dstate == 'a':
691 693 added.append(abs)
692 694
693 695 unknownset = set(unknown)
694 696 toprint = unknownset.copy()
695 697 toprint.update(deleted)
696 698 for abs in sorted(toprint):
697 699 if repo.ui.verbose or not m.exact(abs):
698 700 rel = m.rel(abs)
699 701 if abs in unknownset:
700 702 status = _('adding %s\n') % ((pats and rel) or abs)
701 703 else:
702 704 status = _('removing %s\n') % ((pats and rel) or abs)
703 705 repo.ui.status(status)
704 706
705 707 copies = {}
706 708 if similarity > 0:
707 709 for old, new, score in similar.findrenames(repo,
708 710 added + unknown, removed + deleted, similarity):
709 711 if repo.ui.verbose or not m.exact(old) or not m.exact(new):
710 712 repo.ui.status(_('recording removal of %s as rename to %s '
711 713 '(%d%% similar)\n') %
712 714 (m.rel(old), m.rel(new), score * 100))
713 715 copies[new] = old
714 716
715 717 if not dry_run:
716 718 wctx = repo[None]
717 719 wlock = repo.wlock()
718 720 try:
719 721 wctx.forget(deleted)
720 722 wctx.add(unknown)
721 723 for new, old in copies.iteritems():
722 724 wctx.copy(old, new)
723 725 finally:
724 726 wlock.release()
725 727
726 728 for f in rejected:
727 729 if f in m.files():
728 730 return 1
729 731 return 0
730 732
731 733 def dirstatecopy(ui, repo, wctx, src, dst, dryrun=False, cwd=None):
732 734 """Update the dirstate to reflect the intent of copying src to dst. For
733 735 different reasons it might not end with dst being marked as copied from src.
734 736 """
735 737 origsrc = repo.dirstate.copied(src) or src
736 738 if dst == origsrc: # copying back a copy?
737 739 if repo.dirstate[dst] not in 'mn' and not dryrun:
738 740 repo.dirstate.normallookup(dst)
739 741 else:
740 742 if repo.dirstate[origsrc] == 'a' and origsrc == src:
741 743 if not ui.quiet:
742 744 ui.warn(_("%s has not been committed yet, so no copy "
743 745 "data will be stored for %s.\n")
744 746 % (repo.pathto(origsrc, cwd), repo.pathto(dst, cwd)))
745 747 if repo.dirstate[dst] in '?r' and not dryrun:
746 748 wctx.add([dst])
747 749 elif not dryrun:
748 750 wctx.copy(origsrc, dst)
749 751
750 752 def readrequires(opener, supported):
751 753 '''Reads and parses .hg/requires and checks if all entries found
752 754 are in the list of supported features.'''
753 755 requirements = set(opener.read("requires").splitlines())
754 756 missings = []
755 757 for r in requirements:
756 758 if r not in supported:
757 759 if not r or not r[0].isalnum():
758 760 raise error.RequirementError(_(".hg/requires file is corrupt"))
759 761 missings.append(r)
760 762 missings.sort()
761 763 if missings:
762 764 raise error.RequirementError(
763 765 _("unknown repository format: requires features '%s' (upgrade "
764 766 "Mercurial)") % "', '".join(missings))
765 767 return requirements
766 768
767 769 class filecacheentry(object):
768 770 def __init__(self, path, stat=True):
769 771 self.path = path
770 772 self.cachestat = None
771 773 self._cacheable = None
772 774
773 775 if stat:
774 776 self.cachestat = filecacheentry.stat(self.path)
775 777
776 778 if self.cachestat:
777 779 self._cacheable = self.cachestat.cacheable()
778 780 else:
779 781 # None means we don't know yet
780 782 self._cacheable = None
781 783
782 784 def refresh(self):
783 785 if self.cacheable():
784 786 self.cachestat = filecacheentry.stat(self.path)
785 787
786 788 def cacheable(self):
787 789 if self._cacheable is not None:
788 790 return self._cacheable
789 791
790 792 # we don't know yet, assume it is for now
791 793 return True
792 794
793 795 def changed(self):
794 796 # no point in going further if we can't cache it
795 797 if not self.cacheable():
796 798 return True
797 799
798 800 newstat = filecacheentry.stat(self.path)
799 801
800 802 # we may not know if it's cacheable yet, check again now
801 803 if newstat and self._cacheable is None:
802 804 self._cacheable = newstat.cacheable()
803 805
804 806 # check again
805 807 if not self._cacheable:
806 808 return True
807 809
808 810 if self.cachestat != newstat:
809 811 self.cachestat = newstat
810 812 return True
811 813 else:
812 814 return False
813 815
814 816 @staticmethod
815 817 def stat(path):
816 818 try:
817 819 return util.cachestat(path)
818 820 except OSError, e:
819 821 if e.errno != errno.ENOENT:
820 822 raise
821 823
822 824 class filecache(object):
823 825 '''A property like decorator that tracks a file under .hg/ for updates.
824 826
825 827 Records stat info when called in _filecache.
826 828
827 829 On subsequent calls, compares old stat info with new info, and recreates
828 830 the object when needed, updating the new stat info in _filecache.
829 831
830 832 Mercurial either atomic renames or appends for files under .hg,
831 833 so to ensure the cache is reliable we need the filesystem to be able
832 834 to tell us if a file has been replaced. If it can't, we fallback to
833 835 recreating the object on every call (essentially the same behaviour as
834 836 propertycache).'''
835 837 def __init__(self, path):
836 838 self.path = path
837 839
838 840 def join(self, obj, fname):
839 841 """Used to compute the runtime path of the cached file.
840 842
841 843 Users should subclass filecache and provide their own version of this
842 844 function to call the appropriate join function on 'obj' (an instance
843 845 of the class that its member function was decorated).
844 846 """
845 847 return obj.join(fname)
846 848
847 849 def __call__(self, func):
848 850 self.func = func
849 851 self.name = func.__name__
850 852 return self
851 853
852 854 def __get__(self, obj, type=None):
853 855 # do we need to check if the file changed?
854 856 if self.name in obj.__dict__:
855 857 assert self.name in obj._filecache, self.name
856 858 return obj.__dict__[self.name]
857 859
858 860 entry = obj._filecache.get(self.name)
859 861
860 862 if entry:
861 863 if entry.changed():
862 864 entry.obj = self.func(obj)
863 865 else:
864 866 path = self.join(obj, self.path)
865 867
866 868 # We stat -before- creating the object so our cache doesn't lie if
867 869 # a writer modified between the time we read and stat
868 870 entry = filecacheentry(path)
869 871 entry.obj = self.func(obj)
870 872
871 873 obj._filecache[self.name] = entry
872 874
873 875 obj.__dict__[self.name] = entry.obj
874 876 return entry.obj
875 877
876 878 def __set__(self, obj, value):
877 879 if self.name not in obj._filecache:
878 880 # we add an entry for the missing value because X in __dict__
879 881 # implies X in _filecache
880 882 ce = filecacheentry(self.join(obj, self.path), False)
881 883 obj._filecache[self.name] = ce
882 884 else:
883 885 ce = obj._filecache[self.name]
884 886
885 887 ce.obj = value # update cached copy
886 888 obj.__dict__[self.name] = value # update copy returned by obj.x
887 889
888 890 def __delete__(self, obj):
889 891 try:
890 892 del obj.__dict__[self.name]
891 893 except KeyError:
892 894 raise AttributeError(self.name)
893 895
894 896 class dirs(object):
895 897 '''a multiset of directory names from a dirstate or manifest'''
896 898
897 899 def __init__(self, map, skip=None):
898 900 self._dirs = {}
899 901 addpath = self.addpath
900 902 if util.safehasattr(map, 'iteritems') and skip is not None:
901 903 for f, s in map.iteritems():
902 904 if s[0] != skip:
903 905 addpath(f)
904 906 else:
905 907 for f in map:
906 908 addpath(f)
907 909
908 910 def addpath(self, path):
909 911 dirs = self._dirs
910 912 for base in finddirs(path):
911 913 if base in dirs:
912 914 dirs[base] += 1
913 915 return
914 916 dirs[base] = 1
915 917
916 918 def delpath(self, path):
917 919 dirs = self._dirs
918 920 for base in finddirs(path):
919 921 if dirs[base] > 1:
920 922 dirs[base] -= 1
921 923 return
922 924 del dirs[base]
923 925
924 926 def __iter__(self):
925 927 return self._dirs.iterkeys()
926 928
927 929 def __contains__(self, d):
928 930 return d in self._dirs
929 931
930 932 if util.safehasattr(parsers, 'dirs'):
931 933 dirs = parsers.dirs
932 934
933 935 def finddirs(path):
934 936 pos = path.rfind('/')
935 937 while pos != -1:
936 938 yield path[:pos]
937 939 pos = path.rfind('/', 0, pos)
General Comments 0
You need to be logged in to leave comments. Login now