##// END OF EJS Templates
changegroup: move sorting down into group
Matt Mackall -
r13784:f1e639c7 default
parent child Browse files
Show More
@@ -1,1970 +1,1969 b''
1 # localrepo.py - read/write repository class for mercurial
1 # localrepo.py - read/write repository 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 node import bin, hex, nullid, nullrev, short
8 from node import bin, hex, nullid, nullrev, short
9 from i18n import _
9 from i18n import _
10 import repo, changegroup, subrepo, discovery, pushkey
10 import repo, changegroup, subrepo, discovery, pushkey
11 import changelog, dirstate, filelog, manifest, context, bookmarks
11 import changelog, dirstate, filelog, manifest, context, bookmarks
12 import lock, transaction, store, encoding
12 import lock, transaction, store, encoding
13 import util, extensions, hook, error
13 import util, extensions, hook, error
14 import match as matchmod
14 import match as matchmod
15 import merge as mergemod
15 import merge as mergemod
16 import tags as tagsmod
16 import tags as tagsmod
17 import url as urlmod
17 import url as urlmod
18 from lock import release
18 from lock import release
19 import weakref, errno, os, time, inspect
19 import weakref, errno, os, time, inspect
20 propertycache = util.propertycache
20 propertycache = util.propertycache
21
21
22 class localrepository(repo.repository):
22 class localrepository(repo.repository):
23 capabilities = set(('lookup', 'changegroupsubset', 'branchmap', 'pushkey',
23 capabilities = set(('lookup', 'changegroupsubset', 'branchmap', 'pushkey',
24 'known', 'getbundle'))
24 'known', 'getbundle'))
25 supportedformats = set(('revlogv1', 'parentdelta'))
25 supportedformats = set(('revlogv1', 'parentdelta'))
26 supported = supportedformats | set(('store', 'fncache', 'shared',
26 supported = supportedformats | set(('store', 'fncache', 'shared',
27 'dotencode'))
27 'dotencode'))
28
28
29 def __init__(self, baseui, path=None, create=0):
29 def __init__(self, baseui, path=None, create=0):
30 repo.repository.__init__(self)
30 repo.repository.__init__(self)
31 self.root = os.path.realpath(util.expandpath(path))
31 self.root = os.path.realpath(util.expandpath(path))
32 self.path = os.path.join(self.root, ".hg")
32 self.path = os.path.join(self.root, ".hg")
33 self.origroot = path
33 self.origroot = path
34 self.auditor = util.path_auditor(self.root, self._checknested)
34 self.auditor = util.path_auditor(self.root, self._checknested)
35 self.opener = util.opener(self.path)
35 self.opener = util.opener(self.path)
36 self.wopener = util.opener(self.root)
36 self.wopener = util.opener(self.root)
37 self.baseui = baseui
37 self.baseui = baseui
38 self.ui = baseui.copy()
38 self.ui = baseui.copy()
39
39
40 try:
40 try:
41 self.ui.readconfig(self.join("hgrc"), self.root)
41 self.ui.readconfig(self.join("hgrc"), self.root)
42 extensions.loadall(self.ui)
42 extensions.loadall(self.ui)
43 except IOError:
43 except IOError:
44 pass
44 pass
45
45
46 if not os.path.isdir(self.path):
46 if not os.path.isdir(self.path):
47 if create:
47 if create:
48 if not os.path.exists(path):
48 if not os.path.exists(path):
49 util.makedirs(path)
49 util.makedirs(path)
50 os.mkdir(self.path)
50 os.mkdir(self.path)
51 requirements = ["revlogv1"]
51 requirements = ["revlogv1"]
52 if self.ui.configbool('format', 'usestore', True):
52 if self.ui.configbool('format', 'usestore', True):
53 os.mkdir(os.path.join(self.path, "store"))
53 os.mkdir(os.path.join(self.path, "store"))
54 requirements.append("store")
54 requirements.append("store")
55 if self.ui.configbool('format', 'usefncache', True):
55 if self.ui.configbool('format', 'usefncache', True):
56 requirements.append("fncache")
56 requirements.append("fncache")
57 if self.ui.configbool('format', 'dotencode', True):
57 if self.ui.configbool('format', 'dotencode', True):
58 requirements.append('dotencode')
58 requirements.append('dotencode')
59 # create an invalid changelog
59 # create an invalid changelog
60 self.opener("00changelog.i", "a").write(
60 self.opener("00changelog.i", "a").write(
61 '\0\0\0\2' # represents revlogv2
61 '\0\0\0\2' # represents revlogv2
62 ' dummy changelog to prevent using the old repo layout'
62 ' dummy changelog to prevent using the old repo layout'
63 )
63 )
64 if self.ui.configbool('format', 'parentdelta', False):
64 if self.ui.configbool('format', 'parentdelta', False):
65 requirements.append("parentdelta")
65 requirements.append("parentdelta")
66 else:
66 else:
67 raise error.RepoError(_("repository %s not found") % path)
67 raise error.RepoError(_("repository %s not found") % path)
68 elif create:
68 elif create:
69 raise error.RepoError(_("repository %s already exists") % path)
69 raise error.RepoError(_("repository %s already exists") % path)
70 else:
70 else:
71 # find requirements
71 # find requirements
72 requirements = set()
72 requirements = set()
73 try:
73 try:
74 requirements = set(self.opener("requires").read().splitlines())
74 requirements = set(self.opener("requires").read().splitlines())
75 except IOError, inst:
75 except IOError, inst:
76 if inst.errno != errno.ENOENT:
76 if inst.errno != errno.ENOENT:
77 raise
77 raise
78 for r in requirements - self.supported:
78 for r in requirements - self.supported:
79 raise error.RequirementError(
79 raise error.RequirementError(
80 _("requirement '%s' not supported") % r)
80 _("requirement '%s' not supported") % r)
81
81
82 self.sharedpath = self.path
82 self.sharedpath = self.path
83 try:
83 try:
84 s = os.path.realpath(self.opener("sharedpath").read())
84 s = os.path.realpath(self.opener("sharedpath").read())
85 if not os.path.exists(s):
85 if not os.path.exists(s):
86 raise error.RepoError(
86 raise error.RepoError(
87 _('.hg/sharedpath points to nonexistent directory %s') % s)
87 _('.hg/sharedpath points to nonexistent directory %s') % s)
88 self.sharedpath = s
88 self.sharedpath = s
89 except IOError, inst:
89 except IOError, inst:
90 if inst.errno != errno.ENOENT:
90 if inst.errno != errno.ENOENT:
91 raise
91 raise
92
92
93 self.store = store.store(requirements, self.sharedpath, util.opener)
93 self.store = store.store(requirements, self.sharedpath, util.opener)
94 self.spath = self.store.path
94 self.spath = self.store.path
95 self.sopener = self.store.opener
95 self.sopener = self.store.opener
96 self.sjoin = self.store.join
96 self.sjoin = self.store.join
97 self.opener.createmode = self.store.createmode
97 self.opener.createmode = self.store.createmode
98 self._applyrequirements(requirements)
98 self._applyrequirements(requirements)
99 if create:
99 if create:
100 self._writerequirements()
100 self._writerequirements()
101
101
102 # These two define the set of tags for this repository. _tags
102 # These two define the set of tags for this repository. _tags
103 # maps tag name to node; _tagtypes maps tag name to 'global' or
103 # maps tag name to node; _tagtypes maps tag name to 'global' or
104 # 'local'. (Global tags are defined by .hgtags across all
104 # 'local'. (Global tags are defined by .hgtags across all
105 # heads, and local tags are defined in .hg/localtags.) They
105 # heads, and local tags are defined in .hg/localtags.) They
106 # constitute the in-memory cache of tags.
106 # constitute the in-memory cache of tags.
107 self._tags = None
107 self._tags = None
108 self._tagtypes = None
108 self._tagtypes = None
109
109
110 self._branchcache = None
110 self._branchcache = None
111 self._branchcachetip = None
111 self._branchcachetip = None
112 self.nodetagscache = None
112 self.nodetagscache = None
113 self.filterpats = {}
113 self.filterpats = {}
114 self._datafilters = {}
114 self._datafilters = {}
115 self._transref = self._lockref = self._wlockref = None
115 self._transref = self._lockref = self._wlockref = None
116
116
117 def _applyrequirements(self, requirements):
117 def _applyrequirements(self, requirements):
118 self.requirements = requirements
118 self.requirements = requirements
119 self.sopener.options = {}
119 self.sopener.options = {}
120 if 'parentdelta' in requirements:
120 if 'parentdelta' in requirements:
121 self.sopener.options['parentdelta'] = 1
121 self.sopener.options['parentdelta'] = 1
122
122
123 def _writerequirements(self):
123 def _writerequirements(self):
124 reqfile = self.opener("requires", "w")
124 reqfile = self.opener("requires", "w")
125 for r in self.requirements:
125 for r in self.requirements:
126 reqfile.write("%s\n" % r)
126 reqfile.write("%s\n" % r)
127 reqfile.close()
127 reqfile.close()
128
128
129 def _checknested(self, path):
129 def _checknested(self, path):
130 """Determine if path is a legal nested repository."""
130 """Determine if path is a legal nested repository."""
131 if not path.startswith(self.root):
131 if not path.startswith(self.root):
132 return False
132 return False
133 subpath = path[len(self.root) + 1:]
133 subpath = path[len(self.root) + 1:]
134
134
135 # XXX: Checking against the current working copy is wrong in
135 # XXX: Checking against the current working copy is wrong in
136 # the sense that it can reject things like
136 # the sense that it can reject things like
137 #
137 #
138 # $ hg cat -r 10 sub/x.txt
138 # $ hg cat -r 10 sub/x.txt
139 #
139 #
140 # if sub/ is no longer a subrepository in the working copy
140 # if sub/ is no longer a subrepository in the working copy
141 # parent revision.
141 # parent revision.
142 #
142 #
143 # However, it can of course also allow things that would have
143 # However, it can of course also allow things that would have
144 # been rejected before, such as the above cat command if sub/
144 # been rejected before, such as the above cat command if sub/
145 # is a subrepository now, but was a normal directory before.
145 # is a subrepository now, but was a normal directory before.
146 # The old path auditor would have rejected by mistake since it
146 # The old path auditor would have rejected by mistake since it
147 # panics when it sees sub/.hg/.
147 # panics when it sees sub/.hg/.
148 #
148 #
149 # All in all, checking against the working copy seems sensible
149 # All in all, checking against the working copy seems sensible
150 # since we want to prevent access to nested repositories on
150 # since we want to prevent access to nested repositories on
151 # the filesystem *now*.
151 # the filesystem *now*.
152 ctx = self[None]
152 ctx = self[None]
153 parts = util.splitpath(subpath)
153 parts = util.splitpath(subpath)
154 while parts:
154 while parts:
155 prefix = os.sep.join(parts)
155 prefix = os.sep.join(parts)
156 if prefix in ctx.substate:
156 if prefix in ctx.substate:
157 if prefix == subpath:
157 if prefix == subpath:
158 return True
158 return True
159 else:
159 else:
160 sub = ctx.sub(prefix)
160 sub = ctx.sub(prefix)
161 return sub.checknested(subpath[len(prefix) + 1:])
161 return sub.checknested(subpath[len(prefix) + 1:])
162 else:
162 else:
163 parts.pop()
163 parts.pop()
164 return False
164 return False
165
165
166 @util.propertycache
166 @util.propertycache
167 def _bookmarks(self):
167 def _bookmarks(self):
168 return bookmarks.read(self)
168 return bookmarks.read(self)
169
169
170 @util.propertycache
170 @util.propertycache
171 def _bookmarkcurrent(self):
171 def _bookmarkcurrent(self):
172 return bookmarks.readcurrent(self)
172 return bookmarks.readcurrent(self)
173
173
174 @propertycache
174 @propertycache
175 def changelog(self):
175 def changelog(self):
176 c = changelog.changelog(self.sopener)
176 c = changelog.changelog(self.sopener)
177 if 'HG_PENDING' in os.environ:
177 if 'HG_PENDING' in os.environ:
178 p = os.environ['HG_PENDING']
178 p = os.environ['HG_PENDING']
179 if p.startswith(self.root):
179 if p.startswith(self.root):
180 c.readpending('00changelog.i.a')
180 c.readpending('00changelog.i.a')
181 self.sopener.options['defversion'] = c.version
181 self.sopener.options['defversion'] = c.version
182 return c
182 return c
183
183
184 @propertycache
184 @propertycache
185 def manifest(self):
185 def manifest(self):
186 return manifest.manifest(self.sopener)
186 return manifest.manifest(self.sopener)
187
187
188 @propertycache
188 @propertycache
189 def dirstate(self):
189 def dirstate(self):
190 warned = [0]
190 warned = [0]
191 def validate(node):
191 def validate(node):
192 try:
192 try:
193 r = self.changelog.rev(node)
193 r = self.changelog.rev(node)
194 return node
194 return node
195 except error.LookupError:
195 except error.LookupError:
196 if not warned[0]:
196 if not warned[0]:
197 warned[0] = True
197 warned[0] = True
198 self.ui.warn(_("warning: ignoring unknown"
198 self.ui.warn(_("warning: ignoring unknown"
199 " working parent %s!\n") % short(node))
199 " working parent %s!\n") % short(node))
200 return nullid
200 return nullid
201
201
202 return dirstate.dirstate(self.opener, self.ui, self.root, validate)
202 return dirstate.dirstate(self.opener, self.ui, self.root, validate)
203
203
204 def __getitem__(self, changeid):
204 def __getitem__(self, changeid):
205 if changeid is None:
205 if changeid is None:
206 return context.workingctx(self)
206 return context.workingctx(self)
207 return context.changectx(self, changeid)
207 return context.changectx(self, changeid)
208
208
209 def __contains__(self, changeid):
209 def __contains__(self, changeid):
210 try:
210 try:
211 return bool(self.lookup(changeid))
211 return bool(self.lookup(changeid))
212 except error.RepoLookupError:
212 except error.RepoLookupError:
213 return False
213 return False
214
214
215 def __nonzero__(self):
215 def __nonzero__(self):
216 return True
216 return True
217
217
218 def __len__(self):
218 def __len__(self):
219 return len(self.changelog)
219 return len(self.changelog)
220
220
221 def __iter__(self):
221 def __iter__(self):
222 for i in xrange(len(self)):
222 for i in xrange(len(self)):
223 yield i
223 yield i
224
224
225 def url(self):
225 def url(self):
226 return 'file:' + self.root
226 return 'file:' + self.root
227
227
228 def hook(self, name, throw=False, **args):
228 def hook(self, name, throw=False, **args):
229 return hook.hook(self.ui, self, name, throw, **args)
229 return hook.hook(self.ui, self, name, throw, **args)
230
230
231 tag_disallowed = ':\r\n'
231 tag_disallowed = ':\r\n'
232
232
233 def _tag(self, names, node, message, local, user, date, extra={}):
233 def _tag(self, names, node, message, local, user, date, extra={}):
234 if isinstance(names, str):
234 if isinstance(names, str):
235 allchars = names
235 allchars = names
236 names = (names,)
236 names = (names,)
237 else:
237 else:
238 allchars = ''.join(names)
238 allchars = ''.join(names)
239 for c in self.tag_disallowed:
239 for c in self.tag_disallowed:
240 if c in allchars:
240 if c in allchars:
241 raise util.Abort(_('%r cannot be used in a tag name') % c)
241 raise util.Abort(_('%r cannot be used in a tag name') % c)
242
242
243 branches = self.branchmap()
243 branches = self.branchmap()
244 for name in names:
244 for name in names:
245 self.hook('pretag', throw=True, node=hex(node), tag=name,
245 self.hook('pretag', throw=True, node=hex(node), tag=name,
246 local=local)
246 local=local)
247 if name in branches:
247 if name in branches:
248 self.ui.warn(_("warning: tag %s conflicts with existing"
248 self.ui.warn(_("warning: tag %s conflicts with existing"
249 " branch name\n") % name)
249 " branch name\n") % name)
250
250
251 def writetags(fp, names, munge, prevtags):
251 def writetags(fp, names, munge, prevtags):
252 fp.seek(0, 2)
252 fp.seek(0, 2)
253 if prevtags and prevtags[-1] != '\n':
253 if prevtags and prevtags[-1] != '\n':
254 fp.write('\n')
254 fp.write('\n')
255 for name in names:
255 for name in names:
256 m = munge and munge(name) or name
256 m = munge and munge(name) or name
257 if self._tagtypes and name in self._tagtypes:
257 if self._tagtypes and name in self._tagtypes:
258 old = self._tags.get(name, nullid)
258 old = self._tags.get(name, nullid)
259 fp.write('%s %s\n' % (hex(old), m))
259 fp.write('%s %s\n' % (hex(old), m))
260 fp.write('%s %s\n' % (hex(node), m))
260 fp.write('%s %s\n' % (hex(node), m))
261 fp.close()
261 fp.close()
262
262
263 prevtags = ''
263 prevtags = ''
264 if local:
264 if local:
265 try:
265 try:
266 fp = self.opener('localtags', 'r+')
266 fp = self.opener('localtags', 'r+')
267 except IOError:
267 except IOError:
268 fp = self.opener('localtags', 'a')
268 fp = self.opener('localtags', 'a')
269 else:
269 else:
270 prevtags = fp.read()
270 prevtags = fp.read()
271
271
272 # local tags are stored in the current charset
272 # local tags are stored in the current charset
273 writetags(fp, names, None, prevtags)
273 writetags(fp, names, None, prevtags)
274 for name in names:
274 for name in names:
275 self.hook('tag', node=hex(node), tag=name, local=local)
275 self.hook('tag', node=hex(node), tag=name, local=local)
276 return
276 return
277
277
278 try:
278 try:
279 fp = self.wfile('.hgtags', 'rb+')
279 fp = self.wfile('.hgtags', 'rb+')
280 except IOError:
280 except IOError:
281 fp = self.wfile('.hgtags', 'ab')
281 fp = self.wfile('.hgtags', 'ab')
282 else:
282 else:
283 prevtags = fp.read()
283 prevtags = fp.read()
284
284
285 # committed tags are stored in UTF-8
285 # committed tags are stored in UTF-8
286 writetags(fp, names, encoding.fromlocal, prevtags)
286 writetags(fp, names, encoding.fromlocal, prevtags)
287
287
288 fp.close()
288 fp.close()
289
289
290 if '.hgtags' not in self.dirstate:
290 if '.hgtags' not in self.dirstate:
291 self[None].add(['.hgtags'])
291 self[None].add(['.hgtags'])
292
292
293 m = matchmod.exact(self.root, '', ['.hgtags'])
293 m = matchmod.exact(self.root, '', ['.hgtags'])
294 tagnode = self.commit(message, user, date, extra=extra, match=m)
294 tagnode = self.commit(message, user, date, extra=extra, match=m)
295
295
296 for name in names:
296 for name in names:
297 self.hook('tag', node=hex(node), tag=name, local=local)
297 self.hook('tag', node=hex(node), tag=name, local=local)
298
298
299 return tagnode
299 return tagnode
300
300
301 def tag(self, names, node, message, local, user, date):
301 def tag(self, names, node, message, local, user, date):
302 '''tag a revision with one or more symbolic names.
302 '''tag a revision with one or more symbolic names.
303
303
304 names is a list of strings or, when adding a single tag, names may be a
304 names is a list of strings or, when adding a single tag, names may be a
305 string.
305 string.
306
306
307 if local is True, the tags are stored in a per-repository file.
307 if local is True, the tags are stored in a per-repository file.
308 otherwise, they are stored in the .hgtags file, and a new
308 otherwise, they are stored in the .hgtags file, and a new
309 changeset is committed with the change.
309 changeset is committed with the change.
310
310
311 keyword arguments:
311 keyword arguments:
312
312
313 local: whether to store tags in non-version-controlled file
313 local: whether to store tags in non-version-controlled file
314 (default False)
314 (default False)
315
315
316 message: commit message to use if committing
316 message: commit message to use if committing
317
317
318 user: name of user to use if committing
318 user: name of user to use if committing
319
319
320 date: date tuple to use if committing'''
320 date: date tuple to use if committing'''
321
321
322 if not local:
322 if not local:
323 for x in self.status()[:5]:
323 for x in self.status()[:5]:
324 if '.hgtags' in x:
324 if '.hgtags' in x:
325 raise util.Abort(_('working copy of .hgtags is changed '
325 raise util.Abort(_('working copy of .hgtags is changed '
326 '(please commit .hgtags manually)'))
326 '(please commit .hgtags manually)'))
327
327
328 self.tags() # instantiate the cache
328 self.tags() # instantiate the cache
329 self._tag(names, node, message, local, user, date)
329 self._tag(names, node, message, local, user, date)
330
330
331 def tags(self):
331 def tags(self):
332 '''return a mapping of tag to node'''
332 '''return a mapping of tag to node'''
333 if self._tags is None:
333 if self._tags is None:
334 (self._tags, self._tagtypes) = self._findtags()
334 (self._tags, self._tagtypes) = self._findtags()
335
335
336 return self._tags
336 return self._tags
337
337
338 def _findtags(self):
338 def _findtags(self):
339 '''Do the hard work of finding tags. Return a pair of dicts
339 '''Do the hard work of finding tags. Return a pair of dicts
340 (tags, tagtypes) where tags maps tag name to node, and tagtypes
340 (tags, tagtypes) where tags maps tag name to node, and tagtypes
341 maps tag name to a string like \'global\' or \'local\'.
341 maps tag name to a string like \'global\' or \'local\'.
342 Subclasses or extensions are free to add their own tags, but
342 Subclasses or extensions are free to add their own tags, but
343 should be aware that the returned dicts will be retained for the
343 should be aware that the returned dicts will be retained for the
344 duration of the localrepo object.'''
344 duration of the localrepo object.'''
345
345
346 # XXX what tagtype should subclasses/extensions use? Currently
346 # XXX what tagtype should subclasses/extensions use? Currently
347 # mq and bookmarks add tags, but do not set the tagtype at all.
347 # mq and bookmarks add tags, but do not set the tagtype at all.
348 # Should each extension invent its own tag type? Should there
348 # Should each extension invent its own tag type? Should there
349 # be one tagtype for all such "virtual" tags? Or is the status
349 # be one tagtype for all such "virtual" tags? Or is the status
350 # quo fine?
350 # quo fine?
351
351
352 alltags = {} # map tag name to (node, hist)
352 alltags = {} # map tag name to (node, hist)
353 tagtypes = {}
353 tagtypes = {}
354
354
355 tagsmod.findglobaltags(self.ui, self, alltags, tagtypes)
355 tagsmod.findglobaltags(self.ui, self, alltags, tagtypes)
356 tagsmod.readlocaltags(self.ui, self, alltags, tagtypes)
356 tagsmod.readlocaltags(self.ui, self, alltags, tagtypes)
357
357
358 # Build the return dicts. Have to re-encode tag names because
358 # Build the return dicts. Have to re-encode tag names because
359 # the tags module always uses UTF-8 (in order not to lose info
359 # the tags module always uses UTF-8 (in order not to lose info
360 # writing to the cache), but the rest of Mercurial wants them in
360 # writing to the cache), but the rest of Mercurial wants them in
361 # local encoding.
361 # local encoding.
362 tags = {}
362 tags = {}
363 for (name, (node, hist)) in alltags.iteritems():
363 for (name, (node, hist)) in alltags.iteritems():
364 if node != nullid:
364 if node != nullid:
365 tags[encoding.tolocal(name)] = node
365 tags[encoding.tolocal(name)] = node
366 tags['tip'] = self.changelog.tip()
366 tags['tip'] = self.changelog.tip()
367 tagtypes = dict([(encoding.tolocal(name), value)
367 tagtypes = dict([(encoding.tolocal(name), value)
368 for (name, value) in tagtypes.iteritems()])
368 for (name, value) in tagtypes.iteritems()])
369 return (tags, tagtypes)
369 return (tags, tagtypes)
370
370
371 def tagtype(self, tagname):
371 def tagtype(self, tagname):
372 '''
372 '''
373 return the type of the given tag. result can be:
373 return the type of the given tag. result can be:
374
374
375 'local' : a local tag
375 'local' : a local tag
376 'global' : a global tag
376 'global' : a global tag
377 None : tag does not exist
377 None : tag does not exist
378 '''
378 '''
379
379
380 self.tags()
380 self.tags()
381
381
382 return self._tagtypes.get(tagname)
382 return self._tagtypes.get(tagname)
383
383
384 def tagslist(self):
384 def tagslist(self):
385 '''return a list of tags ordered by revision'''
385 '''return a list of tags ordered by revision'''
386 l = []
386 l = []
387 for t, n in self.tags().iteritems():
387 for t, n in self.tags().iteritems():
388 try:
388 try:
389 r = self.changelog.rev(n)
389 r = self.changelog.rev(n)
390 except:
390 except:
391 r = -2 # sort to the beginning of the list if unknown
391 r = -2 # sort to the beginning of the list if unknown
392 l.append((r, t, n))
392 l.append((r, t, n))
393 return [(t, n) for r, t, n in sorted(l)]
393 return [(t, n) for r, t, n in sorted(l)]
394
394
395 def nodetags(self, node):
395 def nodetags(self, node):
396 '''return the tags associated with a node'''
396 '''return the tags associated with a node'''
397 if not self.nodetagscache:
397 if not self.nodetagscache:
398 self.nodetagscache = {}
398 self.nodetagscache = {}
399 for t, n in self.tags().iteritems():
399 for t, n in self.tags().iteritems():
400 self.nodetagscache.setdefault(n, []).append(t)
400 self.nodetagscache.setdefault(n, []).append(t)
401 for tags in self.nodetagscache.itervalues():
401 for tags in self.nodetagscache.itervalues():
402 tags.sort()
402 tags.sort()
403 return self.nodetagscache.get(node, [])
403 return self.nodetagscache.get(node, [])
404
404
405 def nodebookmarks(self, node):
405 def nodebookmarks(self, node):
406 marks = []
406 marks = []
407 for bookmark, n in self._bookmarks.iteritems():
407 for bookmark, n in self._bookmarks.iteritems():
408 if n == node:
408 if n == node:
409 marks.append(bookmark)
409 marks.append(bookmark)
410 return sorted(marks)
410 return sorted(marks)
411
411
412 def _branchtags(self, partial, lrev):
412 def _branchtags(self, partial, lrev):
413 # TODO: rename this function?
413 # TODO: rename this function?
414 tiprev = len(self) - 1
414 tiprev = len(self) - 1
415 if lrev != tiprev:
415 if lrev != tiprev:
416 ctxgen = (self[r] for r in xrange(lrev + 1, tiprev + 1))
416 ctxgen = (self[r] for r in xrange(lrev + 1, tiprev + 1))
417 self._updatebranchcache(partial, ctxgen)
417 self._updatebranchcache(partial, ctxgen)
418 self._writebranchcache(partial, self.changelog.tip(), tiprev)
418 self._writebranchcache(partial, self.changelog.tip(), tiprev)
419
419
420 return partial
420 return partial
421
421
422 def updatebranchcache(self):
422 def updatebranchcache(self):
423 tip = self.changelog.tip()
423 tip = self.changelog.tip()
424 if self._branchcache is not None and self._branchcachetip == tip:
424 if self._branchcache is not None and self._branchcachetip == tip:
425 return self._branchcache
425 return self._branchcache
426
426
427 oldtip = self._branchcachetip
427 oldtip = self._branchcachetip
428 self._branchcachetip = tip
428 self._branchcachetip = tip
429 if oldtip is None or oldtip not in self.changelog.nodemap:
429 if oldtip is None or oldtip not in self.changelog.nodemap:
430 partial, last, lrev = self._readbranchcache()
430 partial, last, lrev = self._readbranchcache()
431 else:
431 else:
432 lrev = self.changelog.rev(oldtip)
432 lrev = self.changelog.rev(oldtip)
433 partial = self._branchcache
433 partial = self._branchcache
434
434
435 self._branchtags(partial, lrev)
435 self._branchtags(partial, lrev)
436 # this private cache holds all heads (not just tips)
436 # this private cache holds all heads (not just tips)
437 self._branchcache = partial
437 self._branchcache = partial
438
438
439 def branchmap(self):
439 def branchmap(self):
440 '''returns a dictionary {branch: [branchheads]}'''
440 '''returns a dictionary {branch: [branchheads]}'''
441 self.updatebranchcache()
441 self.updatebranchcache()
442 return self._branchcache
442 return self._branchcache
443
443
444 def branchtags(self):
444 def branchtags(self):
445 '''return a dict where branch names map to the tipmost head of
445 '''return a dict where branch names map to the tipmost head of
446 the branch, open heads come before closed'''
446 the branch, open heads come before closed'''
447 bt = {}
447 bt = {}
448 for bn, heads in self.branchmap().iteritems():
448 for bn, heads in self.branchmap().iteritems():
449 tip = heads[-1]
449 tip = heads[-1]
450 for h in reversed(heads):
450 for h in reversed(heads):
451 if 'close' not in self.changelog.read(h)[5]:
451 if 'close' not in self.changelog.read(h)[5]:
452 tip = h
452 tip = h
453 break
453 break
454 bt[bn] = tip
454 bt[bn] = tip
455 return bt
455 return bt
456
456
457 def _readbranchcache(self):
457 def _readbranchcache(self):
458 partial = {}
458 partial = {}
459 try:
459 try:
460 f = self.opener("cache/branchheads")
460 f = self.opener("cache/branchheads")
461 lines = f.read().split('\n')
461 lines = f.read().split('\n')
462 f.close()
462 f.close()
463 except (IOError, OSError):
463 except (IOError, OSError):
464 return {}, nullid, nullrev
464 return {}, nullid, nullrev
465
465
466 try:
466 try:
467 last, lrev = lines.pop(0).split(" ", 1)
467 last, lrev = lines.pop(0).split(" ", 1)
468 last, lrev = bin(last), int(lrev)
468 last, lrev = bin(last), int(lrev)
469 if lrev >= len(self) or self[lrev].node() != last:
469 if lrev >= len(self) or self[lrev].node() != last:
470 # invalidate the cache
470 # invalidate the cache
471 raise ValueError('invalidating branch cache (tip differs)')
471 raise ValueError('invalidating branch cache (tip differs)')
472 for l in lines:
472 for l in lines:
473 if not l:
473 if not l:
474 continue
474 continue
475 node, label = l.split(" ", 1)
475 node, label = l.split(" ", 1)
476 label = encoding.tolocal(label.strip())
476 label = encoding.tolocal(label.strip())
477 partial.setdefault(label, []).append(bin(node))
477 partial.setdefault(label, []).append(bin(node))
478 except KeyboardInterrupt:
478 except KeyboardInterrupt:
479 raise
479 raise
480 except Exception, inst:
480 except Exception, inst:
481 if self.ui.debugflag:
481 if self.ui.debugflag:
482 self.ui.warn(str(inst), '\n')
482 self.ui.warn(str(inst), '\n')
483 partial, last, lrev = {}, nullid, nullrev
483 partial, last, lrev = {}, nullid, nullrev
484 return partial, last, lrev
484 return partial, last, lrev
485
485
486 def _writebranchcache(self, branches, tip, tiprev):
486 def _writebranchcache(self, branches, tip, tiprev):
487 try:
487 try:
488 f = self.opener("cache/branchheads", "w", atomictemp=True)
488 f = self.opener("cache/branchheads", "w", atomictemp=True)
489 f.write("%s %s\n" % (hex(tip), tiprev))
489 f.write("%s %s\n" % (hex(tip), tiprev))
490 for label, nodes in branches.iteritems():
490 for label, nodes in branches.iteritems():
491 for node in nodes:
491 for node in nodes:
492 f.write("%s %s\n" % (hex(node), encoding.fromlocal(label)))
492 f.write("%s %s\n" % (hex(node), encoding.fromlocal(label)))
493 f.rename()
493 f.rename()
494 except (IOError, OSError):
494 except (IOError, OSError):
495 pass
495 pass
496
496
497 def _updatebranchcache(self, partial, ctxgen):
497 def _updatebranchcache(self, partial, ctxgen):
498 # collect new branch entries
498 # collect new branch entries
499 newbranches = {}
499 newbranches = {}
500 for c in ctxgen:
500 for c in ctxgen:
501 newbranches.setdefault(c.branch(), []).append(c.node())
501 newbranches.setdefault(c.branch(), []).append(c.node())
502 # if older branchheads are reachable from new ones, they aren't
502 # if older branchheads are reachable from new ones, they aren't
503 # really branchheads. Note checking parents is insufficient:
503 # really branchheads. Note checking parents is insufficient:
504 # 1 (branch a) -> 2 (branch b) -> 3 (branch a)
504 # 1 (branch a) -> 2 (branch b) -> 3 (branch a)
505 for branch, newnodes in newbranches.iteritems():
505 for branch, newnodes in newbranches.iteritems():
506 bheads = partial.setdefault(branch, [])
506 bheads = partial.setdefault(branch, [])
507 bheads.extend(newnodes)
507 bheads.extend(newnodes)
508 if len(bheads) <= 1:
508 if len(bheads) <= 1:
509 continue
509 continue
510 # starting from tip means fewer passes over reachable
510 # starting from tip means fewer passes over reachable
511 while newnodes:
511 while newnodes:
512 latest = newnodes.pop()
512 latest = newnodes.pop()
513 if latest not in bheads:
513 if latest not in bheads:
514 continue
514 continue
515 minbhrev = self[min([self[bh].rev() for bh in bheads])].node()
515 minbhrev = self[min([self[bh].rev() for bh in bheads])].node()
516 reachable = self.changelog.reachable(latest, minbhrev)
516 reachable = self.changelog.reachable(latest, minbhrev)
517 reachable.remove(latest)
517 reachable.remove(latest)
518 bheads = [b for b in bheads if b not in reachable]
518 bheads = [b for b in bheads if b not in reachable]
519 partial[branch] = bheads
519 partial[branch] = bheads
520
520
521 def lookup(self, key):
521 def lookup(self, key):
522 if isinstance(key, int):
522 if isinstance(key, int):
523 return self.changelog.node(key)
523 return self.changelog.node(key)
524 elif key == '.':
524 elif key == '.':
525 return self.dirstate.parents()[0]
525 return self.dirstate.parents()[0]
526 elif key == 'null':
526 elif key == 'null':
527 return nullid
527 return nullid
528 elif key == 'tip':
528 elif key == 'tip':
529 return self.changelog.tip()
529 return self.changelog.tip()
530 n = self.changelog._match(key)
530 n = self.changelog._match(key)
531 if n:
531 if n:
532 return n
532 return n
533 if key in self._bookmarks:
533 if key in self._bookmarks:
534 return self._bookmarks[key]
534 return self._bookmarks[key]
535 if key in self.tags():
535 if key in self.tags():
536 return self.tags()[key]
536 return self.tags()[key]
537 if key in self.branchtags():
537 if key in self.branchtags():
538 return self.branchtags()[key]
538 return self.branchtags()[key]
539 n = self.changelog._partialmatch(key)
539 n = self.changelog._partialmatch(key)
540 if n:
540 if n:
541 return n
541 return n
542
542
543 # can't find key, check if it might have come from damaged dirstate
543 # can't find key, check if it might have come from damaged dirstate
544 if key in self.dirstate.parents():
544 if key in self.dirstate.parents():
545 raise error.Abort(_("working directory has unknown parent '%s'!")
545 raise error.Abort(_("working directory has unknown parent '%s'!")
546 % short(key))
546 % short(key))
547 try:
547 try:
548 if len(key) == 20:
548 if len(key) == 20:
549 key = hex(key)
549 key = hex(key)
550 except:
550 except:
551 pass
551 pass
552 raise error.RepoLookupError(_("unknown revision '%s'") % key)
552 raise error.RepoLookupError(_("unknown revision '%s'") % key)
553
553
554 def lookupbranch(self, key, remote=None):
554 def lookupbranch(self, key, remote=None):
555 repo = remote or self
555 repo = remote or self
556 if key in repo.branchmap():
556 if key in repo.branchmap():
557 return key
557 return key
558
558
559 repo = (remote and remote.local()) and remote or self
559 repo = (remote and remote.local()) and remote or self
560 return repo[key].branch()
560 return repo[key].branch()
561
561
562 def known(self, nodes):
562 def known(self, nodes):
563 nm = self.changelog.nodemap
563 nm = self.changelog.nodemap
564 return [(n in nm) for n in nodes]
564 return [(n in nm) for n in nodes]
565
565
566 def local(self):
566 def local(self):
567 return True
567 return True
568
568
569 def join(self, f):
569 def join(self, f):
570 return os.path.join(self.path, f)
570 return os.path.join(self.path, f)
571
571
572 def wjoin(self, f):
572 def wjoin(self, f):
573 return os.path.join(self.root, f)
573 return os.path.join(self.root, f)
574
574
575 def file(self, f):
575 def file(self, f):
576 if f[0] == '/':
576 if f[0] == '/':
577 f = f[1:]
577 f = f[1:]
578 return filelog.filelog(self.sopener, f)
578 return filelog.filelog(self.sopener, f)
579
579
580 def changectx(self, changeid):
580 def changectx(self, changeid):
581 return self[changeid]
581 return self[changeid]
582
582
583 def parents(self, changeid=None):
583 def parents(self, changeid=None):
584 '''get list of changectxs for parents of changeid'''
584 '''get list of changectxs for parents of changeid'''
585 return self[changeid].parents()
585 return self[changeid].parents()
586
586
587 def filectx(self, path, changeid=None, fileid=None):
587 def filectx(self, path, changeid=None, fileid=None):
588 """changeid can be a changeset revision, node, or tag.
588 """changeid can be a changeset revision, node, or tag.
589 fileid can be a file revision or node."""
589 fileid can be a file revision or node."""
590 return context.filectx(self, path, changeid, fileid)
590 return context.filectx(self, path, changeid, fileid)
591
591
592 def getcwd(self):
592 def getcwd(self):
593 return self.dirstate.getcwd()
593 return self.dirstate.getcwd()
594
594
595 def pathto(self, f, cwd=None):
595 def pathto(self, f, cwd=None):
596 return self.dirstate.pathto(f, cwd)
596 return self.dirstate.pathto(f, cwd)
597
597
598 def wfile(self, f, mode='r'):
598 def wfile(self, f, mode='r'):
599 return self.wopener(f, mode)
599 return self.wopener(f, mode)
600
600
601 def _link(self, f):
601 def _link(self, f):
602 return os.path.islink(self.wjoin(f))
602 return os.path.islink(self.wjoin(f))
603
603
604 def _loadfilter(self, filter):
604 def _loadfilter(self, filter):
605 if filter not in self.filterpats:
605 if filter not in self.filterpats:
606 l = []
606 l = []
607 for pat, cmd in self.ui.configitems(filter):
607 for pat, cmd in self.ui.configitems(filter):
608 if cmd == '!':
608 if cmd == '!':
609 continue
609 continue
610 mf = matchmod.match(self.root, '', [pat])
610 mf = matchmod.match(self.root, '', [pat])
611 fn = None
611 fn = None
612 params = cmd
612 params = cmd
613 for name, filterfn in self._datafilters.iteritems():
613 for name, filterfn in self._datafilters.iteritems():
614 if cmd.startswith(name):
614 if cmd.startswith(name):
615 fn = filterfn
615 fn = filterfn
616 params = cmd[len(name):].lstrip()
616 params = cmd[len(name):].lstrip()
617 break
617 break
618 if not fn:
618 if not fn:
619 fn = lambda s, c, **kwargs: util.filter(s, c)
619 fn = lambda s, c, **kwargs: util.filter(s, c)
620 # Wrap old filters not supporting keyword arguments
620 # Wrap old filters not supporting keyword arguments
621 if not inspect.getargspec(fn)[2]:
621 if not inspect.getargspec(fn)[2]:
622 oldfn = fn
622 oldfn = fn
623 fn = lambda s, c, **kwargs: oldfn(s, c)
623 fn = lambda s, c, **kwargs: oldfn(s, c)
624 l.append((mf, fn, params))
624 l.append((mf, fn, params))
625 self.filterpats[filter] = l
625 self.filterpats[filter] = l
626 return self.filterpats[filter]
626 return self.filterpats[filter]
627
627
628 def _filter(self, filterpats, filename, data):
628 def _filter(self, filterpats, filename, data):
629 for mf, fn, cmd in filterpats:
629 for mf, fn, cmd in filterpats:
630 if mf(filename):
630 if mf(filename):
631 self.ui.debug("filtering %s through %s\n" % (filename, cmd))
631 self.ui.debug("filtering %s through %s\n" % (filename, cmd))
632 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
632 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
633 break
633 break
634
634
635 return data
635 return data
636
636
637 @propertycache
637 @propertycache
638 def _encodefilterpats(self):
638 def _encodefilterpats(self):
639 return self._loadfilter('encode')
639 return self._loadfilter('encode')
640
640
641 @propertycache
641 @propertycache
642 def _decodefilterpats(self):
642 def _decodefilterpats(self):
643 return self._loadfilter('decode')
643 return self._loadfilter('decode')
644
644
645 def adddatafilter(self, name, filter):
645 def adddatafilter(self, name, filter):
646 self._datafilters[name] = filter
646 self._datafilters[name] = filter
647
647
648 def wread(self, filename):
648 def wread(self, filename):
649 if self._link(filename):
649 if self._link(filename):
650 data = os.readlink(self.wjoin(filename))
650 data = os.readlink(self.wjoin(filename))
651 else:
651 else:
652 data = self.wopener(filename, 'r').read()
652 data = self.wopener(filename, 'r').read()
653 return self._filter(self._encodefilterpats, filename, data)
653 return self._filter(self._encodefilterpats, filename, data)
654
654
655 def wwrite(self, filename, data, flags):
655 def wwrite(self, filename, data, flags):
656 data = self._filter(self._decodefilterpats, filename, data)
656 data = self._filter(self._decodefilterpats, filename, data)
657 if 'l' in flags:
657 if 'l' in flags:
658 self.wopener.symlink(data, filename)
658 self.wopener.symlink(data, filename)
659 else:
659 else:
660 self.wopener(filename, 'w').write(data)
660 self.wopener(filename, 'w').write(data)
661 if 'x' in flags:
661 if 'x' in flags:
662 util.set_flags(self.wjoin(filename), False, True)
662 util.set_flags(self.wjoin(filename), False, True)
663
663
664 def wwritedata(self, filename, data):
664 def wwritedata(self, filename, data):
665 return self._filter(self._decodefilterpats, filename, data)
665 return self._filter(self._decodefilterpats, filename, data)
666
666
667 def transaction(self, desc):
667 def transaction(self, desc):
668 tr = self._transref and self._transref() or None
668 tr = self._transref and self._transref() or None
669 if tr and tr.running():
669 if tr and tr.running():
670 return tr.nest()
670 return tr.nest()
671
671
672 # abort here if the journal already exists
672 # abort here if the journal already exists
673 if os.path.exists(self.sjoin("journal")):
673 if os.path.exists(self.sjoin("journal")):
674 raise error.RepoError(
674 raise error.RepoError(
675 _("abandoned transaction found - run hg recover"))
675 _("abandoned transaction found - run hg recover"))
676
676
677 # save dirstate for rollback
677 # save dirstate for rollback
678 try:
678 try:
679 ds = self.opener("dirstate").read()
679 ds = self.opener("dirstate").read()
680 except IOError:
680 except IOError:
681 ds = ""
681 ds = ""
682 self.opener("journal.dirstate", "w").write(ds)
682 self.opener("journal.dirstate", "w").write(ds)
683 self.opener("journal.branch", "w").write(
683 self.opener("journal.branch", "w").write(
684 encoding.fromlocal(self.dirstate.branch()))
684 encoding.fromlocal(self.dirstate.branch()))
685 self.opener("journal.desc", "w").write("%d\n%s\n" % (len(self), desc))
685 self.opener("journal.desc", "w").write("%d\n%s\n" % (len(self), desc))
686
686
687 renames = [(self.sjoin("journal"), self.sjoin("undo")),
687 renames = [(self.sjoin("journal"), self.sjoin("undo")),
688 (self.join("journal.dirstate"), self.join("undo.dirstate")),
688 (self.join("journal.dirstate"), self.join("undo.dirstate")),
689 (self.join("journal.branch"), self.join("undo.branch")),
689 (self.join("journal.branch"), self.join("undo.branch")),
690 (self.join("journal.desc"), self.join("undo.desc"))]
690 (self.join("journal.desc"), self.join("undo.desc"))]
691 tr = transaction.transaction(self.ui.warn, self.sopener,
691 tr = transaction.transaction(self.ui.warn, self.sopener,
692 self.sjoin("journal"),
692 self.sjoin("journal"),
693 aftertrans(renames),
693 aftertrans(renames),
694 self.store.createmode)
694 self.store.createmode)
695 self._transref = weakref.ref(tr)
695 self._transref = weakref.ref(tr)
696 return tr
696 return tr
697
697
698 def recover(self):
698 def recover(self):
699 lock = self.lock()
699 lock = self.lock()
700 try:
700 try:
701 if os.path.exists(self.sjoin("journal")):
701 if os.path.exists(self.sjoin("journal")):
702 self.ui.status(_("rolling back interrupted transaction\n"))
702 self.ui.status(_("rolling back interrupted transaction\n"))
703 transaction.rollback(self.sopener, self.sjoin("journal"),
703 transaction.rollback(self.sopener, self.sjoin("journal"),
704 self.ui.warn)
704 self.ui.warn)
705 self.invalidate()
705 self.invalidate()
706 return True
706 return True
707 else:
707 else:
708 self.ui.warn(_("no interrupted transaction available\n"))
708 self.ui.warn(_("no interrupted transaction available\n"))
709 return False
709 return False
710 finally:
710 finally:
711 lock.release()
711 lock.release()
712
712
713 def rollback(self, dryrun=False):
713 def rollback(self, dryrun=False):
714 wlock = lock = None
714 wlock = lock = None
715 try:
715 try:
716 wlock = self.wlock()
716 wlock = self.wlock()
717 lock = self.lock()
717 lock = self.lock()
718 if os.path.exists(self.sjoin("undo")):
718 if os.path.exists(self.sjoin("undo")):
719 try:
719 try:
720 args = self.opener("undo.desc", "r").read().splitlines()
720 args = self.opener("undo.desc", "r").read().splitlines()
721 if len(args) >= 3 and self.ui.verbose:
721 if len(args) >= 3 and self.ui.verbose:
722 desc = _("repository tip rolled back to revision %s"
722 desc = _("repository tip rolled back to revision %s"
723 " (undo %s: %s)\n") % (
723 " (undo %s: %s)\n") % (
724 int(args[0]) - 1, args[1], args[2])
724 int(args[0]) - 1, args[1], args[2])
725 elif len(args) >= 2:
725 elif len(args) >= 2:
726 desc = _("repository tip rolled back to revision %s"
726 desc = _("repository tip rolled back to revision %s"
727 " (undo %s)\n") % (
727 " (undo %s)\n") % (
728 int(args[0]) - 1, args[1])
728 int(args[0]) - 1, args[1])
729 except IOError:
729 except IOError:
730 desc = _("rolling back unknown transaction\n")
730 desc = _("rolling back unknown transaction\n")
731 self.ui.status(desc)
731 self.ui.status(desc)
732 if dryrun:
732 if dryrun:
733 return
733 return
734 transaction.rollback(self.sopener, self.sjoin("undo"),
734 transaction.rollback(self.sopener, self.sjoin("undo"),
735 self.ui.warn)
735 self.ui.warn)
736 util.rename(self.join("undo.dirstate"), self.join("dirstate"))
736 util.rename(self.join("undo.dirstate"), self.join("dirstate"))
737 if os.path.exists(self.join('undo.bookmarks')):
737 if os.path.exists(self.join('undo.bookmarks')):
738 util.rename(self.join('undo.bookmarks'),
738 util.rename(self.join('undo.bookmarks'),
739 self.join('bookmarks'))
739 self.join('bookmarks'))
740 try:
740 try:
741 branch = self.opener("undo.branch").read()
741 branch = self.opener("undo.branch").read()
742 self.dirstate.setbranch(branch)
742 self.dirstate.setbranch(branch)
743 except IOError:
743 except IOError:
744 self.ui.warn(_("Named branch could not be reset, "
744 self.ui.warn(_("Named branch could not be reset, "
745 "current branch still is: %s\n")
745 "current branch still is: %s\n")
746 % self.dirstate.branch())
746 % self.dirstate.branch())
747 self.invalidate()
747 self.invalidate()
748 self.dirstate.invalidate()
748 self.dirstate.invalidate()
749 self.destroyed()
749 self.destroyed()
750 parents = tuple([p.rev() for p in self.parents()])
750 parents = tuple([p.rev() for p in self.parents()])
751 if len(parents) > 1:
751 if len(parents) > 1:
752 self.ui.status(_("working directory now based on "
752 self.ui.status(_("working directory now based on "
753 "revisions %d and %d\n") % parents)
753 "revisions %d and %d\n") % parents)
754 else:
754 else:
755 self.ui.status(_("working directory now based on "
755 self.ui.status(_("working directory now based on "
756 "revision %d\n") % parents)
756 "revision %d\n") % parents)
757 else:
757 else:
758 self.ui.warn(_("no rollback information available\n"))
758 self.ui.warn(_("no rollback information available\n"))
759 return 1
759 return 1
760 finally:
760 finally:
761 release(lock, wlock)
761 release(lock, wlock)
762
762
763 def invalidatecaches(self):
763 def invalidatecaches(self):
764 self._tags = None
764 self._tags = None
765 self._tagtypes = None
765 self._tagtypes = None
766 self.nodetagscache = None
766 self.nodetagscache = None
767 self._branchcache = None # in UTF-8
767 self._branchcache = None # in UTF-8
768 self._branchcachetip = None
768 self._branchcachetip = None
769
769
770 def invalidate(self):
770 def invalidate(self):
771 for a in ("changelog", "manifest", "_bookmarks", "_bookmarkcurrent"):
771 for a in ("changelog", "manifest", "_bookmarks", "_bookmarkcurrent"):
772 if a in self.__dict__:
772 if a in self.__dict__:
773 delattr(self, a)
773 delattr(self, a)
774 self.invalidatecaches()
774 self.invalidatecaches()
775
775
776 def _lock(self, lockname, wait, releasefn, acquirefn, desc):
776 def _lock(self, lockname, wait, releasefn, acquirefn, desc):
777 try:
777 try:
778 l = lock.lock(lockname, 0, releasefn, desc=desc)
778 l = lock.lock(lockname, 0, releasefn, desc=desc)
779 except error.LockHeld, inst:
779 except error.LockHeld, inst:
780 if not wait:
780 if not wait:
781 raise
781 raise
782 self.ui.warn(_("waiting for lock on %s held by %r\n") %
782 self.ui.warn(_("waiting for lock on %s held by %r\n") %
783 (desc, inst.locker))
783 (desc, inst.locker))
784 # default to 600 seconds timeout
784 # default to 600 seconds timeout
785 l = lock.lock(lockname, int(self.ui.config("ui", "timeout", "600")),
785 l = lock.lock(lockname, int(self.ui.config("ui", "timeout", "600")),
786 releasefn, desc=desc)
786 releasefn, desc=desc)
787 if acquirefn:
787 if acquirefn:
788 acquirefn()
788 acquirefn()
789 return l
789 return l
790
790
791 def lock(self, wait=True):
791 def lock(self, wait=True):
792 '''Lock the repository store (.hg/store) and return a weak reference
792 '''Lock the repository store (.hg/store) and return a weak reference
793 to the lock. Use this before modifying the store (e.g. committing or
793 to the lock. Use this before modifying the store (e.g. committing or
794 stripping). If you are opening a transaction, get a lock as well.)'''
794 stripping). If you are opening a transaction, get a lock as well.)'''
795 l = self._lockref and self._lockref()
795 l = self._lockref and self._lockref()
796 if l is not None and l.held:
796 if l is not None and l.held:
797 l.lock()
797 l.lock()
798 return l
798 return l
799
799
800 l = self._lock(self.sjoin("lock"), wait, self.store.write,
800 l = self._lock(self.sjoin("lock"), wait, self.store.write,
801 self.invalidate, _('repository %s') % self.origroot)
801 self.invalidate, _('repository %s') % self.origroot)
802 self._lockref = weakref.ref(l)
802 self._lockref = weakref.ref(l)
803 return l
803 return l
804
804
805 def wlock(self, wait=True):
805 def wlock(self, wait=True):
806 '''Lock the non-store parts of the repository (everything under
806 '''Lock the non-store parts of the repository (everything under
807 .hg except .hg/store) and return a weak reference to the lock.
807 .hg except .hg/store) and return a weak reference to the lock.
808 Use this before modifying files in .hg.'''
808 Use this before modifying files in .hg.'''
809 l = self._wlockref and self._wlockref()
809 l = self._wlockref and self._wlockref()
810 if l is not None and l.held:
810 if l is not None and l.held:
811 l.lock()
811 l.lock()
812 return l
812 return l
813
813
814 l = self._lock(self.join("wlock"), wait, self.dirstate.write,
814 l = self._lock(self.join("wlock"), wait, self.dirstate.write,
815 self.dirstate.invalidate, _('working directory of %s') %
815 self.dirstate.invalidate, _('working directory of %s') %
816 self.origroot)
816 self.origroot)
817 self._wlockref = weakref.ref(l)
817 self._wlockref = weakref.ref(l)
818 return l
818 return l
819
819
820 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
820 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
821 """
821 """
822 commit an individual file as part of a larger transaction
822 commit an individual file as part of a larger transaction
823 """
823 """
824
824
825 fname = fctx.path()
825 fname = fctx.path()
826 text = fctx.data()
826 text = fctx.data()
827 flog = self.file(fname)
827 flog = self.file(fname)
828 fparent1 = manifest1.get(fname, nullid)
828 fparent1 = manifest1.get(fname, nullid)
829 fparent2 = fparent2o = manifest2.get(fname, nullid)
829 fparent2 = fparent2o = manifest2.get(fname, nullid)
830
830
831 meta = {}
831 meta = {}
832 copy = fctx.renamed()
832 copy = fctx.renamed()
833 if copy and copy[0] != fname:
833 if copy and copy[0] != fname:
834 # Mark the new revision of this file as a copy of another
834 # Mark the new revision of this file as a copy of another
835 # file. This copy data will effectively act as a parent
835 # file. This copy data will effectively act as a parent
836 # of this new revision. If this is a merge, the first
836 # of this new revision. If this is a merge, the first
837 # parent will be the nullid (meaning "look up the copy data")
837 # parent will be the nullid (meaning "look up the copy data")
838 # and the second one will be the other parent. For example:
838 # and the second one will be the other parent. For example:
839 #
839 #
840 # 0 --- 1 --- 3 rev1 changes file foo
840 # 0 --- 1 --- 3 rev1 changes file foo
841 # \ / rev2 renames foo to bar and changes it
841 # \ / rev2 renames foo to bar and changes it
842 # \- 2 -/ rev3 should have bar with all changes and
842 # \- 2 -/ rev3 should have bar with all changes and
843 # should record that bar descends from
843 # should record that bar descends from
844 # bar in rev2 and foo in rev1
844 # bar in rev2 and foo in rev1
845 #
845 #
846 # this allows this merge to succeed:
846 # this allows this merge to succeed:
847 #
847 #
848 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
848 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
849 # \ / merging rev3 and rev4 should use bar@rev2
849 # \ / merging rev3 and rev4 should use bar@rev2
850 # \- 2 --- 4 as the merge base
850 # \- 2 --- 4 as the merge base
851 #
851 #
852
852
853 cfname = copy[0]
853 cfname = copy[0]
854 crev = manifest1.get(cfname)
854 crev = manifest1.get(cfname)
855 newfparent = fparent2
855 newfparent = fparent2
856
856
857 if manifest2: # branch merge
857 if manifest2: # branch merge
858 if fparent2 == nullid or crev is None: # copied on remote side
858 if fparent2 == nullid or crev is None: # copied on remote side
859 if cfname in manifest2:
859 if cfname in manifest2:
860 crev = manifest2[cfname]
860 crev = manifest2[cfname]
861 newfparent = fparent1
861 newfparent = fparent1
862
862
863 # find source in nearest ancestor if we've lost track
863 # find source in nearest ancestor if we've lost track
864 if not crev:
864 if not crev:
865 self.ui.debug(" %s: searching for copy revision for %s\n" %
865 self.ui.debug(" %s: searching for copy revision for %s\n" %
866 (fname, cfname))
866 (fname, cfname))
867 for ancestor in self[None].ancestors():
867 for ancestor in self[None].ancestors():
868 if cfname in ancestor:
868 if cfname in ancestor:
869 crev = ancestor[cfname].filenode()
869 crev = ancestor[cfname].filenode()
870 break
870 break
871
871
872 if crev:
872 if crev:
873 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
873 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
874 meta["copy"] = cfname
874 meta["copy"] = cfname
875 meta["copyrev"] = hex(crev)
875 meta["copyrev"] = hex(crev)
876 fparent1, fparent2 = nullid, newfparent
876 fparent1, fparent2 = nullid, newfparent
877 else:
877 else:
878 self.ui.warn(_("warning: can't find ancestor for '%s' "
878 self.ui.warn(_("warning: can't find ancestor for '%s' "
879 "copied from '%s'!\n") % (fname, cfname))
879 "copied from '%s'!\n") % (fname, cfname))
880
880
881 elif fparent2 != nullid:
881 elif fparent2 != nullid:
882 # is one parent an ancestor of the other?
882 # is one parent an ancestor of the other?
883 fparentancestor = flog.ancestor(fparent1, fparent2)
883 fparentancestor = flog.ancestor(fparent1, fparent2)
884 if fparentancestor == fparent1:
884 if fparentancestor == fparent1:
885 fparent1, fparent2 = fparent2, nullid
885 fparent1, fparent2 = fparent2, nullid
886 elif fparentancestor == fparent2:
886 elif fparentancestor == fparent2:
887 fparent2 = nullid
887 fparent2 = nullid
888
888
889 # is the file changed?
889 # is the file changed?
890 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
890 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
891 changelist.append(fname)
891 changelist.append(fname)
892 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
892 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
893
893
894 # are just the flags changed during merge?
894 # are just the flags changed during merge?
895 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
895 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
896 changelist.append(fname)
896 changelist.append(fname)
897
897
898 return fparent1
898 return fparent1
899
899
900 def commit(self, text="", user=None, date=None, match=None, force=False,
900 def commit(self, text="", user=None, date=None, match=None, force=False,
901 editor=False, extra={}):
901 editor=False, extra={}):
902 """Add a new revision to current repository.
902 """Add a new revision to current repository.
903
903
904 Revision information is gathered from the working directory,
904 Revision information is gathered from the working directory,
905 match can be used to filter the committed files. If editor is
905 match can be used to filter the committed files. If editor is
906 supplied, it is called to get a commit message.
906 supplied, it is called to get a commit message.
907 """
907 """
908
908
909 def fail(f, msg):
909 def fail(f, msg):
910 raise util.Abort('%s: %s' % (f, msg))
910 raise util.Abort('%s: %s' % (f, msg))
911
911
912 if not match:
912 if not match:
913 match = matchmod.always(self.root, '')
913 match = matchmod.always(self.root, '')
914
914
915 if not force:
915 if not force:
916 vdirs = []
916 vdirs = []
917 match.dir = vdirs.append
917 match.dir = vdirs.append
918 match.bad = fail
918 match.bad = fail
919
919
920 wlock = self.wlock()
920 wlock = self.wlock()
921 try:
921 try:
922 wctx = self[None]
922 wctx = self[None]
923 merge = len(wctx.parents()) > 1
923 merge = len(wctx.parents()) > 1
924
924
925 if (not force and merge and match and
925 if (not force and merge and match and
926 (match.files() or match.anypats())):
926 (match.files() or match.anypats())):
927 raise util.Abort(_('cannot partially commit a merge '
927 raise util.Abort(_('cannot partially commit a merge '
928 '(do not specify files or patterns)'))
928 '(do not specify files or patterns)'))
929
929
930 changes = self.status(match=match, clean=force)
930 changes = self.status(match=match, clean=force)
931 if force:
931 if force:
932 changes[0].extend(changes[6]) # mq may commit unchanged files
932 changes[0].extend(changes[6]) # mq may commit unchanged files
933
933
934 # check subrepos
934 # check subrepos
935 subs = []
935 subs = []
936 removedsubs = set()
936 removedsubs = set()
937 for p in wctx.parents():
937 for p in wctx.parents():
938 removedsubs.update(s for s in p.substate if match(s))
938 removedsubs.update(s for s in p.substate if match(s))
939 for s in wctx.substate:
939 for s in wctx.substate:
940 removedsubs.discard(s)
940 removedsubs.discard(s)
941 if match(s) and wctx.sub(s).dirty():
941 if match(s) and wctx.sub(s).dirty():
942 subs.append(s)
942 subs.append(s)
943 if (subs or removedsubs):
943 if (subs or removedsubs):
944 if (not match('.hgsub') and
944 if (not match('.hgsub') and
945 '.hgsub' in (wctx.modified() + wctx.added())):
945 '.hgsub' in (wctx.modified() + wctx.added())):
946 raise util.Abort(_("can't commit subrepos without .hgsub"))
946 raise util.Abort(_("can't commit subrepos without .hgsub"))
947 if '.hgsubstate' not in changes[0]:
947 if '.hgsubstate' not in changes[0]:
948 changes[0].insert(0, '.hgsubstate')
948 changes[0].insert(0, '.hgsubstate')
949
949
950 if subs and not self.ui.configbool('ui', 'commitsubrepos', True):
950 if subs and not self.ui.configbool('ui', 'commitsubrepos', True):
951 changedsubs = [s for s in subs if wctx.sub(s).dirty(True)]
951 changedsubs = [s for s in subs if wctx.sub(s).dirty(True)]
952 if changedsubs:
952 if changedsubs:
953 raise util.Abort(_("uncommitted changes in subrepo %s")
953 raise util.Abort(_("uncommitted changes in subrepo %s")
954 % changedsubs[0])
954 % changedsubs[0])
955
955
956 # make sure all explicit patterns are matched
956 # make sure all explicit patterns are matched
957 if not force and match.files():
957 if not force and match.files():
958 matched = set(changes[0] + changes[1] + changes[2])
958 matched = set(changes[0] + changes[1] + changes[2])
959
959
960 for f in match.files():
960 for f in match.files():
961 if f == '.' or f in matched or f in wctx.substate:
961 if f == '.' or f in matched or f in wctx.substate:
962 continue
962 continue
963 if f in changes[3]: # missing
963 if f in changes[3]: # missing
964 fail(f, _('file not found!'))
964 fail(f, _('file not found!'))
965 if f in vdirs: # visited directory
965 if f in vdirs: # visited directory
966 d = f + '/'
966 d = f + '/'
967 for mf in matched:
967 for mf in matched:
968 if mf.startswith(d):
968 if mf.startswith(d):
969 break
969 break
970 else:
970 else:
971 fail(f, _("no match under directory!"))
971 fail(f, _("no match under directory!"))
972 elif f not in self.dirstate:
972 elif f not in self.dirstate:
973 fail(f, _("file not tracked!"))
973 fail(f, _("file not tracked!"))
974
974
975 if (not force and not extra.get("close") and not merge
975 if (not force and not extra.get("close") and not merge
976 and not (changes[0] or changes[1] or changes[2])
976 and not (changes[0] or changes[1] or changes[2])
977 and wctx.branch() == wctx.p1().branch()):
977 and wctx.branch() == wctx.p1().branch()):
978 return None
978 return None
979
979
980 ms = mergemod.mergestate(self)
980 ms = mergemod.mergestate(self)
981 for f in changes[0]:
981 for f in changes[0]:
982 if f in ms and ms[f] == 'u':
982 if f in ms and ms[f] == 'u':
983 raise util.Abort(_("unresolved merge conflicts "
983 raise util.Abort(_("unresolved merge conflicts "
984 "(see hg help resolve)"))
984 "(see hg help resolve)"))
985
985
986 cctx = context.workingctx(self, text, user, date, extra, changes)
986 cctx = context.workingctx(self, text, user, date, extra, changes)
987 if editor:
987 if editor:
988 cctx._text = editor(self, cctx, subs)
988 cctx._text = editor(self, cctx, subs)
989 edited = (text != cctx._text)
989 edited = (text != cctx._text)
990
990
991 # commit subs
991 # commit subs
992 if subs or removedsubs:
992 if subs or removedsubs:
993 state = wctx.substate.copy()
993 state = wctx.substate.copy()
994 for s in sorted(subs):
994 for s in sorted(subs):
995 sub = wctx.sub(s)
995 sub = wctx.sub(s)
996 self.ui.status(_('committing subrepository %s\n') %
996 self.ui.status(_('committing subrepository %s\n') %
997 subrepo.subrelpath(sub))
997 subrepo.subrelpath(sub))
998 sr = sub.commit(cctx._text, user, date)
998 sr = sub.commit(cctx._text, user, date)
999 state[s] = (state[s][0], sr)
999 state[s] = (state[s][0], sr)
1000 subrepo.writestate(self, state)
1000 subrepo.writestate(self, state)
1001
1001
1002 # Save commit message in case this transaction gets rolled back
1002 # Save commit message in case this transaction gets rolled back
1003 # (e.g. by a pretxncommit hook). Leave the content alone on
1003 # (e.g. by a pretxncommit hook). Leave the content alone on
1004 # the assumption that the user will use the same editor again.
1004 # the assumption that the user will use the same editor again.
1005 msgfile = self.opener('last-message.txt', 'wb')
1005 msgfile = self.opener('last-message.txt', 'wb')
1006 msgfile.write(cctx._text)
1006 msgfile.write(cctx._text)
1007 msgfile.close()
1007 msgfile.close()
1008
1008
1009 p1, p2 = self.dirstate.parents()
1009 p1, p2 = self.dirstate.parents()
1010 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1010 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1011 try:
1011 try:
1012 self.hook("precommit", throw=True, parent1=hookp1, parent2=hookp2)
1012 self.hook("precommit", throw=True, parent1=hookp1, parent2=hookp2)
1013 ret = self.commitctx(cctx, True)
1013 ret = self.commitctx(cctx, True)
1014 except:
1014 except:
1015 if edited:
1015 if edited:
1016 msgfn = self.pathto(msgfile.name[len(self.root)+1:])
1016 msgfn = self.pathto(msgfile.name[len(self.root)+1:])
1017 self.ui.write(
1017 self.ui.write(
1018 _('note: commit message saved in %s\n') % msgfn)
1018 _('note: commit message saved in %s\n') % msgfn)
1019 raise
1019 raise
1020
1020
1021 # update bookmarks, dirstate and mergestate
1021 # update bookmarks, dirstate and mergestate
1022 parents = (p1, p2)
1022 parents = (p1, p2)
1023 if p2 == nullid:
1023 if p2 == nullid:
1024 parents = (p1,)
1024 parents = (p1,)
1025 bookmarks.update(self, parents, ret)
1025 bookmarks.update(self, parents, ret)
1026 for f in changes[0] + changes[1]:
1026 for f in changes[0] + changes[1]:
1027 self.dirstate.normal(f)
1027 self.dirstate.normal(f)
1028 for f in changes[2]:
1028 for f in changes[2]:
1029 self.dirstate.forget(f)
1029 self.dirstate.forget(f)
1030 self.dirstate.setparents(ret)
1030 self.dirstate.setparents(ret)
1031 ms.reset()
1031 ms.reset()
1032 finally:
1032 finally:
1033 wlock.release()
1033 wlock.release()
1034
1034
1035 self.hook("commit", node=hex(ret), parent1=hookp1, parent2=hookp2)
1035 self.hook("commit", node=hex(ret), parent1=hookp1, parent2=hookp2)
1036 return ret
1036 return ret
1037
1037
1038 def commitctx(self, ctx, error=False):
1038 def commitctx(self, ctx, error=False):
1039 """Add a new revision to current repository.
1039 """Add a new revision to current repository.
1040 Revision information is passed via the context argument.
1040 Revision information is passed via the context argument.
1041 """
1041 """
1042
1042
1043 tr = lock = None
1043 tr = lock = None
1044 removed = list(ctx.removed())
1044 removed = list(ctx.removed())
1045 p1, p2 = ctx.p1(), ctx.p2()
1045 p1, p2 = ctx.p1(), ctx.p2()
1046 m1 = p1.manifest().copy()
1046 m1 = p1.manifest().copy()
1047 m2 = p2.manifest()
1047 m2 = p2.manifest()
1048 user = ctx.user()
1048 user = ctx.user()
1049
1049
1050 lock = self.lock()
1050 lock = self.lock()
1051 try:
1051 try:
1052 tr = self.transaction("commit")
1052 tr = self.transaction("commit")
1053 trp = weakref.proxy(tr)
1053 trp = weakref.proxy(tr)
1054
1054
1055 # check in files
1055 # check in files
1056 new = {}
1056 new = {}
1057 changed = []
1057 changed = []
1058 linkrev = len(self)
1058 linkrev = len(self)
1059 for f in sorted(ctx.modified() + ctx.added()):
1059 for f in sorted(ctx.modified() + ctx.added()):
1060 self.ui.note(f + "\n")
1060 self.ui.note(f + "\n")
1061 try:
1061 try:
1062 fctx = ctx[f]
1062 fctx = ctx[f]
1063 new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
1063 new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
1064 changed)
1064 changed)
1065 m1.set(f, fctx.flags())
1065 m1.set(f, fctx.flags())
1066 except OSError, inst:
1066 except OSError, inst:
1067 self.ui.warn(_("trouble committing %s!\n") % f)
1067 self.ui.warn(_("trouble committing %s!\n") % f)
1068 raise
1068 raise
1069 except IOError, inst:
1069 except IOError, inst:
1070 errcode = getattr(inst, 'errno', errno.ENOENT)
1070 errcode = getattr(inst, 'errno', errno.ENOENT)
1071 if error or errcode and errcode != errno.ENOENT:
1071 if error or errcode and errcode != errno.ENOENT:
1072 self.ui.warn(_("trouble committing %s!\n") % f)
1072 self.ui.warn(_("trouble committing %s!\n") % f)
1073 raise
1073 raise
1074 else:
1074 else:
1075 removed.append(f)
1075 removed.append(f)
1076
1076
1077 # update manifest
1077 # update manifest
1078 m1.update(new)
1078 m1.update(new)
1079 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1079 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1080 drop = [f for f in removed if f in m1]
1080 drop = [f for f in removed if f in m1]
1081 for f in drop:
1081 for f in drop:
1082 del m1[f]
1082 del m1[f]
1083 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1083 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1084 p2.manifestnode(), (new, drop))
1084 p2.manifestnode(), (new, drop))
1085
1085
1086 # update changelog
1086 # update changelog
1087 self.changelog.delayupdate()
1087 self.changelog.delayupdate()
1088 n = self.changelog.add(mn, changed + removed, ctx.description(),
1088 n = self.changelog.add(mn, changed + removed, ctx.description(),
1089 trp, p1.node(), p2.node(),
1089 trp, p1.node(), p2.node(),
1090 user, ctx.date(), ctx.extra().copy())
1090 user, ctx.date(), ctx.extra().copy())
1091 p = lambda: self.changelog.writepending() and self.root or ""
1091 p = lambda: self.changelog.writepending() and self.root or ""
1092 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1092 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1093 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1093 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1094 parent2=xp2, pending=p)
1094 parent2=xp2, pending=p)
1095 self.changelog.finalize(trp)
1095 self.changelog.finalize(trp)
1096 tr.close()
1096 tr.close()
1097
1097
1098 if self._branchcache:
1098 if self._branchcache:
1099 self.updatebranchcache()
1099 self.updatebranchcache()
1100 return n
1100 return n
1101 finally:
1101 finally:
1102 if tr:
1102 if tr:
1103 tr.release()
1103 tr.release()
1104 lock.release()
1104 lock.release()
1105
1105
1106 def destroyed(self):
1106 def destroyed(self):
1107 '''Inform the repository that nodes have been destroyed.
1107 '''Inform the repository that nodes have been destroyed.
1108 Intended for use by strip and rollback, so there's a common
1108 Intended for use by strip and rollback, so there's a common
1109 place for anything that has to be done after destroying history.'''
1109 place for anything that has to be done after destroying history.'''
1110 # XXX it might be nice if we could take the list of destroyed
1110 # XXX it might be nice if we could take the list of destroyed
1111 # nodes, but I don't see an easy way for rollback() to do that
1111 # nodes, but I don't see an easy way for rollback() to do that
1112
1112
1113 # Ensure the persistent tag cache is updated. Doing it now
1113 # Ensure the persistent tag cache is updated. Doing it now
1114 # means that the tag cache only has to worry about destroyed
1114 # means that the tag cache only has to worry about destroyed
1115 # heads immediately after a strip/rollback. That in turn
1115 # heads immediately after a strip/rollback. That in turn
1116 # guarantees that "cachetip == currenttip" (comparing both rev
1116 # guarantees that "cachetip == currenttip" (comparing both rev
1117 # and node) always means no nodes have been added or destroyed.
1117 # and node) always means no nodes have been added or destroyed.
1118
1118
1119 # XXX this is suboptimal when qrefresh'ing: we strip the current
1119 # XXX this is suboptimal when qrefresh'ing: we strip the current
1120 # head, refresh the tag cache, then immediately add a new head.
1120 # head, refresh the tag cache, then immediately add a new head.
1121 # But I think doing it this way is necessary for the "instant
1121 # But I think doing it this way is necessary for the "instant
1122 # tag cache retrieval" case to work.
1122 # tag cache retrieval" case to work.
1123 self.invalidatecaches()
1123 self.invalidatecaches()
1124
1124
1125 def walk(self, match, node=None):
1125 def walk(self, match, node=None):
1126 '''
1126 '''
1127 walk recursively through the directory tree or a given
1127 walk recursively through the directory tree or a given
1128 changeset, finding all files matched by the match
1128 changeset, finding all files matched by the match
1129 function
1129 function
1130 '''
1130 '''
1131 return self[node].walk(match)
1131 return self[node].walk(match)
1132
1132
1133 def status(self, node1='.', node2=None, match=None,
1133 def status(self, node1='.', node2=None, match=None,
1134 ignored=False, clean=False, unknown=False,
1134 ignored=False, clean=False, unknown=False,
1135 listsubrepos=False):
1135 listsubrepos=False):
1136 """return status of files between two nodes or node and working directory
1136 """return status of files between two nodes or node and working directory
1137
1137
1138 If node1 is None, use the first dirstate parent instead.
1138 If node1 is None, use the first dirstate parent instead.
1139 If node2 is None, compare node1 with working directory.
1139 If node2 is None, compare node1 with working directory.
1140 """
1140 """
1141
1141
1142 def mfmatches(ctx):
1142 def mfmatches(ctx):
1143 mf = ctx.manifest().copy()
1143 mf = ctx.manifest().copy()
1144 for fn in mf.keys():
1144 for fn in mf.keys():
1145 if not match(fn):
1145 if not match(fn):
1146 del mf[fn]
1146 del mf[fn]
1147 return mf
1147 return mf
1148
1148
1149 if isinstance(node1, context.changectx):
1149 if isinstance(node1, context.changectx):
1150 ctx1 = node1
1150 ctx1 = node1
1151 else:
1151 else:
1152 ctx1 = self[node1]
1152 ctx1 = self[node1]
1153 if isinstance(node2, context.changectx):
1153 if isinstance(node2, context.changectx):
1154 ctx2 = node2
1154 ctx2 = node2
1155 else:
1155 else:
1156 ctx2 = self[node2]
1156 ctx2 = self[node2]
1157
1157
1158 working = ctx2.rev() is None
1158 working = ctx2.rev() is None
1159 parentworking = working and ctx1 == self['.']
1159 parentworking = working and ctx1 == self['.']
1160 match = match or matchmod.always(self.root, self.getcwd())
1160 match = match or matchmod.always(self.root, self.getcwd())
1161 listignored, listclean, listunknown = ignored, clean, unknown
1161 listignored, listclean, listunknown = ignored, clean, unknown
1162
1162
1163 # load earliest manifest first for caching reasons
1163 # load earliest manifest first for caching reasons
1164 if not working and ctx2.rev() < ctx1.rev():
1164 if not working and ctx2.rev() < ctx1.rev():
1165 ctx2.manifest()
1165 ctx2.manifest()
1166
1166
1167 if not parentworking:
1167 if not parentworking:
1168 def bad(f, msg):
1168 def bad(f, msg):
1169 if f not in ctx1:
1169 if f not in ctx1:
1170 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1170 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1171 match.bad = bad
1171 match.bad = bad
1172
1172
1173 if working: # we need to scan the working dir
1173 if working: # we need to scan the working dir
1174 subrepos = []
1174 subrepos = []
1175 if '.hgsub' in self.dirstate:
1175 if '.hgsub' in self.dirstate:
1176 subrepos = ctx1.substate.keys()
1176 subrepos = ctx1.substate.keys()
1177 s = self.dirstate.status(match, subrepos, listignored,
1177 s = self.dirstate.status(match, subrepos, listignored,
1178 listclean, listunknown)
1178 listclean, listunknown)
1179 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1179 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1180
1180
1181 # check for any possibly clean files
1181 # check for any possibly clean files
1182 if parentworking and cmp:
1182 if parentworking and cmp:
1183 fixup = []
1183 fixup = []
1184 # do a full compare of any files that might have changed
1184 # do a full compare of any files that might have changed
1185 for f in sorted(cmp):
1185 for f in sorted(cmp):
1186 if (f not in ctx1 or ctx2.flags(f) != ctx1.flags(f)
1186 if (f not in ctx1 or ctx2.flags(f) != ctx1.flags(f)
1187 or ctx1[f].cmp(ctx2[f])):
1187 or ctx1[f].cmp(ctx2[f])):
1188 modified.append(f)
1188 modified.append(f)
1189 else:
1189 else:
1190 fixup.append(f)
1190 fixup.append(f)
1191
1191
1192 # update dirstate for files that are actually clean
1192 # update dirstate for files that are actually clean
1193 if fixup:
1193 if fixup:
1194 if listclean:
1194 if listclean:
1195 clean += fixup
1195 clean += fixup
1196
1196
1197 try:
1197 try:
1198 # updating the dirstate is optional
1198 # updating the dirstate is optional
1199 # so we don't wait on the lock
1199 # so we don't wait on the lock
1200 wlock = self.wlock(False)
1200 wlock = self.wlock(False)
1201 try:
1201 try:
1202 for f in fixup:
1202 for f in fixup:
1203 self.dirstate.normal(f)
1203 self.dirstate.normal(f)
1204 finally:
1204 finally:
1205 wlock.release()
1205 wlock.release()
1206 except error.LockError:
1206 except error.LockError:
1207 pass
1207 pass
1208
1208
1209 if not parentworking:
1209 if not parentworking:
1210 mf1 = mfmatches(ctx1)
1210 mf1 = mfmatches(ctx1)
1211 if working:
1211 if working:
1212 # we are comparing working dir against non-parent
1212 # we are comparing working dir against non-parent
1213 # generate a pseudo-manifest for the working dir
1213 # generate a pseudo-manifest for the working dir
1214 mf2 = mfmatches(self['.'])
1214 mf2 = mfmatches(self['.'])
1215 for f in cmp + modified + added:
1215 for f in cmp + modified + added:
1216 mf2[f] = None
1216 mf2[f] = None
1217 mf2.set(f, ctx2.flags(f))
1217 mf2.set(f, ctx2.flags(f))
1218 for f in removed:
1218 for f in removed:
1219 if f in mf2:
1219 if f in mf2:
1220 del mf2[f]
1220 del mf2[f]
1221 else:
1221 else:
1222 # we are comparing two revisions
1222 # we are comparing two revisions
1223 deleted, unknown, ignored = [], [], []
1223 deleted, unknown, ignored = [], [], []
1224 mf2 = mfmatches(ctx2)
1224 mf2 = mfmatches(ctx2)
1225
1225
1226 modified, added, clean = [], [], []
1226 modified, added, clean = [], [], []
1227 for fn in mf2:
1227 for fn in mf2:
1228 if fn in mf1:
1228 if fn in mf1:
1229 if (mf1.flags(fn) != mf2.flags(fn) or
1229 if (mf1.flags(fn) != mf2.flags(fn) or
1230 (mf1[fn] != mf2[fn] and
1230 (mf1[fn] != mf2[fn] and
1231 (mf2[fn] or ctx1[fn].cmp(ctx2[fn])))):
1231 (mf2[fn] or ctx1[fn].cmp(ctx2[fn])))):
1232 modified.append(fn)
1232 modified.append(fn)
1233 elif listclean:
1233 elif listclean:
1234 clean.append(fn)
1234 clean.append(fn)
1235 del mf1[fn]
1235 del mf1[fn]
1236 else:
1236 else:
1237 added.append(fn)
1237 added.append(fn)
1238 removed = mf1.keys()
1238 removed = mf1.keys()
1239
1239
1240 r = modified, added, removed, deleted, unknown, ignored, clean
1240 r = modified, added, removed, deleted, unknown, ignored, clean
1241
1241
1242 if listsubrepos:
1242 if listsubrepos:
1243 for subpath, sub in subrepo.itersubrepos(ctx1, ctx2):
1243 for subpath, sub in subrepo.itersubrepos(ctx1, ctx2):
1244 if working:
1244 if working:
1245 rev2 = None
1245 rev2 = None
1246 else:
1246 else:
1247 rev2 = ctx2.substate[subpath][1]
1247 rev2 = ctx2.substate[subpath][1]
1248 try:
1248 try:
1249 submatch = matchmod.narrowmatcher(subpath, match)
1249 submatch = matchmod.narrowmatcher(subpath, match)
1250 s = sub.status(rev2, match=submatch, ignored=listignored,
1250 s = sub.status(rev2, match=submatch, ignored=listignored,
1251 clean=listclean, unknown=listunknown,
1251 clean=listclean, unknown=listunknown,
1252 listsubrepos=True)
1252 listsubrepos=True)
1253 for rfiles, sfiles in zip(r, s):
1253 for rfiles, sfiles in zip(r, s):
1254 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
1254 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
1255 except error.LookupError:
1255 except error.LookupError:
1256 self.ui.status(_("skipping missing subrepository: %s\n")
1256 self.ui.status(_("skipping missing subrepository: %s\n")
1257 % subpath)
1257 % subpath)
1258
1258
1259 for l in r:
1259 for l in r:
1260 l.sort()
1260 l.sort()
1261 return r
1261 return r
1262
1262
1263 def heads(self, start=None):
1263 def heads(self, start=None):
1264 heads = self.changelog.heads(start)
1264 heads = self.changelog.heads(start)
1265 # sort the output in rev descending order
1265 # sort the output in rev descending order
1266 return sorted(heads, key=self.changelog.rev, reverse=True)
1266 return sorted(heads, key=self.changelog.rev, reverse=True)
1267
1267
1268 def branchheads(self, branch=None, start=None, closed=False):
1268 def branchheads(self, branch=None, start=None, closed=False):
1269 '''return a (possibly filtered) list of heads for the given branch
1269 '''return a (possibly filtered) list of heads for the given branch
1270
1270
1271 Heads are returned in topological order, from newest to oldest.
1271 Heads are returned in topological order, from newest to oldest.
1272 If branch is None, use the dirstate branch.
1272 If branch is None, use the dirstate branch.
1273 If start is not None, return only heads reachable from start.
1273 If start is not None, return only heads reachable from start.
1274 If closed is True, return heads that are marked as closed as well.
1274 If closed is True, return heads that are marked as closed as well.
1275 '''
1275 '''
1276 if branch is None:
1276 if branch is None:
1277 branch = self[None].branch()
1277 branch = self[None].branch()
1278 branches = self.branchmap()
1278 branches = self.branchmap()
1279 if branch not in branches:
1279 if branch not in branches:
1280 return []
1280 return []
1281 # the cache returns heads ordered lowest to highest
1281 # the cache returns heads ordered lowest to highest
1282 bheads = list(reversed(branches[branch]))
1282 bheads = list(reversed(branches[branch]))
1283 if start is not None:
1283 if start is not None:
1284 # filter out the heads that cannot be reached from startrev
1284 # filter out the heads that cannot be reached from startrev
1285 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1285 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1286 bheads = [h for h in bheads if h in fbheads]
1286 bheads = [h for h in bheads if h in fbheads]
1287 if not closed:
1287 if not closed:
1288 bheads = [h for h in bheads if
1288 bheads = [h for h in bheads if
1289 ('close' not in self.changelog.read(h)[5])]
1289 ('close' not in self.changelog.read(h)[5])]
1290 return bheads
1290 return bheads
1291
1291
1292 def branches(self, nodes):
1292 def branches(self, nodes):
1293 if not nodes:
1293 if not nodes:
1294 nodes = [self.changelog.tip()]
1294 nodes = [self.changelog.tip()]
1295 b = []
1295 b = []
1296 for n in nodes:
1296 for n in nodes:
1297 t = n
1297 t = n
1298 while 1:
1298 while 1:
1299 p = self.changelog.parents(n)
1299 p = self.changelog.parents(n)
1300 if p[1] != nullid or p[0] == nullid:
1300 if p[1] != nullid or p[0] == nullid:
1301 b.append((t, n, p[0], p[1]))
1301 b.append((t, n, p[0], p[1]))
1302 break
1302 break
1303 n = p[0]
1303 n = p[0]
1304 return b
1304 return b
1305
1305
1306 def between(self, pairs):
1306 def between(self, pairs):
1307 r = []
1307 r = []
1308
1308
1309 for top, bottom in pairs:
1309 for top, bottom in pairs:
1310 n, l, i = top, [], 0
1310 n, l, i = top, [], 0
1311 f = 1
1311 f = 1
1312
1312
1313 while n != bottom and n != nullid:
1313 while n != bottom and n != nullid:
1314 p = self.changelog.parents(n)[0]
1314 p = self.changelog.parents(n)[0]
1315 if i == f:
1315 if i == f:
1316 l.append(n)
1316 l.append(n)
1317 f = f * 2
1317 f = f * 2
1318 n = p
1318 n = p
1319 i += 1
1319 i += 1
1320
1320
1321 r.append(l)
1321 r.append(l)
1322
1322
1323 return r
1323 return r
1324
1324
1325 def pull(self, remote, heads=None, force=False):
1325 def pull(self, remote, heads=None, force=False):
1326 lock = self.lock()
1326 lock = self.lock()
1327 try:
1327 try:
1328 usecommon = remote.capable('getbundle')
1328 usecommon = remote.capable('getbundle')
1329 tmp = discovery.findcommonincoming(self, remote, heads=heads,
1329 tmp = discovery.findcommonincoming(self, remote, heads=heads,
1330 force=force, commononly=usecommon)
1330 force=force, commononly=usecommon)
1331 common, fetch, rheads = tmp
1331 common, fetch, rheads = tmp
1332 if not fetch:
1332 if not fetch:
1333 self.ui.status(_("no changes found\n"))
1333 self.ui.status(_("no changes found\n"))
1334 result = 0
1334 result = 0
1335 else:
1335 else:
1336 if heads is None and list(common) == [nullid]:
1336 if heads is None and list(common) == [nullid]:
1337 self.ui.status(_("requesting all changes\n"))
1337 self.ui.status(_("requesting all changes\n"))
1338 elif heads is None and remote.capable('changegroupsubset'):
1338 elif heads is None and remote.capable('changegroupsubset'):
1339 # issue1320, avoid a race if remote changed after discovery
1339 # issue1320, avoid a race if remote changed after discovery
1340 heads = rheads
1340 heads = rheads
1341
1341
1342 if usecommon:
1342 if usecommon:
1343 cg = remote.getbundle('pull', common=common,
1343 cg = remote.getbundle('pull', common=common,
1344 heads=heads or rheads)
1344 heads=heads or rheads)
1345 elif heads is None:
1345 elif heads is None:
1346 cg = remote.changegroup(fetch, 'pull')
1346 cg = remote.changegroup(fetch, 'pull')
1347 elif not remote.capable('changegroupsubset'):
1347 elif not remote.capable('changegroupsubset'):
1348 raise util.Abort(_("partial pull cannot be done because "
1348 raise util.Abort(_("partial pull cannot be done because "
1349 "other repository doesn't support "
1349 "other repository doesn't support "
1350 "changegroupsubset."))
1350 "changegroupsubset."))
1351 else:
1351 else:
1352 cg = remote.changegroupsubset(fetch, heads, 'pull')
1352 cg = remote.changegroupsubset(fetch, heads, 'pull')
1353 result = self.addchangegroup(cg, 'pull', remote.url(),
1353 result = self.addchangegroup(cg, 'pull', remote.url(),
1354 lock=lock)
1354 lock=lock)
1355 finally:
1355 finally:
1356 lock.release()
1356 lock.release()
1357
1357
1358 return result
1358 return result
1359
1359
1360 def checkpush(self, force, revs):
1360 def checkpush(self, force, revs):
1361 """Extensions can override this function if additional checks have
1361 """Extensions can override this function if additional checks have
1362 to be performed before pushing, or call it if they override push
1362 to be performed before pushing, or call it if they override push
1363 command.
1363 command.
1364 """
1364 """
1365 pass
1365 pass
1366
1366
1367 def push(self, remote, force=False, revs=None, newbranch=False):
1367 def push(self, remote, force=False, revs=None, newbranch=False):
1368 '''Push outgoing changesets (limited by revs) from the current
1368 '''Push outgoing changesets (limited by revs) from the current
1369 repository to remote. Return an integer:
1369 repository to remote. Return an integer:
1370 - 0 means HTTP error *or* nothing to push
1370 - 0 means HTTP error *or* nothing to push
1371 - 1 means we pushed and remote head count is unchanged *or*
1371 - 1 means we pushed and remote head count is unchanged *or*
1372 we have outgoing changesets but refused to push
1372 we have outgoing changesets but refused to push
1373 - other values as described by addchangegroup()
1373 - other values as described by addchangegroup()
1374 '''
1374 '''
1375 # there are two ways to push to remote repo:
1375 # there are two ways to push to remote repo:
1376 #
1376 #
1377 # addchangegroup assumes local user can lock remote
1377 # addchangegroup assumes local user can lock remote
1378 # repo (local filesystem, old ssh servers).
1378 # repo (local filesystem, old ssh servers).
1379 #
1379 #
1380 # unbundle assumes local user cannot lock remote repo (new ssh
1380 # unbundle assumes local user cannot lock remote repo (new ssh
1381 # servers, http servers).
1381 # servers, http servers).
1382
1382
1383 self.checkpush(force, revs)
1383 self.checkpush(force, revs)
1384 lock = None
1384 lock = None
1385 unbundle = remote.capable('unbundle')
1385 unbundle = remote.capable('unbundle')
1386 if not unbundle:
1386 if not unbundle:
1387 lock = remote.lock()
1387 lock = remote.lock()
1388 try:
1388 try:
1389 cg, remote_heads = discovery.prepush(self, remote, force, revs,
1389 cg, remote_heads = discovery.prepush(self, remote, force, revs,
1390 newbranch)
1390 newbranch)
1391 ret = remote_heads
1391 ret = remote_heads
1392 if cg is not None:
1392 if cg is not None:
1393 if unbundle:
1393 if unbundle:
1394 # local repo finds heads on server, finds out what
1394 # local repo finds heads on server, finds out what
1395 # revs it must push. once revs transferred, if server
1395 # revs it must push. once revs transferred, if server
1396 # finds it has different heads (someone else won
1396 # finds it has different heads (someone else won
1397 # commit/push race), server aborts.
1397 # commit/push race), server aborts.
1398 if force:
1398 if force:
1399 remote_heads = ['force']
1399 remote_heads = ['force']
1400 # ssh: return remote's addchangegroup()
1400 # ssh: return remote's addchangegroup()
1401 # http: return remote's addchangegroup() or 0 for error
1401 # http: return remote's addchangegroup() or 0 for error
1402 ret = remote.unbundle(cg, remote_heads, 'push')
1402 ret = remote.unbundle(cg, remote_heads, 'push')
1403 else:
1403 else:
1404 # we return an integer indicating remote head count change
1404 # we return an integer indicating remote head count change
1405 ret = remote.addchangegroup(cg, 'push', self.url(),
1405 ret = remote.addchangegroup(cg, 'push', self.url(),
1406 lock=lock)
1406 lock=lock)
1407 finally:
1407 finally:
1408 if lock is not None:
1408 if lock is not None:
1409 lock.release()
1409 lock.release()
1410
1410
1411 self.ui.debug("checking for updated bookmarks\n")
1411 self.ui.debug("checking for updated bookmarks\n")
1412 rb = remote.listkeys('bookmarks')
1412 rb = remote.listkeys('bookmarks')
1413 for k in rb.keys():
1413 for k in rb.keys():
1414 if k in self._bookmarks:
1414 if k in self._bookmarks:
1415 nr, nl = rb[k], hex(self._bookmarks[k])
1415 nr, nl = rb[k], hex(self._bookmarks[k])
1416 if nr in self:
1416 if nr in self:
1417 cr = self[nr]
1417 cr = self[nr]
1418 cl = self[nl]
1418 cl = self[nl]
1419 if cl in cr.descendants():
1419 if cl in cr.descendants():
1420 r = remote.pushkey('bookmarks', k, nr, nl)
1420 r = remote.pushkey('bookmarks', k, nr, nl)
1421 if r:
1421 if r:
1422 self.ui.status(_("updating bookmark %s\n") % k)
1422 self.ui.status(_("updating bookmark %s\n") % k)
1423 else:
1423 else:
1424 self.ui.warn(_('updating bookmark %s'
1424 self.ui.warn(_('updating bookmark %s'
1425 ' failed!\n') % k)
1425 ' failed!\n') % k)
1426
1426
1427 return ret
1427 return ret
1428
1428
1429 def changegroupinfo(self, nodes, source):
1429 def changegroupinfo(self, nodes, source):
1430 if self.ui.verbose or source == 'bundle':
1430 if self.ui.verbose or source == 'bundle':
1431 self.ui.status(_("%d changesets found\n") % len(nodes))
1431 self.ui.status(_("%d changesets found\n") % len(nodes))
1432 if self.ui.debugflag:
1432 if self.ui.debugflag:
1433 self.ui.debug("list of changesets:\n")
1433 self.ui.debug("list of changesets:\n")
1434 for node in nodes:
1434 for node in nodes:
1435 self.ui.debug("%s\n" % hex(node))
1435 self.ui.debug("%s\n" % hex(node))
1436
1436
1437 def changegroupsubset(self, bases, heads, source):
1437 def changegroupsubset(self, bases, heads, source):
1438 """Compute a changegroup consisting of all the nodes that are
1438 """Compute a changegroup consisting of all the nodes that are
1439 descendents of any of the bases and ancestors of any of the heads.
1439 descendents of any of the bases and ancestors of any of the heads.
1440 Return a chunkbuffer object whose read() method will return
1440 Return a chunkbuffer object whose read() method will return
1441 successive changegroup chunks.
1441 successive changegroup chunks.
1442
1442
1443 It is fairly complex as determining which filenodes and which
1443 It is fairly complex as determining which filenodes and which
1444 manifest nodes need to be included for the changeset to be complete
1444 manifest nodes need to be included for the changeset to be complete
1445 is non-trivial.
1445 is non-trivial.
1446
1446
1447 Another wrinkle is doing the reverse, figuring out which changeset in
1447 Another wrinkle is doing the reverse, figuring out which changeset in
1448 the changegroup a particular filenode or manifestnode belongs to.
1448 the changegroup a particular filenode or manifestnode belongs to.
1449 """
1449 """
1450 cl = self.changelog
1450 cl = self.changelog
1451 if not bases:
1451 if not bases:
1452 bases = [nullid]
1452 bases = [nullid]
1453 csets, bases, heads = cl.nodesbetween(bases, heads)
1453 csets, bases, heads = cl.nodesbetween(bases, heads)
1454 # We assume that all ancestors of bases are known
1454 # We assume that all ancestors of bases are known
1455 common = set(cl.ancestors(*[cl.rev(n) for n in bases]))
1455 common = set(cl.ancestors(*[cl.rev(n) for n in bases]))
1456 return self._changegroupsubset(common, csets, heads, source)
1456 return self._changegroupsubset(common, csets, heads, source)
1457
1457
1458 def getbundle(self, source, heads=None, common=None):
1458 def getbundle(self, source, heads=None, common=None):
1459 """Like changegroupsubset, but returns the set difference between the
1459 """Like changegroupsubset, but returns the set difference between the
1460 ancestors of heads and the ancestors common.
1460 ancestors of heads and the ancestors common.
1461
1461
1462 If heads is None, use the local heads. If common is None, use [nullid].
1462 If heads is None, use the local heads. If common is None, use [nullid].
1463
1463
1464 The nodes in common might not all be known locally due to the way the
1464 The nodes in common might not all be known locally due to the way the
1465 current discovery protocol works.
1465 current discovery protocol works.
1466 """
1466 """
1467 cl = self.changelog
1467 cl = self.changelog
1468 if common:
1468 if common:
1469 nm = cl.nodemap
1469 nm = cl.nodemap
1470 common = [n for n in common if n in nm]
1470 common = [n for n in common if n in nm]
1471 else:
1471 else:
1472 common = [nullid]
1472 common = [nullid]
1473 if not heads:
1473 if not heads:
1474 heads = cl.heads()
1474 heads = cl.heads()
1475 common, missing = cl.findcommonmissing(common, heads)
1475 common, missing = cl.findcommonmissing(common, heads)
1476 return self._changegroupsubset(common, missing, heads, source)
1476 return self._changegroupsubset(common, missing, heads, source)
1477
1477
1478 def _changegroupsubset(self, commonrevs, csets, heads, source):
1478 def _changegroupsubset(self, commonrevs, csets, heads, source):
1479
1479
1480 cl = self.changelog
1480 cl = self.changelog
1481 mf = self.manifest
1481 mf = self.manifest
1482 mfs = {} # needed manifests
1482 mfs = {} # needed manifests
1483 fnodes = {} # needed file nodes
1483 fnodes = {} # needed file nodes
1484
1484
1485 # can we go through the fast path ?
1485 # can we go through the fast path ?
1486 heads.sort()
1486 heads.sort()
1487 if heads == sorted(self.heads()):
1487 if heads == sorted(self.heads()):
1488 return self._changegroup(csets, source)
1488 return self._changegroup(csets, source)
1489
1489
1490 # slow path
1490 # slow path
1491 self.hook('preoutgoing', throw=True, source=source)
1491 self.hook('preoutgoing', throw=True, source=source)
1492 self.changegroupinfo(csets, source)
1492 self.changegroupinfo(csets, source)
1493
1493
1494 # A function generating function that sets up the initial environment
1494 # A function generating function that sets up the initial environment
1495 # the inner function.
1495 # the inner function.
1496 def filenode_collector(changedfiles):
1496 def filenode_collector(changedfiles):
1497 # This gathers information from each manifestnode included in the
1497 # This gathers information from each manifestnode included in the
1498 # changegroup about which filenodes the manifest node references
1498 # changegroup about which filenodes the manifest node references
1499 # so we can include those in the changegroup too.
1499 # so we can include those in the changegroup too.
1500 #
1500 #
1501 # It also remembers which changenode each filenode belongs to. It
1501 # It also remembers which changenode each filenode belongs to. It
1502 # does this by assuming the a filenode belongs to the changenode
1502 # does this by assuming the a filenode belongs to the changenode
1503 # the first manifest that references it belongs to.
1503 # the first manifest that references it belongs to.
1504 def collect(mnode):
1504 def collect(mnode):
1505 r = mf.rev(mnode)
1505 r = mf.rev(mnode)
1506 clnode = mfs[mnode]
1506 clnode = mfs[mnode]
1507 mdata = mf.readfast(mnode)
1507 mdata = mf.readfast(mnode)
1508 for f in changedfiles:
1508 for f in changedfiles:
1509 if f in mdata:
1509 if f in mdata:
1510 fnodes.setdefault(f, {}).setdefault(mdata[f], clnode)
1510 fnodes.setdefault(f, {}).setdefault(mdata[f], clnode)
1511
1511
1512 return collect
1512 return collect
1513
1513
1514 # If we determine that a particular file or manifest node must be a
1514 # If we determine that a particular file or manifest node must be a
1515 # node that the recipient of the changegroup will already have, we can
1515 # node that the recipient of the changegroup will already have, we can
1516 # also assume the recipient will have all the parents. This function
1516 # also assume the recipient will have all the parents. This function
1517 # prunes them from the set of missing nodes.
1517 # prunes them from the set of missing nodes.
1518 def prune(revlog, missingnodes):
1518 def prune(revlog, missingnodes):
1519 # drop any nodes that claim to be part of a cset in commonrevs
1519 # drop any nodes that claim to be part of a cset in commonrevs
1520 drop = set()
1520 drop = set()
1521 for n in missingnodes:
1521 for n in missingnodes:
1522 if revlog.linkrev(revlog.rev(n)) in commonrevs:
1522 if revlog.linkrev(revlog.rev(n)) in commonrevs:
1523 drop.add(n)
1523 drop.add(n)
1524 for n in drop:
1524 for n in drop:
1525 missingnodes.pop(n, None)
1525 missingnodes.pop(n, None)
1526
1526
1527 # Now that we have all theses utility functions to help out and
1527 # Now that we have all theses utility functions to help out and
1528 # logically divide up the task, generate the group.
1528 # logically divide up the task, generate the group.
1529 def gengroup():
1529 def gengroup():
1530 # The set of changed files starts empty.
1530 # The set of changed files starts empty.
1531 changedfiles = set()
1531 changedfiles = set()
1532
1532
1533 collect = changegroup.collector(cl, mfs, changedfiles)
1533 collect = changegroup.collector(cl, mfs, changedfiles)
1534 count = [0]
1534 count = [0]
1535 def clookup(x):
1535 def clookup(x):
1536 collect(x)
1536 collect(x)
1537 count[0] += 1
1537 count[0] += 1
1538 self.ui.progress(_('bundling'), count[0], unit=_('changesets'))
1538 self.ui.progress(_('bundling'), count[0], unit=_('changesets'))
1539 return x
1539 return x
1540
1540
1541 # Create a changenode group generator that will call our functions
1541 # Create a changenode group generator that will call our functions
1542 # back to lookup the owning changenode and collect information.
1542 # back to lookup the owning changenode and collect information.
1543 for chunk in cl.group(csets, clookup):
1543 for chunk in cl.group(csets, clookup):
1544 yield chunk
1544 yield chunk
1545 changecount = count[0]
1545 changecount = count[0]
1546 efiles = len(changedfiles)
1546 efiles = len(changedfiles)
1547 self.ui.progress(_('bundling'), None)
1547 self.ui.progress(_('bundling'), None)
1548
1548
1549 prune(mf, mfs)
1549 prune(mf, mfs)
1550 # Create a generator for the manifestnodes that calls our lookup
1550 # Create a generator for the manifestnodes that calls our lookup
1551 # and data collection functions back.
1551 # and data collection functions back.
1552 fcollect = filenode_collector(changedfiles)
1552 fcollect = filenode_collector(changedfiles)
1553 count = [0]
1553 count = [0]
1554 def mlookup(x):
1554 def mlookup(x):
1555 fcollect(x)
1555 fcollect(x)
1556 count[0] += 1
1556 count[0] += 1
1557 self.ui.progress(_('bundling'), count[0],
1557 self.ui.progress(_('bundling'), count[0],
1558 unit=_('manifests'), total=changecount)
1558 unit=_('manifests'), total=changecount)
1559 return mfs[x]
1559 return mfs[x]
1560
1560
1561 for chunk in mf.group(sorted(mfs, key=mf.rev), mlookup):
1561 for chunk in mf.group(mfs, mlookup):
1562 yield chunk
1562 yield chunk
1563 self.ui.progress(_('bundling'), None)
1563 self.ui.progress(_('bundling'), None)
1564
1564
1565 mfs.clear()
1565 mfs.clear()
1566
1566
1567 # Go through all our files in order sorted by name.
1567 # Go through all our files in order sorted by name.
1568 for idx, fname in enumerate(sorted(changedfiles)):
1568 for idx, fname in enumerate(sorted(changedfiles)):
1569 filerevlog = self.file(fname)
1569 filerevlog = self.file(fname)
1570 if not len(filerevlog):
1570 if not len(filerevlog):
1571 raise util.Abort(_("empty or missing revlog for %s") % fname)
1571 raise util.Abort(_("empty or missing revlog for %s") % fname)
1572 # Toss out the filenodes that the recipient isn't really
1572 # Toss out the filenodes that the recipient isn't really
1573 # missing.
1573 # missing.
1574 missingfnodes = fnodes.pop(fname, {})
1574 missingfnodes = fnodes.pop(fname, {})
1575 prune(filerevlog, missingfnodes)
1575 prune(filerevlog, missingfnodes)
1576 # If any filenodes are left, generate the group for them,
1576 # If any filenodes are left, generate the group for them,
1577 # otherwise don't bother.
1577 # otherwise don't bother.
1578 if missingfnodes:
1578 if missingfnodes:
1579 yield changegroup.chunkheader(len(fname))
1579 yield changegroup.chunkheader(len(fname))
1580 yield fname
1580 yield fname
1581 # Create a group generator and only pass in a changenode
1581 # Create a group generator and only pass in a changenode
1582 # lookup function as we need to collect no information
1582 # lookup function as we need to collect no information
1583 # from filenodes.
1583 # from filenodes.
1584 def flookup(x):
1584 def flookup(x):
1585 # even though we print the same progress on
1585 # even though we print the same progress on
1586 # most loop iterations, put the progress call
1586 # most loop iterations, put the progress call
1587 # here so that time estimates (if any) can be updated
1587 # here so that time estimates (if any) can be updated
1588 self.ui.progress(
1588 self.ui.progress(
1589 _('bundling'), idx, item=fname,
1589 _('bundling'), idx, item=fname,
1590 unit=_('files'), total=efiles)
1590 unit=_('files'), total=efiles)
1591 return missingfnodes[x]
1591 return missingfnodes[x]
1592
1592
1593 for chunk in filerevlog.group(
1593 for chunk in filerevlog.group(missingfnodes, flookup):
1594 sorted(missingfnodes, key=filerevlog.rev), flookup):
1595 yield chunk
1594 yield chunk
1596 # Signal that no more groups are left.
1595 # Signal that no more groups are left.
1597 yield changegroup.closechunk()
1596 yield changegroup.closechunk()
1598 self.ui.progress(_('bundling'), None)
1597 self.ui.progress(_('bundling'), None)
1599
1598
1600 if csets:
1599 if csets:
1601 self.hook('outgoing', node=hex(csets[0]), source=source)
1600 self.hook('outgoing', node=hex(csets[0]), source=source)
1602
1601
1603 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1602 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1604
1603
1605 def changegroup(self, basenodes, source):
1604 def changegroup(self, basenodes, source):
1606 # to avoid a race we use changegroupsubset() (issue1320)
1605 # to avoid a race we use changegroupsubset() (issue1320)
1607 return self.changegroupsubset(basenodes, self.heads(), source)
1606 return self.changegroupsubset(basenodes, self.heads(), source)
1608
1607
1609 def _changegroup(self, nodes, source):
1608 def _changegroup(self, nodes, source):
1610 """Compute the changegroup of all nodes that we have that a recipient
1609 """Compute the changegroup of all nodes that we have that a recipient
1611 doesn't. Return a chunkbuffer object whose read() method will return
1610 doesn't. Return a chunkbuffer object whose read() method will return
1612 successive changegroup chunks.
1611 successive changegroup chunks.
1613
1612
1614 This is much easier than the previous function as we can assume that
1613 This is much easier than the previous function as we can assume that
1615 the recipient has any changenode we aren't sending them.
1614 the recipient has any changenode we aren't sending them.
1616
1615
1617 nodes is the set of nodes to send"""
1616 nodes is the set of nodes to send"""
1618
1617
1619 self.hook('preoutgoing', throw=True, source=source)
1618 self.hook('preoutgoing', throw=True, source=source)
1620
1619
1621 cl = self.changelog
1620 cl = self.changelog
1622 revset = set([cl.rev(n) for n in nodes])
1621 revset = set([cl.rev(n) for n in nodes])
1623 self.changegroupinfo(nodes, source)
1622 self.changegroupinfo(nodes, source)
1624
1623
1625 def gennodelst(log):
1624 def gennodelst(log):
1626 for r in log:
1625 for r in log:
1627 if log.linkrev(r) in revset:
1626 if log.linkrev(r) in revset:
1628 yield log.node(r)
1627 yield log.node(r)
1629
1628
1630 def lookuplinkrev_func(revlog):
1629 def lookuplinkrev_func(revlog):
1631 def lookuplinkrev(n):
1630 def lookuplinkrev(n):
1632 return cl.node(revlog.linkrev(revlog.rev(n)))
1631 return cl.node(revlog.linkrev(revlog.rev(n)))
1633 return lookuplinkrev
1632 return lookuplinkrev
1634
1633
1635 def gengroup():
1634 def gengroup():
1636 '''yield a sequence of changegroup chunks (strings)'''
1635 '''yield a sequence of changegroup chunks (strings)'''
1637 # construct a list of all changed files
1636 # construct a list of all changed files
1638 changedfiles = set()
1637 changedfiles = set()
1639 mmfs = {}
1638 mmfs = {}
1640
1639
1641 collect = changegroup.collector(cl, mmfs, changedfiles)
1640 collect = changegroup.collector(cl, mmfs, changedfiles)
1642 count = [0]
1641 count = [0]
1643 def clookup(x):
1642 def clookup(x):
1644 count[0] += 1
1643 count[0] += 1
1645 self.ui.progress(_('bundling'), count[0], unit=_('changesets'))
1644 self.ui.progress(_('bundling'), count[0], unit=_('changesets'))
1646 collect(x)
1645 collect(x)
1647 return x
1646 return x
1648
1647
1649 for chunk in cl.group(nodes, clookup):
1648 for chunk in cl.group(nodes, clookup):
1650 yield chunk
1649 yield chunk
1651 efiles = len(changedfiles)
1650 efiles = len(changedfiles)
1652 changecount = count[0]
1651 changecount = count[0]
1653 self.ui.progress(_('bundling'), None)
1652 self.ui.progress(_('bundling'), None)
1654
1653
1655 mnfst = self.manifest
1654 mnfst = self.manifest
1656 nodeiter = gennodelst(mnfst)
1655 nodeiter = gennodelst(mnfst)
1657 mfunc = lookuplinkrev_func(mnfst)
1656 mfunc = lookuplinkrev_func(mnfst)
1658 count = [0]
1657 count = [0]
1659 def mlookup(x):
1658 def mlookup(x):
1660 count[0] += 1
1659 count[0] += 1
1661 self.ui.progress(_('bundling'), count[0],
1660 self.ui.progress(_('bundling'), count[0],
1662 unit=_('manifests'), total=changecount)
1661 unit=_('manifests'), total=changecount)
1663 return mfunc(x)
1662 return mfunc(x)
1664
1663
1665 for chunk in mnfst.group(nodeiter, mlookup):
1664 for chunk in mnfst.group(nodeiter, mlookup):
1666 yield chunk
1665 yield chunk
1667 self.ui.progress(_('bundling'), None)
1666 self.ui.progress(_('bundling'), None)
1668
1667
1669 for idx, fname in enumerate(sorted(changedfiles)):
1668 for idx, fname in enumerate(sorted(changedfiles)):
1670 filerevlog = self.file(fname)
1669 filerevlog = self.file(fname)
1671 if not len(filerevlog):
1670 if not len(filerevlog):
1672 raise util.Abort(_("empty or missing revlog for %s") % fname)
1671 raise util.Abort(_("empty or missing revlog for %s") % fname)
1673 nodeiter = gennodelst(filerevlog)
1672 nodeiter = gennodelst(filerevlog)
1674 nodeiter = list(nodeiter)
1673 nodeiter = list(nodeiter)
1675 if nodeiter:
1674 if nodeiter:
1676 yield changegroup.chunkheader(len(fname))
1675 yield changegroup.chunkheader(len(fname))
1677 yield fname
1676 yield fname
1678 ffunc = lookuplinkrev_func(filerevlog)
1677 ffunc = lookuplinkrev_func(filerevlog)
1679 def flookup(x):
1678 def flookup(x):
1680 self.ui.progress(
1679 self.ui.progress(
1681 _('bundling'), idx, item=fname,
1680 _('bundling'), idx, item=fname,
1682 total=efiles, unit=_('files'))
1681 total=efiles, unit=_('files'))
1683 return ffunc(x)
1682 return ffunc(x)
1684
1683
1685 for chunk in filerevlog.group(nodeiter, flookup):
1684 for chunk in filerevlog.group(nodeiter, flookup):
1686 yield chunk
1685 yield chunk
1687 self.ui.progress(_('bundling'), None)
1686 self.ui.progress(_('bundling'), None)
1688
1687
1689 yield changegroup.closechunk()
1688 yield changegroup.closechunk()
1690
1689
1691 if nodes:
1690 if nodes:
1692 self.hook('outgoing', node=hex(nodes[0]), source=source)
1691 self.hook('outgoing', node=hex(nodes[0]), source=source)
1693
1692
1694 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1693 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1695
1694
1696 def addchangegroup(self, source, srctype, url, emptyok=False, lock=None):
1695 def addchangegroup(self, source, srctype, url, emptyok=False, lock=None):
1697 """Add the changegroup returned by source.read() to this repo.
1696 """Add the changegroup returned by source.read() to this repo.
1698 srctype is a string like 'push', 'pull', or 'unbundle'. url is
1697 srctype is a string like 'push', 'pull', or 'unbundle'. url is
1699 the URL of the repo where this changegroup is coming from.
1698 the URL of the repo where this changegroup is coming from.
1700 If lock is not None, the function takes ownership of the lock
1699 If lock is not None, the function takes ownership of the lock
1701 and releases it after the changegroup is added.
1700 and releases it after the changegroup is added.
1702
1701
1703 Return an integer summarizing the change to this repo:
1702 Return an integer summarizing the change to this repo:
1704 - nothing changed or no source: 0
1703 - nothing changed or no source: 0
1705 - more heads than before: 1+added heads (2..n)
1704 - more heads than before: 1+added heads (2..n)
1706 - fewer heads than before: -1-removed heads (-2..-n)
1705 - fewer heads than before: -1-removed heads (-2..-n)
1707 - number of heads stays the same: 1
1706 - number of heads stays the same: 1
1708 """
1707 """
1709 def csmap(x):
1708 def csmap(x):
1710 self.ui.debug("add changeset %s\n" % short(x))
1709 self.ui.debug("add changeset %s\n" % short(x))
1711 return len(cl)
1710 return len(cl)
1712
1711
1713 def revmap(x):
1712 def revmap(x):
1714 return cl.rev(x)
1713 return cl.rev(x)
1715
1714
1716 if not source:
1715 if not source:
1717 return 0
1716 return 0
1718
1717
1719 self.hook('prechangegroup', throw=True, source=srctype, url=url)
1718 self.hook('prechangegroup', throw=True, source=srctype, url=url)
1720
1719
1721 changesets = files = revisions = 0
1720 changesets = files = revisions = 0
1722 efiles = set()
1721 efiles = set()
1723
1722
1724 # write changelog data to temp files so concurrent readers will not see
1723 # write changelog data to temp files so concurrent readers will not see
1725 # inconsistent view
1724 # inconsistent view
1726 cl = self.changelog
1725 cl = self.changelog
1727 cl.delayupdate()
1726 cl.delayupdate()
1728 oldheads = len(cl.heads())
1727 oldheads = len(cl.heads())
1729
1728
1730 tr = self.transaction("\n".join([srctype, urlmod.hidepassword(url)]))
1729 tr = self.transaction("\n".join([srctype, urlmod.hidepassword(url)]))
1731 try:
1730 try:
1732 trp = weakref.proxy(tr)
1731 trp = weakref.proxy(tr)
1733 # pull off the changeset group
1732 # pull off the changeset group
1734 self.ui.status(_("adding changesets\n"))
1733 self.ui.status(_("adding changesets\n"))
1735 clstart = len(cl)
1734 clstart = len(cl)
1736 class prog(object):
1735 class prog(object):
1737 step = _('changesets')
1736 step = _('changesets')
1738 count = 1
1737 count = 1
1739 ui = self.ui
1738 ui = self.ui
1740 total = None
1739 total = None
1741 def __call__(self):
1740 def __call__(self):
1742 self.ui.progress(self.step, self.count, unit=_('chunks'),
1741 self.ui.progress(self.step, self.count, unit=_('chunks'),
1743 total=self.total)
1742 total=self.total)
1744 self.count += 1
1743 self.count += 1
1745 pr = prog()
1744 pr = prog()
1746 source.callback = pr
1745 source.callback = pr
1747
1746
1748 if (cl.addgroup(source, csmap, trp) is None
1747 if (cl.addgroup(source, csmap, trp) is None
1749 and not emptyok):
1748 and not emptyok):
1750 raise util.Abort(_("received changelog group is empty"))
1749 raise util.Abort(_("received changelog group is empty"))
1751 clend = len(cl)
1750 clend = len(cl)
1752 changesets = clend - clstart
1751 changesets = clend - clstart
1753 for c in xrange(clstart, clend):
1752 for c in xrange(clstart, clend):
1754 efiles.update(self[c].files())
1753 efiles.update(self[c].files())
1755 efiles = len(efiles)
1754 efiles = len(efiles)
1756 self.ui.progress(_('changesets'), None)
1755 self.ui.progress(_('changesets'), None)
1757
1756
1758 # pull off the manifest group
1757 # pull off the manifest group
1759 self.ui.status(_("adding manifests\n"))
1758 self.ui.status(_("adding manifests\n"))
1760 pr.step = _('manifests')
1759 pr.step = _('manifests')
1761 pr.count = 1
1760 pr.count = 1
1762 pr.total = changesets # manifests <= changesets
1761 pr.total = changesets # manifests <= changesets
1763 # no need to check for empty manifest group here:
1762 # no need to check for empty manifest group here:
1764 # if the result of the merge of 1 and 2 is the same in 3 and 4,
1763 # if the result of the merge of 1 and 2 is the same in 3 and 4,
1765 # no new manifest will be created and the manifest group will
1764 # no new manifest will be created and the manifest group will
1766 # be empty during the pull
1765 # be empty during the pull
1767 self.manifest.addgroup(source, revmap, trp)
1766 self.manifest.addgroup(source, revmap, trp)
1768 self.ui.progress(_('manifests'), None)
1767 self.ui.progress(_('manifests'), None)
1769
1768
1770 needfiles = {}
1769 needfiles = {}
1771 if self.ui.configbool('server', 'validate', default=False):
1770 if self.ui.configbool('server', 'validate', default=False):
1772 # validate incoming csets have their manifests
1771 # validate incoming csets have their manifests
1773 for cset in xrange(clstart, clend):
1772 for cset in xrange(clstart, clend):
1774 mfest = self.changelog.read(self.changelog.node(cset))[0]
1773 mfest = self.changelog.read(self.changelog.node(cset))[0]
1775 mfest = self.manifest.readdelta(mfest)
1774 mfest = self.manifest.readdelta(mfest)
1776 # store file nodes we must see
1775 # store file nodes we must see
1777 for f, n in mfest.iteritems():
1776 for f, n in mfest.iteritems():
1778 needfiles.setdefault(f, set()).add(n)
1777 needfiles.setdefault(f, set()).add(n)
1779
1778
1780 # process the files
1779 # process the files
1781 self.ui.status(_("adding file changes\n"))
1780 self.ui.status(_("adding file changes\n"))
1782 pr.step = 'files'
1781 pr.step = 'files'
1783 pr.count = 1
1782 pr.count = 1
1784 pr.total = efiles
1783 pr.total = efiles
1785 source.callback = None
1784 source.callback = None
1786
1785
1787 while 1:
1786 while 1:
1788 f = source.chunk()
1787 f = source.chunk()
1789 if not f:
1788 if not f:
1790 break
1789 break
1791 self.ui.debug("adding %s revisions\n" % f)
1790 self.ui.debug("adding %s revisions\n" % f)
1792 pr()
1791 pr()
1793 fl = self.file(f)
1792 fl = self.file(f)
1794 o = len(fl)
1793 o = len(fl)
1795 if fl.addgroup(source, revmap, trp) is None:
1794 if fl.addgroup(source, revmap, trp) is None:
1796 raise util.Abort(_("received file revlog group is empty"))
1795 raise util.Abort(_("received file revlog group is empty"))
1797 revisions += len(fl) - o
1796 revisions += len(fl) - o
1798 files += 1
1797 files += 1
1799 if f in needfiles:
1798 if f in needfiles:
1800 needs = needfiles[f]
1799 needs = needfiles[f]
1801 for new in xrange(o, len(fl)):
1800 for new in xrange(o, len(fl)):
1802 n = fl.node(new)
1801 n = fl.node(new)
1803 if n in needs:
1802 if n in needs:
1804 needs.remove(n)
1803 needs.remove(n)
1805 if not needs:
1804 if not needs:
1806 del needfiles[f]
1805 del needfiles[f]
1807 self.ui.progress(_('files'), None)
1806 self.ui.progress(_('files'), None)
1808
1807
1809 for f, needs in needfiles.iteritems():
1808 for f, needs in needfiles.iteritems():
1810 fl = self.file(f)
1809 fl = self.file(f)
1811 for n in needs:
1810 for n in needs:
1812 try:
1811 try:
1813 fl.rev(n)
1812 fl.rev(n)
1814 except error.LookupError:
1813 except error.LookupError:
1815 raise util.Abort(
1814 raise util.Abort(
1816 _('missing file data for %s:%s - run hg verify') %
1815 _('missing file data for %s:%s - run hg verify') %
1817 (f, hex(n)))
1816 (f, hex(n)))
1818
1817
1819 newheads = len(cl.heads())
1818 newheads = len(cl.heads())
1820 heads = ""
1819 heads = ""
1821 if oldheads and newheads != oldheads:
1820 if oldheads and newheads != oldheads:
1822 heads = _(" (%+d heads)") % (newheads - oldheads)
1821 heads = _(" (%+d heads)") % (newheads - oldheads)
1823
1822
1824 self.ui.status(_("added %d changesets"
1823 self.ui.status(_("added %d changesets"
1825 " with %d changes to %d files%s\n")
1824 " with %d changes to %d files%s\n")
1826 % (changesets, revisions, files, heads))
1825 % (changesets, revisions, files, heads))
1827
1826
1828 if changesets > 0:
1827 if changesets > 0:
1829 p = lambda: cl.writepending() and self.root or ""
1828 p = lambda: cl.writepending() and self.root or ""
1830 self.hook('pretxnchangegroup', throw=True,
1829 self.hook('pretxnchangegroup', throw=True,
1831 node=hex(cl.node(clstart)), source=srctype,
1830 node=hex(cl.node(clstart)), source=srctype,
1832 url=url, pending=p)
1831 url=url, pending=p)
1833
1832
1834 # make changelog see real files again
1833 # make changelog see real files again
1835 cl.finalize(trp)
1834 cl.finalize(trp)
1836
1835
1837 tr.close()
1836 tr.close()
1838 finally:
1837 finally:
1839 tr.release()
1838 tr.release()
1840 if lock:
1839 if lock:
1841 lock.release()
1840 lock.release()
1842
1841
1843 if changesets > 0:
1842 if changesets > 0:
1844 # forcefully update the on-disk branch cache
1843 # forcefully update the on-disk branch cache
1845 self.ui.debug("updating the branch cache\n")
1844 self.ui.debug("updating the branch cache\n")
1846 self.updatebranchcache()
1845 self.updatebranchcache()
1847 self.hook("changegroup", node=hex(cl.node(clstart)),
1846 self.hook("changegroup", node=hex(cl.node(clstart)),
1848 source=srctype, url=url)
1847 source=srctype, url=url)
1849
1848
1850 for i in xrange(clstart, clend):
1849 for i in xrange(clstart, clend):
1851 self.hook("incoming", node=hex(cl.node(i)),
1850 self.hook("incoming", node=hex(cl.node(i)),
1852 source=srctype, url=url)
1851 source=srctype, url=url)
1853
1852
1854 # never return 0 here:
1853 # never return 0 here:
1855 if newheads < oldheads:
1854 if newheads < oldheads:
1856 return newheads - oldheads - 1
1855 return newheads - oldheads - 1
1857 else:
1856 else:
1858 return newheads - oldheads + 1
1857 return newheads - oldheads + 1
1859
1858
1860
1859
1861 def stream_in(self, remote, requirements):
1860 def stream_in(self, remote, requirements):
1862 lock = self.lock()
1861 lock = self.lock()
1863 try:
1862 try:
1864 fp = remote.stream_out()
1863 fp = remote.stream_out()
1865 l = fp.readline()
1864 l = fp.readline()
1866 try:
1865 try:
1867 resp = int(l)
1866 resp = int(l)
1868 except ValueError:
1867 except ValueError:
1869 raise error.ResponseError(
1868 raise error.ResponseError(
1870 _('Unexpected response from remote server:'), l)
1869 _('Unexpected response from remote server:'), l)
1871 if resp == 1:
1870 if resp == 1:
1872 raise util.Abort(_('operation forbidden by server'))
1871 raise util.Abort(_('operation forbidden by server'))
1873 elif resp == 2:
1872 elif resp == 2:
1874 raise util.Abort(_('locking the remote repository failed'))
1873 raise util.Abort(_('locking the remote repository failed'))
1875 elif resp != 0:
1874 elif resp != 0:
1876 raise util.Abort(_('the server sent an unknown error code'))
1875 raise util.Abort(_('the server sent an unknown error code'))
1877 self.ui.status(_('streaming all changes\n'))
1876 self.ui.status(_('streaming all changes\n'))
1878 l = fp.readline()
1877 l = fp.readline()
1879 try:
1878 try:
1880 total_files, total_bytes = map(int, l.split(' ', 1))
1879 total_files, total_bytes = map(int, l.split(' ', 1))
1881 except (ValueError, TypeError):
1880 except (ValueError, TypeError):
1882 raise error.ResponseError(
1881 raise error.ResponseError(
1883 _('Unexpected response from remote server:'), l)
1882 _('Unexpected response from remote server:'), l)
1884 self.ui.status(_('%d files to transfer, %s of data\n') %
1883 self.ui.status(_('%d files to transfer, %s of data\n') %
1885 (total_files, util.bytecount(total_bytes)))
1884 (total_files, util.bytecount(total_bytes)))
1886 start = time.time()
1885 start = time.time()
1887 for i in xrange(total_files):
1886 for i in xrange(total_files):
1888 # XXX doesn't support '\n' or '\r' in filenames
1887 # XXX doesn't support '\n' or '\r' in filenames
1889 l = fp.readline()
1888 l = fp.readline()
1890 try:
1889 try:
1891 name, size = l.split('\0', 1)
1890 name, size = l.split('\0', 1)
1892 size = int(size)
1891 size = int(size)
1893 except (ValueError, TypeError):
1892 except (ValueError, TypeError):
1894 raise error.ResponseError(
1893 raise error.ResponseError(
1895 _('Unexpected response from remote server:'), l)
1894 _('Unexpected response from remote server:'), l)
1896 self.ui.debug('adding %s (%s)\n' % (name, util.bytecount(size)))
1895 self.ui.debug('adding %s (%s)\n' % (name, util.bytecount(size)))
1897 # for backwards compat, name was partially encoded
1896 # for backwards compat, name was partially encoded
1898 ofp = self.sopener(store.decodedir(name), 'w')
1897 ofp = self.sopener(store.decodedir(name), 'w')
1899 for chunk in util.filechunkiter(fp, limit=size):
1898 for chunk in util.filechunkiter(fp, limit=size):
1900 ofp.write(chunk)
1899 ofp.write(chunk)
1901 ofp.close()
1900 ofp.close()
1902 elapsed = time.time() - start
1901 elapsed = time.time() - start
1903 if elapsed <= 0:
1902 if elapsed <= 0:
1904 elapsed = 0.001
1903 elapsed = 0.001
1905 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
1904 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
1906 (util.bytecount(total_bytes), elapsed,
1905 (util.bytecount(total_bytes), elapsed,
1907 util.bytecount(total_bytes / elapsed)))
1906 util.bytecount(total_bytes / elapsed)))
1908
1907
1909 # new requirements = old non-format requirements + new format-related
1908 # new requirements = old non-format requirements + new format-related
1910 # requirements from the streamed-in repository
1909 # requirements from the streamed-in repository
1911 requirements.update(set(self.requirements) - self.supportedformats)
1910 requirements.update(set(self.requirements) - self.supportedformats)
1912 self._applyrequirements(requirements)
1911 self._applyrequirements(requirements)
1913 self._writerequirements()
1912 self._writerequirements()
1914
1913
1915 self.invalidate()
1914 self.invalidate()
1916 return len(self.heads()) + 1
1915 return len(self.heads()) + 1
1917 finally:
1916 finally:
1918 lock.release()
1917 lock.release()
1919
1918
1920 def clone(self, remote, heads=[], stream=False):
1919 def clone(self, remote, heads=[], stream=False):
1921 '''clone remote repository.
1920 '''clone remote repository.
1922
1921
1923 keyword arguments:
1922 keyword arguments:
1924 heads: list of revs to clone (forces use of pull)
1923 heads: list of revs to clone (forces use of pull)
1925 stream: use streaming clone if possible'''
1924 stream: use streaming clone if possible'''
1926
1925
1927 # now, all clients that can request uncompressed clones can
1926 # now, all clients that can request uncompressed clones can
1928 # read repo formats supported by all servers that can serve
1927 # read repo formats supported by all servers that can serve
1929 # them.
1928 # them.
1930
1929
1931 # if revlog format changes, client will have to check version
1930 # if revlog format changes, client will have to check version
1932 # and format flags on "stream" capability, and use
1931 # and format flags on "stream" capability, and use
1933 # uncompressed only if compatible.
1932 # uncompressed only if compatible.
1934
1933
1935 if stream and not heads:
1934 if stream and not heads:
1936 # 'stream' means remote revlog format is revlogv1 only
1935 # 'stream' means remote revlog format is revlogv1 only
1937 if remote.capable('stream'):
1936 if remote.capable('stream'):
1938 return self.stream_in(remote, set(('revlogv1',)))
1937 return self.stream_in(remote, set(('revlogv1',)))
1939 # otherwise, 'streamreqs' contains the remote revlog format
1938 # otherwise, 'streamreqs' contains the remote revlog format
1940 streamreqs = remote.capable('streamreqs')
1939 streamreqs = remote.capable('streamreqs')
1941 if streamreqs:
1940 if streamreqs:
1942 streamreqs = set(streamreqs.split(','))
1941 streamreqs = set(streamreqs.split(','))
1943 # if we support it, stream in and adjust our requirements
1942 # if we support it, stream in and adjust our requirements
1944 if not streamreqs - self.supportedformats:
1943 if not streamreqs - self.supportedformats:
1945 return self.stream_in(remote, streamreqs)
1944 return self.stream_in(remote, streamreqs)
1946 return self.pull(remote, heads)
1945 return self.pull(remote, heads)
1947
1946
1948 def pushkey(self, namespace, key, old, new):
1947 def pushkey(self, namespace, key, old, new):
1949 return pushkey.push(self, namespace, key, old, new)
1948 return pushkey.push(self, namespace, key, old, new)
1950
1949
1951 def listkeys(self, namespace):
1950 def listkeys(self, namespace):
1952 return pushkey.list(self, namespace)
1951 return pushkey.list(self, namespace)
1953
1952
1954 def debugwireargs(self, one, two, three=None, four=None):
1953 def debugwireargs(self, one, two, three=None, four=None):
1955 '''used to test argument passing over the wire'''
1954 '''used to test argument passing over the wire'''
1956 return "%s %s %s %s" % (one, two, three, four)
1955 return "%s %s %s %s" % (one, two, three, four)
1957
1956
1958 # used to avoid circular references so destructors work
1957 # used to avoid circular references so destructors work
1959 def aftertrans(files):
1958 def aftertrans(files):
1960 renamefiles = [tuple(t) for t in files]
1959 renamefiles = [tuple(t) for t in files]
1961 def a():
1960 def a():
1962 for src, dest in renamefiles:
1961 for src, dest in renamefiles:
1963 util.rename(src, dest)
1962 util.rename(src, dest)
1964 return a
1963 return a
1965
1964
1966 def instance(ui, path, create):
1965 def instance(ui, path, create):
1967 return localrepository(ui, util.drop_scheme('file', path), create)
1966 return localrepository(ui, util.drop_scheme('file', path), create)
1968
1967
1969 def islocal(path):
1968 def islocal(path):
1970 return True
1969 return True
@@ -1,1273 +1,1273 b''
1 # revlog.py - storage back-end for mercurial
1 # revlog.py - storage back-end 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 """Storage back-end for Mercurial.
8 """Storage back-end for Mercurial.
9
9
10 This provides efficient delta storage with O(1) retrieve and append
10 This provides efficient delta storage with O(1) retrieve and append
11 and O(changes) merge between branches.
11 and O(changes) merge between branches.
12 """
12 """
13
13
14 # import stuff from node for others to import from revlog
14 # import stuff from node for others to import from revlog
15 from node import bin, hex, nullid, nullrev, short #@UnusedImport
15 from node import bin, hex, nullid, nullrev, short #@UnusedImport
16 from i18n import _
16 from i18n import _
17 import changegroup, ancestor, mdiff, parsers, error, util
17 import changegroup, ancestor, mdiff, parsers, error, util
18 import struct, zlib, errno
18 import struct, zlib, errno
19
19
20 _pack = struct.pack
20 _pack = struct.pack
21 _unpack = struct.unpack
21 _unpack = struct.unpack
22 _compress = zlib.compress
22 _compress = zlib.compress
23 _decompress = zlib.decompress
23 _decompress = zlib.decompress
24 _sha = util.sha1
24 _sha = util.sha1
25
25
26 # revlog header flags
26 # revlog header flags
27 REVLOGV0 = 0
27 REVLOGV0 = 0
28 REVLOGNG = 1
28 REVLOGNG = 1
29 REVLOGNGINLINEDATA = (1 << 16)
29 REVLOGNGINLINEDATA = (1 << 16)
30 REVLOGSHALLOW = (1 << 17)
30 REVLOGSHALLOW = (1 << 17)
31 REVLOG_DEFAULT_FLAGS = REVLOGNGINLINEDATA
31 REVLOG_DEFAULT_FLAGS = REVLOGNGINLINEDATA
32 REVLOG_DEFAULT_FORMAT = REVLOGNG
32 REVLOG_DEFAULT_FORMAT = REVLOGNG
33 REVLOG_DEFAULT_VERSION = REVLOG_DEFAULT_FORMAT | REVLOG_DEFAULT_FLAGS
33 REVLOG_DEFAULT_VERSION = REVLOG_DEFAULT_FORMAT | REVLOG_DEFAULT_FLAGS
34 REVLOGNG_FLAGS = REVLOGNGINLINEDATA | REVLOGSHALLOW
34 REVLOGNG_FLAGS = REVLOGNGINLINEDATA | REVLOGSHALLOW
35
35
36 # revlog index flags
36 # revlog index flags
37 REVIDX_PARENTDELTA = 1
37 REVIDX_PARENTDELTA = 1
38 REVIDX_PUNCHED_FLAG = 2
38 REVIDX_PUNCHED_FLAG = 2
39 REVIDX_KNOWN_FLAGS = REVIDX_PUNCHED_FLAG | REVIDX_PARENTDELTA
39 REVIDX_KNOWN_FLAGS = REVIDX_PUNCHED_FLAG | REVIDX_PARENTDELTA
40
40
41 # max size of revlog with inline data
41 # max size of revlog with inline data
42 _maxinline = 131072
42 _maxinline = 131072
43 _chunksize = 1048576
43 _chunksize = 1048576
44
44
45 RevlogError = error.RevlogError
45 RevlogError = error.RevlogError
46 LookupError = error.LookupError
46 LookupError = error.LookupError
47
47
48 def getoffset(q):
48 def getoffset(q):
49 return int(q >> 16)
49 return int(q >> 16)
50
50
51 def gettype(q):
51 def gettype(q):
52 return int(q & 0xFFFF)
52 return int(q & 0xFFFF)
53
53
54 def offset_type(offset, type):
54 def offset_type(offset, type):
55 return long(long(offset) << 16 | type)
55 return long(long(offset) << 16 | type)
56
56
57 nullhash = _sha(nullid)
57 nullhash = _sha(nullid)
58
58
59 def hash(text, p1, p2):
59 def hash(text, p1, p2):
60 """generate a hash from the given text and its parent hashes
60 """generate a hash from the given text and its parent hashes
61
61
62 This hash combines both the current file contents and its history
62 This hash combines both the current file contents and its history
63 in a manner that makes it easy to distinguish nodes with the same
63 in a manner that makes it easy to distinguish nodes with the same
64 content in the revision graph.
64 content in the revision graph.
65 """
65 """
66 # As of now, if one of the parent node is null, p2 is null
66 # As of now, if one of the parent node is null, p2 is null
67 if p2 == nullid:
67 if p2 == nullid:
68 # deep copy of a hash is faster than creating one
68 # deep copy of a hash is faster than creating one
69 s = nullhash.copy()
69 s = nullhash.copy()
70 s.update(p1)
70 s.update(p1)
71 else:
71 else:
72 # none of the parent nodes are nullid
72 # none of the parent nodes are nullid
73 l = [p1, p2]
73 l = [p1, p2]
74 l.sort()
74 l.sort()
75 s = _sha(l[0])
75 s = _sha(l[0])
76 s.update(l[1])
76 s.update(l[1])
77 s.update(text)
77 s.update(text)
78 return s.digest()
78 return s.digest()
79
79
80 def compress(text):
80 def compress(text):
81 """ generate a possibly-compressed representation of text """
81 """ generate a possibly-compressed representation of text """
82 if not text:
82 if not text:
83 return ("", text)
83 return ("", text)
84 l = len(text)
84 l = len(text)
85 bin = None
85 bin = None
86 if l < 44:
86 if l < 44:
87 pass
87 pass
88 elif l > 1000000:
88 elif l > 1000000:
89 # zlib makes an internal copy, thus doubling memory usage for
89 # zlib makes an internal copy, thus doubling memory usage for
90 # large files, so lets do this in pieces
90 # large files, so lets do this in pieces
91 z = zlib.compressobj()
91 z = zlib.compressobj()
92 p = []
92 p = []
93 pos = 0
93 pos = 0
94 while pos < l:
94 while pos < l:
95 pos2 = pos + 2**20
95 pos2 = pos + 2**20
96 p.append(z.compress(text[pos:pos2]))
96 p.append(z.compress(text[pos:pos2]))
97 pos = pos2
97 pos = pos2
98 p.append(z.flush())
98 p.append(z.flush())
99 if sum(map(len, p)) < l:
99 if sum(map(len, p)) < l:
100 bin = "".join(p)
100 bin = "".join(p)
101 else:
101 else:
102 bin = _compress(text)
102 bin = _compress(text)
103 if bin is None or len(bin) > l:
103 if bin is None or len(bin) > l:
104 if text[0] == '\0':
104 if text[0] == '\0':
105 return ("", text)
105 return ("", text)
106 return ('u', text)
106 return ('u', text)
107 return ("", bin)
107 return ("", bin)
108
108
109 def decompress(bin):
109 def decompress(bin):
110 """ decompress the given input """
110 """ decompress the given input """
111 if not bin:
111 if not bin:
112 return bin
112 return bin
113 t = bin[0]
113 t = bin[0]
114 if t == '\0':
114 if t == '\0':
115 return bin
115 return bin
116 if t == 'x':
116 if t == 'x':
117 return _decompress(bin)
117 return _decompress(bin)
118 if t == 'u':
118 if t == 'u':
119 return bin[1:]
119 return bin[1:]
120 raise RevlogError(_("unknown compression type %r") % t)
120 raise RevlogError(_("unknown compression type %r") % t)
121
121
122 indexformatv0 = ">4l20s20s20s"
122 indexformatv0 = ">4l20s20s20s"
123 v0shaoffset = 56
123 v0shaoffset = 56
124
124
125 class revlogoldio(object):
125 class revlogoldio(object):
126 def __init__(self):
126 def __init__(self):
127 self.size = struct.calcsize(indexformatv0)
127 self.size = struct.calcsize(indexformatv0)
128
128
129 def parseindex(self, data, inline):
129 def parseindex(self, data, inline):
130 s = self.size
130 s = self.size
131 index = []
131 index = []
132 nodemap = {nullid: nullrev}
132 nodemap = {nullid: nullrev}
133 n = off = 0
133 n = off = 0
134 l = len(data)
134 l = len(data)
135 while off + s <= l:
135 while off + s <= l:
136 cur = data[off:off + s]
136 cur = data[off:off + s]
137 off += s
137 off += s
138 e = _unpack(indexformatv0, cur)
138 e = _unpack(indexformatv0, cur)
139 # transform to revlogv1 format
139 # transform to revlogv1 format
140 e2 = (offset_type(e[0], 0), e[1], -1, e[2], e[3],
140 e2 = (offset_type(e[0], 0), e[1], -1, e[2], e[3],
141 nodemap.get(e[4], nullrev), nodemap.get(e[5], nullrev), e[6])
141 nodemap.get(e[4], nullrev), nodemap.get(e[5], nullrev), e[6])
142 index.append(e2)
142 index.append(e2)
143 nodemap[e[6]] = n
143 nodemap[e[6]] = n
144 n += 1
144 n += 1
145
145
146 # add the magic null revision at -1
146 # add the magic null revision at -1
147 index.append((0, 0, 0, -1, -1, -1, -1, nullid))
147 index.append((0, 0, 0, -1, -1, -1, -1, nullid))
148
148
149 return index, nodemap, None
149 return index, nodemap, None
150
150
151 def packentry(self, entry, node, version, rev):
151 def packentry(self, entry, node, version, rev):
152 if gettype(entry[0]):
152 if gettype(entry[0]):
153 raise RevlogError(_("index entry flags need RevlogNG"))
153 raise RevlogError(_("index entry flags need RevlogNG"))
154 e2 = (getoffset(entry[0]), entry[1], entry[3], entry[4],
154 e2 = (getoffset(entry[0]), entry[1], entry[3], entry[4],
155 node(entry[5]), node(entry[6]), entry[7])
155 node(entry[5]), node(entry[6]), entry[7])
156 return _pack(indexformatv0, *e2)
156 return _pack(indexformatv0, *e2)
157
157
158 # index ng:
158 # index ng:
159 # 6 bytes: offset
159 # 6 bytes: offset
160 # 2 bytes: flags
160 # 2 bytes: flags
161 # 4 bytes: compressed length
161 # 4 bytes: compressed length
162 # 4 bytes: uncompressed length
162 # 4 bytes: uncompressed length
163 # 4 bytes: base rev
163 # 4 bytes: base rev
164 # 4 bytes: link rev
164 # 4 bytes: link rev
165 # 4 bytes: parent 1 rev
165 # 4 bytes: parent 1 rev
166 # 4 bytes: parent 2 rev
166 # 4 bytes: parent 2 rev
167 # 32 bytes: nodeid
167 # 32 bytes: nodeid
168 indexformatng = ">Qiiiiii20s12x"
168 indexformatng = ">Qiiiiii20s12x"
169 ngshaoffset = 32
169 ngshaoffset = 32
170 versionformat = ">I"
170 versionformat = ">I"
171
171
172 class revlogio(object):
172 class revlogio(object):
173 def __init__(self):
173 def __init__(self):
174 self.size = struct.calcsize(indexformatng)
174 self.size = struct.calcsize(indexformatng)
175
175
176 def parseindex(self, data, inline):
176 def parseindex(self, data, inline):
177 # call the C implementation to parse the index data
177 # call the C implementation to parse the index data
178 index, cache = parsers.parse_index2(data, inline)
178 index, cache = parsers.parse_index2(data, inline)
179 return index, None, cache
179 return index, None, cache
180
180
181 def packentry(self, entry, node, version, rev):
181 def packentry(self, entry, node, version, rev):
182 p = _pack(indexformatng, *entry)
182 p = _pack(indexformatng, *entry)
183 if rev == 0:
183 if rev == 0:
184 p = _pack(versionformat, version) + p[4:]
184 p = _pack(versionformat, version) + p[4:]
185 return p
185 return p
186
186
187 class revlog(object):
187 class revlog(object):
188 """
188 """
189 the underlying revision storage object
189 the underlying revision storage object
190
190
191 A revlog consists of two parts, an index and the revision data.
191 A revlog consists of two parts, an index and the revision data.
192
192
193 The index is a file with a fixed record size containing
193 The index is a file with a fixed record size containing
194 information on each revision, including its nodeid (hash), the
194 information on each revision, including its nodeid (hash), the
195 nodeids of its parents, the position and offset of its data within
195 nodeids of its parents, the position and offset of its data within
196 the data file, and the revision it's based on. Finally, each entry
196 the data file, and the revision it's based on. Finally, each entry
197 contains a linkrev entry that can serve as a pointer to external
197 contains a linkrev entry that can serve as a pointer to external
198 data.
198 data.
199
199
200 The revision data itself is a linear collection of data chunks.
200 The revision data itself is a linear collection of data chunks.
201 Each chunk represents a revision and is usually represented as a
201 Each chunk represents a revision and is usually represented as a
202 delta against the previous chunk. To bound lookup time, runs of
202 delta against the previous chunk. To bound lookup time, runs of
203 deltas are limited to about 2 times the length of the original
203 deltas are limited to about 2 times the length of the original
204 version data. This makes retrieval of a version proportional to
204 version data. This makes retrieval of a version proportional to
205 its size, or O(1) relative to the number of revisions.
205 its size, or O(1) relative to the number of revisions.
206
206
207 Both pieces of the revlog are written to in an append-only
207 Both pieces of the revlog are written to in an append-only
208 fashion, which means we never need to rewrite a file to insert or
208 fashion, which means we never need to rewrite a file to insert or
209 remove data, and can use some simple techniques to avoid the need
209 remove data, and can use some simple techniques to avoid the need
210 for locking while reading.
210 for locking while reading.
211 """
211 """
212 def __init__(self, opener, indexfile, shallowroot=None):
212 def __init__(self, opener, indexfile, shallowroot=None):
213 """
213 """
214 create a revlog object
214 create a revlog object
215
215
216 opener is a function that abstracts the file opening operation
216 opener is a function that abstracts the file opening operation
217 and can be used to implement COW semantics or the like.
217 and can be used to implement COW semantics or the like.
218 """
218 """
219 self.indexfile = indexfile
219 self.indexfile = indexfile
220 self.datafile = indexfile[:-2] + ".d"
220 self.datafile = indexfile[:-2] + ".d"
221 self.opener = opener
221 self.opener = opener
222 self._cache = None
222 self._cache = None
223 self._chunkcache = (0, '')
223 self._chunkcache = (0, '')
224 self.index = []
224 self.index = []
225 self._shallowroot = shallowroot
225 self._shallowroot = shallowroot
226 self._parentdelta = 0
226 self._parentdelta = 0
227 self._pcache = {}
227 self._pcache = {}
228 self._nodecache = {nullid: nullrev}
228 self._nodecache = {nullid: nullrev}
229 self._nodepos = None
229 self._nodepos = None
230
230
231 v = REVLOG_DEFAULT_VERSION
231 v = REVLOG_DEFAULT_VERSION
232 if hasattr(opener, 'options') and 'defversion' in opener.options:
232 if hasattr(opener, 'options') and 'defversion' in opener.options:
233 v = opener.options['defversion']
233 v = opener.options['defversion']
234 if v & REVLOGNG:
234 if v & REVLOGNG:
235 v |= REVLOGNGINLINEDATA
235 v |= REVLOGNGINLINEDATA
236 if v & REVLOGNG and 'parentdelta' in opener.options:
236 if v & REVLOGNG and 'parentdelta' in opener.options:
237 self._parentdelta = 1
237 self._parentdelta = 1
238
238
239 if shallowroot:
239 if shallowroot:
240 v |= REVLOGSHALLOW
240 v |= REVLOGSHALLOW
241
241
242 i = ''
242 i = ''
243 try:
243 try:
244 f = self.opener(self.indexfile)
244 f = self.opener(self.indexfile)
245 i = f.read()
245 i = f.read()
246 f.close()
246 f.close()
247 if len(i) > 0:
247 if len(i) > 0:
248 v = struct.unpack(versionformat, i[:4])[0]
248 v = struct.unpack(versionformat, i[:4])[0]
249 except IOError, inst:
249 except IOError, inst:
250 if inst.errno != errno.ENOENT:
250 if inst.errno != errno.ENOENT:
251 raise
251 raise
252
252
253 self.version = v
253 self.version = v
254 self._inline = v & REVLOGNGINLINEDATA
254 self._inline = v & REVLOGNGINLINEDATA
255 self._shallow = v & REVLOGSHALLOW
255 self._shallow = v & REVLOGSHALLOW
256 flags = v & ~0xFFFF
256 flags = v & ~0xFFFF
257 fmt = v & 0xFFFF
257 fmt = v & 0xFFFF
258 if fmt == REVLOGV0 and flags:
258 if fmt == REVLOGV0 and flags:
259 raise RevlogError(_("index %s unknown flags %#04x for format v0")
259 raise RevlogError(_("index %s unknown flags %#04x for format v0")
260 % (self.indexfile, flags >> 16))
260 % (self.indexfile, flags >> 16))
261 elif fmt == REVLOGNG and flags & ~REVLOGNG_FLAGS:
261 elif fmt == REVLOGNG and flags & ~REVLOGNG_FLAGS:
262 raise RevlogError(_("index %s unknown flags %#04x for revlogng")
262 raise RevlogError(_("index %s unknown flags %#04x for revlogng")
263 % (self.indexfile, flags >> 16))
263 % (self.indexfile, flags >> 16))
264 elif fmt > REVLOGNG:
264 elif fmt > REVLOGNG:
265 raise RevlogError(_("index %s unknown format %d")
265 raise RevlogError(_("index %s unknown format %d")
266 % (self.indexfile, fmt))
266 % (self.indexfile, fmt))
267
267
268 self._io = revlogio()
268 self._io = revlogio()
269 if self.version == REVLOGV0:
269 if self.version == REVLOGV0:
270 self._io = revlogoldio()
270 self._io = revlogoldio()
271 try:
271 try:
272 d = self._io.parseindex(i, self._inline)
272 d = self._io.parseindex(i, self._inline)
273 except (ValueError, IndexError):
273 except (ValueError, IndexError):
274 raise RevlogError(_("index %s is corrupted") % (self.indexfile))
274 raise RevlogError(_("index %s is corrupted") % (self.indexfile))
275 self.index, nodemap, self._chunkcache = d
275 self.index, nodemap, self._chunkcache = d
276 if nodemap is not None:
276 if nodemap is not None:
277 self.nodemap = self._nodecache = nodemap
277 self.nodemap = self._nodecache = nodemap
278 if not self._chunkcache:
278 if not self._chunkcache:
279 self._chunkclear()
279 self._chunkclear()
280
280
281 def tip(self):
281 def tip(self):
282 return self.node(len(self.index) - 2)
282 return self.node(len(self.index) - 2)
283 def __len__(self):
283 def __len__(self):
284 return len(self.index) - 1
284 return len(self.index) - 1
285 def __iter__(self):
285 def __iter__(self):
286 for i in xrange(len(self)):
286 for i in xrange(len(self)):
287 yield i
287 yield i
288
288
289 @util.propertycache
289 @util.propertycache
290 def nodemap(self):
290 def nodemap(self):
291 n = self.rev(self.node(0))
291 n = self.rev(self.node(0))
292 return self._nodecache
292 return self._nodecache
293
293
294 def rev(self, node):
294 def rev(self, node):
295 try:
295 try:
296 return self._nodecache[node]
296 return self._nodecache[node]
297 except KeyError:
297 except KeyError:
298 n = self._nodecache
298 n = self._nodecache
299 i = self.index
299 i = self.index
300 p = self._nodepos
300 p = self._nodepos
301 if p is None:
301 if p is None:
302 p = len(i) - 2
302 p = len(i) - 2
303 for r in xrange(p, -1, -1):
303 for r in xrange(p, -1, -1):
304 v = i[r][7]
304 v = i[r][7]
305 n[v] = r
305 n[v] = r
306 if v == node:
306 if v == node:
307 self._nodepos = r - 1
307 self._nodepos = r - 1
308 return r
308 return r
309 raise LookupError(node, self.indexfile, _('no node'))
309 raise LookupError(node, self.indexfile, _('no node'))
310
310
311 def node(self, rev):
311 def node(self, rev):
312 return self.index[rev][7]
312 return self.index[rev][7]
313 def linkrev(self, rev):
313 def linkrev(self, rev):
314 return self.index[rev][4]
314 return self.index[rev][4]
315 def parents(self, node):
315 def parents(self, node):
316 i = self.index
316 i = self.index
317 d = i[self.rev(node)]
317 d = i[self.rev(node)]
318 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
318 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
319 def parentrevs(self, rev):
319 def parentrevs(self, rev):
320 return self.index[rev][5:7]
320 return self.index[rev][5:7]
321 def start(self, rev):
321 def start(self, rev):
322 return int(self.index[rev][0] >> 16)
322 return int(self.index[rev][0] >> 16)
323 def end(self, rev):
323 def end(self, rev):
324 return self.start(rev) + self.length(rev)
324 return self.start(rev) + self.length(rev)
325 def length(self, rev):
325 def length(self, rev):
326 return self.index[rev][1]
326 return self.index[rev][1]
327 def base(self, rev):
327 def base(self, rev):
328 return self.index[rev][3]
328 return self.index[rev][3]
329 def flags(self, rev):
329 def flags(self, rev):
330 return self.index[rev][0] & 0xFFFF
330 return self.index[rev][0] & 0xFFFF
331 def rawsize(self, rev):
331 def rawsize(self, rev):
332 """return the length of the uncompressed text for a given revision"""
332 """return the length of the uncompressed text for a given revision"""
333 l = self.index[rev][2]
333 l = self.index[rev][2]
334 if l >= 0:
334 if l >= 0:
335 return l
335 return l
336
336
337 t = self.revision(self.node(rev))
337 t = self.revision(self.node(rev))
338 return len(t)
338 return len(t)
339 size = rawsize
339 size = rawsize
340
340
341 def reachable(self, node, stop=None):
341 def reachable(self, node, stop=None):
342 """return the set of all nodes ancestral to a given node, including
342 """return the set of all nodes ancestral to a given node, including
343 the node itself, stopping when stop is matched"""
343 the node itself, stopping when stop is matched"""
344 reachable = set((node,))
344 reachable = set((node,))
345 visit = [node]
345 visit = [node]
346 if stop:
346 if stop:
347 stopn = self.rev(stop)
347 stopn = self.rev(stop)
348 else:
348 else:
349 stopn = 0
349 stopn = 0
350 while visit:
350 while visit:
351 n = visit.pop(0)
351 n = visit.pop(0)
352 if n == stop:
352 if n == stop:
353 continue
353 continue
354 if n == nullid:
354 if n == nullid:
355 continue
355 continue
356 for p in self.parents(n):
356 for p in self.parents(n):
357 if self.rev(p) < stopn:
357 if self.rev(p) < stopn:
358 continue
358 continue
359 if p not in reachable:
359 if p not in reachable:
360 reachable.add(p)
360 reachable.add(p)
361 visit.append(p)
361 visit.append(p)
362 return reachable
362 return reachable
363
363
364 def ancestors(self, *revs):
364 def ancestors(self, *revs):
365 """Generate the ancestors of 'revs' in reverse topological order.
365 """Generate the ancestors of 'revs' in reverse topological order.
366
366
367 Yield a sequence of revision numbers starting with the parents
367 Yield a sequence of revision numbers starting with the parents
368 of each revision in revs, i.e., each revision is *not* considered
368 of each revision in revs, i.e., each revision is *not* considered
369 an ancestor of itself. Results are in breadth-first order:
369 an ancestor of itself. Results are in breadth-first order:
370 parents of each rev in revs, then parents of those, etc. Result
370 parents of each rev in revs, then parents of those, etc. Result
371 does not include the null revision."""
371 does not include the null revision."""
372 visit = list(revs)
372 visit = list(revs)
373 seen = set([nullrev])
373 seen = set([nullrev])
374 while visit:
374 while visit:
375 for parent in self.parentrevs(visit.pop(0)):
375 for parent in self.parentrevs(visit.pop(0)):
376 if parent not in seen:
376 if parent not in seen:
377 visit.append(parent)
377 visit.append(parent)
378 seen.add(parent)
378 seen.add(parent)
379 yield parent
379 yield parent
380
380
381 def descendants(self, *revs):
381 def descendants(self, *revs):
382 """Generate the descendants of 'revs' in revision order.
382 """Generate the descendants of 'revs' in revision order.
383
383
384 Yield a sequence of revision numbers starting with a child of
384 Yield a sequence of revision numbers starting with a child of
385 some rev in revs, i.e., each revision is *not* considered a
385 some rev in revs, i.e., each revision is *not* considered a
386 descendant of itself. Results are ordered by revision number (a
386 descendant of itself. Results are ordered by revision number (a
387 topological sort)."""
387 topological sort)."""
388 first = min(revs)
388 first = min(revs)
389 if first == nullrev:
389 if first == nullrev:
390 for i in self:
390 for i in self:
391 yield i
391 yield i
392 return
392 return
393
393
394 seen = set(revs)
394 seen = set(revs)
395 for i in xrange(first + 1, len(self)):
395 for i in xrange(first + 1, len(self)):
396 for x in self.parentrevs(i):
396 for x in self.parentrevs(i):
397 if x != nullrev and x in seen:
397 if x != nullrev and x in seen:
398 seen.add(i)
398 seen.add(i)
399 yield i
399 yield i
400 break
400 break
401
401
402 def findcommonmissing(self, common=None, heads=None):
402 def findcommonmissing(self, common=None, heads=None):
403 """Return a tuple of the ancestors of common and the ancestors of heads
403 """Return a tuple of the ancestors of common and the ancestors of heads
404 that are not ancestors of common.
404 that are not ancestors of common.
405
405
406 More specifically, the second element is a list of nodes N such that
406 More specifically, the second element is a list of nodes N such that
407 every N satisfies the following constraints:
407 every N satisfies the following constraints:
408
408
409 1. N is an ancestor of some node in 'heads'
409 1. N is an ancestor of some node in 'heads'
410 2. N is not an ancestor of any node in 'common'
410 2. N is not an ancestor of any node in 'common'
411
411
412 The list is sorted by revision number, meaning it is
412 The list is sorted by revision number, meaning it is
413 topologically sorted.
413 topologically sorted.
414
414
415 'heads' and 'common' are both lists of node IDs. If heads is
415 'heads' and 'common' are both lists of node IDs. If heads is
416 not supplied, uses all of the revlog's heads. If common is not
416 not supplied, uses all of the revlog's heads. If common is not
417 supplied, uses nullid."""
417 supplied, uses nullid."""
418 if common is None:
418 if common is None:
419 common = [nullid]
419 common = [nullid]
420 if heads is None:
420 if heads is None:
421 heads = self.heads()
421 heads = self.heads()
422
422
423 common = [self.rev(n) for n in common]
423 common = [self.rev(n) for n in common]
424 heads = [self.rev(n) for n in heads]
424 heads = [self.rev(n) for n in heads]
425
425
426 # we want the ancestors, but inclusive
426 # we want the ancestors, but inclusive
427 has = set(self.ancestors(*common))
427 has = set(self.ancestors(*common))
428 has.add(nullrev)
428 has.add(nullrev)
429 has.update(common)
429 has.update(common)
430
430
431 # take all ancestors from heads that aren't in has
431 # take all ancestors from heads that aren't in has
432 missing = set()
432 missing = set()
433 visit = [r for r in heads if r not in has]
433 visit = [r for r in heads if r not in has]
434 while visit:
434 while visit:
435 r = visit.pop(0)
435 r = visit.pop(0)
436 if r in missing:
436 if r in missing:
437 continue
437 continue
438 else:
438 else:
439 missing.add(r)
439 missing.add(r)
440 for p in self.parentrevs(r):
440 for p in self.parentrevs(r):
441 if p not in has:
441 if p not in has:
442 visit.append(p)
442 visit.append(p)
443 missing = list(missing)
443 missing = list(missing)
444 missing.sort()
444 missing.sort()
445 return has, [self.node(r) for r in missing]
445 return has, [self.node(r) for r in missing]
446
446
447 def findmissing(self, common=None, heads=None):
447 def findmissing(self, common=None, heads=None):
448 """Return the ancestors of heads that are not ancestors of common.
448 """Return the ancestors of heads that are not ancestors of common.
449
449
450 More specifically, return a list of nodes N such that every N
450 More specifically, return a list of nodes N such that every N
451 satisfies the following constraints:
451 satisfies the following constraints:
452
452
453 1. N is an ancestor of some node in 'heads'
453 1. N is an ancestor of some node in 'heads'
454 2. N is not an ancestor of any node in 'common'
454 2. N is not an ancestor of any node in 'common'
455
455
456 The list is sorted by revision number, meaning it is
456 The list is sorted by revision number, meaning it is
457 topologically sorted.
457 topologically sorted.
458
458
459 'heads' and 'common' are both lists of node IDs. If heads is
459 'heads' and 'common' are both lists of node IDs. If heads is
460 not supplied, uses all of the revlog's heads. If common is not
460 not supplied, uses all of the revlog's heads. If common is not
461 supplied, uses nullid."""
461 supplied, uses nullid."""
462 _common, missing = self.findcommonmissing(common, heads)
462 _common, missing = self.findcommonmissing(common, heads)
463 return missing
463 return missing
464
464
465 def nodesbetween(self, roots=None, heads=None):
465 def nodesbetween(self, roots=None, heads=None):
466 """Return a topological path from 'roots' to 'heads'.
466 """Return a topological path from 'roots' to 'heads'.
467
467
468 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
468 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
469 topologically sorted list of all nodes N that satisfy both of
469 topologically sorted list of all nodes N that satisfy both of
470 these constraints:
470 these constraints:
471
471
472 1. N is a descendant of some node in 'roots'
472 1. N is a descendant of some node in 'roots'
473 2. N is an ancestor of some node in 'heads'
473 2. N is an ancestor of some node in 'heads'
474
474
475 Every node is considered to be both a descendant and an ancestor
475 Every node is considered to be both a descendant and an ancestor
476 of itself, so every reachable node in 'roots' and 'heads' will be
476 of itself, so every reachable node in 'roots' and 'heads' will be
477 included in 'nodes'.
477 included in 'nodes'.
478
478
479 'outroots' is the list of reachable nodes in 'roots', i.e., the
479 'outroots' is the list of reachable nodes in 'roots', i.e., the
480 subset of 'roots' that is returned in 'nodes'. Likewise,
480 subset of 'roots' that is returned in 'nodes'. Likewise,
481 'outheads' is the subset of 'heads' that is also in 'nodes'.
481 'outheads' is the subset of 'heads' that is also in 'nodes'.
482
482
483 'roots' and 'heads' are both lists of node IDs. If 'roots' is
483 'roots' and 'heads' are both lists of node IDs. If 'roots' is
484 unspecified, uses nullid as the only root. If 'heads' is
484 unspecified, uses nullid as the only root. If 'heads' is
485 unspecified, uses list of all of the revlog's heads."""
485 unspecified, uses list of all of the revlog's heads."""
486 nonodes = ([], [], [])
486 nonodes = ([], [], [])
487 if roots is not None:
487 if roots is not None:
488 roots = list(roots)
488 roots = list(roots)
489 if not roots:
489 if not roots:
490 return nonodes
490 return nonodes
491 lowestrev = min([self.rev(n) for n in roots])
491 lowestrev = min([self.rev(n) for n in roots])
492 else:
492 else:
493 roots = [nullid] # Everybody's a descendent of nullid
493 roots = [nullid] # Everybody's a descendent of nullid
494 lowestrev = nullrev
494 lowestrev = nullrev
495 if (lowestrev == nullrev) and (heads is None):
495 if (lowestrev == nullrev) and (heads is None):
496 # We want _all_ the nodes!
496 # We want _all_ the nodes!
497 return ([self.node(r) for r in self], [nullid], list(self.heads()))
497 return ([self.node(r) for r in self], [nullid], list(self.heads()))
498 if heads is None:
498 if heads is None:
499 # All nodes are ancestors, so the latest ancestor is the last
499 # All nodes are ancestors, so the latest ancestor is the last
500 # node.
500 # node.
501 highestrev = len(self) - 1
501 highestrev = len(self) - 1
502 # Set ancestors to None to signal that every node is an ancestor.
502 # Set ancestors to None to signal that every node is an ancestor.
503 ancestors = None
503 ancestors = None
504 # Set heads to an empty dictionary for later discovery of heads
504 # Set heads to an empty dictionary for later discovery of heads
505 heads = {}
505 heads = {}
506 else:
506 else:
507 heads = list(heads)
507 heads = list(heads)
508 if not heads:
508 if not heads:
509 return nonodes
509 return nonodes
510 ancestors = set()
510 ancestors = set()
511 # Turn heads into a dictionary so we can remove 'fake' heads.
511 # Turn heads into a dictionary so we can remove 'fake' heads.
512 # Also, later we will be using it to filter out the heads we can't
512 # Also, later we will be using it to filter out the heads we can't
513 # find from roots.
513 # find from roots.
514 heads = dict.fromkeys(heads, 0)
514 heads = dict.fromkeys(heads, 0)
515 # Start at the top and keep marking parents until we're done.
515 # Start at the top and keep marking parents until we're done.
516 nodestotag = set(heads)
516 nodestotag = set(heads)
517 # Remember where the top was so we can use it as a limit later.
517 # Remember where the top was so we can use it as a limit later.
518 highestrev = max([self.rev(n) for n in nodestotag])
518 highestrev = max([self.rev(n) for n in nodestotag])
519 while nodestotag:
519 while nodestotag:
520 # grab a node to tag
520 # grab a node to tag
521 n = nodestotag.pop()
521 n = nodestotag.pop()
522 # Never tag nullid
522 # Never tag nullid
523 if n == nullid:
523 if n == nullid:
524 continue
524 continue
525 # A node's revision number represents its place in a
525 # A node's revision number represents its place in a
526 # topologically sorted list of nodes.
526 # topologically sorted list of nodes.
527 r = self.rev(n)
527 r = self.rev(n)
528 if r >= lowestrev:
528 if r >= lowestrev:
529 if n not in ancestors:
529 if n not in ancestors:
530 # If we are possibly a descendent of one of the roots
530 # If we are possibly a descendent of one of the roots
531 # and we haven't already been marked as an ancestor
531 # and we haven't already been marked as an ancestor
532 ancestors.add(n) # Mark as ancestor
532 ancestors.add(n) # Mark as ancestor
533 # Add non-nullid parents to list of nodes to tag.
533 # Add non-nullid parents to list of nodes to tag.
534 nodestotag.update([p for p in self.parents(n) if
534 nodestotag.update([p for p in self.parents(n) if
535 p != nullid])
535 p != nullid])
536 elif n in heads: # We've seen it before, is it a fake head?
536 elif n in heads: # We've seen it before, is it a fake head?
537 # So it is, real heads should not be the ancestors of
537 # So it is, real heads should not be the ancestors of
538 # any other heads.
538 # any other heads.
539 heads.pop(n)
539 heads.pop(n)
540 if not ancestors:
540 if not ancestors:
541 return nonodes
541 return nonodes
542 # Now that we have our set of ancestors, we want to remove any
542 # Now that we have our set of ancestors, we want to remove any
543 # roots that are not ancestors.
543 # roots that are not ancestors.
544
544
545 # If one of the roots was nullid, everything is included anyway.
545 # If one of the roots was nullid, everything is included anyway.
546 if lowestrev > nullrev:
546 if lowestrev > nullrev:
547 # But, since we weren't, let's recompute the lowest rev to not
547 # But, since we weren't, let's recompute the lowest rev to not
548 # include roots that aren't ancestors.
548 # include roots that aren't ancestors.
549
549
550 # Filter out roots that aren't ancestors of heads
550 # Filter out roots that aren't ancestors of heads
551 roots = [n for n in roots if n in ancestors]
551 roots = [n for n in roots if n in ancestors]
552 # Recompute the lowest revision
552 # Recompute the lowest revision
553 if roots:
553 if roots:
554 lowestrev = min([self.rev(n) for n in roots])
554 lowestrev = min([self.rev(n) for n in roots])
555 else:
555 else:
556 # No more roots? Return empty list
556 # No more roots? Return empty list
557 return nonodes
557 return nonodes
558 else:
558 else:
559 # We are descending from nullid, and don't need to care about
559 # We are descending from nullid, and don't need to care about
560 # any other roots.
560 # any other roots.
561 lowestrev = nullrev
561 lowestrev = nullrev
562 roots = [nullid]
562 roots = [nullid]
563 # Transform our roots list into a set.
563 # Transform our roots list into a set.
564 descendents = set(roots)
564 descendents = set(roots)
565 # Also, keep the original roots so we can filter out roots that aren't
565 # Also, keep the original roots so we can filter out roots that aren't
566 # 'real' roots (i.e. are descended from other roots).
566 # 'real' roots (i.e. are descended from other roots).
567 roots = descendents.copy()
567 roots = descendents.copy()
568 # Our topologically sorted list of output nodes.
568 # Our topologically sorted list of output nodes.
569 orderedout = []
569 orderedout = []
570 # Don't start at nullid since we don't want nullid in our output list,
570 # Don't start at nullid since we don't want nullid in our output list,
571 # and if nullid shows up in descedents, empty parents will look like
571 # and if nullid shows up in descedents, empty parents will look like
572 # they're descendents.
572 # they're descendents.
573 for r in xrange(max(lowestrev, 0), highestrev + 1):
573 for r in xrange(max(lowestrev, 0), highestrev + 1):
574 n = self.node(r)
574 n = self.node(r)
575 isdescendent = False
575 isdescendent = False
576 if lowestrev == nullrev: # Everybody is a descendent of nullid
576 if lowestrev == nullrev: # Everybody is a descendent of nullid
577 isdescendent = True
577 isdescendent = True
578 elif n in descendents:
578 elif n in descendents:
579 # n is already a descendent
579 # n is already a descendent
580 isdescendent = True
580 isdescendent = True
581 # This check only needs to be done here because all the roots
581 # This check only needs to be done here because all the roots
582 # will start being marked is descendents before the loop.
582 # will start being marked is descendents before the loop.
583 if n in roots:
583 if n in roots:
584 # If n was a root, check if it's a 'real' root.
584 # If n was a root, check if it's a 'real' root.
585 p = tuple(self.parents(n))
585 p = tuple(self.parents(n))
586 # If any of its parents are descendents, it's not a root.
586 # If any of its parents are descendents, it's not a root.
587 if (p[0] in descendents) or (p[1] in descendents):
587 if (p[0] in descendents) or (p[1] in descendents):
588 roots.remove(n)
588 roots.remove(n)
589 else:
589 else:
590 p = tuple(self.parents(n))
590 p = tuple(self.parents(n))
591 # A node is a descendent if either of its parents are
591 # A node is a descendent if either of its parents are
592 # descendents. (We seeded the dependents list with the roots
592 # descendents. (We seeded the dependents list with the roots
593 # up there, remember?)
593 # up there, remember?)
594 if (p[0] in descendents) or (p[1] in descendents):
594 if (p[0] in descendents) or (p[1] in descendents):
595 descendents.add(n)
595 descendents.add(n)
596 isdescendent = True
596 isdescendent = True
597 if isdescendent and ((ancestors is None) or (n in ancestors)):
597 if isdescendent and ((ancestors is None) or (n in ancestors)):
598 # Only include nodes that are both descendents and ancestors.
598 # Only include nodes that are both descendents and ancestors.
599 orderedout.append(n)
599 orderedout.append(n)
600 if (ancestors is not None) and (n in heads):
600 if (ancestors is not None) and (n in heads):
601 # We're trying to figure out which heads are reachable
601 # We're trying to figure out which heads are reachable
602 # from roots.
602 # from roots.
603 # Mark this head as having been reached
603 # Mark this head as having been reached
604 heads[n] = 1
604 heads[n] = 1
605 elif ancestors is None:
605 elif ancestors is None:
606 # Otherwise, we're trying to discover the heads.
606 # Otherwise, we're trying to discover the heads.
607 # Assume this is a head because if it isn't, the next step
607 # Assume this is a head because if it isn't, the next step
608 # will eventually remove it.
608 # will eventually remove it.
609 heads[n] = 1
609 heads[n] = 1
610 # But, obviously its parents aren't.
610 # But, obviously its parents aren't.
611 for p in self.parents(n):
611 for p in self.parents(n):
612 heads.pop(p, None)
612 heads.pop(p, None)
613 heads = [n for n in heads.iterkeys() if heads[n] != 0]
613 heads = [n for n in heads.iterkeys() if heads[n] != 0]
614 roots = list(roots)
614 roots = list(roots)
615 assert orderedout
615 assert orderedout
616 assert roots
616 assert roots
617 assert heads
617 assert heads
618 return (orderedout, roots, heads)
618 return (orderedout, roots, heads)
619
619
620 def heads(self, start=None, stop=None):
620 def heads(self, start=None, stop=None):
621 """return the list of all nodes that have no children
621 """return the list of all nodes that have no children
622
622
623 if start is specified, only heads that are descendants of
623 if start is specified, only heads that are descendants of
624 start will be returned
624 start will be returned
625 if stop is specified, it will consider all the revs from stop
625 if stop is specified, it will consider all the revs from stop
626 as if they had no children
626 as if they had no children
627 """
627 """
628 if start is None and stop is None:
628 if start is None and stop is None:
629 count = len(self)
629 count = len(self)
630 if not count:
630 if not count:
631 return [nullid]
631 return [nullid]
632 ishead = [1] * (count + 1)
632 ishead = [1] * (count + 1)
633 index = self.index
633 index = self.index
634 for r in xrange(count):
634 for r in xrange(count):
635 e = index[r]
635 e = index[r]
636 ishead[e[5]] = ishead[e[6]] = 0
636 ishead[e[5]] = ishead[e[6]] = 0
637 return [self.node(r) for r in xrange(count) if ishead[r]]
637 return [self.node(r) for r in xrange(count) if ishead[r]]
638
638
639 if start is None:
639 if start is None:
640 start = nullid
640 start = nullid
641 if stop is None:
641 if stop is None:
642 stop = []
642 stop = []
643 stoprevs = set([self.rev(n) for n in stop])
643 stoprevs = set([self.rev(n) for n in stop])
644 startrev = self.rev(start)
644 startrev = self.rev(start)
645 reachable = set((startrev,))
645 reachable = set((startrev,))
646 heads = set((startrev,))
646 heads = set((startrev,))
647
647
648 parentrevs = self.parentrevs
648 parentrevs = self.parentrevs
649 for r in xrange(startrev + 1, len(self)):
649 for r in xrange(startrev + 1, len(self)):
650 for p in parentrevs(r):
650 for p in parentrevs(r):
651 if p in reachable:
651 if p in reachable:
652 if r not in stoprevs:
652 if r not in stoprevs:
653 reachable.add(r)
653 reachable.add(r)
654 heads.add(r)
654 heads.add(r)
655 if p in heads and p not in stoprevs:
655 if p in heads and p not in stoprevs:
656 heads.remove(p)
656 heads.remove(p)
657
657
658 return [self.node(r) for r in heads]
658 return [self.node(r) for r in heads]
659
659
660 def children(self, node):
660 def children(self, node):
661 """find the children of a given node"""
661 """find the children of a given node"""
662 c = []
662 c = []
663 p = self.rev(node)
663 p = self.rev(node)
664 for r in range(p + 1, len(self)):
664 for r in range(p + 1, len(self)):
665 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
665 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
666 if prevs:
666 if prevs:
667 for pr in prevs:
667 for pr in prevs:
668 if pr == p:
668 if pr == p:
669 c.append(self.node(r))
669 c.append(self.node(r))
670 elif p == nullrev:
670 elif p == nullrev:
671 c.append(self.node(r))
671 c.append(self.node(r))
672 return c
672 return c
673
673
674 def descendant(self, start, end):
674 def descendant(self, start, end):
675 if start == nullrev:
675 if start == nullrev:
676 return True
676 return True
677 for i in self.descendants(start):
677 for i in self.descendants(start):
678 if i == end:
678 if i == end:
679 return True
679 return True
680 elif i > end:
680 elif i > end:
681 break
681 break
682 return False
682 return False
683
683
684 def ancestor(self, a, b):
684 def ancestor(self, a, b):
685 """calculate the least common ancestor of nodes a and b"""
685 """calculate the least common ancestor of nodes a and b"""
686
686
687 # fast path, check if it is a descendant
687 # fast path, check if it is a descendant
688 a, b = self.rev(a), self.rev(b)
688 a, b = self.rev(a), self.rev(b)
689 start, end = sorted((a, b))
689 start, end = sorted((a, b))
690 if self.descendant(start, end):
690 if self.descendant(start, end):
691 return self.node(start)
691 return self.node(start)
692
692
693 def parents(rev):
693 def parents(rev):
694 return [p for p in self.parentrevs(rev) if p != nullrev]
694 return [p for p in self.parentrevs(rev) if p != nullrev]
695
695
696 c = ancestor.ancestor(a, b, parents)
696 c = ancestor.ancestor(a, b, parents)
697 if c is None:
697 if c is None:
698 return nullid
698 return nullid
699
699
700 return self.node(c)
700 return self.node(c)
701
701
702 def _match(self, id):
702 def _match(self, id):
703 if isinstance(id, (long, int)):
703 if isinstance(id, (long, int)):
704 # rev
704 # rev
705 return self.node(id)
705 return self.node(id)
706 if len(id) == 20:
706 if len(id) == 20:
707 # possibly a binary node
707 # possibly a binary node
708 # odds of a binary node being all hex in ASCII are 1 in 10**25
708 # odds of a binary node being all hex in ASCII are 1 in 10**25
709 try:
709 try:
710 node = id
710 node = id
711 self.rev(node) # quick search the index
711 self.rev(node) # quick search the index
712 return node
712 return node
713 except LookupError:
713 except LookupError:
714 pass # may be partial hex id
714 pass # may be partial hex id
715 try:
715 try:
716 # str(rev)
716 # str(rev)
717 rev = int(id)
717 rev = int(id)
718 if str(rev) != id:
718 if str(rev) != id:
719 raise ValueError
719 raise ValueError
720 if rev < 0:
720 if rev < 0:
721 rev = len(self) + rev
721 rev = len(self) + rev
722 if rev < 0 or rev >= len(self):
722 if rev < 0 or rev >= len(self):
723 raise ValueError
723 raise ValueError
724 return self.node(rev)
724 return self.node(rev)
725 except (ValueError, OverflowError):
725 except (ValueError, OverflowError):
726 pass
726 pass
727 if len(id) == 40:
727 if len(id) == 40:
728 try:
728 try:
729 # a full hex nodeid?
729 # a full hex nodeid?
730 node = bin(id)
730 node = bin(id)
731 self.rev(node)
731 self.rev(node)
732 return node
732 return node
733 except (TypeError, LookupError):
733 except (TypeError, LookupError):
734 pass
734 pass
735
735
736 def _partialmatch(self, id):
736 def _partialmatch(self, id):
737 if id in self._pcache:
737 if id in self._pcache:
738 return self._pcache[id]
738 return self._pcache[id]
739
739
740 if len(id) < 40:
740 if len(id) < 40:
741 try:
741 try:
742 # hex(node)[:...]
742 # hex(node)[:...]
743 l = len(id) // 2 # grab an even number of digits
743 l = len(id) // 2 # grab an even number of digits
744 prefix = bin(id[:l * 2])
744 prefix = bin(id[:l * 2])
745 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
745 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
746 nl = [n for n in nl if hex(n).startswith(id)]
746 nl = [n for n in nl if hex(n).startswith(id)]
747 if len(nl) > 0:
747 if len(nl) > 0:
748 if len(nl) == 1:
748 if len(nl) == 1:
749 self._pcache[id] = nl[0]
749 self._pcache[id] = nl[0]
750 return nl[0]
750 return nl[0]
751 raise LookupError(id, self.indexfile,
751 raise LookupError(id, self.indexfile,
752 _('ambiguous identifier'))
752 _('ambiguous identifier'))
753 return None
753 return None
754 except TypeError:
754 except TypeError:
755 pass
755 pass
756
756
757 def lookup(self, id):
757 def lookup(self, id):
758 """locate a node based on:
758 """locate a node based on:
759 - revision number or str(revision number)
759 - revision number or str(revision number)
760 - nodeid or subset of hex nodeid
760 - nodeid or subset of hex nodeid
761 """
761 """
762 n = self._match(id)
762 n = self._match(id)
763 if n is not None:
763 if n is not None:
764 return n
764 return n
765 n = self._partialmatch(id)
765 n = self._partialmatch(id)
766 if n:
766 if n:
767 return n
767 return n
768
768
769 raise LookupError(id, self.indexfile, _('no match found'))
769 raise LookupError(id, self.indexfile, _('no match found'))
770
770
771 def cmp(self, node, text):
771 def cmp(self, node, text):
772 """compare text with a given file revision
772 """compare text with a given file revision
773
773
774 returns True if text is different than what is stored.
774 returns True if text is different than what is stored.
775 """
775 """
776 p1, p2 = self.parents(node)
776 p1, p2 = self.parents(node)
777 return hash(text, p1, p2) != node
777 return hash(text, p1, p2) != node
778
778
779 def _addchunk(self, offset, data):
779 def _addchunk(self, offset, data):
780 o, d = self._chunkcache
780 o, d = self._chunkcache
781 # try to add to existing cache
781 # try to add to existing cache
782 if o + len(d) == offset and len(d) + len(data) < _chunksize:
782 if o + len(d) == offset and len(d) + len(data) < _chunksize:
783 self._chunkcache = o, d + data
783 self._chunkcache = o, d + data
784 else:
784 else:
785 self._chunkcache = offset, data
785 self._chunkcache = offset, data
786
786
787 def _loadchunk(self, offset, length):
787 def _loadchunk(self, offset, length):
788 if self._inline:
788 if self._inline:
789 df = self.opener(self.indexfile)
789 df = self.opener(self.indexfile)
790 else:
790 else:
791 df = self.opener(self.datafile)
791 df = self.opener(self.datafile)
792
792
793 readahead = max(65536, length)
793 readahead = max(65536, length)
794 df.seek(offset)
794 df.seek(offset)
795 d = df.read(readahead)
795 d = df.read(readahead)
796 self._addchunk(offset, d)
796 self._addchunk(offset, d)
797 if readahead > length:
797 if readahead > length:
798 return d[:length]
798 return d[:length]
799 return d
799 return d
800
800
801 def _getchunk(self, offset, length):
801 def _getchunk(self, offset, length):
802 o, d = self._chunkcache
802 o, d = self._chunkcache
803 l = len(d)
803 l = len(d)
804
804
805 # is it in the cache?
805 # is it in the cache?
806 cachestart = offset - o
806 cachestart = offset - o
807 cacheend = cachestart + length
807 cacheend = cachestart + length
808 if cachestart >= 0 and cacheend <= l:
808 if cachestart >= 0 and cacheend <= l:
809 if cachestart == 0 and cacheend == l:
809 if cachestart == 0 and cacheend == l:
810 return d # avoid a copy
810 return d # avoid a copy
811 return d[cachestart:cacheend]
811 return d[cachestart:cacheend]
812
812
813 return self._loadchunk(offset, length)
813 return self._loadchunk(offset, length)
814
814
815 def _chunkraw(self, startrev, endrev):
815 def _chunkraw(self, startrev, endrev):
816 start = self.start(startrev)
816 start = self.start(startrev)
817 length = self.end(endrev) - start
817 length = self.end(endrev) - start
818 if self._inline:
818 if self._inline:
819 start += (startrev + 1) * self._io.size
819 start += (startrev + 1) * self._io.size
820 return self._getchunk(start, length)
820 return self._getchunk(start, length)
821
821
822 def _chunk(self, rev):
822 def _chunk(self, rev):
823 return decompress(self._chunkraw(rev, rev))
823 return decompress(self._chunkraw(rev, rev))
824
824
825 def _chunkclear(self):
825 def _chunkclear(self):
826 self._chunkcache = (0, '')
826 self._chunkcache = (0, '')
827
827
828 def deltaparent(self, rev):
828 def deltaparent(self, rev):
829 """return previous revision or parentrev according to flags"""
829 """return previous revision or parentrev according to flags"""
830 if self.flags(rev) & REVIDX_PARENTDELTA:
830 if self.flags(rev) & REVIDX_PARENTDELTA:
831 return self.parentrevs(rev)[0]
831 return self.parentrevs(rev)[0]
832 else:
832 else:
833 return rev - 1
833 return rev - 1
834
834
835 def revdiff(self, rev1, rev2):
835 def revdiff(self, rev1, rev2):
836 """return or calculate a delta between two revisions"""
836 """return or calculate a delta between two revisions"""
837 if self.base(rev2) != rev2 and self.deltaparent(rev2) == rev1:
837 if self.base(rev2) != rev2 and self.deltaparent(rev2) == rev1:
838 return self._chunk(rev2)
838 return self._chunk(rev2)
839
839
840 return mdiff.textdiff(self.revision(self.node(rev1)),
840 return mdiff.textdiff(self.revision(self.node(rev1)),
841 self.revision(self.node(rev2)))
841 self.revision(self.node(rev2)))
842
842
843 def revision(self, node):
843 def revision(self, node):
844 """return an uncompressed revision of a given node"""
844 """return an uncompressed revision of a given node"""
845 cachedrev = None
845 cachedrev = None
846 if node == nullid:
846 if node == nullid:
847 return ""
847 return ""
848 if self._cache:
848 if self._cache:
849 if self._cache[0] == node:
849 if self._cache[0] == node:
850 return self._cache[2]
850 return self._cache[2]
851 cachedrev = self._cache[1]
851 cachedrev = self._cache[1]
852
852
853 # look up what we need to read
853 # look up what we need to read
854 text = None
854 text = None
855 rev = self.rev(node)
855 rev = self.rev(node)
856 base = self.base(rev)
856 base = self.base(rev)
857
857
858 # check rev flags
858 # check rev flags
859 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
859 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
860 raise RevlogError(_('incompatible revision flag %x') %
860 raise RevlogError(_('incompatible revision flag %x') %
861 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
861 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
862
862
863 # build delta chain
863 # build delta chain
864 chain = []
864 chain = []
865 index = self.index # for performance
865 index = self.index # for performance
866 iterrev = rev
866 iterrev = rev
867 e = index[iterrev]
867 e = index[iterrev]
868 while iterrev != base and iterrev != cachedrev:
868 while iterrev != base and iterrev != cachedrev:
869 chain.append(iterrev)
869 chain.append(iterrev)
870 if e[0] & REVIDX_PARENTDELTA:
870 if e[0] & REVIDX_PARENTDELTA:
871 iterrev = e[5]
871 iterrev = e[5]
872 else:
872 else:
873 iterrev -= 1
873 iterrev -= 1
874 e = index[iterrev]
874 e = index[iterrev]
875 chain.reverse()
875 chain.reverse()
876 base = iterrev
876 base = iterrev
877
877
878 if iterrev == cachedrev:
878 if iterrev == cachedrev:
879 # cache hit
879 # cache hit
880 text = self._cache[2]
880 text = self._cache[2]
881
881
882 # drop cache to save memory
882 # drop cache to save memory
883 self._cache = None
883 self._cache = None
884
884
885 self._chunkraw(base, rev)
885 self._chunkraw(base, rev)
886 if text is None:
886 if text is None:
887 text = self._chunk(base)
887 text = self._chunk(base)
888
888
889 bins = [self._chunk(r) for r in chain]
889 bins = [self._chunk(r) for r in chain]
890 text = mdiff.patches(text, bins)
890 text = mdiff.patches(text, bins)
891
891
892 text = self._checkhash(text, node, rev)
892 text = self._checkhash(text, node, rev)
893
893
894 self._cache = (node, rev, text)
894 self._cache = (node, rev, text)
895 return text
895 return text
896
896
897 def _checkhash(self, text, node, rev):
897 def _checkhash(self, text, node, rev):
898 p1, p2 = self.parents(node)
898 p1, p2 = self.parents(node)
899 if (node != hash(text, p1, p2) and
899 if (node != hash(text, p1, p2) and
900 not (self.flags(rev) & REVIDX_PUNCHED_FLAG)):
900 not (self.flags(rev) & REVIDX_PUNCHED_FLAG)):
901 raise RevlogError(_("integrity check failed on %s:%d")
901 raise RevlogError(_("integrity check failed on %s:%d")
902 % (self.indexfile, rev))
902 % (self.indexfile, rev))
903 return text
903 return text
904
904
905 def checkinlinesize(self, tr, fp=None):
905 def checkinlinesize(self, tr, fp=None):
906 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
906 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
907 return
907 return
908
908
909 trinfo = tr.find(self.indexfile)
909 trinfo = tr.find(self.indexfile)
910 if trinfo is None:
910 if trinfo is None:
911 raise RevlogError(_("%s not found in the transaction")
911 raise RevlogError(_("%s not found in the transaction")
912 % self.indexfile)
912 % self.indexfile)
913
913
914 trindex = trinfo[2]
914 trindex = trinfo[2]
915 dataoff = self.start(trindex)
915 dataoff = self.start(trindex)
916
916
917 tr.add(self.datafile, dataoff)
917 tr.add(self.datafile, dataoff)
918
918
919 if fp:
919 if fp:
920 fp.flush()
920 fp.flush()
921 fp.close()
921 fp.close()
922
922
923 df = self.opener(self.datafile, 'w')
923 df = self.opener(self.datafile, 'w')
924 try:
924 try:
925 for r in self:
925 for r in self:
926 df.write(self._chunkraw(r, r))
926 df.write(self._chunkraw(r, r))
927 finally:
927 finally:
928 df.close()
928 df.close()
929
929
930 fp = self.opener(self.indexfile, 'w', atomictemp=True)
930 fp = self.opener(self.indexfile, 'w', atomictemp=True)
931 self.version &= ~(REVLOGNGINLINEDATA)
931 self.version &= ~(REVLOGNGINLINEDATA)
932 self._inline = False
932 self._inline = False
933 for i in self:
933 for i in self:
934 e = self._io.packentry(self.index[i], self.node, self.version, i)
934 e = self._io.packentry(self.index[i], self.node, self.version, i)
935 fp.write(e)
935 fp.write(e)
936
936
937 # if we don't call rename, the temp file will never replace the
937 # if we don't call rename, the temp file will never replace the
938 # real index
938 # real index
939 fp.rename()
939 fp.rename()
940
940
941 tr.replace(self.indexfile, trindex * self._io.size)
941 tr.replace(self.indexfile, trindex * self._io.size)
942 self._chunkclear()
942 self._chunkclear()
943
943
944 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None):
944 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None):
945 """add a revision to the log
945 """add a revision to the log
946
946
947 text - the revision data to add
947 text - the revision data to add
948 transaction - the transaction object used for rollback
948 transaction - the transaction object used for rollback
949 link - the linkrev data to add
949 link - the linkrev data to add
950 p1, p2 - the parent nodeids of the revision
950 p1, p2 - the parent nodeids of the revision
951 cachedelta - an optional precomputed delta
951 cachedelta - an optional precomputed delta
952 """
952 """
953 node = hash(text, p1, p2)
953 node = hash(text, p1, p2)
954 if (node in self.nodemap and
954 if (node in self.nodemap and
955 (not self.flags(self.rev(node)) & REVIDX_PUNCHED_FLAG)):
955 (not self.flags(self.rev(node)) & REVIDX_PUNCHED_FLAG)):
956 return node
956 return node
957
957
958 dfh = None
958 dfh = None
959 if not self._inline:
959 if not self._inline:
960 dfh = self.opener(self.datafile, "a")
960 dfh = self.opener(self.datafile, "a")
961 ifh = self.opener(self.indexfile, "a+")
961 ifh = self.opener(self.indexfile, "a+")
962 try:
962 try:
963 return self._addrevision(node, text, transaction, link, p1, p2,
963 return self._addrevision(node, text, transaction, link, p1, p2,
964 cachedelta, ifh, dfh)
964 cachedelta, ifh, dfh)
965 finally:
965 finally:
966 if dfh:
966 if dfh:
967 dfh.close()
967 dfh.close()
968 ifh.close()
968 ifh.close()
969
969
970 def _addrevision(self, node, text, transaction, link, p1, p2,
970 def _addrevision(self, node, text, transaction, link, p1, p2,
971 cachedelta, ifh, dfh):
971 cachedelta, ifh, dfh):
972
972
973 btext = [text]
973 btext = [text]
974 def buildtext():
974 def buildtext():
975 if btext[0] is not None:
975 if btext[0] is not None:
976 return btext[0]
976 return btext[0]
977 # flush any pending writes here so we can read it in revision
977 # flush any pending writes here so we can read it in revision
978 if dfh:
978 if dfh:
979 dfh.flush()
979 dfh.flush()
980 ifh.flush()
980 ifh.flush()
981 basetext = self.revision(self.node(cachedelta[0]))
981 basetext = self.revision(self.node(cachedelta[0]))
982 btext[0] = mdiff.patch(basetext, cachedelta[1])
982 btext[0] = mdiff.patch(basetext, cachedelta[1])
983 chk = hash(btext[0], p1, p2)
983 chk = hash(btext[0], p1, p2)
984 if chk != node:
984 if chk != node:
985 raise RevlogError(_("consistency error in delta"))
985 raise RevlogError(_("consistency error in delta"))
986 return btext[0]
986 return btext[0]
987
987
988 def builddelta(rev):
988 def builddelta(rev):
989 # can we use the cached delta?
989 # can we use the cached delta?
990 if cachedelta and cachedelta[0] == rev:
990 if cachedelta and cachedelta[0] == rev:
991 delta = cachedelta[1]
991 delta = cachedelta[1]
992 else:
992 else:
993 t = buildtext()
993 t = buildtext()
994 ptext = self.revision(self.node(rev))
994 ptext = self.revision(self.node(rev))
995 delta = mdiff.textdiff(ptext, t)
995 delta = mdiff.textdiff(ptext, t)
996 data = compress(delta)
996 data = compress(delta)
997 l = len(data[1]) + len(data[0])
997 l = len(data[1]) + len(data[0])
998 base = self.base(rev)
998 base = self.base(rev)
999 dist = l + offset - self.start(base)
999 dist = l + offset - self.start(base)
1000 return dist, l, data, base
1000 return dist, l, data, base
1001
1001
1002 curr = len(self)
1002 curr = len(self)
1003 prev = curr - 1
1003 prev = curr - 1
1004 base = curr
1004 base = curr
1005 offset = self.end(prev)
1005 offset = self.end(prev)
1006 flags = 0
1006 flags = 0
1007 d = None
1007 d = None
1008 p1r, p2r = self.rev(p1), self.rev(p2)
1008 p1r, p2r = self.rev(p1), self.rev(p2)
1009
1009
1010 # should we try to build a delta?
1010 # should we try to build a delta?
1011 if prev != nullrev:
1011 if prev != nullrev:
1012 d = builddelta(prev)
1012 d = builddelta(prev)
1013 if self._parentdelta and prev != p1r:
1013 if self._parentdelta and prev != p1r:
1014 d2 = builddelta(p1r)
1014 d2 = builddelta(p1r)
1015 if d2 < d:
1015 if d2 < d:
1016 d = d2
1016 d = d2
1017 flags = REVIDX_PARENTDELTA
1017 flags = REVIDX_PARENTDELTA
1018 dist, l, data, base = d
1018 dist, l, data, base = d
1019
1019
1020 # full versions are inserted when the needed deltas
1020 # full versions are inserted when the needed deltas
1021 # become comparable to the uncompressed text
1021 # become comparable to the uncompressed text
1022 # or the base revision is punched
1022 # or the base revision is punched
1023 if text is None:
1023 if text is None:
1024 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1024 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1025 cachedelta[1])
1025 cachedelta[1])
1026 else:
1026 else:
1027 textlen = len(text)
1027 textlen = len(text)
1028 if (d is None or dist > textlen * 2 or
1028 if (d is None or dist > textlen * 2 or
1029 (self.flags(base) & REVIDX_PUNCHED_FLAG)):
1029 (self.flags(base) & REVIDX_PUNCHED_FLAG)):
1030 text = buildtext()
1030 text = buildtext()
1031 data = compress(text)
1031 data = compress(text)
1032 l = len(data[1]) + len(data[0])
1032 l = len(data[1]) + len(data[0])
1033 base = curr
1033 base = curr
1034
1034
1035 e = (offset_type(offset, flags), l, textlen,
1035 e = (offset_type(offset, flags), l, textlen,
1036 base, link, p1r, p2r, node)
1036 base, link, p1r, p2r, node)
1037 self.index.insert(-1, e)
1037 self.index.insert(-1, e)
1038 self.nodemap[node] = curr
1038 self.nodemap[node] = curr
1039
1039
1040 entry = self._io.packentry(e, self.node, self.version, curr)
1040 entry = self._io.packentry(e, self.node, self.version, curr)
1041 if not self._inline:
1041 if not self._inline:
1042 transaction.add(self.datafile, offset)
1042 transaction.add(self.datafile, offset)
1043 transaction.add(self.indexfile, curr * len(entry))
1043 transaction.add(self.indexfile, curr * len(entry))
1044 if data[0]:
1044 if data[0]:
1045 dfh.write(data[0])
1045 dfh.write(data[0])
1046 dfh.write(data[1])
1046 dfh.write(data[1])
1047 dfh.flush()
1047 dfh.flush()
1048 ifh.write(entry)
1048 ifh.write(entry)
1049 else:
1049 else:
1050 offset += curr * self._io.size
1050 offset += curr * self._io.size
1051 transaction.add(self.indexfile, offset, curr)
1051 transaction.add(self.indexfile, offset, curr)
1052 ifh.write(entry)
1052 ifh.write(entry)
1053 ifh.write(data[0])
1053 ifh.write(data[0])
1054 ifh.write(data[1])
1054 ifh.write(data[1])
1055 self.checkinlinesize(transaction, ifh)
1055 self.checkinlinesize(transaction, ifh)
1056
1056
1057 if type(text) == str: # only accept immutable objects
1057 if type(text) == str: # only accept immutable objects
1058 self._cache = (node, curr, text)
1058 self._cache = (node, curr, text)
1059 return node
1059 return node
1060
1060
1061 def group(self, nodelist, lookup):
1061 def group(self, nodelist, lookup):
1062 """Calculate a delta group, yielding a sequence of changegroup chunks
1062 """Calculate a delta group, yielding a sequence of changegroup chunks
1063 (strings).
1063 (strings).
1064
1064
1065 Given a list of changeset revs, return a set of deltas and
1065 Given a list of changeset revs, return a set of deltas and
1066 metadata corresponding to nodes. The first delta is
1066 metadata corresponding to nodes. The first delta is
1067 first parent(nodelist[0]) -> nodelist[0], the receiver is
1067 first parent(nodelist[0]) -> nodelist[0], the receiver is
1068 guaranteed to have this parent as it has all history before
1068 guaranteed to have this parent as it has all history before
1069 these changesets. In the case firstparent is nullrev the
1069 these changesets. In the case firstparent is nullrev the
1070 changegroup starts with a full revision.
1070 changegroup starts with a full revision.
1071 """
1071 """
1072
1072
1073 revs = [self.rev(n) for n in nodelist]
1073 revs = sorted([self.rev(n) for n in nodelist])
1074
1074
1075 # if we don't have any revisions touched by these changesets, bail
1075 # if we don't have any revisions touched by these changesets, bail
1076 if not revs:
1076 if not revs:
1077 yield changegroup.closechunk()
1077 yield changegroup.closechunk()
1078 return
1078 return
1079
1079
1080 # add the parent of the first rev
1080 # add the parent of the first rev
1081 p = self.parentrevs(revs[0])[0]
1081 p = self.parentrevs(revs[0])[0]
1082 revs.insert(0, p)
1082 revs.insert(0, p)
1083
1083
1084 # build deltas
1084 # build deltas
1085 for r in xrange(len(revs) - 1):
1085 for r in xrange(len(revs) - 1):
1086 a, b = revs[r], revs[r + 1]
1086 a, b = revs[r], revs[r + 1]
1087 nb = self.node(b)
1087 nb = self.node(b)
1088
1088
1089 p = self.parents(nb)
1089 p = self.parents(nb)
1090 meta = nb + p[0] + p[1] + lookup(nb)
1090 meta = nb + p[0] + p[1] + lookup(nb)
1091 if a == nullrev:
1091 if a == nullrev:
1092 d = self.revision(nb)
1092 d = self.revision(nb)
1093 meta += mdiff.trivialdiffheader(len(d))
1093 meta += mdiff.trivialdiffheader(len(d))
1094 else:
1094 else:
1095 d = self.revdiff(a, b)
1095 d = self.revdiff(a, b)
1096 yield changegroup.chunkheader(len(meta) + len(d))
1096 yield changegroup.chunkheader(len(meta) + len(d))
1097 yield meta
1097 yield meta
1098 yield d
1098 yield d
1099
1099
1100 yield changegroup.closechunk()
1100 yield changegroup.closechunk()
1101
1101
1102 def addgroup(self, bundle, linkmapper, transaction):
1102 def addgroup(self, bundle, linkmapper, transaction):
1103 """
1103 """
1104 add a delta group
1104 add a delta group
1105
1105
1106 given a set of deltas, add them to the revision log. the
1106 given a set of deltas, add them to the revision log. the
1107 first delta is against its parent, which should be in our
1107 first delta is against its parent, which should be in our
1108 log, the rest are against the previous delta.
1108 log, the rest are against the previous delta.
1109 """
1109 """
1110
1110
1111 # track the base of the current delta log
1111 # track the base of the current delta log
1112 node = None
1112 node = None
1113
1113
1114 r = len(self)
1114 r = len(self)
1115 end = 0
1115 end = 0
1116 if r:
1116 if r:
1117 end = self.end(r - 1)
1117 end = self.end(r - 1)
1118 ifh = self.opener(self.indexfile, "a+")
1118 ifh = self.opener(self.indexfile, "a+")
1119 isize = r * self._io.size
1119 isize = r * self._io.size
1120 if self._inline:
1120 if self._inline:
1121 transaction.add(self.indexfile, end + isize, r)
1121 transaction.add(self.indexfile, end + isize, r)
1122 dfh = None
1122 dfh = None
1123 else:
1123 else:
1124 transaction.add(self.indexfile, isize, r)
1124 transaction.add(self.indexfile, isize, r)
1125 transaction.add(self.datafile, end)
1125 transaction.add(self.datafile, end)
1126 dfh = self.opener(self.datafile, "a")
1126 dfh = self.opener(self.datafile, "a")
1127
1127
1128 try:
1128 try:
1129 # loop through our set of deltas
1129 # loop through our set of deltas
1130 chain = None
1130 chain = None
1131 while 1:
1131 while 1:
1132 chunkdata = bundle.parsechunk()
1132 chunkdata = bundle.parsechunk()
1133 if not chunkdata:
1133 if not chunkdata:
1134 break
1134 break
1135 node = chunkdata['node']
1135 node = chunkdata['node']
1136 p1 = chunkdata['p1']
1136 p1 = chunkdata['p1']
1137 p2 = chunkdata['p2']
1137 p2 = chunkdata['p2']
1138 cs = chunkdata['cs']
1138 cs = chunkdata['cs']
1139 delta = chunkdata['data']
1139 delta = chunkdata['data']
1140
1140
1141 link = linkmapper(cs)
1141 link = linkmapper(cs)
1142 if (node in self.nodemap and
1142 if (node in self.nodemap and
1143 (not self.flags(self.rev(node)) & REVIDX_PUNCHED_FLAG)):
1143 (not self.flags(self.rev(node)) & REVIDX_PUNCHED_FLAG)):
1144 # this can happen if two branches make the same change
1144 # this can happen if two branches make the same change
1145 chain = node
1145 chain = node
1146 continue
1146 continue
1147
1147
1148 for p in (p1, p2):
1148 for p in (p1, p2):
1149 if not p in self.nodemap:
1149 if not p in self.nodemap:
1150 if self._shallow:
1150 if self._shallow:
1151 # add null entries for missing parents
1151 # add null entries for missing parents
1152 # XXX FIXME
1152 # XXX FIXME
1153 #if base == nullrev:
1153 #if base == nullrev:
1154 # base = len(self)
1154 # base = len(self)
1155 #e = (offset_type(end, REVIDX_PUNCHED_FLAG),
1155 #e = (offset_type(end, REVIDX_PUNCHED_FLAG),
1156 # 0, 0, base, nullrev, nullrev, nullrev, p)
1156 # 0, 0, base, nullrev, nullrev, nullrev, p)
1157 #self.index.insert(-1, e)
1157 #self.index.insert(-1, e)
1158 #self.nodemap[p] = r
1158 #self.nodemap[p] = r
1159 #entry = self._io.packentry(e, self.node,
1159 #entry = self._io.packentry(e, self.node,
1160 # self.version, r)
1160 # self.version, r)
1161 #ifh.write(entry)
1161 #ifh.write(entry)
1162 #t, r = r, r + 1
1162 #t, r = r, r + 1
1163 raise LookupError(p, self.indexfile,
1163 raise LookupError(p, self.indexfile,
1164 _('unknown parent'))
1164 _('unknown parent'))
1165 else:
1165 else:
1166 raise LookupError(p, self.indexfile,
1166 raise LookupError(p, self.indexfile,
1167 _('unknown parent'))
1167 _('unknown parent'))
1168
1168
1169 if not chain:
1169 if not chain:
1170 # retrieve the parent revision of the delta chain
1170 # retrieve the parent revision of the delta chain
1171 chain = p1
1171 chain = p1
1172 if not chain in self.nodemap:
1172 if not chain in self.nodemap:
1173 raise LookupError(chain, self.indexfile, _('unknown base'))
1173 raise LookupError(chain, self.indexfile, _('unknown base'))
1174
1174
1175 chainrev = self.rev(chain)
1175 chainrev = self.rev(chain)
1176 chain = self._addrevision(node, None, transaction, link,
1176 chain = self._addrevision(node, None, transaction, link,
1177 p1, p2, (chainrev, delta), ifh, dfh)
1177 p1, p2, (chainrev, delta), ifh, dfh)
1178 if not dfh and not self._inline:
1178 if not dfh and not self._inline:
1179 # addrevision switched from inline to conventional
1179 # addrevision switched from inline to conventional
1180 # reopen the index
1180 # reopen the index
1181 ifh.close()
1181 ifh.close()
1182 dfh = self.opener(self.datafile, "a")
1182 dfh = self.opener(self.datafile, "a")
1183 ifh = self.opener(self.indexfile, "a")
1183 ifh = self.opener(self.indexfile, "a")
1184 finally:
1184 finally:
1185 if dfh:
1185 if dfh:
1186 dfh.close()
1186 dfh.close()
1187 ifh.close()
1187 ifh.close()
1188
1188
1189 return node
1189 return node
1190
1190
1191 def strip(self, minlink, transaction):
1191 def strip(self, minlink, transaction):
1192 """truncate the revlog on the first revision with a linkrev >= minlink
1192 """truncate the revlog on the first revision with a linkrev >= minlink
1193
1193
1194 This function is called when we're stripping revision minlink and
1194 This function is called when we're stripping revision minlink and
1195 its descendants from the repository.
1195 its descendants from the repository.
1196
1196
1197 We have to remove all revisions with linkrev >= minlink, because
1197 We have to remove all revisions with linkrev >= minlink, because
1198 the equivalent changelog revisions will be renumbered after the
1198 the equivalent changelog revisions will be renumbered after the
1199 strip.
1199 strip.
1200
1200
1201 So we truncate the revlog on the first of these revisions, and
1201 So we truncate the revlog on the first of these revisions, and
1202 trust that the caller has saved the revisions that shouldn't be
1202 trust that the caller has saved the revisions that shouldn't be
1203 removed and that it'll readd them after this truncation.
1203 removed and that it'll readd them after this truncation.
1204 """
1204 """
1205 if len(self) == 0:
1205 if len(self) == 0:
1206 return
1206 return
1207
1207
1208 for rev in self:
1208 for rev in self:
1209 if self.index[rev][4] >= minlink:
1209 if self.index[rev][4] >= minlink:
1210 break
1210 break
1211 else:
1211 else:
1212 return
1212 return
1213
1213
1214 # first truncate the files on disk
1214 # first truncate the files on disk
1215 end = self.start(rev)
1215 end = self.start(rev)
1216 if not self._inline:
1216 if not self._inline:
1217 transaction.add(self.datafile, end)
1217 transaction.add(self.datafile, end)
1218 end = rev * self._io.size
1218 end = rev * self._io.size
1219 else:
1219 else:
1220 end += rev * self._io.size
1220 end += rev * self._io.size
1221
1221
1222 transaction.add(self.indexfile, end)
1222 transaction.add(self.indexfile, end)
1223
1223
1224 # then reset internal state in memory to forget those revisions
1224 # then reset internal state in memory to forget those revisions
1225 self._cache = None
1225 self._cache = None
1226 self._chunkclear()
1226 self._chunkclear()
1227 for x in xrange(rev, len(self)):
1227 for x in xrange(rev, len(self)):
1228 del self.nodemap[self.node(x)]
1228 del self.nodemap[self.node(x)]
1229
1229
1230 del self.index[rev:-1]
1230 del self.index[rev:-1]
1231
1231
1232 def checksize(self):
1232 def checksize(self):
1233 expected = 0
1233 expected = 0
1234 if len(self):
1234 if len(self):
1235 expected = max(0, self.end(len(self) - 1))
1235 expected = max(0, self.end(len(self) - 1))
1236
1236
1237 try:
1237 try:
1238 f = self.opener(self.datafile)
1238 f = self.opener(self.datafile)
1239 f.seek(0, 2)
1239 f.seek(0, 2)
1240 actual = f.tell()
1240 actual = f.tell()
1241 f.close()
1241 f.close()
1242 dd = actual - expected
1242 dd = actual - expected
1243 except IOError, inst:
1243 except IOError, inst:
1244 if inst.errno != errno.ENOENT:
1244 if inst.errno != errno.ENOENT:
1245 raise
1245 raise
1246 dd = 0
1246 dd = 0
1247
1247
1248 try:
1248 try:
1249 f = self.opener(self.indexfile)
1249 f = self.opener(self.indexfile)
1250 f.seek(0, 2)
1250 f.seek(0, 2)
1251 actual = f.tell()
1251 actual = f.tell()
1252 f.close()
1252 f.close()
1253 s = self._io.size
1253 s = self._io.size
1254 i = max(0, actual // s)
1254 i = max(0, actual // s)
1255 di = actual - (i * s)
1255 di = actual - (i * s)
1256 if self._inline:
1256 if self._inline:
1257 databytes = 0
1257 databytes = 0
1258 for r in self:
1258 for r in self:
1259 databytes += max(0, self.length(r))
1259 databytes += max(0, self.length(r))
1260 dd = 0
1260 dd = 0
1261 di = actual - len(self) * s - databytes
1261 di = actual - len(self) * s - databytes
1262 except IOError, inst:
1262 except IOError, inst:
1263 if inst.errno != errno.ENOENT:
1263 if inst.errno != errno.ENOENT:
1264 raise
1264 raise
1265 di = 0
1265 di = 0
1266
1266
1267 return (dd, di)
1267 return (dd, di)
1268
1268
1269 def files(self):
1269 def files(self):
1270 res = [self.indexfile]
1270 res = [self.indexfile]
1271 if not self._inline:
1271 if not self._inline:
1272 res.append(self.datafile)
1272 res.append(self.datafile)
1273 return res
1273 return res
General Comments 0
You need to be logged in to leave comments. Login now