##// END OF EJS Templates
manifest: specify checkambig=True to revlog.__init__, to avoid ambiguity...
FUJIWARA Katsunori -
r29998:14ad8e2a default
parent child Browse files
Show More
@@ -1,1298 +1,1299 b''
1 # manifest.py - manifest revision class for mercurial
1 # manifest.py - manifest revision class for mercurial
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 from __future__ import absolute_import
8 from __future__ import absolute_import
9
9
10 import array
10 import array
11 import heapq
11 import heapq
12 import os
12 import os
13 import struct
13 import struct
14
14
15 from .i18n import _
15 from .i18n import _
16 from . import (
16 from . import (
17 error,
17 error,
18 mdiff,
18 mdiff,
19 parsers,
19 parsers,
20 revlog,
20 revlog,
21 util,
21 util,
22 )
22 )
23
23
24 propertycache = util.propertycache
24 propertycache = util.propertycache
25
25
26 def _parsev1(data):
26 def _parsev1(data):
27 # This method does a little bit of excessive-looking
27 # This method does a little bit of excessive-looking
28 # precondition checking. This is so that the behavior of this
28 # precondition checking. This is so that the behavior of this
29 # class exactly matches its C counterpart to try and help
29 # class exactly matches its C counterpart to try and help
30 # prevent surprise breakage for anyone that develops against
30 # prevent surprise breakage for anyone that develops against
31 # the pure version.
31 # the pure version.
32 if data and data[-1] != '\n':
32 if data and data[-1] != '\n':
33 raise ValueError('Manifest did not end in a newline.')
33 raise ValueError('Manifest did not end in a newline.')
34 prev = None
34 prev = None
35 for l in data.splitlines():
35 for l in data.splitlines():
36 if prev is not None and prev > l:
36 if prev is not None and prev > l:
37 raise ValueError('Manifest lines not in sorted order.')
37 raise ValueError('Manifest lines not in sorted order.')
38 prev = l
38 prev = l
39 f, n = l.split('\0')
39 f, n = l.split('\0')
40 if len(n) > 40:
40 if len(n) > 40:
41 yield f, revlog.bin(n[:40]), n[40:]
41 yield f, revlog.bin(n[:40]), n[40:]
42 else:
42 else:
43 yield f, revlog.bin(n), ''
43 yield f, revlog.bin(n), ''
44
44
45 def _parsev2(data):
45 def _parsev2(data):
46 metadataend = data.find('\n')
46 metadataend = data.find('\n')
47 # Just ignore metadata for now
47 # Just ignore metadata for now
48 pos = metadataend + 1
48 pos = metadataend + 1
49 prevf = ''
49 prevf = ''
50 while pos < len(data):
50 while pos < len(data):
51 end = data.find('\n', pos + 1) # +1 to skip stem length byte
51 end = data.find('\n', pos + 1) # +1 to skip stem length byte
52 if end == -1:
52 if end == -1:
53 raise ValueError('Manifest ended with incomplete file entry.')
53 raise ValueError('Manifest ended with incomplete file entry.')
54 stemlen = ord(data[pos])
54 stemlen = ord(data[pos])
55 items = data[pos + 1:end].split('\0')
55 items = data[pos + 1:end].split('\0')
56 f = prevf[:stemlen] + items[0]
56 f = prevf[:stemlen] + items[0]
57 if prevf > f:
57 if prevf > f:
58 raise ValueError('Manifest entries not in sorted order.')
58 raise ValueError('Manifest entries not in sorted order.')
59 fl = items[1]
59 fl = items[1]
60 # Just ignore metadata (items[2:] for now)
60 # Just ignore metadata (items[2:] for now)
61 n = data[end + 1:end + 21]
61 n = data[end + 1:end + 21]
62 yield f, n, fl
62 yield f, n, fl
63 pos = end + 22
63 pos = end + 22
64 prevf = f
64 prevf = f
65
65
66 def _parse(data):
66 def _parse(data):
67 """Generates (path, node, flags) tuples from a manifest text"""
67 """Generates (path, node, flags) tuples from a manifest text"""
68 if data.startswith('\0'):
68 if data.startswith('\0'):
69 return iter(_parsev2(data))
69 return iter(_parsev2(data))
70 else:
70 else:
71 return iter(_parsev1(data))
71 return iter(_parsev1(data))
72
72
73 def _text(it, usemanifestv2):
73 def _text(it, usemanifestv2):
74 """Given an iterator over (path, node, flags) tuples, returns a manifest
74 """Given an iterator over (path, node, flags) tuples, returns a manifest
75 text"""
75 text"""
76 if usemanifestv2:
76 if usemanifestv2:
77 return _textv2(it)
77 return _textv2(it)
78 else:
78 else:
79 return _textv1(it)
79 return _textv1(it)
80
80
81 def _textv1(it):
81 def _textv1(it):
82 files = []
82 files = []
83 lines = []
83 lines = []
84 _hex = revlog.hex
84 _hex = revlog.hex
85 for f, n, fl in it:
85 for f, n, fl in it:
86 files.append(f)
86 files.append(f)
87 # if this is changed to support newlines in filenames,
87 # if this is changed to support newlines in filenames,
88 # be sure to check the templates/ dir again (especially *-raw.tmpl)
88 # be sure to check the templates/ dir again (especially *-raw.tmpl)
89 lines.append("%s\0%s%s\n" % (f, _hex(n), fl))
89 lines.append("%s\0%s%s\n" % (f, _hex(n), fl))
90
90
91 _checkforbidden(files)
91 _checkforbidden(files)
92 return ''.join(lines)
92 return ''.join(lines)
93
93
94 def _textv2(it):
94 def _textv2(it):
95 files = []
95 files = []
96 lines = ['\0\n']
96 lines = ['\0\n']
97 prevf = ''
97 prevf = ''
98 for f, n, fl in it:
98 for f, n, fl in it:
99 files.append(f)
99 files.append(f)
100 stem = os.path.commonprefix([prevf, f])
100 stem = os.path.commonprefix([prevf, f])
101 stemlen = min(len(stem), 255)
101 stemlen = min(len(stem), 255)
102 lines.append("%c%s\0%s\n%s\n" % (stemlen, f[stemlen:], fl, n))
102 lines.append("%c%s\0%s\n%s\n" % (stemlen, f[stemlen:], fl, n))
103 prevf = f
103 prevf = f
104 _checkforbidden(files)
104 _checkforbidden(files)
105 return ''.join(lines)
105 return ''.join(lines)
106
106
107 class _lazymanifest(dict):
107 class _lazymanifest(dict):
108 """This is the pure implementation of lazymanifest.
108 """This is the pure implementation of lazymanifest.
109
109
110 It has not been optimized *at all* and is not lazy.
110 It has not been optimized *at all* and is not lazy.
111 """
111 """
112
112
113 def __init__(self, data):
113 def __init__(self, data):
114 dict.__init__(self)
114 dict.__init__(self)
115 for f, n, fl in _parse(data):
115 for f, n, fl in _parse(data):
116 self[f] = n, fl
116 self[f] = n, fl
117
117
118 def __setitem__(self, k, v):
118 def __setitem__(self, k, v):
119 node, flag = v
119 node, flag = v
120 assert node is not None
120 assert node is not None
121 if len(node) > 21:
121 if len(node) > 21:
122 node = node[:21] # match c implementation behavior
122 node = node[:21] # match c implementation behavior
123 dict.__setitem__(self, k, (node, flag))
123 dict.__setitem__(self, k, (node, flag))
124
124
125 def __iter__(self):
125 def __iter__(self):
126 return iter(sorted(dict.keys(self)))
126 return iter(sorted(dict.keys(self)))
127
127
128 def iterkeys(self):
128 def iterkeys(self):
129 return iter(sorted(dict.keys(self)))
129 return iter(sorted(dict.keys(self)))
130
130
131 def iterentries(self):
131 def iterentries(self):
132 return ((f, e[0], e[1]) for f, e in sorted(self.iteritems()))
132 return ((f, e[0], e[1]) for f, e in sorted(self.iteritems()))
133
133
134 def copy(self):
134 def copy(self):
135 c = _lazymanifest('')
135 c = _lazymanifest('')
136 c.update(self)
136 c.update(self)
137 return c
137 return c
138
138
139 def diff(self, m2, clean=False):
139 def diff(self, m2, clean=False):
140 '''Finds changes between the current manifest and m2.'''
140 '''Finds changes between the current manifest and m2.'''
141 diff = {}
141 diff = {}
142
142
143 for fn, e1 in self.iteritems():
143 for fn, e1 in self.iteritems():
144 if fn not in m2:
144 if fn not in m2:
145 diff[fn] = e1, (None, '')
145 diff[fn] = e1, (None, '')
146 else:
146 else:
147 e2 = m2[fn]
147 e2 = m2[fn]
148 if e1 != e2:
148 if e1 != e2:
149 diff[fn] = e1, e2
149 diff[fn] = e1, e2
150 elif clean:
150 elif clean:
151 diff[fn] = None
151 diff[fn] = None
152
152
153 for fn, e2 in m2.iteritems():
153 for fn, e2 in m2.iteritems():
154 if fn not in self:
154 if fn not in self:
155 diff[fn] = (None, ''), e2
155 diff[fn] = (None, ''), e2
156
156
157 return diff
157 return diff
158
158
159 def filtercopy(self, filterfn):
159 def filtercopy(self, filterfn):
160 c = _lazymanifest('')
160 c = _lazymanifest('')
161 for f, n, fl in self.iterentries():
161 for f, n, fl in self.iterentries():
162 if filterfn(f):
162 if filterfn(f):
163 c[f] = n, fl
163 c[f] = n, fl
164 return c
164 return c
165
165
166 def text(self):
166 def text(self):
167 """Get the full data of this manifest as a bytestring."""
167 """Get the full data of this manifest as a bytestring."""
168 return _textv1(self.iterentries())
168 return _textv1(self.iterentries())
169
169
170 try:
170 try:
171 _lazymanifest = parsers.lazymanifest
171 _lazymanifest = parsers.lazymanifest
172 except AttributeError:
172 except AttributeError:
173 pass
173 pass
174
174
175 class manifestdict(object):
175 class manifestdict(object):
176 def __init__(self, data=''):
176 def __init__(self, data=''):
177 if data.startswith('\0'):
177 if data.startswith('\0'):
178 #_lazymanifest can not parse v2
178 #_lazymanifest can not parse v2
179 self._lm = _lazymanifest('')
179 self._lm = _lazymanifest('')
180 for f, n, fl in _parsev2(data):
180 for f, n, fl in _parsev2(data):
181 self._lm[f] = n, fl
181 self._lm[f] = n, fl
182 else:
182 else:
183 self._lm = _lazymanifest(data)
183 self._lm = _lazymanifest(data)
184
184
185 def __getitem__(self, key):
185 def __getitem__(self, key):
186 return self._lm[key][0]
186 return self._lm[key][0]
187
187
188 def find(self, key):
188 def find(self, key):
189 return self._lm[key]
189 return self._lm[key]
190
190
191 def __len__(self):
191 def __len__(self):
192 return len(self._lm)
192 return len(self._lm)
193
193
194 def __setitem__(self, key, node):
194 def __setitem__(self, key, node):
195 self._lm[key] = node, self.flags(key, '')
195 self._lm[key] = node, self.flags(key, '')
196
196
197 def __contains__(self, key):
197 def __contains__(self, key):
198 return key in self._lm
198 return key in self._lm
199
199
200 def __delitem__(self, key):
200 def __delitem__(self, key):
201 del self._lm[key]
201 del self._lm[key]
202
202
203 def __iter__(self):
203 def __iter__(self):
204 return self._lm.__iter__()
204 return self._lm.__iter__()
205
205
206 def iterkeys(self):
206 def iterkeys(self):
207 return self._lm.iterkeys()
207 return self._lm.iterkeys()
208
208
209 def keys(self):
209 def keys(self):
210 return list(self.iterkeys())
210 return list(self.iterkeys())
211
211
212 def filesnotin(self, m2):
212 def filesnotin(self, m2):
213 '''Set of files in this manifest that are not in the other'''
213 '''Set of files in this manifest that are not in the other'''
214 diff = self.diff(m2)
214 diff = self.diff(m2)
215 files = set(filepath
215 files = set(filepath
216 for filepath, hashflags in diff.iteritems()
216 for filepath, hashflags in diff.iteritems()
217 if hashflags[1][0] is None)
217 if hashflags[1][0] is None)
218 return files
218 return files
219
219
220 @propertycache
220 @propertycache
221 def _dirs(self):
221 def _dirs(self):
222 return util.dirs(self)
222 return util.dirs(self)
223
223
224 def dirs(self):
224 def dirs(self):
225 return self._dirs
225 return self._dirs
226
226
227 def hasdir(self, dir):
227 def hasdir(self, dir):
228 return dir in self._dirs
228 return dir in self._dirs
229
229
230 def _filesfastpath(self, match):
230 def _filesfastpath(self, match):
231 '''Checks whether we can correctly and quickly iterate over matcher
231 '''Checks whether we can correctly and quickly iterate over matcher
232 files instead of over manifest files.'''
232 files instead of over manifest files.'''
233 files = match.files()
233 files = match.files()
234 return (len(files) < 100 and (match.isexact() or
234 return (len(files) < 100 and (match.isexact() or
235 (match.prefix() and all(fn in self for fn in files))))
235 (match.prefix() and all(fn in self for fn in files))))
236
236
237 def walk(self, match):
237 def walk(self, match):
238 '''Generates matching file names.
238 '''Generates matching file names.
239
239
240 Equivalent to manifest.matches(match).iterkeys(), but without creating
240 Equivalent to manifest.matches(match).iterkeys(), but without creating
241 an entirely new manifest.
241 an entirely new manifest.
242
242
243 It also reports nonexistent files by marking them bad with match.bad().
243 It also reports nonexistent files by marking them bad with match.bad().
244 '''
244 '''
245 if match.always():
245 if match.always():
246 for f in iter(self):
246 for f in iter(self):
247 yield f
247 yield f
248 return
248 return
249
249
250 fset = set(match.files())
250 fset = set(match.files())
251
251
252 # avoid the entire walk if we're only looking for specific files
252 # avoid the entire walk if we're only looking for specific files
253 if self._filesfastpath(match):
253 if self._filesfastpath(match):
254 for fn in sorted(fset):
254 for fn in sorted(fset):
255 yield fn
255 yield fn
256 return
256 return
257
257
258 for fn in self:
258 for fn in self:
259 if fn in fset:
259 if fn in fset:
260 # specified pattern is the exact name
260 # specified pattern is the exact name
261 fset.remove(fn)
261 fset.remove(fn)
262 if match(fn):
262 if match(fn):
263 yield fn
263 yield fn
264
264
265 # for dirstate.walk, files=['.'] means "walk the whole tree".
265 # for dirstate.walk, files=['.'] means "walk the whole tree".
266 # follow that here, too
266 # follow that here, too
267 fset.discard('.')
267 fset.discard('.')
268
268
269 for fn in sorted(fset):
269 for fn in sorted(fset):
270 if not self.hasdir(fn):
270 if not self.hasdir(fn):
271 match.bad(fn, None)
271 match.bad(fn, None)
272
272
273 def matches(self, match):
273 def matches(self, match):
274 '''generate a new manifest filtered by the match argument'''
274 '''generate a new manifest filtered by the match argument'''
275 if match.always():
275 if match.always():
276 return self.copy()
276 return self.copy()
277
277
278 if self._filesfastpath(match):
278 if self._filesfastpath(match):
279 m = manifestdict()
279 m = manifestdict()
280 lm = self._lm
280 lm = self._lm
281 for fn in match.files():
281 for fn in match.files():
282 if fn in lm:
282 if fn in lm:
283 m._lm[fn] = lm[fn]
283 m._lm[fn] = lm[fn]
284 return m
284 return m
285
285
286 m = manifestdict()
286 m = manifestdict()
287 m._lm = self._lm.filtercopy(match)
287 m._lm = self._lm.filtercopy(match)
288 return m
288 return m
289
289
290 def diff(self, m2, clean=False):
290 def diff(self, m2, clean=False):
291 '''Finds changes between the current manifest and m2.
291 '''Finds changes between the current manifest and m2.
292
292
293 Args:
293 Args:
294 m2: the manifest to which this manifest should be compared.
294 m2: the manifest to which this manifest should be compared.
295 clean: if true, include files unchanged between these manifests
295 clean: if true, include files unchanged between these manifests
296 with a None value in the returned dictionary.
296 with a None value in the returned dictionary.
297
297
298 The result is returned as a dict with filename as key and
298 The result is returned as a dict with filename as key and
299 values of the form ((n1,fl1),(n2,fl2)), where n1/n2 is the
299 values of the form ((n1,fl1),(n2,fl2)), where n1/n2 is the
300 nodeid in the current/other manifest and fl1/fl2 is the flag
300 nodeid in the current/other manifest and fl1/fl2 is the flag
301 in the current/other manifest. Where the file does not exist,
301 in the current/other manifest. Where the file does not exist,
302 the nodeid will be None and the flags will be the empty
302 the nodeid will be None and the flags will be the empty
303 string.
303 string.
304 '''
304 '''
305 return self._lm.diff(m2._lm, clean)
305 return self._lm.diff(m2._lm, clean)
306
306
307 def setflag(self, key, flag):
307 def setflag(self, key, flag):
308 self._lm[key] = self[key], flag
308 self._lm[key] = self[key], flag
309
309
310 def get(self, key, default=None):
310 def get(self, key, default=None):
311 try:
311 try:
312 return self._lm[key][0]
312 return self._lm[key][0]
313 except KeyError:
313 except KeyError:
314 return default
314 return default
315
315
316 def flags(self, key, default=''):
316 def flags(self, key, default=''):
317 try:
317 try:
318 return self._lm[key][1]
318 return self._lm[key][1]
319 except KeyError:
319 except KeyError:
320 return default
320 return default
321
321
322 def copy(self):
322 def copy(self):
323 c = manifestdict()
323 c = manifestdict()
324 c._lm = self._lm.copy()
324 c._lm = self._lm.copy()
325 return c
325 return c
326
326
327 def iteritems(self):
327 def iteritems(self):
328 return (x[:2] for x in self._lm.iterentries())
328 return (x[:2] for x in self._lm.iterentries())
329
329
330 def iterentries(self):
330 def iterentries(self):
331 return self._lm.iterentries()
331 return self._lm.iterentries()
332
332
333 def text(self, usemanifestv2=False):
333 def text(self, usemanifestv2=False):
334 if usemanifestv2:
334 if usemanifestv2:
335 return _textv2(self._lm.iterentries())
335 return _textv2(self._lm.iterentries())
336 else:
336 else:
337 # use (probably) native version for v1
337 # use (probably) native version for v1
338 return self._lm.text()
338 return self._lm.text()
339
339
340 def fastdelta(self, base, changes):
340 def fastdelta(self, base, changes):
341 """Given a base manifest text as an array.array and a list of changes
341 """Given a base manifest text as an array.array and a list of changes
342 relative to that text, compute a delta that can be used by revlog.
342 relative to that text, compute a delta that can be used by revlog.
343 """
343 """
344 delta = []
344 delta = []
345 dstart = None
345 dstart = None
346 dend = None
346 dend = None
347 dline = [""]
347 dline = [""]
348 start = 0
348 start = 0
349 # zero copy representation of base as a buffer
349 # zero copy representation of base as a buffer
350 addbuf = util.buffer(base)
350 addbuf = util.buffer(base)
351
351
352 changes = list(changes)
352 changes = list(changes)
353 if len(changes) < 1000:
353 if len(changes) < 1000:
354 # start with a readonly loop that finds the offset of
354 # start with a readonly loop that finds the offset of
355 # each line and creates the deltas
355 # each line and creates the deltas
356 for f, todelete in changes:
356 for f, todelete in changes:
357 # bs will either be the index of the item or the insert point
357 # bs will either be the index of the item or the insert point
358 start, end = _msearch(addbuf, f, start)
358 start, end = _msearch(addbuf, f, start)
359 if not todelete:
359 if not todelete:
360 h, fl = self._lm[f]
360 h, fl = self._lm[f]
361 l = "%s\0%s%s\n" % (f, revlog.hex(h), fl)
361 l = "%s\0%s%s\n" % (f, revlog.hex(h), fl)
362 else:
362 else:
363 if start == end:
363 if start == end:
364 # item we want to delete was not found, error out
364 # item we want to delete was not found, error out
365 raise AssertionError(
365 raise AssertionError(
366 _("failed to remove %s from manifest") % f)
366 _("failed to remove %s from manifest") % f)
367 l = ""
367 l = ""
368 if dstart is not None and dstart <= start and dend >= start:
368 if dstart is not None and dstart <= start and dend >= start:
369 if dend < end:
369 if dend < end:
370 dend = end
370 dend = end
371 if l:
371 if l:
372 dline.append(l)
372 dline.append(l)
373 else:
373 else:
374 if dstart is not None:
374 if dstart is not None:
375 delta.append([dstart, dend, "".join(dline)])
375 delta.append([dstart, dend, "".join(dline)])
376 dstart = start
376 dstart = start
377 dend = end
377 dend = end
378 dline = [l]
378 dline = [l]
379
379
380 if dstart is not None:
380 if dstart is not None:
381 delta.append([dstart, dend, "".join(dline)])
381 delta.append([dstart, dend, "".join(dline)])
382 # apply the delta to the base, and get a delta for addrevision
382 # apply the delta to the base, and get a delta for addrevision
383 deltatext, arraytext = _addlistdelta(base, delta)
383 deltatext, arraytext = _addlistdelta(base, delta)
384 else:
384 else:
385 # For large changes, it's much cheaper to just build the text and
385 # For large changes, it's much cheaper to just build the text and
386 # diff it.
386 # diff it.
387 arraytext = array.array('c', self.text())
387 arraytext = array.array('c', self.text())
388 deltatext = mdiff.textdiff(base, arraytext)
388 deltatext = mdiff.textdiff(base, arraytext)
389
389
390 return arraytext, deltatext
390 return arraytext, deltatext
391
391
392 def _msearch(m, s, lo=0, hi=None):
392 def _msearch(m, s, lo=0, hi=None):
393 '''return a tuple (start, end) that says where to find s within m.
393 '''return a tuple (start, end) that says where to find s within m.
394
394
395 If the string is found m[start:end] are the line containing
395 If the string is found m[start:end] are the line containing
396 that string. If start == end the string was not found and
396 that string. If start == end the string was not found and
397 they indicate the proper sorted insertion point.
397 they indicate the proper sorted insertion point.
398
398
399 m should be a buffer or a string
399 m should be a buffer or a string
400 s is a string'''
400 s is a string'''
401 def advance(i, c):
401 def advance(i, c):
402 while i < lenm and m[i] != c:
402 while i < lenm and m[i] != c:
403 i += 1
403 i += 1
404 return i
404 return i
405 if not s:
405 if not s:
406 return (lo, lo)
406 return (lo, lo)
407 lenm = len(m)
407 lenm = len(m)
408 if not hi:
408 if not hi:
409 hi = lenm
409 hi = lenm
410 while lo < hi:
410 while lo < hi:
411 mid = (lo + hi) // 2
411 mid = (lo + hi) // 2
412 start = mid
412 start = mid
413 while start > 0 and m[start - 1] != '\n':
413 while start > 0 and m[start - 1] != '\n':
414 start -= 1
414 start -= 1
415 end = advance(start, '\0')
415 end = advance(start, '\0')
416 if m[start:end] < s:
416 if m[start:end] < s:
417 # we know that after the null there are 40 bytes of sha1
417 # we know that after the null there are 40 bytes of sha1
418 # this translates to the bisect lo = mid + 1
418 # this translates to the bisect lo = mid + 1
419 lo = advance(end + 40, '\n') + 1
419 lo = advance(end + 40, '\n') + 1
420 else:
420 else:
421 # this translates to the bisect hi = mid
421 # this translates to the bisect hi = mid
422 hi = start
422 hi = start
423 end = advance(lo, '\0')
423 end = advance(lo, '\0')
424 found = m[lo:end]
424 found = m[lo:end]
425 if s == found:
425 if s == found:
426 # we know that after the null there are 40 bytes of sha1
426 # we know that after the null there are 40 bytes of sha1
427 end = advance(end + 40, '\n')
427 end = advance(end + 40, '\n')
428 return (lo, end + 1)
428 return (lo, end + 1)
429 else:
429 else:
430 return (lo, lo)
430 return (lo, lo)
431
431
432 def _checkforbidden(l):
432 def _checkforbidden(l):
433 """Check filenames for illegal characters."""
433 """Check filenames for illegal characters."""
434 for f in l:
434 for f in l:
435 if '\n' in f or '\r' in f:
435 if '\n' in f or '\r' in f:
436 raise error.RevlogError(
436 raise error.RevlogError(
437 _("'\\n' and '\\r' disallowed in filenames: %r") % f)
437 _("'\\n' and '\\r' disallowed in filenames: %r") % f)
438
438
439
439
440 # apply the changes collected during the bisect loop to our addlist
440 # apply the changes collected during the bisect loop to our addlist
441 # return a delta suitable for addrevision
441 # return a delta suitable for addrevision
442 def _addlistdelta(addlist, x):
442 def _addlistdelta(addlist, x):
443 # for large addlist arrays, building a new array is cheaper
443 # for large addlist arrays, building a new array is cheaper
444 # than repeatedly modifying the existing one
444 # than repeatedly modifying the existing one
445 currentposition = 0
445 currentposition = 0
446 newaddlist = array.array('c')
446 newaddlist = array.array('c')
447
447
448 for start, end, content in x:
448 for start, end, content in x:
449 newaddlist += addlist[currentposition:start]
449 newaddlist += addlist[currentposition:start]
450 if content:
450 if content:
451 newaddlist += array.array('c', content)
451 newaddlist += array.array('c', content)
452
452
453 currentposition = end
453 currentposition = end
454
454
455 newaddlist += addlist[currentposition:]
455 newaddlist += addlist[currentposition:]
456
456
457 deltatext = "".join(struct.pack(">lll", start, end, len(content))
457 deltatext = "".join(struct.pack(">lll", start, end, len(content))
458 + content for start, end, content in x)
458 + content for start, end, content in x)
459 return deltatext, newaddlist
459 return deltatext, newaddlist
460
460
461 def _splittopdir(f):
461 def _splittopdir(f):
462 if '/' in f:
462 if '/' in f:
463 dir, subpath = f.split('/', 1)
463 dir, subpath = f.split('/', 1)
464 return dir + '/', subpath
464 return dir + '/', subpath
465 else:
465 else:
466 return '', f
466 return '', f
467
467
468 _noop = lambda s: None
468 _noop = lambda s: None
469
469
470 class treemanifest(object):
470 class treemanifest(object):
471 def __init__(self, dir='', text=''):
471 def __init__(self, dir='', text=''):
472 self._dir = dir
472 self._dir = dir
473 self._node = revlog.nullid
473 self._node = revlog.nullid
474 self._loadfunc = _noop
474 self._loadfunc = _noop
475 self._copyfunc = _noop
475 self._copyfunc = _noop
476 self._dirty = False
476 self._dirty = False
477 self._dirs = {}
477 self._dirs = {}
478 # Using _lazymanifest here is a little slower than plain old dicts
478 # Using _lazymanifest here is a little slower than plain old dicts
479 self._files = {}
479 self._files = {}
480 self._flags = {}
480 self._flags = {}
481 if text:
481 if text:
482 def readsubtree(subdir, subm):
482 def readsubtree(subdir, subm):
483 raise AssertionError('treemanifest constructor only accepts '
483 raise AssertionError('treemanifest constructor only accepts '
484 'flat manifests')
484 'flat manifests')
485 self.parse(text, readsubtree)
485 self.parse(text, readsubtree)
486 self._dirty = True # Mark flat manifest dirty after parsing
486 self._dirty = True # Mark flat manifest dirty after parsing
487
487
488 def _subpath(self, path):
488 def _subpath(self, path):
489 return self._dir + path
489 return self._dir + path
490
490
491 def __len__(self):
491 def __len__(self):
492 self._load()
492 self._load()
493 size = len(self._files)
493 size = len(self._files)
494 for m in self._dirs.values():
494 for m in self._dirs.values():
495 size += m.__len__()
495 size += m.__len__()
496 return size
496 return size
497
497
498 def _isempty(self):
498 def _isempty(self):
499 self._load() # for consistency; already loaded by all callers
499 self._load() # for consistency; already loaded by all callers
500 return (not self._files and (not self._dirs or
500 return (not self._files and (not self._dirs or
501 all(m._isempty() for m in self._dirs.values())))
501 all(m._isempty() for m in self._dirs.values())))
502
502
503 def __repr__(self):
503 def __repr__(self):
504 return ('<treemanifest dir=%s, node=%s, loaded=%s, dirty=%s at 0x%x>' %
504 return ('<treemanifest dir=%s, node=%s, loaded=%s, dirty=%s at 0x%x>' %
505 (self._dir, revlog.hex(self._node),
505 (self._dir, revlog.hex(self._node),
506 bool(self._loadfunc is _noop),
506 bool(self._loadfunc is _noop),
507 self._dirty, id(self)))
507 self._dirty, id(self)))
508
508
509 def dir(self):
509 def dir(self):
510 '''The directory that this tree manifest represents, including a
510 '''The directory that this tree manifest represents, including a
511 trailing '/'. Empty string for the repo root directory.'''
511 trailing '/'. Empty string for the repo root directory.'''
512 return self._dir
512 return self._dir
513
513
514 def node(self):
514 def node(self):
515 '''This node of this instance. nullid for unsaved instances. Should
515 '''This node of this instance. nullid for unsaved instances. Should
516 be updated when the instance is read or written from a revlog.
516 be updated when the instance is read or written from a revlog.
517 '''
517 '''
518 assert not self._dirty
518 assert not self._dirty
519 return self._node
519 return self._node
520
520
521 def setnode(self, node):
521 def setnode(self, node):
522 self._node = node
522 self._node = node
523 self._dirty = False
523 self._dirty = False
524
524
525 def iterentries(self):
525 def iterentries(self):
526 self._load()
526 self._load()
527 for p, n in sorted(self._dirs.items() + self._files.items()):
527 for p, n in sorted(self._dirs.items() + self._files.items()):
528 if p in self._files:
528 if p in self._files:
529 yield self._subpath(p), n, self._flags.get(p, '')
529 yield self._subpath(p), n, self._flags.get(p, '')
530 else:
530 else:
531 for x in n.iterentries():
531 for x in n.iterentries():
532 yield x
532 yield x
533
533
534 def iteritems(self):
534 def iteritems(self):
535 self._load()
535 self._load()
536 for p, n in sorted(self._dirs.items() + self._files.items()):
536 for p, n in sorted(self._dirs.items() + self._files.items()):
537 if p in self._files:
537 if p in self._files:
538 yield self._subpath(p), n
538 yield self._subpath(p), n
539 else:
539 else:
540 for f, sn in n.iteritems():
540 for f, sn in n.iteritems():
541 yield f, sn
541 yield f, sn
542
542
543 def iterkeys(self):
543 def iterkeys(self):
544 self._load()
544 self._load()
545 for p in sorted(self._dirs.keys() + self._files.keys()):
545 for p in sorted(self._dirs.keys() + self._files.keys()):
546 if p in self._files:
546 if p in self._files:
547 yield self._subpath(p)
547 yield self._subpath(p)
548 else:
548 else:
549 for f in self._dirs[p].iterkeys():
549 for f in self._dirs[p].iterkeys():
550 yield f
550 yield f
551
551
552 def keys(self):
552 def keys(self):
553 return list(self.iterkeys())
553 return list(self.iterkeys())
554
554
555 def __iter__(self):
555 def __iter__(self):
556 return self.iterkeys()
556 return self.iterkeys()
557
557
558 def __contains__(self, f):
558 def __contains__(self, f):
559 if f is None:
559 if f is None:
560 return False
560 return False
561 self._load()
561 self._load()
562 dir, subpath = _splittopdir(f)
562 dir, subpath = _splittopdir(f)
563 if dir:
563 if dir:
564 if dir not in self._dirs:
564 if dir not in self._dirs:
565 return False
565 return False
566 return self._dirs[dir].__contains__(subpath)
566 return self._dirs[dir].__contains__(subpath)
567 else:
567 else:
568 return f in self._files
568 return f in self._files
569
569
570 def get(self, f, default=None):
570 def get(self, f, default=None):
571 self._load()
571 self._load()
572 dir, subpath = _splittopdir(f)
572 dir, subpath = _splittopdir(f)
573 if dir:
573 if dir:
574 if dir not in self._dirs:
574 if dir not in self._dirs:
575 return default
575 return default
576 return self._dirs[dir].get(subpath, default)
576 return self._dirs[dir].get(subpath, default)
577 else:
577 else:
578 return self._files.get(f, default)
578 return self._files.get(f, default)
579
579
580 def __getitem__(self, f):
580 def __getitem__(self, f):
581 self._load()
581 self._load()
582 dir, subpath = _splittopdir(f)
582 dir, subpath = _splittopdir(f)
583 if dir:
583 if dir:
584 return self._dirs[dir].__getitem__(subpath)
584 return self._dirs[dir].__getitem__(subpath)
585 else:
585 else:
586 return self._files[f]
586 return self._files[f]
587
587
588 def flags(self, f):
588 def flags(self, f):
589 self._load()
589 self._load()
590 dir, subpath = _splittopdir(f)
590 dir, subpath = _splittopdir(f)
591 if dir:
591 if dir:
592 if dir not in self._dirs:
592 if dir not in self._dirs:
593 return ''
593 return ''
594 return self._dirs[dir].flags(subpath)
594 return self._dirs[dir].flags(subpath)
595 else:
595 else:
596 if f in self._dirs:
596 if f in self._dirs:
597 return ''
597 return ''
598 return self._flags.get(f, '')
598 return self._flags.get(f, '')
599
599
600 def find(self, f):
600 def find(self, f):
601 self._load()
601 self._load()
602 dir, subpath = _splittopdir(f)
602 dir, subpath = _splittopdir(f)
603 if dir:
603 if dir:
604 return self._dirs[dir].find(subpath)
604 return self._dirs[dir].find(subpath)
605 else:
605 else:
606 return self._files[f], self._flags.get(f, '')
606 return self._files[f], self._flags.get(f, '')
607
607
608 def __delitem__(self, f):
608 def __delitem__(self, f):
609 self._load()
609 self._load()
610 dir, subpath = _splittopdir(f)
610 dir, subpath = _splittopdir(f)
611 if dir:
611 if dir:
612 self._dirs[dir].__delitem__(subpath)
612 self._dirs[dir].__delitem__(subpath)
613 # If the directory is now empty, remove it
613 # If the directory is now empty, remove it
614 if self._dirs[dir]._isempty():
614 if self._dirs[dir]._isempty():
615 del self._dirs[dir]
615 del self._dirs[dir]
616 else:
616 else:
617 del self._files[f]
617 del self._files[f]
618 if f in self._flags:
618 if f in self._flags:
619 del self._flags[f]
619 del self._flags[f]
620 self._dirty = True
620 self._dirty = True
621
621
622 def __setitem__(self, f, n):
622 def __setitem__(self, f, n):
623 assert n is not None
623 assert n is not None
624 self._load()
624 self._load()
625 dir, subpath = _splittopdir(f)
625 dir, subpath = _splittopdir(f)
626 if dir:
626 if dir:
627 if dir not in self._dirs:
627 if dir not in self._dirs:
628 self._dirs[dir] = treemanifest(self._subpath(dir))
628 self._dirs[dir] = treemanifest(self._subpath(dir))
629 self._dirs[dir].__setitem__(subpath, n)
629 self._dirs[dir].__setitem__(subpath, n)
630 else:
630 else:
631 self._files[f] = n[:21] # to match manifestdict's behavior
631 self._files[f] = n[:21] # to match manifestdict's behavior
632 self._dirty = True
632 self._dirty = True
633
633
634 def _load(self):
634 def _load(self):
635 if self._loadfunc is not _noop:
635 if self._loadfunc is not _noop:
636 lf, self._loadfunc = self._loadfunc, _noop
636 lf, self._loadfunc = self._loadfunc, _noop
637 lf(self)
637 lf(self)
638 elif self._copyfunc is not _noop:
638 elif self._copyfunc is not _noop:
639 cf, self._copyfunc = self._copyfunc, _noop
639 cf, self._copyfunc = self._copyfunc, _noop
640 cf(self)
640 cf(self)
641
641
642 def setflag(self, f, flags):
642 def setflag(self, f, flags):
643 """Set the flags (symlink, executable) for path f."""
643 """Set the flags (symlink, executable) for path f."""
644 self._load()
644 self._load()
645 dir, subpath = _splittopdir(f)
645 dir, subpath = _splittopdir(f)
646 if dir:
646 if dir:
647 if dir not in self._dirs:
647 if dir not in self._dirs:
648 self._dirs[dir] = treemanifest(self._subpath(dir))
648 self._dirs[dir] = treemanifest(self._subpath(dir))
649 self._dirs[dir].setflag(subpath, flags)
649 self._dirs[dir].setflag(subpath, flags)
650 else:
650 else:
651 self._flags[f] = flags
651 self._flags[f] = flags
652 self._dirty = True
652 self._dirty = True
653
653
654 def copy(self):
654 def copy(self):
655 copy = treemanifest(self._dir)
655 copy = treemanifest(self._dir)
656 copy._node = self._node
656 copy._node = self._node
657 copy._dirty = self._dirty
657 copy._dirty = self._dirty
658 if self._copyfunc is _noop:
658 if self._copyfunc is _noop:
659 def _copyfunc(s):
659 def _copyfunc(s):
660 self._load()
660 self._load()
661 for d in self._dirs:
661 for d in self._dirs:
662 s._dirs[d] = self._dirs[d].copy()
662 s._dirs[d] = self._dirs[d].copy()
663 s._files = dict.copy(self._files)
663 s._files = dict.copy(self._files)
664 s._flags = dict.copy(self._flags)
664 s._flags = dict.copy(self._flags)
665 if self._loadfunc is _noop:
665 if self._loadfunc is _noop:
666 _copyfunc(copy)
666 _copyfunc(copy)
667 else:
667 else:
668 copy._copyfunc = _copyfunc
668 copy._copyfunc = _copyfunc
669 else:
669 else:
670 copy._copyfunc = self._copyfunc
670 copy._copyfunc = self._copyfunc
671 return copy
671 return copy
672
672
673 def filesnotin(self, m2):
673 def filesnotin(self, m2):
674 '''Set of files in this manifest that are not in the other'''
674 '''Set of files in this manifest that are not in the other'''
675 files = set()
675 files = set()
676 def _filesnotin(t1, t2):
676 def _filesnotin(t1, t2):
677 if t1._node == t2._node and not t1._dirty and not t2._dirty:
677 if t1._node == t2._node and not t1._dirty and not t2._dirty:
678 return
678 return
679 t1._load()
679 t1._load()
680 t2._load()
680 t2._load()
681 for d, m1 in t1._dirs.iteritems():
681 for d, m1 in t1._dirs.iteritems():
682 if d in t2._dirs:
682 if d in t2._dirs:
683 m2 = t2._dirs[d]
683 m2 = t2._dirs[d]
684 _filesnotin(m1, m2)
684 _filesnotin(m1, m2)
685 else:
685 else:
686 files.update(m1.iterkeys())
686 files.update(m1.iterkeys())
687
687
688 for fn in t1._files.iterkeys():
688 for fn in t1._files.iterkeys():
689 if fn not in t2._files:
689 if fn not in t2._files:
690 files.add(t1._subpath(fn))
690 files.add(t1._subpath(fn))
691
691
692 _filesnotin(self, m2)
692 _filesnotin(self, m2)
693 return files
693 return files
694
694
695 @propertycache
695 @propertycache
696 def _alldirs(self):
696 def _alldirs(self):
697 return util.dirs(self)
697 return util.dirs(self)
698
698
699 def dirs(self):
699 def dirs(self):
700 return self._alldirs
700 return self._alldirs
701
701
702 def hasdir(self, dir):
702 def hasdir(self, dir):
703 self._load()
703 self._load()
704 topdir, subdir = _splittopdir(dir)
704 topdir, subdir = _splittopdir(dir)
705 if topdir:
705 if topdir:
706 if topdir in self._dirs:
706 if topdir in self._dirs:
707 return self._dirs[topdir].hasdir(subdir)
707 return self._dirs[topdir].hasdir(subdir)
708 return False
708 return False
709 return (dir + '/') in self._dirs
709 return (dir + '/') in self._dirs
710
710
711 def walk(self, match):
711 def walk(self, match):
712 '''Generates matching file names.
712 '''Generates matching file names.
713
713
714 Equivalent to manifest.matches(match).iterkeys(), but without creating
714 Equivalent to manifest.matches(match).iterkeys(), but without creating
715 an entirely new manifest.
715 an entirely new manifest.
716
716
717 It also reports nonexistent files by marking them bad with match.bad().
717 It also reports nonexistent files by marking them bad with match.bad().
718 '''
718 '''
719 if match.always():
719 if match.always():
720 for f in iter(self):
720 for f in iter(self):
721 yield f
721 yield f
722 return
722 return
723
723
724 fset = set(match.files())
724 fset = set(match.files())
725
725
726 for fn in self._walk(match):
726 for fn in self._walk(match):
727 if fn in fset:
727 if fn in fset:
728 # specified pattern is the exact name
728 # specified pattern is the exact name
729 fset.remove(fn)
729 fset.remove(fn)
730 yield fn
730 yield fn
731
731
732 # for dirstate.walk, files=['.'] means "walk the whole tree".
732 # for dirstate.walk, files=['.'] means "walk the whole tree".
733 # follow that here, too
733 # follow that here, too
734 fset.discard('.')
734 fset.discard('.')
735
735
736 for fn in sorted(fset):
736 for fn in sorted(fset):
737 if not self.hasdir(fn):
737 if not self.hasdir(fn):
738 match.bad(fn, None)
738 match.bad(fn, None)
739
739
740 def _walk(self, match):
740 def _walk(self, match):
741 '''Recursively generates matching file names for walk().'''
741 '''Recursively generates matching file names for walk().'''
742 if not match.visitdir(self._dir[:-1] or '.'):
742 if not match.visitdir(self._dir[:-1] or '.'):
743 return
743 return
744
744
745 # yield this dir's files and walk its submanifests
745 # yield this dir's files and walk its submanifests
746 self._load()
746 self._load()
747 for p in sorted(self._dirs.keys() + self._files.keys()):
747 for p in sorted(self._dirs.keys() + self._files.keys()):
748 if p in self._files:
748 if p in self._files:
749 fullp = self._subpath(p)
749 fullp = self._subpath(p)
750 if match(fullp):
750 if match(fullp):
751 yield fullp
751 yield fullp
752 else:
752 else:
753 for f in self._dirs[p]._walk(match):
753 for f in self._dirs[p]._walk(match):
754 yield f
754 yield f
755
755
756 def matches(self, match):
756 def matches(self, match):
757 '''generate a new manifest filtered by the match argument'''
757 '''generate a new manifest filtered by the match argument'''
758 if match.always():
758 if match.always():
759 return self.copy()
759 return self.copy()
760
760
761 return self._matches(match)
761 return self._matches(match)
762
762
763 def _matches(self, match):
763 def _matches(self, match):
764 '''recursively generate a new manifest filtered by the match argument.
764 '''recursively generate a new manifest filtered by the match argument.
765 '''
765 '''
766
766
767 visit = match.visitdir(self._dir[:-1] or '.')
767 visit = match.visitdir(self._dir[:-1] or '.')
768 if visit == 'all':
768 if visit == 'all':
769 return self.copy()
769 return self.copy()
770 ret = treemanifest(self._dir)
770 ret = treemanifest(self._dir)
771 if not visit:
771 if not visit:
772 return ret
772 return ret
773
773
774 self._load()
774 self._load()
775 for fn in self._files:
775 for fn in self._files:
776 fullp = self._subpath(fn)
776 fullp = self._subpath(fn)
777 if not match(fullp):
777 if not match(fullp):
778 continue
778 continue
779 ret._files[fn] = self._files[fn]
779 ret._files[fn] = self._files[fn]
780 if fn in self._flags:
780 if fn in self._flags:
781 ret._flags[fn] = self._flags[fn]
781 ret._flags[fn] = self._flags[fn]
782
782
783 for dir, subm in self._dirs.iteritems():
783 for dir, subm in self._dirs.iteritems():
784 m = subm._matches(match)
784 m = subm._matches(match)
785 if not m._isempty():
785 if not m._isempty():
786 ret._dirs[dir] = m
786 ret._dirs[dir] = m
787
787
788 if not ret._isempty():
788 if not ret._isempty():
789 ret._dirty = True
789 ret._dirty = True
790 return ret
790 return ret
791
791
792 def diff(self, m2, clean=False):
792 def diff(self, m2, clean=False):
793 '''Finds changes between the current manifest and m2.
793 '''Finds changes between the current manifest and m2.
794
794
795 Args:
795 Args:
796 m2: the manifest to which this manifest should be compared.
796 m2: the manifest to which this manifest should be compared.
797 clean: if true, include files unchanged between these manifests
797 clean: if true, include files unchanged between these manifests
798 with a None value in the returned dictionary.
798 with a None value in the returned dictionary.
799
799
800 The result is returned as a dict with filename as key and
800 The result is returned as a dict with filename as key and
801 values of the form ((n1,fl1),(n2,fl2)), where n1/n2 is the
801 values of the form ((n1,fl1),(n2,fl2)), where n1/n2 is the
802 nodeid in the current/other manifest and fl1/fl2 is the flag
802 nodeid in the current/other manifest and fl1/fl2 is the flag
803 in the current/other manifest. Where the file does not exist,
803 in the current/other manifest. Where the file does not exist,
804 the nodeid will be None and the flags will be the empty
804 the nodeid will be None and the flags will be the empty
805 string.
805 string.
806 '''
806 '''
807 result = {}
807 result = {}
808 emptytree = treemanifest()
808 emptytree = treemanifest()
809 def _diff(t1, t2):
809 def _diff(t1, t2):
810 if t1._node == t2._node and not t1._dirty and not t2._dirty:
810 if t1._node == t2._node and not t1._dirty and not t2._dirty:
811 return
811 return
812 t1._load()
812 t1._load()
813 t2._load()
813 t2._load()
814 for d, m1 in t1._dirs.iteritems():
814 for d, m1 in t1._dirs.iteritems():
815 m2 = t2._dirs.get(d, emptytree)
815 m2 = t2._dirs.get(d, emptytree)
816 _diff(m1, m2)
816 _diff(m1, m2)
817
817
818 for d, m2 in t2._dirs.iteritems():
818 for d, m2 in t2._dirs.iteritems():
819 if d not in t1._dirs:
819 if d not in t1._dirs:
820 _diff(emptytree, m2)
820 _diff(emptytree, m2)
821
821
822 for fn, n1 in t1._files.iteritems():
822 for fn, n1 in t1._files.iteritems():
823 fl1 = t1._flags.get(fn, '')
823 fl1 = t1._flags.get(fn, '')
824 n2 = t2._files.get(fn, None)
824 n2 = t2._files.get(fn, None)
825 fl2 = t2._flags.get(fn, '')
825 fl2 = t2._flags.get(fn, '')
826 if n1 != n2 or fl1 != fl2:
826 if n1 != n2 or fl1 != fl2:
827 result[t1._subpath(fn)] = ((n1, fl1), (n2, fl2))
827 result[t1._subpath(fn)] = ((n1, fl1), (n2, fl2))
828 elif clean:
828 elif clean:
829 result[t1._subpath(fn)] = None
829 result[t1._subpath(fn)] = None
830
830
831 for fn, n2 in t2._files.iteritems():
831 for fn, n2 in t2._files.iteritems():
832 if fn not in t1._files:
832 if fn not in t1._files:
833 fl2 = t2._flags.get(fn, '')
833 fl2 = t2._flags.get(fn, '')
834 result[t2._subpath(fn)] = ((None, ''), (n2, fl2))
834 result[t2._subpath(fn)] = ((None, ''), (n2, fl2))
835
835
836 _diff(self, m2)
836 _diff(self, m2)
837 return result
837 return result
838
838
839 def unmodifiedsince(self, m2):
839 def unmodifiedsince(self, m2):
840 return not self._dirty and not m2._dirty and self._node == m2._node
840 return not self._dirty and not m2._dirty and self._node == m2._node
841
841
842 def parse(self, text, readsubtree):
842 def parse(self, text, readsubtree):
843 for f, n, fl in _parse(text):
843 for f, n, fl in _parse(text):
844 if fl == 't':
844 if fl == 't':
845 f = f + '/'
845 f = f + '/'
846 self._dirs[f] = readsubtree(self._subpath(f), n)
846 self._dirs[f] = readsubtree(self._subpath(f), n)
847 elif '/' in f:
847 elif '/' in f:
848 # This is a flat manifest, so use __setitem__ and setflag rather
848 # This is a flat manifest, so use __setitem__ and setflag rather
849 # than assigning directly to _files and _flags, so we can
849 # than assigning directly to _files and _flags, so we can
850 # assign a path in a subdirectory, and to mark dirty (compared
850 # assign a path in a subdirectory, and to mark dirty (compared
851 # to nullid).
851 # to nullid).
852 self[f] = n
852 self[f] = n
853 if fl:
853 if fl:
854 self.setflag(f, fl)
854 self.setflag(f, fl)
855 else:
855 else:
856 # Assigning to _files and _flags avoids marking as dirty,
856 # Assigning to _files and _flags avoids marking as dirty,
857 # and should be a little faster.
857 # and should be a little faster.
858 self._files[f] = n
858 self._files[f] = n
859 if fl:
859 if fl:
860 self._flags[f] = fl
860 self._flags[f] = fl
861
861
862 def text(self, usemanifestv2=False):
862 def text(self, usemanifestv2=False):
863 """Get the full data of this manifest as a bytestring."""
863 """Get the full data of this manifest as a bytestring."""
864 self._load()
864 self._load()
865 return _text(self.iterentries(), usemanifestv2)
865 return _text(self.iterentries(), usemanifestv2)
866
866
867 def dirtext(self, usemanifestv2=False):
867 def dirtext(self, usemanifestv2=False):
868 """Get the full data of this directory as a bytestring. Make sure that
868 """Get the full data of this directory as a bytestring. Make sure that
869 any submanifests have been written first, so their nodeids are correct.
869 any submanifests have been written first, so their nodeids are correct.
870 """
870 """
871 self._load()
871 self._load()
872 flags = self.flags
872 flags = self.flags
873 dirs = [(d[:-1], self._dirs[d]._node, 't') for d in self._dirs]
873 dirs = [(d[:-1], self._dirs[d]._node, 't') for d in self._dirs]
874 files = [(f, self._files[f], flags(f)) for f in self._files]
874 files = [(f, self._files[f], flags(f)) for f in self._files]
875 return _text(sorted(dirs + files), usemanifestv2)
875 return _text(sorted(dirs + files), usemanifestv2)
876
876
877 def read(self, gettext, readsubtree):
877 def read(self, gettext, readsubtree):
878 def _load_for_read(s):
878 def _load_for_read(s):
879 s.parse(gettext(), readsubtree)
879 s.parse(gettext(), readsubtree)
880 s._dirty = False
880 s._dirty = False
881 self._loadfunc = _load_for_read
881 self._loadfunc = _load_for_read
882
882
883 def writesubtrees(self, m1, m2, writesubtree):
883 def writesubtrees(self, m1, m2, writesubtree):
884 self._load() # for consistency; should never have any effect here
884 self._load() # for consistency; should never have any effect here
885 m1._load()
885 m1._load()
886 m2._load()
886 m2._load()
887 emptytree = treemanifest()
887 emptytree = treemanifest()
888 for d, subm in self._dirs.iteritems():
888 for d, subm in self._dirs.iteritems():
889 subp1 = m1._dirs.get(d, emptytree)._node
889 subp1 = m1._dirs.get(d, emptytree)._node
890 subp2 = m2._dirs.get(d, emptytree)._node
890 subp2 = m2._dirs.get(d, emptytree)._node
891 if subp1 == revlog.nullid:
891 if subp1 == revlog.nullid:
892 subp1, subp2 = subp2, subp1
892 subp1, subp2 = subp2, subp1
893 writesubtree(subm, subp1, subp2)
893 writesubtree(subm, subp1, subp2)
894
894
895 class manifestrevlog(revlog.revlog):
895 class manifestrevlog(revlog.revlog):
896 '''A revlog that stores manifest texts. This is responsible for caching the
896 '''A revlog that stores manifest texts. This is responsible for caching the
897 full-text manifest contents.
897 full-text manifest contents.
898 '''
898 '''
899 def __init__(self, opener, dir='', dirlogcache=None):
899 def __init__(self, opener, dir='', dirlogcache=None):
900 # During normal operations, we expect to deal with not more than four
900 # During normal operations, we expect to deal with not more than four
901 # revs at a time (such as during commit --amend). When rebasing large
901 # revs at a time (such as during commit --amend). When rebasing large
902 # stacks of commits, the number can go up, hence the config knob below.
902 # stacks of commits, the number can go up, hence the config knob below.
903 cachesize = 4
903 cachesize = 4
904 usetreemanifest = False
904 usetreemanifest = False
905 usemanifestv2 = False
905 usemanifestv2 = False
906 opts = getattr(opener, 'options', None)
906 opts = getattr(opener, 'options', None)
907 if opts is not None:
907 if opts is not None:
908 cachesize = opts.get('manifestcachesize', cachesize)
908 cachesize = opts.get('manifestcachesize', cachesize)
909 usetreemanifest = opts.get('treemanifest', usetreemanifest)
909 usetreemanifest = opts.get('treemanifest', usetreemanifest)
910 usemanifestv2 = opts.get('manifestv2', usemanifestv2)
910 usemanifestv2 = opts.get('manifestv2', usemanifestv2)
911
911
912 self._treeondisk = usetreemanifest
912 self._treeondisk = usetreemanifest
913 self._usemanifestv2 = usemanifestv2
913 self._usemanifestv2 = usemanifestv2
914
914
915 self._fulltextcache = util.lrucachedict(cachesize)
915 self._fulltextcache = util.lrucachedict(cachesize)
916
916
917 indexfile = "00manifest.i"
917 indexfile = "00manifest.i"
918 if dir:
918 if dir:
919 assert self._treeondisk, 'opts is %r' % opts
919 assert self._treeondisk, 'opts is %r' % opts
920 if not dir.endswith('/'):
920 if not dir.endswith('/'):
921 dir = dir + '/'
921 dir = dir + '/'
922 indexfile = "meta/" + dir + "00manifest.i"
922 indexfile = "meta/" + dir + "00manifest.i"
923 self._dir = dir
923 self._dir = dir
924 # The dirlogcache is kept on the root manifest log
924 # The dirlogcache is kept on the root manifest log
925 if dir:
925 if dir:
926 self._dirlogcache = dirlogcache
926 self._dirlogcache = dirlogcache
927 else:
927 else:
928 self._dirlogcache = {'': self}
928 self._dirlogcache = {'': self}
929
929
930 super(manifestrevlog, self).__init__(opener, indexfile)
930 super(manifestrevlog, self).__init__(opener, indexfile,
931 checkambig=bool(dir))
931
932
932 @property
933 @property
933 def fulltextcache(self):
934 def fulltextcache(self):
934 return self._fulltextcache
935 return self._fulltextcache
935
936
936 def clearcaches(self):
937 def clearcaches(self):
937 super(manifestrevlog, self).clearcaches()
938 super(manifestrevlog, self).clearcaches()
938 self._fulltextcache.clear()
939 self._fulltextcache.clear()
939 self._dirlogcache = {'': self}
940 self._dirlogcache = {'': self}
940
941
941 def dirlog(self, dir):
942 def dirlog(self, dir):
942 if dir:
943 if dir:
943 assert self._treeondisk
944 assert self._treeondisk
944 if dir not in self._dirlogcache:
945 if dir not in self._dirlogcache:
945 self._dirlogcache[dir] = manifestrevlog(self.opener, dir,
946 self._dirlogcache[dir] = manifestrevlog(self.opener, dir,
946 self._dirlogcache)
947 self._dirlogcache)
947 return self._dirlogcache[dir]
948 return self._dirlogcache[dir]
948
949
949 def add(self, m, transaction, link, p1, p2, added, removed):
950 def add(self, m, transaction, link, p1, p2, added, removed):
950 if (p1 in self.fulltextcache and util.safehasattr(m, 'fastdelta')
951 if (p1 in self.fulltextcache and util.safehasattr(m, 'fastdelta')
951 and not self._usemanifestv2):
952 and not self._usemanifestv2):
952 # If our first parent is in the manifest cache, we can
953 # If our first parent is in the manifest cache, we can
953 # compute a delta here using properties we know about the
954 # compute a delta here using properties we know about the
954 # manifest up-front, which may save time later for the
955 # manifest up-front, which may save time later for the
955 # revlog layer.
956 # revlog layer.
956
957
957 _checkforbidden(added)
958 _checkforbidden(added)
958 # combine the changed lists into one sorted iterator
959 # combine the changed lists into one sorted iterator
959 work = heapq.merge([(x, False) for x in added],
960 work = heapq.merge([(x, False) for x in added],
960 [(x, True) for x in removed])
961 [(x, True) for x in removed])
961
962
962 arraytext, deltatext = m.fastdelta(self.fulltextcache[p1], work)
963 arraytext, deltatext = m.fastdelta(self.fulltextcache[p1], work)
963 cachedelta = self.rev(p1), deltatext
964 cachedelta = self.rev(p1), deltatext
964 text = util.buffer(arraytext)
965 text = util.buffer(arraytext)
965 n = self.addrevision(text, transaction, link, p1, p2, cachedelta)
966 n = self.addrevision(text, transaction, link, p1, p2, cachedelta)
966 else:
967 else:
967 # The first parent manifest isn't already loaded, so we'll
968 # The first parent manifest isn't already loaded, so we'll
968 # just encode a fulltext of the manifest and pass that
969 # just encode a fulltext of the manifest and pass that
969 # through to the revlog layer, and let it handle the delta
970 # through to the revlog layer, and let it handle the delta
970 # process.
971 # process.
971 if self._treeondisk:
972 if self._treeondisk:
972 m1 = self.read(p1)
973 m1 = self.read(p1)
973 m2 = self.read(p2)
974 m2 = self.read(p2)
974 n = self._addtree(m, transaction, link, m1, m2)
975 n = self._addtree(m, transaction, link, m1, m2)
975 arraytext = None
976 arraytext = None
976 else:
977 else:
977 text = m.text(self._usemanifestv2)
978 text = m.text(self._usemanifestv2)
978 n = self.addrevision(text, transaction, link, p1, p2)
979 n = self.addrevision(text, transaction, link, p1, p2)
979 arraytext = array.array('c', text)
980 arraytext = array.array('c', text)
980
981
981 self.fulltextcache[n] = arraytext
982 self.fulltextcache[n] = arraytext
982
983
983 return n
984 return n
984
985
985 def _addtree(self, m, transaction, link, m1, m2):
986 def _addtree(self, m, transaction, link, m1, m2):
986 # If the manifest is unchanged compared to one parent,
987 # If the manifest is unchanged compared to one parent,
987 # don't write a new revision
988 # don't write a new revision
988 if m.unmodifiedsince(m1) or m.unmodifiedsince(m2):
989 if m.unmodifiedsince(m1) or m.unmodifiedsince(m2):
989 return m.node()
990 return m.node()
990 def writesubtree(subm, subp1, subp2):
991 def writesubtree(subm, subp1, subp2):
991 sublog = self.dirlog(subm.dir())
992 sublog = self.dirlog(subm.dir())
992 sublog.add(subm, transaction, link, subp1, subp2, None, None)
993 sublog.add(subm, transaction, link, subp1, subp2, None, None)
993 m.writesubtrees(m1, m2, writesubtree)
994 m.writesubtrees(m1, m2, writesubtree)
994 text = m.dirtext(self._usemanifestv2)
995 text = m.dirtext(self._usemanifestv2)
995 # Double-check whether contents are unchanged to one parent
996 # Double-check whether contents are unchanged to one parent
996 if text == m1.dirtext(self._usemanifestv2):
997 if text == m1.dirtext(self._usemanifestv2):
997 n = m1.node()
998 n = m1.node()
998 elif text == m2.dirtext(self._usemanifestv2):
999 elif text == m2.dirtext(self._usemanifestv2):
999 n = m2.node()
1000 n = m2.node()
1000 else:
1001 else:
1001 n = self.addrevision(text, transaction, link, m1.node(), m2.node())
1002 n = self.addrevision(text, transaction, link, m1.node(), m2.node())
1002 # Save nodeid so parent manifest can calculate its nodeid
1003 # Save nodeid so parent manifest can calculate its nodeid
1003 m.setnode(n)
1004 m.setnode(n)
1004 return n
1005 return n
1005
1006
1006 class manifestlog(object):
1007 class manifestlog(object):
1007 """A collection class representing the collection of manifest snapshots
1008 """A collection class representing the collection of manifest snapshots
1008 referenced by commits in the repository.
1009 referenced by commits in the repository.
1009
1010
1010 In this situation, 'manifest' refers to the abstract concept of a snapshot
1011 In this situation, 'manifest' refers to the abstract concept of a snapshot
1011 of the list of files in the given commit. Consumers of the output of this
1012 of the list of files in the given commit. Consumers of the output of this
1012 class do not care about the implementation details of the actual manifests
1013 class do not care about the implementation details of the actual manifests
1013 they receive (i.e. tree or flat or lazily loaded, etc)."""
1014 they receive (i.e. tree or flat or lazily loaded, etc)."""
1014 def __init__(self, opener, repo):
1015 def __init__(self, opener, repo):
1015 self._repo = repo
1016 self._repo = repo
1016
1017
1017 usetreemanifest = False
1018 usetreemanifest = False
1018
1019
1019 opts = getattr(opener, 'options', None)
1020 opts = getattr(opener, 'options', None)
1020 if opts is not None:
1021 if opts is not None:
1021 usetreemanifest = opts.get('treemanifest', usetreemanifest)
1022 usetreemanifest = opts.get('treemanifest', usetreemanifest)
1022 self._treeinmem = usetreemanifest
1023 self._treeinmem = usetreemanifest
1023
1024
1024 # We'll separate this into it's own cache once oldmanifest is no longer
1025 # We'll separate this into it's own cache once oldmanifest is no longer
1025 # used
1026 # used
1026 self._mancache = repo.manifest._mancache
1027 self._mancache = repo.manifest._mancache
1027
1028
1028 @property
1029 @property
1029 def _revlog(self):
1030 def _revlog(self):
1030 return self._repo.manifest
1031 return self._repo.manifest
1031
1032
1032 def __getitem__(self, node):
1033 def __getitem__(self, node):
1033 """Retrieves the manifest instance for the given node. Throws a KeyError
1034 """Retrieves the manifest instance for the given node. Throws a KeyError
1034 if not found.
1035 if not found.
1035 """
1036 """
1036 if node in self._mancache:
1037 if node in self._mancache:
1037 cachemf = self._mancache[node]
1038 cachemf = self._mancache[node]
1038 # The old manifest may put non-ctx manifests in the cache, so skip
1039 # The old manifest may put non-ctx manifests in the cache, so skip
1039 # those since they don't implement the full api.
1040 # those since they don't implement the full api.
1040 if (isinstance(cachemf, manifestctx) or
1041 if (isinstance(cachemf, manifestctx) or
1041 isinstance(cachemf, treemanifestctx)):
1042 isinstance(cachemf, treemanifestctx)):
1042 return cachemf
1043 return cachemf
1043
1044
1044 if self._treeinmem:
1045 if self._treeinmem:
1045 m = treemanifestctx(self._revlog, '', node)
1046 m = treemanifestctx(self._revlog, '', node)
1046 else:
1047 else:
1047 m = manifestctx(self._revlog, node)
1048 m = manifestctx(self._revlog, node)
1048 if node != revlog.nullid:
1049 if node != revlog.nullid:
1049 self._mancache[node] = m
1050 self._mancache[node] = m
1050 return m
1051 return m
1051
1052
1052 def add(self, m, transaction, link, p1, p2, added, removed):
1053 def add(self, m, transaction, link, p1, p2, added, removed):
1053 return self._revlog.add(m, transaction, link, p1, p2, added, removed)
1054 return self._revlog.add(m, transaction, link, p1, p2, added, removed)
1054
1055
1055 class manifestctx(object):
1056 class manifestctx(object):
1056 """A class representing a single revision of a manifest, including its
1057 """A class representing a single revision of a manifest, including its
1057 contents, its parent revs, and its linkrev.
1058 contents, its parent revs, and its linkrev.
1058 """
1059 """
1059 def __init__(self, revlog, node):
1060 def __init__(self, revlog, node):
1060 self._revlog = revlog
1061 self._revlog = revlog
1061 self._data = None
1062 self._data = None
1062
1063
1063 self._node = node
1064 self._node = node
1064
1065
1065 # TODO: We eventually want p1, p2, and linkrev exposed on this class,
1066 # TODO: We eventually want p1, p2, and linkrev exposed on this class,
1066 # but let's add it later when something needs it and we can load it
1067 # but let's add it later when something needs it and we can load it
1067 # lazily.
1068 # lazily.
1068 #self.p1, self.p2 = revlog.parents(node)
1069 #self.p1, self.p2 = revlog.parents(node)
1069 #rev = revlog.rev(node)
1070 #rev = revlog.rev(node)
1070 #self.linkrev = revlog.linkrev(rev)
1071 #self.linkrev = revlog.linkrev(rev)
1071
1072
1072 def node(self):
1073 def node(self):
1073 return self._node
1074 return self._node
1074
1075
1075 def read(self):
1076 def read(self):
1076 if not self._data:
1077 if not self._data:
1077 if self._node == revlog.nullid:
1078 if self._node == revlog.nullid:
1078 self._data = manifestdict()
1079 self._data = manifestdict()
1079 else:
1080 else:
1080 text = self._revlog.revision(self._node)
1081 text = self._revlog.revision(self._node)
1081 arraytext = array.array('c', text)
1082 arraytext = array.array('c', text)
1082 self._revlog._fulltextcache[self._node] = arraytext
1083 self._revlog._fulltextcache[self._node] = arraytext
1083 self._data = manifestdict(text)
1084 self._data = manifestdict(text)
1084 return self._data
1085 return self._data
1085
1086
1086 def readfast(self):
1087 def readfast(self):
1087 rl = self._revlog
1088 rl = self._revlog
1088 r = rl.rev(self._node)
1089 r = rl.rev(self._node)
1089 deltaparent = rl.deltaparent(r)
1090 deltaparent = rl.deltaparent(r)
1090 if deltaparent != revlog.nullrev and deltaparent in rl.parentrevs(r):
1091 if deltaparent != revlog.nullrev and deltaparent in rl.parentrevs(r):
1091 return self.readdelta()
1092 return self.readdelta()
1092 return self.read()
1093 return self.read()
1093
1094
1094 def readdelta(self):
1095 def readdelta(self):
1095 revlog = self._revlog
1096 revlog = self._revlog
1096 if revlog._usemanifestv2:
1097 if revlog._usemanifestv2:
1097 # Need to perform a slow delta
1098 # Need to perform a slow delta
1098 r0 = revlog.deltaparent(revlog.rev(self._node))
1099 r0 = revlog.deltaparent(revlog.rev(self._node))
1099 m0 = manifestctx(revlog, revlog.node(r0)).read()
1100 m0 = manifestctx(revlog, revlog.node(r0)).read()
1100 m1 = self.read()
1101 m1 = self.read()
1101 md = manifestdict()
1102 md = manifestdict()
1102 for f, ((n0, fl0), (n1, fl1)) in m0.diff(m1).iteritems():
1103 for f, ((n0, fl0), (n1, fl1)) in m0.diff(m1).iteritems():
1103 if n1:
1104 if n1:
1104 md[f] = n1
1105 md[f] = n1
1105 if fl1:
1106 if fl1:
1106 md.setflag(f, fl1)
1107 md.setflag(f, fl1)
1107 return md
1108 return md
1108
1109
1109 r = revlog.rev(self._node)
1110 r = revlog.rev(self._node)
1110 d = mdiff.patchtext(revlog.revdiff(revlog.deltaparent(r), r))
1111 d = mdiff.patchtext(revlog.revdiff(revlog.deltaparent(r), r))
1111 return manifestdict(d)
1112 return manifestdict(d)
1112
1113
1113 class treemanifestctx(object):
1114 class treemanifestctx(object):
1114 def __init__(self, revlog, dir, node):
1115 def __init__(self, revlog, dir, node):
1115 revlog = revlog.dirlog(dir)
1116 revlog = revlog.dirlog(dir)
1116 self._revlog = revlog
1117 self._revlog = revlog
1117 self._dir = dir
1118 self._dir = dir
1118 self._data = None
1119 self._data = None
1119
1120
1120 self._node = node
1121 self._node = node
1121
1122
1122 # TODO: Load p1/p2/linkrev lazily. They need to be lazily loaded so that
1123 # TODO: Load p1/p2/linkrev lazily. They need to be lazily loaded so that
1123 # we can instantiate treemanifestctx objects for directories we don't
1124 # we can instantiate treemanifestctx objects for directories we don't
1124 # have on disk.
1125 # have on disk.
1125 #self.p1, self.p2 = revlog.parents(node)
1126 #self.p1, self.p2 = revlog.parents(node)
1126 #rev = revlog.rev(node)
1127 #rev = revlog.rev(node)
1127 #self.linkrev = revlog.linkrev(rev)
1128 #self.linkrev = revlog.linkrev(rev)
1128
1129
1129 def read(self):
1130 def read(self):
1130 if not self._data:
1131 if not self._data:
1131 if self._node == revlog.nullid:
1132 if self._node == revlog.nullid:
1132 self._data = treemanifest()
1133 self._data = treemanifest()
1133 elif self._revlog._treeondisk:
1134 elif self._revlog._treeondisk:
1134 m = treemanifest(dir=self._dir)
1135 m = treemanifest(dir=self._dir)
1135 def gettext():
1136 def gettext():
1136 return self._revlog.revision(self._node)
1137 return self._revlog.revision(self._node)
1137 def readsubtree(dir, subm):
1138 def readsubtree(dir, subm):
1138 return treemanifestctx(self._revlog, dir, subm).read()
1139 return treemanifestctx(self._revlog, dir, subm).read()
1139 m.read(gettext, readsubtree)
1140 m.read(gettext, readsubtree)
1140 m.setnode(self._node)
1141 m.setnode(self._node)
1141 self._data = m
1142 self._data = m
1142 else:
1143 else:
1143 text = self._revlog.revision(self._node)
1144 text = self._revlog.revision(self._node)
1144 arraytext = array.array('c', text)
1145 arraytext = array.array('c', text)
1145 self._revlog.fulltextcache[self._node] = arraytext
1146 self._revlog.fulltextcache[self._node] = arraytext
1146 self._data = treemanifest(dir=self._dir, text=text)
1147 self._data = treemanifest(dir=self._dir, text=text)
1147
1148
1148 return self._data
1149 return self._data
1149
1150
1150 def node(self):
1151 def node(self):
1151 return self._node
1152 return self._node
1152
1153
1153 def readdelta(self):
1154 def readdelta(self):
1154 # Need to perform a slow delta
1155 # Need to perform a slow delta
1155 revlog = self._revlog
1156 revlog = self._revlog
1156 r0 = revlog.deltaparent(revlog.rev(self._node))
1157 r0 = revlog.deltaparent(revlog.rev(self._node))
1157 m0 = treemanifestctx(revlog, revlog.node(r0), dir=self._dir).read()
1158 m0 = treemanifestctx(revlog, revlog.node(r0), dir=self._dir).read()
1158 m1 = self.read()
1159 m1 = self.read()
1159 md = treemanifest(dir=self._dir)
1160 md = treemanifest(dir=self._dir)
1160 for f, ((n0, fl0), (n1, fl1)) in m0.diff(m1).iteritems():
1161 for f, ((n0, fl0), (n1, fl1)) in m0.diff(m1).iteritems():
1161 if n1:
1162 if n1:
1162 md[f] = n1
1163 md[f] = n1
1163 if fl1:
1164 if fl1:
1164 md.setflag(f, fl1)
1165 md.setflag(f, fl1)
1165 return md
1166 return md
1166
1167
1167 def readfast(self):
1168 def readfast(self):
1168 rl = self._revlog
1169 rl = self._revlog
1169 r = rl.rev(self._node)
1170 r = rl.rev(self._node)
1170 deltaparent = rl.deltaparent(r)
1171 deltaparent = rl.deltaparent(r)
1171 if deltaparent != revlog.nullrev and deltaparent in rl.parentrevs(r):
1172 if deltaparent != revlog.nullrev and deltaparent in rl.parentrevs(r):
1172 return self.readdelta()
1173 return self.readdelta()
1173 return self.read()
1174 return self.read()
1174
1175
1175 class manifest(manifestrevlog):
1176 class manifest(manifestrevlog):
1176 def __init__(self, opener, dir='', dirlogcache=None):
1177 def __init__(self, opener, dir='', dirlogcache=None):
1177 '''The 'dir' and 'dirlogcache' arguments are for internal use by
1178 '''The 'dir' and 'dirlogcache' arguments are for internal use by
1178 manifest.manifest only. External users should create a root manifest
1179 manifest.manifest only. External users should create a root manifest
1179 log with manifest.manifest(opener) and call dirlog() on it.
1180 log with manifest.manifest(opener) and call dirlog() on it.
1180 '''
1181 '''
1181 # During normal operations, we expect to deal with not more than four
1182 # During normal operations, we expect to deal with not more than four
1182 # revs at a time (such as during commit --amend). When rebasing large
1183 # revs at a time (such as during commit --amend). When rebasing large
1183 # stacks of commits, the number can go up, hence the config knob below.
1184 # stacks of commits, the number can go up, hence the config knob below.
1184 cachesize = 4
1185 cachesize = 4
1185 usetreemanifest = False
1186 usetreemanifest = False
1186 opts = getattr(opener, 'options', None)
1187 opts = getattr(opener, 'options', None)
1187 if opts is not None:
1188 if opts is not None:
1188 cachesize = opts.get('manifestcachesize', cachesize)
1189 cachesize = opts.get('manifestcachesize', cachesize)
1189 usetreemanifest = opts.get('treemanifest', usetreemanifest)
1190 usetreemanifest = opts.get('treemanifest', usetreemanifest)
1190 self._mancache = util.lrucachedict(cachesize)
1191 self._mancache = util.lrucachedict(cachesize)
1191 self._treeinmem = usetreemanifest
1192 self._treeinmem = usetreemanifest
1192 super(manifest, self).__init__(opener, dir=dir, dirlogcache=dirlogcache)
1193 super(manifest, self).__init__(opener, dir=dir, dirlogcache=dirlogcache)
1193
1194
1194 def _newmanifest(self, data=''):
1195 def _newmanifest(self, data=''):
1195 if self._treeinmem:
1196 if self._treeinmem:
1196 return treemanifest(self._dir, data)
1197 return treemanifest(self._dir, data)
1197 return manifestdict(data)
1198 return manifestdict(data)
1198
1199
1199 def dirlog(self, dir):
1200 def dirlog(self, dir):
1200 """This overrides the base revlog implementation to allow construction
1201 """This overrides the base revlog implementation to allow construction
1201 'manifest' types instead of manifestrevlog types. This is only needed
1202 'manifest' types instead of manifestrevlog types. This is only needed
1202 until we migrate off the 'manifest' type."""
1203 until we migrate off the 'manifest' type."""
1203 if dir:
1204 if dir:
1204 assert self._treeondisk
1205 assert self._treeondisk
1205 if dir not in self._dirlogcache:
1206 if dir not in self._dirlogcache:
1206 self._dirlogcache[dir] = manifest(self.opener, dir,
1207 self._dirlogcache[dir] = manifest(self.opener, dir,
1207 self._dirlogcache)
1208 self._dirlogcache)
1208 return self._dirlogcache[dir]
1209 return self._dirlogcache[dir]
1209
1210
1210 def _slowreaddelta(self, node):
1211 def _slowreaddelta(self, node):
1211 r0 = self.deltaparent(self.rev(node))
1212 r0 = self.deltaparent(self.rev(node))
1212 m0 = self.read(self.node(r0))
1213 m0 = self.read(self.node(r0))
1213 m1 = self.read(node)
1214 m1 = self.read(node)
1214 md = self._newmanifest()
1215 md = self._newmanifest()
1215 for f, ((n0, fl0), (n1, fl1)) in m0.diff(m1).iteritems():
1216 for f, ((n0, fl0), (n1, fl1)) in m0.diff(m1).iteritems():
1216 if n1:
1217 if n1:
1217 md[f] = n1
1218 md[f] = n1
1218 if fl1:
1219 if fl1:
1219 md.setflag(f, fl1)
1220 md.setflag(f, fl1)
1220 return md
1221 return md
1221
1222
1222 def readdelta(self, node):
1223 def readdelta(self, node):
1223 if self._usemanifestv2 or self._treeondisk:
1224 if self._usemanifestv2 or self._treeondisk:
1224 return self._slowreaddelta(node)
1225 return self._slowreaddelta(node)
1225 r = self.rev(node)
1226 r = self.rev(node)
1226 d = mdiff.patchtext(self.revdiff(self.deltaparent(r), r))
1227 d = mdiff.patchtext(self.revdiff(self.deltaparent(r), r))
1227 return self._newmanifest(d)
1228 return self._newmanifest(d)
1228
1229
1229 def readshallowdelta(self, node):
1230 def readshallowdelta(self, node):
1230 '''For flat manifests, this is the same as readdelta(). For
1231 '''For flat manifests, this is the same as readdelta(). For
1231 treemanifests, this will read the delta for this revlog's directory,
1232 treemanifests, this will read the delta for this revlog's directory,
1232 without recursively reading subdirectory manifests. Instead, any
1233 without recursively reading subdirectory manifests. Instead, any
1233 subdirectory entry will be reported as it appears in the manifests, i.e.
1234 subdirectory entry will be reported as it appears in the manifests, i.e.
1234 the subdirectory will be reported among files and distinguished only by
1235 the subdirectory will be reported among files and distinguished only by
1235 its 't' flag.'''
1236 its 't' flag.'''
1236 if not self._treeondisk:
1237 if not self._treeondisk:
1237 return self.readdelta(node)
1238 return self.readdelta(node)
1238 if self._usemanifestv2:
1239 if self._usemanifestv2:
1239 raise error.Abort(
1240 raise error.Abort(
1240 _("readshallowdelta() not implemented for manifestv2"))
1241 _("readshallowdelta() not implemented for manifestv2"))
1241 r = self.rev(node)
1242 r = self.rev(node)
1242 d = mdiff.patchtext(self.revdiff(self.deltaparent(r), r))
1243 d = mdiff.patchtext(self.revdiff(self.deltaparent(r), r))
1243 return manifestdict(d)
1244 return manifestdict(d)
1244
1245
1245 def readshallowfast(self, node):
1246 def readshallowfast(self, node):
1246 '''like readfast(), but calls readshallowdelta() instead of readdelta()
1247 '''like readfast(), but calls readshallowdelta() instead of readdelta()
1247 '''
1248 '''
1248 r = self.rev(node)
1249 r = self.rev(node)
1249 deltaparent = self.deltaparent(r)
1250 deltaparent = self.deltaparent(r)
1250 if deltaparent != revlog.nullrev and deltaparent in self.parentrevs(r):
1251 if deltaparent != revlog.nullrev and deltaparent in self.parentrevs(r):
1251 return self.readshallowdelta(node)
1252 return self.readshallowdelta(node)
1252 return self.readshallow(node)
1253 return self.readshallow(node)
1253
1254
1254 def read(self, node):
1255 def read(self, node):
1255 if node == revlog.nullid:
1256 if node == revlog.nullid:
1256 return self._newmanifest() # don't upset local cache
1257 return self._newmanifest() # don't upset local cache
1257 if node in self._mancache:
1258 if node in self._mancache:
1258 cached = self._mancache[node]
1259 cached = self._mancache[node]
1259 if (isinstance(cached, manifestctx) or
1260 if (isinstance(cached, manifestctx) or
1260 isinstance(cached, treemanifestctx)):
1261 isinstance(cached, treemanifestctx)):
1261 cached = cached.read()
1262 cached = cached.read()
1262 return cached
1263 return cached
1263 if self._treeondisk:
1264 if self._treeondisk:
1264 def gettext():
1265 def gettext():
1265 return self.revision(node)
1266 return self.revision(node)
1266 def readsubtree(dir, subm):
1267 def readsubtree(dir, subm):
1267 return self.dirlog(dir).read(subm)
1268 return self.dirlog(dir).read(subm)
1268 m = self._newmanifest()
1269 m = self._newmanifest()
1269 m.read(gettext, readsubtree)
1270 m.read(gettext, readsubtree)
1270 m.setnode(node)
1271 m.setnode(node)
1271 arraytext = None
1272 arraytext = None
1272 else:
1273 else:
1273 text = self.revision(node)
1274 text = self.revision(node)
1274 m = self._newmanifest(text)
1275 m = self._newmanifest(text)
1275 arraytext = array.array('c', text)
1276 arraytext = array.array('c', text)
1276 self._mancache[node] = m
1277 self._mancache[node] = m
1277 self.fulltextcache[node] = arraytext
1278 self.fulltextcache[node] = arraytext
1278 return m
1279 return m
1279
1280
1280 def readshallow(self, node):
1281 def readshallow(self, node):
1281 '''Reads the manifest in this directory. When using flat manifests,
1282 '''Reads the manifest in this directory. When using flat manifests,
1282 this manifest will generally have files in subdirectories in it. Does
1283 this manifest will generally have files in subdirectories in it. Does
1283 not cache the manifest as the callers generally do not read the same
1284 not cache the manifest as the callers generally do not read the same
1284 version twice.'''
1285 version twice.'''
1285 return manifestdict(self.revision(node))
1286 return manifestdict(self.revision(node))
1286
1287
1287 def find(self, node, f):
1288 def find(self, node, f):
1288 '''look up entry for a single file efficiently.
1289 '''look up entry for a single file efficiently.
1289 return (node, flags) pair if found, (None, None) if not.'''
1290 return (node, flags) pair if found, (None, None) if not.'''
1290 m = self.read(node)
1291 m = self.read(node)
1291 try:
1292 try:
1292 return m.find(f)
1293 return m.find(f)
1293 except KeyError:
1294 except KeyError:
1294 return None, None
1295 return None, None
1295
1296
1296 def clearcaches(self):
1297 def clearcaches(self):
1297 super(manifest, self).clearcaches()
1298 super(manifest, self).clearcaches()
1298 self._mancache.clear()
1299 self._mancache.clear()
General Comments 0
You need to be logged in to leave comments. Login now