##// END OF EJS Templates
localrepo: reset _phasesdirty flag after writing
Idan Kamara -
r16113:3f75fb83 stable
parent child Browse files
Show More
@@ -1,2310 +1,2311 b''
1 # localrepo.py - read/write repository class for mercurial
1 # localrepo.py - read/write repository class for mercurial
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 from node import bin, hex, nullid, nullrev, short
8 from node import bin, hex, nullid, nullrev, short
9 from i18n import _
9 from i18n import _
10 import repo, changegroup, subrepo, discovery, pushkey
10 import repo, changegroup, subrepo, discovery, pushkey
11 import changelog, dirstate, filelog, manifest, context, bookmarks, 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 self._dirtyphases = False
952 for k, ce in self._filecache.items():
953 for k, ce in self._filecache.items():
953 if k == 'dirstate':
954 if k == 'dirstate':
954 continue
955 continue
955 ce.refresh()
956 ce.refresh()
956
957
957 l = self._lock(self.sjoin("lock"), wait, unlock,
958 l = self._lock(self.sjoin("lock"), wait, unlock,
958 self.invalidate, _('repository %s') % self.origroot)
959 self.invalidate, _('repository %s') % self.origroot)
959 self._lockref = weakref.ref(l)
960 self._lockref = weakref.ref(l)
960 return l
961 return l
961
962
962 def wlock(self, wait=True):
963 def wlock(self, wait=True):
963 '''Lock the non-store parts of the repository (everything under
964 '''Lock the non-store parts of the repository (everything under
964 .hg except .hg/store) and return a weak reference to the lock.
965 .hg except .hg/store) and return a weak reference to the lock.
965 Use this before modifying files in .hg.'''
966 Use this before modifying files in .hg.'''
966 l = self._wlockref and self._wlockref()
967 l = self._wlockref and self._wlockref()
967 if l is not None and l.held:
968 if l is not None and l.held:
968 l.lock()
969 l.lock()
969 return l
970 return l
970
971
971 def unlock():
972 def unlock():
972 self.dirstate.write()
973 self.dirstate.write()
973 ce = self._filecache.get('dirstate')
974 ce = self._filecache.get('dirstate')
974 if ce:
975 if ce:
975 ce.refresh()
976 ce.refresh()
976
977
977 l = self._lock(self.join("wlock"), wait, unlock,
978 l = self._lock(self.join("wlock"), wait, unlock,
978 self.invalidatedirstate, _('working directory of %s') %
979 self.invalidatedirstate, _('working directory of %s') %
979 self.origroot)
980 self.origroot)
980 self._wlockref = weakref.ref(l)
981 self._wlockref = weakref.ref(l)
981 return l
982 return l
982
983
983 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
984 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
984 """
985 """
985 commit an individual file as part of a larger transaction
986 commit an individual file as part of a larger transaction
986 """
987 """
987
988
988 fname = fctx.path()
989 fname = fctx.path()
989 text = fctx.data()
990 text = fctx.data()
990 flog = self.file(fname)
991 flog = self.file(fname)
991 fparent1 = manifest1.get(fname, nullid)
992 fparent1 = manifest1.get(fname, nullid)
992 fparent2 = fparent2o = manifest2.get(fname, nullid)
993 fparent2 = fparent2o = manifest2.get(fname, nullid)
993
994
994 meta = {}
995 meta = {}
995 copy = fctx.renamed()
996 copy = fctx.renamed()
996 if copy and copy[0] != fname:
997 if copy and copy[0] != fname:
997 # Mark the new revision of this file as a copy of another
998 # Mark the new revision of this file as a copy of another
998 # file. This copy data will effectively act as a parent
999 # file. This copy data will effectively act as a parent
999 # of this new revision. If this is a merge, the first
1000 # of this new revision. If this is a merge, the first
1000 # parent will be the nullid (meaning "look up the copy data")
1001 # parent will be the nullid (meaning "look up the copy data")
1001 # and the second one will be the other parent. For example:
1002 # and the second one will be the other parent. For example:
1002 #
1003 #
1003 # 0 --- 1 --- 3 rev1 changes file foo
1004 # 0 --- 1 --- 3 rev1 changes file foo
1004 # \ / rev2 renames foo to bar and changes it
1005 # \ / rev2 renames foo to bar and changes it
1005 # \- 2 -/ rev3 should have bar with all changes and
1006 # \- 2 -/ rev3 should have bar with all changes and
1006 # should record that bar descends from
1007 # should record that bar descends from
1007 # bar in rev2 and foo in rev1
1008 # bar in rev2 and foo in rev1
1008 #
1009 #
1009 # this allows this merge to succeed:
1010 # this allows this merge to succeed:
1010 #
1011 #
1011 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
1012 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
1012 # \ / merging rev3 and rev4 should use bar@rev2
1013 # \ / merging rev3 and rev4 should use bar@rev2
1013 # \- 2 --- 4 as the merge base
1014 # \- 2 --- 4 as the merge base
1014 #
1015 #
1015
1016
1016 cfname = copy[0]
1017 cfname = copy[0]
1017 crev = manifest1.get(cfname)
1018 crev = manifest1.get(cfname)
1018 newfparent = fparent2
1019 newfparent = fparent2
1019
1020
1020 if manifest2: # branch merge
1021 if manifest2: # branch merge
1021 if fparent2 == nullid or crev is None: # copied on remote side
1022 if fparent2 == nullid or crev is None: # copied on remote side
1022 if cfname in manifest2:
1023 if cfname in manifest2:
1023 crev = manifest2[cfname]
1024 crev = manifest2[cfname]
1024 newfparent = fparent1
1025 newfparent = fparent1
1025
1026
1026 # find source in nearest ancestor if we've lost track
1027 # find source in nearest ancestor if we've lost track
1027 if not crev:
1028 if not crev:
1028 self.ui.debug(" %s: searching for copy revision for %s\n" %
1029 self.ui.debug(" %s: searching for copy revision for %s\n" %
1029 (fname, cfname))
1030 (fname, cfname))
1030 for ancestor in self[None].ancestors():
1031 for ancestor in self[None].ancestors():
1031 if cfname in ancestor:
1032 if cfname in ancestor:
1032 crev = ancestor[cfname].filenode()
1033 crev = ancestor[cfname].filenode()
1033 break
1034 break
1034
1035
1035 if crev:
1036 if crev:
1036 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
1037 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
1037 meta["copy"] = cfname
1038 meta["copy"] = cfname
1038 meta["copyrev"] = hex(crev)
1039 meta["copyrev"] = hex(crev)
1039 fparent1, fparent2 = nullid, newfparent
1040 fparent1, fparent2 = nullid, newfparent
1040 else:
1041 else:
1041 self.ui.warn(_("warning: can't find ancestor for '%s' "
1042 self.ui.warn(_("warning: can't find ancestor for '%s' "
1042 "copied from '%s'!\n") % (fname, cfname))
1043 "copied from '%s'!\n") % (fname, cfname))
1043
1044
1044 elif fparent2 != nullid:
1045 elif fparent2 != nullid:
1045 # is one parent an ancestor of the other?
1046 # is one parent an ancestor of the other?
1046 fparentancestor = flog.ancestor(fparent1, fparent2)
1047 fparentancestor = flog.ancestor(fparent1, fparent2)
1047 if fparentancestor == fparent1:
1048 if fparentancestor == fparent1:
1048 fparent1, fparent2 = fparent2, nullid
1049 fparent1, fparent2 = fparent2, nullid
1049 elif fparentancestor == fparent2:
1050 elif fparentancestor == fparent2:
1050 fparent2 = nullid
1051 fparent2 = nullid
1051
1052
1052 # is the file changed?
1053 # is the file changed?
1053 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
1054 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
1054 changelist.append(fname)
1055 changelist.append(fname)
1055 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
1056 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
1056
1057
1057 # are just the flags changed during merge?
1058 # are just the flags changed during merge?
1058 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
1059 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
1059 changelist.append(fname)
1060 changelist.append(fname)
1060
1061
1061 return fparent1
1062 return fparent1
1062
1063
1063 def commit(self, text="", user=None, date=None, match=None, force=False,
1064 def commit(self, text="", user=None, date=None, match=None, force=False,
1064 editor=False, extra={}):
1065 editor=False, extra={}):
1065 """Add a new revision to current repository.
1066 """Add a new revision to current repository.
1066
1067
1067 Revision information is gathered from the working directory,
1068 Revision information is gathered from the working directory,
1068 match can be used to filter the committed files. If editor is
1069 match can be used to filter the committed files. If editor is
1069 supplied, it is called to get a commit message.
1070 supplied, it is called to get a commit message.
1070 """
1071 """
1071
1072
1072 def fail(f, msg):
1073 def fail(f, msg):
1073 raise util.Abort('%s: %s' % (f, msg))
1074 raise util.Abort('%s: %s' % (f, msg))
1074
1075
1075 if not match:
1076 if not match:
1076 match = matchmod.always(self.root, '')
1077 match = matchmod.always(self.root, '')
1077
1078
1078 if not force:
1079 if not force:
1079 vdirs = []
1080 vdirs = []
1080 match.dir = vdirs.append
1081 match.dir = vdirs.append
1081 match.bad = fail
1082 match.bad = fail
1082
1083
1083 wlock = self.wlock()
1084 wlock = self.wlock()
1084 try:
1085 try:
1085 wctx = self[None]
1086 wctx = self[None]
1086 merge = len(wctx.parents()) > 1
1087 merge = len(wctx.parents()) > 1
1087
1088
1088 if (not force and merge and match and
1089 if (not force and merge and match and
1089 (match.files() or match.anypats())):
1090 (match.files() or match.anypats())):
1090 raise util.Abort(_('cannot partially commit a merge '
1091 raise util.Abort(_('cannot partially commit a merge '
1091 '(do not specify files or patterns)'))
1092 '(do not specify files or patterns)'))
1092
1093
1093 changes = self.status(match=match, clean=force)
1094 changes = self.status(match=match, clean=force)
1094 if force:
1095 if force:
1095 changes[0].extend(changes[6]) # mq may commit unchanged files
1096 changes[0].extend(changes[6]) # mq may commit unchanged files
1096
1097
1097 # check subrepos
1098 # check subrepos
1098 subs = []
1099 subs = []
1099 removedsubs = set()
1100 removedsubs = set()
1100 if '.hgsub' in wctx:
1101 if '.hgsub' in wctx:
1101 # only manage subrepos and .hgsubstate if .hgsub is present
1102 # only manage subrepos and .hgsubstate if .hgsub is present
1102 for p in wctx.parents():
1103 for p in wctx.parents():
1103 removedsubs.update(s for s in p.substate if match(s))
1104 removedsubs.update(s for s in p.substate if match(s))
1104 for s in wctx.substate:
1105 for s in wctx.substate:
1105 removedsubs.discard(s)
1106 removedsubs.discard(s)
1106 if match(s) and wctx.sub(s).dirty():
1107 if match(s) and wctx.sub(s).dirty():
1107 subs.append(s)
1108 subs.append(s)
1108 if (subs or removedsubs):
1109 if (subs or removedsubs):
1109 if (not match('.hgsub') and
1110 if (not match('.hgsub') and
1110 '.hgsub' in (wctx.modified() + wctx.added())):
1111 '.hgsub' in (wctx.modified() + wctx.added())):
1111 raise util.Abort(
1112 raise util.Abort(
1112 _("can't commit subrepos without .hgsub"))
1113 _("can't commit subrepos without .hgsub"))
1113 if '.hgsubstate' not in changes[0]:
1114 if '.hgsubstate' not in changes[0]:
1114 changes[0].insert(0, '.hgsubstate')
1115 changes[0].insert(0, '.hgsubstate')
1115 if '.hgsubstate' in changes[2]:
1116 if '.hgsubstate' in changes[2]:
1116 changes[2].remove('.hgsubstate')
1117 changes[2].remove('.hgsubstate')
1117 elif '.hgsub' in changes[2]:
1118 elif '.hgsub' in changes[2]:
1118 # clean up .hgsubstate when .hgsub is removed
1119 # clean up .hgsubstate when .hgsub is removed
1119 if ('.hgsubstate' in wctx and
1120 if ('.hgsubstate' in wctx and
1120 '.hgsubstate' not in changes[0] + changes[1] + changes[2]):
1121 '.hgsubstate' not in changes[0] + changes[1] + changes[2]):
1121 changes[2].insert(0, '.hgsubstate')
1122 changes[2].insert(0, '.hgsubstate')
1122
1123
1123 if subs and not self.ui.configbool('ui', 'commitsubrepos', False):
1124 if subs and not self.ui.configbool('ui', 'commitsubrepos', False):
1124 changedsubs = [s for s in subs if wctx.sub(s).dirty(True)]
1125 changedsubs = [s for s in subs if wctx.sub(s).dirty(True)]
1125 if changedsubs:
1126 if changedsubs:
1126 raise util.Abort(_("uncommitted changes in subrepo %s")
1127 raise util.Abort(_("uncommitted changes in subrepo %s")
1127 % changedsubs[0],
1128 % changedsubs[0],
1128 hint=_("use --subrepos for recursive commit"))
1129 hint=_("use --subrepos for recursive commit"))
1129
1130
1130 # make sure all explicit patterns are matched
1131 # make sure all explicit patterns are matched
1131 if not force and match.files():
1132 if not force and match.files():
1132 matched = set(changes[0] + changes[1] + changes[2])
1133 matched = set(changes[0] + changes[1] + changes[2])
1133
1134
1134 for f in match.files():
1135 for f in match.files():
1135 if f == '.' or f in matched or f in wctx.substate:
1136 if f == '.' or f in matched or f in wctx.substate:
1136 continue
1137 continue
1137 if f in changes[3]: # missing
1138 if f in changes[3]: # missing
1138 fail(f, _('file not found!'))
1139 fail(f, _('file not found!'))
1139 if f in vdirs: # visited directory
1140 if f in vdirs: # visited directory
1140 d = f + '/'
1141 d = f + '/'
1141 for mf in matched:
1142 for mf in matched:
1142 if mf.startswith(d):
1143 if mf.startswith(d):
1143 break
1144 break
1144 else:
1145 else:
1145 fail(f, _("no match under directory!"))
1146 fail(f, _("no match under directory!"))
1146 elif f not in self.dirstate:
1147 elif f not in self.dirstate:
1147 fail(f, _("file not tracked!"))
1148 fail(f, _("file not tracked!"))
1148
1149
1149 if (not force and not extra.get("close") and not merge
1150 if (not force and not extra.get("close") and not merge
1150 and not (changes[0] or changes[1] or changes[2])
1151 and not (changes[0] or changes[1] or changes[2])
1151 and wctx.branch() == wctx.p1().branch()):
1152 and wctx.branch() == wctx.p1().branch()):
1152 return None
1153 return None
1153
1154
1154 ms = mergemod.mergestate(self)
1155 ms = mergemod.mergestate(self)
1155 for f in changes[0]:
1156 for f in changes[0]:
1156 if f in ms and ms[f] == 'u':
1157 if f in ms and ms[f] == 'u':
1157 raise util.Abort(_("unresolved merge conflicts "
1158 raise util.Abort(_("unresolved merge conflicts "
1158 "(see hg help resolve)"))
1159 "(see hg help resolve)"))
1159
1160
1160 cctx = context.workingctx(self, text, user, date, extra, changes)
1161 cctx = context.workingctx(self, text, user, date, extra, changes)
1161 if editor:
1162 if editor:
1162 cctx._text = editor(self, cctx, subs)
1163 cctx._text = editor(self, cctx, subs)
1163 edited = (text != cctx._text)
1164 edited = (text != cctx._text)
1164
1165
1165 # commit subs
1166 # commit subs
1166 if subs or removedsubs:
1167 if subs or removedsubs:
1167 state = wctx.substate.copy()
1168 state = wctx.substate.copy()
1168 for s in sorted(subs):
1169 for s in sorted(subs):
1169 sub = wctx.sub(s)
1170 sub = wctx.sub(s)
1170 self.ui.status(_('committing subrepository %s\n') %
1171 self.ui.status(_('committing subrepository %s\n') %
1171 subrepo.subrelpath(sub))
1172 subrepo.subrelpath(sub))
1172 sr = sub.commit(cctx._text, user, date)
1173 sr = sub.commit(cctx._text, user, date)
1173 state[s] = (state[s][0], sr)
1174 state[s] = (state[s][0], sr)
1174 subrepo.writestate(self, state)
1175 subrepo.writestate(self, state)
1175
1176
1176 # Save commit message in case this transaction gets rolled back
1177 # Save commit message in case this transaction gets rolled back
1177 # (e.g. by a pretxncommit hook). Leave the content alone on
1178 # (e.g. by a pretxncommit hook). Leave the content alone on
1178 # the assumption that the user will use the same editor again.
1179 # the assumption that the user will use the same editor again.
1179 msgfn = self.savecommitmessage(cctx._text)
1180 msgfn = self.savecommitmessage(cctx._text)
1180
1181
1181 p1, p2 = self.dirstate.parents()
1182 p1, p2 = self.dirstate.parents()
1182 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1183 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1183 try:
1184 try:
1184 self.hook("precommit", throw=True, parent1=hookp1, parent2=hookp2)
1185 self.hook("precommit", throw=True, parent1=hookp1, parent2=hookp2)
1185 ret = self.commitctx(cctx, True)
1186 ret = self.commitctx(cctx, True)
1186 except:
1187 except:
1187 if edited:
1188 if edited:
1188 self.ui.write(
1189 self.ui.write(
1189 _('note: commit message saved in %s\n') % msgfn)
1190 _('note: commit message saved in %s\n') % msgfn)
1190 raise
1191 raise
1191
1192
1192 # update bookmarks, dirstate and mergestate
1193 # update bookmarks, dirstate and mergestate
1193 bookmarks.update(self, p1, ret)
1194 bookmarks.update(self, p1, ret)
1194 for f in changes[0] + changes[1]:
1195 for f in changes[0] + changes[1]:
1195 self.dirstate.normal(f)
1196 self.dirstate.normal(f)
1196 for f in changes[2]:
1197 for f in changes[2]:
1197 self.dirstate.drop(f)
1198 self.dirstate.drop(f)
1198 self.dirstate.setparents(ret)
1199 self.dirstate.setparents(ret)
1199 ms.reset()
1200 ms.reset()
1200 finally:
1201 finally:
1201 wlock.release()
1202 wlock.release()
1202
1203
1203 self.hook("commit", node=hex(ret), parent1=hookp1, parent2=hookp2)
1204 self.hook("commit", node=hex(ret), parent1=hookp1, parent2=hookp2)
1204 return ret
1205 return ret
1205
1206
1206 def commitctx(self, ctx, error=False):
1207 def commitctx(self, ctx, error=False):
1207 """Add a new revision to current repository.
1208 """Add a new revision to current repository.
1208 Revision information is passed via the context argument.
1209 Revision information is passed via the context argument.
1209 """
1210 """
1210
1211
1211 tr = lock = None
1212 tr = lock = None
1212 removed = list(ctx.removed())
1213 removed = list(ctx.removed())
1213 p1, p2 = ctx.p1(), ctx.p2()
1214 p1, p2 = ctx.p1(), ctx.p2()
1214 user = ctx.user()
1215 user = ctx.user()
1215
1216
1216 lock = self.lock()
1217 lock = self.lock()
1217 try:
1218 try:
1218 tr = self.transaction("commit")
1219 tr = self.transaction("commit")
1219 trp = weakref.proxy(tr)
1220 trp = weakref.proxy(tr)
1220
1221
1221 if ctx.files():
1222 if ctx.files():
1222 m1 = p1.manifest().copy()
1223 m1 = p1.manifest().copy()
1223 m2 = p2.manifest()
1224 m2 = p2.manifest()
1224
1225
1225 # check in files
1226 # check in files
1226 new = {}
1227 new = {}
1227 changed = []
1228 changed = []
1228 linkrev = len(self)
1229 linkrev = len(self)
1229 for f in sorted(ctx.modified() + ctx.added()):
1230 for f in sorted(ctx.modified() + ctx.added()):
1230 self.ui.note(f + "\n")
1231 self.ui.note(f + "\n")
1231 try:
1232 try:
1232 fctx = ctx[f]
1233 fctx = ctx[f]
1233 new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
1234 new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
1234 changed)
1235 changed)
1235 m1.set(f, fctx.flags())
1236 m1.set(f, fctx.flags())
1236 except OSError, inst:
1237 except OSError, inst:
1237 self.ui.warn(_("trouble committing %s!\n") % f)
1238 self.ui.warn(_("trouble committing %s!\n") % f)
1238 raise
1239 raise
1239 except IOError, inst:
1240 except IOError, inst:
1240 errcode = getattr(inst, 'errno', errno.ENOENT)
1241 errcode = getattr(inst, 'errno', errno.ENOENT)
1241 if error or errcode and errcode != errno.ENOENT:
1242 if error or errcode and errcode != errno.ENOENT:
1242 self.ui.warn(_("trouble committing %s!\n") % f)
1243 self.ui.warn(_("trouble committing %s!\n") % f)
1243 raise
1244 raise
1244 else:
1245 else:
1245 removed.append(f)
1246 removed.append(f)
1246
1247
1247 # update manifest
1248 # update manifest
1248 m1.update(new)
1249 m1.update(new)
1249 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1250 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]
1251 drop = [f for f in removed if f in m1]
1251 for f in drop:
1252 for f in drop:
1252 del m1[f]
1253 del m1[f]
1253 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1254 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1254 p2.manifestnode(), (new, drop))
1255 p2.manifestnode(), (new, drop))
1255 files = changed + removed
1256 files = changed + removed
1256 else:
1257 else:
1257 mn = p1.manifestnode()
1258 mn = p1.manifestnode()
1258 files = []
1259 files = []
1259
1260
1260 # update changelog
1261 # update changelog
1261 self.changelog.delayupdate()
1262 self.changelog.delayupdate()
1262 n = self.changelog.add(mn, files, ctx.description(),
1263 n = self.changelog.add(mn, files, ctx.description(),
1263 trp, p1.node(), p2.node(),
1264 trp, p1.node(), p2.node(),
1264 user, ctx.date(), ctx.extra().copy())
1265 user, ctx.date(), ctx.extra().copy())
1265 p = lambda: self.changelog.writepending() and self.root or ""
1266 p = lambda: self.changelog.writepending() and self.root or ""
1266 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1267 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1267 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1268 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1268 parent2=xp2, pending=p)
1269 parent2=xp2, pending=p)
1269 self.changelog.finalize(trp)
1270 self.changelog.finalize(trp)
1270 # set the new commit is proper phase
1271 # set the new commit is proper phase
1271 targetphase = phases.newcommitphase(self.ui)
1272 targetphase = phases.newcommitphase(self.ui)
1272 if targetphase:
1273 if targetphase:
1273 # retract boundary do not alter parent changeset.
1274 # retract boundary do not alter parent changeset.
1274 # if a parent have higher the resulting phase will
1275 # if a parent have higher the resulting phase will
1275 # be compliant anyway
1276 # be compliant anyway
1276 #
1277 #
1277 # 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
1278 phases.retractboundary(self, targetphase, [n])
1279 phases.retractboundary(self, targetphase, [n])
1279 tr.close()
1280 tr.close()
1280 self.updatebranchcache()
1281 self.updatebranchcache()
1281 return n
1282 return n
1282 finally:
1283 finally:
1283 if tr:
1284 if tr:
1284 tr.release()
1285 tr.release()
1285 lock.release()
1286 lock.release()
1286
1287
1287 def destroyed(self):
1288 def destroyed(self):
1288 '''Inform the repository that nodes have been destroyed.
1289 '''Inform the repository that nodes have been destroyed.
1289 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
1290 place for anything that has to be done after destroying history.'''
1291 place for anything that has to be done after destroying history.'''
1291 # 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
1292 # 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
1293
1294
1294 # Ensure the persistent tag cache is updated. Doing it now
1295 # Ensure the persistent tag cache is updated. Doing it now
1295 # means that the tag cache only has to worry about destroyed
1296 # means that the tag cache only has to worry about destroyed
1296 # heads immediately after a strip/rollback. That in turn
1297 # heads immediately after a strip/rollback. That in turn
1297 # guarantees that "cachetip == currenttip" (comparing both rev
1298 # guarantees that "cachetip == currenttip" (comparing both rev
1298 # and node) always means no nodes have been added or destroyed.
1299 # and node) always means no nodes have been added or destroyed.
1299
1300
1300 # XXX this is suboptimal when qrefresh'ing: we strip the current
1301 # XXX this is suboptimal when qrefresh'ing: we strip the current
1301 # head, refresh the tag cache, then immediately add a new head.
1302 # head, refresh the tag cache, then immediately add a new head.
1302 # 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
1303 # tag cache retrieval" case to work.
1304 # tag cache retrieval" case to work.
1304 self.invalidatecaches()
1305 self.invalidatecaches()
1305
1306
1306 def walk(self, match, node=None):
1307 def walk(self, match, node=None):
1307 '''
1308 '''
1308 walk recursively through the directory tree or a given
1309 walk recursively through the directory tree or a given
1309 changeset, finding all files matched by the match
1310 changeset, finding all files matched by the match
1310 function
1311 function
1311 '''
1312 '''
1312 return self[node].walk(match)
1313 return self[node].walk(match)
1313
1314
1314 def status(self, node1='.', node2=None, match=None,
1315 def status(self, node1='.', node2=None, match=None,
1315 ignored=False, clean=False, unknown=False,
1316 ignored=False, clean=False, unknown=False,
1316 listsubrepos=False):
1317 listsubrepos=False):
1317 """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
1318
1319
1319 If node1 is None, use the first dirstate parent instead.
1320 If node1 is None, use the first dirstate parent instead.
1320 If node2 is None, compare node1 with working directory.
1321 If node2 is None, compare node1 with working directory.
1321 """
1322 """
1322
1323
1323 def mfmatches(ctx):
1324 def mfmatches(ctx):
1324 mf = ctx.manifest().copy()
1325 mf = ctx.manifest().copy()
1325 for fn in mf.keys():
1326 for fn in mf.keys():
1326 if not match(fn):
1327 if not match(fn):
1327 del mf[fn]
1328 del mf[fn]
1328 return mf
1329 return mf
1329
1330
1330 if isinstance(node1, context.changectx):
1331 if isinstance(node1, context.changectx):
1331 ctx1 = node1
1332 ctx1 = node1
1332 else:
1333 else:
1333 ctx1 = self[node1]
1334 ctx1 = self[node1]
1334 if isinstance(node2, context.changectx):
1335 if isinstance(node2, context.changectx):
1335 ctx2 = node2
1336 ctx2 = node2
1336 else:
1337 else:
1337 ctx2 = self[node2]
1338 ctx2 = self[node2]
1338
1339
1339 working = ctx2.rev() is None
1340 working = ctx2.rev() is None
1340 parentworking = working and ctx1 == self['.']
1341 parentworking = working and ctx1 == self['.']
1341 match = match or matchmod.always(self.root, self.getcwd())
1342 match = match or matchmod.always(self.root, self.getcwd())
1342 listignored, listclean, listunknown = ignored, clean, unknown
1343 listignored, listclean, listunknown = ignored, clean, unknown
1343
1344
1344 # load earliest manifest first for caching reasons
1345 # load earliest manifest first for caching reasons
1345 if not working and ctx2.rev() < ctx1.rev():
1346 if not working and ctx2.rev() < ctx1.rev():
1346 ctx2.manifest()
1347 ctx2.manifest()
1347
1348
1348 if not parentworking:
1349 if not parentworking:
1349 def bad(f, msg):
1350 def bad(f, msg):
1350 if f not in ctx1:
1351 if f not in ctx1:
1351 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1352 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1352 match.bad = bad
1353 match.bad = bad
1353
1354
1354 if working: # we need to scan the working dir
1355 if working: # we need to scan the working dir
1355 subrepos = []
1356 subrepos = []
1356 if '.hgsub' in self.dirstate:
1357 if '.hgsub' in self.dirstate:
1357 subrepos = ctx2.substate.keys()
1358 subrepos = ctx2.substate.keys()
1358 s = self.dirstate.status(match, subrepos, listignored,
1359 s = self.dirstate.status(match, subrepos, listignored,
1359 listclean, listunknown)
1360 listclean, listunknown)
1360 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1361 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1361
1362
1362 # check for any possibly clean files
1363 # check for any possibly clean files
1363 if parentworking and cmp:
1364 if parentworking and cmp:
1364 fixup = []
1365 fixup = []
1365 # do a full compare of any files that might have changed
1366 # do a full compare of any files that might have changed
1366 for f in sorted(cmp):
1367 for f in sorted(cmp):
1367 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)
1368 or ctx1[f].cmp(ctx2[f])):
1369 or ctx1[f].cmp(ctx2[f])):
1369 modified.append(f)
1370 modified.append(f)
1370 else:
1371 else:
1371 fixup.append(f)
1372 fixup.append(f)
1372
1373
1373 # update dirstate for files that are actually clean
1374 # update dirstate for files that are actually clean
1374 if fixup:
1375 if fixup:
1375 if listclean:
1376 if listclean:
1376 clean += fixup
1377 clean += fixup
1377
1378
1378 try:
1379 try:
1379 # updating the dirstate is optional
1380 # updating the dirstate is optional
1380 # so we don't wait on the lock
1381 # so we don't wait on the lock
1381 wlock = self.wlock(False)
1382 wlock = self.wlock(False)
1382 try:
1383 try:
1383 for f in fixup:
1384 for f in fixup:
1384 self.dirstate.normal(f)
1385 self.dirstate.normal(f)
1385 finally:
1386 finally:
1386 wlock.release()
1387 wlock.release()
1387 except error.LockError:
1388 except error.LockError:
1388 pass
1389 pass
1389
1390
1390 if not parentworking:
1391 if not parentworking:
1391 mf1 = mfmatches(ctx1)
1392 mf1 = mfmatches(ctx1)
1392 if working:
1393 if working:
1393 # we are comparing working dir against non-parent
1394 # we are comparing working dir against non-parent
1394 # generate a pseudo-manifest for the working dir
1395 # generate a pseudo-manifest for the working dir
1395 mf2 = mfmatches(self['.'])
1396 mf2 = mfmatches(self['.'])
1396 for f in cmp + modified + added:
1397 for f in cmp + modified + added:
1397 mf2[f] = None
1398 mf2[f] = None
1398 mf2.set(f, ctx2.flags(f))
1399 mf2.set(f, ctx2.flags(f))
1399 for f in removed:
1400 for f in removed:
1400 if f in mf2:
1401 if f in mf2:
1401 del mf2[f]
1402 del mf2[f]
1402 else:
1403 else:
1403 # we are comparing two revisions
1404 # we are comparing two revisions
1404 deleted, unknown, ignored = [], [], []
1405 deleted, unknown, ignored = [], [], []
1405 mf2 = mfmatches(ctx2)
1406 mf2 = mfmatches(ctx2)
1406
1407
1407 modified, added, clean = [], [], []
1408 modified, added, clean = [], [], []
1408 for fn in mf2:
1409 for fn in mf2:
1409 if fn in mf1:
1410 if fn in mf1:
1410 if (fn not in deleted and
1411 if (fn not in deleted and
1411 (mf1.flags(fn) != mf2.flags(fn) or
1412 (mf1.flags(fn) != mf2.flags(fn) or
1412 (mf1[fn] != mf2[fn] and
1413 (mf1[fn] != mf2[fn] and
1413 (mf2[fn] or ctx1[fn].cmp(ctx2[fn]))))):
1414 (mf2[fn] or ctx1[fn].cmp(ctx2[fn]))))):
1414 modified.append(fn)
1415 modified.append(fn)
1415 elif listclean:
1416 elif listclean:
1416 clean.append(fn)
1417 clean.append(fn)
1417 del mf1[fn]
1418 del mf1[fn]
1418 elif fn not in deleted:
1419 elif fn not in deleted:
1419 added.append(fn)
1420 added.append(fn)
1420 removed = mf1.keys()
1421 removed = mf1.keys()
1421
1422
1422 if working and modified and not self.dirstate._checklink:
1423 if working and modified and not self.dirstate._checklink:
1423 # Symlink placeholders may get non-symlink-like contents
1424 # Symlink placeholders may get non-symlink-like contents
1424 # via user error or dereferencing by NFS or Samba servers,
1425 # via user error or dereferencing by NFS or Samba servers,
1425 # 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
1426 # symlink
1427 # symlink
1427 sane = []
1428 sane = []
1428 for f in modified:
1429 for f in modified:
1429 if ctx2.flags(f) == 'l':
1430 if ctx2.flags(f) == 'l':
1430 d = ctx2[f].data()
1431 d = ctx2[f].data()
1431 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):
1432 self.ui.debug('ignoring suspect symlink placeholder'
1433 self.ui.debug('ignoring suspect symlink placeholder'
1433 ' "%s"\n' % f)
1434 ' "%s"\n' % f)
1434 continue
1435 continue
1435 sane.append(f)
1436 sane.append(f)
1436 modified = sane
1437 modified = sane
1437
1438
1438 r = modified, added, removed, deleted, unknown, ignored, clean
1439 r = modified, added, removed, deleted, unknown, ignored, clean
1439
1440
1440 if listsubrepos:
1441 if listsubrepos:
1441 for subpath, sub in subrepo.itersubrepos(ctx1, ctx2):
1442 for subpath, sub in subrepo.itersubrepos(ctx1, ctx2):
1442 if working:
1443 if working:
1443 rev2 = None
1444 rev2 = None
1444 else:
1445 else:
1445 rev2 = ctx2.substate[subpath][1]
1446 rev2 = ctx2.substate[subpath][1]
1446 try:
1447 try:
1447 submatch = matchmod.narrowmatcher(subpath, match)
1448 submatch = matchmod.narrowmatcher(subpath, match)
1448 s = sub.status(rev2, match=submatch, ignored=listignored,
1449 s = sub.status(rev2, match=submatch, ignored=listignored,
1449 clean=listclean, unknown=listunknown,
1450 clean=listclean, unknown=listunknown,
1450 listsubrepos=True)
1451 listsubrepos=True)
1451 for rfiles, sfiles in zip(r, s):
1452 for rfiles, sfiles in zip(r, s):
1452 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
1453 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
1453 except error.LookupError:
1454 except error.LookupError:
1454 self.ui.status(_("skipping missing subrepository: %s\n")
1455 self.ui.status(_("skipping missing subrepository: %s\n")
1455 % subpath)
1456 % subpath)
1456
1457
1457 for l in r:
1458 for l in r:
1458 l.sort()
1459 l.sort()
1459 return r
1460 return r
1460
1461
1461 def heads(self, start=None):
1462 def heads(self, start=None):
1462 heads = self.changelog.heads(start)
1463 heads = self.changelog.heads(start)
1463 # sort the output in rev descending order
1464 # sort the output in rev descending order
1464 return sorted(heads, key=self.changelog.rev, reverse=True)
1465 return sorted(heads, key=self.changelog.rev, reverse=True)
1465
1466
1466 def branchheads(self, branch=None, start=None, closed=False):
1467 def branchheads(self, branch=None, start=None, closed=False):
1467 '''return a (possibly filtered) list of heads for the given branch
1468 '''return a (possibly filtered) list of heads for the given branch
1468
1469
1469 Heads are returned in topological order, from newest to oldest.
1470 Heads are returned in topological order, from newest to oldest.
1470 If branch is None, use the dirstate branch.
1471 If branch is None, use the dirstate branch.
1471 If start is not None, return only heads reachable from start.
1472 If start is not None, return only heads reachable from start.
1472 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.
1473 '''
1474 '''
1474 if branch is None:
1475 if branch is None:
1475 branch = self[None].branch()
1476 branch = self[None].branch()
1476 branches = self.branchmap()
1477 branches = self.branchmap()
1477 if branch not in branches:
1478 if branch not in branches:
1478 return []
1479 return []
1479 # the cache returns heads ordered lowest to highest
1480 # the cache returns heads ordered lowest to highest
1480 bheads = list(reversed(branches[branch]))
1481 bheads = list(reversed(branches[branch]))
1481 if start is not None:
1482 if start is not None:
1482 # filter out the heads that cannot be reached from startrev
1483 # filter out the heads that cannot be reached from startrev
1483 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1484 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1484 bheads = [h for h in bheads if h in fbheads]
1485 bheads = [h for h in bheads if h in fbheads]
1485 if not closed:
1486 if not closed:
1486 bheads = [h for h in bheads if
1487 bheads = [h for h in bheads if
1487 ('close' not in self.changelog.read(h)[5])]
1488 ('close' not in self.changelog.read(h)[5])]
1488 return bheads
1489 return bheads
1489
1490
1490 def branches(self, nodes):
1491 def branches(self, nodes):
1491 if not nodes:
1492 if not nodes:
1492 nodes = [self.changelog.tip()]
1493 nodes = [self.changelog.tip()]
1493 b = []
1494 b = []
1494 for n in nodes:
1495 for n in nodes:
1495 t = n
1496 t = n
1496 while True:
1497 while True:
1497 p = self.changelog.parents(n)
1498 p = self.changelog.parents(n)
1498 if p[1] != nullid or p[0] == nullid:
1499 if p[1] != nullid or p[0] == nullid:
1499 b.append((t, n, p[0], p[1]))
1500 b.append((t, n, p[0], p[1]))
1500 break
1501 break
1501 n = p[0]
1502 n = p[0]
1502 return b
1503 return b
1503
1504
1504 def between(self, pairs):
1505 def between(self, pairs):
1505 r = []
1506 r = []
1506
1507
1507 for top, bottom in pairs:
1508 for top, bottom in pairs:
1508 n, l, i = top, [], 0
1509 n, l, i = top, [], 0
1509 f = 1
1510 f = 1
1510
1511
1511 while n != bottom and n != nullid:
1512 while n != bottom and n != nullid:
1512 p = self.changelog.parents(n)[0]
1513 p = self.changelog.parents(n)[0]
1513 if i == f:
1514 if i == f:
1514 l.append(n)
1515 l.append(n)
1515 f = f * 2
1516 f = f * 2
1516 n = p
1517 n = p
1517 i += 1
1518 i += 1
1518
1519
1519 r.append(l)
1520 r.append(l)
1520
1521
1521 return r
1522 return r
1522
1523
1523 def pull(self, remote, heads=None, force=False):
1524 def pull(self, remote, heads=None, force=False):
1524 lock = self.lock()
1525 lock = self.lock()
1525 try:
1526 try:
1526 tmp = discovery.findcommonincoming(self, remote, heads=heads,
1527 tmp = discovery.findcommonincoming(self, remote, heads=heads,
1527 force=force)
1528 force=force)
1528 common, fetch, rheads = tmp
1529 common, fetch, rheads = tmp
1529 if not fetch:
1530 if not fetch:
1530 self.ui.status(_("no changes found\n"))
1531 self.ui.status(_("no changes found\n"))
1531 added = []
1532 added = []
1532 result = 0
1533 result = 0
1533 else:
1534 else:
1534 if heads is None and list(common) == [nullid]:
1535 if heads is None and list(common) == [nullid]:
1535 self.ui.status(_("requesting all changes\n"))
1536 self.ui.status(_("requesting all changes\n"))
1536 elif heads is None and remote.capable('changegroupsubset'):
1537 elif heads is None and remote.capable('changegroupsubset'):
1537 # issue1320, avoid a race if remote changed after discovery
1538 # issue1320, avoid a race if remote changed after discovery
1538 heads = rheads
1539 heads = rheads
1539
1540
1540 if remote.capable('getbundle'):
1541 if remote.capable('getbundle'):
1541 cg = remote.getbundle('pull', common=common,
1542 cg = remote.getbundle('pull', common=common,
1542 heads=heads or rheads)
1543 heads=heads or rheads)
1543 elif heads is None:
1544 elif heads is None:
1544 cg = remote.changegroup(fetch, 'pull')
1545 cg = remote.changegroup(fetch, 'pull')
1545 elif not remote.capable('changegroupsubset'):
1546 elif not remote.capable('changegroupsubset'):
1546 raise util.Abort(_("partial pull cannot be done because "
1547 raise util.Abort(_("partial pull cannot be done because "
1547 "other repository doesn't support "
1548 "other repository doesn't support "
1548 "changegroupsubset."))
1549 "changegroupsubset."))
1549 else:
1550 else:
1550 cg = remote.changegroupsubset(fetch, heads, 'pull')
1551 cg = remote.changegroupsubset(fetch, heads, 'pull')
1551 clstart = len(self.changelog)
1552 clstart = len(self.changelog)
1552 result = self.addchangegroup(cg, 'pull', remote.url())
1553 result = self.addchangegroup(cg, 'pull', remote.url())
1553 clend = len(self.changelog)
1554 clend = len(self.changelog)
1554 added = [self.changelog.node(r) for r in xrange(clstart, clend)]
1555 added = [self.changelog.node(r) for r in xrange(clstart, clend)]
1555
1556
1556 # compute target subset
1557 # compute target subset
1557 if heads is None:
1558 if heads is None:
1558 # We pulled every thing possible
1559 # We pulled every thing possible
1559 # sync on everything common
1560 # sync on everything common
1560 subset = common + added
1561 subset = common + added
1561 else:
1562 else:
1562 # We pulled a specific subset
1563 # We pulled a specific subset
1563 # sync on this subset
1564 # sync on this subset
1564 subset = heads
1565 subset = heads
1565
1566
1566 # Get remote phases data from remote
1567 # Get remote phases data from remote
1567 remotephases = remote.listkeys('phases')
1568 remotephases = remote.listkeys('phases')
1568 publishing = bool(remotephases.get('publishing', False))
1569 publishing = bool(remotephases.get('publishing', False))
1569 if remotephases and not publishing:
1570 if remotephases and not publishing:
1570 # remote is new and unpublishing
1571 # remote is new and unpublishing
1571 pheads, _dr = phases.analyzeremotephases(self, subset,
1572 pheads, _dr = phases.analyzeremotephases(self, subset,
1572 remotephases)
1573 remotephases)
1573 phases.advanceboundary(self, phases.public, pheads)
1574 phases.advanceboundary(self, phases.public, pheads)
1574 phases.advanceboundary(self, phases.draft, subset)
1575 phases.advanceboundary(self, phases.draft, subset)
1575 else:
1576 else:
1576 # Remote is old or publishing all common changesets
1577 # Remote is old or publishing all common changesets
1577 # should be seen as public
1578 # should be seen as public
1578 phases.advanceboundary(self, phases.public, subset)
1579 phases.advanceboundary(self, phases.public, subset)
1579 finally:
1580 finally:
1580 lock.release()
1581 lock.release()
1581
1582
1582 return result
1583 return result
1583
1584
1584 def checkpush(self, force, revs):
1585 def checkpush(self, force, revs):
1585 """Extensions can override this function if additional checks have
1586 """Extensions can override this function if additional checks have
1586 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
1587 command.
1588 command.
1588 """
1589 """
1589 pass
1590 pass
1590
1591
1591 def push(self, remote, force=False, revs=None, newbranch=False):
1592 def push(self, remote, force=False, revs=None, newbranch=False):
1592 '''Push outgoing changesets (limited by revs) from the current
1593 '''Push outgoing changesets (limited by revs) from the current
1593 repository to remote. Return an integer:
1594 repository to remote. Return an integer:
1594 - None means nothing to push
1595 - None means nothing to push
1595 - 0 means HTTP error
1596 - 0 means HTTP error
1596 - 1 means we pushed and remote head count is unchanged *or*
1597 - 1 means we pushed and remote head count is unchanged *or*
1597 we have outgoing changesets but refused to push
1598 we have outgoing changesets but refused to push
1598 - other values as described by addchangegroup()
1599 - other values as described by addchangegroup()
1599 '''
1600 '''
1600 # there are two ways to push to remote repo:
1601 # there are two ways to push to remote repo:
1601 #
1602 #
1602 # addchangegroup assumes local user can lock remote
1603 # addchangegroup assumes local user can lock remote
1603 # repo (local filesystem, old ssh servers).
1604 # repo (local filesystem, old ssh servers).
1604 #
1605 #
1605 # unbundle assumes local user cannot lock remote repo (new ssh
1606 # unbundle assumes local user cannot lock remote repo (new ssh
1606 # servers, http servers).
1607 # servers, http servers).
1607
1608
1608 # get local lock as we might write phase data
1609 # get local lock as we might write phase data
1609 locallock = self.lock()
1610 locallock = self.lock()
1610 try:
1611 try:
1611 self.checkpush(force, revs)
1612 self.checkpush(force, revs)
1612 lock = None
1613 lock = None
1613 unbundle = remote.capable('unbundle')
1614 unbundle = remote.capable('unbundle')
1614 if not unbundle:
1615 if not unbundle:
1615 lock = remote.lock()
1616 lock = remote.lock()
1616 try:
1617 try:
1617 # discovery
1618 # discovery
1618 fci = discovery.findcommonincoming
1619 fci = discovery.findcommonincoming
1619 commoninc = fci(self, remote, force=force)
1620 commoninc = fci(self, remote, force=force)
1620 common, inc, remoteheads = commoninc
1621 common, inc, remoteheads = commoninc
1621 fco = discovery.findcommonoutgoing
1622 fco = discovery.findcommonoutgoing
1622 outgoing = fco(self, remote, onlyheads=revs,
1623 outgoing = fco(self, remote, onlyheads=revs,
1623 commoninc=commoninc, force=force)
1624 commoninc=commoninc, force=force)
1624
1625
1625
1626
1626 if not outgoing.missing:
1627 if not outgoing.missing:
1627 # nothing to push
1628 # nothing to push
1628 scmutil.nochangesfound(self.ui, outgoing.excluded)
1629 scmutil.nochangesfound(self.ui, outgoing.excluded)
1629 ret = None
1630 ret = None
1630 else:
1631 else:
1631 # something to push
1632 # something to push
1632 if not force:
1633 if not force:
1633 discovery.checkheads(self, remote, outgoing,
1634 discovery.checkheads(self, remote, outgoing,
1634 remoteheads, newbranch,
1635 remoteheads, newbranch,
1635 bool(inc))
1636 bool(inc))
1636
1637
1637 # create a changegroup from local
1638 # create a changegroup from local
1638 if revs is None and not outgoing.excluded:
1639 if revs is None and not outgoing.excluded:
1639 # push everything,
1640 # push everything,
1640 # use the fast path, no race possible on push
1641 # use the fast path, no race possible on push
1641 cg = self._changegroup(outgoing.missing, 'push')
1642 cg = self._changegroup(outgoing.missing, 'push')
1642 else:
1643 else:
1643 cg = self.getlocalbundle('push', outgoing)
1644 cg = self.getlocalbundle('push', outgoing)
1644
1645
1645 # apply changegroup to remote
1646 # apply changegroup to remote
1646 if unbundle:
1647 if unbundle:
1647 # local repo finds heads on server, finds out what
1648 # local repo finds heads on server, finds out what
1648 # revs it must push. once revs transferred, if server
1649 # revs it must push. once revs transferred, if server
1649 # finds it has different heads (someone else won
1650 # finds it has different heads (someone else won
1650 # commit/push race), server aborts.
1651 # commit/push race), server aborts.
1651 if force:
1652 if force:
1652 remoteheads = ['force']
1653 remoteheads = ['force']
1653 # ssh: return remote's addchangegroup()
1654 # ssh: return remote's addchangegroup()
1654 # http: return remote's addchangegroup() or 0 for error
1655 # http: return remote's addchangegroup() or 0 for error
1655 ret = remote.unbundle(cg, remoteheads, 'push')
1656 ret = remote.unbundle(cg, remoteheads, 'push')
1656 else:
1657 else:
1657 # we return an integer indicating remote head count change
1658 # we return an integer indicating remote head count change
1658 ret = remote.addchangegroup(cg, 'push', self.url())
1659 ret = remote.addchangegroup(cg, 'push', self.url())
1659
1660
1660 if ret:
1661 if ret:
1661 # push succeed, synchonize target of the push
1662 # push succeed, synchonize target of the push
1662 cheads = outgoing.missingheads
1663 cheads = outgoing.missingheads
1663 elif revs is None:
1664 elif revs is None:
1664 # All out push fails. synchronize all common
1665 # All out push fails. synchronize all common
1665 cheads = outgoing.commonheads
1666 cheads = outgoing.commonheads
1666 else:
1667 else:
1667 # I want cheads = heads(::missingheads and ::commonheads)
1668 # I want cheads = heads(::missingheads and ::commonheads)
1668 # (missingheads is revs with secret changeset filtered out)
1669 # (missingheads is revs with secret changeset filtered out)
1669 #
1670 #
1670 # This can be expressed as:
1671 # This can be expressed as:
1671 # cheads = ( (missingheads and ::commonheads)
1672 # cheads = ( (missingheads and ::commonheads)
1672 # + (commonheads and ::missingheads))"
1673 # + (commonheads and ::missingheads))"
1673 # )
1674 # )
1674 #
1675 #
1675 # while trying to push we already computed the following:
1676 # while trying to push we already computed the following:
1676 # common = (::commonheads)
1677 # common = (::commonheads)
1677 # missing = ((commonheads::missingheads) - commonheads)
1678 # missing = ((commonheads::missingheads) - commonheads)
1678 #
1679 #
1679 # We can pick:
1680 # We can pick:
1680 # * missingheads part of comon (::commonheads)
1681 # * missingheads part of comon (::commonheads)
1681 common = set(outgoing.common)
1682 common = set(outgoing.common)
1682 cheads = [node for node in revs if node in common]
1683 cheads = [node for node in revs if node in common]
1683 # and
1684 # and
1684 # * commonheads parents on missing
1685 # * commonheads parents on missing
1685 revset = self.set('%ln and parents(roots(%ln))',
1686 revset = self.set('%ln and parents(roots(%ln))',
1686 outgoing.commonheads,
1687 outgoing.commonheads,
1687 outgoing.missing)
1688 outgoing.missing)
1688 cheads.extend(c.node() for c in revset)
1689 cheads.extend(c.node() for c in revset)
1689 # even when we don't push, exchanging phase data is useful
1690 # even when we don't push, exchanging phase data is useful
1690 remotephases = remote.listkeys('phases')
1691 remotephases = remote.listkeys('phases')
1691 if not remotephases: # old server or public only repo
1692 if not remotephases: # old server or public only repo
1692 phases.advanceboundary(self, phases.public, cheads)
1693 phases.advanceboundary(self, phases.public, cheads)
1693 # don't push any phase data as there is nothing to push
1694 # don't push any phase data as there is nothing to push
1694 else:
1695 else:
1695 ana = phases.analyzeremotephases(self, cheads, remotephases)
1696 ana = phases.analyzeremotephases(self, cheads, remotephases)
1696 pheads, droots = ana
1697 pheads, droots = ana
1697 ### Apply remote phase on local
1698 ### Apply remote phase on local
1698 if remotephases.get('publishing', False):
1699 if remotephases.get('publishing', False):
1699 phases.advanceboundary(self, phases.public, cheads)
1700 phases.advanceboundary(self, phases.public, cheads)
1700 else: # publish = False
1701 else: # publish = False
1701 phases.advanceboundary(self, phases.public, pheads)
1702 phases.advanceboundary(self, phases.public, pheads)
1702 phases.advanceboundary(self, phases.draft, cheads)
1703 phases.advanceboundary(self, phases.draft, cheads)
1703 ### Apply local phase on remote
1704 ### Apply local phase on remote
1704
1705
1705 # Get the list of all revs draft on remote by public here.
1706 # Get the list of all revs draft on remote by public here.
1706 # XXX Beware that revset break if droots is not strictly
1707 # XXX Beware that revset break if droots is not strictly
1707 # XXX root we may want to ensure it is but it is costly
1708 # XXX root we may want to ensure it is but it is costly
1708 outdated = self.set('heads((%ln::%ln) and public())',
1709 outdated = self.set('heads((%ln::%ln) and public())',
1709 droots, cheads)
1710 droots, cheads)
1710 for newremotehead in outdated:
1711 for newremotehead in outdated:
1711 r = remote.pushkey('phases',
1712 r = remote.pushkey('phases',
1712 newremotehead.hex(),
1713 newremotehead.hex(),
1713 str(phases.draft),
1714 str(phases.draft),
1714 str(phases.public))
1715 str(phases.public))
1715 if not r:
1716 if not r:
1716 self.ui.warn(_('updating %s to public failed!\n')
1717 self.ui.warn(_('updating %s to public failed!\n')
1717 % newremotehead)
1718 % newremotehead)
1718 finally:
1719 finally:
1719 if lock is not None:
1720 if lock is not None:
1720 lock.release()
1721 lock.release()
1721 finally:
1722 finally:
1722 locallock.release()
1723 locallock.release()
1723
1724
1724 self.ui.debug("checking for updated bookmarks\n")
1725 self.ui.debug("checking for updated bookmarks\n")
1725 rb = remote.listkeys('bookmarks')
1726 rb = remote.listkeys('bookmarks')
1726 for k in rb.keys():
1727 for k in rb.keys():
1727 if k in self._bookmarks:
1728 if k in self._bookmarks:
1728 nr, nl = rb[k], hex(self._bookmarks[k])
1729 nr, nl = rb[k], hex(self._bookmarks[k])
1729 if nr in self:
1730 if nr in self:
1730 cr = self[nr]
1731 cr = self[nr]
1731 cl = self[nl]
1732 cl = self[nl]
1732 if cl in cr.descendants():
1733 if cl in cr.descendants():
1733 r = remote.pushkey('bookmarks', k, nr, nl)
1734 r = remote.pushkey('bookmarks', k, nr, nl)
1734 if r:
1735 if r:
1735 self.ui.status(_("updating bookmark %s\n") % k)
1736 self.ui.status(_("updating bookmark %s\n") % k)
1736 else:
1737 else:
1737 self.ui.warn(_('updating bookmark %s'
1738 self.ui.warn(_('updating bookmark %s'
1738 ' failed!\n') % k)
1739 ' failed!\n') % k)
1739
1740
1740 return ret
1741 return ret
1741
1742
1742 def changegroupinfo(self, nodes, source):
1743 def changegroupinfo(self, nodes, source):
1743 if self.ui.verbose or source == 'bundle':
1744 if self.ui.verbose or source == 'bundle':
1744 self.ui.status(_("%d changesets found\n") % len(nodes))
1745 self.ui.status(_("%d changesets found\n") % len(nodes))
1745 if self.ui.debugflag:
1746 if self.ui.debugflag:
1746 self.ui.debug("list of changesets:\n")
1747 self.ui.debug("list of changesets:\n")
1747 for node in nodes:
1748 for node in nodes:
1748 self.ui.debug("%s\n" % hex(node))
1749 self.ui.debug("%s\n" % hex(node))
1749
1750
1750 def changegroupsubset(self, bases, heads, source):
1751 def changegroupsubset(self, bases, heads, source):
1751 """Compute a changegroup consisting of all the nodes that are
1752 """Compute a changegroup consisting of all the nodes that are
1752 descendants of any of the bases and ancestors of any of the heads.
1753 descendants of any of the bases and ancestors of any of the heads.
1753 Return a chunkbuffer object whose read() method will return
1754 Return a chunkbuffer object whose read() method will return
1754 successive changegroup chunks.
1755 successive changegroup chunks.
1755
1756
1756 It is fairly complex as determining which filenodes and which
1757 It is fairly complex as determining which filenodes and which
1757 manifest nodes need to be included for the changeset to be complete
1758 manifest nodes need to be included for the changeset to be complete
1758 is non-trivial.
1759 is non-trivial.
1759
1760
1760 Another wrinkle is doing the reverse, figuring out which changeset in
1761 Another wrinkle is doing the reverse, figuring out which changeset in
1761 the changegroup a particular filenode or manifestnode belongs to.
1762 the changegroup a particular filenode or manifestnode belongs to.
1762 """
1763 """
1763 cl = self.changelog
1764 cl = self.changelog
1764 if not bases:
1765 if not bases:
1765 bases = [nullid]
1766 bases = [nullid]
1766 csets, bases, heads = cl.nodesbetween(bases, heads)
1767 csets, bases, heads = cl.nodesbetween(bases, heads)
1767 # We assume that all ancestors of bases are known
1768 # We assume that all ancestors of bases are known
1768 common = set(cl.ancestors(*[cl.rev(n) for n in bases]))
1769 common = set(cl.ancestors(*[cl.rev(n) for n in bases]))
1769 return self._changegroupsubset(common, csets, heads, source)
1770 return self._changegroupsubset(common, csets, heads, source)
1770
1771
1771 def getlocalbundle(self, source, outgoing):
1772 def getlocalbundle(self, source, outgoing):
1772 """Like getbundle, but taking a discovery.outgoing as an argument.
1773 """Like getbundle, but taking a discovery.outgoing as an argument.
1773
1774
1774 This is only implemented for local repos and reuses potentially
1775 This is only implemented for local repos and reuses potentially
1775 precomputed sets in outgoing."""
1776 precomputed sets in outgoing."""
1776 if not outgoing.missing:
1777 if not outgoing.missing:
1777 return None
1778 return None
1778 return self._changegroupsubset(outgoing.common,
1779 return self._changegroupsubset(outgoing.common,
1779 outgoing.missing,
1780 outgoing.missing,
1780 outgoing.missingheads,
1781 outgoing.missingheads,
1781 source)
1782 source)
1782
1783
1783 def getbundle(self, source, heads=None, common=None):
1784 def getbundle(self, source, heads=None, common=None):
1784 """Like changegroupsubset, but returns the set difference between the
1785 """Like changegroupsubset, but returns the set difference between the
1785 ancestors of heads and the ancestors common.
1786 ancestors of heads and the ancestors common.
1786
1787
1787 If heads is None, use the local heads. If common is None, use [nullid].
1788 If heads is None, use the local heads. If common is None, use [nullid].
1788
1789
1789 The nodes in common might not all be known locally due to the way the
1790 The nodes in common might not all be known locally due to the way the
1790 current discovery protocol works.
1791 current discovery protocol works.
1791 """
1792 """
1792 cl = self.changelog
1793 cl = self.changelog
1793 if common:
1794 if common:
1794 nm = cl.nodemap
1795 nm = cl.nodemap
1795 common = [n for n in common if n in nm]
1796 common = [n for n in common if n in nm]
1796 else:
1797 else:
1797 common = [nullid]
1798 common = [nullid]
1798 if not heads:
1799 if not heads:
1799 heads = cl.heads()
1800 heads = cl.heads()
1800 return self.getlocalbundle(source,
1801 return self.getlocalbundle(source,
1801 discovery.outgoing(cl, common, heads))
1802 discovery.outgoing(cl, common, heads))
1802
1803
1803 def _changegroupsubset(self, commonrevs, csets, heads, source):
1804 def _changegroupsubset(self, commonrevs, csets, heads, source):
1804
1805
1805 cl = self.changelog
1806 cl = self.changelog
1806 mf = self.manifest
1807 mf = self.manifest
1807 mfs = {} # needed manifests
1808 mfs = {} # needed manifests
1808 fnodes = {} # needed file nodes
1809 fnodes = {} # needed file nodes
1809 changedfiles = set()
1810 changedfiles = set()
1810 fstate = ['', {}]
1811 fstate = ['', {}]
1811 count = [0]
1812 count = [0]
1812
1813
1813 # can we go through the fast path ?
1814 # can we go through the fast path ?
1814 heads.sort()
1815 heads.sort()
1815 if heads == sorted(self.heads()):
1816 if heads == sorted(self.heads()):
1816 return self._changegroup(csets, source)
1817 return self._changegroup(csets, source)
1817
1818
1818 # slow path
1819 # slow path
1819 self.hook('preoutgoing', throw=True, source=source)
1820 self.hook('preoutgoing', throw=True, source=source)
1820 self.changegroupinfo(csets, source)
1821 self.changegroupinfo(csets, source)
1821
1822
1822 # filter any nodes that claim to be part of the known set
1823 # filter any nodes that claim to be part of the known set
1823 def prune(revlog, missing):
1824 def prune(revlog, missing):
1824 return [n for n in missing
1825 return [n for n in missing
1825 if revlog.linkrev(revlog.rev(n)) not in commonrevs]
1826 if revlog.linkrev(revlog.rev(n)) not in commonrevs]
1826
1827
1827 def lookup(revlog, x):
1828 def lookup(revlog, x):
1828 if revlog == cl:
1829 if revlog == cl:
1829 c = cl.read(x)
1830 c = cl.read(x)
1830 changedfiles.update(c[3])
1831 changedfiles.update(c[3])
1831 mfs.setdefault(c[0], x)
1832 mfs.setdefault(c[0], x)
1832 count[0] += 1
1833 count[0] += 1
1833 self.ui.progress(_('bundling'), count[0],
1834 self.ui.progress(_('bundling'), count[0],
1834 unit=_('changesets'), total=len(csets))
1835 unit=_('changesets'), total=len(csets))
1835 return x
1836 return x
1836 elif revlog == mf:
1837 elif revlog == mf:
1837 clnode = mfs[x]
1838 clnode = mfs[x]
1838 mdata = mf.readfast(x)
1839 mdata = mf.readfast(x)
1839 for f in changedfiles:
1840 for f in changedfiles:
1840 if f in mdata:
1841 if f in mdata:
1841 fnodes.setdefault(f, {}).setdefault(mdata[f], clnode)
1842 fnodes.setdefault(f, {}).setdefault(mdata[f], clnode)
1842 count[0] += 1
1843 count[0] += 1
1843 self.ui.progress(_('bundling'), count[0],
1844 self.ui.progress(_('bundling'), count[0],
1844 unit=_('manifests'), total=len(mfs))
1845 unit=_('manifests'), total=len(mfs))
1845 return mfs[x]
1846 return mfs[x]
1846 else:
1847 else:
1847 self.ui.progress(
1848 self.ui.progress(
1848 _('bundling'), count[0], item=fstate[0],
1849 _('bundling'), count[0], item=fstate[0],
1849 unit=_('files'), total=len(changedfiles))
1850 unit=_('files'), total=len(changedfiles))
1850 return fstate[1][x]
1851 return fstate[1][x]
1851
1852
1852 bundler = changegroup.bundle10(lookup)
1853 bundler = changegroup.bundle10(lookup)
1853 reorder = self.ui.config('bundle', 'reorder', 'auto')
1854 reorder = self.ui.config('bundle', 'reorder', 'auto')
1854 if reorder == 'auto':
1855 if reorder == 'auto':
1855 reorder = None
1856 reorder = None
1856 else:
1857 else:
1857 reorder = util.parsebool(reorder)
1858 reorder = util.parsebool(reorder)
1858
1859
1859 def gengroup():
1860 def gengroup():
1860 # Create a changenode group generator that will call our functions
1861 # Create a changenode group generator that will call our functions
1861 # back to lookup the owning changenode and collect information.
1862 # back to lookup the owning changenode and collect information.
1862 for chunk in cl.group(csets, bundler, reorder=reorder):
1863 for chunk in cl.group(csets, bundler, reorder=reorder):
1863 yield chunk
1864 yield chunk
1864 self.ui.progress(_('bundling'), None)
1865 self.ui.progress(_('bundling'), None)
1865
1866
1866 # Create a generator for the manifestnodes that calls our lookup
1867 # Create a generator for the manifestnodes that calls our lookup
1867 # and data collection functions back.
1868 # and data collection functions back.
1868 count[0] = 0
1869 count[0] = 0
1869 for chunk in mf.group(prune(mf, mfs), bundler, reorder=reorder):
1870 for chunk in mf.group(prune(mf, mfs), bundler, reorder=reorder):
1870 yield chunk
1871 yield chunk
1871 self.ui.progress(_('bundling'), None)
1872 self.ui.progress(_('bundling'), None)
1872
1873
1873 mfs.clear()
1874 mfs.clear()
1874
1875
1875 # Go through all our files in order sorted by name.
1876 # Go through all our files in order sorted by name.
1876 count[0] = 0
1877 count[0] = 0
1877 for fname in sorted(changedfiles):
1878 for fname in sorted(changedfiles):
1878 filerevlog = self.file(fname)
1879 filerevlog = self.file(fname)
1879 if not len(filerevlog):
1880 if not len(filerevlog):
1880 raise util.Abort(_("empty or missing revlog for %s") % fname)
1881 raise util.Abort(_("empty or missing revlog for %s") % fname)
1881 fstate[0] = fname
1882 fstate[0] = fname
1882 fstate[1] = fnodes.pop(fname, {})
1883 fstate[1] = fnodes.pop(fname, {})
1883
1884
1884 nodelist = prune(filerevlog, fstate[1])
1885 nodelist = prune(filerevlog, fstate[1])
1885 if nodelist:
1886 if nodelist:
1886 count[0] += 1
1887 count[0] += 1
1887 yield bundler.fileheader(fname)
1888 yield bundler.fileheader(fname)
1888 for chunk in filerevlog.group(nodelist, bundler, reorder):
1889 for chunk in filerevlog.group(nodelist, bundler, reorder):
1889 yield chunk
1890 yield chunk
1890
1891
1891 # Signal that no more groups are left.
1892 # Signal that no more groups are left.
1892 yield bundler.close()
1893 yield bundler.close()
1893 self.ui.progress(_('bundling'), None)
1894 self.ui.progress(_('bundling'), None)
1894
1895
1895 if csets:
1896 if csets:
1896 self.hook('outgoing', node=hex(csets[0]), source=source)
1897 self.hook('outgoing', node=hex(csets[0]), source=source)
1897
1898
1898 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1899 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1899
1900
1900 def changegroup(self, basenodes, source):
1901 def changegroup(self, basenodes, source):
1901 # to avoid a race we use changegroupsubset() (issue1320)
1902 # to avoid a race we use changegroupsubset() (issue1320)
1902 return self.changegroupsubset(basenodes, self.heads(), source)
1903 return self.changegroupsubset(basenodes, self.heads(), source)
1903
1904
1904 def _changegroup(self, nodes, source):
1905 def _changegroup(self, nodes, source):
1905 """Compute the changegroup of all nodes that we have that a recipient
1906 """Compute the changegroup of all nodes that we have that a recipient
1906 doesn't. Return a chunkbuffer object whose read() method will return
1907 doesn't. Return a chunkbuffer object whose read() method will return
1907 successive changegroup chunks.
1908 successive changegroup chunks.
1908
1909
1909 This is much easier than the previous function as we can assume that
1910 This is much easier than the previous function as we can assume that
1910 the recipient has any changenode we aren't sending them.
1911 the recipient has any changenode we aren't sending them.
1911
1912
1912 nodes is the set of nodes to send"""
1913 nodes is the set of nodes to send"""
1913
1914
1914 cl = self.changelog
1915 cl = self.changelog
1915 mf = self.manifest
1916 mf = self.manifest
1916 mfs = {}
1917 mfs = {}
1917 changedfiles = set()
1918 changedfiles = set()
1918 fstate = ['']
1919 fstate = ['']
1919 count = [0]
1920 count = [0]
1920
1921
1921 self.hook('preoutgoing', throw=True, source=source)
1922 self.hook('preoutgoing', throw=True, source=source)
1922 self.changegroupinfo(nodes, source)
1923 self.changegroupinfo(nodes, source)
1923
1924
1924 revset = set([cl.rev(n) for n in nodes])
1925 revset = set([cl.rev(n) for n in nodes])
1925
1926
1926 def gennodelst(log):
1927 def gennodelst(log):
1927 return [log.node(r) for r in log if log.linkrev(r) in revset]
1928 return [log.node(r) for r in log if log.linkrev(r) in revset]
1928
1929
1929 def lookup(revlog, x):
1930 def lookup(revlog, x):
1930 if revlog == cl:
1931 if revlog == cl:
1931 c = cl.read(x)
1932 c = cl.read(x)
1932 changedfiles.update(c[3])
1933 changedfiles.update(c[3])
1933 mfs.setdefault(c[0], x)
1934 mfs.setdefault(c[0], x)
1934 count[0] += 1
1935 count[0] += 1
1935 self.ui.progress(_('bundling'), count[0],
1936 self.ui.progress(_('bundling'), count[0],
1936 unit=_('changesets'), total=len(nodes))
1937 unit=_('changesets'), total=len(nodes))
1937 return x
1938 return x
1938 elif revlog == mf:
1939 elif revlog == mf:
1939 count[0] += 1
1940 count[0] += 1
1940 self.ui.progress(_('bundling'), count[0],
1941 self.ui.progress(_('bundling'), count[0],
1941 unit=_('manifests'), total=len(mfs))
1942 unit=_('manifests'), total=len(mfs))
1942 return cl.node(revlog.linkrev(revlog.rev(x)))
1943 return cl.node(revlog.linkrev(revlog.rev(x)))
1943 else:
1944 else:
1944 self.ui.progress(
1945 self.ui.progress(
1945 _('bundling'), count[0], item=fstate[0],
1946 _('bundling'), count[0], item=fstate[0],
1946 total=len(changedfiles), unit=_('files'))
1947 total=len(changedfiles), unit=_('files'))
1947 return cl.node(revlog.linkrev(revlog.rev(x)))
1948 return cl.node(revlog.linkrev(revlog.rev(x)))
1948
1949
1949 bundler = changegroup.bundle10(lookup)
1950 bundler = changegroup.bundle10(lookup)
1950 reorder = self.ui.config('bundle', 'reorder', 'auto')
1951 reorder = self.ui.config('bundle', 'reorder', 'auto')
1951 if reorder == 'auto':
1952 if reorder == 'auto':
1952 reorder = None
1953 reorder = None
1953 else:
1954 else:
1954 reorder = util.parsebool(reorder)
1955 reorder = util.parsebool(reorder)
1955
1956
1956 def gengroup():
1957 def gengroup():
1957 '''yield a sequence of changegroup chunks (strings)'''
1958 '''yield a sequence of changegroup chunks (strings)'''
1958 # construct a list of all changed files
1959 # construct a list of all changed files
1959
1960
1960 for chunk in cl.group(nodes, bundler, reorder=reorder):
1961 for chunk in cl.group(nodes, bundler, reorder=reorder):
1961 yield chunk
1962 yield chunk
1962 self.ui.progress(_('bundling'), None)
1963 self.ui.progress(_('bundling'), None)
1963
1964
1964 count[0] = 0
1965 count[0] = 0
1965 for chunk in mf.group(gennodelst(mf), bundler, reorder=reorder):
1966 for chunk in mf.group(gennodelst(mf), bundler, reorder=reorder):
1966 yield chunk
1967 yield chunk
1967 self.ui.progress(_('bundling'), None)
1968 self.ui.progress(_('bundling'), None)
1968
1969
1969 count[0] = 0
1970 count[0] = 0
1970 for fname in sorted(changedfiles):
1971 for fname in sorted(changedfiles):
1971 filerevlog = self.file(fname)
1972 filerevlog = self.file(fname)
1972 if not len(filerevlog):
1973 if not len(filerevlog):
1973 raise util.Abort(_("empty or missing revlog for %s") % fname)
1974 raise util.Abort(_("empty or missing revlog for %s") % fname)
1974 fstate[0] = fname
1975 fstate[0] = fname
1975 nodelist = gennodelst(filerevlog)
1976 nodelist = gennodelst(filerevlog)
1976 if nodelist:
1977 if nodelist:
1977 count[0] += 1
1978 count[0] += 1
1978 yield bundler.fileheader(fname)
1979 yield bundler.fileheader(fname)
1979 for chunk in filerevlog.group(nodelist, bundler, reorder):
1980 for chunk in filerevlog.group(nodelist, bundler, reorder):
1980 yield chunk
1981 yield chunk
1981 yield bundler.close()
1982 yield bundler.close()
1982 self.ui.progress(_('bundling'), None)
1983 self.ui.progress(_('bundling'), None)
1983
1984
1984 if nodes:
1985 if nodes:
1985 self.hook('outgoing', node=hex(nodes[0]), source=source)
1986 self.hook('outgoing', node=hex(nodes[0]), source=source)
1986
1987
1987 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1988 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
1988
1989
1989 def addchangegroup(self, source, srctype, url, emptyok=False):
1990 def addchangegroup(self, source, srctype, url, emptyok=False):
1990 """Add the changegroup returned by source.read() to this repo.
1991 """Add the changegroup returned by source.read() to this repo.
1991 srctype is a string like 'push', 'pull', or 'unbundle'. url is
1992 srctype is a string like 'push', 'pull', or 'unbundle'. url is
1992 the URL of the repo where this changegroup is coming from.
1993 the URL of the repo where this changegroup is coming from.
1993
1994
1994 Return an integer summarizing the change to this repo:
1995 Return an integer summarizing the change to this repo:
1995 - nothing changed or no source: 0
1996 - nothing changed or no source: 0
1996 - more heads than before: 1+added heads (2..n)
1997 - more heads than before: 1+added heads (2..n)
1997 - fewer heads than before: -1-removed heads (-2..-n)
1998 - fewer heads than before: -1-removed heads (-2..-n)
1998 - number of heads stays the same: 1
1999 - number of heads stays the same: 1
1999 """
2000 """
2000 def csmap(x):
2001 def csmap(x):
2001 self.ui.debug("add changeset %s\n" % short(x))
2002 self.ui.debug("add changeset %s\n" % short(x))
2002 return len(cl)
2003 return len(cl)
2003
2004
2004 def revmap(x):
2005 def revmap(x):
2005 return cl.rev(x)
2006 return cl.rev(x)
2006
2007
2007 if not source:
2008 if not source:
2008 return 0
2009 return 0
2009
2010
2010 self.hook('prechangegroup', throw=True, source=srctype, url=url)
2011 self.hook('prechangegroup', throw=True, source=srctype, url=url)
2011
2012
2012 changesets = files = revisions = 0
2013 changesets = files = revisions = 0
2013 efiles = set()
2014 efiles = set()
2014
2015
2015 # write changelog data to temp files so concurrent readers will not see
2016 # write changelog data to temp files so concurrent readers will not see
2016 # inconsistent view
2017 # inconsistent view
2017 cl = self.changelog
2018 cl = self.changelog
2018 cl.delayupdate()
2019 cl.delayupdate()
2019 oldheads = cl.heads()
2020 oldheads = cl.heads()
2020
2021
2021 tr = self.transaction("\n".join([srctype, util.hidepassword(url)]))
2022 tr = self.transaction("\n".join([srctype, util.hidepassword(url)]))
2022 try:
2023 try:
2023 trp = weakref.proxy(tr)
2024 trp = weakref.proxy(tr)
2024 # pull off the changeset group
2025 # pull off the changeset group
2025 self.ui.status(_("adding changesets\n"))
2026 self.ui.status(_("adding changesets\n"))
2026 clstart = len(cl)
2027 clstart = len(cl)
2027 class prog(object):
2028 class prog(object):
2028 step = _('changesets')
2029 step = _('changesets')
2029 count = 1
2030 count = 1
2030 ui = self.ui
2031 ui = self.ui
2031 total = None
2032 total = None
2032 def __call__(self):
2033 def __call__(self):
2033 self.ui.progress(self.step, self.count, unit=_('chunks'),
2034 self.ui.progress(self.step, self.count, unit=_('chunks'),
2034 total=self.total)
2035 total=self.total)
2035 self.count += 1
2036 self.count += 1
2036 pr = prog()
2037 pr = prog()
2037 source.callback = pr
2038 source.callback = pr
2038
2039
2039 source.changelogheader()
2040 source.changelogheader()
2040 srccontent = cl.addgroup(source, csmap, trp)
2041 srccontent = cl.addgroup(source, csmap, trp)
2041 if not (srccontent or emptyok):
2042 if not (srccontent or emptyok):
2042 raise util.Abort(_("received changelog group is empty"))
2043 raise util.Abort(_("received changelog group is empty"))
2043 clend = len(cl)
2044 clend = len(cl)
2044 changesets = clend - clstart
2045 changesets = clend - clstart
2045 for c in xrange(clstart, clend):
2046 for c in xrange(clstart, clend):
2046 efiles.update(self[c].files())
2047 efiles.update(self[c].files())
2047 efiles = len(efiles)
2048 efiles = len(efiles)
2048 self.ui.progress(_('changesets'), None)
2049 self.ui.progress(_('changesets'), None)
2049
2050
2050 # pull off the manifest group
2051 # pull off the manifest group
2051 self.ui.status(_("adding manifests\n"))
2052 self.ui.status(_("adding manifests\n"))
2052 pr.step = _('manifests')
2053 pr.step = _('manifests')
2053 pr.count = 1
2054 pr.count = 1
2054 pr.total = changesets # manifests <= changesets
2055 pr.total = changesets # manifests <= changesets
2055 # no need to check for empty manifest group here:
2056 # 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,
2057 # 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
2058 # no new manifest will be created and the manifest group will
2058 # be empty during the pull
2059 # be empty during the pull
2059 source.manifestheader()
2060 source.manifestheader()
2060 self.manifest.addgroup(source, revmap, trp)
2061 self.manifest.addgroup(source, revmap, trp)
2061 self.ui.progress(_('manifests'), None)
2062 self.ui.progress(_('manifests'), None)
2062
2063
2063 needfiles = {}
2064 needfiles = {}
2064 if self.ui.configbool('server', 'validate', default=False):
2065 if self.ui.configbool('server', 'validate', default=False):
2065 # validate incoming csets have their manifests
2066 # validate incoming csets have their manifests
2066 for cset in xrange(clstart, clend):
2067 for cset in xrange(clstart, clend):
2067 mfest = self.changelog.read(self.changelog.node(cset))[0]
2068 mfest = self.changelog.read(self.changelog.node(cset))[0]
2068 mfest = self.manifest.readdelta(mfest)
2069 mfest = self.manifest.readdelta(mfest)
2069 # store file nodes we must see
2070 # store file nodes we must see
2070 for f, n in mfest.iteritems():
2071 for f, n in mfest.iteritems():
2071 needfiles.setdefault(f, set()).add(n)
2072 needfiles.setdefault(f, set()).add(n)
2072
2073
2073 # process the files
2074 # process the files
2074 self.ui.status(_("adding file changes\n"))
2075 self.ui.status(_("adding file changes\n"))
2075 pr.step = _('files')
2076 pr.step = _('files')
2076 pr.count = 1
2077 pr.count = 1
2077 pr.total = efiles
2078 pr.total = efiles
2078 source.callback = None
2079 source.callback = None
2079
2080
2080 while True:
2081 while True:
2081 chunkdata = source.filelogheader()
2082 chunkdata = source.filelogheader()
2082 if not chunkdata:
2083 if not chunkdata:
2083 break
2084 break
2084 f = chunkdata["filename"]
2085 f = chunkdata["filename"]
2085 self.ui.debug("adding %s revisions\n" % f)
2086 self.ui.debug("adding %s revisions\n" % f)
2086 pr()
2087 pr()
2087 fl = self.file(f)
2088 fl = self.file(f)
2088 o = len(fl)
2089 o = len(fl)
2089 if not fl.addgroup(source, revmap, trp):
2090 if not fl.addgroup(source, revmap, trp):
2090 raise util.Abort(_("received file revlog group is empty"))
2091 raise util.Abort(_("received file revlog group is empty"))
2091 revisions += len(fl) - o
2092 revisions += len(fl) - o
2092 files += 1
2093 files += 1
2093 if f in needfiles:
2094 if f in needfiles:
2094 needs = needfiles[f]
2095 needs = needfiles[f]
2095 for new in xrange(o, len(fl)):
2096 for new in xrange(o, len(fl)):
2096 n = fl.node(new)
2097 n = fl.node(new)
2097 if n in needs:
2098 if n in needs:
2098 needs.remove(n)
2099 needs.remove(n)
2099 if not needs:
2100 if not needs:
2100 del needfiles[f]
2101 del needfiles[f]
2101 self.ui.progress(_('files'), None)
2102 self.ui.progress(_('files'), None)
2102
2103
2103 for f, needs in needfiles.iteritems():
2104 for f, needs in needfiles.iteritems():
2104 fl = self.file(f)
2105 fl = self.file(f)
2105 for n in needs:
2106 for n in needs:
2106 try:
2107 try:
2107 fl.rev(n)
2108 fl.rev(n)
2108 except error.LookupError:
2109 except error.LookupError:
2109 raise util.Abort(
2110 raise util.Abort(
2110 _('missing file data for %s:%s - run hg verify') %
2111 _('missing file data for %s:%s - run hg verify') %
2111 (f, hex(n)))
2112 (f, hex(n)))
2112
2113
2113 dh = 0
2114 dh = 0
2114 if oldheads:
2115 if oldheads:
2115 heads = cl.heads()
2116 heads = cl.heads()
2116 dh = len(heads) - len(oldheads)
2117 dh = len(heads) - len(oldheads)
2117 for h in heads:
2118 for h in heads:
2118 if h not in oldheads and 'close' in self[h].extra():
2119 if h not in oldheads and 'close' in self[h].extra():
2119 dh -= 1
2120 dh -= 1
2120 htext = ""
2121 htext = ""
2121 if dh:
2122 if dh:
2122 htext = _(" (%+d heads)") % dh
2123 htext = _(" (%+d heads)") % dh
2123
2124
2124 self.ui.status(_("added %d changesets"
2125 self.ui.status(_("added %d changesets"
2125 " with %d changes to %d files%s\n")
2126 " with %d changes to %d files%s\n")
2126 % (changesets, revisions, files, htext))
2127 % (changesets, revisions, files, htext))
2127
2128
2128 if changesets > 0:
2129 if changesets > 0:
2129 p = lambda: cl.writepending() and self.root or ""
2130 p = lambda: cl.writepending() and self.root or ""
2130 self.hook('pretxnchangegroup', throw=True,
2131 self.hook('pretxnchangegroup', throw=True,
2131 node=hex(cl.node(clstart)), source=srctype,
2132 node=hex(cl.node(clstart)), source=srctype,
2132 url=url, pending=p)
2133 url=url, pending=p)
2133
2134
2134 added = [cl.node(r) for r in xrange(clstart, clend)]
2135 added = [cl.node(r) for r in xrange(clstart, clend)]
2135 publishing = self.ui.configbool('phases', 'publish', True)
2136 publishing = self.ui.configbool('phases', 'publish', True)
2136 if srctype == 'push':
2137 if srctype == 'push':
2137 # Old server can not push the boundary themself.
2138 # Old server can not push the boundary themself.
2138 # New server won't push the boundary if changeset already
2139 # New server won't push the boundary if changeset already
2139 # existed locally as secrete
2140 # existed locally as secrete
2140 #
2141 #
2141 # We should not use added here but the list of all change in
2142 # We should not use added here but the list of all change in
2142 # the bundle
2143 # the bundle
2143 if publishing:
2144 if publishing:
2144 phases.advanceboundary(self, phases.public, srccontent)
2145 phases.advanceboundary(self, phases.public, srccontent)
2145 else:
2146 else:
2146 phases.advanceboundary(self, phases.draft, srccontent)
2147 phases.advanceboundary(self, phases.draft, srccontent)
2147 phases.retractboundary(self, phases.draft, added)
2148 phases.retractboundary(self, phases.draft, added)
2148 elif srctype != 'strip':
2149 elif srctype != 'strip':
2149 # publishing only alter behavior during push
2150 # publishing only alter behavior during push
2150 #
2151 #
2151 # strip should not touch boundary at all
2152 # strip should not touch boundary at all
2152 phases.retractboundary(self, phases.draft, added)
2153 phases.retractboundary(self, phases.draft, added)
2153
2154
2154 # make changelog see real files again
2155 # make changelog see real files again
2155 cl.finalize(trp)
2156 cl.finalize(trp)
2156
2157
2157 tr.close()
2158 tr.close()
2158
2159
2159 if changesets > 0:
2160 if changesets > 0:
2160 def runhooks():
2161 def runhooks():
2161 # forcefully update the on-disk branch cache
2162 # forcefully update the on-disk branch cache
2162 self.ui.debug("updating the branch cache\n")
2163 self.ui.debug("updating the branch cache\n")
2163 self.updatebranchcache()
2164 self.updatebranchcache()
2164 self.hook("changegroup", node=hex(cl.node(clstart)),
2165 self.hook("changegroup", node=hex(cl.node(clstart)),
2165 source=srctype, url=url)
2166 source=srctype, url=url)
2166
2167
2167 for n in added:
2168 for n in added:
2168 self.hook("incoming", node=hex(n), source=srctype,
2169 self.hook("incoming", node=hex(n), source=srctype,
2169 url=url)
2170 url=url)
2170 self._afterlock(runhooks)
2171 self._afterlock(runhooks)
2171
2172
2172 finally:
2173 finally:
2173 tr.release()
2174 tr.release()
2174 # never return 0 here:
2175 # never return 0 here:
2175 if dh < 0:
2176 if dh < 0:
2176 return dh - 1
2177 return dh - 1
2177 else:
2178 else:
2178 return dh + 1
2179 return dh + 1
2179
2180
2180 def stream_in(self, remote, requirements):
2181 def stream_in(self, remote, requirements):
2181 lock = self.lock()
2182 lock = self.lock()
2182 try:
2183 try:
2183 fp = remote.stream_out()
2184 fp = remote.stream_out()
2184 l = fp.readline()
2185 l = fp.readline()
2185 try:
2186 try:
2186 resp = int(l)
2187 resp = int(l)
2187 except ValueError:
2188 except ValueError:
2188 raise error.ResponseError(
2189 raise error.ResponseError(
2189 _('Unexpected response from remote server:'), l)
2190 _('Unexpected response from remote server:'), l)
2190 if resp == 1:
2191 if resp == 1:
2191 raise util.Abort(_('operation forbidden by server'))
2192 raise util.Abort(_('operation forbidden by server'))
2192 elif resp == 2:
2193 elif resp == 2:
2193 raise util.Abort(_('locking the remote repository failed'))
2194 raise util.Abort(_('locking the remote repository failed'))
2194 elif resp != 0:
2195 elif resp != 0:
2195 raise util.Abort(_('the server sent an unknown error code'))
2196 raise util.Abort(_('the server sent an unknown error code'))
2196 self.ui.status(_('streaming all changes\n'))
2197 self.ui.status(_('streaming all changes\n'))
2197 l = fp.readline()
2198 l = fp.readline()
2198 try:
2199 try:
2199 total_files, total_bytes = map(int, l.split(' ', 1))
2200 total_files, total_bytes = map(int, l.split(' ', 1))
2200 except (ValueError, TypeError):
2201 except (ValueError, TypeError):
2201 raise error.ResponseError(
2202 raise error.ResponseError(
2202 _('Unexpected response from remote server:'), l)
2203 _('Unexpected response from remote server:'), l)
2203 self.ui.status(_('%d files to transfer, %s of data\n') %
2204 self.ui.status(_('%d files to transfer, %s of data\n') %
2204 (total_files, util.bytecount(total_bytes)))
2205 (total_files, util.bytecount(total_bytes)))
2205 start = time.time()
2206 start = time.time()
2206 for i in xrange(total_files):
2207 for i in xrange(total_files):
2207 # XXX doesn't support '\n' or '\r' in filenames
2208 # XXX doesn't support '\n' or '\r' in filenames
2208 l = fp.readline()
2209 l = fp.readline()
2209 try:
2210 try:
2210 name, size = l.split('\0', 1)
2211 name, size = l.split('\0', 1)
2211 size = int(size)
2212 size = int(size)
2212 except (ValueError, TypeError):
2213 except (ValueError, TypeError):
2213 raise error.ResponseError(
2214 raise error.ResponseError(
2214 _('Unexpected response from remote server:'), l)
2215 _('Unexpected response from remote server:'), l)
2215 self.ui.debug('adding %s (%s)\n' % (name, util.bytecount(size)))
2216 self.ui.debug('adding %s (%s)\n' % (name, util.bytecount(size)))
2216 # for backwards compat, name was partially encoded
2217 # for backwards compat, name was partially encoded
2217 ofp = self.sopener(store.decodedir(name), 'w')
2218 ofp = self.sopener(store.decodedir(name), 'w')
2218 for chunk in util.filechunkiter(fp, limit=size):
2219 for chunk in util.filechunkiter(fp, limit=size):
2219 ofp.write(chunk)
2220 ofp.write(chunk)
2220 ofp.close()
2221 ofp.close()
2221 elapsed = time.time() - start
2222 elapsed = time.time() - start
2222 if elapsed <= 0:
2223 if elapsed <= 0:
2223 elapsed = 0.001
2224 elapsed = 0.001
2224 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
2225 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
2225 (util.bytecount(total_bytes), elapsed,
2226 (util.bytecount(total_bytes), elapsed,
2226 util.bytecount(total_bytes / elapsed)))
2227 util.bytecount(total_bytes / elapsed)))
2227
2228
2228 # new requirements = old non-format requirements + new format-related
2229 # new requirements = old non-format requirements + new format-related
2229 # requirements from the streamed-in repository
2230 # requirements from the streamed-in repository
2230 requirements.update(set(self.requirements) - self.supportedformats)
2231 requirements.update(set(self.requirements) - self.supportedformats)
2231 self._applyrequirements(requirements)
2232 self._applyrequirements(requirements)
2232 self._writerequirements()
2233 self._writerequirements()
2233
2234
2234 self.invalidate()
2235 self.invalidate()
2235 return len(self.heads()) + 1
2236 return len(self.heads()) + 1
2236 finally:
2237 finally:
2237 lock.release()
2238 lock.release()
2238
2239
2239 def clone(self, remote, heads=[], stream=False):
2240 def clone(self, remote, heads=[], stream=False):
2240 '''clone remote repository.
2241 '''clone remote repository.
2241
2242
2242 keyword arguments:
2243 keyword arguments:
2243 heads: list of revs to clone (forces use of pull)
2244 heads: list of revs to clone (forces use of pull)
2244 stream: use streaming clone if possible'''
2245 stream: use streaming clone if possible'''
2245
2246
2246 # now, all clients that can request uncompressed clones can
2247 # now, all clients that can request uncompressed clones can
2247 # read repo formats supported by all servers that can serve
2248 # read repo formats supported by all servers that can serve
2248 # them.
2249 # them.
2249
2250
2250 # if revlog format changes, client will have to check version
2251 # if revlog format changes, client will have to check version
2251 # and format flags on "stream" capability, and use
2252 # and format flags on "stream" capability, and use
2252 # uncompressed only if compatible.
2253 # uncompressed only if compatible.
2253
2254
2254 if stream and not heads:
2255 if stream and not heads:
2255 # 'stream' means remote revlog format is revlogv1 only
2256 # 'stream' means remote revlog format is revlogv1 only
2256 if remote.capable('stream'):
2257 if remote.capable('stream'):
2257 return self.stream_in(remote, set(('revlogv1',)))
2258 return self.stream_in(remote, set(('revlogv1',)))
2258 # otherwise, 'streamreqs' contains the remote revlog format
2259 # otherwise, 'streamreqs' contains the remote revlog format
2259 streamreqs = remote.capable('streamreqs')
2260 streamreqs = remote.capable('streamreqs')
2260 if streamreqs:
2261 if streamreqs:
2261 streamreqs = set(streamreqs.split(','))
2262 streamreqs = set(streamreqs.split(','))
2262 # if we support it, stream in and adjust our requirements
2263 # if we support it, stream in and adjust our requirements
2263 if not streamreqs - self.supportedformats:
2264 if not streamreqs - self.supportedformats:
2264 return self.stream_in(remote, streamreqs)
2265 return self.stream_in(remote, streamreqs)
2265 return self.pull(remote, heads)
2266 return self.pull(remote, heads)
2266
2267
2267 def pushkey(self, namespace, key, old, new):
2268 def pushkey(self, namespace, key, old, new):
2268 self.hook('prepushkey', throw=True, namespace=namespace, key=key,
2269 self.hook('prepushkey', throw=True, namespace=namespace, key=key,
2269 old=old, new=new)
2270 old=old, new=new)
2270 ret = pushkey.push(self, namespace, key, old, new)
2271 ret = pushkey.push(self, namespace, key, old, new)
2271 self.hook('pushkey', namespace=namespace, key=key, old=old, new=new,
2272 self.hook('pushkey', namespace=namespace, key=key, old=old, new=new,
2272 ret=ret)
2273 ret=ret)
2273 return ret
2274 return ret
2274
2275
2275 def listkeys(self, namespace):
2276 def listkeys(self, namespace):
2276 self.hook('prelistkeys', throw=True, namespace=namespace)
2277 self.hook('prelistkeys', throw=True, namespace=namespace)
2277 values = pushkey.list(self, namespace)
2278 values = pushkey.list(self, namespace)
2278 self.hook('listkeys', namespace=namespace, values=values)
2279 self.hook('listkeys', namespace=namespace, values=values)
2279 return values
2280 return values
2280
2281
2281 def debugwireargs(self, one, two, three=None, four=None, five=None):
2282 def debugwireargs(self, one, two, three=None, four=None, five=None):
2282 '''used to test argument passing over the wire'''
2283 '''used to test argument passing over the wire'''
2283 return "%s %s %s %s %s" % (one, two, three, four, five)
2284 return "%s %s %s %s %s" % (one, two, three, four, five)
2284
2285
2285 def savecommitmessage(self, text):
2286 def savecommitmessage(self, text):
2286 fp = self.opener('last-message.txt', 'wb')
2287 fp = self.opener('last-message.txt', 'wb')
2287 try:
2288 try:
2288 fp.write(text)
2289 fp.write(text)
2289 finally:
2290 finally:
2290 fp.close()
2291 fp.close()
2291 return self.pathto(fp.name[len(self.root)+1:])
2292 return self.pathto(fp.name[len(self.root)+1:])
2292
2293
2293 # used to avoid circular references so destructors work
2294 # used to avoid circular references so destructors work
2294 def aftertrans(files):
2295 def aftertrans(files):
2295 renamefiles = [tuple(t) for t in files]
2296 renamefiles = [tuple(t) for t in files]
2296 def a():
2297 def a():
2297 for src, dest in renamefiles:
2298 for src, dest in renamefiles:
2298 util.rename(src, dest)
2299 util.rename(src, dest)
2299 return a
2300 return a
2300
2301
2301 def undoname(fn):
2302 def undoname(fn):
2302 base, name = os.path.split(fn)
2303 base, name = os.path.split(fn)
2303 assert name.startswith('journal')
2304 assert name.startswith('journal')
2304 return os.path.join(base, name.replace('journal', 'undo', 1))
2305 return os.path.join(base, name.replace('journal', 'undo', 1))
2305
2306
2306 def instance(ui, path, create):
2307 def instance(ui, path, create):
2307 return localrepository(ui, util.urllocalpath(path), create)
2308 return localrepository(ui, util.urllocalpath(path), create)
2308
2309
2309 def islocal(path):
2310 def islocal(path):
2310 return True
2311 return True
General Comments 0
You need to be logged in to leave comments. Login now