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