##// END OF EJS Templates
Beginnings of transaction undo support
mpm@selenic.com -
r95:589f507b default
parent child Browse files
Show More
@@ -1,825 +1,826 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, sha, socket, os, time, base64, re, urllib2
9 9 import urllib
10 10 from mercurial import byterange
11 11 from mercurial.transaction import *
12 12 from mercurial.revlog import *
13 13 from difflib import SequenceMatcher
14 14
15 15 class filelog(revlog):
16 16 def __init__(self, opener, path):
17 17 s = self.encodepath(path)
18 18 revlog.__init__(self, opener, os.path.join("data", s + "i"),
19 19 os.path.join("data", s))
20 20
21 21 def encodepath(self, path):
22 22 s = sha.sha(path).digest()
23 23 s = base64.encodestring(s)[:-3]
24 24 s = re.sub("\+", "%", s)
25 25 s = re.sub("/", "_", s)
26 26 return s
27 27
28 28 def read(self, node):
29 29 return self.revision(node)
30 30 def add(self, text, transaction, link, p1=None, p2=None):
31 31 return self.addrevision(text, transaction, link, p1, p2)
32 32
33 33 def annotate(self, node):
34 34 revs = []
35 35 while node != nullid:
36 36 revs.append(node)
37 37 node = self.parents(node)[0]
38 38 revs.reverse()
39 39 prev = []
40 40 annotate = []
41 41 for node in revs:
42 42 curr = self.read(node).splitlines(1)
43 43 linkrev = self.linkrev(node)
44 44 sm = SequenceMatcher(None, prev, curr)
45 45 offset = 0
46 46 for o, m, n, s, t in sm.get_opcodes():
47 47 if o in ('insert','replace'):
48 48 annotate[m+offset:n+offset] = \
49 49 [ (linkrev, l) for l in curr[s:t]]
50 50 if o == 'insert':
51 51 offset += m-n
52 52 elif o == 'delete':
53 53 del annotate[m+offset:n+offset]
54 54 offset -= m-n
55 55 assert len(annotate) == len(curr)
56 56 prev = curr
57 57 return annotate
58 58
59 59 class manifest(revlog):
60 60 def __init__(self, opener):
61 61 self.mapcache = None
62 62 self.listcache = None
63 63 self.addlist = None
64 64 revlog.__init__(self, opener, "00manifest.i", "00manifest.d")
65 65
66 66 def read(self, node):
67 67 if self.mapcache and self.mapcache[0] == node:
68 68 return self.mapcache[1].copy()
69 69 text = self.revision(node)
70 70 map = {}
71 71 self.listcache = (text, text.splitlines(1))
72 72 for l in self.listcache[1]:
73 73 (f, n) = l.split('\0')
74 74 map[f] = bin(n[:40])
75 75 self.mapcache = (node, map)
76 76 return map
77 77
78 78 def diff(self, a, b):
79 79 # this is sneaky, as we're not actually using a and b
80 80 if self.listcache and len(self.listcache[0]) == len(a):
81 81 return mdiff.diff(self.listcache[1], self.addlist, 1)
82 82 else:
83 83 return mdiff.textdiff(a, b)
84 84
85 85 def add(self, map, transaction, link, p1=None, p2=None):
86 86 files = map.keys()
87 87 files.sort()
88 88
89 89 self.addlist = ["%s\000%s\n" % (f, hex(map[f])) for f in files]
90 90 text = "".join(self.addlist)
91 91
92 92 n = self.addrevision(text, transaction, link, p1, p2)
93 93 self.mapcache = (n, map)
94 94 self.listcache = (text, self.addlist)
95 95
96 96 return n
97 97
98 98 class changelog(revlog):
99 99 def __init__(self, opener):
100 100 revlog.__init__(self, opener, "00changelog.i", "00changelog.d")
101 101
102 102 def extract(self, text):
103 103 if not text:
104 104 return (nullid, "", "0", [], "")
105 105 last = text.index("\n\n")
106 106 desc = text[last + 2:]
107 107 l = text[:last].splitlines()
108 108 manifest = bin(l[0])
109 109 user = l[1]
110 110 date = l[2]
111 111 files = l[3:]
112 112 return (manifest, user, date, files, desc)
113 113
114 114 def read(self, node):
115 115 return self.extract(self.revision(node))
116 116
117 117 def add(self, manifest, list, desc, transaction, p1=None, p2=None):
118 118 user = (os.environ.get("HGUSER") or
119 119 os.environ.get("EMAIL") or
120 120 os.environ.get("LOGNAME", "unknown") + '@' + socket.getfqdn())
121 121 date = "%d %d" % (time.time(), time.timezone)
122 122 list.sort()
123 123 l = [hex(manifest), user, date] + list + ["", desc]
124 124 text = "\n".join(l)
125 125 return self.addrevision(text, transaction, self.count(), p1, p2)
126 126
127 127 class dircache:
128 128 def __init__(self, opener, ui):
129 129 self.opener = opener
130 130 self.dirty = 0
131 131 self.ui = ui
132 132 self.map = None
133 133 def __del__(self):
134 134 if self.dirty: self.write()
135 135 def __getitem__(self, key):
136 136 try:
137 137 return self.map[key]
138 138 except TypeError:
139 139 self.read()
140 140 return self[key]
141 141
142 142 def read(self):
143 143 if self.map is not None: return self.map
144 144
145 145 self.map = {}
146 146 try:
147 147 st = self.opener("dircache").read()
148 148 except: return
149 149
150 150 pos = 0
151 151 while pos < len(st):
152 152 e = struct.unpack(">llll", st[pos:pos+16])
153 153 l = e[3]
154 154 pos += 16
155 155 f = st[pos:pos + l]
156 156 self.map[f] = e[:3]
157 157 pos += l
158 158
159 159 def update(self, files):
160 160 if not files: return
161 161 self.read()
162 162 self.dirty = 1
163 163 for f in files:
164 164 try:
165 165 s = os.stat(f)
166 166 self.map[f] = (s.st_mode, s.st_size, s.st_mtime)
167 167 except IOError:
168 168 self.remove(f)
169 169
170 170 def taint(self, files):
171 171 if not files: return
172 172 self.read()
173 173 self.dirty = 1
174 174 for f in files:
175 175 self.map[f] = (0, -1, 0)
176 176
177 177 def remove(self, files):
178 178 if not files: return
179 179 self.read()
180 180 self.dirty = 1
181 181 for f in files:
182 182 try:
183 183 del self.map[f]
184 184 except KeyError:
185 185 self.ui.warn("Not in dircache: %s\n" % f)
186 186 pass
187 187
188 188 def clear(self):
189 189 self.map = {}
190 190 self.dirty = 1
191 191
192 192 def write(self):
193 193 st = self.opener("dircache", "w")
194 194 for f, e in self.map.items():
195 195 e = struct.pack(">llll", e[0], e[1], e[2], len(f))
196 196 st.write(e + f)
197 197 self.dirty = 0
198 198
199 199 def copy(self):
200 200 self.read()
201 201 return self.map.copy()
202 202
203 203 # used to avoid circular references so destructors work
204 204 def opener(base):
205 205 p = base
206 206 def o(path, mode="r"):
207 207 if p[:7] == "http://":
208 208 f = os.path.join(p, urllib.quote(path))
209 209 return httprangereader(f)
210 210
211 211 f = os.path.join(p, path)
212 212
213 213 if mode != "r" and os.path.isfile(f):
214 214 s = os.stat(f)
215 215 if s.st_nlink > 1:
216 216 file(f + ".tmp", "w").write(file(f).read())
217 217 os.rename(f+".tmp", f)
218 218
219 219 return file(f, mode)
220 220
221 221 return o
222 222
223 223 class localrepository:
224 224 def __init__(self, ui, path=None, create=0):
225 225 self.remote = 0
226 226 if path and path[:7] == "http://":
227 227 self.remote = 1
228 228 self.path = path
229 229 else:
230 230 if not path:
231 231 p = os.getcwd()
232 232 while not os.path.isdir(os.path.join(p, ".hg")):
233 233 p = os.path.dirname(p)
234 234 if p == "/": raise "No repo found"
235 235 path = p
236 236 self.path = os.path.join(path, ".hg")
237 237
238 238 self.root = path
239 239 self.ui = ui
240 240
241 241 if create:
242 242 os.mkdir(self.path)
243 243 os.mkdir(self.join("data"))
244 244
245 245 self.opener = opener(self.path)
246 246 self.manifest = manifest(self.opener)
247 247 self.changelog = changelog(self.opener)
248 248 self.ignorelist = None
249 249 self.tags = None
250 250
251 251 if not self.remote:
252 252 self.dircache = dircache(self.opener, ui)
253 253 try:
254 254 self.current = bin(self.opener("current").read())
255 255 except IOError:
256 256 self.current = None
257 257
258 258 def setcurrent(self, node):
259 259 self.current = node
260 260 self.opener("current", "w").write(hex(node))
261 261
262 262 def ignore(self, f):
263 263 if self.ignorelist is None:
264 264 self.ignorelist = []
265 265 try:
266 266 l = open(os.path.join(self.root, ".hgignore"))
267 267 for pat in l:
268 268 if pat != "\n":
269 269 self.ignorelist.append(re.compile(pat[:-1]))
270 270 except IOError: pass
271 271 for pat in self.ignorelist:
272 272 if pat.search(f): return True
273 273 return False
274 274
275 275 def lookup(self, key):
276 276 if self.tags is None:
277 277 self.tags = {}
278 278 try:
279 279 fl = self.file(".hgtags")
280 280 for l in fl.revision(fl.tip()).splitlines():
281 281 if l:
282 282 n, k = l.split(" ")
283 283 self.tags[k] = bin(n)
284 284 except KeyError: pass
285 285 try:
286 286 return self.tags[key]
287 287 except KeyError:
288 288 return self.changelog.lookup(key)
289 289
290 290 def join(self, f):
291 291 return os.path.join(self.path, f)
292 292
293 293 def file(self, f):
294 294 return filelog(self.opener, f)
295 295
296 296 def transaction(self):
297 return transaction(self.opener, self.join("journal"))
297 return transaction(self.opener, self.join("journal"),
298 self.join("undo"))
298 299
299 300 def commit(self, parent, update = None, text = ""):
300 301 tr = self.transaction()
301 302
302 303 try:
303 304 remove = [ l[:-1] for l in self.opener("to-remove") ]
304 305 os.unlink(self.join("to-remove"))
305 306
306 307 except IOError:
307 308 remove = []
308 309
309 310 if update == None:
310 311 update = self.diffdir(self.root, parent)[0]
311 312
312 313 # check in files
313 314 new = {}
314 315 linkrev = self.changelog.count()
315 316 for f in update:
316 317 self.ui.note(f + "\n")
317 318 try:
318 319 t = file(f).read()
319 320 except IOError:
320 321 remove.append(f)
321 322 continue
322 323 r = self.file(f)
323 324 new[f] = r.add(t, tr, linkrev)
324 325
325 326 # update manifest
326 327 mmap = self.manifest.read(self.manifest.tip())
327 328 mmap.update(new)
328 329 for f in remove:
329 330 del mmap[f]
330 331 mnode = self.manifest.add(mmap, tr, linkrev)
331 332
332 333 # add changeset
333 334 new = new.keys()
334 335 new.sort()
335 336
336 337 edittext = text + "\n"+"".join(["HG: changed %s\n" % f for f in new])
337 338 edittext += "".join(["HG: removed %s\n" % f for f in remove])
338 339 edittext = self.ui.edit(edittext)
339 340
340 341 n = self.changelog.add(mnode, new, edittext, tr)
341 342 tr.close()
342 343
343 344 self.setcurrent(n)
344 345 self.dircache.update(new)
345 346 self.dircache.remove(remove)
346 347
347 348 def checkdir(self, path):
348 349 d = os.path.dirname(path)
349 350 if not d: return
350 351 if not os.path.isdir(d):
351 352 self.checkdir(d)
352 353 os.mkdir(d)
353 354
354 355 def checkout(self, node):
355 356 # checkout is really dumb at the moment
356 357 # it ought to basically merge
357 358 change = self.changelog.read(node)
358 359 mmap = self.manifest.read(change[0])
359 360
360 361 l = mmap.keys()
361 362 l.sort()
362 363 stats = []
363 364 for f in l:
364 365 self.ui.note(f + "\n")
365 366 r = self.file(f)
366 367 t = r.revision(mmap[f])
367 368 try:
368 369 file(f, "w").write(t)
369 370 except:
370 371 self.checkdir(f)
371 372 file(f, "w").write(t)
372 373
373 374 self.setcurrent(node)
374 375 self.dircache.clear()
375 376 self.dircache.update(l)
376 377
377 378 def diffdir(self, path, changeset):
378 379 changed = []
379 380 mf = {}
380 381 added = []
381 382
382 383 if changeset:
383 384 change = self.changelog.read(changeset)
384 385 mf = self.manifest.read(change[0])
385 386
386 387 if changeset == self.current:
387 388 dc = self.dircache.copy()
388 389 else:
389 390 dc = dict.fromkeys(mf)
390 391
391 392 def fcmp(fn):
392 393 t1 = file(os.path.join(self.root, fn)).read()
393 394 t2 = self.file(fn).revision(mf[fn])
394 395 return cmp(t1, t2)
395 396
396 397 for dir, subdirs, files in os.walk(self.root):
397 398 d = dir[len(self.root)+1:]
398 399 if ".hg" in subdirs: subdirs.remove(".hg")
399 400
400 401 for f in files:
401 402 fn = os.path.join(d, f)
402 403 try: s = os.stat(os.path.join(self.root, fn))
403 404 except: continue
404 405 if fn in dc:
405 406 c = dc[fn]
406 407 del dc[fn]
407 408 if not c:
408 409 if fcmp(fn):
409 410 changed.append(fn)
410 411 elif c[1] != s.st_size:
411 412 changed.append(fn)
412 413 elif c[0] != s.st_mode or c[2] != s.st_mtime:
413 414 if fcmp(fn):
414 415 changed.append(fn)
415 416 else:
416 417 if self.ignore(fn): continue
417 418 added.append(fn)
418 419
419 420 deleted = dc.keys()
420 421 deleted.sort()
421 422
422 423 return (changed, added, deleted)
423 424
424 425 def diffrevs(self, node1, node2):
425 426 changed, added = [], []
426 427
427 428 change = self.changelog.read(node1)
428 429 mf1 = self.manifest.read(change[0])
429 430 change = self.changelog.read(node2)
430 431 mf2 = self.manifest.read(change[0])
431 432
432 433 for fn in mf2:
433 434 if mf1.has_key(fn):
434 435 if mf1[fn] != mf2[fn]:
435 436 changed.append(fn)
436 437 del mf1[fn]
437 438 else:
438 439 added.append(fn)
439 440
440 441 deleted = mf1.keys()
441 442 deleted.sort()
442 443
443 444 return (changed, added, deleted)
444 445
445 446 def add(self, list):
446 447 self.dircache.taint(list)
447 448
448 449 def remove(self, list):
449 450 dl = self.opener("to-remove", "a")
450 451 for f in list:
451 452 dl.write(f + "\n")
452 453
453 454 def branches(self, nodes):
454 455 if not nodes: nodes = [self.changelog.tip()]
455 456 b = []
456 457 for n in nodes:
457 458 t = n
458 459 while n:
459 460 p = self.changelog.parents(n)
460 461 if p[1] != nullid or p[0] == nullid:
461 462 b.append((t, n, p[0], p[1]))
462 463 break
463 464 n = p[0]
464 465 return b
465 466
466 467 def between(self, pairs):
467 468 r = []
468 469
469 470 for top, bottom in pairs:
470 471 n, l, i = top, [], 0
471 472 f = 1
472 473
473 474 while n != bottom:
474 475 p = self.changelog.parents(n)[0]
475 476 if i == f:
476 477 l.append(n)
477 478 f = f * 2
478 479 n = p
479 480 i += 1
480 481
481 482 r.append(l)
482 483
483 484 return r
484 485
485 486 def newer(self, nodes):
486 487 m = {}
487 488 nl = []
488 489 pm = {}
489 490 cl = self.changelog
490 491 t = l = cl.count()
491 492
492 493 # find the lowest numbered node
493 494 for n in nodes:
494 495 l = min(l, cl.rev(n))
495 496 m[n] = 1
496 497
497 498 for i in xrange(l, t):
498 499 n = cl.node(i)
499 500 if n in m: # explicitly listed
500 501 pm[n] = 1
501 502 nl.append(n)
502 503 continue
503 504 for p in cl.parents(n):
504 505 if p in pm: # parent listed
505 506 pm[n] = 1
506 507 nl.append(n)
507 508 break
508 509
509 510 return nl
510 511
511 512 def getchangegroup(self, remote):
512 513 tip = remote.branches([])[0]
513 514 self.ui.debug("remote tip branch is %s:%s\n" %
514 515 (short(tip[0]), short(tip[1])))
515 516 m = self.changelog.nodemap
516 517 unknown = [tip]
517 518 search = []
518 519 fetch = []
519 520
520 521 if tip[0] in m:
521 522 self.ui.note("nothing to do!\n")
522 523 return None
523 524
524 525 while unknown:
525 526 n = unknown.pop(0)
526 527 if n == nullid: break
527 528 if n[1] and n[1] in m: # do we know the base?
528 529 self.ui.debug("found incomplete branch %s\n" % short(n[1]))
529 530 search.append(n) # schedule branch range for scanning
530 531 else:
531 532 if n[2] in m and n[3] in m:
532 533 if n[1] not in fetch:
533 534 self.ui.debug("found new changeset %s\n" %
534 535 short(n[1]))
535 536 fetch.append(n[1]) # earliest unknown
536 537 continue
537 538 for b in remote.branches([n[2], n[3]]):
538 539 if b[0] not in m:
539 540 unknown.append(b)
540 541
541 542 while search:
542 543 n = search.pop(0)
543 544 l = remote.between([(n[0], n[1])])[0]
544 545 p = n[0]
545 546 f = 1
546 547 for i in l + [n[1]]:
547 548 if i in m:
548 549 if f <= 2:
549 550 self.ui.debug("found new branch changeset %s\n" %
550 551 short(p))
551 552 fetch.append(p)
552 553 else:
553 554 self.ui.debug("narrowed branch search to %s:%s\n"
554 555 % (short(p), short(i)))
555 556 search.append((p, i))
556 557 break
557 558 p, f = i, f * 2
558 559
559 560 for f in fetch:
560 561 if f in m:
561 562 raise "already have", short(f[:4])
562 563
563 564 self.ui.note("adding new changesets starting at " +
564 565 " ".join([short(f) for f in fetch]) + "\n")
565 566
566 567 return remote.changegroup(fetch)
567 568
568 569 def changegroup(self, basenodes):
569 570 nodes = self.newer(basenodes)
570 571
571 572 # construct the link map
572 573 linkmap = {}
573 574 for n in nodes:
574 575 linkmap[self.changelog.rev(n)] = n
575 576
576 577 # construct a list of all changed files
577 578 changed = {}
578 579 for n in nodes:
579 580 c = self.changelog.read(n)
580 581 for f in c[3]:
581 582 changed[f] = 1
582 583 changed = changed.keys()
583 584 changed.sort()
584 585
585 586 # the changegroup is changesets + manifests + all file revs
586 587 revs = [ self.changelog.rev(n) for n in nodes ]
587 588
588 589 yield self.changelog.group(linkmap)
589 590 yield self.manifest.group(linkmap)
590 591
591 592 for f in changed:
592 593 g = self.file(f).group(linkmap)
593 594 if not g: raise "couldn't find change to %s" % f
594 595 l = struct.pack(">l", len(f))
595 596 yield "".join([l, f, g])
596 597
597 598 def addchangegroup(self, generator):
598 599 class genread:
599 600 def __init__(self, generator):
600 601 self.g = generator
601 602 self.buf = ""
602 603 def read(self, l):
603 604 while l > len(self.buf):
604 605 try:
605 606 self.buf += self.g.next()
606 607 except StopIteration:
607 608 break
608 609 d, self.buf = self.buf[:l], self.buf[l:]
609 610 return d
610 611
611 612 if not generator: return
612 613 source = genread(generator)
613 614
614 615 def getchunk(add = 0):
615 616 d = source.read(4)
616 617 if not d: return ""
617 618 l = struct.unpack(">l", d)[0]
618 619 return source.read(l - 4 + add)
619 620
620 621 tr = self.transaction()
621 622 simple = True
622 623
623 624 self.ui.status("adding changesets\n")
624 625 # pull off the changeset group
625 626 def report(x):
626 627 self.ui.debug("add changeset %s\n" % short(x))
627 628 return self.changelog.count()
628 629
629 630 csg = getchunk()
630 631 co = self.changelog.tip()
631 632 cn = self.changelog.addgroup(csg, report, tr)
632 633
633 634 self.ui.status("adding manifests\n")
634 635 # pull off the manifest group
635 636 mfg = getchunk()
636 637 mm = self.manifest.tip()
637 638 mo = self.manifest.addgroup(mfg, lambda x: self.changelog.rev(x), tr)
638 639
639 640 # do we need a resolve?
640 641 if self.changelog.ancestor(co, cn) != co:
641 642 simple = False
642 643 resolverev = self.changelog.count()
643 644
644 645 # process the files
645 646 self.ui.status("adding files\n")
646 647 new = {}
647 648 while 1:
648 649 f = getchunk(4)
649 650 if not f: break
650 651 fg = getchunk()
651 652 self.ui.debug("adding %s revisions\n" % f)
652 653 fl = self.file(f)
653 654 o = fl.tip()
654 655 n = fl.addgroup(fg, lambda x: self.changelog.rev(x), tr)
655 656 if not simple:
656 657 if o == n: continue
657 658 # this file has changed between branches, so it must be
658 659 # represented in the merge changeset
659 660 new[f] = self.merge3(fl, f, o, n, tr, resolverev)
660 661
661 662 # For simple merges, we don't need to resolve manifests or changesets
662 663 if simple:
663 664 self.ui.debug("simple merge, skipping resolve\n")
664 665 tr.close()
665 666 return
666 667
667 668 # resolve the manifest to point to all the merged files
668 669 self.ui.status("resolving manifests\n")
669 670 ma = self.manifest.ancestor(mm, mo)
670 671 omap = self.manifest.read(mo) # other
671 672 amap = self.manifest.read(ma) # ancestor
672 673 mmap = self.manifest.read(mm) # mine
673 674 self.ui.debug("ancestor %s local %s other %s\n" %
674 675 (short(ma), short(mm), short(mo)))
675 676 nmap = {}
676 677
677 678 for f, mid in mmap.iteritems():
678 679 if f in omap:
679 680 if mid != omap[f]:
680 681 self.ui.debug("%s versions differ\n" % f)
681 682 if f in new: self.ui.debug("%s updated in resolve\n" % f)
682 683 # use merged version or local version
683 684 nmap[f] = new.get(f, mid)
684 685 else:
685 686 nmap[f] = mid # keep ours
686 687 del omap[f]
687 688 elif f in amap:
688 689 if mid != amap[f]:
689 690 self.ui.debug("local changed %s which other deleted\n" % f)
690 691 pass # we should prompt here
691 692 else:
692 693 self.ui.debug("other deleted %s\n" % f)
693 694 pass # other deleted it
694 695 else:
695 696 self.ui.debug("local created %s\n" %f)
696 697 nmap[f] = mid # we created it
697 698
698 699 del mmap
699 700
700 701 for f, oid in omap.iteritems():
701 702 if f in amap:
702 703 if oid != amap[f]:
703 704 self.ui.debug("other changed %s which we deleted\n" % f)
704 705 pass # this is the nasty case, we should prompt
705 706 else:
706 707 pass # probably safe
707 708 else:
708 709 self.ui.debug("remote created %s\n" % f)
709 710 nmap[f] = new.get(f, oid) # remote created it
710 711
711 712 del omap
712 713 del amap
713 714
714 715 node = self.manifest.add(nmap, tr, resolverev, mm, mo)
715 716
716 717 # Now all files and manifests are merged, we add the changed files
717 718 # and manifest id to the changelog
718 719 self.ui.status("committing merge changeset\n")
719 720 new = new.keys()
720 721 new.sort()
721 722 if co == cn: cn = -1
722 723
723 724 edittext = "\nHG: merge resolve\n" + \
724 725 "".join(["HG: changed %s\n" % f for f in new])
725 726 edittext = self.ui.edit(edittext)
726 727 n = self.changelog.add(node, new, edittext, tr, co, cn)
727 728
728 729 tr.close()
729 730
730 731 class remoterepository:
731 732 def __init__(self, ui, path):
732 733 self.url = path.replace("hg://", "http://", 1)
733 734 self.ui = ui
734 735
735 736 def do_cmd(self, cmd, **args):
736 737 self.ui.debug("sending %s command\n" % cmd)
737 738 q = {"cmd": cmd}
738 739 q.update(args)
739 740 qs = urllib.urlencode(q)
740 741 cu = "%s?%s" % (self.url, qs)
741 742 return urllib.urlopen(cu)
742 743
743 744 def branches(self, nodes):
744 745 n = " ".join(map(hex, nodes))
745 746 d = self.do_cmd("branches", nodes=n).read()
746 747 br = [ map(bin, b.split(" ")) for b in d.splitlines() ]
747 748 return br
748 749
749 750 def between(self, pairs):
750 751 n = "\n".join(["-".join(map(hex, p)) for p in pairs])
751 752 d = self.do_cmd("between", pairs=n).read()
752 753 p = [ map(bin, l.split(" ")) for l in d.splitlines() ]
753 754 return p
754 755
755 756 def changegroup(self, nodes):
756 757 n = " ".join(map(hex, nodes))
757 758 zd = zlib.decompressobj()
758 759 f = self.do_cmd("changegroup", roots=n)
759 760 while 1:
760 761 d = f.read(4096)
761 762 if not d:
762 763 yield zd.flush()
763 764 break
764 765 yield zd.decompress(d)
765 766
766 767 def repository(ui, path=None, create=0):
767 768 if path and path[:5] == "hg://":
768 769 return remoterepository(ui, path)
769 770 else:
770 771 return localrepository(ui, path, create)
771 772
772 773 class ui:
773 774 def __init__(self, verbose=False, debug=False, quiet=False):
774 775 self.quiet = quiet and not verbose and not debug
775 776 self.verbose = verbose or debug
776 777 self.debugflag = debug
777 778 def write(self, *args):
778 779 for a in args:
779 780 sys.stdout.write(str(a))
780 781 def prompt(self, msg, pat):
781 782 while 1:
782 783 sys.stdout.write(msg)
783 784 r = sys.stdin.readline()[:-1]
784 785 if re.match(pat, r):
785 786 return r
786 787 def status(self, *msg):
787 788 if not self.quiet: self.write(*msg)
788 789 def warn(self, msg):
789 790 self.write(*msg)
790 791 def note(self, msg):
791 792 if self.verbose: self.write(*msg)
792 793 def debug(self, msg):
793 794 if self.debugflag: self.write(*msg)
794 795 def edit(self, text):
795 796 (fd, name) = tempfile.mkstemp("hg")
796 797 f = os.fdopen(fd, "w")
797 798 f.write(text)
798 799 f.close()
799 800
800 801 editor = os.environ.get("EDITOR", "vi")
801 802 r = os.system("%s %s" % (editor, name))
802 803 if r:
803 804 raise "Edit failed!"
804 805
805 806 t = open(name).read()
806 807 t = re.sub("(?m)^HG:.*\n", "", t)
807 808
808 809 return t
809 810
810 811
811 812 class httprangereader:
812 813 def __init__(self, url):
813 814 self.url = url
814 815 self.pos = 0
815 816 def seek(self, pos):
816 817 self.pos = pos
817 818 def read(self, bytes=None):
818 819 opener = urllib2.build_opener(byterange.HTTPRangeHandler())
819 820 urllib2.install_opener(opener)
820 821 req = urllib2.Request(self.url)
821 822 end = ''
822 823 if bytes: end = self.pos + bytes
823 824 req.add_header('Range', 'bytes=%d-%s' % (self.pos, end))
824 825 f = urllib2.urlopen(req)
825 826 return f.read()
@@ -1,65 +1,68 b''
1 1 # transaction.py - simple journalling scheme for mercurial
2 2 #
3 3 # This transaction scheme is intended to gracefully handle program
4 4 # errors and interruptions. More serious failures like system crashes
5 5 # can be recovered with an fsck-like tool. As the whole repository is
6 6 # effectively log-structured, this should amount to simply truncating
7 7 # anything that isn't referenced in the changelog.
8 8 #
9 9 # Copyright 2005 Matt Mackall <mpm@selenic.com>
10 10 #
11 11 # This software may be used and distributed according to the terms
12 12 # of the GNU General Public License, incorporated herein by reference.
13 13
14 14 import os
15 15
16 16 class transaction:
17 def __init__(self, opener, journal):
17 def __init__(self, opener, journal, after = None):
18 18 self.opener = opener
19 self.after = after
19 20 self.entries = []
20 21 self.map = {}
21 22 self.journal = journal
22 23
23 24 # abort here if the journal already exists
24 25 if os.path.exists(self.journal):
25 print "journal already exists, recovering"
26 self.recover()
26 raise "journal already exists!"
27 27
28 28 self.file = open(self.journal, "w")
29 29
30 30 def __del__(self):
31 31 if self.entries: self.abort()
32 32 try: os.unlink(self.journal)
33 33 except: pass
34 34
35 35 def add(self, file, offset):
36 36 if file in self.map: return
37 37 self.entries.append((file, offset))
38 38 self.map[file] = 1
39 39 # add enough data to the journal to do the truncate
40 40 self.file.write("%s\0%d\n" % (file, offset))
41 41 self.file.flush()
42 42
43 43 def close(self):
44 44 self.file.close()
45 45 self.entries = []
46 os.unlink(self.journal)
46 if self.after:
47 os.rename(self.journal, self.after)
48 else:
49 os.unlink(self.journal)
47 50
48 51 def abort(self):
49 52 if not self.entries: return
50 53
51 54 print "transaction abort!"
52 55
53 56 for f, o in self.entries:
54 57 self.opener(f, "a").truncate(o)
55 58
56 59 self.entries = []
57 60
58 61 print "rollback completed"
59 62
60 63 def recover(self):
61 64 for l in open(self.journal).readlines():
62 65 f, o = l.split('\0')
63 66 self.opener(f, "a").truncate(int(o))
64 67 os.unlink(self.journal)
65 68
General Comments 0
You need to be logged in to leave comments. Login now