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