##// END OF EJS Templates
push: fix coding errors in unused branch...
Matt Mackall -
r16020:7460f3a1 stable
parent child Browse files
Show More
@@ -1,2310 +1,2310
1 # localrepo.py - read/write repository class for mercurial
1 # localrepo.py - read/write repository class for mercurial
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 from node import bin, hex, nullid, nullrev, short
8 from node import bin, hex, nullid, nullrev, short
9 from i18n import _
9 from i18n import _
10 import repo, changegroup, subrepo, discovery, pushkey
10 import repo, changegroup, subrepo, discovery, pushkey
11 import changelog, dirstate, filelog, manifest, context, bookmarks, phases
11 import changelog, dirstate, filelog, manifest, context, bookmarks, phases
12 import lock, transaction, store, encoding
12 import lock, transaction, store, encoding
13 import scmutil, util, extensions, hook, error, revset
13 import scmutil, util, extensions, hook, error, revset
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 from lock import release
17 from lock import release
18 import weakref, errno, os, time, inspect
18 import weakref, errno, os, time, inspect
19 propertycache = util.propertycache
19 propertycache = util.propertycache
20 filecache = scmutil.filecache
20 filecache = scmutil.filecache
21
21
22 class localrepository(repo.repository):
22 class localrepository(repo.repository):
23 capabilities = set(('lookup', 'changegroupsubset', 'branchmap', 'pushkey',
23 capabilities = set(('lookup', 'changegroupsubset', 'branchmap', 'pushkey',
24 'known', 'getbundle'))
24 'known', 'getbundle'))
25 supportedformats = set(('revlogv1', 'generaldelta'))
25 supportedformats = set(('revlogv1', 'generaldelta'))
26 supported = supportedformats | set(('store', 'fncache', 'shared',
26 supported = supportedformats | set(('store', 'fncache', 'shared',
27 'dotencode'))
27 'dotencode'))
28
28
29 def __init__(self, baseui, path=None, create=False):
29 def __init__(self, baseui, path=None, create=False):
30 repo.repository.__init__(self)
30 repo.repository.__init__(self)
31 self.root = os.path.realpath(util.expandpath(path))
31 self.root = os.path.realpath(util.expandpath(path))
32 self.path = os.path.join(self.root, ".hg")
32 self.path = os.path.join(self.root, ".hg")
33 self.origroot = path
33 self.origroot = path
34 self.auditor = scmutil.pathauditor(self.root, self._checknested)
34 self.auditor = scmutil.pathauditor(self.root, self._checknested)
35 self.opener = scmutil.opener(self.path)
35 self.opener = scmutil.opener(self.path)
36 self.wopener = scmutil.opener(self.root)
36 self.wopener = scmutil.opener(self.root)
37 self.baseui = baseui
37 self.baseui = baseui
38 self.ui = baseui.copy()
38 self.ui = baseui.copy()
39 self._dirtyphases = False
39 self._dirtyphases = False
40 # A list of callback to shape the phase if no data were found.
40 # A list of callback to shape the phase if no data were found.
41 # Callback are in the form: func(repo, roots) --> processed root.
41 # Callback are in the form: func(repo, roots) --> processed root.
42 # This list it to be filled by extension during repo setup
42 # This list it to be filled by extension during repo setup
43 self._phasedefaults = []
43 self._phasedefaults = []
44
44
45 try:
45 try:
46 self.ui.readconfig(self.join("hgrc"), self.root)
46 self.ui.readconfig(self.join("hgrc"), self.root)
47 extensions.loadall(self.ui)
47 extensions.loadall(self.ui)
48 except IOError:
48 except IOError:
49 pass
49 pass
50
50
51 if not os.path.isdir(self.path):
51 if not os.path.isdir(self.path):
52 if create:
52 if create:
53 if not os.path.exists(path):
53 if not os.path.exists(path):
54 util.makedirs(path)
54 util.makedirs(path)
55 util.makedir(self.path, notindexed=True)
55 util.makedir(self.path, notindexed=True)
56 requirements = ["revlogv1"]
56 requirements = ["revlogv1"]
57 if self.ui.configbool('format', 'usestore', True):
57 if self.ui.configbool('format', 'usestore', True):
58 os.mkdir(os.path.join(self.path, "store"))
58 os.mkdir(os.path.join(self.path, "store"))
59 requirements.append("store")
59 requirements.append("store")
60 if self.ui.configbool('format', 'usefncache', True):
60 if self.ui.configbool('format', 'usefncache', True):
61 requirements.append("fncache")
61 requirements.append("fncache")
62 if self.ui.configbool('format', 'dotencode', True):
62 if self.ui.configbool('format', 'dotencode', True):
63 requirements.append('dotencode')
63 requirements.append('dotencode')
64 # create an invalid changelog
64 # create an invalid changelog
65 self.opener.append(
65 self.opener.append(
66 "00changelog.i",
66 "00changelog.i",
67 '\0\0\0\2' # represents revlogv2
67 '\0\0\0\2' # represents revlogv2
68 ' dummy changelog to prevent using the old repo layout'
68 ' dummy changelog to prevent using the old repo layout'
69 )
69 )
70 if self.ui.configbool('format', 'generaldelta', False):
70 if self.ui.configbool('format', 'generaldelta', False):
71 requirements.append("generaldelta")
71 requirements.append("generaldelta")
72 requirements = set(requirements)
72 requirements = set(requirements)
73 else:
73 else:
74 raise error.RepoError(_("repository %s not found") % path)
74 raise error.RepoError(_("repository %s not found") % path)
75 elif create:
75 elif create:
76 raise error.RepoError(_("repository %s already exists") % path)
76 raise error.RepoError(_("repository %s already exists") % path)
77 else:
77 else:
78 try:
78 try:
79 requirements = scmutil.readrequires(self.opener, self.supported)
79 requirements = scmutil.readrequires(self.opener, self.supported)
80 except IOError, inst:
80 except IOError, inst:
81 if inst.errno != errno.ENOENT:
81 if inst.errno != errno.ENOENT:
82 raise
82 raise
83 requirements = set()
83 requirements = set()
84
84
85 self.sharedpath = self.path
85 self.sharedpath = self.path
86 try:
86 try:
87 s = os.path.realpath(self.opener.read("sharedpath").rstrip('\n'))
87 s = os.path.realpath(self.opener.read("sharedpath").rstrip('\n'))
88 if not os.path.exists(s):
88 if not os.path.exists(s):
89 raise error.RepoError(
89 raise error.RepoError(
90 _('.hg/sharedpath points to nonexistent directory %s') % s)
90 _('.hg/sharedpath points to nonexistent directory %s') % s)
91 self.sharedpath = s
91 self.sharedpath = s
92 except IOError, inst:
92 except IOError, inst:
93 if inst.errno != errno.ENOENT:
93 if inst.errno != errno.ENOENT:
94 raise
94 raise
95
95
96 self.store = store.store(requirements, self.sharedpath, scmutil.opener)
96 self.store = store.store(requirements, self.sharedpath, scmutil.opener)
97 self.spath = self.store.path
97 self.spath = self.store.path
98 self.sopener = self.store.opener
98 self.sopener = self.store.opener
99 self.sjoin = self.store.join
99 self.sjoin = self.store.join
100 self.opener.createmode = self.store.createmode
100 self.opener.createmode = self.store.createmode
101 self._applyrequirements(requirements)
101 self._applyrequirements(requirements)
102 if create:
102 if create:
103 self._writerequirements()
103 self._writerequirements()
104
104
105
105
106 self._branchcache = None
106 self._branchcache = None
107 self._branchcachetip = None
107 self._branchcachetip = None
108 self.filterpats = {}
108 self.filterpats = {}
109 self._datafilters = {}
109 self._datafilters = {}
110 self._transref = self._lockref = self._wlockref = None
110 self._transref = self._lockref = self._wlockref = None
111
111
112 # A cache for various files under .hg/ that tracks file changes,
112 # A cache for various files under .hg/ that tracks file changes,
113 # (used by the filecache decorator)
113 # (used by the filecache decorator)
114 #
114 #
115 # Maps a property name to its util.filecacheentry
115 # Maps a property name to its util.filecacheentry
116 self._filecache = {}
116 self._filecache = {}
117
117
118 def _applyrequirements(self, requirements):
118 def _applyrequirements(self, requirements):
119 self.requirements = requirements
119 self.requirements = requirements
120 openerreqs = set(('revlogv1', 'generaldelta'))
120 openerreqs = set(('revlogv1', 'generaldelta'))
121 self.sopener.options = dict((r, 1) for r in requirements
121 self.sopener.options = dict((r, 1) for r in requirements
122 if r in openerreqs)
122 if r in openerreqs)
123
123
124 def _writerequirements(self):
124 def _writerequirements(self):
125 reqfile = self.opener("requires", "w")
125 reqfile = self.opener("requires", "w")
126 for r in self.requirements:
126 for r in self.requirements:
127 reqfile.write("%s\n" % r)
127 reqfile.write("%s\n" % r)
128 reqfile.close()
128 reqfile.close()
129
129
130 def _checknested(self, path):
130 def _checknested(self, path):
131 """Determine if path is a legal nested repository."""
131 """Determine if path is a legal nested repository."""
132 if not path.startswith(self.root):
132 if not path.startswith(self.root):
133 return False
133 return False
134 subpath = path[len(self.root) + 1:]
134 subpath = path[len(self.root) + 1:]
135 normsubpath = util.pconvert(subpath)
135 normsubpath = util.pconvert(subpath)
136
136
137 # XXX: Checking against the current working copy is wrong in
137 # XXX: Checking against the current working copy is wrong in
138 # the sense that it can reject things like
138 # the sense that it can reject things like
139 #
139 #
140 # $ hg cat -r 10 sub/x.txt
140 # $ hg cat -r 10 sub/x.txt
141 #
141 #
142 # if sub/ is no longer a subrepository in the working copy
142 # if sub/ is no longer a subrepository in the working copy
143 # parent revision.
143 # parent revision.
144 #
144 #
145 # However, it can of course also allow things that would have
145 # However, it can of course also allow things that would have
146 # been rejected before, such as the above cat command if sub/
146 # been rejected before, such as the above cat command if sub/
147 # is a subrepository now, but was a normal directory before.
147 # is a subrepository now, but was a normal directory before.
148 # The old path auditor would have rejected by mistake since it
148 # The old path auditor would have rejected by mistake since it
149 # panics when it sees sub/.hg/.
149 # panics when it sees sub/.hg/.
150 #
150 #
151 # All in all, checking against the working copy seems sensible
151 # All in all, checking against the working copy seems sensible
152 # since we want to prevent access to nested repositories on
152 # since we want to prevent access to nested repositories on
153 # the filesystem *now*.
153 # the filesystem *now*.
154 ctx = self[None]
154 ctx = self[None]
155 parts = util.splitpath(subpath)
155 parts = util.splitpath(subpath)
156 while parts:
156 while parts:
157 prefix = '/'.join(parts)
157 prefix = '/'.join(parts)
158 if prefix in ctx.substate:
158 if prefix in ctx.substate:
159 if prefix == normsubpath:
159 if prefix == normsubpath:
160 return True
160 return True
161 else:
161 else:
162 sub = ctx.sub(prefix)
162 sub = ctx.sub(prefix)
163 return sub.checknested(subpath[len(prefix) + 1:])
163 return sub.checknested(subpath[len(prefix) + 1:])
164 else:
164 else:
165 parts.pop()
165 parts.pop()
166 return False
166 return False
167
167
168 @filecache('bookmarks')
168 @filecache('bookmarks')
169 def _bookmarks(self):
169 def _bookmarks(self):
170 return bookmarks.read(self)
170 return bookmarks.read(self)
171
171
172 @filecache('bookmarks.current')
172 @filecache('bookmarks.current')
173 def _bookmarkcurrent(self):
173 def _bookmarkcurrent(self):
174 return bookmarks.readcurrent(self)
174 return bookmarks.readcurrent(self)
175
175
176 def _writebookmarks(self, marks):
176 def _writebookmarks(self, marks):
177 bookmarks.write(self)
177 bookmarks.write(self)
178
178
179 @filecache('phaseroots', True)
179 @filecache('phaseroots', True)
180 def _phaseroots(self):
180 def _phaseroots(self):
181 self._dirtyphases = False
181 self._dirtyphases = False
182 phaseroots = phases.readroots(self)
182 phaseroots = phases.readroots(self)
183 phases.filterunknown(self, phaseroots)
183 phases.filterunknown(self, phaseroots)
184 return phaseroots
184 return phaseroots
185
185
186 @propertycache
186 @propertycache
187 def _phaserev(self):
187 def _phaserev(self):
188 cache = [phases.public] * len(self)
188 cache = [phases.public] * len(self)
189 for phase in phases.trackedphases:
189 for phase in phases.trackedphases:
190 roots = map(self.changelog.rev, self._phaseroots[phase])
190 roots = map(self.changelog.rev, self._phaseroots[phase])
191 if roots:
191 if roots:
192 for rev in roots:
192 for rev in roots:
193 cache[rev] = phase
193 cache[rev] = phase
194 for rev in self.changelog.descendants(*roots):
194 for rev in self.changelog.descendants(*roots):
195 cache[rev] = phase
195 cache[rev] = phase
196 return cache
196 return cache
197
197
198 @filecache('00changelog.i', True)
198 @filecache('00changelog.i', True)
199 def changelog(self):
199 def changelog(self):
200 c = changelog.changelog(self.sopener)
200 c = changelog.changelog(self.sopener)
201 if 'HG_PENDING' in os.environ:
201 if 'HG_PENDING' in os.environ:
202 p = os.environ['HG_PENDING']
202 p = os.environ['HG_PENDING']
203 if p.startswith(self.root):
203 if p.startswith(self.root):
204 c.readpending('00changelog.i.a')
204 c.readpending('00changelog.i.a')
205 return c
205 return c
206
206
207 @filecache('00manifest.i', True)
207 @filecache('00manifest.i', True)
208 def manifest(self):
208 def manifest(self):
209 return manifest.manifest(self.sopener)
209 return manifest.manifest(self.sopener)
210
210
211 @filecache('dirstate')
211 @filecache('dirstate')
212 def dirstate(self):
212 def dirstate(self):
213 warned = [0]
213 warned = [0]
214 def validate(node):
214 def validate(node):
215 try:
215 try:
216 self.changelog.rev(node)
216 self.changelog.rev(node)
217 return node
217 return node
218 except error.LookupError:
218 except error.LookupError:
219 if not warned[0]:
219 if not warned[0]:
220 warned[0] = True
220 warned[0] = True
221 self.ui.warn(_("warning: ignoring unknown"
221 self.ui.warn(_("warning: ignoring unknown"
222 " working parent %s!\n") % short(node))
222 " working parent %s!\n") % short(node))
223 return nullid
223 return nullid
224
224
225 return dirstate.dirstate(self.opener, self.ui, self.root, validate)
225 return dirstate.dirstate(self.opener, self.ui, self.root, validate)
226
226
227 def __getitem__(self, changeid):
227 def __getitem__(self, changeid):
228 if changeid is None:
228 if changeid is None:
229 return context.workingctx(self)
229 return context.workingctx(self)
230 return context.changectx(self, changeid)
230 return context.changectx(self, changeid)
231
231
232 def __contains__(self, changeid):
232 def __contains__(self, changeid):
233 try:
233 try:
234 return bool(self.lookup(changeid))
234 return bool(self.lookup(changeid))
235 except error.RepoLookupError:
235 except error.RepoLookupError:
236 return False
236 return False
237
237
238 def __nonzero__(self):
238 def __nonzero__(self):
239 return True
239 return True
240
240
241 def __len__(self):
241 def __len__(self):
242 return len(self.changelog)
242 return len(self.changelog)
243
243
244 def __iter__(self):
244 def __iter__(self):
245 for i in xrange(len(self)):
245 for i in xrange(len(self)):
246 yield i
246 yield i
247
247
248 def revs(self, expr, *args):
248 def revs(self, expr, *args):
249 '''Return a list of revisions matching the given revset'''
249 '''Return a list of revisions matching the given revset'''
250 expr = revset.formatspec(expr, *args)
250 expr = revset.formatspec(expr, *args)
251 m = revset.match(None, expr)
251 m = revset.match(None, expr)
252 return [r for r in m(self, range(len(self)))]
252 return [r for r in m(self, range(len(self)))]
253
253
254 def set(self, expr, *args):
254 def set(self, expr, *args):
255 '''
255 '''
256 Yield a context for each matching revision, after doing arg
256 Yield a context for each matching revision, after doing arg
257 replacement via revset.formatspec
257 replacement via revset.formatspec
258 '''
258 '''
259 for r in self.revs(expr, *args):
259 for r in self.revs(expr, *args):
260 yield self[r]
260 yield self[r]
261
261
262 def url(self):
262 def url(self):
263 return 'file:' + self.root
263 return 'file:' + self.root
264
264
265 def hook(self, name, throw=False, **args):
265 def hook(self, name, throw=False, **args):
266 return hook.hook(self.ui, self, name, throw, **args)
266 return hook.hook(self.ui, self, name, throw, **args)
267
267
268 tag_disallowed = ':\r\n'
268 tag_disallowed = ':\r\n'
269
269
270 def _tag(self, names, node, message, local, user, date, extra={}):
270 def _tag(self, names, node, message, local, user, date, extra={}):
271 if isinstance(names, str):
271 if isinstance(names, str):
272 allchars = names
272 allchars = names
273 names = (names,)
273 names = (names,)
274 else:
274 else:
275 allchars = ''.join(names)
275 allchars = ''.join(names)
276 for c in self.tag_disallowed:
276 for c in self.tag_disallowed:
277 if c in allchars:
277 if c in allchars:
278 raise util.Abort(_('%r cannot be used in a tag name') % c)
278 raise util.Abort(_('%r cannot be used in a tag name') % c)
279
279
280 branches = self.branchmap()
280 branches = self.branchmap()
281 for name in names:
281 for name in names:
282 self.hook('pretag', throw=True, node=hex(node), tag=name,
282 self.hook('pretag', throw=True, node=hex(node), tag=name,
283 local=local)
283 local=local)
284 if name in branches:
284 if name in branches:
285 self.ui.warn(_("warning: tag %s conflicts with existing"
285 self.ui.warn(_("warning: tag %s conflicts with existing"
286 " branch name\n") % name)
286 " branch name\n") % name)
287
287
288 def writetags(fp, names, munge, prevtags):
288 def writetags(fp, names, munge, prevtags):
289 fp.seek(0, 2)
289 fp.seek(0, 2)
290 if prevtags and prevtags[-1] != '\n':
290 if prevtags and prevtags[-1] != '\n':
291 fp.write('\n')
291 fp.write('\n')
292 for name in names:
292 for name in names:
293 m = munge and munge(name) or name
293 m = munge and munge(name) or name
294 if self._tagscache.tagtypes and name in self._tagscache.tagtypes:
294 if self._tagscache.tagtypes and name in self._tagscache.tagtypes:
295 old = self.tags().get(name, nullid)
295 old = self.tags().get(name, nullid)
296 fp.write('%s %s\n' % (hex(old), m))
296 fp.write('%s %s\n' % (hex(old), m))
297 fp.write('%s %s\n' % (hex(node), m))
297 fp.write('%s %s\n' % (hex(node), m))
298 fp.close()
298 fp.close()
299
299
300 prevtags = ''
300 prevtags = ''
301 if local:
301 if local:
302 try:
302 try:
303 fp = self.opener('localtags', 'r+')
303 fp = self.opener('localtags', 'r+')
304 except IOError:
304 except IOError:
305 fp = self.opener('localtags', 'a')
305 fp = self.opener('localtags', 'a')
306 else:
306 else:
307 prevtags = fp.read()
307 prevtags = fp.read()
308
308
309 # local tags are stored in the current charset
309 # local tags are stored in the current charset
310 writetags(fp, names, None, prevtags)
310 writetags(fp, names, None, prevtags)
311 for name in names:
311 for name in names:
312 self.hook('tag', node=hex(node), tag=name, local=local)
312 self.hook('tag', node=hex(node), tag=name, local=local)
313 return
313 return
314
314
315 try:
315 try:
316 fp = self.wfile('.hgtags', 'rb+')
316 fp = self.wfile('.hgtags', 'rb+')
317 except IOError, e:
317 except IOError, e:
318 if e.errno != errno.ENOENT:
318 if e.errno != errno.ENOENT:
319 raise
319 raise
320 fp = self.wfile('.hgtags', 'ab')
320 fp = self.wfile('.hgtags', 'ab')
321 else:
321 else:
322 prevtags = fp.read()
322 prevtags = fp.read()
323
323
324 # committed tags are stored in UTF-8
324 # committed tags are stored in UTF-8
325 writetags(fp, names, encoding.fromlocal, prevtags)
325 writetags(fp, names, encoding.fromlocal, prevtags)
326
326
327 fp.close()
327 fp.close()
328
328
329 self.invalidatecaches()
329 self.invalidatecaches()
330
330
331 if '.hgtags' not in self.dirstate:
331 if '.hgtags' not in self.dirstate:
332 self[None].add(['.hgtags'])
332 self[None].add(['.hgtags'])
333
333
334 m = matchmod.exact(self.root, '', ['.hgtags'])
334 m = matchmod.exact(self.root, '', ['.hgtags'])
335 tagnode = self.commit(message, user, date, extra=extra, match=m)
335 tagnode = self.commit(message, user, date, extra=extra, match=m)
336
336
337 for name in names:
337 for name in names:
338 self.hook('tag', node=hex(node), tag=name, local=local)
338 self.hook('tag', node=hex(node), tag=name, local=local)
339
339
340 return tagnode
340 return tagnode
341
341
342 def tag(self, names, node, message, local, user, date):
342 def tag(self, names, node, message, local, user, date):
343 '''tag a revision with one or more symbolic names.
343 '''tag a revision with one or more symbolic names.
344
344
345 names is a list of strings or, when adding a single tag, names may be a
345 names is a list of strings or, when adding a single tag, names may be a
346 string.
346 string.
347
347
348 if local is True, the tags are stored in a per-repository file.
348 if local is True, the tags are stored in a per-repository file.
349 otherwise, they are stored in the .hgtags file, and a new
349 otherwise, they are stored in the .hgtags file, and a new
350 changeset is committed with the change.
350 changeset is committed with the change.
351
351
352 keyword arguments:
352 keyword arguments:
353
353
354 local: whether to store tags in non-version-controlled file
354 local: whether to store tags in non-version-controlled file
355 (default False)
355 (default False)
356
356
357 message: commit message to use if committing
357 message: commit message to use if committing
358
358
359 user: name of user to use if committing
359 user: name of user to use if committing
360
360
361 date: date tuple to use if committing'''
361 date: date tuple to use if committing'''
362
362
363 if not local:
363 if not local:
364 for x in self.status()[:5]:
364 for x in self.status()[:5]:
365 if '.hgtags' in x:
365 if '.hgtags' in x:
366 raise util.Abort(_('working copy of .hgtags is changed '
366 raise util.Abort(_('working copy of .hgtags is changed '
367 '(please commit .hgtags manually)'))
367 '(please commit .hgtags manually)'))
368
368
369 self.tags() # instantiate the cache
369 self.tags() # instantiate the cache
370 self._tag(names, node, message, local, user, date)
370 self._tag(names, node, message, local, user, date)
371
371
372 @propertycache
372 @propertycache
373 def _tagscache(self):
373 def _tagscache(self):
374 '''Returns a tagscache object that contains various tags related caches.'''
374 '''Returns a tagscache object that contains various tags related caches.'''
375
375
376 # This simplifies its cache management by having one decorated
376 # This simplifies its cache management by having one decorated
377 # function (this one) and the rest simply fetch things from it.
377 # function (this one) and the rest simply fetch things from it.
378 class tagscache(object):
378 class tagscache(object):
379 def __init__(self):
379 def __init__(self):
380 # These two define the set of tags for this repository. tags
380 # These two define the set of tags for this repository. tags
381 # maps tag name to node; tagtypes maps tag name to 'global' or
381 # maps tag name to node; tagtypes maps tag name to 'global' or
382 # 'local'. (Global tags are defined by .hgtags across all
382 # 'local'. (Global tags are defined by .hgtags across all
383 # heads, and local tags are defined in .hg/localtags.)
383 # heads, and local tags are defined in .hg/localtags.)
384 # They constitute the in-memory cache of tags.
384 # They constitute the in-memory cache of tags.
385 self.tags = self.tagtypes = None
385 self.tags = self.tagtypes = None
386
386
387 self.nodetagscache = self.tagslist = None
387 self.nodetagscache = self.tagslist = None
388
388
389 cache = tagscache()
389 cache = tagscache()
390 cache.tags, cache.tagtypes = self._findtags()
390 cache.tags, cache.tagtypes = self._findtags()
391
391
392 return cache
392 return cache
393
393
394 def tags(self):
394 def tags(self):
395 '''return a mapping of tag to node'''
395 '''return a mapping of tag to node'''
396 return self._tagscache.tags
396 return self._tagscache.tags
397
397
398 def _findtags(self):
398 def _findtags(self):
399 '''Do the hard work of finding tags. Return a pair of dicts
399 '''Do the hard work of finding tags. Return a pair of dicts
400 (tags, tagtypes) where tags maps tag name to node, and tagtypes
400 (tags, tagtypes) where tags maps tag name to node, and tagtypes
401 maps tag name to a string like \'global\' or \'local\'.
401 maps tag name to a string like \'global\' or \'local\'.
402 Subclasses or extensions are free to add their own tags, but
402 Subclasses or extensions are free to add their own tags, but
403 should be aware that the returned dicts will be retained for the
403 should be aware that the returned dicts will be retained for the
404 duration of the localrepo object.'''
404 duration of the localrepo object.'''
405
405
406 # XXX what tagtype should subclasses/extensions use? Currently
406 # XXX what tagtype should subclasses/extensions use? Currently
407 # mq and bookmarks add tags, but do not set the tagtype at all.
407 # mq and bookmarks add tags, but do not set the tagtype at all.
408 # Should each extension invent its own tag type? Should there
408 # Should each extension invent its own tag type? Should there
409 # be one tagtype for all such "virtual" tags? Or is the status
409 # be one tagtype for all such "virtual" tags? Or is the status
410 # quo fine?
410 # quo fine?
411
411
412 alltags = {} # map tag name to (node, hist)
412 alltags = {} # map tag name to (node, hist)
413 tagtypes = {}
413 tagtypes = {}
414
414
415 tagsmod.findglobaltags(self.ui, self, alltags, tagtypes)
415 tagsmod.findglobaltags(self.ui, self, alltags, tagtypes)
416 tagsmod.readlocaltags(self.ui, self, alltags, tagtypes)
416 tagsmod.readlocaltags(self.ui, self, alltags, tagtypes)
417
417
418 # Build the return dicts. Have to re-encode tag names because
418 # Build the return dicts. Have to re-encode tag names because
419 # the tags module always uses UTF-8 (in order not to lose info
419 # the tags module always uses UTF-8 (in order not to lose info
420 # writing to the cache), but the rest of Mercurial wants them in
420 # writing to the cache), but the rest of Mercurial wants them in
421 # local encoding.
421 # local encoding.
422 tags = {}
422 tags = {}
423 for (name, (node, hist)) in alltags.iteritems():
423 for (name, (node, hist)) in alltags.iteritems():
424 if node != nullid:
424 if node != nullid:
425 try:
425 try:
426 # ignore tags to unknown nodes
426 # ignore tags to unknown nodes
427 self.changelog.lookup(node)
427 self.changelog.lookup(node)
428 tags[encoding.tolocal(name)] = node
428 tags[encoding.tolocal(name)] = node
429 except error.LookupError:
429 except error.LookupError:
430 pass
430 pass
431 tags['tip'] = self.changelog.tip()
431 tags['tip'] = self.changelog.tip()
432 tagtypes = dict([(encoding.tolocal(name), value)
432 tagtypes = dict([(encoding.tolocal(name), value)
433 for (name, value) in tagtypes.iteritems()])
433 for (name, value) in tagtypes.iteritems()])
434 return (tags, tagtypes)
434 return (tags, tagtypes)
435
435
436 def tagtype(self, tagname):
436 def tagtype(self, tagname):
437 '''
437 '''
438 return the type of the given tag. result can be:
438 return the type of the given tag. result can be:
439
439
440 'local' : a local tag
440 'local' : a local tag
441 'global' : a global tag
441 'global' : a global tag
442 None : tag does not exist
442 None : tag does not exist
443 '''
443 '''
444
444
445 return self._tagscache.tagtypes.get(tagname)
445 return self._tagscache.tagtypes.get(tagname)
446
446
447 def tagslist(self):
447 def tagslist(self):
448 '''return a list of tags ordered by revision'''
448 '''return a list of tags ordered by revision'''
449 if not self._tagscache.tagslist:
449 if not self._tagscache.tagslist:
450 l = []
450 l = []
451 for t, n in self.tags().iteritems():
451 for t, n in self.tags().iteritems():
452 r = self.changelog.rev(n)
452 r = self.changelog.rev(n)
453 l.append((r, t, n))
453 l.append((r, t, n))
454 self._tagscache.tagslist = [(t, n) for r, t, n in sorted(l)]
454 self._tagscache.tagslist = [(t, n) for r, t, n in sorted(l)]
455
455
456 return self._tagscache.tagslist
456 return self._tagscache.tagslist
457
457
458 def nodetags(self, node):
458 def nodetags(self, node):
459 '''return the tags associated with a node'''
459 '''return the tags associated with a node'''
460 if not self._tagscache.nodetagscache:
460 if not self._tagscache.nodetagscache:
461 nodetagscache = {}
461 nodetagscache = {}
462 for t, n in self.tags().iteritems():
462 for t, n in self.tags().iteritems():
463 nodetagscache.setdefault(n, []).append(t)
463 nodetagscache.setdefault(n, []).append(t)
464 for tags in nodetagscache.itervalues():
464 for tags in nodetagscache.itervalues():
465 tags.sort()
465 tags.sort()
466 self._tagscache.nodetagscache = nodetagscache
466 self._tagscache.nodetagscache = nodetagscache
467 return self._tagscache.nodetagscache.get(node, [])
467 return self._tagscache.nodetagscache.get(node, [])
468
468
469 def nodebookmarks(self, node):
469 def nodebookmarks(self, node):
470 marks = []
470 marks = []
471 for bookmark, n in self._bookmarks.iteritems():
471 for bookmark, n in self._bookmarks.iteritems():
472 if n == node:
472 if n == node:
473 marks.append(bookmark)
473 marks.append(bookmark)
474 return sorted(marks)
474 return sorted(marks)
475
475
476 def _branchtags(self, partial, lrev):
476 def _branchtags(self, partial, lrev):
477 # TODO: rename this function?
477 # TODO: rename this function?
478 tiprev = len(self) - 1
478 tiprev = len(self) - 1
479 if lrev != tiprev:
479 if lrev != tiprev:
480 ctxgen = (self[r] for r in xrange(lrev + 1, tiprev + 1))
480 ctxgen = (self[r] for r in xrange(lrev + 1, tiprev + 1))
481 self._updatebranchcache(partial, ctxgen)
481 self._updatebranchcache(partial, ctxgen)
482 self._writebranchcache(partial, self.changelog.tip(), tiprev)
482 self._writebranchcache(partial, self.changelog.tip(), tiprev)
483
483
484 return partial
484 return partial
485
485
486 def updatebranchcache(self):
486 def updatebranchcache(self):
487 tip = self.changelog.tip()
487 tip = self.changelog.tip()
488 if self._branchcache is not None and self._branchcachetip == tip:
488 if self._branchcache is not None and self._branchcachetip == tip:
489 return
489 return
490
490
491 oldtip = self._branchcachetip
491 oldtip = self._branchcachetip
492 self._branchcachetip = tip
492 self._branchcachetip = tip
493 if oldtip is None or oldtip not in self.changelog.nodemap:
493 if oldtip is None or oldtip not in self.changelog.nodemap:
494 partial, last, lrev = self._readbranchcache()
494 partial, last, lrev = self._readbranchcache()
495 else:
495 else:
496 lrev = self.changelog.rev(oldtip)
496 lrev = self.changelog.rev(oldtip)
497 partial = self._branchcache
497 partial = self._branchcache
498
498
499 self._branchtags(partial, lrev)
499 self._branchtags(partial, lrev)
500 # this private cache holds all heads (not just tips)
500 # this private cache holds all heads (not just tips)
501 self._branchcache = partial
501 self._branchcache = partial
502
502
503 def branchmap(self):
503 def branchmap(self):
504 '''returns a dictionary {branch: [branchheads]}'''
504 '''returns a dictionary {branch: [branchheads]}'''
505 self.updatebranchcache()
505 self.updatebranchcache()
506 return self._branchcache
506 return self._branchcache
507
507
508 def branchtags(self):
508 def branchtags(self):
509 '''return a dict where branch names map to the tipmost head of
509 '''return a dict where branch names map to the tipmost head of
510 the branch, open heads come before closed'''
510 the branch, open heads come before closed'''
511 bt = {}
511 bt = {}
512 for bn, heads in self.branchmap().iteritems():
512 for bn, heads in self.branchmap().iteritems():
513 tip = heads[-1]
513 tip = heads[-1]
514 for h in reversed(heads):
514 for h in reversed(heads):
515 if 'close' not in self.changelog.read(h)[5]:
515 if 'close' not in self.changelog.read(h)[5]:
516 tip = h
516 tip = h
517 break
517 break
518 bt[bn] = tip
518 bt[bn] = tip
519 return bt
519 return bt
520
520
521 def _readbranchcache(self):
521 def _readbranchcache(self):
522 partial = {}
522 partial = {}
523 try:
523 try:
524 f = self.opener("cache/branchheads")
524 f = self.opener("cache/branchheads")
525 lines = f.read().split('\n')
525 lines = f.read().split('\n')
526 f.close()
526 f.close()
527 except (IOError, OSError):
527 except (IOError, OSError):
528 return {}, nullid, nullrev
528 return {}, nullid, nullrev
529
529
530 try:
530 try:
531 last, lrev = lines.pop(0).split(" ", 1)
531 last, lrev = lines.pop(0).split(" ", 1)
532 last, lrev = bin(last), int(lrev)
532 last, lrev = bin(last), int(lrev)
533 if lrev >= len(self) or self[lrev].node() != last:
533 if lrev >= len(self) or self[lrev].node() != last:
534 # invalidate the cache
534 # invalidate the cache
535 raise ValueError('invalidating branch cache (tip differs)')
535 raise ValueError('invalidating branch cache (tip differs)')
536 for l in lines:
536 for l in lines:
537 if not l:
537 if not l:
538 continue
538 continue
539 node, label = l.split(" ", 1)
539 node, label = l.split(" ", 1)
540 label = encoding.tolocal(label.strip())
540 label = encoding.tolocal(label.strip())
541 partial.setdefault(label, []).append(bin(node))
541 partial.setdefault(label, []).append(bin(node))
542 except KeyboardInterrupt:
542 except KeyboardInterrupt:
543 raise
543 raise
544 except Exception, inst:
544 except Exception, inst:
545 if self.ui.debugflag:
545 if self.ui.debugflag:
546 self.ui.warn(str(inst), '\n')
546 self.ui.warn(str(inst), '\n')
547 partial, last, lrev = {}, nullid, nullrev
547 partial, last, lrev = {}, nullid, nullrev
548 return partial, last, lrev
548 return partial, last, lrev
549
549
550 def _writebranchcache(self, branches, tip, tiprev):
550 def _writebranchcache(self, branches, tip, tiprev):
551 try:
551 try:
552 f = self.opener("cache/branchheads", "w", atomictemp=True)
552 f = self.opener("cache/branchheads", "w", atomictemp=True)
553 f.write("%s %s\n" % (hex(tip), tiprev))
553 f.write("%s %s\n" % (hex(tip), tiprev))
554 for label, nodes in branches.iteritems():
554 for label, nodes in branches.iteritems():
555 for node in nodes:
555 for node in nodes:
556 f.write("%s %s\n" % (hex(node), encoding.fromlocal(label)))
556 f.write("%s %s\n" % (hex(node), encoding.fromlocal(label)))
557 f.close()
557 f.close()
558 except (IOError, OSError):
558 except (IOError, OSError):
559 pass
559 pass
560
560
561 def _updatebranchcache(self, partial, ctxgen):
561 def _updatebranchcache(self, partial, ctxgen):
562 # collect new branch entries
562 # collect new branch entries
563 newbranches = {}
563 newbranches = {}
564 for c in ctxgen:
564 for c in ctxgen:
565 newbranches.setdefault(c.branch(), []).append(c.node())
565 newbranches.setdefault(c.branch(), []).append(c.node())
566 # if older branchheads are reachable from new ones, they aren't
566 # if older branchheads are reachable from new ones, they aren't
567 # really branchheads. Note checking parents is insufficient:
567 # really branchheads. Note checking parents is insufficient:
568 # 1 (branch a) -> 2 (branch b) -> 3 (branch a)
568 # 1 (branch a) -> 2 (branch b) -> 3 (branch a)
569 for branch, newnodes in newbranches.iteritems():
569 for branch, newnodes in newbranches.iteritems():
570 bheads = partial.setdefault(branch, [])
570 bheads = partial.setdefault(branch, [])
571 bheads.extend(newnodes)
571 bheads.extend(newnodes)
572 if len(bheads) <= 1:
572 if len(bheads) <= 1:
573 continue
573 continue
574 bheads = sorted(bheads, key=lambda x: self[x].rev())
574 bheads = sorted(bheads, key=lambda x: self[x].rev())
575 # starting from tip means fewer passes over reachable
575 # starting from tip means fewer passes over reachable
576 while newnodes:
576 while newnodes:
577 latest = newnodes.pop()
577 latest = newnodes.pop()
578 if latest not in bheads:
578 if latest not in bheads:
579 continue
579 continue
580 minbhrev = self[bheads[0]].node()
580 minbhrev = self[bheads[0]].node()
581 reachable = self.changelog.reachable(latest, minbhrev)
581 reachable = self.changelog.reachable(latest, minbhrev)
582 reachable.remove(latest)
582 reachable.remove(latest)
583 if reachable:
583 if reachable:
584 bheads = [b for b in bheads if b not in reachable]
584 bheads = [b for b in bheads if b not in reachable]
585 partial[branch] = bheads
585 partial[branch] = bheads
586
586
587 def lookup(self, key):
587 def lookup(self, key):
588 if isinstance(key, int):
588 if isinstance(key, int):
589 return self.changelog.node(key)
589 return self.changelog.node(key)
590 elif key == '.':
590 elif key == '.':
591 return self.dirstate.p1()
591 return self.dirstate.p1()
592 elif key == 'null':
592 elif key == 'null':
593 return nullid
593 return nullid
594 elif key == 'tip':
594 elif key == 'tip':
595 return self.changelog.tip()
595 return self.changelog.tip()
596 n = self.changelog._match(key)
596 n = self.changelog._match(key)
597 if n:
597 if n:
598 return n
598 return n
599 if key in self._bookmarks:
599 if key in self._bookmarks:
600 return self._bookmarks[key]
600 return self._bookmarks[key]
601 if key in self.tags():
601 if key in self.tags():
602 return self.tags()[key]
602 return self.tags()[key]
603 if key in self.branchtags():
603 if key in self.branchtags():
604 return self.branchtags()[key]
604 return self.branchtags()[key]
605 n = self.changelog._partialmatch(key)
605 n = self.changelog._partialmatch(key)
606 if n:
606 if n:
607 return n
607 return n
608
608
609 # can't find key, check if it might have come from damaged dirstate
609 # can't find key, check if it might have come from damaged dirstate
610 if key in self.dirstate.parents():
610 if key in self.dirstate.parents():
611 raise error.Abort(_("working directory has unknown parent '%s'!")
611 raise error.Abort(_("working directory has unknown parent '%s'!")
612 % short(key))
612 % short(key))
613 try:
613 try:
614 if len(key) == 20:
614 if len(key) == 20:
615 key = hex(key)
615 key = hex(key)
616 except TypeError:
616 except TypeError:
617 pass
617 pass
618 raise error.RepoLookupError(_("unknown revision '%s'") % key)
618 raise error.RepoLookupError(_("unknown revision '%s'") % key)
619
619
620 def lookupbranch(self, key, remote=None):
620 def lookupbranch(self, key, remote=None):
621 repo = remote or self
621 repo = remote or self
622 if key in repo.branchmap():
622 if key in repo.branchmap():
623 return key
623 return key
624
624
625 repo = (remote and remote.local()) and remote or self
625 repo = (remote and remote.local()) and remote or self
626 return repo[key].branch()
626 return repo[key].branch()
627
627
628 def known(self, nodes):
628 def known(self, nodes):
629 nm = self.changelog.nodemap
629 nm = self.changelog.nodemap
630 result = []
630 result = []
631 for n in nodes:
631 for n in nodes:
632 r = nm.get(n)
632 r = nm.get(n)
633 resp = not (r is None or self._phaserev[r] >= phases.secret)
633 resp = not (r is None or self._phaserev[r] >= phases.secret)
634 result.append(resp)
634 result.append(resp)
635 return result
635 return result
636
636
637 def local(self):
637 def local(self):
638 return self
638 return self
639
639
640 def join(self, f):
640 def join(self, f):
641 return os.path.join(self.path, f)
641 return os.path.join(self.path, f)
642
642
643 def wjoin(self, f):
643 def wjoin(self, f):
644 return os.path.join(self.root, f)
644 return os.path.join(self.root, f)
645
645
646 def file(self, f):
646 def file(self, f):
647 if f[0] == '/':
647 if f[0] == '/':
648 f = f[1:]
648 f = f[1:]
649 return filelog.filelog(self.sopener, f)
649 return filelog.filelog(self.sopener, f)
650
650
651 def changectx(self, changeid):
651 def changectx(self, changeid):
652 return self[changeid]
652 return self[changeid]
653
653
654 def parents(self, changeid=None):
654 def parents(self, changeid=None):
655 '''get list of changectxs for parents of changeid'''
655 '''get list of changectxs for parents of changeid'''
656 return self[changeid].parents()
656 return self[changeid].parents()
657
657
658 def filectx(self, path, changeid=None, fileid=None):
658 def filectx(self, path, changeid=None, fileid=None):
659 """changeid can be a changeset revision, node, or tag.
659 """changeid can be a changeset revision, node, or tag.
660 fileid can be a file revision or node."""
660 fileid can be a file revision or node."""
661 return context.filectx(self, path, changeid, fileid)
661 return context.filectx(self, path, changeid, fileid)
662
662
663 def getcwd(self):
663 def getcwd(self):
664 return self.dirstate.getcwd()
664 return self.dirstate.getcwd()
665
665
666 def pathto(self, f, cwd=None):
666 def pathto(self, f, cwd=None):
667 return self.dirstate.pathto(f, cwd)
667 return self.dirstate.pathto(f, cwd)
668
668
669 def wfile(self, f, mode='r'):
669 def wfile(self, f, mode='r'):
670 return self.wopener(f, mode)
670 return self.wopener(f, mode)
671
671
672 def _link(self, f):
672 def _link(self, f):
673 return os.path.islink(self.wjoin(f))
673 return os.path.islink(self.wjoin(f))
674
674
675 def _loadfilter(self, filter):
675 def _loadfilter(self, filter):
676 if filter not in self.filterpats:
676 if filter not in self.filterpats:
677 l = []
677 l = []
678 for pat, cmd in self.ui.configitems(filter):
678 for pat, cmd in self.ui.configitems(filter):
679 if cmd == '!':
679 if cmd == '!':
680 continue
680 continue
681 mf = matchmod.match(self.root, '', [pat])
681 mf = matchmod.match(self.root, '', [pat])
682 fn = None
682 fn = None
683 params = cmd
683 params = cmd
684 for name, filterfn in self._datafilters.iteritems():
684 for name, filterfn in self._datafilters.iteritems():
685 if cmd.startswith(name):
685 if cmd.startswith(name):
686 fn = filterfn
686 fn = filterfn
687 params = cmd[len(name):].lstrip()
687 params = cmd[len(name):].lstrip()
688 break
688 break
689 if not fn:
689 if not fn:
690 fn = lambda s, c, **kwargs: util.filter(s, c)
690 fn = lambda s, c, **kwargs: util.filter(s, c)
691 # Wrap old filters not supporting keyword arguments
691 # Wrap old filters not supporting keyword arguments
692 if not inspect.getargspec(fn)[2]:
692 if not inspect.getargspec(fn)[2]:
693 oldfn = fn
693 oldfn = fn
694 fn = lambda s, c, **kwargs: oldfn(s, c)
694 fn = lambda s, c, **kwargs: oldfn(s, c)
695 l.append((mf, fn, params))
695 l.append((mf, fn, params))
696 self.filterpats[filter] = l
696 self.filterpats[filter] = l
697 return self.filterpats[filter]
697 return self.filterpats[filter]
698
698
699 def _filter(self, filterpats, filename, data):
699 def _filter(self, filterpats, filename, data):
700 for mf, fn, cmd in filterpats:
700 for mf, fn, cmd in filterpats:
701 if mf(filename):
701 if mf(filename):
702 self.ui.debug("filtering %s through %s\n" % (filename, cmd))
702 self.ui.debug("filtering %s through %s\n" % (filename, cmd))
703 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
703 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
704 break
704 break
705
705
706 return data
706 return data
707
707
708 @propertycache
708 @propertycache
709 def _encodefilterpats(self):
709 def _encodefilterpats(self):
710 return self._loadfilter('encode')
710 return self._loadfilter('encode')
711
711
712 @propertycache
712 @propertycache
713 def _decodefilterpats(self):
713 def _decodefilterpats(self):
714 return self._loadfilter('decode')
714 return self._loadfilter('decode')
715
715
716 def adddatafilter(self, name, filter):
716 def adddatafilter(self, name, filter):
717 self._datafilters[name] = filter
717 self._datafilters[name] = filter
718
718
719 def wread(self, filename):
719 def wread(self, filename):
720 if self._link(filename):
720 if self._link(filename):
721 data = os.readlink(self.wjoin(filename))
721 data = os.readlink(self.wjoin(filename))
722 else:
722 else:
723 data = self.wopener.read(filename)
723 data = self.wopener.read(filename)
724 return self._filter(self._encodefilterpats, filename, data)
724 return self._filter(self._encodefilterpats, filename, data)
725
725
726 def wwrite(self, filename, data, flags):
726 def wwrite(self, filename, data, flags):
727 data = self._filter(self._decodefilterpats, filename, data)
727 data = self._filter(self._decodefilterpats, filename, data)
728 if 'l' in flags:
728 if 'l' in flags:
729 self.wopener.symlink(data, filename)
729 self.wopener.symlink(data, filename)
730 else:
730 else:
731 self.wopener.write(filename, data)
731 self.wopener.write(filename, data)
732 if 'x' in flags:
732 if 'x' in flags:
733 util.setflags(self.wjoin(filename), False, True)
733 util.setflags(self.wjoin(filename), False, True)
734
734
735 def wwritedata(self, filename, data):
735 def wwritedata(self, filename, data):
736 return self._filter(self._decodefilterpats, filename, data)
736 return self._filter(self._decodefilterpats, filename, data)
737
737
738 def transaction(self, desc):
738 def transaction(self, desc):
739 tr = self._transref and self._transref() or None
739 tr = self._transref and self._transref() or None
740 if tr and tr.running():
740 if tr and tr.running():
741 return tr.nest()
741 return tr.nest()
742
742
743 # abort here if the journal already exists
743 # abort here if the journal already exists
744 if os.path.exists(self.sjoin("journal")):
744 if os.path.exists(self.sjoin("journal")):
745 raise error.RepoError(
745 raise error.RepoError(
746 _("abandoned transaction found - run hg recover"))
746 _("abandoned transaction found - run hg recover"))
747
747
748 journalfiles = self._writejournal(desc)
748 journalfiles = self._writejournal(desc)
749 renames = [(x, undoname(x)) for x in journalfiles]
749 renames = [(x, undoname(x)) for x in journalfiles]
750
750
751 tr = transaction.transaction(self.ui.warn, self.sopener,
751 tr = transaction.transaction(self.ui.warn, self.sopener,
752 self.sjoin("journal"),
752 self.sjoin("journal"),
753 aftertrans(renames),
753 aftertrans(renames),
754 self.store.createmode)
754 self.store.createmode)
755 self._transref = weakref.ref(tr)
755 self._transref = weakref.ref(tr)
756 return tr
756 return tr
757
757
758 def _writejournal(self, desc):
758 def _writejournal(self, desc):
759 # save dirstate for rollback
759 # save dirstate for rollback
760 try:
760 try:
761 ds = self.opener.read("dirstate")
761 ds = self.opener.read("dirstate")
762 except IOError:
762 except IOError:
763 ds = ""
763 ds = ""
764 self.opener.write("journal.dirstate", ds)
764 self.opener.write("journal.dirstate", ds)
765 self.opener.write("journal.branch",
765 self.opener.write("journal.branch",
766 encoding.fromlocal(self.dirstate.branch()))
766 encoding.fromlocal(self.dirstate.branch()))
767 self.opener.write("journal.desc",
767 self.opener.write("journal.desc",
768 "%d\n%s\n" % (len(self), desc))
768 "%d\n%s\n" % (len(self), desc))
769
769
770 bkname = self.join('bookmarks')
770 bkname = self.join('bookmarks')
771 if os.path.exists(bkname):
771 if os.path.exists(bkname):
772 util.copyfile(bkname, self.join('journal.bookmarks'))
772 util.copyfile(bkname, self.join('journal.bookmarks'))
773 else:
773 else:
774 self.opener.write('journal.bookmarks', '')
774 self.opener.write('journal.bookmarks', '')
775 phasesname = self.sjoin('phaseroots')
775 phasesname = self.sjoin('phaseroots')
776 if os.path.exists(phasesname):
776 if os.path.exists(phasesname):
777 util.copyfile(phasesname, self.sjoin('journal.phaseroots'))
777 util.copyfile(phasesname, self.sjoin('journal.phaseroots'))
778 else:
778 else:
779 self.sopener.write('journal.phaseroots', '')
779 self.sopener.write('journal.phaseroots', '')
780
780
781 return (self.sjoin('journal'), self.join('journal.dirstate'),
781 return (self.sjoin('journal'), self.join('journal.dirstate'),
782 self.join('journal.branch'), self.join('journal.desc'),
782 self.join('journal.branch'), self.join('journal.desc'),
783 self.join('journal.bookmarks'),
783 self.join('journal.bookmarks'),
784 self.sjoin('journal.phaseroots'))
784 self.sjoin('journal.phaseroots'))
785
785
786 def recover(self):
786 def recover(self):
787 lock = self.lock()
787 lock = self.lock()
788 try:
788 try:
789 if os.path.exists(self.sjoin("journal")):
789 if os.path.exists(self.sjoin("journal")):
790 self.ui.status(_("rolling back interrupted transaction\n"))
790 self.ui.status(_("rolling back interrupted transaction\n"))
791 transaction.rollback(self.sopener, self.sjoin("journal"),
791 transaction.rollback(self.sopener, self.sjoin("journal"),
792 self.ui.warn)
792 self.ui.warn)
793 self.invalidate()
793 self.invalidate()
794 return True
794 return True
795 else:
795 else:
796 self.ui.warn(_("no interrupted transaction available\n"))
796 self.ui.warn(_("no interrupted transaction available\n"))
797 return False
797 return False
798 finally:
798 finally:
799 lock.release()
799 lock.release()
800
800
801 def rollback(self, dryrun=False, force=False):
801 def rollback(self, dryrun=False, force=False):
802 wlock = lock = None
802 wlock = lock = None
803 try:
803 try:
804 wlock = self.wlock()
804 wlock = self.wlock()
805 lock = self.lock()
805 lock = self.lock()
806 if os.path.exists(self.sjoin("undo")):
806 if os.path.exists(self.sjoin("undo")):
807 return self._rollback(dryrun, force)
807 return self._rollback(dryrun, force)
808 else:
808 else:
809 self.ui.warn(_("no rollback information available\n"))
809 self.ui.warn(_("no rollback information available\n"))
810 return 1
810 return 1
811 finally:
811 finally:
812 release(lock, wlock)
812 release(lock, wlock)
813
813
814 def _rollback(self, dryrun, force):
814 def _rollback(self, dryrun, force):
815 ui = self.ui
815 ui = self.ui
816 try:
816 try:
817 args = self.opener.read('undo.desc').splitlines()
817 args = self.opener.read('undo.desc').splitlines()
818 (oldlen, desc, detail) = (int(args[0]), args[1], None)
818 (oldlen, desc, detail) = (int(args[0]), args[1], None)
819 if len(args) >= 3:
819 if len(args) >= 3:
820 detail = args[2]
820 detail = args[2]
821 oldtip = oldlen - 1
821 oldtip = oldlen - 1
822
822
823 if detail and ui.verbose:
823 if detail and ui.verbose:
824 msg = (_('repository tip rolled back to revision %s'
824 msg = (_('repository tip rolled back to revision %s'
825 ' (undo %s: %s)\n')
825 ' (undo %s: %s)\n')
826 % (oldtip, desc, detail))
826 % (oldtip, desc, detail))
827 else:
827 else:
828 msg = (_('repository tip rolled back to revision %s'
828 msg = (_('repository tip rolled back to revision %s'
829 ' (undo %s)\n')
829 ' (undo %s)\n')
830 % (oldtip, desc))
830 % (oldtip, desc))
831 except IOError:
831 except IOError:
832 msg = _('rolling back unknown transaction\n')
832 msg = _('rolling back unknown transaction\n')
833 desc = None
833 desc = None
834
834
835 if not force and self['.'] != self['tip'] and desc == 'commit':
835 if not force and self['.'] != self['tip'] and desc == 'commit':
836 raise util.Abort(
836 raise util.Abort(
837 _('rollback of last commit while not checked out '
837 _('rollback of last commit while not checked out '
838 'may lose data'), hint=_('use -f to force'))
838 'may lose data'), hint=_('use -f to force'))
839
839
840 ui.status(msg)
840 ui.status(msg)
841 if dryrun:
841 if dryrun:
842 return 0
842 return 0
843
843
844 parents = self.dirstate.parents()
844 parents = self.dirstate.parents()
845 transaction.rollback(self.sopener, self.sjoin('undo'), ui.warn)
845 transaction.rollback(self.sopener, self.sjoin('undo'), ui.warn)
846 if os.path.exists(self.join('undo.bookmarks')):
846 if os.path.exists(self.join('undo.bookmarks')):
847 util.rename(self.join('undo.bookmarks'),
847 util.rename(self.join('undo.bookmarks'),
848 self.join('bookmarks'))
848 self.join('bookmarks'))
849 if os.path.exists(self.sjoin('undo.phaseroots')):
849 if os.path.exists(self.sjoin('undo.phaseroots')):
850 util.rename(self.sjoin('undo.phaseroots'),
850 util.rename(self.sjoin('undo.phaseroots'),
851 self.sjoin('phaseroots'))
851 self.sjoin('phaseroots'))
852 self.invalidate()
852 self.invalidate()
853
853
854 parentgone = (parents[0] not in self.changelog.nodemap or
854 parentgone = (parents[0] not in self.changelog.nodemap or
855 parents[1] not in self.changelog.nodemap)
855 parents[1] not in self.changelog.nodemap)
856 if parentgone:
856 if parentgone:
857 util.rename(self.join('undo.dirstate'), self.join('dirstate'))
857 util.rename(self.join('undo.dirstate'), self.join('dirstate'))
858 try:
858 try:
859 branch = self.opener.read('undo.branch')
859 branch = self.opener.read('undo.branch')
860 self.dirstate.setbranch(branch)
860 self.dirstate.setbranch(branch)
861 except IOError:
861 except IOError:
862 ui.warn(_('named branch could not be reset: '
862 ui.warn(_('named branch could not be reset: '
863 'current branch is still \'%s\'\n')
863 'current branch is still \'%s\'\n')
864 % self.dirstate.branch())
864 % self.dirstate.branch())
865
865
866 self.dirstate.invalidate()
866 self.dirstate.invalidate()
867 parents = tuple([p.rev() for p in self.parents()])
867 parents = tuple([p.rev() for p in self.parents()])
868 if len(parents) > 1:
868 if len(parents) > 1:
869 ui.status(_('working directory now based on '
869 ui.status(_('working directory now based on '
870 'revisions %d and %d\n') % parents)
870 'revisions %d and %d\n') % parents)
871 else:
871 else:
872 ui.status(_('working directory now based on '
872 ui.status(_('working directory now based on '
873 'revision %d\n') % parents)
873 'revision %d\n') % parents)
874 self.destroyed()
874 self.destroyed()
875 return 0
875 return 0
876
876
877 def invalidatecaches(self):
877 def invalidatecaches(self):
878 def delcache(name):
878 def delcache(name):
879 try:
879 try:
880 delattr(self, name)
880 delattr(self, name)
881 except AttributeError:
881 except AttributeError:
882 pass
882 pass
883
883
884 delcache('_tagscache')
884 delcache('_tagscache')
885 delcache('_phaserev')
885 delcache('_phaserev')
886
886
887 self._branchcache = None # in UTF-8
887 self._branchcache = None # in UTF-8
888 self._branchcachetip = None
888 self._branchcachetip = None
889
889
890 def invalidatedirstate(self):
890 def invalidatedirstate(self):
891 '''Invalidates the dirstate, causing the next call to dirstate
891 '''Invalidates the dirstate, causing the next call to dirstate
892 to check if it was modified since the last time it was read,
892 to check if it was modified since the last time it was read,
893 rereading it if it has.
893 rereading it if it has.
894
894
895 This is different to dirstate.invalidate() that it doesn't always
895 This is different to dirstate.invalidate() that it doesn't always
896 rereads the dirstate. Use dirstate.invalidate() if you want to
896 rereads the dirstate. Use dirstate.invalidate() if you want to
897 explicitly read the dirstate again (i.e. restoring it to a previous
897 explicitly read the dirstate again (i.e. restoring it to a previous
898 known good state).'''
898 known good state).'''
899 try:
899 try:
900 delattr(self, 'dirstate')
900 delattr(self, 'dirstate')
901 except AttributeError:
901 except AttributeError:
902 pass
902 pass
903
903
904 def invalidate(self):
904 def invalidate(self):
905 for k in self._filecache:
905 for k in self._filecache:
906 # dirstate is invalidated separately in invalidatedirstate()
906 # dirstate is invalidated separately in invalidatedirstate()
907 if k == 'dirstate':
907 if k == 'dirstate':
908 continue
908 continue
909
909
910 try:
910 try:
911 delattr(self, k)
911 delattr(self, k)
912 except AttributeError:
912 except AttributeError:
913 pass
913 pass
914 self.invalidatecaches()
914 self.invalidatecaches()
915
915
916 def _lock(self, lockname, wait, releasefn, acquirefn, desc):
916 def _lock(self, lockname, wait, releasefn, acquirefn, desc):
917 try:
917 try:
918 l = lock.lock(lockname, 0, releasefn, desc=desc)
918 l = lock.lock(lockname, 0, releasefn, desc=desc)
919 except error.LockHeld, inst:
919 except error.LockHeld, inst:
920 if not wait:
920 if not wait:
921 raise
921 raise
922 self.ui.warn(_("waiting for lock on %s held by %r\n") %
922 self.ui.warn(_("waiting for lock on %s held by %r\n") %
923 (desc, inst.locker))
923 (desc, inst.locker))
924 # default to 600 seconds timeout
924 # default to 600 seconds timeout
925 l = lock.lock(lockname, int(self.ui.config("ui", "timeout", "600")),
925 l = lock.lock(lockname, int(self.ui.config("ui", "timeout", "600")),
926 releasefn, desc=desc)
926 releasefn, desc=desc)
927 if acquirefn:
927 if acquirefn:
928 acquirefn()
928 acquirefn()
929 return l
929 return l
930
930
931 def _afterlock(self, callback):
931 def _afterlock(self, callback):
932 """add a callback to the current repository lock.
932 """add a callback to the current repository lock.
933
933
934 The callback will be executed on lock release."""
934 The callback will be executed on lock release."""
935 l = self._lockref and self._lockref()
935 l = self._lockref and self._lockref()
936 if l:
936 if l:
937 l.postrelease.append(callback)
937 l.postrelease.append(callback)
938
938
939 def lock(self, wait=True):
939 def lock(self, wait=True):
940 '''Lock the repository store (.hg/store) and return a weak reference
940 '''Lock the repository store (.hg/store) and return a weak reference
941 to the lock. Use this before modifying the store (e.g. committing or
941 to the lock. Use this before modifying the store (e.g. committing or
942 stripping). If you are opening a transaction, get a lock as well.)'''
942 stripping). If you are opening a transaction, get a lock as well.)'''
943 l = self._lockref and self._lockref()
943 l = self._lockref and self._lockref()
944 if l is not None and l.held:
944 if l is not None and l.held:
945 l.lock()
945 l.lock()
946 return l
946 return l
947
947
948 def unlock():
948 def unlock():
949 self.store.write()
949 self.store.write()
950 if self._dirtyphases:
950 if self._dirtyphases:
951 phases.writeroots(self)
951 phases.writeroots(self)
952 for k, ce in self._filecache.items():
952 for k, ce in self._filecache.items():
953 if k == 'dirstate':
953 if k == 'dirstate':
954 continue
954 continue
955 ce.refresh()
955 ce.refresh()
956
956
957 l = self._lock(self.sjoin("lock"), wait, unlock,
957 l = self._lock(self.sjoin("lock"), wait, unlock,
958 self.invalidate, _('repository %s') % self.origroot)
958 self.invalidate, _('repository %s') % self.origroot)
959 self._lockref = weakref.ref(l)
959 self._lockref = weakref.ref(l)
960 return l
960 return l
961
961
962 def wlock(self, wait=True):
962 def wlock(self, wait=True):
963 '''Lock the non-store parts of the repository (everything under
963 '''Lock the non-store parts of the repository (everything under
964 .hg except .hg/store) and return a weak reference to the lock.
964 .hg except .hg/store) and return a weak reference to the lock.
965 Use this before modifying files in .hg.'''
965 Use this before modifying files in .hg.'''
966 l = self._wlockref and self._wlockref()
966 l = self._wlockref and self._wlockref()
967 if l is not None and l.held:
967 if l is not None and l.held:
968 l.lock()
968 l.lock()
969 return l
969 return l
970
970
971 def unlock():
971 def unlock():
972 self.dirstate.write()
972 self.dirstate.write()
973 ce = self._filecache.get('dirstate')
973 ce = self._filecache.get('dirstate')
974 if ce:
974 if ce:
975 ce.refresh()
975 ce.refresh()
976
976
977 l = self._lock(self.join("wlock"), wait, unlock,
977 l = self._lock(self.join("wlock"), wait, unlock,
978 self.invalidatedirstate, _('working directory of %s') %
978 self.invalidatedirstate, _('working directory of %s') %
979 self.origroot)
979 self.origroot)
980 self._wlockref = weakref.ref(l)
980 self._wlockref = weakref.ref(l)
981 return l
981 return l
982
982
983 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
983 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
984 """
984 """
985 commit an individual file as part of a larger transaction
985 commit an individual file as part of a larger transaction
986 """
986 """
987
987
988 fname = fctx.path()
988 fname = fctx.path()
989 text = fctx.data()
989 text = fctx.data()
990 flog = self.file(fname)
990 flog = self.file(fname)
991 fparent1 = manifest1.get(fname, nullid)
991 fparent1 = manifest1.get(fname, nullid)
992 fparent2 = fparent2o = manifest2.get(fname, nullid)
992 fparent2 = fparent2o = manifest2.get(fname, nullid)
993
993
994 meta = {}
994 meta = {}
995 copy = fctx.renamed()
995 copy = fctx.renamed()
996 if copy and copy[0] != fname:
996 if copy and copy[0] != fname:
997 # Mark the new revision of this file as a copy of another
997 # Mark the new revision of this file as a copy of another
998 # file. This copy data will effectively act as a parent
998 # file. This copy data will effectively act as a parent
999 # of this new revision. If this is a merge, the first
999 # of this new revision. If this is a merge, the first
1000 # parent will be the nullid (meaning "look up the copy data")
1000 # parent will be the nullid (meaning "look up the copy data")
1001 # and the second one will be the other parent. For example:
1001 # and the second one will be the other parent. For example:
1002 #
1002 #
1003 # 0 --- 1 --- 3 rev1 changes file foo
1003 # 0 --- 1 --- 3 rev1 changes file foo
1004 # \ / rev2 renames foo to bar and changes it
1004 # \ / rev2 renames foo to bar and changes it
1005 # \- 2 -/ rev3 should have bar with all changes and
1005 # \- 2 -/ rev3 should have bar with all changes and
1006 # should record that bar descends from
1006 # should record that bar descends from
1007 # bar in rev2 and foo in rev1
1007 # bar in rev2 and foo in rev1
1008 #
1008 #
1009 # this allows this merge to succeed:
1009 # this allows this merge to succeed:
1010 #
1010 #
1011 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
1011 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
1012 # \ / merging rev3 and rev4 should use bar@rev2
1012 # \ / merging rev3 and rev4 should use bar@rev2
1013 # \- 2 --- 4 as the merge base
1013 # \- 2 --- 4 as the merge base
1014 #
1014 #
1015
1015
1016 cfname = copy[0]
1016 cfname = copy[0]
1017 crev = manifest1.get(cfname)
1017 crev = manifest1.get(cfname)
1018 newfparent = fparent2
1018 newfparent = fparent2
1019
1019
1020 if manifest2: # branch merge
1020 if manifest2: # branch merge
1021 if fparent2 == nullid or crev is None: # copied on remote side
1021 if fparent2 == nullid or crev is None: # copied on remote side
1022 if cfname in manifest2:
1022 if cfname in manifest2:
1023 crev = manifest2[cfname]
1023 crev = manifest2[cfname]
1024 newfparent = fparent1
1024 newfparent = fparent1
1025
1025
1026 # find source in nearest ancestor if we've lost track
1026 # find source in nearest ancestor if we've lost track
1027 if not crev:
1027 if not crev:
1028 self.ui.debug(" %s: searching for copy revision for %s\n" %
1028 self.ui.debug(" %s: searching for copy revision for %s\n" %
1029 (fname, cfname))
1029 (fname, cfname))
1030 for ancestor in self[None].ancestors():
1030 for ancestor in self[None].ancestors():
1031 if cfname in ancestor:
1031 if cfname in ancestor:
1032 crev = ancestor[cfname].filenode()
1032 crev = ancestor[cfname].filenode()
1033 break
1033 break
1034
1034
1035 if crev:
1035 if crev:
1036 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
1036 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
1037 meta["copy"] = cfname
1037 meta["copy"] = cfname
1038 meta["copyrev"] = hex(crev)
1038 meta["copyrev"] = hex(crev)
1039 fparent1, fparent2 = nullid, newfparent
1039 fparent1, fparent2 = nullid, newfparent
1040 else:
1040 else:
1041 self.ui.warn(_("warning: can't find ancestor for '%s' "
1041 self.ui.warn(_("warning: can't find ancestor for '%s' "
1042 "copied from '%s'!\n") % (fname, cfname))
1042 "copied from '%s'!\n") % (fname, cfname))
1043
1043
1044 elif fparent2 != nullid:
1044 elif fparent2 != nullid:
1045 # is one parent an ancestor of the other?
1045 # is one parent an ancestor of the other?
1046 fparentancestor = flog.ancestor(fparent1, fparent2)
1046 fparentancestor = flog.ancestor(fparent1, fparent2)
1047 if fparentancestor == fparent1:
1047 if fparentancestor == fparent1:
1048 fparent1, fparent2 = fparent2, nullid
1048 fparent1, fparent2 = fparent2, nullid
1049 elif fparentancestor == fparent2:
1049 elif fparentancestor == fparent2:
1050 fparent2 = nullid
1050 fparent2 = nullid
1051
1051
1052 # is the file changed?
1052 # is the file changed?
1053 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
1053 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
1054 changelist.append(fname)
1054 changelist.append(fname)
1055 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
1055 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
1056
1056
1057 # are just the flags changed during merge?
1057 # are just the flags changed during merge?
1058 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
1058 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
1059 changelist.append(fname)
1059 changelist.append(fname)
1060
1060
1061 return fparent1
1061 return fparent1
1062
1062
1063 def commit(self, text="", user=None, date=None, match=None, force=False,
1063 def commit(self, text="", user=None, date=None, match=None, force=False,
1064 editor=False, extra={}):
1064 editor=False, extra={}):
1065 """Add a new revision to current repository.
1065 """Add a new revision to current repository.
1066
1066
1067 Revision information is gathered from the working directory,
1067 Revision information is gathered from the working directory,
1068 match can be used to filter the committed files. If editor is
1068 match can be used to filter the committed files. If editor is
1069 supplied, it is called to get a commit message.
1069 supplied, it is called to get a commit message.
1070 """
1070 """
1071
1071
1072 def fail(f, msg):
1072 def fail(f, msg):
1073 raise util.Abort('%s: %s' % (f, msg))
1073 raise util.Abort('%s: %s' % (f, msg))
1074
1074
1075 if not match:
1075 if not match:
1076 match = matchmod.always(self.root, '')
1076 match = matchmod.always(self.root, '')
1077
1077
1078 if not force:
1078 if not force:
1079 vdirs = []
1079 vdirs = []
1080 match.dir = vdirs.append
1080 match.dir = vdirs.append
1081 match.bad = fail
1081 match.bad = fail
1082
1082
1083 wlock = self.wlock()
1083 wlock = self.wlock()
1084 try:
1084 try:
1085 wctx = self[None]
1085 wctx = self[None]
1086 merge = len(wctx.parents()) > 1
1086 merge = len(wctx.parents()) > 1
1087
1087
1088 if (not force and merge and match and
1088 if (not force and merge and match and
1089 (match.files() or match.anypats())):
1089 (match.files() or match.anypats())):
1090 raise util.Abort(_('cannot partially commit a merge '
1090 raise util.Abort(_('cannot partially commit a merge '
1091 '(do not specify files or patterns)'))
1091 '(do not specify files or patterns)'))
1092
1092
1093 changes = self.status(match=match, clean=force)
1093 changes = self.status(match=match, clean=force)
1094 if force:
1094 if force:
1095 changes[0].extend(changes[6]) # mq may commit unchanged files
1095 changes[0].extend(changes[6]) # mq may commit unchanged files
1096
1096
1097 # check subrepos
1097 # check subrepos
1098 subs = []
1098 subs = []
1099 removedsubs = set()
1099 removedsubs = set()
1100 if '.hgsub' in wctx:
1100 if '.hgsub' in wctx:
1101 # only manage subrepos and .hgsubstate if .hgsub is present
1101 # only manage subrepos and .hgsubstate if .hgsub is present
1102 for p in wctx.parents():
1102 for p in wctx.parents():
1103 removedsubs.update(s for s in p.substate if match(s))
1103 removedsubs.update(s for s in p.substate if match(s))
1104 for s in wctx.substate:
1104 for s in wctx.substate:
1105 removedsubs.discard(s)
1105 removedsubs.discard(s)
1106 if match(s) and wctx.sub(s).dirty():
1106 if match(s) and wctx.sub(s).dirty():
1107 subs.append(s)
1107 subs.append(s)
1108 if (subs or removedsubs):
1108 if (subs or removedsubs):
1109 if (not match('.hgsub') and
1109 if (not match('.hgsub') and
1110 '.hgsub' in (wctx.modified() + wctx.added())):
1110 '.hgsub' in (wctx.modified() + wctx.added())):
1111 raise util.Abort(
1111 raise util.Abort(
1112 _("can't commit subrepos without .hgsub"))
1112 _("can't commit subrepos without .hgsub"))
1113 if '.hgsubstate' not in changes[0]:
1113 if '.hgsubstate' not in changes[0]:
1114 changes[0].insert(0, '.hgsubstate')
1114 changes[0].insert(0, '.hgsubstate')
1115 if '.hgsubstate' in changes[2]:
1115 if '.hgsubstate' in changes[2]:
1116 changes[2].remove('.hgsubstate')
1116 changes[2].remove('.hgsubstate')
1117 elif '.hgsub' in changes[2]:
1117 elif '.hgsub' in changes[2]:
1118 # clean up .hgsubstate when .hgsub is removed
1118 # clean up .hgsubstate when .hgsub is removed
1119 if ('.hgsubstate' in wctx and
1119 if ('.hgsubstate' in wctx and
1120 '.hgsubstate' not in changes[0] + changes[1] + changes[2]):
1120 '.hgsubstate' not in changes[0] + changes[1] + changes[2]):
1121 changes[2].insert(0, '.hgsubstate')
1121 changes[2].insert(0, '.hgsubstate')
1122
1122
1123 if subs and not self.ui.configbool('ui', 'commitsubrepos', False):
1123 if subs and not self.ui.configbool('ui', 'commitsubrepos', False):
1124 changedsubs = [s for s in subs if wctx.sub(s).dirty(True)]
1124 changedsubs = [s for s in subs if wctx.sub(s).dirty(True)]
1125 if changedsubs:
1125 if changedsubs:
1126 raise util.Abort(_("uncommitted changes in subrepo %s")
1126 raise util.Abort(_("uncommitted changes in subrepo %s")
1127 % changedsubs[0],
1127 % changedsubs[0],
1128 hint=_("use --subrepos for recursive commit"))
1128 hint=_("use --subrepos for recursive commit"))
1129
1129
1130 # make sure all explicit patterns are matched
1130 # make sure all explicit patterns are matched
1131 if not force and match.files():
1131 if not force and match.files():
1132 matched = set(changes[0] + changes[1] + changes[2])
1132 matched = set(changes[0] + changes[1] + changes[2])
1133
1133
1134 for f in match.files():
1134 for f in match.files():
1135 if f == '.' or f in matched or f in wctx.substate:
1135 if f == '.' or f in matched or f in wctx.substate:
1136 continue
1136 continue
1137 if f in changes[3]: # missing
1137 if f in changes[3]: # missing
1138 fail(f, _('file not found!'))
1138 fail(f, _('file not found!'))
1139 if f in vdirs: # visited directory
1139 if f in vdirs: # visited directory
1140 d = f + '/'
1140 d = f + '/'
1141 for mf in matched:
1141 for mf in matched:
1142 if mf.startswith(d):
1142 if mf.startswith(d):
1143 break
1143 break
1144 else:
1144 else:
1145 fail(f, _("no match under directory!"))
1145 fail(f, _("no match under directory!"))
1146 elif f not in self.dirstate:
1146 elif f not in self.dirstate:
1147 fail(f, _("file not tracked!"))
1147 fail(f, _("file not tracked!"))
1148
1148
1149 if (not force and not extra.get("close") and not merge
1149 if (not force and not extra.get("close") and not merge
1150 and not (changes[0] or changes[1] or changes[2])
1150 and not (changes[0] or changes[1] or changes[2])
1151 and wctx.branch() == wctx.p1().branch()):
1151 and wctx.branch() == wctx.p1().branch()):
1152 return None
1152 return None
1153
1153
1154 ms = mergemod.mergestate(self)
1154 ms = mergemod.mergestate(self)
1155 for f in changes[0]:
1155 for f in changes[0]:
1156 if f in ms and ms[f] == 'u':
1156 if f in ms and ms[f] == 'u':
1157 raise util.Abort(_("unresolved merge conflicts "
1157 raise util.Abort(_("unresolved merge conflicts "
1158 "(see hg help resolve)"))
1158 "(see hg help resolve)"))
1159
1159
1160 cctx = context.workingctx(self, text, user, date, extra, changes)
1160 cctx = context.workingctx(self, text, user, date, extra, changes)
1161 if editor:
1161 if editor:
1162 cctx._text = editor(self, cctx, subs)
1162 cctx._text = editor(self, cctx, subs)
1163 edited = (text != cctx._text)
1163 edited = (text != cctx._text)
1164
1164
1165 # commit subs
1165 # commit subs
1166 if subs or removedsubs:
1166 if subs or removedsubs:
1167 state = wctx.substate.copy()
1167 state = wctx.substate.copy()
1168 for s in sorted(subs):
1168 for s in sorted(subs):
1169 sub = wctx.sub(s)
1169 sub = wctx.sub(s)
1170 self.ui.status(_('committing subrepository %s\n') %
1170 self.ui.status(_('committing subrepository %s\n') %
1171 subrepo.subrelpath(sub))
1171 subrepo.subrelpath(sub))
1172 sr = sub.commit(cctx._text, user, date)
1172 sr = sub.commit(cctx._text, user, date)
1173 state[s] = (state[s][0], sr)
1173 state[s] = (state[s][0], sr)
1174 subrepo.writestate(self, state)
1174 subrepo.writestate(self, state)
1175
1175
1176 # Save commit message in case this transaction gets rolled back
1176 # Save commit message in case this transaction gets rolled back
1177 # (e.g. by a pretxncommit hook). Leave the content alone on
1177 # (e.g. by a pretxncommit hook). Leave the content alone on
1178 # the assumption that the user will use the same editor again.
1178 # the assumption that the user will use the same editor again.
1179 msgfn = self.savecommitmessage(cctx._text)
1179 msgfn = self.savecommitmessage(cctx._text)
1180
1180
1181 p1, p2 = self.dirstate.parents()
1181 p1, p2 = self.dirstate.parents()
1182 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1182 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1183 try:
1183 try:
1184 self.hook("precommit", throw=True, parent1=hookp1, parent2=hookp2)
1184 self.hook("precommit", throw=True, parent1=hookp1, parent2=hookp2)
1185 ret = self.commitctx(cctx, True)
1185 ret = self.commitctx(cctx, True)
1186 except:
1186 except:
1187 if edited:
1187 if edited:
1188 self.ui.write(
1188 self.ui.write(
1189 _('note: commit message saved in %s\n') % msgfn)
1189 _('note: commit message saved in %s\n') % msgfn)
1190 raise
1190 raise
1191
1191
1192 # update bookmarks, dirstate and mergestate
1192 # update bookmarks, dirstate and mergestate
1193 bookmarks.update(self, p1, ret)
1193 bookmarks.update(self, p1, ret)
1194 for f in changes[0] + changes[1]:
1194 for f in changes[0] + changes[1]:
1195 self.dirstate.normal(f)
1195 self.dirstate.normal(f)
1196 for f in changes[2]:
1196 for f in changes[2]:
1197 self.dirstate.drop(f)
1197 self.dirstate.drop(f)
1198 self.dirstate.setparents(ret)
1198 self.dirstate.setparents(ret)
1199 ms.reset()
1199 ms.reset()
1200 finally:
1200 finally:
1201 wlock.release()
1201 wlock.release()
1202
1202
1203 self.hook("commit", node=hex(ret), parent1=hookp1, parent2=hookp2)
1203 self.hook("commit", node=hex(ret), parent1=hookp1, parent2=hookp2)
1204 return ret
1204 return ret
1205
1205
1206 def commitctx(self, ctx, error=False):
1206 def commitctx(self, ctx, error=False):
1207 """Add a new revision to current repository.
1207 """Add a new revision to current repository.
1208 Revision information is passed via the context argument.
1208 Revision information is passed via the context argument.
1209 """
1209 """
1210
1210
1211 tr = lock = None
1211 tr = lock = None
1212 removed = list(ctx.removed())
1212 removed = list(ctx.removed())
1213 p1, p2 = ctx.p1(), ctx.p2()
1213 p1, p2 = ctx.p1(), ctx.p2()
1214 user = ctx.user()
1214 user = ctx.user()
1215
1215
1216 lock = self.lock()
1216 lock = self.lock()
1217 try:
1217 try:
1218 tr = self.transaction("commit")
1218 tr = self.transaction("commit")
1219 trp = weakref.proxy(tr)
1219 trp = weakref.proxy(tr)
1220
1220
1221 if ctx.files():
1221 if ctx.files():
1222 m1 = p1.manifest().copy()
1222 m1 = p1.manifest().copy()
1223 m2 = p2.manifest()
1223 m2 = p2.manifest()
1224
1224
1225 # check in files
1225 # check in files
1226 new = {}
1226 new = {}
1227 changed = []
1227 changed = []
1228 linkrev = len(self)
1228 linkrev = len(self)
1229 for f in sorted(ctx.modified() + ctx.added()):
1229 for f in sorted(ctx.modified() + ctx.added()):
1230 self.ui.note(f + "\n")
1230 self.ui.note(f + "\n")
1231 try:
1231 try:
1232 fctx = ctx[f]
1232 fctx = ctx[f]
1233 new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
1233 new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
1234 changed)
1234 changed)
1235 m1.set(f, fctx.flags())
1235 m1.set(f, fctx.flags())
1236 except OSError, inst:
1236 except OSError, inst:
1237 self.ui.warn(_("trouble committing %s!\n") % f)
1237 self.ui.warn(_("trouble committing %s!\n") % f)
1238 raise
1238 raise
1239 except IOError, inst:
1239 except IOError, inst:
1240 errcode = getattr(inst, 'errno', errno.ENOENT)
1240 errcode = getattr(inst, 'errno', errno.ENOENT)
1241 if error or errcode and errcode != errno.ENOENT:
1241 if error or errcode and errcode != errno.ENOENT:
1242 self.ui.warn(_("trouble committing %s!\n") % f)
1242 self.ui.warn(_("trouble committing %s!\n") % f)
1243 raise
1243 raise
1244 else:
1244 else:
1245 removed.append(f)
1245 removed.append(f)
1246
1246
1247 # update manifest
1247 # update manifest
1248 m1.update(new)
1248 m1.update(new)
1249 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1249 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1250 drop = [f for f in removed if f in m1]
1250 drop = [f for f in removed if f in m1]
1251 for f in drop:
1251 for f in drop:
1252 del m1[f]
1252 del m1[f]
1253 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1253 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1254 p2.manifestnode(), (new, drop))
1254 p2.manifestnode(), (new, drop))
1255 files = changed + removed
1255 files = changed + removed
1256 else:
1256 else:
1257 mn = p1.manifestnode()
1257 mn = p1.manifestnode()
1258 files = []
1258 files = []
1259
1259
1260 # update changelog
1260 # update changelog
1261 self.changelog.delayupdate()
1261 self.changelog.delayupdate()
1262 n = self.changelog.add(mn, files, ctx.description(),
1262 n = self.changelog.add(mn, files, ctx.description(),
1263 trp, p1.node(), p2.node(),
1263 trp, p1.node(), p2.node(),
1264 user, ctx.date(), ctx.extra().copy())
1264 user, ctx.date(), ctx.extra().copy())
1265 p = lambda: self.changelog.writepending() and self.root or ""
1265 p = lambda: self.changelog.writepending() and self.root or ""
1266 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1266 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1267 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1267 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1268 parent2=xp2, pending=p)
1268 parent2=xp2, pending=p)
1269 self.changelog.finalize(trp)
1269 self.changelog.finalize(trp)
1270 # set the new commit is proper phase
1270 # set the new commit is proper phase
1271 targetphase = self.ui.configint('phases', 'new-commit',
1271 targetphase = self.ui.configint('phases', 'new-commit',
1272 phases.draft)
1272 phases.draft)
1273 if targetphase:
1273 if targetphase:
1274 # retract boundary do not alter parent changeset.
1274 # retract boundary do not alter parent changeset.
1275 # if a parent have higher the resulting phase will
1275 # if a parent have higher the resulting phase will
1276 # be compliant anyway
1276 # be compliant anyway
1277 #
1277 #
1278 # if minimal phase was 0 we don't need to retract anything
1278 # if minimal phase was 0 we don't need to retract anything
1279 phases.retractboundary(self, targetphase, [n])
1279 phases.retractboundary(self, targetphase, [n])
1280 tr.close()
1280 tr.close()
1281 self.updatebranchcache()
1281 self.updatebranchcache()
1282 return n
1282 return n
1283 finally:
1283 finally:
1284 if tr:
1284 if tr:
1285 tr.release()
1285 tr.release()
1286 lock.release()
1286 lock.release()
1287
1287
1288 def destroyed(self):
1288 def destroyed(self):
1289 '''Inform the repository that nodes have been destroyed.
1289 '''Inform the repository that nodes have been destroyed.
1290 Intended for use by strip and rollback, so there's a common
1290 Intended for use by strip and rollback, so there's a common
1291 place for anything that has to be done after destroying history.'''
1291 place for anything that has to be done after destroying history.'''
1292 # XXX it might be nice if we could take the list of destroyed
1292 # XXX it might be nice if we could take the list of destroyed
1293 # nodes, but I don't see an easy way for rollback() to do that
1293 # nodes, but I don't see an easy way for rollback() to do that
1294
1294
1295 # Ensure the persistent tag cache is updated. Doing it now
1295 # Ensure the persistent tag cache is updated. Doing it now
1296 # means that the tag cache only has to worry about destroyed
1296 # means that the tag cache only has to worry about destroyed
1297 # heads immediately after a strip/rollback. That in turn
1297 # heads immediately after a strip/rollback. That in turn
1298 # guarantees that "cachetip == currenttip" (comparing both rev
1298 # guarantees that "cachetip == currenttip" (comparing both rev
1299 # and node) always means no nodes have been added or destroyed.
1299 # and node) always means no nodes have been added or destroyed.
1300
1300
1301 # XXX this is suboptimal when qrefresh'ing: we strip the current
1301 # XXX this is suboptimal when qrefresh'ing: we strip the current
1302 # head, refresh the tag cache, then immediately add a new head.
1302 # head, refresh the tag cache, then immediately add a new head.
1303 # But I think doing it this way is necessary for the "instant
1303 # But I think doing it this way is necessary for the "instant
1304 # tag cache retrieval" case to work.
1304 # tag cache retrieval" case to work.
1305 self.invalidatecaches()
1305 self.invalidatecaches()
1306
1306
1307 def walk(self, match, node=None):
1307 def walk(self, match, node=None):
1308 '''
1308 '''
1309 walk recursively through the directory tree or a given
1309 walk recursively through the directory tree or a given
1310 changeset, finding all files matched by the match
1310 changeset, finding all files matched by the match
1311 function
1311 function
1312 '''
1312 '''
1313 return self[node].walk(match)
1313 return self[node].walk(match)
1314
1314
1315 def status(self, node1='.', node2=None, match=None,
1315 def status(self, node1='.', node2=None, match=None,
1316 ignored=False, clean=False, unknown=False,
1316 ignored=False, clean=False, unknown=False,
1317 listsubrepos=False):
1317 listsubrepos=False):
1318 """return status of files between two nodes or node and working directory
1318 """return status of files between two nodes or node and working directory
1319
1319
1320 If node1 is None, use the first dirstate parent instead.
1320 If node1 is None, use the first dirstate parent instead.
1321 If node2 is None, compare node1 with working directory.
1321 If node2 is None, compare node1 with working directory.
1322 """
1322 """
1323
1323
1324 def mfmatches(ctx):
1324 def mfmatches(ctx):
1325 mf = ctx.manifest().copy()
1325 mf = ctx.manifest().copy()
1326 for fn in mf.keys():
1326 for fn in mf.keys():
1327 if not match(fn):
1327 if not match(fn):
1328 del mf[fn]
1328 del mf[fn]
1329 return mf
1329 return mf
1330
1330
1331 if isinstance(node1, context.changectx):
1331 if isinstance(node1, context.changectx):
1332 ctx1 = node1
1332 ctx1 = node1
1333 else:
1333 else:
1334 ctx1 = self[node1]
1334 ctx1 = self[node1]
1335 if isinstance(node2, context.changectx):
1335 if isinstance(node2, context.changectx):
1336 ctx2 = node2
1336 ctx2 = node2
1337 else:
1337 else:
1338 ctx2 = self[node2]
1338 ctx2 = self[node2]
1339
1339
1340 working = ctx2.rev() is None
1340 working = ctx2.rev() is None
1341 parentworking = working and ctx1 == self['.']
1341 parentworking = working and ctx1 == self['.']
1342 match = match or matchmod.always(self.root, self.getcwd())
1342 match = match or matchmod.always(self.root, self.getcwd())
1343 listignored, listclean, listunknown = ignored, clean, unknown
1343 listignored, listclean, listunknown = ignored, clean, unknown
1344
1344
1345 # load earliest manifest first for caching reasons
1345 # load earliest manifest first for caching reasons
1346 if not working and ctx2.rev() < ctx1.rev():
1346 if not working and ctx2.rev() < ctx1.rev():
1347 ctx2.manifest()
1347 ctx2.manifest()
1348
1348
1349 if not parentworking:
1349 if not parentworking:
1350 def bad(f, msg):
1350 def bad(f, msg):
1351 if f not in ctx1:
1351 if f not in ctx1:
1352 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1352 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1353 match.bad = bad
1353 match.bad = bad
1354
1354
1355 if working: # we need to scan the working dir
1355 if working: # we need to scan the working dir
1356 subrepos = []
1356 subrepos = []
1357 if '.hgsub' in self.dirstate:
1357 if '.hgsub' in self.dirstate:
1358 subrepos = ctx2.substate.keys()
1358 subrepos = ctx2.substate.keys()
1359 s = self.dirstate.status(match, subrepos, listignored,
1359 s = self.dirstate.status(match, subrepos, listignored,
1360 listclean, listunknown)
1360 listclean, listunknown)
1361 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1361 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1362
1362
1363 # check for any possibly clean files
1363 # check for any possibly clean files
1364 if parentworking and cmp:
1364 if parentworking and cmp:
1365 fixup = []
1365 fixup = []
1366 # do a full compare of any files that might have changed
1366 # do a full compare of any files that might have changed
1367 for f in sorted(cmp):
1367 for f in sorted(cmp):
1368 if (f not in ctx1 or ctx2.flags(f) != ctx1.flags(f)
1368 if (f not in ctx1 or ctx2.flags(f) != ctx1.flags(f)
1369 or ctx1[f].cmp(ctx2[f])):
1369 or ctx1[f].cmp(ctx2[f])):
1370 modified.append(f)
1370 modified.append(f)
1371 else:
1371 else:
1372 fixup.append(f)
1372 fixup.append(f)
1373
1373
1374 # update dirstate for files that are actually clean
1374 # update dirstate for files that are actually clean
1375 if fixup:
1375 if fixup:
1376 if listclean:
1376 if listclean:
1377 clean += fixup
1377 clean += fixup
1378
1378
1379 try:
1379 try:
1380 # updating the dirstate is optional
1380 # updating the dirstate is optional
1381 # so we don't wait on the lock
1381 # so we don't wait on the lock
1382 wlock = self.wlock(False)
1382 wlock = self.wlock(False)
1383 try:
1383 try:
1384 for f in fixup:
1384 for f in fixup:
1385 self.dirstate.normal(f)
1385 self.dirstate.normal(f)
1386 finally:
1386 finally:
1387 wlock.release()
1387 wlock.release()
1388 except error.LockError:
1388 except error.LockError:
1389 pass
1389 pass
1390
1390
1391 if not parentworking:
1391 if not parentworking:
1392 mf1 = mfmatches(ctx1)
1392 mf1 = mfmatches(ctx1)
1393 if working:
1393 if working:
1394 # we are comparing working dir against non-parent
1394 # we are comparing working dir against non-parent
1395 # generate a pseudo-manifest for the working dir
1395 # generate a pseudo-manifest for the working dir
1396 mf2 = mfmatches(self['.'])
1396 mf2 = mfmatches(self['.'])
1397 for f in cmp + modified + added:
1397 for f in cmp + modified + added:
1398 mf2[f] = None
1398 mf2[f] = None
1399 mf2.set(f, ctx2.flags(f))
1399 mf2.set(f, ctx2.flags(f))
1400 for f in removed:
1400 for f in removed:
1401 if f in mf2:
1401 if f in mf2:
1402 del mf2[f]
1402 del mf2[f]
1403 else:
1403 else:
1404 # we are comparing two revisions
1404 # we are comparing two revisions
1405 deleted, unknown, ignored = [], [], []
1405 deleted, unknown, ignored = [], [], []
1406 mf2 = mfmatches(ctx2)
1406 mf2 = mfmatches(ctx2)
1407
1407
1408 modified, added, clean = [], [], []
1408 modified, added, clean = [], [], []
1409 for fn in mf2:
1409 for fn in mf2:
1410 if fn in mf1:
1410 if fn in mf1:
1411 if (fn not in deleted and
1411 if (fn not in deleted and
1412 (mf1.flags(fn) != mf2.flags(fn) or
1412 (mf1.flags(fn) != mf2.flags(fn) or
1413 (mf1[fn] != mf2[fn] and
1413 (mf1[fn] != mf2[fn] and
1414 (mf2[fn] or ctx1[fn].cmp(ctx2[fn]))))):
1414 (mf2[fn] or ctx1[fn].cmp(ctx2[fn]))))):
1415 modified.append(fn)
1415 modified.append(fn)
1416 elif listclean:
1416 elif listclean:
1417 clean.append(fn)
1417 clean.append(fn)
1418 del mf1[fn]
1418 del mf1[fn]
1419 elif fn not in deleted:
1419 elif fn not in deleted:
1420 added.append(fn)
1420 added.append(fn)
1421 removed = mf1.keys()
1421 removed = mf1.keys()
1422
1422
1423 if working and modified and not self.dirstate._checklink:
1423 if working and modified and not self.dirstate._checklink:
1424 # Symlink placeholders may get non-symlink-like contents
1424 # Symlink placeholders may get non-symlink-like contents
1425 # via user error or dereferencing by NFS or Samba servers,
1425 # via user error or dereferencing by NFS or Samba servers,
1426 # so we filter out any placeholders that don't look like a
1426 # so we filter out any placeholders that don't look like a
1427 # symlink
1427 # symlink
1428 sane = []
1428 sane = []
1429 for f in modified:
1429 for f in modified:
1430 if ctx2.flags(f) == 'l':
1430 if ctx2.flags(f) == 'l':
1431 d = ctx2[f].data()
1431 d = ctx2[f].data()
1432 if len(d) >= 1024 or '\n' in d or util.binary(d):
1432 if len(d) >= 1024 or '\n' in d or util.binary(d):
1433 self.ui.debug('ignoring suspect symlink placeholder'
1433 self.ui.debug('ignoring suspect symlink placeholder'
1434 ' "%s"\n' % f)
1434 ' "%s"\n' % f)
1435 continue
1435 continue
1436 sane.append(f)
1436 sane.append(f)
1437 modified = sane
1437 modified = sane
1438
1438
1439 r = modified, added, removed, deleted, unknown, ignored, clean
1439 r = modified, added, removed, deleted, unknown, ignored, clean
1440
1440
1441 if listsubrepos:
1441 if listsubrepos:
1442 for subpath, sub in subrepo.itersubrepos(ctx1, ctx2):
1442 for subpath, sub in subrepo.itersubrepos(ctx1, ctx2):
1443 if working:
1443 if working:
1444 rev2 = None
1444 rev2 = None
1445 else:
1445 else:
1446 rev2 = ctx2.substate[subpath][1]
1446 rev2 = ctx2.substate[subpath][1]
1447 try:
1447 try:
1448 submatch = matchmod.narrowmatcher(subpath, match)
1448 submatch = matchmod.narrowmatcher(subpath, match)
1449 s = sub.status(rev2, match=submatch, ignored=listignored,
1449 s = sub.status(rev2, match=submatch, ignored=listignored,
1450 clean=listclean, unknown=listunknown,
1450 clean=listclean, unknown=listunknown,
1451 listsubrepos=True)
1451 listsubrepos=True)
1452 for rfiles, sfiles in zip(r, s):
1452 for rfiles, sfiles in zip(r, s):
1453 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
1453 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
1454 except error.LookupError:
1454 except error.LookupError:
1455 self.ui.status(_("skipping missing subrepository: %s\n")
1455 self.ui.status(_("skipping missing subrepository: %s\n")
1456 % subpath)
1456 % subpath)
1457
1457
1458 for l in r:
1458 for l in r:
1459 l.sort()
1459 l.sort()
1460 return r
1460 return r
1461
1461
1462 def heads(self, start=None):
1462 def heads(self, start=None):
1463 heads = self.changelog.heads(start)
1463 heads = self.changelog.heads(start)
1464 # sort the output in rev descending order
1464 # sort the output in rev descending order
1465 return sorted(heads, key=self.changelog.rev, reverse=True)
1465 return sorted(heads, key=self.changelog.rev, reverse=True)
1466
1466
1467 def branchheads(self, branch=None, start=None, closed=False):
1467 def branchheads(self, branch=None, start=None, closed=False):
1468 '''return a (possibly filtered) list of heads for the given branch
1468 '''return a (possibly filtered) list of heads for the given branch
1469
1469
1470 Heads are returned in topological order, from newest to oldest.
1470 Heads are returned in topological order, from newest to oldest.
1471 If branch is None, use the dirstate branch.
1471 If branch is None, use the dirstate branch.
1472 If start is not None, return only heads reachable from start.
1472 If start is not None, return only heads reachable from start.
1473 If closed is True, return heads that are marked as closed as well.
1473 If closed is True, return heads that are marked as closed as well.
1474 '''
1474 '''
1475 if branch is None:
1475 if branch is None:
1476 branch = self[None].branch()
1476 branch = self[None].branch()
1477 branches = self.branchmap()
1477 branches = self.branchmap()
1478 if branch not in branches:
1478 if branch not in branches:
1479 return []
1479 return []
1480 # the cache returns heads ordered lowest to highest
1480 # the cache returns heads ordered lowest to highest
1481 bheads = list(reversed(branches[branch]))
1481 bheads = list(reversed(branches[branch]))
1482 if start is not None:
1482 if start is not None:
1483 # filter out the heads that cannot be reached from startrev
1483 # filter out the heads that cannot be reached from startrev
1484 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1484 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1485 bheads = [h for h in bheads if h in fbheads]
1485 bheads = [h for h in bheads if h in fbheads]
1486 if not closed:
1486 if not closed:
1487 bheads = [h for h in bheads if
1487 bheads = [h for h in bheads if
1488 ('close' not in self.changelog.read(h)[5])]
1488 ('close' not in self.changelog.read(h)[5])]
1489 return bheads
1489 return bheads
1490
1490
1491 def branches(self, nodes):
1491 def branches(self, nodes):
1492 if not nodes:
1492 if not nodes:
1493 nodes = [self.changelog.tip()]
1493 nodes = [self.changelog.tip()]
1494 b = []
1494 b = []
1495 for n in nodes:
1495 for n in nodes:
1496 t = n
1496 t = n
1497 while True:
1497 while True:
1498 p = self.changelog.parents(n)
1498 p = self.changelog.parents(n)
1499 if p[1] != nullid or p[0] == nullid:
1499 if p[1] != nullid or p[0] == nullid:
1500 b.append((t, n, p[0], p[1]))
1500 b.append((t, n, p[0], p[1]))
1501 break
1501 break
1502 n = p[0]
1502 n = p[0]
1503 return b
1503 return b
1504
1504
1505 def between(self, pairs):
1505 def between(self, pairs):
1506 r = []
1506 r = []
1507
1507
1508 for top, bottom in pairs:
1508 for top, bottom in pairs:
1509 n, l, i = top, [], 0
1509 n, l, i = top, [], 0
1510 f = 1
1510 f = 1
1511
1511
1512 while n != bottom and n != nullid:
1512 while n != bottom and n != nullid:
1513 p = self.changelog.parents(n)[0]
1513 p = self.changelog.parents(n)[0]
1514 if i == f:
1514 if i == f:
1515 l.append(n)
1515 l.append(n)
1516 f = f * 2
1516 f = f * 2
1517 n = p
1517 n = p
1518 i += 1
1518 i += 1
1519
1519
1520 r.append(l)
1520 r.append(l)
1521
1521
1522 return r
1522 return r
1523
1523
1524 def pull(self, remote, heads=None, force=False):
1524 def pull(self, remote, heads=None, force=False):
1525 lock = self.lock()
1525 lock = self.lock()
1526 try:
1526 try:
1527 tmp = discovery.findcommonincoming(self, remote, heads=heads,
1527 tmp = discovery.findcommonincoming(self, remote, heads=heads,
1528 force=force)
1528 force=force)
1529 common, fetch, rheads = tmp
1529 common, fetch, rheads = tmp
1530 if not fetch:
1530 if not fetch:
1531 self.ui.status(_("no changes found\n"))
1531 self.ui.status(_("no changes found\n"))
1532 added = []
1532 added = []
1533 result = 0
1533 result = 0
1534 else:
1534 else:
1535 if heads is None and list(common) == [nullid]:
1535 if heads is None and list(common) == [nullid]:
1536 self.ui.status(_("requesting all changes\n"))
1536 self.ui.status(_("requesting all changes\n"))
1537 elif heads is None and remote.capable('changegroupsubset'):
1537 elif heads is None and remote.capable('changegroupsubset'):
1538 # issue1320, avoid a race if remote changed after discovery
1538 # issue1320, avoid a race if remote changed after discovery
1539 heads = rheads
1539 heads = rheads
1540
1540
1541 if remote.capable('getbundle'):
1541 if remote.capable('getbundle'):
1542 cg = remote.getbundle('pull', common=common,
1542 cg = remote.getbundle('pull', common=common,
1543 heads=heads or rheads)
1543 heads=heads or rheads)
1544 elif heads is None:
1544 elif heads is None:
1545 cg = remote.changegroup(fetch, 'pull')
1545 cg = remote.changegroup(fetch, 'pull')
1546 elif not remote.capable('changegroupsubset'):
1546 elif not remote.capable('changegroupsubset'):
1547 raise util.Abort(_("partial pull cannot be done because "
1547 raise util.Abort(_("partial pull cannot be done because "
1548 "other repository doesn't support "
1548 "other repository doesn't support "
1549 "changegroupsubset."))
1549 "changegroupsubset."))
1550 else:
1550 else:
1551 cg = remote.changegroupsubset(fetch, heads, 'pull')
1551 cg = remote.changegroupsubset(fetch, heads, 'pull')
1552 clstart = len(self.changelog)
1552 clstart = len(self.changelog)
1553 result = self.addchangegroup(cg, 'pull', remote.url())
1553 result = self.addchangegroup(cg, 'pull', remote.url())
1554 clend = len(self.changelog)
1554 clend = len(self.changelog)
1555 added = [self.changelog.node(r) for r in xrange(clstart, clend)]
1555 added = [self.changelog.node(r) for r in xrange(clstart, clend)]
1556
1556
1557 # compute target subset
1557 # compute target subset
1558 if heads is None:
1558 if heads is None:
1559 # We pulled every thing possible
1559 # We pulled every thing possible
1560 # sync on everything common
1560 # sync on everything common
1561 subset = common + added
1561 subset = common + added
1562 else:
1562 else:
1563 # We pulled a specific subset
1563 # We pulled a specific subset
1564 # sync on this subset
1564 # sync on this subset
1565 subset = heads
1565 subset = heads
1566
1566
1567 # Get remote phases data from remote
1567 # Get remote phases data from remote
1568 remotephases = remote.listkeys('phases')
1568 remotephases = remote.listkeys('phases')
1569 publishing = bool(remotephases.get('publishing', False))
1569 publishing = bool(remotephases.get('publishing', False))
1570 if remotephases and not publishing:
1570 if remotephases and not publishing:
1571 # remote is new and unpublishing
1571 # remote is new and unpublishing
1572 pheads, _dr = phases.analyzeremotephases(self, subset,
1572 pheads, _dr = phases.analyzeremotephases(self, subset,
1573 remotephases)
1573 remotephases)
1574 phases.advanceboundary(self, phases.public, pheads)
1574 phases.advanceboundary(self, phases.public, pheads)
1575 phases.advanceboundary(self, phases.draft, subset)
1575 phases.advanceboundary(self, phases.draft, subset)
1576 else:
1576 else:
1577 # Remote is old or publishing all common changesets
1577 # Remote is old or publishing all common changesets
1578 # should be seen as public
1578 # should be seen as public
1579 phases.advanceboundary(self, phases.public, subset)
1579 phases.advanceboundary(self, phases.public, subset)
1580 finally:
1580 finally:
1581 lock.release()
1581 lock.release()
1582
1582
1583 return result
1583 return result
1584
1584
1585 def checkpush(self, force, revs):
1585 def checkpush(self, force, revs):
1586 """Extensions can override this function if additional checks have
1586 """Extensions can override this function if additional checks have
1587 to be performed before pushing, or call it if they override push
1587 to be performed before pushing, or call it if they override push
1588 command.
1588 command.
1589 """
1589 """
1590 pass
1590 pass
1591
1591
1592 def push(self, remote, force=False, revs=None, newbranch=False):
1592 def push(self, remote, force=False, revs=None, newbranch=False):
1593 '''Push outgoing changesets (limited by revs) from the current
1593 '''Push outgoing changesets (limited by revs) from the current
1594 repository to remote. Return an integer:
1594 repository to remote. Return an integer:
1595 - 0 means HTTP error *or* nothing to push
1595 - 0 means HTTP error *or* nothing to push
1596 - 1 means we pushed and remote head count is unchanged *or*
1596 - 1 means we pushed and remote head count is unchanged *or*
1597 we have outgoing changesets but refused to push
1597 we have outgoing changesets but refused to push
1598 - other values as described by addchangegroup()
1598 - other values as described by addchangegroup()
1599 '''
1599 '''
1600 # there are two ways to push to remote repo:
1600 # there are two ways to push to remote repo:
1601 #
1601 #
1602 # addchangegroup assumes local user can lock remote
1602 # addchangegroup assumes local user can lock remote
1603 # repo (local filesystem, old ssh servers).
1603 # repo (local filesystem, old ssh servers).
1604 #
1604 #
1605 # unbundle assumes local user cannot lock remote repo (new ssh
1605 # unbundle assumes local user cannot lock remote repo (new ssh
1606 # servers, http servers).
1606 # servers, http servers).
1607
1607
1608 # get local lock as we might write phase data
1608 # get local lock as we might write phase data
1609 locallock = self.lock()
1609 locallock = self.lock()
1610 try:
1610 try:
1611 self.checkpush(force, revs)
1611 self.checkpush(force, revs)
1612 lock = None
1612 lock = None
1613 unbundle = remote.capable('unbundle')
1613 unbundle = remote.capable('unbundle')
1614 if not unbundle:
1614 if not unbundle:
1615 lock = remote.lock()
1615 lock = remote.lock()
1616 try:
1616 try:
1617 # discovery
1617 # discovery
1618 fci = discovery.findcommonincoming
1618 fci = discovery.findcommonincoming
1619 commoninc = fci(self, remote, force=force)
1619 commoninc = fci(self, remote, force=force)
1620 common, inc, remoteheads = commoninc
1620 common, inc, remoteheads = commoninc
1621 fco = discovery.findcommonoutgoing
1621 fco = discovery.findcommonoutgoing
1622 outgoing = fco(self, remote, onlyheads=revs,
1622 outgoing = fco(self, remote, onlyheads=revs,
1623 commoninc=commoninc, force=force)
1623 commoninc=commoninc, force=force)
1624
1624
1625
1625
1626 if not outgoing.missing:
1626 if not outgoing.missing:
1627 # nothing to push
1627 # nothing to push
1628 scmutil.nochangesfound(self.ui, outgoing.excluded)
1628 scmutil.nochangesfound(self.ui, outgoing.excluded)
1629 ret = 1
1629 ret = 1
1630 else:
1630 else:
1631 # something to push
1631 # something to push
1632 if not force:
1632 if not force:
1633 discovery.checkheads(self, remote, outgoing,
1633 discovery.checkheads(self, remote, outgoing,
1634 remoteheads, newbranch,
1634 remoteheads, newbranch,
1635 bool(inc))
1635 bool(inc))
1636
1636
1637 # create a changegroup from local
1637 # create a changegroup from local
1638 if revs is None and not outgoing.excluded:
1638 if revs is None and not outgoing.excluded:
1639 # push everything,
1639 # push everything,
1640 # use the fast path, no race possible on push
1640 # use the fast path, no race possible on push
1641 cg = self._changegroup(outgoing.missing, 'push')
1641 cg = self._changegroup(outgoing.missing, 'push')
1642 else:
1642 else:
1643 cg = self.getlocalbundle('push', outgoing)
1643 cg = self.getlocalbundle('push', outgoing)
1644
1644
1645 # apply changegroup to remote
1645 # apply changegroup to remote
1646 if unbundle:
1646 if unbundle:
1647 # local repo finds heads on server, finds out what
1647 # local repo finds heads on server, finds out what
1648 # revs it must push. once revs transferred, if server
1648 # revs it must push. once revs transferred, if server
1649 # finds it has different heads (someone else won
1649 # finds it has different heads (someone else won
1650 # commit/push race), server aborts.
1650 # commit/push race), server aborts.
1651 if force:
1651 if force:
1652 remoteheads = ['force']
1652 remoteheads = ['force']
1653 # ssh: return remote's addchangegroup()
1653 # ssh: return remote's addchangegroup()
1654 # http: return remote's addchangegroup() or 0 for error
1654 # http: return remote's addchangegroup() or 0 for error
1655 ret = remote.unbundle(cg, remoteheads, 'push')
1655 ret = remote.unbundle(cg, remoteheads, 'push')
1656 else:
1656 else:
1657 # we return an integer indicating remote head count change
1657 # we return an integer indicating remote head count change
1658 ret = remote.addchangegroup(cg, 'push', self.url())
1658 ret = remote.addchangegroup(cg, 'push', self.url())
1659
1659
1660 if ret:
1660 if ret:
1661 # push succeed, synchonize target of the push
1661 # push succeed, synchonize target of the push
1662 cheads = outgoing.missingheads
1662 cheads = outgoing.missingheads
1663 elif revs is None:
1663 elif revs is None:
1664 # All out push fails. synchronize all common
1664 # All out push fails. synchronize all common
1665 cheads = outgoing.commonheads
1665 cheads = outgoing.commonheads
1666 else:
1666 else:
1667 # I want cheads = heads(::missingheads and ::commonheads)
1667 # I want cheads = heads(::missingheads and ::commonheads)
1668 # (missingheads is revs with secret changeset filtered out)
1668 # (missingheads is revs with secret changeset filtered out)
1669 #
1669 #
1670 # This can be expressed as:
1670 # This can be expressed as:
1671 # cheads = ( (missingheads and ::commonheads)
1671 # cheads = ( (missingheads and ::commonheads)
1672 # + (commonheads and ::missingheads))"
1672 # + (commonheads and ::missingheads))"
1673 # )
1673 # )
1674 #
1674 #
1675 # while trying to push we already computed the following:
1675 # while trying to push we already computed the following:
1676 # common = (::commonheads)
1676 # common = (::commonheads)
1677 # missing = ((commonheads::missingheads) - commonheads)
1677 # missing = ((commonheads::missingheads) - commonheads)
1678 #
1678 #
1679 # We can pick:
1679 # We can pick:
1680 # * missingheads part of comon (::commonheads)
1680 # * missingheads part of comon (::commonheads)
1681 common = set(outgoing.common)
1681 common = set(outgoing.common)
1682 cheads = [n for node in revs if n in common]
1682 cheads = [node for node in revs if node in common]
1683 # and
1683 # and
1684 # * commonheads parents on missing
1684 # * commonheads parents on missing
1685 rvset = repo.revset('%ln and parents(roots(%ln))',
1685 revset = self.set('%ln and parents(roots(%ln))',
1686 outgoing.commonheads,
1686 outgoing.commonheads,
1687 outgoing.missing)
1687 outgoing.missing)
1688 cheads.extend(c.node() for c in rvset)
1688 cheads.extend(c.node() for c in revset)
1689 # even when we don't push, exchanging phase data is useful
1689 # even when we don't push, exchanging phase data is useful
1690 remotephases = remote.listkeys('phases')
1690 remotephases = remote.listkeys('phases')
1691 if not remotephases: # old server or public only repo
1691 if not remotephases: # old server or public only repo
1692 phases.advanceboundary(self, phases.public, cheads)
1692 phases.advanceboundary(self, phases.public, cheads)
1693 # don't push any phase data as there is nothing to push
1693 # don't push any phase data as there is nothing to push
1694 else:
1694 else:
1695 ana = phases.analyzeremotephases(self, cheads, remotephases)
1695 ana = phases.analyzeremotephases(self, cheads, remotephases)
1696 pheads, droots = ana
1696 pheads, droots = ana
1697 ### Apply remote phase on local
1697 ### Apply remote phase on local
1698 if remotephases.get('publishing', False):
1698 if remotephases.get('publishing', False):
1699 phases.advanceboundary(self, phases.public, cheads)
1699 phases.advanceboundary(self, phases.public, cheads)
1700 else: # publish = False
1700 else: # publish = False
1701 phases.advanceboundary(self, phases.public, pheads)
1701 phases.advanceboundary(self, phases.public, pheads)
1702 phases.advanceboundary(self, phases.draft, cheads)
1702 phases.advanceboundary(self, phases.draft, cheads)
1703 ### Apply local phase on remote
1703 ### Apply local phase on remote
1704
1704
1705 # Get the list of all revs draft on remote by public here.
1705 # Get the list of all revs draft on remote by public here.
1706 # XXX Beware that revset break if droots is not strictly
1706 # XXX Beware that revset break if droots is not strictly
1707 # XXX root we may want to ensure it is but it is costly
1707 # XXX root we may want to ensure it is but it is costly
1708 outdated = self.set('heads((%ln::%ln) and public())',
1708 outdated = self.set('heads((%ln::%ln) and public())',
1709 droots, cheads)
1709 droots, cheads)
1710 for newremotehead in outdated:
1710 for newremotehead in outdated:
1711 r = remote.pushkey('phases',
1711 r = remote.pushkey('phases',
1712 newremotehead.hex(),
1712 newremotehead.hex(),
1713 str(phases.draft),
1713 str(phases.draft),
1714 str(phases.public))
1714 str(phases.public))
1715 if not r:
1715 if not r:
1716 self.ui.warn(_('updating %s to public failed!\n')
1716 self.ui.warn(_('updating %s to public failed!\n')
1717 % newremotehead)
1717 % newremotehead)
1718 finally:
1718 finally:
1719 if lock is not None:
1719 if lock is not None:
1720 lock.release()
1720 lock.release()
1721 finally:
1721 finally:
1722 locallock.release()
1722 locallock.release()
1723
1723
1724 self.ui.debug("checking for updated bookmarks\n")
1724 self.ui.debug("checking for updated bookmarks\n")
1725 rb = remote.listkeys('bookmarks')
1725 rb = remote.listkeys('bookmarks')
1726 for k in rb.keys():
1726 for k in rb.keys():
1727 if k in self._bookmarks:
1727 if k in self._bookmarks:
1728 nr, nl = rb[k], hex(self._bookmarks[k])
1728 nr, nl = rb[k], hex(self._bookmarks[k])
1729 if nr in self:
1729 if nr in self:
1730 cr = self[nr]
1730 cr = self[nr]
1731 cl = self[nl]
1731 cl = self[nl]
1732 if cl in cr.descendants():
1732 if cl in cr.descendants():
1733 r = remote.pushkey('bookmarks', k, nr, nl)
1733 r = remote.pushkey('bookmarks', k, nr, nl)
1734 if r:
1734 if r:
1735 self.ui.status(_("updating bookmark %s\n") % k)
1735 self.ui.status(_("updating bookmark %s\n") % k)
1736 else:
1736 else:
1737 self.ui.warn(_('updating bookmark %s'
1737 self.ui.warn(_('updating bookmark %s'
1738 ' failed!\n') % k)
1738 ' failed!\n') % k)
1739
1739
1740 return ret
1740 return ret
1741
1741
1742 def changegroupinfo(self, nodes, source):
1742 def changegroupinfo(self, nodes, source):
1743 if self.ui.verbose or source == 'bundle':
1743 if self.ui.verbose or source == 'bundle':
1744 self.ui.status(_("%d changesets found\n") % len(nodes))
1744 self.ui.status(_("%d changesets found\n") % len(nodes))
1745 if self.ui.debugflag:
1745 if self.ui.debugflag:
1746 self.ui.debug("list of changesets:\n")
1746 self.ui.debug("list of changesets:\n")
1747 for node in nodes:
1747 for node in nodes:
1748 self.ui.debug("%s\n" % hex(node))
1748 self.ui.debug("%s\n" % hex(node))
1749
1749
1750 def changegroupsubset(self, bases, heads, source):
1750 def changegroupsubset(self, bases, heads, source):
1751 """Compute a changegroup consisting of all the nodes that are
1751 """Compute a changegroup consisting of all the nodes that are
1752 descendants of any of the bases and ancestors of any of the heads.
1752 descendants of any of the bases and ancestors of any of the heads.
1753 Return a chunkbuffer object whose read() method will return
1753 Return a chunkbuffer object whose read() method will return
1754 successive changegroup chunks.
1754 successive changegroup chunks.
1755
1755
1756 It is fairly complex as determining which filenodes and which
1756 It is fairly complex as determining which filenodes and which
1757 manifest nodes need to be included for the changeset to be complete
1757 manifest nodes need to be included for the changeset to be complete
1758 is non-trivial.
1758 is non-trivial.
1759
1759
1760 Another wrinkle is doing the reverse, figuring out which changeset in
1760 Another wrinkle is doing the reverse, figuring out which changeset in
1761 the changegroup a particular filenode or manifestnode belongs to.
1761 the changegroup a particular filenode or manifestnode belongs to.
1762 """
1762 """
1763 cl = self.changelog
1763 cl = self.changelog
1764 if not bases:
1764 if not bases:
1765 bases = [nullid]
1765 bases = [nullid]
1766 csets, bases, heads = cl.nodesbetween(bases, heads)
1766 csets, bases, heads = cl.nodesbetween(bases, heads)
1767 # We assume that all ancestors of bases are known
1767 # We assume that all ancestors of bases are known
1768 common = set(cl.ancestors(*[cl.rev(n) for n in bases]))
1768 common = set(cl.ancestors(*[cl.rev(n) for n in bases]))
1769 return self._changegroupsubset(common, csets, heads, source)
1769 return self._changegroupsubset(common, csets, heads, source)
1770
1770
1771 def getlocalbundle(self, source, outgoing):
1771 def getlocalbundle(self, source, outgoing):
1772 """Like getbundle, but taking a discovery.outgoing as an argument.
1772 """Like getbundle, but taking a discovery.outgoing as an argument.
1773
1773
1774 This is only implemented for local repos and reuses potentially
1774 This is only implemented for local repos and reuses potentially
1775 precomputed sets in outgoing."""
1775 precomputed sets in outgoing."""
1776 if not outgoing.missing:
1776 if not outgoing.missing:
1777 return None
1777 return None
1778 return self._changegroupsubset(outgoing.common,
1778 return self._changegroupsubset(outgoing.common,
1779 outgoing.missing,
1779 outgoing.missing,
1780 outgoing.missingheads,
1780 outgoing.missingheads,
1781 source)
1781 source)
1782
1782
1783 def getbundle(self, source, heads=None, common=None):
1783 def getbundle(self, source, heads=None, common=None):
1784 """Like changegroupsubset, but returns the set difference between the
1784 """Like changegroupsubset, but returns the set difference between the
1785 ancestors of heads and the ancestors common.
1785 ancestors of heads and the ancestors common.
1786
1786
1787 If heads is None, use the local heads. If common is None, use [nullid].
1787 If heads is None, use the local heads. If common is None, use [nullid].
1788
1788
1789 The nodes in common might not all be known locally due to the way the
1789 The nodes in common might not all be known locally due to the way the
1790 current discovery protocol works.
1790 current discovery protocol works.
1791 """
1791 """
1792 cl = self.changelog
1792 cl = self.changelog
1793 if common:
1793 if common:
1794 nm = cl.nodemap
1794 nm = cl.nodemap
1795 common = [n for n in common if n in nm]
1795 common = [n for n in common if n in nm]
1796 else:
1796 else:
1797 common = [nullid]
1797 common = [nullid]
1798 if not heads:
1798 if not heads:
1799 heads = cl.heads()
1799 heads = cl.heads()
1800 return self.getlocalbundle(source,
1800 return self.getlocalbundle(source,
1801 discovery.outgoing(cl, common, heads))
1801 discovery.outgoing(cl, common, heads))
1802
1802
1803 def _changegroupsubset(self, commonrevs, csets, heads, source):
1803 def _changegroupsubset(self, commonrevs, csets, heads, source):
1804
1804
1805 cl = self.changelog
1805 cl = self.changelog
1806 mf = self.manifest
1806 mf = self.manifest
1807 mfs = {} # needed manifests
1807 mfs = {} # needed manifests
1808 fnodes = {} # needed file nodes
1808 fnodes = {} # needed file nodes
1809 changedfiles = set()
1809 changedfiles = set()
1810 fstate = ['', {}]
1810 fstate = ['', {}]
1811 count = [0]
1811 count = [0]
1812
1812
1813 # can we go through the fast path ?
1813 # can we go through the fast path ?
1814 heads.sort()
1814 heads.sort()
1815 if heads == sorted(self.heads()):
1815 if heads == sorted(self.heads()):
1816 return self._changegroup(csets, source)
1816 return self._changegroup(csets, source)
1817
1817
1818 # slow path
1818 # slow path
1819 self.hook('preoutgoing', throw=True, source=source)
1819 self.hook('preoutgoing', throw=True, source=source)
1820 self.changegroupinfo(csets, source)
1820 self.changegroupinfo(csets, source)
1821
1821
1822 # filter any nodes that claim to be part of the known set
1822 # filter any nodes that claim to be part of the known set
1823 def prune(revlog, missing):
1823 def prune(revlog, missing):
1824 return [n for n in missing
1824 return [n for n in missing
1825 if revlog.linkrev(revlog.rev(n)) not in commonrevs]
1825 if revlog.linkrev(revlog.rev(n)) not in commonrevs]
1826
1826
1827 def lookup(revlog, x):
1827 def lookup(revlog, x):
1828 if revlog == cl:
1828 if revlog == cl:
1829 c = cl.read(x)
1829 c = cl.read(x)
1830 changedfiles.update(c[3])
1830 changedfiles.update(c[3])
1831 mfs.setdefault(c[0], x)
1831 mfs.setdefault(c[0], x)
1832 count[0] += 1
1832 count[0] += 1
1833 self.ui.progress(_('bundling'), count[0],
1833 self.ui.progress(_('bundling'), count[0],
1834 unit=_('changesets'), total=len(csets))
1834 unit=_('changesets'), total=len(csets))
1835 return x
1835 return x
1836 elif revlog == mf:
1836 elif revlog == mf:
1837 clnode = mfs[x]
1837 clnode = mfs[x]
1838 mdata = mf.readfast(x)
1838 mdata = mf.readfast(x)
1839 for f in changedfiles:
1839 for f in changedfiles:
1840 if f in mdata:
1840 if f in mdata:
1841 fnodes.setdefault(f, {}).setdefault(mdata[f], clnode)
1841 fnodes.setdefault(f, {}).setdefault(mdata[f], clnode)
1842 count[0] += 1
1842 count[0] += 1
1843 self.ui.progress(_('bundling'), count[0],
1843 self.ui.progress(_('bundling'), count[0],
1844 unit=_('manifests'), total=len(mfs))
1844 unit=_('manifests'), total=len(mfs))
1845 return mfs[x]
1845 return mfs[x]
1846 else:
1846 else:
1847 self.ui.progress(
1847 self.ui.progress(
1848 _('bundling'), count[0], item=fstate[0],
1848 _('bundling'), count[0], item=fstate[0],
1849 unit=_('files'), total=len(changedfiles))
1849 unit=_('files'), total=len(changedfiles))
1850 return fstate[1][x]
1850 return fstate[1][x]
1851
1851
1852 bundler = changegroup.bundle10(lookup)
1852 bundler = changegroup.bundle10(lookup)
1853 reorder = self.ui.config('bundle', 'reorder', 'auto')
1853 reorder = self.ui.config('bundle', 'reorder', 'auto')
1854 if reorder == 'auto':
1854 if reorder == 'auto':
1855 reorder = None
1855 reorder = None
1856 else:
1856 else:
1857 reorder = util.parsebool(reorder)
1857 reorder = util.parsebool(reorder)
1858
1858
1859 def gengroup():
1859 def gengroup():
1860 # Create a changenode group generator that will call our functions
1860 # Create a changenode group generator that will call our functions
1861 # back to lookup the owning changenode and collect information.
1861 # back to lookup the owning changenode and collect information.
1862 for chunk in cl.group(csets, bundler, reorder=reorder):
1862 for chunk in cl.group(csets, bundler, reorder=reorder):
1863 yield chunk
1863 yield chunk
1864 self.ui.progress(_('bundling'), None)
1864 self.ui.progress(_('bundling'), None)
1865
1865
1866 # Create a generator for the manifestnodes that calls our lookup
1866 # Create a generator for the manifestnodes that calls our lookup
1867 # and data collection functions back.
1867 # and data collection functions back.
1868 count[0] = 0
1868 count[0] = 0
1869 for chunk in mf.group(prune(mf, mfs), bundler, reorder=reorder):
1869 for chunk in mf.group(prune(mf, mfs), bundler, reorder=reorder):
1870 yield chunk
1870 yield chunk
1871 self.ui.progress(_('bundling'), None)
1871 self.ui.progress(_('bundling'), None)
1872
1872
1873 mfs.clear()
1873 mfs.clear()
1874
1874
1875 # Go through all our files in order sorted by name.
1875 # Go through all our files in order sorted by name.
1876 count[0] = 0
1876 count[0] = 0
1877 for fname in sorted(changedfiles):
1877 for fname in sorted(changedfiles):
1878 filerevlog = self.file(fname)
1878 filerevlog = self.file(fname)
1879 if not len(filerevlog):
1879 if not len(filerevlog):
1880 raise util.Abort(_("empty or missing revlog for %s") % fname)
1880 raise util.Abort(_("empty or missing revlog for %s") % fname)
1881 fstate[0] = fname
1881 fstate[0] = fname
1882 fstate[1] = fnodes.pop(fname, {})
1882 fstate[1] = fnodes.pop(fname, {})
1883
1883
1884 nodelist = prune(filerevlog, fstate[1])
1884 nodelist = prune(filerevlog, fstate[1])
1885 if nodelist:
1885 if nodelist:
1886 count[0] += 1
1886 count[0] += 1
1887 yield bundler.fileheader(fname)
1887 yield bundler.fileheader(fname)
1888 for chunk in filerevlog.group(nodelist, bundler, reorder):
1888 for chunk in filerevlog.group(nodelist, bundler, reorder):
1889 yield chunk
1889 yield chunk
1890
1890
1891 # Signal that no more groups are left.
1891 # Signal that no more groups are left.
1892 yield bundler.close()
1892 yield bundler.close()
1893 self.ui.progress(_('bundling'), None)
1893 self.ui.progress(_('bundling'), None)
1894
1894
1895 if csets:
1895 if csets:
1896 self.hook('outgoing', node=hex(csets[0]), source=source)
1896 self.hook('outgoing', node=hex(csets[0]), source=source)
1897
1897
1898 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1898 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1899
1899
1900 def changegroup(self, basenodes, source):
1900 def changegroup(self, basenodes, source):
1901 # to avoid a race we use changegroupsubset() (issue1320)
1901 # to avoid a race we use changegroupsubset() (issue1320)
1902 return self.changegroupsubset(basenodes, self.heads(), source)
1902 return self.changegroupsubset(basenodes, self.heads(), source)
1903
1903
1904 def _changegroup(self, nodes, source):
1904 def _changegroup(self, nodes, source):
1905 """Compute the changegroup of all nodes that we have that a recipient
1905 """Compute the changegroup of all nodes that we have that a recipient
1906 doesn't. Return a chunkbuffer object whose read() method will return
1906 doesn't. Return a chunkbuffer object whose read() method will return
1907 successive changegroup chunks.
1907 successive changegroup chunks.
1908
1908
1909 This is much easier than the previous function as we can assume that
1909 This is much easier than the previous function as we can assume that
1910 the recipient has any changenode we aren't sending them.
1910 the recipient has any changenode we aren't sending them.
1911
1911
1912 nodes is the set of nodes to send"""
1912 nodes is the set of nodes to send"""
1913
1913
1914 cl = self.changelog
1914 cl = self.changelog
1915 mf = self.manifest
1915 mf = self.manifest
1916 mfs = {}
1916 mfs = {}
1917 changedfiles = set()
1917 changedfiles = set()
1918 fstate = ['']
1918 fstate = ['']
1919 count = [0]
1919 count = [0]
1920
1920
1921 self.hook('preoutgoing', throw=True, source=source)
1921 self.hook('preoutgoing', throw=True, source=source)
1922 self.changegroupinfo(nodes, source)
1922 self.changegroupinfo(nodes, source)
1923
1923
1924 revset = set([cl.rev(n) for n in nodes])
1924 revset = set([cl.rev(n) for n in nodes])
1925
1925
1926 def gennodelst(log):
1926 def gennodelst(log):
1927 return [log.node(r) for r in log if log.linkrev(r) in revset]
1927 return [log.node(r) for r in log if log.linkrev(r) in revset]
1928
1928
1929 def lookup(revlog, x):
1929 def lookup(revlog, x):
1930 if revlog == cl:
1930 if revlog == cl:
1931 c = cl.read(x)
1931 c = cl.read(x)
1932 changedfiles.update(c[3])
1932 changedfiles.update(c[3])
1933 mfs.setdefault(c[0], x)
1933 mfs.setdefault(c[0], x)
1934 count[0] += 1
1934 count[0] += 1
1935 self.ui.progress(_('bundling'), count[0],
1935 self.ui.progress(_('bundling'), count[0],
1936 unit=_('changesets'), total=len(nodes))
1936 unit=_('changesets'), total=len(nodes))
1937 return x
1937 return x
1938 elif revlog == mf:
1938 elif revlog == mf:
1939 count[0] += 1
1939 count[0] += 1
1940 self.ui.progress(_('bundling'), count[0],
1940 self.ui.progress(_('bundling'), count[0],
1941 unit=_('manifests'), total=len(mfs))
1941 unit=_('manifests'), total=len(mfs))
1942 return cl.node(revlog.linkrev(revlog.rev(x)))
1942 return cl.node(revlog.linkrev(revlog.rev(x)))
1943 else:
1943 else:
1944 self.ui.progress(
1944 self.ui.progress(
1945 _('bundling'), count[0], item=fstate[0],
1945 _('bundling'), count[0], item=fstate[0],
1946 total=len(changedfiles), unit=_('files'))
1946 total=len(changedfiles), unit=_('files'))
1947 return cl.node(revlog.linkrev(revlog.rev(x)))
1947 return cl.node(revlog.linkrev(revlog.rev(x)))
1948
1948
1949 bundler = changegroup.bundle10(lookup)
1949 bundler = changegroup.bundle10(lookup)
1950 reorder = self.ui.config('bundle', 'reorder', 'auto')
1950 reorder = self.ui.config('bundle', 'reorder', 'auto')
1951 if reorder == 'auto':
1951 if reorder == 'auto':
1952 reorder = None
1952 reorder = None
1953 else:
1953 else:
1954 reorder = util.parsebool(reorder)
1954 reorder = util.parsebool(reorder)
1955
1955
1956 def gengroup():
1956 def gengroup():
1957 '''yield a sequence of changegroup chunks (strings)'''
1957 '''yield a sequence of changegroup chunks (strings)'''
1958 # construct a list of all changed files
1958 # construct a list of all changed files
1959
1959
1960 for chunk in cl.group(nodes, bundler, reorder=reorder):
1960 for chunk in cl.group(nodes, bundler, reorder=reorder):
1961 yield chunk
1961 yield chunk
1962 self.ui.progress(_('bundling'), None)
1962 self.ui.progress(_('bundling'), None)
1963
1963
1964 count[0] = 0
1964 count[0] = 0
1965 for chunk in mf.group(gennodelst(mf), bundler, reorder=reorder):
1965 for chunk in mf.group(gennodelst(mf), bundler, reorder=reorder):
1966 yield chunk
1966 yield chunk
1967 self.ui.progress(_('bundling'), None)
1967 self.ui.progress(_('bundling'), None)
1968
1968
1969 count[0] = 0
1969 count[0] = 0
1970 for fname in sorted(changedfiles):
1970 for fname in sorted(changedfiles):
1971 filerevlog = self.file(fname)
1971 filerevlog = self.file(fname)
1972 if not len(filerevlog):
1972 if not len(filerevlog):
1973 raise util.Abort(_("empty or missing revlog for %s") % fname)
1973 raise util.Abort(_("empty or missing revlog for %s") % fname)
1974 fstate[0] = fname
1974 fstate[0] = fname
1975 nodelist = gennodelst(filerevlog)
1975 nodelist = gennodelst(filerevlog)
1976 if nodelist:
1976 if nodelist:
1977 count[0] += 1
1977 count[0] += 1
1978 yield bundler.fileheader(fname)
1978 yield bundler.fileheader(fname)
1979 for chunk in filerevlog.group(nodelist, bundler, reorder):
1979 for chunk in filerevlog.group(nodelist, bundler, reorder):
1980 yield chunk
1980 yield chunk
1981 yield bundler.close()
1981 yield bundler.close()
1982 self.ui.progress(_('bundling'), None)
1982 self.ui.progress(_('bundling'), None)
1983
1983
1984 if nodes:
1984 if nodes:
1985 self.hook('outgoing', node=hex(nodes[0]), source=source)
1985 self.hook('outgoing', node=hex(nodes[0]), source=source)
1986
1986
1987 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1987 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1988
1988
1989 def addchangegroup(self, source, srctype, url, emptyok=False):
1989 def addchangegroup(self, source, srctype, url, emptyok=False):
1990 """Add the changegroup returned by source.read() to this repo.
1990 """Add the changegroup returned by source.read() to this repo.
1991 srctype is a string like 'push', 'pull', or 'unbundle'. url is
1991 srctype is a string like 'push', 'pull', or 'unbundle'. url is
1992 the URL of the repo where this changegroup is coming from.
1992 the URL of the repo where this changegroup is coming from.
1993
1993
1994 Return an integer summarizing the change to this repo:
1994 Return an integer summarizing the change to this repo:
1995 - nothing changed or no source: 0
1995 - nothing changed or no source: 0
1996 - more heads than before: 1+added heads (2..n)
1996 - more heads than before: 1+added heads (2..n)
1997 - fewer heads than before: -1-removed heads (-2..-n)
1997 - fewer heads than before: -1-removed heads (-2..-n)
1998 - number of heads stays the same: 1
1998 - number of heads stays the same: 1
1999 """
1999 """
2000 def csmap(x):
2000 def csmap(x):
2001 self.ui.debug("add changeset %s\n" % short(x))
2001 self.ui.debug("add changeset %s\n" % short(x))
2002 return len(cl)
2002 return len(cl)
2003
2003
2004 def revmap(x):
2004 def revmap(x):
2005 return cl.rev(x)
2005 return cl.rev(x)
2006
2006
2007 if not source:
2007 if not source:
2008 return 0
2008 return 0
2009
2009
2010 self.hook('prechangegroup', throw=True, source=srctype, url=url)
2010 self.hook('prechangegroup', throw=True, source=srctype, url=url)
2011
2011
2012 changesets = files = revisions = 0
2012 changesets = files = revisions = 0
2013 efiles = set()
2013 efiles = set()
2014
2014
2015 # write changelog data to temp files so concurrent readers will not see
2015 # write changelog data to temp files so concurrent readers will not see
2016 # inconsistent view
2016 # inconsistent view
2017 cl = self.changelog
2017 cl = self.changelog
2018 cl.delayupdate()
2018 cl.delayupdate()
2019 oldheads = cl.heads()
2019 oldheads = cl.heads()
2020
2020
2021 tr = self.transaction("\n".join([srctype, util.hidepassword(url)]))
2021 tr = self.transaction("\n".join([srctype, util.hidepassword(url)]))
2022 try:
2022 try:
2023 trp = weakref.proxy(tr)
2023 trp = weakref.proxy(tr)
2024 # pull off the changeset group
2024 # pull off the changeset group
2025 self.ui.status(_("adding changesets\n"))
2025 self.ui.status(_("adding changesets\n"))
2026 clstart = len(cl)
2026 clstart = len(cl)
2027 class prog(object):
2027 class prog(object):
2028 step = _('changesets')
2028 step = _('changesets')
2029 count = 1
2029 count = 1
2030 ui = self.ui
2030 ui = self.ui
2031 total = None
2031 total = None
2032 def __call__(self):
2032 def __call__(self):
2033 self.ui.progress(self.step, self.count, unit=_('chunks'),
2033 self.ui.progress(self.step, self.count, unit=_('chunks'),
2034 total=self.total)
2034 total=self.total)
2035 self.count += 1
2035 self.count += 1
2036 pr = prog()
2036 pr = prog()
2037 source.callback = pr
2037 source.callback = pr
2038
2038
2039 source.changelogheader()
2039 source.changelogheader()
2040 srccontent = cl.addgroup(source, csmap, trp)
2040 srccontent = cl.addgroup(source, csmap, trp)
2041 if not (srccontent or emptyok):
2041 if not (srccontent or emptyok):
2042 raise util.Abort(_("received changelog group is empty"))
2042 raise util.Abort(_("received changelog group is empty"))
2043 clend = len(cl)
2043 clend = len(cl)
2044 changesets = clend - clstart
2044 changesets = clend - clstart
2045 for c in xrange(clstart, clend):
2045 for c in xrange(clstart, clend):
2046 efiles.update(self[c].files())
2046 efiles.update(self[c].files())
2047 efiles = len(efiles)
2047 efiles = len(efiles)
2048 self.ui.progress(_('changesets'), None)
2048 self.ui.progress(_('changesets'), None)
2049
2049
2050 # pull off the manifest group
2050 # pull off the manifest group
2051 self.ui.status(_("adding manifests\n"))
2051 self.ui.status(_("adding manifests\n"))
2052 pr.step = _('manifests')
2052 pr.step = _('manifests')
2053 pr.count = 1
2053 pr.count = 1
2054 pr.total = changesets # manifests <= changesets
2054 pr.total = changesets # manifests <= changesets
2055 # no need to check for empty manifest group here:
2055 # no need to check for empty manifest group here:
2056 # if the result of the merge of 1 and 2 is the same in 3 and 4,
2056 # if the result of the merge of 1 and 2 is the same in 3 and 4,
2057 # no new manifest will be created and the manifest group will
2057 # no new manifest will be created and the manifest group will
2058 # be empty during the pull
2058 # be empty during the pull
2059 source.manifestheader()
2059 source.manifestheader()
2060 self.manifest.addgroup(source, revmap, trp)
2060 self.manifest.addgroup(source, revmap, trp)
2061 self.ui.progress(_('manifests'), None)
2061 self.ui.progress(_('manifests'), None)
2062
2062
2063 needfiles = {}
2063 needfiles = {}
2064 if self.ui.configbool('server', 'validate', default=False):
2064 if self.ui.configbool('server', 'validate', default=False):
2065 # validate incoming csets have their manifests
2065 # validate incoming csets have their manifests
2066 for cset in xrange(clstart, clend):
2066 for cset in xrange(clstart, clend):
2067 mfest = self.changelog.read(self.changelog.node(cset))[0]
2067 mfest = self.changelog.read(self.changelog.node(cset))[0]
2068 mfest = self.manifest.readdelta(mfest)
2068 mfest = self.manifest.readdelta(mfest)
2069 # store file nodes we must see
2069 # store file nodes we must see
2070 for f, n in mfest.iteritems():
2070 for f, n in mfest.iteritems():
2071 needfiles.setdefault(f, set()).add(n)
2071 needfiles.setdefault(f, set()).add(n)
2072
2072
2073 # process the files
2073 # process the files
2074 self.ui.status(_("adding file changes\n"))
2074 self.ui.status(_("adding file changes\n"))
2075 pr.step = _('files')
2075 pr.step = _('files')
2076 pr.count = 1
2076 pr.count = 1
2077 pr.total = efiles
2077 pr.total = efiles
2078 source.callback = None
2078 source.callback = None
2079
2079
2080 while True:
2080 while True:
2081 chunkdata = source.filelogheader()
2081 chunkdata = source.filelogheader()
2082 if not chunkdata:
2082 if not chunkdata:
2083 break
2083 break
2084 f = chunkdata["filename"]
2084 f = chunkdata["filename"]
2085 self.ui.debug("adding %s revisions\n" % f)
2085 self.ui.debug("adding %s revisions\n" % f)
2086 pr()
2086 pr()
2087 fl = self.file(f)
2087 fl = self.file(f)
2088 o = len(fl)
2088 o = len(fl)
2089 if not fl.addgroup(source, revmap, trp):
2089 if not fl.addgroup(source, revmap, trp):
2090 raise util.Abort(_("received file revlog group is empty"))
2090 raise util.Abort(_("received file revlog group is empty"))
2091 revisions += len(fl) - o
2091 revisions += len(fl) - o
2092 files += 1
2092 files += 1
2093 if f in needfiles:
2093 if f in needfiles:
2094 needs = needfiles[f]
2094 needs = needfiles[f]
2095 for new in xrange(o, len(fl)):
2095 for new in xrange(o, len(fl)):
2096 n = fl.node(new)
2096 n = fl.node(new)
2097 if n in needs:
2097 if n in needs:
2098 needs.remove(n)
2098 needs.remove(n)
2099 if not needs:
2099 if not needs:
2100 del needfiles[f]
2100 del needfiles[f]
2101 self.ui.progress(_('files'), None)
2101 self.ui.progress(_('files'), None)
2102
2102
2103 for f, needs in needfiles.iteritems():
2103 for f, needs in needfiles.iteritems():
2104 fl = self.file(f)
2104 fl = self.file(f)
2105 for n in needs:
2105 for n in needs:
2106 try:
2106 try:
2107 fl.rev(n)
2107 fl.rev(n)
2108 except error.LookupError:
2108 except error.LookupError:
2109 raise util.Abort(
2109 raise util.Abort(
2110 _('missing file data for %s:%s - run hg verify') %
2110 _('missing file data for %s:%s - run hg verify') %
2111 (f, hex(n)))
2111 (f, hex(n)))
2112
2112
2113 dh = 0
2113 dh = 0
2114 if oldheads:
2114 if oldheads:
2115 heads = cl.heads()
2115 heads = cl.heads()
2116 dh = len(heads) - len(oldheads)
2116 dh = len(heads) - len(oldheads)
2117 for h in heads:
2117 for h in heads:
2118 if h not in oldheads and 'close' in self[h].extra():
2118 if h not in oldheads and 'close' in self[h].extra():
2119 dh -= 1
2119 dh -= 1
2120 htext = ""
2120 htext = ""
2121 if dh:
2121 if dh:
2122 htext = _(" (%+d heads)") % dh
2122 htext = _(" (%+d heads)") % dh
2123
2123
2124 self.ui.status(_("added %d changesets"
2124 self.ui.status(_("added %d changesets"
2125 " with %d changes to %d files%s\n")
2125 " with %d changes to %d files%s\n")
2126 % (changesets, revisions, files, htext))
2126 % (changesets, revisions, files, htext))
2127
2127
2128 if changesets > 0:
2128 if changesets > 0:
2129 p = lambda: cl.writepending() and self.root or ""
2129 p = lambda: cl.writepending() and self.root or ""
2130 self.hook('pretxnchangegroup', throw=True,
2130 self.hook('pretxnchangegroup', throw=True,
2131 node=hex(cl.node(clstart)), source=srctype,
2131 node=hex(cl.node(clstart)), source=srctype,
2132 url=url, pending=p)
2132 url=url, pending=p)
2133
2133
2134 added = [cl.node(r) for r in xrange(clstart, clend)]
2134 added = [cl.node(r) for r in xrange(clstart, clend)]
2135 publishing = self.ui.configbool('phases', 'publish', True)
2135 publishing = self.ui.configbool('phases', 'publish', True)
2136 if srctype == 'push':
2136 if srctype == 'push':
2137 # Old server can not push the boundary themself.
2137 # Old server can not push the boundary themself.
2138 # New server won't push the boundary if changeset already
2138 # New server won't push the boundary if changeset already
2139 # existed locally as secrete
2139 # existed locally as secrete
2140 #
2140 #
2141 # We should not use added here but the list of all change in
2141 # We should not use added here but the list of all change in
2142 # the bundle
2142 # the bundle
2143 if publishing:
2143 if publishing:
2144 phases.advanceboundary(self, phases.public, srccontent)
2144 phases.advanceboundary(self, phases.public, srccontent)
2145 else:
2145 else:
2146 phases.advanceboundary(self, phases.draft, srccontent)
2146 phases.advanceboundary(self, phases.draft, srccontent)
2147 phases.retractboundary(self, phases.draft, added)
2147 phases.retractboundary(self, phases.draft, added)
2148 elif srctype != 'strip':
2148 elif srctype != 'strip':
2149 # publishing only alter behavior during push
2149 # publishing only alter behavior during push
2150 #
2150 #
2151 # strip should not touch boundary at all
2151 # strip should not touch boundary at all
2152 phases.retractboundary(self, phases.draft, added)
2152 phases.retractboundary(self, phases.draft, added)
2153
2153
2154 # make changelog see real files again
2154 # make changelog see real files again
2155 cl.finalize(trp)
2155 cl.finalize(trp)
2156
2156
2157 tr.close()
2157 tr.close()
2158
2158
2159 if changesets > 0:
2159 if changesets > 0:
2160 def runhooks():
2160 def runhooks():
2161 # forcefully update the on-disk branch cache
2161 # forcefully update the on-disk branch cache
2162 self.ui.debug("updating the branch cache\n")
2162 self.ui.debug("updating the branch cache\n")
2163 self.updatebranchcache()
2163 self.updatebranchcache()
2164 self.hook("changegroup", node=hex(cl.node(clstart)),
2164 self.hook("changegroup", node=hex(cl.node(clstart)),
2165 source=srctype, url=url)
2165 source=srctype, url=url)
2166
2166
2167 for n in added:
2167 for n in added:
2168 self.hook("incoming", node=hex(n), source=srctype,
2168 self.hook("incoming", node=hex(n), source=srctype,
2169 url=url)
2169 url=url)
2170 self._afterlock(runhooks)
2170 self._afterlock(runhooks)
2171
2171
2172 finally:
2172 finally:
2173 tr.release()
2173 tr.release()
2174 # never return 0 here:
2174 # never return 0 here:
2175 if dh < 0:
2175 if dh < 0:
2176 return dh - 1
2176 return dh - 1
2177 else:
2177 else:
2178 return dh + 1
2178 return dh + 1
2179
2179
2180 def stream_in(self, remote, requirements):
2180 def stream_in(self, remote, requirements):
2181 lock = self.lock()
2181 lock = self.lock()
2182 try:
2182 try:
2183 fp = remote.stream_out()
2183 fp = remote.stream_out()
2184 l = fp.readline()
2184 l = fp.readline()
2185 try:
2185 try:
2186 resp = int(l)
2186 resp = int(l)
2187 except ValueError:
2187 except ValueError:
2188 raise error.ResponseError(
2188 raise error.ResponseError(
2189 _('Unexpected response from remote server:'), l)
2189 _('Unexpected response from remote server:'), l)
2190 if resp == 1:
2190 if resp == 1:
2191 raise util.Abort(_('operation forbidden by server'))
2191 raise util.Abort(_('operation forbidden by server'))
2192 elif resp == 2:
2192 elif resp == 2:
2193 raise util.Abort(_('locking the remote repository failed'))
2193 raise util.Abort(_('locking the remote repository failed'))
2194 elif resp != 0:
2194 elif resp != 0:
2195 raise util.Abort(_('the server sent an unknown error code'))
2195 raise util.Abort(_('the server sent an unknown error code'))
2196 self.ui.status(_('streaming all changes\n'))
2196 self.ui.status(_('streaming all changes\n'))
2197 l = fp.readline()
2197 l = fp.readline()
2198 try:
2198 try:
2199 total_files, total_bytes = map(int, l.split(' ', 1))
2199 total_files, total_bytes = map(int, l.split(' ', 1))
2200 except (ValueError, TypeError):
2200 except (ValueError, TypeError):
2201 raise error.ResponseError(
2201 raise error.ResponseError(
2202 _('Unexpected response from remote server:'), l)
2202 _('Unexpected response from remote server:'), l)
2203 self.ui.status(_('%d files to transfer, %s of data\n') %
2203 self.ui.status(_('%d files to transfer, %s of data\n') %
2204 (total_files, util.bytecount(total_bytes)))
2204 (total_files, util.bytecount(total_bytes)))
2205 start = time.time()
2205 start = time.time()
2206 for i in xrange(total_files):
2206 for i in xrange(total_files):
2207 # XXX doesn't support '\n' or '\r' in filenames
2207 # XXX doesn't support '\n' or '\r' in filenames
2208 l = fp.readline()
2208 l = fp.readline()
2209 try:
2209 try:
2210 name, size = l.split('\0', 1)
2210 name, size = l.split('\0', 1)
2211 size = int(size)
2211 size = int(size)
2212 except (ValueError, TypeError):
2212 except (ValueError, TypeError):
2213 raise error.ResponseError(
2213 raise error.ResponseError(
2214 _('Unexpected response from remote server:'), l)
2214 _('Unexpected response from remote server:'), l)
2215 self.ui.debug('adding %s (%s)\n' % (name, util.bytecount(size)))
2215 self.ui.debug('adding %s (%s)\n' % (name, util.bytecount(size)))
2216 # for backwards compat, name was partially encoded
2216 # for backwards compat, name was partially encoded
2217 ofp = self.sopener(store.decodedir(name), 'w')
2217 ofp = self.sopener(store.decodedir(name), 'w')
2218 for chunk in util.filechunkiter(fp, limit=size):
2218 for chunk in util.filechunkiter(fp, limit=size):
2219 ofp.write(chunk)
2219 ofp.write(chunk)
2220 ofp.close()
2220 ofp.close()
2221 elapsed = time.time() - start
2221 elapsed = time.time() - start
2222 if elapsed <= 0:
2222 if elapsed <= 0:
2223 elapsed = 0.001
2223 elapsed = 0.001
2224 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
2224 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
2225 (util.bytecount(total_bytes), elapsed,
2225 (util.bytecount(total_bytes), elapsed,
2226 util.bytecount(total_bytes / elapsed)))
2226 util.bytecount(total_bytes / elapsed)))
2227
2227
2228 # new requirements = old non-format requirements + new format-related
2228 # new requirements = old non-format requirements + new format-related
2229 # requirements from the streamed-in repository
2229 # requirements from the streamed-in repository
2230 requirements.update(set(self.requirements) - self.supportedformats)
2230 requirements.update(set(self.requirements) - self.supportedformats)
2231 self._applyrequirements(requirements)
2231 self._applyrequirements(requirements)
2232 self._writerequirements()
2232 self._writerequirements()
2233
2233
2234 self.invalidate()
2234 self.invalidate()
2235 return len(self.heads()) + 1
2235 return len(self.heads()) + 1
2236 finally:
2236 finally:
2237 lock.release()
2237 lock.release()
2238
2238
2239 def clone(self, remote, heads=[], stream=False):
2239 def clone(self, remote, heads=[], stream=False):
2240 '''clone remote repository.
2240 '''clone remote repository.
2241
2241
2242 keyword arguments:
2242 keyword arguments:
2243 heads: list of revs to clone (forces use of pull)
2243 heads: list of revs to clone (forces use of pull)
2244 stream: use streaming clone if possible'''
2244 stream: use streaming clone if possible'''
2245
2245
2246 # now, all clients that can request uncompressed clones can
2246 # now, all clients that can request uncompressed clones can
2247 # read repo formats supported by all servers that can serve
2247 # read repo formats supported by all servers that can serve
2248 # them.
2248 # them.
2249
2249
2250 # if revlog format changes, client will have to check version
2250 # if revlog format changes, client will have to check version
2251 # and format flags on "stream" capability, and use
2251 # and format flags on "stream" capability, and use
2252 # uncompressed only if compatible.
2252 # uncompressed only if compatible.
2253
2253
2254 if stream and not heads:
2254 if stream and not heads:
2255 # 'stream' means remote revlog format is revlogv1 only
2255 # 'stream' means remote revlog format is revlogv1 only
2256 if remote.capable('stream'):
2256 if remote.capable('stream'):
2257 return self.stream_in(remote, set(('revlogv1',)))
2257 return self.stream_in(remote, set(('revlogv1',)))
2258 # otherwise, 'streamreqs' contains the remote revlog format
2258 # otherwise, 'streamreqs' contains the remote revlog format
2259 streamreqs = remote.capable('streamreqs')
2259 streamreqs = remote.capable('streamreqs')
2260 if streamreqs:
2260 if streamreqs:
2261 streamreqs = set(streamreqs.split(','))
2261 streamreqs = set(streamreqs.split(','))
2262 # if we support it, stream in and adjust our requirements
2262 # if we support it, stream in and adjust our requirements
2263 if not streamreqs - self.supportedformats:
2263 if not streamreqs - self.supportedformats:
2264 return self.stream_in(remote, streamreqs)
2264 return self.stream_in(remote, streamreqs)
2265 return self.pull(remote, heads)
2265 return self.pull(remote, heads)
2266
2266
2267 def pushkey(self, namespace, key, old, new):
2267 def pushkey(self, namespace, key, old, new):
2268 self.hook('prepushkey', throw=True, namespace=namespace, key=key,
2268 self.hook('prepushkey', throw=True, namespace=namespace, key=key,
2269 old=old, new=new)
2269 old=old, new=new)
2270 ret = pushkey.push(self, namespace, key, old, new)
2270 ret = pushkey.push(self, namespace, key, old, new)
2271 self.hook('pushkey', namespace=namespace, key=key, old=old, new=new,
2271 self.hook('pushkey', namespace=namespace, key=key, old=old, new=new,
2272 ret=ret)
2272 ret=ret)
2273 return ret
2273 return ret
2274
2274
2275 def listkeys(self, namespace):
2275 def listkeys(self, namespace):
2276 self.hook('prelistkeys', throw=True, namespace=namespace)
2276 self.hook('prelistkeys', throw=True, namespace=namespace)
2277 values = pushkey.list(self, namespace)
2277 values = pushkey.list(self, namespace)
2278 self.hook('listkeys', namespace=namespace, values=values)
2278 self.hook('listkeys', namespace=namespace, values=values)
2279 return values
2279 return values
2280
2280
2281 def debugwireargs(self, one, two, three=None, four=None, five=None):
2281 def debugwireargs(self, one, two, three=None, four=None, five=None):
2282 '''used to test argument passing over the wire'''
2282 '''used to test argument passing over the wire'''
2283 return "%s %s %s %s %s" % (one, two, three, four, five)
2283 return "%s %s %s %s %s" % (one, two, three, four, five)
2284
2284
2285 def savecommitmessage(self, text):
2285 def savecommitmessage(self, text):
2286 fp = self.opener('last-message.txt', 'wb')
2286 fp = self.opener('last-message.txt', 'wb')
2287 try:
2287 try:
2288 fp.write(text)
2288 fp.write(text)
2289 finally:
2289 finally:
2290 fp.close()
2290 fp.close()
2291 return self.pathto(fp.name[len(self.root)+1:])
2291 return self.pathto(fp.name[len(self.root)+1:])
2292
2292
2293 # used to avoid circular references so destructors work
2293 # used to avoid circular references so destructors work
2294 def aftertrans(files):
2294 def aftertrans(files):
2295 renamefiles = [tuple(t) for t in files]
2295 renamefiles = [tuple(t) for t in files]
2296 def a():
2296 def a():
2297 for src, dest in renamefiles:
2297 for src, dest in renamefiles:
2298 util.rename(src, dest)
2298 util.rename(src, dest)
2299 return a
2299 return a
2300
2300
2301 def undoname(fn):
2301 def undoname(fn):
2302 base, name = os.path.split(fn)
2302 base, name = os.path.split(fn)
2303 assert name.startswith('journal')
2303 assert name.startswith('journal')
2304 return os.path.join(base, name.replace('journal', 'undo', 1))
2304 return os.path.join(base, name.replace('journal', 'undo', 1))
2305
2305
2306 def instance(ui, path, create):
2306 def instance(ui, path, create):
2307 return localrepository(ui, util.urllocalpath(path), create)
2307 return localrepository(ui, util.urllocalpath(path), create)
2308
2308
2309 def islocal(path):
2309 def islocal(path):
2310 return True
2310 return True
General Comments 0
You need to be logged in to leave comments. Login now