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