##// END OF EJS Templates
status: check cmp list in order
Matt Mackall -
r8395:475552c0 default
parent child Browse files
Show More
@@ -1,2172 +1,2172 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, incorporated herein by reference.
7 7
8 8 from node import bin, hex, nullid, nullrev, short
9 9 from i18n import _
10 10 import repo, changegroup
11 11 import changelog, dirstate, filelog, manifest, context
12 12 import lock, transaction, store, encoding
13 13 import util, extensions, hook, error
14 14 import match as match_
15 15 import merge as merge_
16 16 from lock import release
17 17 import weakref, stat, errno, os, time, inspect
18 18 propertycache = util.propertycache
19 19
20 20 class localrepository(repo.repository):
21 21 capabilities = set(('lookup', 'changegroupsubset'))
22 22 supported = set('revlogv1 store fncache'.split())
23 23
24 24 def __init__(self, baseui, path=None, create=0):
25 25 repo.repository.__init__(self)
26 26 self.root = os.path.realpath(path)
27 27 self.path = os.path.join(self.root, ".hg")
28 28 self.origroot = path
29 29 self.opener = util.opener(self.path)
30 30 self.wopener = util.opener(self.root)
31 31
32 32 if not os.path.isdir(self.path):
33 33 if create:
34 34 if not os.path.exists(path):
35 35 os.mkdir(path)
36 36 os.mkdir(self.path)
37 37 requirements = ["revlogv1"]
38 38 if baseui.configbool('format', 'usestore', True):
39 39 os.mkdir(os.path.join(self.path, "store"))
40 40 requirements.append("store")
41 41 if baseui.configbool('format', 'usefncache', True):
42 42 requirements.append("fncache")
43 43 # create an invalid changelog
44 44 self.opener("00changelog.i", "a").write(
45 45 '\0\0\0\2' # represents revlogv2
46 46 ' dummy changelog to prevent using the old repo layout'
47 47 )
48 48 reqfile = self.opener("requires", "w")
49 49 for r in requirements:
50 50 reqfile.write("%s\n" % r)
51 51 reqfile.close()
52 52 else:
53 53 raise error.RepoError(_("repository %s not found") % path)
54 54 elif create:
55 55 raise error.RepoError(_("repository %s already exists") % path)
56 56 else:
57 57 # find requirements
58 58 requirements = set()
59 59 try:
60 60 requirements = set(self.opener("requires").read().splitlines())
61 61 except IOError, inst:
62 62 if inst.errno != errno.ENOENT:
63 63 raise
64 64 for r in requirements - self.supported:
65 65 raise error.RepoError(_("requirement '%s' not supported") % r)
66 66
67 67 self.store = store.store(requirements, self.path, util.opener)
68 68 self.spath = self.store.path
69 69 self.sopener = self.store.opener
70 70 self.sjoin = self.store.join
71 71 self.opener.createmode = self.store.createmode
72 72
73 73 self.baseui = baseui
74 74 self.ui = baseui.copy()
75 75 try:
76 76 self.ui.readconfig(self.join("hgrc"), self.root)
77 77 extensions.loadall(self.ui)
78 78 except IOError:
79 79 pass
80 80
81 81 self.tagscache = None
82 82 self._tagstypecache = None
83 83 self.branchcache = None
84 84 self._ubranchcache = None # UTF-8 version of branchcache
85 85 self._branchcachetip = None
86 86 self.nodetagscache = None
87 87 self.filterpats = {}
88 88 self._datafilters = {}
89 89 self._transref = self._lockref = self._wlockref = None
90 90
91 91 @propertycache
92 92 def changelog(self):
93 93 c = changelog.changelog(self.sopener)
94 94 if 'HG_PENDING' in os.environ:
95 95 p = os.environ['HG_PENDING']
96 96 if p.startswith(self.root):
97 97 c.readpending('00changelog.i.a')
98 98 self.sopener.defversion = c.version
99 99 return c
100 100
101 101 @propertycache
102 102 def manifest(self):
103 103 return manifest.manifest(self.sopener)
104 104
105 105 @propertycache
106 106 def dirstate(self):
107 107 return dirstate.dirstate(self.opener, self.ui, self.root)
108 108
109 109 def __getitem__(self, changeid):
110 110 if changeid == None:
111 111 return context.workingctx(self)
112 112 return context.changectx(self, changeid)
113 113
114 114 def __nonzero__(self):
115 115 return True
116 116
117 117 def __len__(self):
118 118 return len(self.changelog)
119 119
120 120 def __iter__(self):
121 121 for i in xrange(len(self)):
122 122 yield i
123 123
124 124 def url(self):
125 125 return 'file:' + self.root
126 126
127 127 def hook(self, name, throw=False, **args):
128 128 return hook.hook(self.ui, self, name, throw, **args)
129 129
130 130 tag_disallowed = ':\r\n'
131 131
132 132 def _tag(self, names, node, message, local, user, date, parent=None,
133 133 extra={}):
134 134 use_dirstate = parent is None
135 135
136 136 if isinstance(names, str):
137 137 allchars = names
138 138 names = (names,)
139 139 else:
140 140 allchars = ''.join(names)
141 141 for c in self.tag_disallowed:
142 142 if c in allchars:
143 143 raise util.Abort(_('%r cannot be used in a tag name') % c)
144 144
145 145 for name in names:
146 146 self.hook('pretag', throw=True, node=hex(node), tag=name,
147 147 local=local)
148 148
149 149 def writetags(fp, names, munge, prevtags):
150 150 fp.seek(0, 2)
151 151 if prevtags and prevtags[-1] != '\n':
152 152 fp.write('\n')
153 153 for name in names:
154 154 m = munge and munge(name) or name
155 155 if self._tagstypecache and name in self._tagstypecache:
156 156 old = self.tagscache.get(name, nullid)
157 157 fp.write('%s %s\n' % (hex(old), m))
158 158 fp.write('%s %s\n' % (hex(node), m))
159 159 fp.close()
160 160
161 161 prevtags = ''
162 162 if local:
163 163 try:
164 164 fp = self.opener('localtags', 'r+')
165 165 except IOError:
166 166 fp = self.opener('localtags', 'a')
167 167 else:
168 168 prevtags = fp.read()
169 169
170 170 # local tags are stored in the current charset
171 171 writetags(fp, names, None, prevtags)
172 172 for name in names:
173 173 self.hook('tag', node=hex(node), tag=name, local=local)
174 174 return
175 175
176 176 if use_dirstate:
177 177 try:
178 178 fp = self.wfile('.hgtags', 'rb+')
179 179 except IOError:
180 180 fp = self.wfile('.hgtags', 'ab')
181 181 else:
182 182 prevtags = fp.read()
183 183 else:
184 184 try:
185 185 prevtags = self.filectx('.hgtags', parent).data()
186 186 except error.LookupError:
187 187 pass
188 188 fp = self.wfile('.hgtags', 'wb')
189 189 if prevtags:
190 190 fp.write(prevtags)
191 191
192 192 # committed tags are stored in UTF-8
193 193 writetags(fp, names, encoding.fromlocal, prevtags)
194 194
195 195 if use_dirstate and '.hgtags' not in self.dirstate:
196 196 self.add(['.hgtags'])
197 197
198 198 tagnode = self.commit(['.hgtags'], message, user, date, p1=parent,
199 199 extra=extra)
200 200
201 201 for name in names:
202 202 self.hook('tag', node=hex(node), tag=name, local=local)
203 203
204 204 return tagnode
205 205
206 206 def tag(self, names, node, message, local, user, date):
207 207 '''tag a revision with one or more symbolic names.
208 208
209 209 names is a list of strings or, when adding a single tag, names may be a
210 210 string.
211 211
212 212 if local is True, the tags are stored in a per-repository file.
213 213 otherwise, they are stored in the .hgtags file, and a new
214 214 changeset is committed with the change.
215 215
216 216 keyword arguments:
217 217
218 218 local: whether to store tags in non-version-controlled file
219 219 (default False)
220 220
221 221 message: commit message to use if committing
222 222
223 223 user: name of user to use if committing
224 224
225 225 date: date tuple to use if committing'''
226 226
227 227 for x in self.status()[:5]:
228 228 if '.hgtags' in x:
229 229 raise util.Abort(_('working copy of .hgtags is changed '
230 230 '(please commit .hgtags manually)'))
231 231
232 232 self.tags() # instantiate the cache
233 233 self._tag(names, node, message, local, user, date)
234 234
235 235 def tags(self):
236 236 '''return a mapping of tag to node'''
237 237 if self.tagscache:
238 238 return self.tagscache
239 239
240 240 globaltags = {}
241 241 tagtypes = {}
242 242
243 243 def readtags(lines, fn, tagtype):
244 244 filetags = {}
245 245 count = 0
246 246
247 247 def warn(msg):
248 248 self.ui.warn(_("%s, line %s: %s\n") % (fn, count, msg))
249 249
250 250 for l in lines:
251 251 count += 1
252 252 if not l:
253 253 continue
254 254 s = l.split(" ", 1)
255 255 if len(s) != 2:
256 256 warn(_("cannot parse entry"))
257 257 continue
258 258 node, key = s
259 259 key = encoding.tolocal(key.strip()) # stored in UTF-8
260 260 try:
261 261 bin_n = bin(node)
262 262 except TypeError:
263 263 warn(_("node '%s' is not well formed") % node)
264 264 continue
265 265 if bin_n not in self.changelog.nodemap:
266 266 warn(_("tag '%s' refers to unknown node") % key)
267 267 continue
268 268
269 269 h = []
270 270 if key in filetags:
271 271 n, h = filetags[key]
272 272 h.append(n)
273 273 filetags[key] = (bin_n, h)
274 274
275 275 for k, nh in filetags.iteritems():
276 276 if k not in globaltags:
277 277 globaltags[k] = nh
278 278 tagtypes[k] = tagtype
279 279 continue
280 280
281 281 # we prefer the global tag if:
282 282 # it supercedes us OR
283 283 # mutual supercedes and it has a higher rank
284 284 # otherwise we win because we're tip-most
285 285 an, ah = nh
286 286 bn, bh = globaltags[k]
287 287 if (bn != an and an in bh and
288 288 (bn not in ah or len(bh) > len(ah))):
289 289 an = bn
290 290 ah.extend([n for n in bh if n not in ah])
291 291 globaltags[k] = an, ah
292 292 tagtypes[k] = tagtype
293 293
294 294 # read the tags file from each head, ending with the tip
295 295 f = None
296 296 for rev, node, fnode in self._hgtagsnodes():
297 297 f = (f and f.filectx(fnode) or
298 298 self.filectx('.hgtags', fileid=fnode))
299 299 readtags(f.data().splitlines(), f, "global")
300 300
301 301 try:
302 302 data = encoding.fromlocal(self.opener("localtags").read())
303 303 # localtags are stored in the local character set
304 304 # while the internal tag table is stored in UTF-8
305 305 readtags(data.splitlines(), "localtags", "local")
306 306 except IOError:
307 307 pass
308 308
309 309 self.tagscache = {}
310 310 self._tagstypecache = {}
311 311 for k, nh in globaltags.iteritems():
312 312 n = nh[0]
313 313 if n != nullid:
314 314 self.tagscache[k] = n
315 315 self._tagstypecache[k] = tagtypes[k]
316 316 self.tagscache['tip'] = self.changelog.tip()
317 317 return self.tagscache
318 318
319 319 def tagtype(self, tagname):
320 320 '''
321 321 return the type of the given tag. result can be:
322 322
323 323 'local' : a local tag
324 324 'global' : a global tag
325 325 None : tag does not exist
326 326 '''
327 327
328 328 self.tags()
329 329
330 330 return self._tagstypecache.get(tagname)
331 331
332 332 def _hgtagsnodes(self):
333 333 last = {}
334 334 ret = []
335 335 for node in reversed(self.heads()):
336 336 c = self[node]
337 337 rev = c.rev()
338 338 try:
339 339 fnode = c.filenode('.hgtags')
340 340 except error.LookupError:
341 341 continue
342 342 ret.append((rev, node, fnode))
343 343 if fnode in last:
344 344 ret[last[fnode]] = None
345 345 last[fnode] = len(ret) - 1
346 346 return [item for item in ret if item]
347 347
348 348 def tagslist(self):
349 349 '''return a list of tags ordered by revision'''
350 350 l = []
351 351 for t, n in self.tags().iteritems():
352 352 try:
353 353 r = self.changelog.rev(n)
354 354 except:
355 355 r = -2 # sort to the beginning of the list if unknown
356 356 l.append((r, t, n))
357 357 return [(t, n) for r, t, n in sorted(l)]
358 358
359 359 def nodetags(self, node):
360 360 '''return the tags associated with a node'''
361 361 if not self.nodetagscache:
362 362 self.nodetagscache = {}
363 363 for t, n in self.tags().iteritems():
364 364 self.nodetagscache.setdefault(n, []).append(t)
365 365 return self.nodetagscache.get(node, [])
366 366
367 367 def _branchtags(self, partial, lrev):
368 368 # TODO: rename this function?
369 369 tiprev = len(self) - 1
370 370 if lrev != tiprev:
371 371 self._updatebranchcache(partial, lrev+1, tiprev+1)
372 372 self._writebranchcache(partial, self.changelog.tip(), tiprev)
373 373
374 374 return partial
375 375
376 376 def _branchheads(self):
377 377 tip = self.changelog.tip()
378 378 if self.branchcache is not None and self._branchcachetip == tip:
379 379 return self.branchcache
380 380
381 381 oldtip = self._branchcachetip
382 382 self._branchcachetip = tip
383 383 if self.branchcache is None:
384 384 self.branchcache = {} # avoid recursion in changectx
385 385 else:
386 386 self.branchcache.clear() # keep using the same dict
387 387 if oldtip is None or oldtip not in self.changelog.nodemap:
388 388 partial, last, lrev = self._readbranchcache()
389 389 else:
390 390 lrev = self.changelog.rev(oldtip)
391 391 partial = self._ubranchcache
392 392
393 393 self._branchtags(partial, lrev)
394 394 # this private cache holds all heads (not just tips)
395 395 self._ubranchcache = partial
396 396
397 397 # the branch cache is stored on disk as UTF-8, but in the local
398 398 # charset internally
399 399 for k, v in partial.iteritems():
400 400 self.branchcache[encoding.tolocal(k)] = v
401 401 return self.branchcache
402 402
403 403
404 404 def branchtags(self):
405 405 '''return a dict where branch names map to the tipmost head of
406 406 the branch, open heads come before closed'''
407 407 bt = {}
408 408 for bn, heads in self._branchheads().iteritems():
409 409 head = None
410 410 for i in range(len(heads)-1, -1, -1):
411 411 h = heads[i]
412 412 if 'close' not in self.changelog.read(h)[5]:
413 413 head = h
414 414 break
415 415 # no open heads were found
416 416 if head is None:
417 417 head = heads[-1]
418 418 bt[bn] = head
419 419 return bt
420 420
421 421
422 422 def _readbranchcache(self):
423 423 partial = {}
424 424 try:
425 425 f = self.opener("branchheads.cache")
426 426 lines = f.read().split('\n')
427 427 f.close()
428 428 except (IOError, OSError):
429 429 return {}, nullid, nullrev
430 430
431 431 try:
432 432 last, lrev = lines.pop(0).split(" ", 1)
433 433 last, lrev = bin(last), int(lrev)
434 434 if lrev >= len(self) or self[lrev].node() != last:
435 435 # invalidate the cache
436 436 raise ValueError('invalidating branch cache (tip differs)')
437 437 for l in lines:
438 438 if not l: continue
439 439 node, label = l.split(" ", 1)
440 440 partial.setdefault(label.strip(), []).append(bin(node))
441 441 except KeyboardInterrupt:
442 442 raise
443 443 except Exception, inst:
444 444 if self.ui.debugflag:
445 445 self.ui.warn(str(inst), '\n')
446 446 partial, last, lrev = {}, nullid, nullrev
447 447 return partial, last, lrev
448 448
449 449 def _writebranchcache(self, branches, tip, tiprev):
450 450 try:
451 451 f = self.opener("branchheads.cache", "w", atomictemp=True)
452 452 f.write("%s %s\n" % (hex(tip), tiprev))
453 453 for label, nodes in branches.iteritems():
454 454 for node in nodes:
455 455 f.write("%s %s\n" % (hex(node), label))
456 456 f.rename()
457 457 except (IOError, OSError):
458 458 pass
459 459
460 460 def _updatebranchcache(self, partial, start, end):
461 461 for r in xrange(start, end):
462 462 c = self[r]
463 463 b = c.branch()
464 464 bheads = partial.setdefault(b, [])
465 465 bheads.append(c.node())
466 466 for p in c.parents():
467 467 pn = p.node()
468 468 if pn in bheads:
469 469 bheads.remove(pn)
470 470
471 471 def lookup(self, key):
472 472 if isinstance(key, int):
473 473 return self.changelog.node(key)
474 474 elif key == '.':
475 475 return self.dirstate.parents()[0]
476 476 elif key == 'null':
477 477 return nullid
478 478 elif key == 'tip':
479 479 return self.changelog.tip()
480 480 n = self.changelog._match(key)
481 481 if n:
482 482 return n
483 483 if key in self.tags():
484 484 return self.tags()[key]
485 485 if key in self.branchtags():
486 486 return self.branchtags()[key]
487 487 n = self.changelog._partialmatch(key)
488 488 if n:
489 489 return n
490 490 try:
491 491 if len(key) == 20:
492 492 key = hex(key)
493 493 except:
494 494 pass
495 495 raise error.RepoError(_("unknown revision '%s'") % key)
496 496
497 497 def local(self):
498 498 return True
499 499
500 500 def join(self, f):
501 501 return os.path.join(self.path, f)
502 502
503 503 def wjoin(self, f):
504 504 return os.path.join(self.root, f)
505 505
506 506 def rjoin(self, f):
507 507 return os.path.join(self.root, util.pconvert(f))
508 508
509 509 def file(self, f):
510 510 if f[0] == '/':
511 511 f = f[1:]
512 512 return filelog.filelog(self.sopener, f)
513 513
514 514 def changectx(self, changeid):
515 515 return self[changeid]
516 516
517 517 def parents(self, changeid=None):
518 518 '''get list of changectxs for parents of changeid'''
519 519 return self[changeid].parents()
520 520
521 521 def filectx(self, path, changeid=None, fileid=None):
522 522 """changeid can be a changeset revision, node, or tag.
523 523 fileid can be a file revision or node."""
524 524 return context.filectx(self, path, changeid, fileid)
525 525
526 526 def getcwd(self):
527 527 return self.dirstate.getcwd()
528 528
529 529 def pathto(self, f, cwd=None):
530 530 return self.dirstate.pathto(f, cwd)
531 531
532 532 def wfile(self, f, mode='r'):
533 533 return self.wopener(f, mode)
534 534
535 535 def _link(self, f):
536 536 return os.path.islink(self.wjoin(f))
537 537
538 538 def _filter(self, filter, filename, data):
539 539 if filter not in self.filterpats:
540 540 l = []
541 541 for pat, cmd in self.ui.configitems(filter):
542 542 if cmd == '!':
543 543 continue
544 544 mf = util.matcher(self.root, "", [pat], [], [])[1]
545 545 fn = None
546 546 params = cmd
547 547 for name, filterfn in self._datafilters.iteritems():
548 548 if cmd.startswith(name):
549 549 fn = filterfn
550 550 params = cmd[len(name):].lstrip()
551 551 break
552 552 if not fn:
553 553 fn = lambda s, c, **kwargs: util.filter(s, c)
554 554 # Wrap old filters not supporting keyword arguments
555 555 if not inspect.getargspec(fn)[2]:
556 556 oldfn = fn
557 557 fn = lambda s, c, **kwargs: oldfn(s, c)
558 558 l.append((mf, fn, params))
559 559 self.filterpats[filter] = l
560 560
561 561 for mf, fn, cmd in self.filterpats[filter]:
562 562 if mf(filename):
563 563 self.ui.debug(_("filtering %s through %s\n") % (filename, cmd))
564 564 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
565 565 break
566 566
567 567 return data
568 568
569 569 def adddatafilter(self, name, filter):
570 570 self._datafilters[name] = filter
571 571
572 572 def wread(self, filename):
573 573 if self._link(filename):
574 574 data = os.readlink(self.wjoin(filename))
575 575 else:
576 576 data = self.wopener(filename, 'r').read()
577 577 return self._filter("encode", filename, data)
578 578
579 579 def wwrite(self, filename, data, flags):
580 580 data = self._filter("decode", filename, data)
581 581 try:
582 582 os.unlink(self.wjoin(filename))
583 583 except OSError:
584 584 pass
585 585 if 'l' in flags:
586 586 self.wopener.symlink(data, filename)
587 587 else:
588 588 self.wopener(filename, 'w').write(data)
589 589 if 'x' in flags:
590 590 util.set_flags(self.wjoin(filename), False, True)
591 591
592 592 def wwritedata(self, filename, data):
593 593 return self._filter("decode", filename, data)
594 594
595 595 def transaction(self):
596 596 tr = self._transref and self._transref() or None
597 597 if tr and tr.running():
598 598 return tr.nest()
599 599
600 600 # abort here if the journal already exists
601 601 if os.path.exists(self.sjoin("journal")):
602 602 raise error.RepoError(_("journal already exists - run hg recover"))
603 603
604 604 # save dirstate for rollback
605 605 try:
606 606 ds = self.opener("dirstate").read()
607 607 except IOError:
608 608 ds = ""
609 609 self.opener("journal.dirstate", "w").write(ds)
610 610 self.opener("journal.branch", "w").write(self.dirstate.branch())
611 611
612 612 renames = [(self.sjoin("journal"), self.sjoin("undo")),
613 613 (self.join("journal.dirstate"), self.join("undo.dirstate")),
614 614 (self.join("journal.branch"), self.join("undo.branch"))]
615 615 tr = transaction.transaction(self.ui.warn, self.sopener,
616 616 self.sjoin("journal"),
617 617 aftertrans(renames),
618 618 self.store.createmode)
619 619 self._transref = weakref.ref(tr)
620 620 return tr
621 621
622 622 def recover(self):
623 623 lock = self.lock()
624 624 try:
625 625 if os.path.exists(self.sjoin("journal")):
626 626 self.ui.status(_("rolling back interrupted transaction\n"))
627 627 transaction.rollback(self.sopener, self.sjoin("journal"), self.ui.warn)
628 628 self.invalidate()
629 629 return True
630 630 else:
631 631 self.ui.warn(_("no interrupted transaction available\n"))
632 632 return False
633 633 finally:
634 634 lock.release()
635 635
636 636 def rollback(self):
637 637 wlock = lock = None
638 638 try:
639 639 wlock = self.wlock()
640 640 lock = self.lock()
641 641 if os.path.exists(self.sjoin("undo")):
642 642 self.ui.status(_("rolling back last transaction\n"))
643 643 transaction.rollback(self.sopener, self.sjoin("undo"), self.ui.warn)
644 644 util.rename(self.join("undo.dirstate"), self.join("dirstate"))
645 645 try:
646 646 branch = self.opener("undo.branch").read()
647 647 self.dirstate.setbranch(branch)
648 648 except IOError:
649 649 self.ui.warn(_("Named branch could not be reset, "
650 650 "current branch still is: %s\n")
651 651 % encoding.tolocal(self.dirstate.branch()))
652 652 self.invalidate()
653 653 self.dirstate.invalidate()
654 654 else:
655 655 self.ui.warn(_("no rollback information available\n"))
656 656 finally:
657 657 release(lock, wlock)
658 658
659 659 def invalidate(self):
660 660 for a in "changelog manifest".split():
661 661 if a in self.__dict__:
662 662 delattr(self, a)
663 663 self.tagscache = None
664 664 self._tagstypecache = None
665 665 self.nodetagscache = None
666 666 self.branchcache = None
667 667 self._ubranchcache = None
668 668 self._branchcachetip = None
669 669
670 670 def _lock(self, lockname, wait, releasefn, acquirefn, desc):
671 671 try:
672 672 l = lock.lock(lockname, 0, releasefn, desc=desc)
673 673 except error.LockHeld, inst:
674 674 if not wait:
675 675 raise
676 676 self.ui.warn(_("waiting for lock on %s held by %r\n") %
677 677 (desc, inst.locker))
678 678 # default to 600 seconds timeout
679 679 l = lock.lock(lockname, int(self.ui.config("ui", "timeout", "600")),
680 680 releasefn, desc=desc)
681 681 if acquirefn:
682 682 acquirefn()
683 683 return l
684 684
685 685 def lock(self, wait=True):
686 686 l = self._lockref and self._lockref()
687 687 if l is not None and l.held:
688 688 l.lock()
689 689 return l
690 690
691 691 l = self._lock(self.sjoin("lock"), wait, None, self.invalidate,
692 692 _('repository %s') % self.origroot)
693 693 self._lockref = weakref.ref(l)
694 694 return l
695 695
696 696 def wlock(self, wait=True):
697 697 l = self._wlockref and self._wlockref()
698 698 if l is not None and l.held:
699 699 l.lock()
700 700 return l
701 701
702 702 l = self._lock(self.join("wlock"), wait, self.dirstate.write,
703 703 self.dirstate.invalidate, _('working directory of %s') %
704 704 self.origroot)
705 705 self._wlockref = weakref.ref(l)
706 706 return l
707 707
708 708 def filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
709 709 """
710 710 commit an individual file as part of a larger transaction
711 711 """
712 712
713 713 fname = fctx.path()
714 714 text = fctx.data()
715 715 flog = self.file(fname)
716 716 fparent1 = manifest1.get(fname, nullid)
717 717 fparent2 = manifest2.get(fname, nullid)
718 718
719 719 meta = {}
720 720 copy = fctx.renamed()
721 721 if copy and copy[0] != fname:
722 722 # Mark the new revision of this file as a copy of another
723 723 # file. This copy data will effectively act as a parent
724 724 # of this new revision. If this is a merge, the first
725 725 # parent will be the nullid (meaning "look up the copy data")
726 726 # and the second one will be the other parent. For example:
727 727 #
728 728 # 0 --- 1 --- 3 rev1 changes file foo
729 729 # \ / rev2 renames foo to bar and changes it
730 730 # \- 2 -/ rev3 should have bar with all changes and
731 731 # should record that bar descends from
732 732 # bar in rev2 and foo in rev1
733 733 #
734 734 # this allows this merge to succeed:
735 735 #
736 736 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
737 737 # \ / merging rev3 and rev4 should use bar@rev2
738 738 # \- 2 --- 4 as the merge base
739 739 #
740 740
741 741 cfname = copy[0]
742 742 crev = manifest1.get(cfname)
743 743 newfparent = fparent2
744 744
745 745 if manifest2: # branch merge
746 746 if fparent2 == nullid or crev is None: # copied on remote side
747 747 if cfname in manifest2:
748 748 crev = manifest2[cfname]
749 749 newfparent = fparent1
750 750
751 751 # find source in nearest ancestor if we've lost track
752 752 if not crev:
753 753 self.ui.debug(_(" %s: searching for copy revision for %s\n") %
754 754 (fname, cfname))
755 755 for ancestor in self['.'].ancestors():
756 756 if cfname in ancestor:
757 757 crev = ancestor[cfname].filenode()
758 758 break
759 759
760 760 self.ui.debug(_(" %s: copy %s:%s\n") % (fname, cfname, hex(crev)))
761 761 meta["copy"] = cfname
762 762 meta["copyrev"] = hex(crev)
763 763 fparent1, fparent2 = nullid, newfparent
764 764 elif fparent2 != nullid:
765 765 # is one parent an ancestor of the other?
766 766 fparentancestor = flog.ancestor(fparent1, fparent2)
767 767 if fparentancestor == fparent1:
768 768 fparent1, fparent2 = fparent2, nullid
769 769 elif fparentancestor == fparent2:
770 770 fparent2 = nullid
771 771
772 772 # is the file unmodified from the parent? report existing entry
773 773 if fparent2 == nullid and not flog.cmp(fparent1, text) and not meta:
774 774 return fparent1
775 775
776 776 changelist.append(fname)
777 777 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
778 778
779 779 def rawcommit(self, files, text, user, date, p1=None, p2=None, extra={}):
780 780 if p1 is None:
781 781 p1, p2 = self.dirstate.parents()
782 782 return self.commit(files=files, text=text, user=user, date=date,
783 783 p1=p1, p2=p2, extra=extra, empty_ok=True)
784 784
785 785 def commit(self, files=None, text="", user=None, date=None,
786 786 match=None, force=False, force_editor=False,
787 787 p1=None, p2=None, extra={}, empty_ok=False):
788 788 wlock = lock = None
789 789 if extra.get("close"):
790 790 force = True
791 791 if files:
792 792 files = list(set(files))
793 793 try:
794 794 wlock = self.wlock()
795 795 lock = self.lock()
796 796 use_dirstate = (p1 is None) # not rawcommit
797 797
798 798 if use_dirstate:
799 799 p1, p2 = self.dirstate.parents()
800 800 update_dirstate = True
801 801
802 802 if (not force and p2 != nullid and
803 803 (match and (match.files() or match.anypats()))):
804 804 raise util.Abort(_('cannot partially commit a merge '
805 805 '(do not specify files or patterns)'))
806 806
807 807 if files:
808 808 modified, removed = [], []
809 809 for f in files:
810 810 s = self.dirstate[f]
811 811 if s in 'nma':
812 812 modified.append(f)
813 813 elif s == 'r':
814 814 removed.append(f)
815 815 else:
816 816 self.ui.warn(_("%s not tracked!\n") % f)
817 817 changes = [modified, [], removed, [], []]
818 818 else:
819 819 changes = self.status(match=match)
820 820 else:
821 821 p1, p2 = p1, p2 or nullid
822 822 update_dirstate = (self.dirstate.parents()[0] == p1)
823 823 changes = [files, [], [], [], []]
824 824
825 825 ms = merge_.mergestate(self)
826 826 for f in changes[0]:
827 827 if f in ms and ms[f] == 'u':
828 828 raise util.Abort(_("unresolved merge conflicts "
829 829 "(see hg resolve)"))
830 830 wctx = context.workingctx(self, (p1, p2), text, user, date,
831 831 extra, changes)
832 832 r = self._commitctx(wctx, force, force_editor, empty_ok,
833 833 use_dirstate, update_dirstate)
834 834 ms.reset()
835 835 return r
836 836
837 837 finally:
838 838 release(lock, wlock)
839 839
840 840 def commitctx(self, ctx):
841 841 """Add a new revision to current repository.
842 842
843 843 Revision information is passed in the context.memctx argument.
844 844 commitctx() does not touch the working directory.
845 845 """
846 846 wlock = lock = None
847 847 try:
848 848 wlock = self.wlock()
849 849 lock = self.lock()
850 850 return self._commitctx(ctx, force=True, force_editor=False,
851 851 empty_ok=True, use_dirstate=False,
852 852 update_dirstate=False)
853 853 finally:
854 854 release(lock, wlock)
855 855
856 856 def _commitctx(self, wctx, force=False, force_editor=False, empty_ok=False,
857 857 use_dirstate=True, update_dirstate=True):
858 858 tr = None
859 859 valid = 0 # don't save the dirstate if this isn't set
860 860 try:
861 861 commit = sorted(wctx.modified() + wctx.added())
862 862 remove = wctx.removed()
863 863 extra = wctx.extra().copy()
864 864 branchname = extra['branch']
865 865 user = wctx.user()
866 866 text = wctx.description()
867 867
868 868 p1, p2 = [p.node() for p in wctx.parents()]
869 869 c1 = self.changelog.read(p1)
870 870 c2 = self.changelog.read(p2)
871 871 m1 = self.manifest.read(c1[0]).copy()
872 872 m2 = self.manifest.read(c2[0])
873 873
874 874 if use_dirstate:
875 875 oldname = c1[5].get("branch") # stored in UTF-8
876 876 if (not commit and not remove and not force and p2 == nullid
877 877 and branchname == oldname):
878 878 self.ui.status(_("nothing changed\n"))
879 879 return None
880 880
881 881 xp1 = hex(p1)
882 882 if p2 == nullid: xp2 = ''
883 883 else: xp2 = hex(p2)
884 884
885 885 self.hook("precommit", throw=True, parent1=xp1, parent2=xp2)
886 886
887 887 tr = self.transaction()
888 888 trp = weakref.proxy(tr)
889 889
890 890 # check in files
891 891 new = {}
892 892 changed = []
893 893 linkrev = len(self)
894 894 for f in commit:
895 895 self.ui.note(f + "\n")
896 896 try:
897 897 fctx = wctx.filectx(f)
898 898 newflags = fctx.flags()
899 899 new[f] = self.filecommit(fctx, m1, m2, linkrev, trp, changed)
900 900 if ((not changed or changed[-1] != f) and
901 901 m2.get(f) != new[f]):
902 902 # mention the file in the changelog if some
903 903 # flag changed, even if there was no content
904 904 # change.
905 905 if m1.flags(f) != newflags:
906 906 changed.append(f)
907 907 m1.set(f, newflags)
908 908 if use_dirstate:
909 909 self.dirstate.normal(f)
910 910
911 911 except (OSError, IOError):
912 912 if use_dirstate:
913 913 self.ui.warn(_("trouble committing %s!\n") % f)
914 914 raise
915 915 else:
916 916 remove.append(f)
917 917
918 918 updated, added = [], []
919 919 for f in sorted(changed):
920 920 if f in m1 or f in m2:
921 921 updated.append(f)
922 922 else:
923 923 added.append(f)
924 924
925 925 # update manifest
926 926 m1.update(new)
927 927 removed = [f for f in sorted(remove) if f in m1 or f in m2]
928 928 removed1 = []
929 929
930 930 for f in removed:
931 931 if f in m1:
932 932 del m1[f]
933 933 removed1.append(f)
934 934 mn = self.manifest.add(m1, trp, linkrev, c1[0], c2[0],
935 935 (new, removed1))
936 936
937 937 # add changeset
938 938 if (not empty_ok and not text) or force_editor:
939 939 edittext = []
940 940 if text:
941 941 edittext.append(text)
942 942 edittext.append("")
943 943 edittext.append("") # Empty line between message and comments.
944 944 edittext.append(_("HG: Enter commit message."
945 945 " Lines beginning with 'HG:' are removed."))
946 946 edittext.append("HG: --")
947 947 edittext.append(_("HG: user: %s") % user)
948 948 if p2 != nullid:
949 949 edittext.append(_("HG: branch merge"))
950 950 if branchname:
951 951 edittext.append(_("HG: branch '%s'")
952 952 % encoding.tolocal(branchname))
953 953 edittext.extend([_("HG: added %s") % f for f in added])
954 954 edittext.extend([_("HG: changed %s") % f for f in updated])
955 955 edittext.extend([_("HG: removed %s") % f for f in removed])
956 956 if not added and not updated and not removed:
957 957 edittext.append(_("HG: no files changed"))
958 958 edittext.append("")
959 959 # run editor in the repository root
960 960 olddir = os.getcwd()
961 961 os.chdir(self.root)
962 962 text = self.ui.edit("\n".join(edittext), user)
963 963 os.chdir(olddir)
964 964
965 965 lines = [line.rstrip() for line in text.rstrip().splitlines()]
966 966 while lines and not lines[0]:
967 967 del lines[0]
968 968 if not lines and use_dirstate:
969 969 raise util.Abort(_("empty commit message"))
970 970 text = '\n'.join(lines)
971 971
972 972 self.changelog.delayupdate()
973 973 n = self.changelog.add(mn, changed + removed, text, trp, p1, p2,
974 974 user, wctx.date(), extra)
975 975 p = lambda: self.changelog.writepending() and self.root or ""
976 976 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
977 977 parent2=xp2, pending=p)
978 978 self.changelog.finalize(trp)
979 979 tr.close()
980 980
981 981 if self.branchcache:
982 982 self.branchtags()
983 983
984 984 if use_dirstate or update_dirstate:
985 985 self.dirstate.setparents(n)
986 986 if use_dirstate:
987 987 for f in removed:
988 988 self.dirstate.forget(f)
989 989 valid = 1 # our dirstate updates are complete
990 990
991 991 self.hook("commit", node=hex(n), parent1=xp1, parent2=xp2)
992 992 return n
993 993 finally:
994 994 if not valid: # don't save our updated dirstate
995 995 self.dirstate.invalidate()
996 996 del tr
997 997
998 998 def walk(self, match, node=None):
999 999 '''
1000 1000 walk recursively through the directory tree or a given
1001 1001 changeset, finding all files matched by the match
1002 1002 function
1003 1003 '''
1004 1004 return self[node].walk(match)
1005 1005
1006 1006 def status(self, node1='.', node2=None, match=None,
1007 1007 ignored=False, clean=False, unknown=False):
1008 1008 """return status of files between two nodes or node and working directory
1009 1009
1010 1010 If node1 is None, use the first dirstate parent instead.
1011 1011 If node2 is None, compare node1 with working directory.
1012 1012 """
1013 1013
1014 1014 def mfmatches(ctx):
1015 1015 mf = ctx.manifest().copy()
1016 1016 for fn in mf.keys():
1017 1017 if not match(fn):
1018 1018 del mf[fn]
1019 1019 return mf
1020 1020
1021 1021 if isinstance(node1, context.changectx):
1022 1022 ctx1 = node1
1023 1023 else:
1024 1024 ctx1 = self[node1]
1025 1025 if isinstance(node2, context.changectx):
1026 1026 ctx2 = node2
1027 1027 else:
1028 1028 ctx2 = self[node2]
1029 1029
1030 1030 working = ctx2.rev() is None
1031 1031 parentworking = working and ctx1 == self['.']
1032 1032 match = match or match_.always(self.root, self.getcwd())
1033 1033 listignored, listclean, listunknown = ignored, clean, unknown
1034 1034
1035 1035 # load earliest manifest first for caching reasons
1036 1036 if not working and ctx2.rev() < ctx1.rev():
1037 1037 ctx2.manifest()
1038 1038
1039 1039 if not parentworking:
1040 1040 def bad(f, msg):
1041 1041 if f not in ctx1:
1042 1042 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1043 1043 return False
1044 1044 match.bad = bad
1045 1045
1046 1046 if working: # we need to scan the working dir
1047 1047 s = self.dirstate.status(match, listignored, listclean, listunknown)
1048 1048 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1049 1049
1050 1050 # check for any possibly clean files
1051 1051 if parentworking and cmp:
1052 1052 fixup = []
1053 1053 # do a full compare of any files that might have changed
1054 for f in cmp:
1054 for f in sorted(cmp):
1055 1055 if (f not in ctx1 or ctx2.flags(f) != ctx1.flags(f)
1056 1056 or ctx1[f].cmp(ctx2[f].data())):
1057 1057 modified.append(f)
1058 1058 else:
1059 1059 fixup.append(f)
1060 1060
1061 1061 if listclean:
1062 1062 clean += fixup
1063 1063
1064 1064 # update dirstate for files that are actually clean
1065 1065 if fixup:
1066 1066 wlock = None
1067 1067 try:
1068 1068 try:
1069 1069 # updating the dirstate is optional
1070 1070 # so we don't wait on the lock
1071 1071 wlock = self.wlock(False)
1072 1072 for f in fixup:
1073 1073 self.dirstate.normal(f)
1074 1074 except error.LockError:
1075 1075 pass
1076 1076 finally:
1077 1077 release(wlock)
1078 1078
1079 1079 if not parentworking:
1080 1080 mf1 = mfmatches(ctx1)
1081 1081 if working:
1082 1082 # we are comparing working dir against non-parent
1083 1083 # generate a pseudo-manifest for the working dir
1084 1084 mf2 = mfmatches(self['.'])
1085 1085 for f in cmp + modified + added:
1086 1086 mf2[f] = None
1087 1087 mf2.set(f, ctx2.flags(f))
1088 1088 for f in removed:
1089 1089 if f in mf2:
1090 1090 del mf2[f]
1091 1091 else:
1092 1092 # we are comparing two revisions
1093 1093 deleted, unknown, ignored = [], [], []
1094 1094 mf2 = mfmatches(ctx2)
1095 1095
1096 1096 modified, added, clean = [], [], []
1097 1097 for fn in mf2:
1098 1098 if fn in mf1:
1099 1099 if (mf1.flags(fn) != mf2.flags(fn) or
1100 1100 (mf1[fn] != mf2[fn] and
1101 1101 (mf2[fn] or ctx1[fn].cmp(ctx2[fn].data())))):
1102 1102 modified.append(fn)
1103 1103 elif listclean:
1104 1104 clean.append(fn)
1105 1105 del mf1[fn]
1106 1106 else:
1107 1107 added.append(fn)
1108 1108 removed = mf1.keys()
1109 1109
1110 1110 r = modified, added, removed, deleted, unknown, ignored, clean
1111 1111 [l.sort() for l in r]
1112 1112 return r
1113 1113
1114 1114 def add(self, list):
1115 1115 wlock = self.wlock()
1116 1116 try:
1117 1117 rejected = []
1118 1118 for f in list:
1119 1119 p = self.wjoin(f)
1120 1120 try:
1121 1121 st = os.lstat(p)
1122 1122 except:
1123 1123 self.ui.warn(_("%s does not exist!\n") % f)
1124 1124 rejected.append(f)
1125 1125 continue
1126 1126 if st.st_size > 10000000:
1127 1127 self.ui.warn(_("%s: files over 10MB may cause memory and"
1128 1128 " performance problems\n"
1129 1129 "(use 'hg revert %s' to unadd the file)\n")
1130 1130 % (f, f))
1131 1131 if not (stat.S_ISREG(st.st_mode) or stat.S_ISLNK(st.st_mode)):
1132 1132 self.ui.warn(_("%s not added: only files and symlinks "
1133 1133 "supported currently\n") % f)
1134 1134 rejected.append(p)
1135 1135 elif self.dirstate[f] in 'amn':
1136 1136 self.ui.warn(_("%s already tracked!\n") % f)
1137 1137 elif self.dirstate[f] == 'r':
1138 1138 self.dirstate.normallookup(f)
1139 1139 else:
1140 1140 self.dirstate.add(f)
1141 1141 return rejected
1142 1142 finally:
1143 1143 wlock.release()
1144 1144
1145 1145 def forget(self, list):
1146 1146 wlock = self.wlock()
1147 1147 try:
1148 1148 for f in list:
1149 1149 if self.dirstate[f] != 'a':
1150 1150 self.ui.warn(_("%s not added!\n") % f)
1151 1151 else:
1152 1152 self.dirstate.forget(f)
1153 1153 finally:
1154 1154 wlock.release()
1155 1155
1156 1156 def remove(self, list, unlink=False):
1157 1157 wlock = None
1158 1158 try:
1159 1159 if unlink:
1160 1160 for f in list:
1161 1161 try:
1162 1162 util.unlink(self.wjoin(f))
1163 1163 except OSError, inst:
1164 1164 if inst.errno != errno.ENOENT:
1165 1165 raise
1166 1166 wlock = self.wlock()
1167 1167 for f in list:
1168 1168 if unlink and os.path.exists(self.wjoin(f)):
1169 1169 self.ui.warn(_("%s still exists!\n") % f)
1170 1170 elif self.dirstate[f] == 'a':
1171 1171 self.dirstate.forget(f)
1172 1172 elif f not in self.dirstate:
1173 1173 self.ui.warn(_("%s not tracked!\n") % f)
1174 1174 else:
1175 1175 self.dirstate.remove(f)
1176 1176 finally:
1177 1177 release(wlock)
1178 1178
1179 1179 def undelete(self, list):
1180 1180 manifests = [self.manifest.read(self.changelog.read(p)[0])
1181 1181 for p in self.dirstate.parents() if p != nullid]
1182 1182 wlock = self.wlock()
1183 1183 try:
1184 1184 for f in list:
1185 1185 if self.dirstate[f] != 'r':
1186 1186 self.ui.warn(_("%s not removed!\n") % f)
1187 1187 else:
1188 1188 m = f in manifests[0] and manifests[0] or manifests[1]
1189 1189 t = self.file(f).read(m[f])
1190 1190 self.wwrite(f, t, m.flags(f))
1191 1191 self.dirstate.normal(f)
1192 1192 finally:
1193 1193 wlock.release()
1194 1194
1195 1195 def copy(self, source, dest):
1196 1196 p = self.wjoin(dest)
1197 1197 if not (os.path.exists(p) or os.path.islink(p)):
1198 1198 self.ui.warn(_("%s does not exist!\n") % dest)
1199 1199 elif not (os.path.isfile(p) or os.path.islink(p)):
1200 1200 self.ui.warn(_("copy failed: %s is not a file or a "
1201 1201 "symbolic link\n") % dest)
1202 1202 else:
1203 1203 wlock = self.wlock()
1204 1204 try:
1205 1205 if self.dirstate[dest] in '?r':
1206 1206 self.dirstate.add(dest)
1207 1207 self.dirstate.copy(source, dest)
1208 1208 finally:
1209 1209 wlock.release()
1210 1210
1211 1211 def heads(self, start=None, closed=True):
1212 1212 heads = self.changelog.heads(start)
1213 1213 def display(head):
1214 1214 if closed:
1215 1215 return True
1216 1216 extras = self.changelog.read(head)[5]
1217 1217 return ('close' not in extras)
1218 1218 # sort the output in rev descending order
1219 1219 heads = [(-self.changelog.rev(h), h) for h in heads if display(h)]
1220 1220 return [n for (r, n) in sorted(heads)]
1221 1221
1222 1222 def branchheads(self, branch=None, start=None, closed=True):
1223 1223 if branch is None:
1224 1224 branch = self[None].branch()
1225 1225 branches = self._branchheads()
1226 1226 if branch not in branches:
1227 1227 return []
1228 1228 bheads = branches[branch]
1229 1229 # the cache returns heads ordered lowest to highest
1230 1230 bheads.reverse()
1231 1231 if start is not None:
1232 1232 # filter out the heads that cannot be reached from startrev
1233 1233 bheads = self.changelog.nodesbetween([start], bheads)[2]
1234 1234 if not closed:
1235 1235 bheads = [h for h in bheads if
1236 1236 ('close' not in self.changelog.read(h)[5])]
1237 1237 return bheads
1238 1238
1239 1239 def branches(self, nodes):
1240 1240 if not nodes:
1241 1241 nodes = [self.changelog.tip()]
1242 1242 b = []
1243 1243 for n in nodes:
1244 1244 t = n
1245 1245 while 1:
1246 1246 p = self.changelog.parents(n)
1247 1247 if p[1] != nullid or p[0] == nullid:
1248 1248 b.append((t, n, p[0], p[1]))
1249 1249 break
1250 1250 n = p[0]
1251 1251 return b
1252 1252
1253 1253 def between(self, pairs):
1254 1254 r = []
1255 1255
1256 1256 for top, bottom in pairs:
1257 1257 n, l, i = top, [], 0
1258 1258 f = 1
1259 1259
1260 1260 while n != bottom and n != nullid:
1261 1261 p = self.changelog.parents(n)[0]
1262 1262 if i == f:
1263 1263 l.append(n)
1264 1264 f = f * 2
1265 1265 n = p
1266 1266 i += 1
1267 1267
1268 1268 r.append(l)
1269 1269
1270 1270 return r
1271 1271
1272 1272 def findincoming(self, remote, base=None, heads=None, force=False):
1273 1273 """Return list of roots of the subsets of missing nodes from remote
1274 1274
1275 1275 If base dict is specified, assume that these nodes and their parents
1276 1276 exist on the remote side and that no child of a node of base exists
1277 1277 in both remote and self.
1278 1278 Furthermore base will be updated to include the nodes that exists
1279 1279 in self and remote but no children exists in self and remote.
1280 1280 If a list of heads is specified, return only nodes which are heads
1281 1281 or ancestors of these heads.
1282 1282
1283 1283 All the ancestors of base are in self and in remote.
1284 1284 All the descendants of the list returned are missing in self.
1285 1285 (and so we know that the rest of the nodes are missing in remote, see
1286 1286 outgoing)
1287 1287 """
1288 1288 return self.findcommonincoming(remote, base, heads, force)[1]
1289 1289
1290 1290 def findcommonincoming(self, remote, base=None, heads=None, force=False):
1291 1291 """Return a tuple (common, missing roots, heads) used to identify
1292 1292 missing nodes from remote.
1293 1293
1294 1294 If base dict is specified, assume that these nodes and their parents
1295 1295 exist on the remote side and that no child of a node of base exists
1296 1296 in both remote and self.
1297 1297 Furthermore base will be updated to include the nodes that exists
1298 1298 in self and remote but no children exists in self and remote.
1299 1299 If a list of heads is specified, return only nodes which are heads
1300 1300 or ancestors of these heads.
1301 1301
1302 1302 All the ancestors of base are in self and in remote.
1303 1303 """
1304 1304 m = self.changelog.nodemap
1305 1305 search = []
1306 1306 fetch = set()
1307 1307 seen = set()
1308 1308 seenbranch = set()
1309 1309 if base == None:
1310 1310 base = {}
1311 1311
1312 1312 if not heads:
1313 1313 heads = remote.heads()
1314 1314
1315 1315 if self.changelog.tip() == nullid:
1316 1316 base[nullid] = 1
1317 1317 if heads != [nullid]:
1318 1318 return [nullid], [nullid], list(heads)
1319 1319 return [nullid], [], []
1320 1320
1321 1321 # assume we're closer to the tip than the root
1322 1322 # and start by examining the heads
1323 1323 self.ui.status(_("searching for changes\n"))
1324 1324
1325 1325 unknown = []
1326 1326 for h in heads:
1327 1327 if h not in m:
1328 1328 unknown.append(h)
1329 1329 else:
1330 1330 base[h] = 1
1331 1331
1332 1332 heads = unknown
1333 1333 if not unknown:
1334 1334 return base.keys(), [], []
1335 1335
1336 1336 req = set(unknown)
1337 1337 reqcnt = 0
1338 1338
1339 1339 # search through remote branches
1340 1340 # a 'branch' here is a linear segment of history, with four parts:
1341 1341 # head, root, first parent, second parent
1342 1342 # (a branch always has two parents (or none) by definition)
1343 1343 unknown = remote.branches(unknown)
1344 1344 while unknown:
1345 1345 r = []
1346 1346 while unknown:
1347 1347 n = unknown.pop(0)
1348 1348 if n[0] in seen:
1349 1349 continue
1350 1350
1351 1351 self.ui.debug(_("examining %s:%s\n")
1352 1352 % (short(n[0]), short(n[1])))
1353 1353 if n[0] == nullid: # found the end of the branch
1354 1354 pass
1355 1355 elif n in seenbranch:
1356 1356 self.ui.debug(_("branch already found\n"))
1357 1357 continue
1358 1358 elif n[1] and n[1] in m: # do we know the base?
1359 1359 self.ui.debug(_("found incomplete branch %s:%s\n")
1360 1360 % (short(n[0]), short(n[1])))
1361 1361 search.append(n[0:2]) # schedule branch range for scanning
1362 1362 seenbranch.add(n)
1363 1363 else:
1364 1364 if n[1] not in seen and n[1] not in fetch:
1365 1365 if n[2] in m and n[3] in m:
1366 1366 self.ui.debug(_("found new changeset %s\n") %
1367 1367 short(n[1]))
1368 1368 fetch.add(n[1]) # earliest unknown
1369 1369 for p in n[2:4]:
1370 1370 if p in m:
1371 1371 base[p] = 1 # latest known
1372 1372
1373 1373 for p in n[2:4]:
1374 1374 if p not in req and p not in m:
1375 1375 r.append(p)
1376 1376 req.add(p)
1377 1377 seen.add(n[0])
1378 1378
1379 1379 if r:
1380 1380 reqcnt += 1
1381 1381 self.ui.debug(_("request %d: %s\n") %
1382 1382 (reqcnt, " ".join(map(short, r))))
1383 1383 for p in xrange(0, len(r), 10):
1384 1384 for b in remote.branches(r[p:p+10]):
1385 1385 self.ui.debug(_("received %s:%s\n") %
1386 1386 (short(b[0]), short(b[1])))
1387 1387 unknown.append(b)
1388 1388
1389 1389 # do binary search on the branches we found
1390 1390 while search:
1391 1391 newsearch = []
1392 1392 reqcnt += 1
1393 1393 for n, l in zip(search, remote.between(search)):
1394 1394 l.append(n[1])
1395 1395 p = n[0]
1396 1396 f = 1
1397 1397 for i in l:
1398 1398 self.ui.debug(_("narrowing %d:%d %s\n") % (f, len(l), short(i)))
1399 1399 if i in m:
1400 1400 if f <= 2:
1401 1401 self.ui.debug(_("found new branch changeset %s\n") %
1402 1402 short(p))
1403 1403 fetch.add(p)
1404 1404 base[i] = 1
1405 1405 else:
1406 1406 self.ui.debug(_("narrowed branch search to %s:%s\n")
1407 1407 % (short(p), short(i)))
1408 1408 newsearch.append((p, i))
1409 1409 break
1410 1410 p, f = i, f * 2
1411 1411 search = newsearch
1412 1412
1413 1413 # sanity check our fetch list
1414 1414 for f in fetch:
1415 1415 if f in m:
1416 1416 raise error.RepoError(_("already have changeset ")
1417 1417 + short(f[:4]))
1418 1418
1419 1419 if base.keys() == [nullid]:
1420 1420 if force:
1421 1421 self.ui.warn(_("warning: repository is unrelated\n"))
1422 1422 else:
1423 1423 raise util.Abort(_("repository is unrelated"))
1424 1424
1425 1425 self.ui.debug(_("found new changesets starting at ") +
1426 1426 " ".join([short(f) for f in fetch]) + "\n")
1427 1427
1428 1428 self.ui.debug(_("%d total queries\n") % reqcnt)
1429 1429
1430 1430 return base.keys(), list(fetch), heads
1431 1431
1432 1432 def findoutgoing(self, remote, base=None, heads=None, force=False):
1433 1433 """Return list of nodes that are roots of subsets not in remote
1434 1434
1435 1435 If base dict is specified, assume that these nodes and their parents
1436 1436 exist on the remote side.
1437 1437 If a list of heads is specified, return only nodes which are heads
1438 1438 or ancestors of these heads, and return a second element which
1439 1439 contains all remote heads which get new children.
1440 1440 """
1441 1441 if base == None:
1442 1442 base = {}
1443 1443 self.findincoming(remote, base, heads, force=force)
1444 1444
1445 1445 self.ui.debug(_("common changesets up to ")
1446 1446 + " ".join(map(short, base.keys())) + "\n")
1447 1447
1448 1448 remain = set(self.changelog.nodemap)
1449 1449
1450 1450 # prune everything remote has from the tree
1451 1451 remain.remove(nullid)
1452 1452 remove = base.keys()
1453 1453 while remove:
1454 1454 n = remove.pop(0)
1455 1455 if n in remain:
1456 1456 remain.remove(n)
1457 1457 for p in self.changelog.parents(n):
1458 1458 remove.append(p)
1459 1459
1460 1460 # find every node whose parents have been pruned
1461 1461 subset = []
1462 1462 # find every remote head that will get new children
1463 1463 updated_heads = {}
1464 1464 for n in remain:
1465 1465 p1, p2 = self.changelog.parents(n)
1466 1466 if p1 not in remain and p2 not in remain:
1467 1467 subset.append(n)
1468 1468 if heads:
1469 1469 if p1 in heads:
1470 1470 updated_heads[p1] = True
1471 1471 if p2 in heads:
1472 1472 updated_heads[p2] = True
1473 1473
1474 1474 # this is the set of all roots we have to push
1475 1475 if heads:
1476 1476 return subset, updated_heads.keys()
1477 1477 else:
1478 1478 return subset
1479 1479
1480 1480 def pull(self, remote, heads=None, force=False):
1481 1481 lock = self.lock()
1482 1482 try:
1483 1483 common, fetch, rheads = self.findcommonincoming(remote, heads=heads,
1484 1484 force=force)
1485 1485 if fetch == [nullid]:
1486 1486 self.ui.status(_("requesting all changes\n"))
1487 1487
1488 1488 if not fetch:
1489 1489 self.ui.status(_("no changes found\n"))
1490 1490 return 0
1491 1491
1492 1492 if heads is None and remote.capable('changegroupsubset'):
1493 1493 heads = rheads
1494 1494
1495 1495 if heads is None:
1496 1496 cg = remote.changegroup(fetch, 'pull')
1497 1497 else:
1498 1498 if not remote.capable('changegroupsubset'):
1499 1499 raise util.Abort(_("Partial pull cannot be done because other repository doesn't support changegroupsubset."))
1500 1500 cg = remote.changegroupsubset(fetch, heads, 'pull')
1501 1501 return self.addchangegroup(cg, 'pull', remote.url())
1502 1502 finally:
1503 1503 lock.release()
1504 1504
1505 1505 def push(self, remote, force=False, revs=None):
1506 1506 # there are two ways to push to remote repo:
1507 1507 #
1508 1508 # addchangegroup assumes local user can lock remote
1509 1509 # repo (local filesystem, old ssh servers).
1510 1510 #
1511 1511 # unbundle assumes local user cannot lock remote repo (new ssh
1512 1512 # servers, http servers).
1513 1513
1514 1514 if remote.capable('unbundle'):
1515 1515 return self.push_unbundle(remote, force, revs)
1516 1516 return self.push_addchangegroup(remote, force, revs)
1517 1517
1518 1518 def prepush(self, remote, force, revs):
1519 1519 common = {}
1520 1520 remote_heads = remote.heads()
1521 1521 inc = self.findincoming(remote, common, remote_heads, force=force)
1522 1522
1523 1523 update, updated_heads = self.findoutgoing(remote, common, remote_heads)
1524 1524 if revs is not None:
1525 1525 msng_cl, bases, heads = self.changelog.nodesbetween(update, revs)
1526 1526 else:
1527 1527 bases, heads = update, self.changelog.heads()
1528 1528
1529 1529 if not bases:
1530 1530 self.ui.status(_("no changes found\n"))
1531 1531 return None, 1
1532 1532 elif not force:
1533 1533 # check if we're creating new remote heads
1534 1534 # to be a remote head after push, node must be either
1535 1535 # - unknown locally
1536 1536 # - a local outgoing head descended from update
1537 1537 # - a remote head that's known locally and not
1538 1538 # ancestral to an outgoing head
1539 1539
1540 1540 warn = 0
1541 1541
1542 1542 if remote_heads == [nullid]:
1543 1543 warn = 0
1544 1544 elif not revs and len(heads) > len(remote_heads):
1545 1545 warn = 1
1546 1546 else:
1547 1547 newheads = list(heads)
1548 1548 for r in remote_heads:
1549 1549 if r in self.changelog.nodemap:
1550 1550 desc = self.changelog.heads(r, heads)
1551 1551 l = [h for h in heads if h in desc]
1552 1552 if not l:
1553 1553 newheads.append(r)
1554 1554 else:
1555 1555 newheads.append(r)
1556 1556 if len(newheads) > len(remote_heads):
1557 1557 warn = 1
1558 1558
1559 1559 if warn:
1560 1560 self.ui.warn(_("abort: push creates new remote heads!\n"))
1561 1561 self.ui.status(_("(did you forget to merge?"
1562 1562 " use push -f to force)\n"))
1563 1563 return None, 0
1564 1564 elif inc:
1565 1565 self.ui.warn(_("note: unsynced remote changes!\n"))
1566 1566
1567 1567
1568 1568 if revs is None:
1569 1569 # use the fast path, no race possible on push
1570 1570 cg = self._changegroup(common.keys(), 'push')
1571 1571 else:
1572 1572 cg = self.changegroupsubset(update, revs, 'push')
1573 1573 return cg, remote_heads
1574 1574
1575 1575 def push_addchangegroup(self, remote, force, revs):
1576 1576 lock = remote.lock()
1577 1577 try:
1578 1578 ret = self.prepush(remote, force, revs)
1579 1579 if ret[0] is not None:
1580 1580 cg, remote_heads = ret
1581 1581 return remote.addchangegroup(cg, 'push', self.url())
1582 1582 return ret[1]
1583 1583 finally:
1584 1584 lock.release()
1585 1585
1586 1586 def push_unbundle(self, remote, force, revs):
1587 1587 # local repo finds heads on server, finds out what revs it
1588 1588 # must push. once revs transferred, if server finds it has
1589 1589 # different heads (someone else won commit/push race), server
1590 1590 # aborts.
1591 1591
1592 1592 ret = self.prepush(remote, force, revs)
1593 1593 if ret[0] is not None:
1594 1594 cg, remote_heads = ret
1595 1595 if force: remote_heads = ['force']
1596 1596 return remote.unbundle(cg, remote_heads, 'push')
1597 1597 return ret[1]
1598 1598
1599 1599 def changegroupinfo(self, nodes, source):
1600 1600 if self.ui.verbose or source == 'bundle':
1601 1601 self.ui.status(_("%d changesets found\n") % len(nodes))
1602 1602 if self.ui.debugflag:
1603 1603 self.ui.debug(_("list of changesets:\n"))
1604 1604 for node in nodes:
1605 1605 self.ui.debug("%s\n" % hex(node))
1606 1606
1607 1607 def changegroupsubset(self, bases, heads, source, extranodes=None):
1608 1608 """This function generates a changegroup consisting of all the nodes
1609 1609 that are descendents of any of the bases, and ancestors of any of
1610 1610 the heads.
1611 1611
1612 1612 It is fairly complex as determining which filenodes and which
1613 1613 manifest nodes need to be included for the changeset to be complete
1614 1614 is non-trivial.
1615 1615
1616 1616 Another wrinkle is doing the reverse, figuring out which changeset in
1617 1617 the changegroup a particular filenode or manifestnode belongs to.
1618 1618
1619 1619 The caller can specify some nodes that must be included in the
1620 1620 changegroup using the extranodes argument. It should be a dict
1621 1621 where the keys are the filenames (or 1 for the manifest), and the
1622 1622 values are lists of (node, linknode) tuples, where node is a wanted
1623 1623 node and linknode is the changelog node that should be transmitted as
1624 1624 the linkrev.
1625 1625 """
1626 1626
1627 1627 if extranodes is None:
1628 1628 # can we go through the fast path ?
1629 1629 heads.sort()
1630 1630 allheads = self.heads()
1631 1631 allheads.sort()
1632 1632 if heads == allheads:
1633 1633 common = []
1634 1634 # parents of bases are known from both sides
1635 1635 for n in bases:
1636 1636 for p in self.changelog.parents(n):
1637 1637 if p != nullid:
1638 1638 common.append(p)
1639 1639 return self._changegroup(common, source)
1640 1640
1641 1641 self.hook('preoutgoing', throw=True, source=source)
1642 1642
1643 1643 # Set up some initial variables
1644 1644 # Make it easy to refer to self.changelog
1645 1645 cl = self.changelog
1646 1646 # msng is short for missing - compute the list of changesets in this
1647 1647 # changegroup.
1648 1648 msng_cl_lst, bases, heads = cl.nodesbetween(bases, heads)
1649 1649 self.changegroupinfo(msng_cl_lst, source)
1650 1650 # Some bases may turn out to be superfluous, and some heads may be
1651 1651 # too. nodesbetween will return the minimal set of bases and heads
1652 1652 # necessary to re-create the changegroup.
1653 1653
1654 1654 # Known heads are the list of heads that it is assumed the recipient
1655 1655 # of this changegroup will know about.
1656 1656 knownheads = {}
1657 1657 # We assume that all parents of bases are known heads.
1658 1658 for n in bases:
1659 1659 for p in cl.parents(n):
1660 1660 if p != nullid:
1661 1661 knownheads[p] = 1
1662 1662 knownheads = knownheads.keys()
1663 1663 if knownheads:
1664 1664 # Now that we know what heads are known, we can compute which
1665 1665 # changesets are known. The recipient must know about all
1666 1666 # changesets required to reach the known heads from the null
1667 1667 # changeset.
1668 1668 has_cl_set, junk, junk = cl.nodesbetween(None, knownheads)
1669 1669 junk = None
1670 1670 # Transform the list into a set.
1671 1671 has_cl_set = set(has_cl_set)
1672 1672 else:
1673 1673 # If there were no known heads, the recipient cannot be assumed to
1674 1674 # know about any changesets.
1675 1675 has_cl_set = set()
1676 1676
1677 1677 # Make it easy to refer to self.manifest
1678 1678 mnfst = self.manifest
1679 1679 # We don't know which manifests are missing yet
1680 1680 msng_mnfst_set = {}
1681 1681 # Nor do we know which filenodes are missing.
1682 1682 msng_filenode_set = {}
1683 1683
1684 1684 junk = mnfst.index[len(mnfst) - 1] # Get around a bug in lazyindex
1685 1685 junk = None
1686 1686
1687 1687 # A changeset always belongs to itself, so the changenode lookup
1688 1688 # function for a changenode is identity.
1689 1689 def identity(x):
1690 1690 return x
1691 1691
1692 1692 # A function generating function. Sets up an environment for the
1693 1693 # inner function.
1694 1694 def cmp_by_rev_func(revlog):
1695 1695 # Compare two nodes by their revision number in the environment's
1696 1696 # revision history. Since the revision number both represents the
1697 1697 # most efficient order to read the nodes in, and represents a
1698 1698 # topological sorting of the nodes, this function is often useful.
1699 1699 def cmp_by_rev(a, b):
1700 1700 return cmp(revlog.rev(a), revlog.rev(b))
1701 1701 return cmp_by_rev
1702 1702
1703 1703 # If we determine that a particular file or manifest node must be a
1704 1704 # node that the recipient of the changegroup will already have, we can
1705 1705 # also assume the recipient will have all the parents. This function
1706 1706 # prunes them from the set of missing nodes.
1707 1707 def prune_parents(revlog, hasset, msngset):
1708 1708 haslst = hasset.keys()
1709 1709 haslst.sort(cmp_by_rev_func(revlog))
1710 1710 for node in haslst:
1711 1711 parentlst = [p for p in revlog.parents(node) if p != nullid]
1712 1712 while parentlst:
1713 1713 n = parentlst.pop()
1714 1714 if n not in hasset:
1715 1715 hasset[n] = 1
1716 1716 p = [p for p in revlog.parents(n) if p != nullid]
1717 1717 parentlst.extend(p)
1718 1718 for n in hasset:
1719 1719 msngset.pop(n, None)
1720 1720
1721 1721 # This is a function generating function used to set up an environment
1722 1722 # for the inner function to execute in.
1723 1723 def manifest_and_file_collector(changedfileset):
1724 1724 # This is an information gathering function that gathers
1725 1725 # information from each changeset node that goes out as part of
1726 1726 # the changegroup. The information gathered is a list of which
1727 1727 # manifest nodes are potentially required (the recipient may
1728 1728 # already have them) and total list of all files which were
1729 1729 # changed in any changeset in the changegroup.
1730 1730 #
1731 1731 # We also remember the first changenode we saw any manifest
1732 1732 # referenced by so we can later determine which changenode 'owns'
1733 1733 # the manifest.
1734 1734 def collect_manifests_and_files(clnode):
1735 1735 c = cl.read(clnode)
1736 1736 for f in c[3]:
1737 1737 # This is to make sure we only have one instance of each
1738 1738 # filename string for each filename.
1739 1739 changedfileset.setdefault(f, f)
1740 1740 msng_mnfst_set.setdefault(c[0], clnode)
1741 1741 return collect_manifests_and_files
1742 1742
1743 1743 # Figure out which manifest nodes (of the ones we think might be part
1744 1744 # of the changegroup) the recipient must know about and remove them
1745 1745 # from the changegroup.
1746 1746 def prune_manifests():
1747 1747 has_mnfst_set = {}
1748 1748 for n in msng_mnfst_set:
1749 1749 # If a 'missing' manifest thinks it belongs to a changenode
1750 1750 # the recipient is assumed to have, obviously the recipient
1751 1751 # must have that manifest.
1752 1752 linknode = cl.node(mnfst.linkrev(mnfst.rev(n)))
1753 1753 if linknode in has_cl_set:
1754 1754 has_mnfst_set[n] = 1
1755 1755 prune_parents(mnfst, has_mnfst_set, msng_mnfst_set)
1756 1756
1757 1757 # Use the information collected in collect_manifests_and_files to say
1758 1758 # which changenode any manifestnode belongs to.
1759 1759 def lookup_manifest_link(mnfstnode):
1760 1760 return msng_mnfst_set[mnfstnode]
1761 1761
1762 1762 # A function generating function that sets up the initial environment
1763 1763 # the inner function.
1764 1764 def filenode_collector(changedfiles):
1765 1765 next_rev = [0]
1766 1766 # This gathers information from each manifestnode included in the
1767 1767 # changegroup about which filenodes the manifest node references
1768 1768 # so we can include those in the changegroup too.
1769 1769 #
1770 1770 # It also remembers which changenode each filenode belongs to. It
1771 1771 # does this by assuming the a filenode belongs to the changenode
1772 1772 # the first manifest that references it belongs to.
1773 1773 def collect_msng_filenodes(mnfstnode):
1774 1774 r = mnfst.rev(mnfstnode)
1775 1775 if r == next_rev[0]:
1776 1776 # If the last rev we looked at was the one just previous,
1777 1777 # we only need to see a diff.
1778 1778 deltamf = mnfst.readdelta(mnfstnode)
1779 1779 # For each line in the delta
1780 1780 for f, fnode in deltamf.iteritems():
1781 1781 f = changedfiles.get(f, None)
1782 1782 # And if the file is in the list of files we care
1783 1783 # about.
1784 1784 if f is not None:
1785 1785 # Get the changenode this manifest belongs to
1786 1786 clnode = msng_mnfst_set[mnfstnode]
1787 1787 # Create the set of filenodes for the file if
1788 1788 # there isn't one already.
1789 1789 ndset = msng_filenode_set.setdefault(f, {})
1790 1790 # And set the filenode's changelog node to the
1791 1791 # manifest's if it hasn't been set already.
1792 1792 ndset.setdefault(fnode, clnode)
1793 1793 else:
1794 1794 # Otherwise we need a full manifest.
1795 1795 m = mnfst.read(mnfstnode)
1796 1796 # For every file in we care about.
1797 1797 for f in changedfiles:
1798 1798 fnode = m.get(f, None)
1799 1799 # If it's in the manifest
1800 1800 if fnode is not None:
1801 1801 # See comments above.
1802 1802 clnode = msng_mnfst_set[mnfstnode]
1803 1803 ndset = msng_filenode_set.setdefault(f, {})
1804 1804 ndset.setdefault(fnode, clnode)
1805 1805 # Remember the revision we hope to see next.
1806 1806 next_rev[0] = r + 1
1807 1807 return collect_msng_filenodes
1808 1808
1809 1809 # We have a list of filenodes we think we need for a file, lets remove
1810 1810 # all those we know the recipient must have.
1811 1811 def prune_filenodes(f, filerevlog):
1812 1812 msngset = msng_filenode_set[f]
1813 1813 hasset = {}
1814 1814 # If a 'missing' filenode thinks it belongs to a changenode we
1815 1815 # assume the recipient must have, then the recipient must have
1816 1816 # that filenode.
1817 1817 for n in msngset:
1818 1818 clnode = cl.node(filerevlog.linkrev(filerevlog.rev(n)))
1819 1819 if clnode in has_cl_set:
1820 1820 hasset[n] = 1
1821 1821 prune_parents(filerevlog, hasset, msngset)
1822 1822
1823 1823 # A function generator function that sets up the a context for the
1824 1824 # inner function.
1825 1825 def lookup_filenode_link_func(fname):
1826 1826 msngset = msng_filenode_set[fname]
1827 1827 # Lookup the changenode the filenode belongs to.
1828 1828 def lookup_filenode_link(fnode):
1829 1829 return msngset[fnode]
1830 1830 return lookup_filenode_link
1831 1831
1832 1832 # Add the nodes that were explicitly requested.
1833 1833 def add_extra_nodes(name, nodes):
1834 1834 if not extranodes or name not in extranodes:
1835 1835 return
1836 1836
1837 1837 for node, linknode in extranodes[name]:
1838 1838 if node not in nodes:
1839 1839 nodes[node] = linknode
1840 1840
1841 1841 # Now that we have all theses utility functions to help out and
1842 1842 # logically divide up the task, generate the group.
1843 1843 def gengroup():
1844 1844 # The set of changed files starts empty.
1845 1845 changedfiles = {}
1846 1846 # Create a changenode group generator that will call our functions
1847 1847 # back to lookup the owning changenode and collect information.
1848 1848 group = cl.group(msng_cl_lst, identity,
1849 1849 manifest_and_file_collector(changedfiles))
1850 1850 for chnk in group:
1851 1851 yield chnk
1852 1852
1853 1853 # The list of manifests has been collected by the generator
1854 1854 # calling our functions back.
1855 1855 prune_manifests()
1856 1856 add_extra_nodes(1, msng_mnfst_set)
1857 1857 msng_mnfst_lst = msng_mnfst_set.keys()
1858 1858 # Sort the manifestnodes by revision number.
1859 1859 msng_mnfst_lst.sort(cmp_by_rev_func(mnfst))
1860 1860 # Create a generator for the manifestnodes that calls our lookup
1861 1861 # and data collection functions back.
1862 1862 group = mnfst.group(msng_mnfst_lst, lookup_manifest_link,
1863 1863 filenode_collector(changedfiles))
1864 1864 for chnk in group:
1865 1865 yield chnk
1866 1866
1867 1867 # These are no longer needed, dereference and toss the memory for
1868 1868 # them.
1869 1869 msng_mnfst_lst = None
1870 1870 msng_mnfst_set.clear()
1871 1871
1872 1872 if extranodes:
1873 1873 for fname in extranodes:
1874 1874 if isinstance(fname, int):
1875 1875 continue
1876 1876 msng_filenode_set.setdefault(fname, {})
1877 1877 changedfiles[fname] = 1
1878 1878 # Go through all our files in order sorted by name.
1879 1879 for fname in sorted(changedfiles):
1880 1880 filerevlog = self.file(fname)
1881 1881 if not len(filerevlog):
1882 1882 raise util.Abort(_("empty or missing revlog for %s") % fname)
1883 1883 # Toss out the filenodes that the recipient isn't really
1884 1884 # missing.
1885 1885 if fname in msng_filenode_set:
1886 1886 prune_filenodes(fname, filerevlog)
1887 1887 add_extra_nodes(fname, msng_filenode_set[fname])
1888 1888 msng_filenode_lst = msng_filenode_set[fname].keys()
1889 1889 else:
1890 1890 msng_filenode_lst = []
1891 1891 # If any filenodes are left, generate the group for them,
1892 1892 # otherwise don't bother.
1893 1893 if len(msng_filenode_lst) > 0:
1894 1894 yield changegroup.chunkheader(len(fname))
1895 1895 yield fname
1896 1896 # Sort the filenodes by their revision #
1897 1897 msng_filenode_lst.sort(cmp_by_rev_func(filerevlog))
1898 1898 # Create a group generator and only pass in a changenode
1899 1899 # lookup function as we need to collect no information
1900 1900 # from filenodes.
1901 1901 group = filerevlog.group(msng_filenode_lst,
1902 1902 lookup_filenode_link_func(fname))
1903 1903 for chnk in group:
1904 1904 yield chnk
1905 1905 if fname in msng_filenode_set:
1906 1906 # Don't need this anymore, toss it to free memory.
1907 1907 del msng_filenode_set[fname]
1908 1908 # Signal that no more groups are left.
1909 1909 yield changegroup.closechunk()
1910 1910
1911 1911 if msng_cl_lst:
1912 1912 self.hook('outgoing', node=hex(msng_cl_lst[0]), source=source)
1913 1913
1914 1914 return util.chunkbuffer(gengroup())
1915 1915
1916 1916 def changegroup(self, basenodes, source):
1917 1917 # to avoid a race we use changegroupsubset() (issue1320)
1918 1918 return self.changegroupsubset(basenodes, self.heads(), source)
1919 1919
1920 1920 def _changegroup(self, common, source):
1921 1921 """Generate a changegroup of all nodes that we have that a recipient
1922 1922 doesn't.
1923 1923
1924 1924 This is much easier than the previous function as we can assume that
1925 1925 the recipient has any changenode we aren't sending them.
1926 1926
1927 1927 common is the set of common nodes between remote and self"""
1928 1928
1929 1929 self.hook('preoutgoing', throw=True, source=source)
1930 1930
1931 1931 cl = self.changelog
1932 1932 nodes = cl.findmissing(common)
1933 1933 revset = set([cl.rev(n) for n in nodes])
1934 1934 self.changegroupinfo(nodes, source)
1935 1935
1936 1936 def identity(x):
1937 1937 return x
1938 1938
1939 1939 def gennodelst(log):
1940 1940 for r in log:
1941 1941 if log.linkrev(r) in revset:
1942 1942 yield log.node(r)
1943 1943
1944 1944 def changed_file_collector(changedfileset):
1945 1945 def collect_changed_files(clnode):
1946 1946 c = cl.read(clnode)
1947 1947 for fname in c[3]:
1948 1948 changedfileset[fname] = 1
1949 1949 return collect_changed_files
1950 1950
1951 1951 def lookuprevlink_func(revlog):
1952 1952 def lookuprevlink(n):
1953 1953 return cl.node(revlog.linkrev(revlog.rev(n)))
1954 1954 return lookuprevlink
1955 1955
1956 1956 def gengroup():
1957 1957 # construct a list of all changed files
1958 1958 changedfiles = {}
1959 1959
1960 1960 for chnk in cl.group(nodes, identity,
1961 1961 changed_file_collector(changedfiles)):
1962 1962 yield chnk
1963 1963
1964 1964 mnfst = self.manifest
1965 1965 nodeiter = gennodelst(mnfst)
1966 1966 for chnk in mnfst.group(nodeiter, lookuprevlink_func(mnfst)):
1967 1967 yield chnk
1968 1968
1969 1969 for fname in sorted(changedfiles):
1970 1970 filerevlog = self.file(fname)
1971 1971 if not len(filerevlog):
1972 1972 raise util.Abort(_("empty or missing revlog for %s") % fname)
1973 1973 nodeiter = gennodelst(filerevlog)
1974 1974 nodeiter = list(nodeiter)
1975 1975 if nodeiter:
1976 1976 yield changegroup.chunkheader(len(fname))
1977 1977 yield fname
1978 1978 lookup = lookuprevlink_func(filerevlog)
1979 1979 for chnk in filerevlog.group(nodeiter, lookup):
1980 1980 yield chnk
1981 1981
1982 1982 yield changegroup.closechunk()
1983 1983
1984 1984 if nodes:
1985 1985 self.hook('outgoing', node=hex(nodes[0]), source=source)
1986 1986
1987 1987 return util.chunkbuffer(gengroup())
1988 1988
1989 1989 def addchangegroup(self, source, srctype, url, emptyok=False):
1990 1990 """add changegroup to repo.
1991 1991
1992 1992 return values:
1993 1993 - nothing changed or no source: 0
1994 1994 - more heads than before: 1+added heads (2..n)
1995 1995 - less heads than before: -1-removed heads (-2..-n)
1996 1996 - number of heads stays the same: 1
1997 1997 """
1998 1998 def csmap(x):
1999 1999 self.ui.debug(_("add changeset %s\n") % short(x))
2000 2000 return len(cl)
2001 2001
2002 2002 def revmap(x):
2003 2003 return cl.rev(x)
2004 2004
2005 2005 if not source:
2006 2006 return 0
2007 2007
2008 2008 self.hook('prechangegroup', throw=True, source=srctype, url=url)
2009 2009
2010 2010 changesets = files = revisions = 0
2011 2011
2012 2012 # write changelog data to temp files so concurrent readers will not see
2013 2013 # inconsistent view
2014 2014 cl = self.changelog
2015 2015 cl.delayupdate()
2016 2016 oldheads = len(cl.heads())
2017 2017
2018 2018 tr = self.transaction()
2019 2019 try:
2020 2020 trp = weakref.proxy(tr)
2021 2021 # pull off the changeset group
2022 2022 self.ui.status(_("adding changesets\n"))
2023 2023 clstart = len(cl)
2024 2024 chunkiter = changegroup.chunkiter(source)
2025 2025 if cl.addgroup(chunkiter, csmap, trp) is None and not emptyok:
2026 2026 raise util.Abort(_("received changelog group is empty"))
2027 2027 clend = len(cl)
2028 2028 changesets = clend - clstart
2029 2029
2030 2030 # pull off the manifest group
2031 2031 self.ui.status(_("adding manifests\n"))
2032 2032 chunkiter = changegroup.chunkiter(source)
2033 2033 # no need to check for empty manifest group here:
2034 2034 # if the result of the merge of 1 and 2 is the same in 3 and 4,
2035 2035 # no new manifest will be created and the manifest group will
2036 2036 # be empty during the pull
2037 2037 self.manifest.addgroup(chunkiter, revmap, trp)
2038 2038
2039 2039 # process the files
2040 2040 self.ui.status(_("adding file changes\n"))
2041 2041 while 1:
2042 2042 f = changegroup.getchunk(source)
2043 2043 if not f:
2044 2044 break
2045 2045 self.ui.debug(_("adding %s revisions\n") % f)
2046 2046 fl = self.file(f)
2047 2047 o = len(fl)
2048 2048 chunkiter = changegroup.chunkiter(source)
2049 2049 if fl.addgroup(chunkiter, revmap, trp) is None:
2050 2050 raise util.Abort(_("received file revlog group is empty"))
2051 2051 revisions += len(fl) - o
2052 2052 files += 1
2053 2053
2054 2054 newheads = len(cl.heads())
2055 2055 heads = ""
2056 2056 if oldheads and newheads != oldheads:
2057 2057 heads = _(" (%+d heads)") % (newheads - oldheads)
2058 2058
2059 2059 self.ui.status(_("added %d changesets"
2060 2060 " with %d changes to %d files%s\n")
2061 2061 % (changesets, revisions, files, heads))
2062 2062
2063 2063 if changesets > 0:
2064 2064 p = lambda: cl.writepending() and self.root or ""
2065 2065 self.hook('pretxnchangegroup', throw=True,
2066 2066 node=hex(cl.node(clstart)), source=srctype,
2067 2067 url=url, pending=p)
2068 2068
2069 2069 # make changelog see real files again
2070 2070 cl.finalize(trp)
2071 2071
2072 2072 tr.close()
2073 2073 finally:
2074 2074 del tr
2075 2075
2076 2076 if changesets > 0:
2077 2077 # forcefully update the on-disk branch cache
2078 2078 self.ui.debug(_("updating the branch cache\n"))
2079 2079 self.branchtags()
2080 2080 self.hook("changegroup", node=hex(cl.node(clstart)),
2081 2081 source=srctype, url=url)
2082 2082
2083 2083 for i in xrange(clstart, clend):
2084 2084 self.hook("incoming", node=hex(cl.node(i)),
2085 2085 source=srctype, url=url)
2086 2086
2087 2087 # never return 0 here:
2088 2088 if newheads < oldheads:
2089 2089 return newheads - oldheads - 1
2090 2090 else:
2091 2091 return newheads - oldheads + 1
2092 2092
2093 2093
2094 2094 def stream_in(self, remote):
2095 2095 fp = remote.stream_out()
2096 2096 l = fp.readline()
2097 2097 try:
2098 2098 resp = int(l)
2099 2099 except ValueError:
2100 2100 raise error.ResponseError(
2101 2101 _('Unexpected response from remote server:'), l)
2102 2102 if resp == 1:
2103 2103 raise util.Abort(_('operation forbidden by server'))
2104 2104 elif resp == 2:
2105 2105 raise util.Abort(_('locking the remote repository failed'))
2106 2106 elif resp != 0:
2107 2107 raise util.Abort(_('the server sent an unknown error code'))
2108 2108 self.ui.status(_('streaming all changes\n'))
2109 2109 l = fp.readline()
2110 2110 try:
2111 2111 total_files, total_bytes = map(int, l.split(' ', 1))
2112 2112 except (ValueError, TypeError):
2113 2113 raise error.ResponseError(
2114 2114 _('Unexpected response from remote server:'), l)
2115 2115 self.ui.status(_('%d files to transfer, %s of data\n') %
2116 2116 (total_files, util.bytecount(total_bytes)))
2117 2117 start = time.time()
2118 2118 for i in xrange(total_files):
2119 2119 # XXX doesn't support '\n' or '\r' in filenames
2120 2120 l = fp.readline()
2121 2121 try:
2122 2122 name, size = l.split('\0', 1)
2123 2123 size = int(size)
2124 2124 except (ValueError, TypeError):
2125 2125 raise error.ResponseError(
2126 2126 _('Unexpected response from remote server:'), l)
2127 2127 self.ui.debug(_('adding %s (%s)\n') % (name, util.bytecount(size)))
2128 2128 ofp = self.sopener(name, 'w')
2129 2129 for chunk in util.filechunkiter(fp, limit=size):
2130 2130 ofp.write(chunk)
2131 2131 ofp.close()
2132 2132 elapsed = time.time() - start
2133 2133 if elapsed <= 0:
2134 2134 elapsed = 0.001
2135 2135 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
2136 2136 (util.bytecount(total_bytes), elapsed,
2137 2137 util.bytecount(total_bytes / elapsed)))
2138 2138 self.invalidate()
2139 2139 return len(self.heads()) + 1
2140 2140
2141 2141 def clone(self, remote, heads=[], stream=False):
2142 2142 '''clone remote repository.
2143 2143
2144 2144 keyword arguments:
2145 2145 heads: list of revs to clone (forces use of pull)
2146 2146 stream: use streaming clone if possible'''
2147 2147
2148 2148 # now, all clients that can request uncompressed clones can
2149 2149 # read repo formats supported by all servers that can serve
2150 2150 # them.
2151 2151
2152 2152 # if revlog format changes, client will have to check version
2153 2153 # and format flags on "stream" capability, and use
2154 2154 # uncompressed only if compatible.
2155 2155
2156 2156 if stream and not heads and remote.capable('stream'):
2157 2157 return self.stream_in(remote)
2158 2158 return self.pull(remote, heads)
2159 2159
2160 2160 # used to avoid circular references so destructors work
2161 2161 def aftertrans(files):
2162 2162 renamefiles = [tuple(t) for t in files]
2163 2163 def a():
2164 2164 for src, dest in renamefiles:
2165 2165 util.rename(src, dest)
2166 2166 return a
2167 2167
2168 2168 def instance(ui, path, create):
2169 2169 return localrepository(ui, util.drop_scheme('file', path), create)
2170 2170
2171 2171 def islocal(path):
2172 2172 return True
General Comments 0
You need to be logged in to leave comments. Login now