##// END OF EJS Templates
Fix yet another resolve corner case...
mpm@selenic.com -
r344:db419f14 default
parent child Browse files
Show More
@@ -1,1382 +1,1381 b''
1 1 # hg.py - repository classes for mercurial
2 2 #
3 3 # Copyright 2005 Matt Mackall <mpm@selenic.com>
4 4 #
5 5 # This software may be used and distributed according to the terms
6 6 # of the GNU General Public License, incorporated herein by reference.
7 7
8 8 import sys, struct, os
9 9 from revlog import *
10 10 from demandload import *
11 11 demandload(globals(), "re lock urllib urllib2 transaction time socket")
12 12 demandload(globals(), "tempfile byterange difflib")
13 13
14 14 def is_exec(f):
15 15 return (os.stat(f).st_mode & 0100 != 0)
16 16
17 17 def set_exec(f, mode):
18 18 s = os.stat(f).st_mode
19 19 if (s & 0100 != 0) == mode:
20 20 return
21 21 if mode:
22 22 # Turn on +x for every +r bit when making a file executable
23 23 # and obey umask.
24 24 umask = os.umask(0)
25 25 os.umask(umask)
26 26 os.chmod(f, s | (s & 0444) >> 2 & ~umask)
27 27 else:
28 28 os.chmod(f, s & 0666)
29 29
30 30 class filelog(revlog):
31 31 def __init__(self, opener, path):
32 32 revlog.__init__(self, opener,
33 33 os.path.join("data", path + ".i"),
34 34 os.path.join("data", path + ".d"))
35 35
36 36 def read(self, node):
37 37 return self.revision(node)
38 38 def add(self, text, transaction, link, p1=None, p2=None):
39 39 return self.addrevision(text, transaction, link, p1, p2)
40 40
41 41 def annotate(self, node):
42 42
43 43 def decorate(text, rev):
44 44 return [(rev, l) for l in text.splitlines(1)]
45 45
46 46 def strip(annotation):
47 47 return [e[1] for e in annotation]
48 48
49 49 def pair(parent, child):
50 50 new = []
51 51 sm = difflib.SequenceMatcher(None, strip(parent), strip(child))
52 52 for o, m, n, s, t in sm.get_opcodes():
53 53 if o == 'equal':
54 54 new += parent[m:n]
55 55 else:
56 56 new += child[s:t]
57 57 return new
58 58
59 59 # find all ancestors
60 60 needed = {node:1}
61 61 visit = [node]
62 62 while visit:
63 63 n = visit.pop(0)
64 64 for p in self.parents(n):
65 65 if p not in needed:
66 66 needed[p] = 1
67 67 visit.append(p)
68 68 else:
69 69 # count how many times we'll use this
70 70 needed[p] += 1
71 71
72 72 # sort by revision which is a topological order
73 73 visit = needed.keys()
74 74 visit = [ (self.rev(n), n) for n in visit ]
75 75 visit.sort()
76 76 visit = [ p[1] for p in visit ]
77 77 hist = {}
78 78
79 79 for n in visit:
80 80 curr = decorate(self.read(n), self.linkrev(n))
81 81 for p in self.parents(n):
82 82 if p != nullid:
83 83 curr = pair(hist[p], curr)
84 84 # trim the history of unneeded revs
85 85 needed[p] -= 1
86 86 if not needed[p]:
87 87 del hist[p]
88 88 hist[n] = curr
89 89
90 90 return hist[n]
91 91
92 92 class manifest(revlog):
93 93 def __init__(self, opener):
94 94 self.mapcache = None
95 95 self.listcache = None
96 96 self.addlist = None
97 97 revlog.__init__(self, opener, "00manifest.i", "00manifest.d")
98 98
99 99 def read(self, node):
100 100 if node == nullid: return {} # don't upset local cache
101 101 if self.mapcache and self.mapcache[0] == node:
102 102 return self.mapcache[1].copy()
103 103 text = self.revision(node)
104 104 map = {}
105 105 flag = {}
106 106 self.listcache = (text, text.splitlines(1))
107 107 for l in self.listcache[1]:
108 108 (f, n) = l.split('\0')
109 109 map[f] = bin(n[:40])
110 110 flag[f] = (n[40:-1] == "x")
111 111 self.mapcache = (node, map, flag)
112 112 return map
113 113
114 114 def readflags(self, node):
115 115 if node == nullid: return {} # don't upset local cache
116 116 if self.mapcache or self.mapcache[0] != node:
117 117 self.read(node)
118 118 return self.mapcache[2]
119 119
120 120 def diff(self, a, b):
121 121 # this is sneaky, as we're not actually using a and b
122 122 if self.listcache and self.addlist and self.listcache[0] == a:
123 123 d = mdiff.diff(self.listcache[1], self.addlist, 1)
124 124 if mdiff.patch(a, d) != b:
125 125 sys.stderr.write("*** sortdiff failed, falling back ***\n")
126 126 return mdiff.textdiff(a, b)
127 127 return d
128 128 else:
129 129 return mdiff.textdiff(a, b)
130 130
131 131 def add(self, map, flags, transaction, link, p1=None, p2=None):
132 132 files = map.keys()
133 133 files.sort()
134 134
135 135 self.addlist = ["%s\000%s%s\n" %
136 136 (f, hex(map[f]), flags[f] and "x" or '')
137 137 for f in files]
138 138 text = "".join(self.addlist)
139 139
140 140 n = self.addrevision(text, transaction, link, p1, p2)
141 141 self.mapcache = (n, map, flags)
142 142 self.listcache = (text, self.addlist)
143 143 self.addlist = None
144 144
145 145 return n
146 146
147 147 class changelog(revlog):
148 148 def __init__(self, opener):
149 149 revlog.__init__(self, opener, "00changelog.i", "00changelog.d")
150 150
151 151 def extract(self, text):
152 152 if not text:
153 153 return (nullid, "", "0", [], "")
154 154 last = text.index("\n\n")
155 155 desc = text[last + 2:]
156 156 l = text[:last].splitlines()
157 157 manifest = bin(l[0])
158 158 user = l[1]
159 159 date = l[2]
160 160 files = l[3:]
161 161 return (manifest, user, date, files, desc)
162 162
163 163 def read(self, node):
164 164 return self.extract(self.revision(node))
165 165
166 166 def add(self, manifest, list, desc, transaction, p1=None, p2=None,
167 167 user=None, date=None):
168 168 user = (user or
169 169 os.environ.get("HGUSER") or
170 170 os.environ.get("EMAIL") or
171 171 os.environ.get("LOGNAME", "unknown") + '@' + socket.getfqdn())
172 172 date = date or "%d %d" % (time.time(), time.timezone)
173 173 list.sort()
174 174 l = [hex(manifest), user, date] + list + ["", desc]
175 175 text = "\n".join(l)
176 176 return self.addrevision(text, transaction, self.count(), p1, p2)
177 177
178 178 class dirstate:
179 179 def __init__(self, opener, ui, root):
180 180 self.opener = opener
181 181 self.root = root
182 182 self.dirty = 0
183 183 self.ui = ui
184 184 self.map = None
185 185 self.pl = None
186 186
187 187 def __del__(self):
188 188 if self.dirty:
189 189 self.write()
190 190
191 191 def __getitem__(self, key):
192 192 try:
193 193 return self.map[key]
194 194 except TypeError:
195 195 self.read()
196 196 return self[key]
197 197
198 198 def __contains__(self, key):
199 199 if not self.map: self.read()
200 200 return key in self.map
201 201
202 202 def parents(self):
203 203 if not self.pl:
204 204 self.read()
205 205 return self.pl
206 206
207 207 def setparents(self, p1, p2 = nullid):
208 208 self.dirty = 1
209 209 self.pl = p1, p2
210 210
211 211 def state(self, key):
212 212 try:
213 213 return self[key][0]
214 214 except KeyError:
215 215 return "?"
216 216
217 217 def read(self):
218 218 if self.map is not None: return self.map
219 219
220 220 self.map = {}
221 221 self.pl = [nullid, nullid]
222 222 try:
223 223 st = self.opener("dirstate").read()
224 224 if not st: return
225 225 except: return
226 226
227 227 self.pl = [st[:20], st[20: 40]]
228 228
229 229 pos = 40
230 230 while pos < len(st):
231 231 e = struct.unpack(">cllll", st[pos:pos+17])
232 232 l = e[4]
233 233 pos += 17
234 234 f = st[pos:pos + l]
235 235 self.map[f] = e[:4]
236 236 pos += l
237 237
238 238 def update(self, files, state):
239 239 ''' current states:
240 240 n normal
241 241 m needs merging
242 242 r marked for removal
243 243 a marked for addition'''
244 244
245 245 if not files: return
246 246 self.read()
247 247 self.dirty = 1
248 248 for f in files:
249 249 if state == "r":
250 250 self.map[f] = ('r', 0, 0, 0)
251 251 else:
252 252 s = os.stat(os.path.join(self.root, f))
253 253 self.map[f] = (state, s.st_mode, s.st_size, s.st_mtime)
254 254
255 255 def forget(self, files):
256 256 if not files: return
257 257 self.read()
258 258 self.dirty = 1
259 259 for f in files:
260 260 try:
261 261 del self.map[f]
262 262 except KeyError:
263 263 self.ui.warn("not in dirstate: %s!\n" % f)
264 264 pass
265 265
266 266 def clear(self):
267 267 self.map = {}
268 268 self.dirty = 1
269 269
270 270 def write(self):
271 271 st = self.opener("dirstate", "w")
272 272 st.write("".join(self.pl))
273 273 for f, e in self.map.items():
274 274 e = struct.pack(">cllll", e[0], e[1], e[2], e[3], len(f))
275 275 st.write(e + f)
276 276 self.dirty = 0
277 277
278 278 def copy(self):
279 279 self.read()
280 280 return self.map.copy()
281 281
282 282 # used to avoid circular references so destructors work
283 283 def opener(base):
284 284 p = base
285 285 def o(path, mode="r"):
286 286 if p[:7] == "http://":
287 287 f = os.path.join(p, urllib.quote(path))
288 288 return httprangereader(f)
289 289
290 290 f = os.path.join(p, path)
291 291
292 292 mode += "b" # for that other OS
293 293
294 294 if mode[0] != "r":
295 295 try:
296 296 s = os.stat(f)
297 297 except OSError:
298 298 d = os.path.dirname(f)
299 299 if not os.path.isdir(d):
300 300 os.makedirs(d)
301 301 else:
302 302 if s.st_nlink > 1:
303 303 file(f + ".tmp", "w").write(file(f).read())
304 304 os.rename(f+".tmp", f)
305 305
306 306 return file(f, mode)
307 307
308 308 return o
309 309
310 310 class localrepository:
311 311 def __init__(self, ui, path=None, create=0):
312 312 self.remote = 0
313 313 if path and path[:7] == "http://":
314 314 self.remote = 1
315 315 self.path = path
316 316 else:
317 317 if not path:
318 318 p = os.getcwd()
319 319 while not os.path.isdir(os.path.join(p, ".hg")):
320 320 p = os.path.dirname(p)
321 321 if p == "/": raise "No repo found"
322 322 path = p
323 323 self.path = os.path.join(path, ".hg")
324 324
325 325 self.root = path
326 326 self.ui = ui
327 327
328 328 if create:
329 329 os.mkdir(self.path)
330 330 os.mkdir(self.join("data"))
331 331
332 332 self.opener = opener(self.path)
333 333 self.wopener = opener(self.root)
334 334 self.manifest = manifest(self.opener)
335 335 self.changelog = changelog(self.opener)
336 336 self.ignorelist = None
337 337 self.tagscache = None
338 338 self.nodetagscache = None
339 339
340 340 if not self.remote:
341 341 self.dirstate = dirstate(self.opener, ui, self.root)
342 342 try:
343 343 self.ui.readconfig(self.opener("hgrc"))
344 344 except IOError: pass
345 345
346 346 def ignore(self, f):
347 347 if self.ignorelist is None:
348 348 self.ignorelist = []
349 349 try:
350 350 l = self.wfile(".hgignore")
351 351 for pat in l:
352 352 if pat != "\n":
353 353 self.ignorelist.append(re.compile(pat[:-1]))
354 354 except IOError: pass
355 355 for pat in self.ignorelist:
356 356 if pat.search(f): return True
357 357 return False
358 358
359 359 def tags(self):
360 360 '''return a mapping of tag to node'''
361 361 if not self.tagscache:
362 362 self.tagscache = {}
363 363 try:
364 364 # read each head of the tags file, ending with the tip
365 365 # and add each tag found to the map, with "newer" ones
366 366 # taking precedence
367 367 fl = self.file(".hgtags")
368 368 h = fl.heads()
369 369 h.reverse()
370 370 for r in h:
371 371 for l in fl.revision(r).splitlines():
372 372 if l:
373 373 n, k = l.split(" ")
374 374 self.tagscache[k] = bin(n)
375 375 except KeyError: pass
376 376 self.tagscache['tip'] = self.changelog.tip()
377 377
378 378 return self.tagscache
379 379
380 380 def tagslist(self):
381 381 '''return a list of tags ordered by revision'''
382 382 l = []
383 383 for t,n in self.tags().items():
384 384 try:
385 385 r = self.changelog.rev(n)
386 386 except:
387 387 r = -2 # sort to the beginning of the list if unknown
388 388 l.append((r,t,n))
389 389 l.sort()
390 390 return [(t,n) for r,t,n in l]
391 391
392 392 def nodetags(self, node):
393 393 '''return the tags associated with a node'''
394 394 if not self.nodetagscache:
395 395 self.nodetagscache = {}
396 396 for t,n in self.tags().items():
397 397 self.nodetagscache.setdefault(n,[]).append(t)
398 398 return self.nodetagscache.get(node, [])
399 399
400 400 def lookup(self, key):
401 401 try:
402 402 return self.tags()[key]
403 403 except KeyError:
404 404 return self.changelog.lookup(key)
405 405
406 406 def join(self, f):
407 407 return os.path.join(self.path, f)
408 408
409 409 def wjoin(self, f):
410 410 return os.path.join(self.root, f)
411 411
412 412 def file(self, f):
413 413 if f[0] == '/': f = f[1:]
414 414 return filelog(self.opener, f)
415 415
416 416 def wfile(self, f, mode='r'):
417 417 return self.wopener(f, mode)
418 418
419 419 def transaction(self):
420 420 # save dirstate for undo
421 421 try:
422 422 ds = self.opener("dirstate").read()
423 423 except IOError:
424 424 ds = ""
425 425 self.opener("undo.dirstate", "w").write(ds)
426 426
427 427 return transaction.transaction(self.opener, self.join("journal"),
428 428 self.join("undo"))
429 429
430 430 def recover(self):
431 431 lock = self.lock()
432 432 if os.path.exists(self.join("recover")):
433 433 self.ui.status("attempting to rollback interrupted transaction\n")
434 434 return transaction.rollback(self.opener, self.join("recover"))
435 435 else:
436 436 self.ui.warn("no interrupted transaction available\n")
437 437
438 438 def undo(self):
439 439 lock = self.lock()
440 440 if os.path.exists(self.join("undo")):
441 441 self.ui.status("attempting to rollback last transaction\n")
442 442 transaction.rollback(self.opener, self.join("undo"))
443 443 self.dirstate = None
444 444 os.rename(self.join("undo.dirstate"), self.join("dirstate"))
445 445 self.dirstate = dirstate(self.opener, self.ui, self.root)
446 446 else:
447 447 self.ui.warn("no undo information available\n")
448 448
449 449 def lock(self, wait = 1):
450 450 try:
451 451 return lock.lock(self.join("lock"), 0)
452 452 except lock.LockHeld, inst:
453 453 if wait:
454 454 self.ui.warn("waiting for lock held by %s\n" % inst.args[0])
455 455 return lock.lock(self.join("lock"), wait)
456 456 raise inst
457 457
458 458 def rawcommit(self, files, text, user, date, p1=None, p2=None):
459 459 p1 = p1 or self.dirstate.parents()[0] or nullid
460 460 p2 = p2 or self.dirstate.parents()[1] or nullid
461 461 c1 = self.changelog.read(p1)
462 462 c2 = self.changelog.read(p2)
463 463 m1 = self.manifest.read(c1[0])
464 464 mf1 = self.manifest.readflags(c1[0])
465 465 m2 = self.manifest.read(c2[0])
466 466
467 467 tr = self.transaction()
468 468 mm = m1.copy()
469 469 mfm = mf1.copy()
470 470 linkrev = self.changelog.count()
471 471 self.dirstate.setparents(p1, p2)
472 472 for f in files:
473 473 try:
474 474 t = self.wfile(f).read()
475 475 tm = is_exec(self.wjoin(f))
476 476 r = self.file(f)
477 477 mfm[f] = tm
478 478 mm[f] = r.add(t, tr, linkrev,
479 479 m1.get(f, nullid), m2.get(f, nullid))
480 480 self.dirstate.update([f], "n")
481 481 except IOError:
482 482 try:
483 483 del mm[f]
484 484 del mfm[f]
485 485 self.dirstate.forget([f])
486 486 except:
487 487 # deleted from p2?
488 488 pass
489 489
490 490 mnode = self.manifest.add(mm, mfm, tr, linkrev, c1[0], c2[0])
491 491 n = self.changelog.add(mnode, files, text, tr, p1, p2, user, date)
492 492 tr.close()
493 493
494 494 def commit(self, files = None, text = "", user = None, date = None):
495 495 commit = []
496 496 remove = []
497 497 if files:
498 498 for f in files:
499 499 s = self.dirstate.state(f)
500 500 if s in 'nmai':
501 501 commit.append(f)
502 502 elif s == 'r':
503 503 remove.append(f)
504 504 else:
505 505 self.ui.warn("%s not tracked!\n" % f)
506 506 else:
507 507 (c, a, d, u) = self.diffdir(self.root)
508 508 commit = c + a
509 509 remove = d
510 510
511 511 if not commit and not remove:
512 512 self.ui.status("nothing changed\n")
513 513 return
514 514
515 515 p1, p2 = self.dirstate.parents()
516 516 c1 = self.changelog.read(p1)
517 517 c2 = self.changelog.read(p2)
518 518 m1 = self.manifest.read(c1[0])
519 519 mf1 = self.manifest.readflags(c1[0])
520 520 m2 = self.manifest.read(c2[0])
521 521 lock = self.lock()
522 522 tr = self.transaction()
523 523
524 524 # check in files
525 525 new = {}
526 526 linkrev = self.changelog.count()
527 527 commit.sort()
528 528 for f in commit:
529 529 self.ui.note(f + "\n")
530 530 try:
531 531 fp = self.wjoin(f)
532 532 mf1[f] = is_exec(fp)
533 533 t = file(fp).read()
534 534 except IOError:
535 535 self.warn("trouble committing %s!\n" % f)
536 536 raise
537 537
538 538 r = self.file(f)
539 539 fp1 = m1.get(f, nullid)
540 540 fp2 = m2.get(f, nullid)
541 541 new[f] = r.add(t, tr, linkrev, fp1, fp2)
542 542
543 543 # update manifest
544 544 m1.update(new)
545 545 for f in remove: del m1[f]
546 546 mn = self.manifest.add(m1, mf1, tr, linkrev, c1[0], c2[0])
547 547
548 548 # add changeset
549 549 new = new.keys()
550 550 new.sort()
551 551
552 552 if not text:
553 553 edittext = "\n" + "HG: manifest hash %s\n" % hex(mn)
554 554 edittext += "".join(["HG: changed %s\n" % f for f in new])
555 555 edittext += "".join(["HG: removed %s\n" % f for f in remove])
556 556 edittext = self.ui.edit(edittext)
557 557 if not edittext.rstrip():
558 558 return 1
559 559 text = edittext
560 560
561 561 n = self.changelog.add(mn, new, text, tr, p1, p2, user, date)
562 562 tr.close()
563 563
564 564 self.dirstate.setparents(n)
565 565 self.dirstate.update(new, "n")
566 566 self.dirstate.forget(remove)
567 567
568 568 def diffdir(self, path, changeset = None):
569 569 changed = []
570 570 added = []
571 571 unknown = []
572 572 mf = {}
573 573
574 574 if changeset:
575 575 change = self.changelog.read(changeset)
576 576 mf = self.manifest.read(change[0])
577 577 dc = dict.fromkeys(mf)
578 578 else:
579 579 changeset = self.dirstate.parents()[0]
580 580 change = self.changelog.read(changeset)
581 581 mf = self.manifest.read(change[0])
582 582 dc = self.dirstate.copy()
583 583
584 584 def fcmp(fn):
585 585 t1 = self.wfile(fn).read()
586 586 t2 = self.file(fn).revision(mf[fn])
587 587 return cmp(t1, t2)
588 588
589 589 for dir, subdirs, files in os.walk(path):
590 590 d = dir[len(self.root)+1:]
591 591 if ".hg" in subdirs: subdirs.remove(".hg")
592 592
593 593 for f in files:
594 594 fn = os.path.join(d, f)
595 595 try: s = os.stat(os.path.join(self.root, fn))
596 596 except: continue
597 597 if fn in dc:
598 598 c = dc[fn]
599 599 del dc[fn]
600 600 if not c:
601 601 if fcmp(fn):
602 602 changed.append(fn)
603 603 elif c[0] == 'm':
604 604 changed.append(fn)
605 605 elif c[0] == 'a':
606 606 added.append(fn)
607 607 elif c[0] == 'r':
608 608 unknown.append(fn)
609 609 elif c[2] != s.st_size or (c[1] ^ s.st_mode) & 0100:
610 610 changed.append(fn)
611 611 elif c[1] != s.st_mode or c[3] != s.st_mtime:
612 612 if fcmp(fn):
613 613 changed.append(fn)
614 614 else:
615 615 if self.ignore(fn): continue
616 616 unknown.append(fn)
617 617
618 618 deleted = dc.keys()
619 619 deleted.sort()
620 620
621 621 return (changed, added, deleted, unknown)
622 622
623 623 def diffrevs(self, node1, node2):
624 624 changed, added = [], []
625 625
626 626 change = self.changelog.read(node1)
627 627 mf1 = self.manifest.read(change[0])
628 628 change = self.changelog.read(node2)
629 629 mf2 = self.manifest.read(change[0])
630 630
631 631 for fn in mf2:
632 632 if mf1.has_key(fn):
633 633 if mf1[fn] != mf2[fn]:
634 634 changed.append(fn)
635 635 del mf1[fn]
636 636 else:
637 637 added.append(fn)
638 638
639 639 deleted = mf1.keys()
640 640 deleted.sort()
641 641
642 642 return (changed, added, deleted)
643 643
644 644 def add(self, list):
645 645 for f in list:
646 646 p = self.wjoin(f)
647 647 if not os.path.isfile(p):
648 648 self.ui.warn("%s does not exist!\n" % f)
649 649 elif self.dirstate.state(f) == 'n':
650 650 self.ui.warn("%s already tracked!\n" % f)
651 651 else:
652 652 self.dirstate.update([f], "a")
653 653
654 654 def forget(self, list):
655 655 for f in list:
656 656 if self.dirstate.state(f) not in 'ai':
657 657 self.ui.warn("%s not added!\n" % f)
658 658 else:
659 659 self.dirstate.forget([f])
660 660
661 661 def remove(self, list):
662 662 for f in list:
663 663 p = self.wjoin(f)
664 664 if os.path.isfile(p):
665 665 self.ui.warn("%s still exists!\n" % f)
666 666 elif f not in self.dirstate:
667 667 self.ui.warn("%s not tracked!\n" % f)
668 668 else:
669 669 self.dirstate.update([f], "r")
670 670
671 671 def heads(self):
672 672 return self.changelog.heads()
673 673
674 674 def branches(self, nodes):
675 675 if not nodes: nodes = [self.changelog.tip()]
676 676 b = []
677 677 for n in nodes:
678 678 t = n
679 679 while n:
680 680 p = self.changelog.parents(n)
681 681 if p[1] != nullid or p[0] == nullid:
682 682 b.append((t, n, p[0], p[1]))
683 683 break
684 684 n = p[0]
685 685 return b
686 686
687 687 def between(self, pairs):
688 688 r = []
689 689
690 690 for top, bottom in pairs:
691 691 n, l, i = top, [], 0
692 692 f = 1
693 693
694 694 while n != bottom:
695 695 p = self.changelog.parents(n)[0]
696 696 if i == f:
697 697 l.append(n)
698 698 f = f * 2
699 699 n = p
700 700 i += 1
701 701
702 702 r.append(l)
703 703
704 704 return r
705 705
706 706 def newer(self, nodes):
707 707 m = {}
708 708 nl = []
709 709 pm = {}
710 710 cl = self.changelog
711 711 t = l = cl.count()
712 712
713 713 # find the lowest numbered node
714 714 for n in nodes:
715 715 l = min(l, cl.rev(n))
716 716 m[n] = 1
717 717
718 718 for i in xrange(l, t):
719 719 n = cl.node(i)
720 720 if n in m: # explicitly listed
721 721 pm[n] = 1
722 722 nl.append(n)
723 723 continue
724 724 for p in cl.parents(n):
725 725 if p in pm: # parent listed
726 726 pm[n] = 1
727 727 nl.append(n)
728 728 break
729 729
730 730 return nl
731 731
732 732 def getchangegroup(self, remote):
733 733 m = self.changelog.nodemap
734 734 search = []
735 735 fetch = []
736 736 seen = {}
737 737 seenbranch = {}
738 738
739 739 # if we have an empty repo, fetch everything
740 740 if self.changelog.tip() == nullid:
741 741 self.ui.status("requesting all changes\n")
742 742 return remote.changegroup([nullid])
743 743
744 744 # otherwise, assume we're closer to the tip than the root
745 745 self.ui.status("searching for changes\n")
746 746 heads = remote.heads()
747 747 unknown = []
748 748 for h in heads:
749 749 if h not in m:
750 750 unknown.append(h)
751 751
752 752 if not unknown:
753 753 self.ui.status("nothing to do!\n")
754 754 return None
755 755
756 756 rep = {}
757 757 reqcnt = 0
758 758
759 759 unknown = remote.branches(unknown)
760 760 while unknown:
761 761 r = []
762 762 while unknown:
763 763 n = unknown.pop(0)
764 764 if n[0] in seen:
765 765 continue
766 766
767 767 self.ui.debug("examining %s:%s\n" % (short(n[0]), short(n[1])))
768 768 if n[0] == nullid:
769 769 break
770 770 if n in seenbranch:
771 771 self.ui.debug("branch already found\n")
772 772 continue
773 773 if n[1] and n[1] in m: # do we know the base?
774 774 self.ui.debug("found incomplete branch %s:%s\n"
775 775 % (short(n[0]), short(n[1])))
776 776 search.append(n) # schedule branch range for scanning
777 777 seenbranch[n] = 1
778 778 else:
779 779 if n[1] not in seen and n[1] not in fetch:
780 780 if n[2] in m and n[3] in m:
781 781 self.ui.debug("found new changeset %s\n" %
782 782 short(n[1]))
783 783 fetch.append(n[1]) # earliest unknown
784 784 continue
785 785
786 786 for a in n[2:4]:
787 787 if a not in rep:
788 788 r.append(a)
789 789 rep[a] = 1
790 790
791 791 seen[n[0]] = 1
792 792
793 793 if r:
794 794 reqcnt += 1
795 795 self.ui.debug("request %d: %s\n" %
796 796 (reqcnt, " ".join(map(short, r))))
797 797 for p in range(0, len(r), 10):
798 798 for b in remote.branches(r[p:p+10]):
799 799 self.ui.debug("received %s:%s\n" %
800 800 (short(b[0]), short(b[1])))
801 801 if b[0] not in m and b[0] not in seen:
802 802 unknown.append(b)
803 803
804 804 while search:
805 805 n = search.pop(0)
806 806 reqcnt += 1
807 807 l = remote.between([(n[0], n[1])])[0]
808 808 l.append(n[1])
809 809 p = n[0]
810 810 f = 1
811 811 for i in l:
812 812 self.ui.debug("narrowing %d:%d %s\n" % (f, len(l), short(i)))
813 813 if i in m:
814 814 if f <= 2:
815 815 self.ui.debug("found new branch changeset %s\n" %
816 816 short(p))
817 817 fetch.append(p)
818 818 else:
819 819 self.ui.debug("narrowed branch search to %s:%s\n"
820 820 % (short(p), short(i)))
821 821 search.append((p, i))
822 822 break
823 823 p, f = i, f * 2
824 824
825 825 for f in fetch:
826 826 if f in m:
827 827 raise "already have", short(f[:4])
828 828
829 829 self.ui.note("adding new changesets starting at " +
830 830 " ".join([short(f) for f in fetch]) + "\n")
831 831
832 832 self.ui.debug("%d total queries\n" % reqcnt)
833 833
834 834 return remote.changegroup(fetch)
835 835
836 836 def changegroup(self, basenodes):
837 837 nodes = self.newer(basenodes)
838 838
839 839 # construct the link map
840 840 linkmap = {}
841 841 for n in nodes:
842 842 linkmap[self.changelog.rev(n)] = n
843 843
844 844 # construct a list of all changed files
845 845 changed = {}
846 846 for n in nodes:
847 847 c = self.changelog.read(n)
848 848 for f in c[3]:
849 849 changed[f] = 1
850 850 changed = changed.keys()
851 851 changed.sort()
852 852
853 853 # the changegroup is changesets + manifests + all file revs
854 854 revs = [ self.changelog.rev(n) for n in nodes ]
855 855
856 856 for y in self.changelog.group(linkmap): yield y
857 857 for y in self.manifest.group(linkmap): yield y
858 858 for f in changed:
859 859 yield struct.pack(">l", len(f) + 4) + f
860 860 g = self.file(f).group(linkmap)
861 861 for y in g:
862 862 yield y
863 863
864 864 def addchangegroup(self, generator):
865 865
866 866 class genread:
867 867 def __init__(self, generator):
868 868 self.g = generator
869 869 self.buf = ""
870 870 def read(self, l):
871 871 while l > len(self.buf):
872 872 try:
873 873 self.buf += self.g.next()
874 874 except StopIteration:
875 875 break
876 876 d, self.buf = self.buf[:l], self.buf[l:]
877 877 return d
878 878
879 879 def getchunk():
880 880 d = source.read(4)
881 881 if not d: return ""
882 882 l = struct.unpack(">l", d)[0]
883 883 if l <= 4: return ""
884 884 return source.read(l - 4)
885 885
886 886 def getgroup():
887 887 while 1:
888 888 c = getchunk()
889 889 if not c: break
890 890 yield c
891 891
892 892 def csmap(x):
893 893 self.ui.debug("add changeset %s\n" % short(x))
894 894 return self.changelog.count()
895 895
896 896 def revmap(x):
897 897 return self.changelog.rev(x)
898 898
899 899 if not generator: return
900 900 changesets = files = revisions = 0
901 901
902 902 source = genread(generator)
903 903 lock = self.lock()
904 904 tr = self.transaction()
905 905
906 906 # pull off the changeset group
907 907 self.ui.status("adding changesets\n")
908 908 co = self.changelog.tip()
909 909 cn = self.changelog.addgroup(getgroup(), csmap, tr, 1) # unique
910 910 changesets = self.changelog.rev(cn) - self.changelog.rev(co)
911 911
912 912 # pull off the manifest group
913 913 self.ui.status("adding manifests\n")
914 914 mm = self.manifest.tip()
915 915 mo = self.manifest.addgroup(getgroup(), revmap, tr)
916 916
917 917 # process the files
918 918 self.ui.status("adding file revisions\n")
919 919 while 1:
920 920 f = getchunk()
921 921 if not f: break
922 922 self.ui.debug("adding %s revisions\n" % f)
923 923 fl = self.file(f)
924 924 o = fl.tip()
925 925 n = fl.addgroup(getgroup(), revmap, tr)
926 926 revisions += fl.rev(n) - fl.rev(o)
927 927 files += 1
928 928
929 929 self.ui.status(("modified %d files, added %d changesets" +
930 930 " and %d new revisions\n")
931 931 % (files, changesets, revisions))
932 932
933 933 tr.close()
934 934 return
935 935
936 936 def update(self, node, allow=False, force=False):
937 937 pl = self.dirstate.parents()
938 938 if not force and pl[1] != nullid:
939 939 self.ui.warn("aborting: outstanding uncommitted merges\n")
940 940 return
941 941
942 942 p1, p2 = pl[0], node
943 943 pa = self.changelog.ancestor(p1, p2)
944 944 m1n = self.changelog.read(p1)[0]
945 945 m2n = self.changelog.read(p2)[0]
946 946 man = self.manifest.ancestor(m1n, m2n)
947 947 m1 = self.manifest.read(m1n)
948 948 mf1 = self.manifest.readflags(m1n)
949 949 m2 = self.manifest.read(m2n)
950 950 mf2 = self.manifest.readflags(m2n)
951 951 ma = self.manifest.read(man)
952 952 mfa = self.manifest.readflags(m2n)
953 953
954 954 (c, a, d, u) = self.diffdir(self.root)
955 955
956 956 # resolve the manifest to determine which files
957 957 # we care about merging
958 958 self.ui.note("resolving manifests\n")
959 959 self.ui.debug(" ancestor %s local %s remote %s\n" %
960 960 (short(man), short(m1n), short(m2n)))
961 961
962 962 merge = {}
963 963 get = {}
964 964 remove = []
965 965 mark = {}
966 966
967 967 # construct a working dir manifest
968 968 mw = m1.copy()
969 969 mfw = mf1.copy()
970 970 for f in a + c + u:
971 971 mw[f] = ""
972 972 mfw[f] = is_exec(self.wjoin(f))
973 973 for f in d:
974 974 if f in mw: del mw[f]
975 975
976 976 for f, n in mw.iteritems():
977 977 if f in m2:
978 978 s = 0
979 979
980 980 # are files different?
981 981 if n != m2[f]:
982 982 a = ma.get(f, nullid)
983 983 # are both different from the ancestor?
984 984 if n != a and m2[f] != a:
985 985 self.ui.debug(" %s versions differ, resolve\n" % f)
986 986 merge[f] = (m1.get(f, nullid), m2[f])
987 987 # merge executable bits
988 988 # "if we changed or they changed, change in merge"
989 989 a, b, c = mfa.get(f, 0), mfw[f], mf2[f]
990 990 mode = ((a^b) | (a^c)) ^ a
991 991 merge[f] = (m1.get(f, nullid), m2[f], mode)
992 992 s = 1
993 993 # are we clobbering?
994 994 # is remote's version newer?
995 995 # or are we going back in time?
996 996 elif force or m2[f] != a or (p2 == pa and mw[f] == m1[f]):
997 997 self.ui.debug(" remote %s is newer, get\n" % f)
998 998 get[f] = m2[f]
999 999 s = 1
1000 1000 else:
1001 1001 mark[f] = 1
1002 1002
1003 1003 if not s and mfw[f] != mf2[f]:
1004 1004 if force:
1005 1005 self.ui.debug(" updating permissions for %s\n" % f)
1006 1006 set_exec(self.wjoin(f), mf2[f])
1007 1007 else:
1008 1008 a, b, c = mfa.get(f, 0), mfw[f], mf2[f]
1009 1009 mode = ((a^b) | (a^c)) ^ a
1010 1010 if mode != b:
1011 1011 self.ui.debug(" updating permissions for %s\n" % f)
1012 1012 set_exec(self.wjoin(f), mode)
1013 1013 mark[f] = 1
1014 1014 del m2[f]
1015 1015 elif f in ma:
1016 1016 if not force and n != ma[f]:
1017 1017 r = self.ui.prompt(
1018 1018 (" local changed %s which remote deleted\n" % f) +
1019 1019 "(k)eep or (d)elete?", "[kd]", "k")
1020 1020 if r == "d":
1021 1021 remove.append(f)
1022 1022 else:
1023 1023 self.ui.debug("other deleted %s\n" % f)
1024 1024 remove.append(f) # other deleted it
1025 1025 else:
1026 1026 if n == m1.get(f, nullid): # same as parent
1027 self.ui.debug("remote deleted %s\n" % f)
1028 remove.append(f)
1027 self.ui.debug("local created %s, keeping\n" % f)
1029 1028 else:
1030 1029 self.ui.debug("working dir created %s, keeping\n" % f)
1031 1030
1032 1031 for f, n in m2.iteritems():
1033 1032 if f[0] == "/": continue
1034 1033 if not force and f in ma and n != ma[f]:
1035 1034 r = self.ui.prompt(
1036 1035 ("remote changed %s which local deleted\n" % f) +
1037 1036 "(k)eep or (d)elete?", "[kd]", "k")
1038 1037 if r == "d": remove.append(f)
1039 1038 else:
1040 1039 self.ui.debug("remote created %s\n" % f)
1041 1040 get[f] = n
1042 1041
1043 1042 del mw, m1, m2, ma
1044 1043
1045 1044 if force:
1046 1045 for f in merge:
1047 1046 get[f] = merge[f][1]
1048 1047 merge = {}
1049 1048
1050 1049 if pa == p1 or pa == p2:
1051 1050 # we don't need to do any magic, just jump to the new rev
1052 1051 mode = 'n'
1053 1052 p1, p2 = p2, nullid
1054 1053 else:
1055 1054 if not allow:
1056 1055 self.ui.status("this update spans a branch" +
1057 1056 " affecting the following files:\n")
1058 1057 fl = merge.keys() + get.keys()
1059 1058 fl.sort()
1060 1059 for f in fl:
1061 1060 cf = ""
1062 1061 if f in merge: cf = " (resolve)"
1063 1062 self.ui.status(" %s%s\n" % (f, cf))
1064 1063 self.ui.warn("aborting update spanning branches!\n")
1065 1064 self.ui.status("(use update -m to perform a branch merge)\n")
1066 1065 return 1
1067 1066 # we have to remember what files we needed to get/change
1068 1067 # because any file that's different from either one of its
1069 1068 # parents must be in the changeset
1070 1069 mode = 'm'
1071 1070 self.dirstate.update(mark.keys(), "m")
1072 1071
1073 1072 self.dirstate.setparents(p1, p2)
1074 1073
1075 1074 # get the files we don't need to change
1076 1075 files = get.keys()
1077 1076 files.sort()
1078 1077 for f in files:
1079 1078 if f[0] == "/": continue
1080 1079 self.ui.note("getting %s\n" % f)
1081 1080 t = self.file(f).read(get[f])
1082 1081 try:
1083 1082 self.wfile(f, "w").write(t)
1084 1083 except IOError:
1085 1084 os.makedirs(os.path.dirname(self.wjoin(f)))
1086 1085 self.wfile(f, "w").write(t)
1087 1086 set_exec(self.wjoin(f), mf2[f])
1088 1087 self.dirstate.update([f], mode)
1089 1088
1090 1089 # merge the tricky bits
1091 1090 files = merge.keys()
1092 1091 files.sort()
1093 1092 for f in files:
1094 1093 self.ui.status("merging %s\n" % f)
1095 1094 m, o, flag = merge[f]
1096 1095 self.merge3(f, m, o)
1097 1096 set_exec(self.wjoin(f), flag)
1098 1097 self.dirstate.update([f], 'm')
1099 1098
1100 1099 for f in remove:
1101 1100 self.ui.note("removing %s\n" % f)
1102 1101 os.unlink(f)
1103 1102 if mode == 'n':
1104 1103 self.dirstate.forget(remove)
1105 1104 else:
1106 1105 self.dirstate.update(remove, 'r')
1107 1106
1108 1107 def merge3(self, fn, my, other):
1109 1108 """perform a 3-way merge in the working directory"""
1110 1109
1111 1110 def temp(prefix, node):
1112 1111 pre = "%s~%s." % (os.path.basename(fn), prefix)
1113 1112 (fd, name) = tempfile.mkstemp("", pre)
1114 1113 f = os.fdopen(fd, "w")
1115 1114 f.write(fl.revision(node))
1116 1115 f.close()
1117 1116 return name
1118 1117
1119 1118 fl = self.file(fn)
1120 1119 base = fl.ancestor(my, other)
1121 1120 a = self.wjoin(fn)
1122 1121 b = temp("other", other)
1123 1122 c = temp("base", base)
1124 1123
1125 1124 self.ui.note("resolving %s\n" % fn)
1126 1125 self.ui.debug("file %s: other %s ancestor %s\n" %
1127 1126 (fn, short(other), short(base)))
1128 1127
1129 1128 cmd = os.environ.get("HGMERGE", "hgmerge")
1130 1129 r = os.system("%s %s %s %s" % (cmd, a, b, c))
1131 1130 if r:
1132 1131 self.ui.warn("merging %s failed!\n" % fn)
1133 1132
1134 1133 os.unlink(b)
1135 1134 os.unlink(c)
1136 1135
1137 1136 def verify(self):
1138 1137 filelinkrevs = {}
1139 1138 filenodes = {}
1140 1139 changesets = revisions = files = 0
1141 1140 errors = 0
1142 1141
1143 1142 seen = {}
1144 1143 self.ui.status("checking changesets\n")
1145 1144 for i in range(self.changelog.count()):
1146 1145 changesets += 1
1147 1146 n = self.changelog.node(i)
1148 1147 if n in seen:
1149 1148 self.ui.warn("duplicate changeset at revision %d\n" % i)
1150 1149 errors += 1
1151 1150 seen[n] = 1
1152 1151
1153 1152 for p in self.changelog.parents(n):
1154 1153 if p not in self.changelog.nodemap:
1155 1154 self.ui.warn("changeset %s has unknown parent %s\n" %
1156 1155 (short(n), short(p)))
1157 1156 errors += 1
1158 1157 try:
1159 1158 changes = self.changelog.read(n)
1160 1159 except Exception, inst:
1161 1160 self.ui.warn("unpacking changeset %s: %s\n" % (short(n), inst))
1162 1161 errors += 1
1163 1162
1164 1163 for f in changes[3]:
1165 1164 filelinkrevs.setdefault(f, []).append(i)
1166 1165
1167 1166 seen = {}
1168 1167 self.ui.status("checking manifests\n")
1169 1168 for i in range(self.manifest.count()):
1170 1169 n = self.manifest.node(i)
1171 1170 if n in seen:
1172 1171 self.ui.warn("duplicate manifest at revision %d\n" % i)
1173 1172 errors += 1
1174 1173 seen[n] = 1
1175 1174
1176 1175 for p in self.manifest.parents(n):
1177 1176 if p not in self.manifest.nodemap:
1178 1177 self.ui.warn("manifest %s has unknown parent %s\n" %
1179 1178 (short(n), short(p)))
1180 1179 errors += 1
1181 1180
1182 1181 try:
1183 1182 delta = mdiff.patchtext(self.manifest.delta(n))
1184 1183 except KeyboardInterrupt:
1185 1184 print "aborted"
1186 1185 sys.exit(0)
1187 1186 except Exception, inst:
1188 1187 self.ui.warn("unpacking manifest %s: %s\n"
1189 1188 % (short(n), inst))
1190 1189 errors += 1
1191 1190
1192 1191 ff = [ l.split('\0') for l in delta.splitlines() ]
1193 1192 for f, fn in ff:
1194 1193 filenodes.setdefault(f, {})[bin(fn[:40])] = 1
1195 1194
1196 1195 self.ui.status("crosschecking files in changesets and manifests\n")
1197 1196 for f in filenodes:
1198 1197 if f not in filelinkrevs:
1199 1198 self.ui.warn("file %s in manifest but not in changesets\n" % f)
1200 1199 errors += 1
1201 1200
1202 1201 for f in filelinkrevs:
1203 1202 if f not in filenodes:
1204 1203 self.ui.warn("file %s in changeset but not in manifest\n" % f)
1205 1204 errors += 1
1206 1205
1207 1206 self.ui.status("checking files\n")
1208 1207 ff = filenodes.keys()
1209 1208 ff.sort()
1210 1209 for f in ff:
1211 1210 if f == "/dev/null": continue
1212 1211 files += 1
1213 1212 fl = self.file(f)
1214 1213 nodes = { nullid: 1 }
1215 1214 seen = {}
1216 1215 for i in range(fl.count()):
1217 1216 revisions += 1
1218 1217 n = fl.node(i)
1219 1218
1220 1219 if n in seen:
1221 1220 self.ui.warn("%s: duplicate revision %d\n" % (f, i))
1222 1221 errors += 1
1223 1222
1224 1223 if n not in filenodes[f]:
1225 1224 self.ui.warn("%s: %d:%s not in manifests\n"
1226 1225 % (f, i, short(n)))
1227 1226 print len(filenodes[f].keys()), fl.count(), f
1228 1227 errors += 1
1229 1228 else:
1230 1229 del filenodes[f][n]
1231 1230
1232 1231 flr = fl.linkrev(n)
1233 1232 if flr not in filelinkrevs[f]:
1234 1233 self.ui.warn("%s:%s points to unexpected changeset %d\n"
1235 1234 % (f, short(n), fl.linkrev(n)))
1236 1235 errors += 1
1237 1236 else:
1238 1237 filelinkrevs[f].remove(flr)
1239 1238
1240 1239 # verify contents
1241 1240 try:
1242 1241 t = fl.read(n)
1243 1242 except Exception, inst:
1244 1243 self.ui.warn("unpacking file %s %s: %s\n"
1245 1244 % (f, short(n), inst))
1246 1245 errors += 1
1247 1246
1248 1247 # verify parents
1249 1248 (p1, p2) = fl.parents(n)
1250 1249 if p1 not in nodes:
1251 1250 self.ui.warn("file %s:%s unknown parent 1 %s" %
1252 1251 (f, short(n), short(p1)))
1253 1252 errors += 1
1254 1253 if p2 not in nodes:
1255 1254 self.ui.warn("file %s:%s unknown parent 2 %s" %
1256 1255 (f, short(n), short(p1)))
1257 1256 errors += 1
1258 1257 nodes[n] = 1
1259 1258
1260 1259 # cross-check
1261 1260 for node in filenodes[f]:
1262 1261 self.ui.warn("node %s in manifests not in %s\n"
1263 1262 % (hex(n), f))
1264 1263 errors += 1
1265 1264
1266 1265 self.ui.status("%d files, %d changesets, %d total revisions\n" %
1267 1266 (files, changesets, revisions))
1268 1267
1269 1268 if errors:
1270 1269 self.ui.warn("%d integrity errors encountered!\n" % errors)
1271 1270 return 1
1272 1271
1273 1272 class remoterepository:
1274 1273 def __init__(self, ui, path):
1275 1274 self.url = path
1276 1275 self.ui = ui
1277 1276 no_list = [ "localhost", "127.0.0.1" ]
1278 1277 host = ui.config("http_proxy", "host")
1279 1278 user = ui.config("http_proxy", "user")
1280 1279 passwd = ui.config("http_proxy", "passwd")
1281 1280 no = ui.config("http_proxy", "no")
1282 1281 if no:
1283 1282 no_list = no_list + no.split(",")
1284 1283
1285 1284 no_proxy = 0
1286 1285 for h in no_list:
1287 1286 if (path.startswith("http://" + h + "/") or
1288 1287 path.startswith("http://" + h + ":") or
1289 1288 path == "http://" + h):
1290 1289 no_proxy = 1
1291 1290
1292 1291 # Note: urllib2 takes proxy values from the environment and those will
1293 1292 # take precedence
1294 1293
1295 1294 proxy_handler = urllib2.BaseHandler()
1296 1295 if host and not no_proxy:
1297 1296 proxy_handler = urllib2.ProxyHandler({"http" : "http://" + host})
1298 1297
1299 1298 authinfo = None
1300 1299 if user and passwd:
1301 1300 passmgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
1302 1301 passmgr.add_password(None, host, user, passwd)
1303 1302 authinfo = urllib2.ProxyBasicAuthHandler(passmgr)
1304 1303
1305 1304 opener = urllib2.build_opener(proxy_handler, authinfo)
1306 1305 urllib2.install_opener(opener)
1307 1306
1308 1307 def do_cmd(self, cmd, **args):
1309 1308 self.ui.debug("sending %s command\n" % cmd)
1310 1309 q = {"cmd": cmd}
1311 1310 q.update(args)
1312 1311 qs = urllib.urlencode(q)
1313 1312 cu = "%s?%s" % (self.url, qs)
1314 1313 return urllib2.urlopen(cu)
1315 1314
1316 1315 def heads(self):
1317 1316 d = self.do_cmd("heads").read()
1318 1317 try:
1319 1318 return map(bin, d[:-1].split(" "))
1320 1319 except:
1321 1320 self.ui.warn("unexpected response:\n" + d[:400] + "\n...\n")
1322 1321 raise
1323 1322
1324 1323 def branches(self, nodes):
1325 1324 n = " ".join(map(hex, nodes))
1326 1325 d = self.do_cmd("branches", nodes=n).read()
1327 1326 try:
1328 1327 br = [ tuple(map(bin, b.split(" "))) for b in d.splitlines() ]
1329 1328 return br
1330 1329 except:
1331 1330 self.ui.warn("unexpected response:\n" + d[:400] + "\n...\n")
1332 1331 raise
1333 1332
1334 1333 def between(self, pairs):
1335 1334 n = "\n".join(["-".join(map(hex, p)) for p in pairs])
1336 1335 d = self.do_cmd("between", pairs=n).read()
1337 1336 try:
1338 1337 p = [ l and map(bin, l.split(" ")) or [] for l in d.splitlines() ]
1339 1338 return p
1340 1339 except:
1341 1340 self.ui.warn("unexpected response:\n" + d[:400] + "\n...\n")
1342 1341 raise
1343 1342
1344 1343 def changegroup(self, nodes):
1345 1344 n = " ".join(map(hex, nodes))
1346 1345 zd = zlib.decompressobj()
1347 1346 f = self.do_cmd("changegroup", roots=n)
1348 1347 bytes = 0
1349 1348 while 1:
1350 1349 d = f.read(4096)
1351 1350 bytes += len(d)
1352 1351 if not d:
1353 1352 yield zd.flush()
1354 1353 break
1355 1354 yield zd.decompress(d)
1356 1355 self.ui.note("%d bytes of data transfered\n" % bytes)
1357 1356
1358 1357 def repository(ui, path=None, create=0):
1359 1358 if path and path[:7] == "http://":
1360 1359 return remoterepository(ui, path)
1361 1360 if path and path[:5] == "hg://":
1362 1361 return remoterepository(ui, path.replace("hg://", "http://"))
1363 1362 if path and path[:11] == "old-http://":
1364 1363 return localrepository(ui, path.replace("old-http://", "http://"))
1365 1364 else:
1366 1365 return localrepository(ui, path, create)
1367 1366
1368 1367 class httprangereader:
1369 1368 def __init__(self, url):
1370 1369 self.url = url
1371 1370 self.pos = 0
1372 1371 def seek(self, pos):
1373 1372 self.pos = pos
1374 1373 def read(self, bytes=None):
1375 1374 opener = urllib2.build_opener(byterange.HTTPRangeHandler())
1376 1375 urllib2.install_opener(opener)
1377 1376 req = urllib2.Request(self.url)
1378 1377 end = ''
1379 1378 if bytes: end = self.pos + bytes
1380 1379 req.add_header('Range', 'bytes=%d-%s' % (self.pos, end))
1381 1380 f = urllib2.urlopen(req)
1382 1381 return f.read()
General Comments 0
You need to be logged in to leave comments. Login now