##// END OF EJS Templates
blackbox: only show new heads on incoming...
Durham Goode -
r18677:539210ed default
parent child Browse files
Show More
@@ -1,2591 +1,2591 b''
1 # localrepo.py - read/write repository class for mercurial
1 # localrepo.py - read/write repository class for mercurial
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7 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 a repo has an unfilteredpropertycache value for <name>"""
52 """check if a repo has an unfilteredpropertycache 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('served')
69 self._repo = repo.filtered('served')
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 sorted(self.requirements):
265 for r in sorted(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 overwritten by filtered repo."""
313 Intended to be overwritten 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 # (possibly subclass of the repo)
319 # (possibly 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 self.destroying()
898 self.destroying()
899 transaction.rollback(self.sopener, self.sjoin('undo'), ui.warn)
899 transaction.rollback(self.sopener, self.sjoin('undo'), ui.warn)
900 if os.path.exists(self.join('undo.bookmarks')):
900 if os.path.exists(self.join('undo.bookmarks')):
901 util.rename(self.join('undo.bookmarks'),
901 util.rename(self.join('undo.bookmarks'),
902 self.join('bookmarks'))
902 self.join('bookmarks'))
903 if os.path.exists(self.sjoin('undo.phaseroots')):
903 if os.path.exists(self.sjoin('undo.phaseroots')):
904 util.rename(self.sjoin('undo.phaseroots'),
904 util.rename(self.sjoin('undo.phaseroots'),
905 self.sjoin('phaseroots'))
905 self.sjoin('phaseroots'))
906 self.invalidate()
906 self.invalidate()
907
907
908 parentgone = (parents[0] not in self.changelog.nodemap or
908 parentgone = (parents[0] not in self.changelog.nodemap or
909 parents[1] not in self.changelog.nodemap)
909 parents[1] not in self.changelog.nodemap)
910 if parentgone:
910 if parentgone:
911 util.rename(self.join('undo.dirstate'), self.join('dirstate'))
911 util.rename(self.join('undo.dirstate'), self.join('dirstate'))
912 try:
912 try:
913 branch = self.opener.read('undo.branch')
913 branch = self.opener.read('undo.branch')
914 self.dirstate.setbranch(encoding.tolocal(branch))
914 self.dirstate.setbranch(encoding.tolocal(branch))
915 except IOError:
915 except IOError:
916 ui.warn(_('named branch could not be reset: '
916 ui.warn(_('named branch could not be reset: '
917 'current branch is still \'%s\'\n')
917 'current branch is still \'%s\'\n')
918 % self.dirstate.branch())
918 % self.dirstate.branch())
919
919
920 self.dirstate.invalidate()
920 self.dirstate.invalidate()
921 parents = tuple([p.rev() for p in self.parents()])
921 parents = tuple([p.rev() for p in self.parents()])
922 if len(parents) > 1:
922 if len(parents) > 1:
923 ui.status(_('working directory now based on '
923 ui.status(_('working directory now based on '
924 'revisions %d and %d\n') % parents)
924 'revisions %d and %d\n') % parents)
925 else:
925 else:
926 ui.status(_('working directory now based on '
926 ui.status(_('working directory now based on '
927 'revision %d\n') % parents)
927 'revision %d\n') % parents)
928 # TODO: if we know which new heads may result from this rollback, pass
928 # TODO: if we know which new heads may result from this rollback, pass
929 # them to destroy(), which will prevent the branchhead cache from being
929 # them to destroy(), which will prevent the branchhead cache from being
930 # invalidated.
930 # invalidated.
931 self.destroyed()
931 self.destroyed()
932 return 0
932 return 0
933
933
934 def invalidatecaches(self):
934 def invalidatecaches(self):
935
935
936 if '_tagscache' in vars(self):
936 if '_tagscache' in vars(self):
937 # can't use delattr on proxy
937 # can't use delattr on proxy
938 del self.__dict__['_tagscache']
938 del self.__dict__['_tagscache']
939
939
940 self.unfiltered()._branchcaches.clear()
940 self.unfiltered()._branchcaches.clear()
941 self.invalidatevolatilesets()
941 self.invalidatevolatilesets()
942
942
943 def invalidatevolatilesets(self):
943 def invalidatevolatilesets(self):
944 self.filteredrevcache.clear()
944 self.filteredrevcache.clear()
945 obsolete.clearobscaches(self)
945 obsolete.clearobscaches(self)
946
946
947 def invalidatedirstate(self):
947 def invalidatedirstate(self):
948 '''Invalidates the dirstate, causing the next call to dirstate
948 '''Invalidates the dirstate, causing the next call to dirstate
949 to check if it was modified since the last time it was read,
949 to check if it was modified since the last time it was read,
950 rereading it if it has.
950 rereading it if it has.
951
951
952 This is different to dirstate.invalidate() that it doesn't always
952 This is different to dirstate.invalidate() that it doesn't always
953 rereads the dirstate. Use dirstate.invalidate() if you want to
953 rereads the dirstate. Use dirstate.invalidate() if you want to
954 explicitly read the dirstate again (i.e. restoring it to a previous
954 explicitly read the dirstate again (i.e. restoring it to a previous
955 known good state).'''
955 known good state).'''
956 if hasunfilteredcache(self, 'dirstate'):
956 if hasunfilteredcache(self, 'dirstate'):
957 for k in self.dirstate._filecache:
957 for k in self.dirstate._filecache:
958 try:
958 try:
959 delattr(self.dirstate, k)
959 delattr(self.dirstate, k)
960 except AttributeError:
960 except AttributeError:
961 pass
961 pass
962 delattr(self.unfiltered(), 'dirstate')
962 delattr(self.unfiltered(), 'dirstate')
963
963
964 def invalidate(self):
964 def invalidate(self):
965 unfiltered = self.unfiltered() # all file caches are stored unfiltered
965 unfiltered = self.unfiltered() # all file caches are stored unfiltered
966 for k in self._filecache:
966 for k in self._filecache:
967 # dirstate is invalidated separately in invalidatedirstate()
967 # dirstate is invalidated separately in invalidatedirstate()
968 if k == 'dirstate':
968 if k == 'dirstate':
969 continue
969 continue
970
970
971 try:
971 try:
972 delattr(unfiltered, k)
972 delattr(unfiltered, k)
973 except AttributeError:
973 except AttributeError:
974 pass
974 pass
975 self.invalidatecaches()
975 self.invalidatecaches()
976
976
977 def _lock(self, lockname, wait, releasefn, acquirefn, desc):
977 def _lock(self, lockname, wait, releasefn, acquirefn, desc):
978 try:
978 try:
979 l = lock.lock(lockname, 0, releasefn, desc=desc)
979 l = lock.lock(lockname, 0, releasefn, desc=desc)
980 except error.LockHeld, inst:
980 except error.LockHeld, inst:
981 if not wait:
981 if not wait:
982 raise
982 raise
983 self.ui.warn(_("waiting for lock on %s held by %r\n") %
983 self.ui.warn(_("waiting for lock on %s held by %r\n") %
984 (desc, inst.locker))
984 (desc, inst.locker))
985 # default to 600 seconds timeout
985 # default to 600 seconds timeout
986 l = lock.lock(lockname, int(self.ui.config("ui", "timeout", "600")),
986 l = lock.lock(lockname, int(self.ui.config("ui", "timeout", "600")),
987 releasefn, desc=desc)
987 releasefn, desc=desc)
988 if acquirefn:
988 if acquirefn:
989 acquirefn()
989 acquirefn()
990 return l
990 return l
991
991
992 def _afterlock(self, callback):
992 def _afterlock(self, callback):
993 """add a callback to the current repository lock.
993 """add a callback to the current repository lock.
994
994
995 The callback will be executed on lock release."""
995 The callback will be executed on lock release."""
996 l = self._lockref and self._lockref()
996 l = self._lockref and self._lockref()
997 if l:
997 if l:
998 l.postrelease.append(callback)
998 l.postrelease.append(callback)
999 else:
999 else:
1000 callback()
1000 callback()
1001
1001
1002 def lock(self, wait=True):
1002 def lock(self, wait=True):
1003 '''Lock the repository store (.hg/store) and return a weak reference
1003 '''Lock the repository store (.hg/store) and return a weak reference
1004 to the lock. Use this before modifying the store (e.g. committing or
1004 to the lock. Use this before modifying the store (e.g. committing or
1005 stripping). If you are opening a transaction, get a lock as well.)'''
1005 stripping). If you are opening a transaction, get a lock as well.)'''
1006 l = self._lockref and self._lockref()
1006 l = self._lockref and self._lockref()
1007 if l is not None and l.held:
1007 if l is not None and l.held:
1008 l.lock()
1008 l.lock()
1009 return l
1009 return l
1010
1010
1011 def unlock():
1011 def unlock():
1012 self.store.write()
1012 self.store.write()
1013 if hasunfilteredcache(self, '_phasecache'):
1013 if hasunfilteredcache(self, '_phasecache'):
1014 self._phasecache.write()
1014 self._phasecache.write()
1015 for k, ce in self._filecache.items():
1015 for k, ce in self._filecache.items():
1016 if k == 'dirstate' or k not in self.__dict__:
1016 if k == 'dirstate' or k not in self.__dict__:
1017 continue
1017 continue
1018 ce.refresh()
1018 ce.refresh()
1019
1019
1020 l = self._lock(self.sjoin("lock"), wait, unlock,
1020 l = self._lock(self.sjoin("lock"), wait, unlock,
1021 self.invalidate, _('repository %s') % self.origroot)
1021 self.invalidate, _('repository %s') % self.origroot)
1022 self._lockref = weakref.ref(l)
1022 self._lockref = weakref.ref(l)
1023 return l
1023 return l
1024
1024
1025 def wlock(self, wait=True):
1025 def wlock(self, wait=True):
1026 '''Lock the non-store parts of the repository (everything under
1026 '''Lock the non-store parts of the repository (everything under
1027 .hg except .hg/store) and return a weak reference to the lock.
1027 .hg except .hg/store) and return a weak reference to the lock.
1028 Use this before modifying files in .hg.'''
1028 Use this before modifying files in .hg.'''
1029 l = self._wlockref and self._wlockref()
1029 l = self._wlockref and self._wlockref()
1030 if l is not None and l.held:
1030 if l is not None and l.held:
1031 l.lock()
1031 l.lock()
1032 return l
1032 return l
1033
1033
1034 def unlock():
1034 def unlock():
1035 self.dirstate.write()
1035 self.dirstate.write()
1036 self._filecache['dirstate'].refresh()
1036 self._filecache['dirstate'].refresh()
1037
1037
1038 l = self._lock(self.join("wlock"), wait, unlock,
1038 l = self._lock(self.join("wlock"), wait, unlock,
1039 self.invalidatedirstate, _('working directory of %s') %
1039 self.invalidatedirstate, _('working directory of %s') %
1040 self.origroot)
1040 self.origroot)
1041 self._wlockref = weakref.ref(l)
1041 self._wlockref = weakref.ref(l)
1042 return l
1042 return l
1043
1043
1044 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
1044 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
1045 """
1045 """
1046 commit an individual file as part of a larger transaction
1046 commit an individual file as part of a larger transaction
1047 """
1047 """
1048
1048
1049 fname = fctx.path()
1049 fname = fctx.path()
1050 text = fctx.data()
1050 text = fctx.data()
1051 flog = self.file(fname)
1051 flog = self.file(fname)
1052 fparent1 = manifest1.get(fname, nullid)
1052 fparent1 = manifest1.get(fname, nullid)
1053 fparent2 = fparent2o = manifest2.get(fname, nullid)
1053 fparent2 = fparent2o = manifest2.get(fname, nullid)
1054
1054
1055 meta = {}
1055 meta = {}
1056 copy = fctx.renamed()
1056 copy = fctx.renamed()
1057 if copy and copy[0] != fname:
1057 if copy and copy[0] != fname:
1058 # Mark the new revision of this file as a copy of another
1058 # Mark the new revision of this file as a copy of another
1059 # file. This copy data will effectively act as a parent
1059 # file. This copy data will effectively act as a parent
1060 # of this new revision. If this is a merge, the first
1060 # of this new revision. If this is a merge, the first
1061 # parent will be the nullid (meaning "look up the copy data")
1061 # parent will be the nullid (meaning "look up the copy data")
1062 # and the second one will be the other parent. For example:
1062 # and the second one will be the other parent. For example:
1063 #
1063 #
1064 # 0 --- 1 --- 3 rev1 changes file foo
1064 # 0 --- 1 --- 3 rev1 changes file foo
1065 # \ / rev2 renames foo to bar and changes it
1065 # \ / rev2 renames foo to bar and changes it
1066 # \- 2 -/ rev3 should have bar with all changes and
1066 # \- 2 -/ rev3 should have bar with all changes and
1067 # should record that bar descends from
1067 # should record that bar descends from
1068 # bar in rev2 and foo in rev1
1068 # bar in rev2 and foo in rev1
1069 #
1069 #
1070 # this allows this merge to succeed:
1070 # this allows this merge to succeed:
1071 #
1071 #
1072 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
1072 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
1073 # \ / merging rev3 and rev4 should use bar@rev2
1073 # \ / merging rev3 and rev4 should use bar@rev2
1074 # \- 2 --- 4 as the merge base
1074 # \- 2 --- 4 as the merge base
1075 #
1075 #
1076
1076
1077 cfname = copy[0]
1077 cfname = copy[0]
1078 crev = manifest1.get(cfname)
1078 crev = manifest1.get(cfname)
1079 newfparent = fparent2
1079 newfparent = fparent2
1080
1080
1081 if manifest2: # branch merge
1081 if manifest2: # branch merge
1082 if fparent2 == nullid or crev is None: # copied on remote side
1082 if fparent2 == nullid or crev is None: # copied on remote side
1083 if cfname in manifest2:
1083 if cfname in manifest2:
1084 crev = manifest2[cfname]
1084 crev = manifest2[cfname]
1085 newfparent = fparent1
1085 newfparent = fparent1
1086
1086
1087 # find source in nearest ancestor if we've lost track
1087 # find source in nearest ancestor if we've lost track
1088 if not crev:
1088 if not crev:
1089 self.ui.debug(" %s: searching for copy revision for %s\n" %
1089 self.ui.debug(" %s: searching for copy revision for %s\n" %
1090 (fname, cfname))
1090 (fname, cfname))
1091 for ancestor in self[None].ancestors():
1091 for ancestor in self[None].ancestors():
1092 if cfname in ancestor:
1092 if cfname in ancestor:
1093 crev = ancestor[cfname].filenode()
1093 crev = ancestor[cfname].filenode()
1094 break
1094 break
1095
1095
1096 if crev:
1096 if crev:
1097 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
1097 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
1098 meta["copy"] = cfname
1098 meta["copy"] = cfname
1099 meta["copyrev"] = hex(crev)
1099 meta["copyrev"] = hex(crev)
1100 fparent1, fparent2 = nullid, newfparent
1100 fparent1, fparent2 = nullid, newfparent
1101 else:
1101 else:
1102 self.ui.warn(_("warning: can't find ancestor for '%s' "
1102 self.ui.warn(_("warning: can't find ancestor for '%s' "
1103 "copied from '%s'!\n") % (fname, cfname))
1103 "copied from '%s'!\n") % (fname, cfname))
1104
1104
1105 elif fparent2 != nullid:
1105 elif fparent2 != nullid:
1106 # is one parent an ancestor of the other?
1106 # is one parent an ancestor of the other?
1107 fparentancestor = flog.ancestor(fparent1, fparent2)
1107 fparentancestor = flog.ancestor(fparent1, fparent2)
1108 if fparentancestor == fparent1:
1108 if fparentancestor == fparent1:
1109 fparent1, fparent2 = fparent2, nullid
1109 fparent1, fparent2 = fparent2, nullid
1110 elif fparentancestor == fparent2:
1110 elif fparentancestor == fparent2:
1111 fparent2 = nullid
1111 fparent2 = nullid
1112
1112
1113 # is the file changed?
1113 # is the file changed?
1114 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
1114 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
1115 changelist.append(fname)
1115 changelist.append(fname)
1116 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
1116 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
1117
1117
1118 # are just the flags changed during merge?
1118 # are just the flags changed during merge?
1119 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
1119 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
1120 changelist.append(fname)
1120 changelist.append(fname)
1121
1121
1122 return fparent1
1122 return fparent1
1123
1123
1124 @unfilteredmethod
1124 @unfilteredmethod
1125 def commit(self, text="", user=None, date=None, match=None, force=False,
1125 def commit(self, text="", user=None, date=None, match=None, force=False,
1126 editor=False, extra={}):
1126 editor=False, extra={}):
1127 """Add a new revision to current repository.
1127 """Add a new revision to current repository.
1128
1128
1129 Revision information is gathered from the working directory,
1129 Revision information is gathered from the working directory,
1130 match can be used to filter the committed files. If editor is
1130 match can be used to filter the committed files. If editor is
1131 supplied, it is called to get a commit message.
1131 supplied, it is called to get a commit message.
1132 """
1132 """
1133
1133
1134 def fail(f, msg):
1134 def fail(f, msg):
1135 raise util.Abort('%s: %s' % (f, msg))
1135 raise util.Abort('%s: %s' % (f, msg))
1136
1136
1137 if not match:
1137 if not match:
1138 match = matchmod.always(self.root, '')
1138 match = matchmod.always(self.root, '')
1139
1139
1140 if not force:
1140 if not force:
1141 vdirs = []
1141 vdirs = []
1142 match.dir = vdirs.append
1142 match.dir = vdirs.append
1143 match.bad = fail
1143 match.bad = fail
1144
1144
1145 wlock = self.wlock()
1145 wlock = self.wlock()
1146 try:
1146 try:
1147 wctx = self[None]
1147 wctx = self[None]
1148 merge = len(wctx.parents()) > 1
1148 merge = len(wctx.parents()) > 1
1149
1149
1150 if (not force and merge and match and
1150 if (not force and merge and match and
1151 (match.files() or match.anypats())):
1151 (match.files() or match.anypats())):
1152 raise util.Abort(_('cannot partially commit a merge '
1152 raise util.Abort(_('cannot partially commit a merge '
1153 '(do not specify files or patterns)'))
1153 '(do not specify files or patterns)'))
1154
1154
1155 changes = self.status(match=match, clean=force)
1155 changes = self.status(match=match, clean=force)
1156 if force:
1156 if force:
1157 changes[0].extend(changes[6]) # mq may commit unchanged files
1157 changes[0].extend(changes[6]) # mq may commit unchanged files
1158
1158
1159 # check subrepos
1159 # check subrepos
1160 subs = []
1160 subs = []
1161 commitsubs = set()
1161 commitsubs = set()
1162 newstate = wctx.substate.copy()
1162 newstate = wctx.substate.copy()
1163 # only manage subrepos and .hgsubstate if .hgsub is present
1163 # only manage subrepos and .hgsubstate if .hgsub is present
1164 if '.hgsub' in wctx:
1164 if '.hgsub' in wctx:
1165 # we'll decide whether to track this ourselves, thanks
1165 # we'll decide whether to track this ourselves, thanks
1166 if '.hgsubstate' in changes[0]:
1166 if '.hgsubstate' in changes[0]:
1167 changes[0].remove('.hgsubstate')
1167 changes[0].remove('.hgsubstate')
1168 if '.hgsubstate' in changes[2]:
1168 if '.hgsubstate' in changes[2]:
1169 changes[2].remove('.hgsubstate')
1169 changes[2].remove('.hgsubstate')
1170
1170
1171 # compare current state to last committed state
1171 # compare current state to last committed state
1172 # build new substate based on last committed state
1172 # build new substate based on last committed state
1173 oldstate = wctx.p1().substate
1173 oldstate = wctx.p1().substate
1174 for s in sorted(newstate.keys()):
1174 for s in sorted(newstate.keys()):
1175 if not match(s):
1175 if not match(s):
1176 # ignore working copy, use old state if present
1176 # ignore working copy, use old state if present
1177 if s in oldstate:
1177 if s in oldstate:
1178 newstate[s] = oldstate[s]
1178 newstate[s] = oldstate[s]
1179 continue
1179 continue
1180 if not force:
1180 if not force:
1181 raise util.Abort(
1181 raise util.Abort(
1182 _("commit with new subrepo %s excluded") % s)
1182 _("commit with new subrepo %s excluded") % s)
1183 if wctx.sub(s).dirty(True):
1183 if wctx.sub(s).dirty(True):
1184 if not self.ui.configbool('ui', 'commitsubrepos'):
1184 if not self.ui.configbool('ui', 'commitsubrepos'):
1185 raise util.Abort(
1185 raise util.Abort(
1186 _("uncommitted changes in subrepo %s") % s,
1186 _("uncommitted changes in subrepo %s") % s,
1187 hint=_("use --subrepos for recursive commit"))
1187 hint=_("use --subrepos for recursive commit"))
1188 subs.append(s)
1188 subs.append(s)
1189 commitsubs.add(s)
1189 commitsubs.add(s)
1190 else:
1190 else:
1191 bs = wctx.sub(s).basestate()
1191 bs = wctx.sub(s).basestate()
1192 newstate[s] = (newstate[s][0], bs, newstate[s][2])
1192 newstate[s] = (newstate[s][0], bs, newstate[s][2])
1193 if oldstate.get(s, (None, None, None))[1] != bs:
1193 if oldstate.get(s, (None, None, None))[1] != bs:
1194 subs.append(s)
1194 subs.append(s)
1195
1195
1196 # check for removed subrepos
1196 # check for removed subrepos
1197 for p in wctx.parents():
1197 for p in wctx.parents():
1198 r = [s for s in p.substate if s not in newstate]
1198 r = [s for s in p.substate if s not in newstate]
1199 subs += [s for s in r if match(s)]
1199 subs += [s for s in r if match(s)]
1200 if subs:
1200 if subs:
1201 if (not match('.hgsub') and
1201 if (not match('.hgsub') and
1202 '.hgsub' in (wctx.modified() + wctx.added())):
1202 '.hgsub' in (wctx.modified() + wctx.added())):
1203 raise util.Abort(
1203 raise util.Abort(
1204 _("can't commit subrepos without .hgsub"))
1204 _("can't commit subrepos without .hgsub"))
1205 changes[0].insert(0, '.hgsubstate')
1205 changes[0].insert(0, '.hgsubstate')
1206
1206
1207 elif '.hgsub' in changes[2]:
1207 elif '.hgsub' in changes[2]:
1208 # clean up .hgsubstate when .hgsub is removed
1208 # clean up .hgsubstate when .hgsub is removed
1209 if ('.hgsubstate' in wctx and
1209 if ('.hgsubstate' in wctx and
1210 '.hgsubstate' not in changes[0] + changes[1] + changes[2]):
1210 '.hgsubstate' not in changes[0] + changes[1] + changes[2]):
1211 changes[2].insert(0, '.hgsubstate')
1211 changes[2].insert(0, '.hgsubstate')
1212
1212
1213 # make sure all explicit patterns are matched
1213 # make sure all explicit patterns are matched
1214 if not force and match.files():
1214 if not force and match.files():
1215 matched = set(changes[0] + changes[1] + changes[2])
1215 matched = set(changes[0] + changes[1] + changes[2])
1216
1216
1217 for f in match.files():
1217 for f in match.files():
1218 f = self.dirstate.normalize(f)
1218 f = self.dirstate.normalize(f)
1219 if f == '.' or f in matched or f in wctx.substate:
1219 if f == '.' or f in matched or f in wctx.substate:
1220 continue
1220 continue
1221 if f in changes[3]: # missing
1221 if f in changes[3]: # missing
1222 fail(f, _('file not found!'))
1222 fail(f, _('file not found!'))
1223 if f in vdirs: # visited directory
1223 if f in vdirs: # visited directory
1224 d = f + '/'
1224 d = f + '/'
1225 for mf in matched:
1225 for mf in matched:
1226 if mf.startswith(d):
1226 if mf.startswith(d):
1227 break
1227 break
1228 else:
1228 else:
1229 fail(f, _("no match under directory!"))
1229 fail(f, _("no match under directory!"))
1230 elif f not in self.dirstate:
1230 elif f not in self.dirstate:
1231 fail(f, _("file not tracked!"))
1231 fail(f, _("file not tracked!"))
1232
1232
1233 cctx = context.workingctx(self, text, user, date, extra, changes)
1233 cctx = context.workingctx(self, text, user, date, extra, changes)
1234
1234
1235 if (not force and not extra.get("close") and not merge
1235 if (not force and not extra.get("close") and not merge
1236 and not cctx.files()
1236 and not cctx.files()
1237 and wctx.branch() == wctx.p1().branch()):
1237 and wctx.branch() == wctx.p1().branch()):
1238 return None
1238 return None
1239
1239
1240 if merge and cctx.deleted():
1240 if merge and cctx.deleted():
1241 raise util.Abort(_("cannot commit merge with missing files"))
1241 raise util.Abort(_("cannot commit merge with missing files"))
1242
1242
1243 ms = mergemod.mergestate(self)
1243 ms = mergemod.mergestate(self)
1244 for f in changes[0]:
1244 for f in changes[0]:
1245 if f in ms and ms[f] == 'u':
1245 if f in ms and ms[f] == 'u':
1246 raise util.Abort(_("unresolved merge conflicts "
1246 raise util.Abort(_("unresolved merge conflicts "
1247 "(see hg help resolve)"))
1247 "(see hg help resolve)"))
1248
1248
1249 if editor:
1249 if editor:
1250 cctx._text = editor(self, cctx, subs)
1250 cctx._text = editor(self, cctx, subs)
1251 edited = (text != cctx._text)
1251 edited = (text != cctx._text)
1252
1252
1253 # commit subs and write new state
1253 # commit subs and write new state
1254 if subs:
1254 if subs:
1255 for s in sorted(commitsubs):
1255 for s in sorted(commitsubs):
1256 sub = wctx.sub(s)
1256 sub = wctx.sub(s)
1257 self.ui.status(_('committing subrepository %s\n') %
1257 self.ui.status(_('committing subrepository %s\n') %
1258 subrepo.subrelpath(sub))
1258 subrepo.subrelpath(sub))
1259 sr = sub.commit(cctx._text, user, date)
1259 sr = sub.commit(cctx._text, user, date)
1260 newstate[s] = (newstate[s][0], sr)
1260 newstate[s] = (newstate[s][0], sr)
1261 subrepo.writestate(self, newstate)
1261 subrepo.writestate(self, newstate)
1262
1262
1263 # Save commit message in case this transaction gets rolled back
1263 # Save commit message in case this transaction gets rolled back
1264 # (e.g. by a pretxncommit hook). Leave the content alone on
1264 # (e.g. by a pretxncommit hook). Leave the content alone on
1265 # the assumption that the user will use the same editor again.
1265 # the assumption that the user will use the same editor again.
1266 msgfn = self.savecommitmessage(cctx._text)
1266 msgfn = self.savecommitmessage(cctx._text)
1267
1267
1268 p1, p2 = self.dirstate.parents()
1268 p1, p2 = self.dirstate.parents()
1269 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1269 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1270 try:
1270 try:
1271 self.hook("precommit", throw=True, parent1=hookp1,
1271 self.hook("precommit", throw=True, parent1=hookp1,
1272 parent2=hookp2)
1272 parent2=hookp2)
1273 ret = self.commitctx(cctx, True)
1273 ret = self.commitctx(cctx, True)
1274 except: # re-raises
1274 except: # re-raises
1275 if edited:
1275 if edited:
1276 self.ui.write(
1276 self.ui.write(
1277 _('note: commit message saved in %s\n') % msgfn)
1277 _('note: commit message saved in %s\n') % msgfn)
1278 raise
1278 raise
1279
1279
1280 # update bookmarks, dirstate and mergestate
1280 # update bookmarks, dirstate and mergestate
1281 bookmarks.update(self, [p1, p2], ret)
1281 bookmarks.update(self, [p1, p2], ret)
1282 cctx.markcommitted(ret)
1282 cctx.markcommitted(ret)
1283 ms.reset()
1283 ms.reset()
1284 finally:
1284 finally:
1285 wlock.release()
1285 wlock.release()
1286
1286
1287 def commithook(node=hex(ret), parent1=hookp1, parent2=hookp2):
1287 def commithook(node=hex(ret), parent1=hookp1, parent2=hookp2):
1288 self.hook("commit", node=node, parent1=parent1, parent2=parent2)
1288 self.hook("commit", node=node, parent1=parent1, parent2=parent2)
1289 self._afterlock(commithook)
1289 self._afterlock(commithook)
1290 return ret
1290 return ret
1291
1291
1292 @unfilteredmethod
1292 @unfilteredmethod
1293 def commitctx(self, ctx, error=False):
1293 def commitctx(self, ctx, error=False):
1294 """Add a new revision to current repository.
1294 """Add a new revision to current repository.
1295 Revision information is passed via the context argument.
1295 Revision information is passed via the context argument.
1296 """
1296 """
1297
1297
1298 tr = lock = None
1298 tr = lock = None
1299 removed = list(ctx.removed())
1299 removed = list(ctx.removed())
1300 p1, p2 = ctx.p1(), ctx.p2()
1300 p1, p2 = ctx.p1(), ctx.p2()
1301 user = ctx.user()
1301 user = ctx.user()
1302
1302
1303 lock = self.lock()
1303 lock = self.lock()
1304 try:
1304 try:
1305 tr = self.transaction("commit")
1305 tr = self.transaction("commit")
1306 trp = weakref.proxy(tr)
1306 trp = weakref.proxy(tr)
1307
1307
1308 if ctx.files():
1308 if ctx.files():
1309 m1 = p1.manifest().copy()
1309 m1 = p1.manifest().copy()
1310 m2 = p2.manifest()
1310 m2 = p2.manifest()
1311
1311
1312 # check in files
1312 # check in files
1313 new = {}
1313 new = {}
1314 changed = []
1314 changed = []
1315 linkrev = len(self)
1315 linkrev = len(self)
1316 for f in sorted(ctx.modified() + ctx.added()):
1316 for f in sorted(ctx.modified() + ctx.added()):
1317 self.ui.note(f + "\n")
1317 self.ui.note(f + "\n")
1318 try:
1318 try:
1319 fctx = ctx[f]
1319 fctx = ctx[f]
1320 new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
1320 new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
1321 changed)
1321 changed)
1322 m1.set(f, fctx.flags())
1322 m1.set(f, fctx.flags())
1323 except OSError, inst:
1323 except OSError, inst:
1324 self.ui.warn(_("trouble committing %s!\n") % f)
1324 self.ui.warn(_("trouble committing %s!\n") % f)
1325 raise
1325 raise
1326 except IOError, inst:
1326 except IOError, inst:
1327 errcode = getattr(inst, 'errno', errno.ENOENT)
1327 errcode = getattr(inst, 'errno', errno.ENOENT)
1328 if error or errcode and errcode != errno.ENOENT:
1328 if error or errcode and errcode != errno.ENOENT:
1329 self.ui.warn(_("trouble committing %s!\n") % f)
1329 self.ui.warn(_("trouble committing %s!\n") % f)
1330 raise
1330 raise
1331 else:
1331 else:
1332 removed.append(f)
1332 removed.append(f)
1333
1333
1334 # update manifest
1334 # update manifest
1335 m1.update(new)
1335 m1.update(new)
1336 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1336 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1337 drop = [f for f in removed if f in m1]
1337 drop = [f for f in removed if f in m1]
1338 for f in drop:
1338 for f in drop:
1339 del m1[f]
1339 del m1[f]
1340 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1340 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1341 p2.manifestnode(), (new, drop))
1341 p2.manifestnode(), (new, drop))
1342 files = changed + removed
1342 files = changed + removed
1343 else:
1343 else:
1344 mn = p1.manifestnode()
1344 mn = p1.manifestnode()
1345 files = []
1345 files = []
1346
1346
1347 # update changelog
1347 # update changelog
1348 self.changelog.delayupdate()
1348 self.changelog.delayupdate()
1349 n = self.changelog.add(mn, files, ctx.description(),
1349 n = self.changelog.add(mn, files, ctx.description(),
1350 trp, p1.node(), p2.node(),
1350 trp, p1.node(), p2.node(),
1351 user, ctx.date(), ctx.extra().copy())
1351 user, ctx.date(), ctx.extra().copy())
1352 p = lambda: self.changelog.writepending() and self.root or ""
1352 p = lambda: self.changelog.writepending() and self.root or ""
1353 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1353 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1354 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1354 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1355 parent2=xp2, pending=p)
1355 parent2=xp2, pending=p)
1356 self.changelog.finalize(trp)
1356 self.changelog.finalize(trp)
1357 # set the new commit is proper phase
1357 # set the new commit is proper phase
1358 targetphase = phases.newcommitphase(self.ui)
1358 targetphase = phases.newcommitphase(self.ui)
1359 if targetphase:
1359 if targetphase:
1360 # retract boundary do not alter parent changeset.
1360 # retract boundary do not alter parent changeset.
1361 # if a parent have higher the resulting phase will
1361 # if a parent have higher the resulting phase will
1362 # be compliant anyway
1362 # be compliant anyway
1363 #
1363 #
1364 # if minimal phase was 0 we don't need to retract anything
1364 # if minimal phase was 0 we don't need to retract anything
1365 phases.retractboundary(self, targetphase, [n])
1365 phases.retractboundary(self, targetphase, [n])
1366 tr.close()
1366 tr.close()
1367 branchmap.updatecache(self.filtered('served'))
1367 branchmap.updatecache(self.filtered('served'))
1368 return n
1368 return n
1369 finally:
1369 finally:
1370 if tr:
1370 if tr:
1371 tr.release()
1371 tr.release()
1372 lock.release()
1372 lock.release()
1373
1373
1374 @unfilteredmethod
1374 @unfilteredmethod
1375 def destroying(self):
1375 def destroying(self):
1376 '''Inform the repository that nodes are about to be destroyed.
1376 '''Inform the repository that nodes are about to be destroyed.
1377 Intended for use by strip and rollback, so there's a common
1377 Intended for use by strip and rollback, so there's a common
1378 place for anything that has to be done before destroying history.
1378 place for anything that has to be done before destroying history.
1379
1379
1380 This is mostly useful for saving state that is in memory and waiting
1380 This is mostly useful for saving state that is in memory and waiting
1381 to be flushed when the current lock is released. Because a call to
1381 to be flushed when the current lock is released. Because a call to
1382 destroyed is imminent, the repo will be invalidated causing those
1382 destroyed is imminent, the repo will be invalidated causing those
1383 changes to stay in memory (waiting for the next unlock), or vanish
1383 changes to stay in memory (waiting for the next unlock), or vanish
1384 completely.
1384 completely.
1385 '''
1385 '''
1386 # When using the same lock to commit and strip, the phasecache is left
1386 # When using the same lock to commit and strip, the phasecache is left
1387 # dirty after committing. Then when we strip, the repo is invalidated,
1387 # dirty after committing. Then when we strip, the repo is invalidated,
1388 # causing those changes to disappear.
1388 # causing those changes to disappear.
1389 if '_phasecache' in vars(self):
1389 if '_phasecache' in vars(self):
1390 self._phasecache.write()
1390 self._phasecache.write()
1391
1391
1392 @unfilteredmethod
1392 @unfilteredmethod
1393 def destroyed(self):
1393 def destroyed(self):
1394 '''Inform the repository that nodes have been destroyed.
1394 '''Inform the repository that nodes have been destroyed.
1395 Intended for use by strip and rollback, so there's a common
1395 Intended for use by strip and rollback, so there's a common
1396 place for anything that has to be done after destroying history.
1396 place for anything that has to be done after destroying history.
1397 '''
1397 '''
1398 # When one tries to:
1398 # When one tries to:
1399 # 1) destroy nodes thus calling this method (e.g. strip)
1399 # 1) destroy nodes thus calling this method (e.g. strip)
1400 # 2) use phasecache somewhere (e.g. commit)
1400 # 2) use phasecache somewhere (e.g. commit)
1401 #
1401 #
1402 # then 2) will fail because the phasecache contains nodes that were
1402 # then 2) will fail because the phasecache contains nodes that were
1403 # removed. We can either remove phasecache from the filecache,
1403 # removed. We can either remove phasecache from the filecache,
1404 # causing it to reload next time it is accessed, or simply filter
1404 # causing it to reload next time it is accessed, or simply filter
1405 # the removed nodes now and write the updated cache.
1405 # the removed nodes now and write the updated cache.
1406 if '_phasecache' in self._filecache:
1406 if '_phasecache' in self._filecache:
1407 self._phasecache.filterunknown(self)
1407 self._phasecache.filterunknown(self)
1408 self._phasecache.write()
1408 self._phasecache.write()
1409
1409
1410 # update the 'served' branch cache to help read only server process
1410 # update the 'served' branch cache to help read only server process
1411 # Thanks to branchcache collaboration this is done from the nearest
1411 # Thanks to branchcache collaboration this is done from the nearest
1412 # filtered subset and it is expected to be fast.
1412 # filtered subset and it is expected to be fast.
1413 branchmap.updatecache(self.filtered('served'))
1413 branchmap.updatecache(self.filtered('served'))
1414
1414
1415 # Ensure the persistent tag cache is updated. Doing it now
1415 # Ensure the persistent tag cache is updated. Doing it now
1416 # means that the tag cache only has to worry about destroyed
1416 # means that the tag cache only has to worry about destroyed
1417 # heads immediately after a strip/rollback. That in turn
1417 # heads immediately after a strip/rollback. That in turn
1418 # guarantees that "cachetip == currenttip" (comparing both rev
1418 # guarantees that "cachetip == currenttip" (comparing both rev
1419 # and node) always means no nodes have been added or destroyed.
1419 # and node) always means no nodes have been added or destroyed.
1420
1420
1421 # XXX this is suboptimal when qrefresh'ing: we strip the current
1421 # XXX this is suboptimal when qrefresh'ing: we strip the current
1422 # head, refresh the tag cache, then immediately add a new head.
1422 # head, refresh the tag cache, then immediately add a new head.
1423 # But I think doing it this way is necessary for the "instant
1423 # But I think doing it this way is necessary for the "instant
1424 # tag cache retrieval" case to work.
1424 # tag cache retrieval" case to work.
1425 self.invalidate()
1425 self.invalidate()
1426
1426
1427 def walk(self, match, node=None):
1427 def walk(self, match, node=None):
1428 '''
1428 '''
1429 walk recursively through the directory tree or a given
1429 walk recursively through the directory tree or a given
1430 changeset, finding all files matched by the match
1430 changeset, finding all files matched by the match
1431 function
1431 function
1432 '''
1432 '''
1433 return self[node].walk(match)
1433 return self[node].walk(match)
1434
1434
1435 def status(self, node1='.', node2=None, match=None,
1435 def status(self, node1='.', node2=None, match=None,
1436 ignored=False, clean=False, unknown=False,
1436 ignored=False, clean=False, unknown=False,
1437 listsubrepos=False):
1437 listsubrepos=False):
1438 """return status of files between two nodes or node and working
1438 """return status of files between two nodes or node and working
1439 directory.
1439 directory.
1440
1440
1441 If node1 is None, use the first dirstate parent instead.
1441 If node1 is None, use the first dirstate parent instead.
1442 If node2 is None, compare node1 with working directory.
1442 If node2 is None, compare node1 with working directory.
1443 """
1443 """
1444
1444
1445 def mfmatches(ctx):
1445 def mfmatches(ctx):
1446 mf = ctx.manifest().copy()
1446 mf = ctx.manifest().copy()
1447 if match.always():
1447 if match.always():
1448 return mf
1448 return mf
1449 for fn in mf.keys():
1449 for fn in mf.keys():
1450 if not match(fn):
1450 if not match(fn):
1451 del mf[fn]
1451 del mf[fn]
1452 return mf
1452 return mf
1453
1453
1454 if isinstance(node1, context.changectx):
1454 if isinstance(node1, context.changectx):
1455 ctx1 = node1
1455 ctx1 = node1
1456 else:
1456 else:
1457 ctx1 = self[node1]
1457 ctx1 = self[node1]
1458 if isinstance(node2, context.changectx):
1458 if isinstance(node2, context.changectx):
1459 ctx2 = node2
1459 ctx2 = node2
1460 else:
1460 else:
1461 ctx2 = self[node2]
1461 ctx2 = self[node2]
1462
1462
1463 working = ctx2.rev() is None
1463 working = ctx2.rev() is None
1464 parentworking = working and ctx1 == self['.']
1464 parentworking = working and ctx1 == self['.']
1465 match = match or matchmod.always(self.root, self.getcwd())
1465 match = match or matchmod.always(self.root, self.getcwd())
1466 listignored, listclean, listunknown = ignored, clean, unknown
1466 listignored, listclean, listunknown = ignored, clean, unknown
1467
1467
1468 # load earliest manifest first for caching reasons
1468 # load earliest manifest first for caching reasons
1469 if not working and ctx2.rev() < ctx1.rev():
1469 if not working and ctx2.rev() < ctx1.rev():
1470 ctx2.manifest()
1470 ctx2.manifest()
1471
1471
1472 if not parentworking:
1472 if not parentworking:
1473 def bad(f, msg):
1473 def bad(f, msg):
1474 # 'f' may be a directory pattern from 'match.files()',
1474 # 'f' may be a directory pattern from 'match.files()',
1475 # so 'f not in ctx1' is not enough
1475 # so 'f not in ctx1' is not enough
1476 if f not in ctx1 and f not in ctx1.dirs():
1476 if f not in ctx1 and f not in ctx1.dirs():
1477 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1477 self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
1478 match.bad = bad
1478 match.bad = bad
1479
1479
1480 if working: # we need to scan the working dir
1480 if working: # we need to scan the working dir
1481 subrepos = []
1481 subrepos = []
1482 if '.hgsub' in self.dirstate:
1482 if '.hgsub' in self.dirstate:
1483 subrepos = sorted(ctx2.substate)
1483 subrepos = sorted(ctx2.substate)
1484 s = self.dirstate.status(match, subrepos, listignored,
1484 s = self.dirstate.status(match, subrepos, listignored,
1485 listclean, listunknown)
1485 listclean, listunknown)
1486 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1486 cmp, modified, added, removed, deleted, unknown, ignored, clean = s
1487
1487
1488 # check for any possibly clean files
1488 # check for any possibly clean files
1489 if parentworking and cmp:
1489 if parentworking and cmp:
1490 fixup = []
1490 fixup = []
1491 # do a full compare of any files that might have changed
1491 # do a full compare of any files that might have changed
1492 for f in sorted(cmp):
1492 for f in sorted(cmp):
1493 if (f not in ctx1 or ctx2.flags(f) != ctx1.flags(f)
1493 if (f not in ctx1 or ctx2.flags(f) != ctx1.flags(f)
1494 or ctx1[f].cmp(ctx2[f])):
1494 or ctx1[f].cmp(ctx2[f])):
1495 modified.append(f)
1495 modified.append(f)
1496 else:
1496 else:
1497 fixup.append(f)
1497 fixup.append(f)
1498
1498
1499 # update dirstate for files that are actually clean
1499 # update dirstate for files that are actually clean
1500 if fixup:
1500 if fixup:
1501 if listclean:
1501 if listclean:
1502 clean += fixup
1502 clean += fixup
1503
1503
1504 try:
1504 try:
1505 # updating the dirstate is optional
1505 # updating the dirstate is optional
1506 # so we don't wait on the lock
1506 # so we don't wait on the lock
1507 wlock = self.wlock(False)
1507 wlock = self.wlock(False)
1508 try:
1508 try:
1509 for f in fixup:
1509 for f in fixup:
1510 self.dirstate.normal(f)
1510 self.dirstate.normal(f)
1511 finally:
1511 finally:
1512 wlock.release()
1512 wlock.release()
1513 except error.LockError:
1513 except error.LockError:
1514 pass
1514 pass
1515
1515
1516 if not parentworking:
1516 if not parentworking:
1517 mf1 = mfmatches(ctx1)
1517 mf1 = mfmatches(ctx1)
1518 if working:
1518 if working:
1519 # we are comparing working dir against non-parent
1519 # we are comparing working dir against non-parent
1520 # generate a pseudo-manifest for the working dir
1520 # generate a pseudo-manifest for the working dir
1521 mf2 = mfmatches(self['.'])
1521 mf2 = mfmatches(self['.'])
1522 for f in cmp + modified + added:
1522 for f in cmp + modified + added:
1523 mf2[f] = None
1523 mf2[f] = None
1524 mf2.set(f, ctx2.flags(f))
1524 mf2.set(f, ctx2.flags(f))
1525 for f in removed:
1525 for f in removed:
1526 if f in mf2:
1526 if f in mf2:
1527 del mf2[f]
1527 del mf2[f]
1528 else:
1528 else:
1529 # we are comparing two revisions
1529 # we are comparing two revisions
1530 deleted, unknown, ignored = [], [], []
1530 deleted, unknown, ignored = [], [], []
1531 mf2 = mfmatches(ctx2)
1531 mf2 = mfmatches(ctx2)
1532
1532
1533 modified, added, clean = [], [], []
1533 modified, added, clean = [], [], []
1534 withflags = mf1.withflags() | mf2.withflags()
1534 withflags = mf1.withflags() | mf2.withflags()
1535 for fn in mf2:
1535 for fn in mf2:
1536 if fn in mf1:
1536 if fn in mf1:
1537 if (fn not in deleted and
1537 if (fn not in deleted and
1538 ((fn in withflags and mf1.flags(fn) != mf2.flags(fn)) or
1538 ((fn in withflags and mf1.flags(fn) != mf2.flags(fn)) or
1539 (mf1[fn] != mf2[fn] and
1539 (mf1[fn] != mf2[fn] and
1540 (mf2[fn] or ctx1[fn].cmp(ctx2[fn]))))):
1540 (mf2[fn] or ctx1[fn].cmp(ctx2[fn]))))):
1541 modified.append(fn)
1541 modified.append(fn)
1542 elif listclean:
1542 elif listclean:
1543 clean.append(fn)
1543 clean.append(fn)
1544 del mf1[fn]
1544 del mf1[fn]
1545 elif fn not in deleted:
1545 elif fn not in deleted:
1546 added.append(fn)
1546 added.append(fn)
1547 removed = mf1.keys()
1547 removed = mf1.keys()
1548
1548
1549 if working and modified and not self.dirstate._checklink:
1549 if working and modified and not self.dirstate._checklink:
1550 # Symlink placeholders may get non-symlink-like contents
1550 # Symlink placeholders may get non-symlink-like contents
1551 # via user error or dereferencing by NFS or Samba servers,
1551 # via user error or dereferencing by NFS or Samba servers,
1552 # so we filter out any placeholders that don't look like a
1552 # so we filter out any placeholders that don't look like a
1553 # symlink
1553 # symlink
1554 sane = []
1554 sane = []
1555 for f in modified:
1555 for f in modified:
1556 if ctx2.flags(f) == 'l':
1556 if ctx2.flags(f) == 'l':
1557 d = ctx2[f].data()
1557 d = ctx2[f].data()
1558 if len(d) >= 1024 or '\n' in d or util.binary(d):
1558 if len(d) >= 1024 or '\n' in d or util.binary(d):
1559 self.ui.debug('ignoring suspect symlink placeholder'
1559 self.ui.debug('ignoring suspect symlink placeholder'
1560 ' "%s"\n' % f)
1560 ' "%s"\n' % f)
1561 continue
1561 continue
1562 sane.append(f)
1562 sane.append(f)
1563 modified = sane
1563 modified = sane
1564
1564
1565 r = modified, added, removed, deleted, unknown, ignored, clean
1565 r = modified, added, removed, deleted, unknown, ignored, clean
1566
1566
1567 if listsubrepos:
1567 if listsubrepos:
1568 for subpath, sub in subrepo.itersubrepos(ctx1, ctx2):
1568 for subpath, sub in subrepo.itersubrepos(ctx1, ctx2):
1569 if working:
1569 if working:
1570 rev2 = None
1570 rev2 = None
1571 else:
1571 else:
1572 rev2 = ctx2.substate[subpath][1]
1572 rev2 = ctx2.substate[subpath][1]
1573 try:
1573 try:
1574 submatch = matchmod.narrowmatcher(subpath, match)
1574 submatch = matchmod.narrowmatcher(subpath, match)
1575 s = sub.status(rev2, match=submatch, ignored=listignored,
1575 s = sub.status(rev2, match=submatch, ignored=listignored,
1576 clean=listclean, unknown=listunknown,
1576 clean=listclean, unknown=listunknown,
1577 listsubrepos=True)
1577 listsubrepos=True)
1578 for rfiles, sfiles in zip(r, s):
1578 for rfiles, sfiles in zip(r, s):
1579 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
1579 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
1580 except error.LookupError:
1580 except error.LookupError:
1581 self.ui.status(_("skipping missing subrepository: %s\n")
1581 self.ui.status(_("skipping missing subrepository: %s\n")
1582 % subpath)
1582 % subpath)
1583
1583
1584 for l in r:
1584 for l in r:
1585 l.sort()
1585 l.sort()
1586 return r
1586 return r
1587
1587
1588 def heads(self, start=None):
1588 def heads(self, start=None):
1589 heads = self.changelog.heads(start)
1589 heads = self.changelog.heads(start)
1590 # sort the output in rev descending order
1590 # sort the output in rev descending order
1591 return sorted(heads, key=self.changelog.rev, reverse=True)
1591 return sorted(heads, key=self.changelog.rev, reverse=True)
1592
1592
1593 def branchheads(self, branch=None, start=None, closed=False):
1593 def branchheads(self, branch=None, start=None, closed=False):
1594 '''return a (possibly filtered) list of heads for the given branch
1594 '''return a (possibly filtered) list of heads for the given branch
1595
1595
1596 Heads are returned in topological order, from newest to oldest.
1596 Heads are returned in topological order, from newest to oldest.
1597 If branch is None, use the dirstate branch.
1597 If branch is None, use the dirstate branch.
1598 If start is not None, return only heads reachable from start.
1598 If start is not None, return only heads reachable from start.
1599 If closed is True, return heads that are marked as closed as well.
1599 If closed is True, return heads that are marked as closed as well.
1600 '''
1600 '''
1601 if branch is None:
1601 if branch is None:
1602 branch = self[None].branch()
1602 branch = self[None].branch()
1603 branches = self.branchmap()
1603 branches = self.branchmap()
1604 if branch not in branches:
1604 if branch not in branches:
1605 return []
1605 return []
1606 # the cache returns heads ordered lowest to highest
1606 # the cache returns heads ordered lowest to highest
1607 bheads = list(reversed(branches[branch]))
1607 bheads = list(reversed(branches[branch]))
1608 if start is not None:
1608 if start is not None:
1609 # filter out the heads that cannot be reached from startrev
1609 # filter out the heads that cannot be reached from startrev
1610 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1610 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1611 bheads = [h for h in bheads if h in fbheads]
1611 bheads = [h for h in bheads if h in fbheads]
1612 if not closed:
1612 if not closed:
1613 bheads = [h for h in bheads if not self[h].closesbranch()]
1613 bheads = [h for h in bheads if not self[h].closesbranch()]
1614 return bheads
1614 return bheads
1615
1615
1616 def branches(self, nodes):
1616 def branches(self, nodes):
1617 if not nodes:
1617 if not nodes:
1618 nodes = [self.changelog.tip()]
1618 nodes = [self.changelog.tip()]
1619 b = []
1619 b = []
1620 for n in nodes:
1620 for n in nodes:
1621 t = n
1621 t = n
1622 while True:
1622 while True:
1623 p = self.changelog.parents(n)
1623 p = self.changelog.parents(n)
1624 if p[1] != nullid or p[0] == nullid:
1624 if p[1] != nullid or p[0] == nullid:
1625 b.append((t, n, p[0], p[1]))
1625 b.append((t, n, p[0], p[1]))
1626 break
1626 break
1627 n = p[0]
1627 n = p[0]
1628 return b
1628 return b
1629
1629
1630 def between(self, pairs):
1630 def between(self, pairs):
1631 r = []
1631 r = []
1632
1632
1633 for top, bottom in pairs:
1633 for top, bottom in pairs:
1634 n, l, i = top, [], 0
1634 n, l, i = top, [], 0
1635 f = 1
1635 f = 1
1636
1636
1637 while n != bottom and n != nullid:
1637 while n != bottom and n != nullid:
1638 p = self.changelog.parents(n)[0]
1638 p = self.changelog.parents(n)[0]
1639 if i == f:
1639 if i == f:
1640 l.append(n)
1640 l.append(n)
1641 f = f * 2
1641 f = f * 2
1642 n = p
1642 n = p
1643 i += 1
1643 i += 1
1644
1644
1645 r.append(l)
1645 r.append(l)
1646
1646
1647 return r
1647 return r
1648
1648
1649 def pull(self, remote, heads=None, force=False):
1649 def pull(self, remote, heads=None, force=False):
1650 # don't open transaction for nothing or you break future useful
1650 # don't open transaction for nothing or you break future useful
1651 # rollback call
1651 # rollback call
1652 tr = None
1652 tr = None
1653 trname = 'pull\n' + util.hidepassword(remote.url())
1653 trname = 'pull\n' + util.hidepassword(remote.url())
1654 lock = self.lock()
1654 lock = self.lock()
1655 try:
1655 try:
1656 tmp = discovery.findcommonincoming(self, remote, heads=heads,
1656 tmp = discovery.findcommonincoming(self, remote, heads=heads,
1657 force=force)
1657 force=force)
1658 common, fetch, rheads = tmp
1658 common, fetch, rheads = tmp
1659 if not fetch:
1659 if not fetch:
1660 self.ui.status(_("no changes found\n"))
1660 self.ui.status(_("no changes found\n"))
1661 added = []
1661 added = []
1662 result = 0
1662 result = 0
1663 else:
1663 else:
1664 tr = self.transaction(trname)
1664 tr = self.transaction(trname)
1665 if heads is None and list(common) == [nullid]:
1665 if heads is None and list(common) == [nullid]:
1666 self.ui.status(_("requesting all changes\n"))
1666 self.ui.status(_("requesting all changes\n"))
1667 elif heads is None and remote.capable('changegroupsubset'):
1667 elif heads is None and remote.capable('changegroupsubset'):
1668 # issue1320, avoid a race if remote changed after discovery
1668 # issue1320, avoid a race if remote changed after discovery
1669 heads = rheads
1669 heads = rheads
1670
1670
1671 if remote.capable('getbundle'):
1671 if remote.capable('getbundle'):
1672 cg = remote.getbundle('pull', common=common,
1672 cg = remote.getbundle('pull', common=common,
1673 heads=heads or rheads)
1673 heads=heads or rheads)
1674 elif heads is None:
1674 elif heads is None:
1675 cg = remote.changegroup(fetch, 'pull')
1675 cg = remote.changegroup(fetch, 'pull')
1676 elif not remote.capable('changegroupsubset'):
1676 elif not remote.capable('changegroupsubset'):
1677 raise util.Abort(_("partial pull cannot be done because "
1677 raise util.Abort(_("partial pull cannot be done because "
1678 "other repository doesn't support "
1678 "other repository doesn't support "
1679 "changegroupsubset."))
1679 "changegroupsubset."))
1680 else:
1680 else:
1681 cg = remote.changegroupsubset(fetch, heads, 'pull')
1681 cg = remote.changegroupsubset(fetch, heads, 'pull')
1682 # we use unfiltered changelog here because hidden revision must
1682 # we use unfiltered changelog here because hidden revision must
1683 # be taken in account for phase synchronization. They may
1683 # be taken in account for phase synchronization. They may
1684 # becomes public and becomes visible again.
1684 # becomes public and becomes visible again.
1685 cl = self.unfiltered().changelog
1685 cl = self.unfiltered().changelog
1686 clstart = len(cl)
1686 clstart = len(cl)
1687 result = self.addchangegroup(cg, 'pull', remote.url())
1687 result = self.addchangegroup(cg, 'pull', remote.url())
1688 clend = len(cl)
1688 clend = len(cl)
1689 added = [cl.node(r) for r in xrange(clstart, clend)]
1689 added = [cl.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 (self.ui.configbool('ui', '_usedassubrepo', False)
1868 if (self.ui.configbool('ui', '_usedassubrepo', False)
1869 and remotephases # server supports phases
1869 and remotephases # server supports phases
1870 and ret is None # nothing was pushed
1870 and ret is None # nothing was pushed
1871 and remotephases.get('publishing', False)):
1871 and remotephases.get('publishing', False)):
1872 # When:
1872 # When:
1873 # - this is a subrepo push
1873 # - this is a subrepo push
1874 # - and remote support phase
1874 # - and remote support phase
1875 # - and no changeset was pushed
1875 # - and no changeset was pushed
1876 # - and remote is publishing
1876 # - and remote is publishing
1877 # We may be in issue 3871 case!
1877 # We may be in issue 3871 case!
1878 # We drop the possible phase synchronisation done by
1878 # We drop the possible phase synchronisation done by
1879 # courtesy to publish changesets possibly locally draft
1879 # courtesy to publish changesets possibly locally draft
1880 # on the remote.
1880 # on the remote.
1881 remotephases = {'publishing': 'True'}
1881 remotephases = {'publishing': 'True'}
1882 if not remotephases: # old server or public only repo
1882 if not remotephases: # old server or public only repo
1883 phases.advanceboundary(self, phases.public, cheads)
1883 phases.advanceboundary(self, phases.public, cheads)
1884 # don't push any phase data as there is nothing to push
1884 # don't push any phase data as there is nothing to push
1885 else:
1885 else:
1886 ana = phases.analyzeremotephases(self, cheads, remotephases)
1886 ana = phases.analyzeremotephases(self, cheads, remotephases)
1887 pheads, droots = ana
1887 pheads, droots = ana
1888 ### Apply remote phase on local
1888 ### Apply remote phase on local
1889 if remotephases.get('publishing', False):
1889 if remotephases.get('publishing', False):
1890 phases.advanceboundary(self, phases.public, cheads)
1890 phases.advanceboundary(self, phases.public, cheads)
1891 else: # publish = False
1891 else: # publish = False
1892 phases.advanceboundary(self, phases.public, pheads)
1892 phases.advanceboundary(self, phases.public, pheads)
1893 phases.advanceboundary(self, phases.draft, cheads)
1893 phases.advanceboundary(self, phases.draft, cheads)
1894 ### Apply local phase on remote
1894 ### Apply local phase on remote
1895
1895
1896 # Get the list of all revs draft on remote by public here.
1896 # Get the list of all revs draft on remote by public here.
1897 # XXX Beware that revset break if droots is not strictly
1897 # XXX Beware that revset break if droots is not strictly
1898 # XXX root we may want to ensure it is but it is costly
1898 # XXX root we may want to ensure it is but it is costly
1899 outdated = unfi.set('heads((%ln::%ln) and public())',
1899 outdated = unfi.set('heads((%ln::%ln) and public())',
1900 droots, cheads)
1900 droots, cheads)
1901 for newremotehead in outdated:
1901 for newremotehead in outdated:
1902 r = remote.pushkey('phases',
1902 r = remote.pushkey('phases',
1903 newremotehead.hex(),
1903 newremotehead.hex(),
1904 str(phases.draft),
1904 str(phases.draft),
1905 str(phases.public))
1905 str(phases.public))
1906 if not r:
1906 if not r:
1907 self.ui.warn(_('updating %s to public failed!\n')
1907 self.ui.warn(_('updating %s to public failed!\n')
1908 % newremotehead)
1908 % newremotehead)
1909 self.ui.debug('try to push obsolete markers to remote\n')
1909 self.ui.debug('try to push obsolete markers to remote\n')
1910 if (obsolete._enabled and self.obsstore and
1910 if (obsolete._enabled and self.obsstore and
1911 'obsolete' in remote.listkeys('namespaces')):
1911 'obsolete' in remote.listkeys('namespaces')):
1912 rslts = []
1912 rslts = []
1913 remotedata = self.listkeys('obsolete')
1913 remotedata = self.listkeys('obsolete')
1914 for key in sorted(remotedata, reverse=True):
1914 for key in sorted(remotedata, reverse=True):
1915 # reverse sort to ensure we end with dump0
1915 # reverse sort to ensure we end with dump0
1916 data = remotedata[key]
1916 data = remotedata[key]
1917 rslts.append(remote.pushkey('obsolete', key, '', data))
1917 rslts.append(remote.pushkey('obsolete', key, '', data))
1918 if [r for r in rslts if not r]:
1918 if [r for r in rslts if not r]:
1919 msg = _('failed to push some obsolete markers!\n')
1919 msg = _('failed to push some obsolete markers!\n')
1920 self.ui.warn(msg)
1920 self.ui.warn(msg)
1921 finally:
1921 finally:
1922 if lock is not None:
1922 if lock is not None:
1923 lock.release()
1923 lock.release()
1924 finally:
1924 finally:
1925 locallock.release()
1925 locallock.release()
1926
1926
1927 self.ui.debug("checking for updated bookmarks\n")
1927 self.ui.debug("checking for updated bookmarks\n")
1928 rb = remote.listkeys('bookmarks')
1928 rb = remote.listkeys('bookmarks')
1929 for k in rb.keys():
1929 for k in rb.keys():
1930 if k in unfi._bookmarks:
1930 if k in unfi._bookmarks:
1931 nr, nl = rb[k], hex(self._bookmarks[k])
1931 nr, nl = rb[k], hex(self._bookmarks[k])
1932 if nr in unfi:
1932 if nr in unfi:
1933 cr = unfi[nr]
1933 cr = unfi[nr]
1934 cl = unfi[nl]
1934 cl = unfi[nl]
1935 if bookmarks.validdest(unfi, cr, cl):
1935 if bookmarks.validdest(unfi, cr, cl):
1936 r = remote.pushkey('bookmarks', k, nr, nl)
1936 r = remote.pushkey('bookmarks', k, nr, nl)
1937 if r:
1937 if r:
1938 self.ui.status(_("updating bookmark %s\n") % k)
1938 self.ui.status(_("updating bookmark %s\n") % k)
1939 else:
1939 else:
1940 self.ui.warn(_('updating bookmark %s'
1940 self.ui.warn(_('updating bookmark %s'
1941 ' failed!\n') % k)
1941 ' failed!\n') % k)
1942
1942
1943 return ret
1943 return ret
1944
1944
1945 def changegroupinfo(self, nodes, source):
1945 def changegroupinfo(self, nodes, source):
1946 if self.ui.verbose or source == 'bundle':
1946 if self.ui.verbose or source == 'bundle':
1947 self.ui.status(_("%d changesets found\n") % len(nodes))
1947 self.ui.status(_("%d changesets found\n") % len(nodes))
1948 if self.ui.debugflag:
1948 if self.ui.debugflag:
1949 self.ui.debug("list of changesets:\n")
1949 self.ui.debug("list of changesets:\n")
1950 for node in nodes:
1950 for node in nodes:
1951 self.ui.debug("%s\n" % hex(node))
1951 self.ui.debug("%s\n" % hex(node))
1952
1952
1953 def changegroupsubset(self, bases, heads, source):
1953 def changegroupsubset(self, bases, heads, source):
1954 """Compute a changegroup consisting of all the nodes that are
1954 """Compute a changegroup consisting of all the nodes that are
1955 descendants of any of the bases and ancestors of any of the heads.
1955 descendants of any of the bases and ancestors of any of the heads.
1956 Return a chunkbuffer object whose read() method will return
1956 Return a chunkbuffer object whose read() method will return
1957 successive changegroup chunks.
1957 successive changegroup chunks.
1958
1958
1959 It is fairly complex as determining which filenodes and which
1959 It is fairly complex as determining which filenodes and which
1960 manifest nodes need to be included for the changeset to be complete
1960 manifest nodes need to be included for the changeset to be complete
1961 is non-trivial.
1961 is non-trivial.
1962
1962
1963 Another wrinkle is doing the reverse, figuring out which changeset in
1963 Another wrinkle is doing the reverse, figuring out which changeset in
1964 the changegroup a particular filenode or manifestnode belongs to.
1964 the changegroup a particular filenode or manifestnode belongs to.
1965 """
1965 """
1966 cl = self.changelog
1966 cl = self.changelog
1967 if not bases:
1967 if not bases:
1968 bases = [nullid]
1968 bases = [nullid]
1969 csets, bases, heads = cl.nodesbetween(bases, heads)
1969 csets, bases, heads = cl.nodesbetween(bases, heads)
1970 # We assume that all ancestors of bases are known
1970 # We assume that all ancestors of bases are known
1971 common = cl.ancestors([cl.rev(n) for n in bases])
1971 common = cl.ancestors([cl.rev(n) for n in bases])
1972 return self._changegroupsubset(common, csets, heads, source)
1972 return self._changegroupsubset(common, csets, heads, source)
1973
1973
1974 def getlocalbundle(self, source, outgoing):
1974 def getlocalbundle(self, source, outgoing):
1975 """Like getbundle, but taking a discovery.outgoing as an argument.
1975 """Like getbundle, but taking a discovery.outgoing as an argument.
1976
1976
1977 This is only implemented for local repos and reuses potentially
1977 This is only implemented for local repos and reuses potentially
1978 precomputed sets in outgoing."""
1978 precomputed sets in outgoing."""
1979 if not outgoing.missing:
1979 if not outgoing.missing:
1980 return None
1980 return None
1981 return self._changegroupsubset(outgoing.common,
1981 return self._changegroupsubset(outgoing.common,
1982 outgoing.missing,
1982 outgoing.missing,
1983 outgoing.missingheads,
1983 outgoing.missingheads,
1984 source)
1984 source)
1985
1985
1986 def getbundle(self, source, heads=None, common=None):
1986 def getbundle(self, source, heads=None, common=None):
1987 """Like changegroupsubset, but returns the set difference between the
1987 """Like changegroupsubset, but returns the set difference between the
1988 ancestors of heads and the ancestors common.
1988 ancestors of heads and the ancestors common.
1989
1989
1990 If heads is None, use the local heads. If common is None, use [nullid].
1990 If heads is None, use the local heads. If common is None, use [nullid].
1991
1991
1992 The nodes in common might not all be known locally due to the way the
1992 The nodes in common might not all be known locally due to the way the
1993 current discovery protocol works.
1993 current discovery protocol works.
1994 """
1994 """
1995 cl = self.changelog
1995 cl = self.changelog
1996 if common:
1996 if common:
1997 hasnode = cl.hasnode
1997 hasnode = cl.hasnode
1998 common = [n for n in common if hasnode(n)]
1998 common = [n for n in common if hasnode(n)]
1999 else:
1999 else:
2000 common = [nullid]
2000 common = [nullid]
2001 if not heads:
2001 if not heads:
2002 heads = cl.heads()
2002 heads = cl.heads()
2003 return self.getlocalbundle(source,
2003 return self.getlocalbundle(source,
2004 discovery.outgoing(cl, common, heads))
2004 discovery.outgoing(cl, common, heads))
2005
2005
2006 @unfilteredmethod
2006 @unfilteredmethod
2007 def _changegroupsubset(self, commonrevs, csets, heads, source):
2007 def _changegroupsubset(self, commonrevs, csets, heads, source):
2008
2008
2009 cl = self.changelog
2009 cl = self.changelog
2010 mf = self.manifest
2010 mf = self.manifest
2011 mfs = {} # needed manifests
2011 mfs = {} # needed manifests
2012 fnodes = {} # needed file nodes
2012 fnodes = {} # needed file nodes
2013 changedfiles = set()
2013 changedfiles = set()
2014 fstate = ['', {}]
2014 fstate = ['', {}]
2015 count = [0, 0]
2015 count = [0, 0]
2016
2016
2017 # can we go through the fast path ?
2017 # can we go through the fast path ?
2018 heads.sort()
2018 heads.sort()
2019 if heads == sorted(self.heads()):
2019 if heads == sorted(self.heads()):
2020 return self._changegroup(csets, source)
2020 return self._changegroup(csets, source)
2021
2021
2022 # slow path
2022 # slow path
2023 self.hook('preoutgoing', throw=True, source=source)
2023 self.hook('preoutgoing', throw=True, source=source)
2024 self.changegroupinfo(csets, source)
2024 self.changegroupinfo(csets, source)
2025
2025
2026 # filter any nodes that claim to be part of the known set
2026 # filter any nodes that claim to be part of the known set
2027 def prune(revlog, missing):
2027 def prune(revlog, missing):
2028 rr, rl = revlog.rev, revlog.linkrev
2028 rr, rl = revlog.rev, revlog.linkrev
2029 return [n for n in missing
2029 return [n for n in missing
2030 if rl(rr(n)) not in commonrevs]
2030 if rl(rr(n)) not in commonrevs]
2031
2031
2032 progress = self.ui.progress
2032 progress = self.ui.progress
2033 _bundling = _('bundling')
2033 _bundling = _('bundling')
2034 _changesets = _('changesets')
2034 _changesets = _('changesets')
2035 _manifests = _('manifests')
2035 _manifests = _('manifests')
2036 _files = _('files')
2036 _files = _('files')
2037
2037
2038 def lookup(revlog, x):
2038 def lookup(revlog, x):
2039 if revlog == cl:
2039 if revlog == cl:
2040 c = cl.read(x)
2040 c = cl.read(x)
2041 changedfiles.update(c[3])
2041 changedfiles.update(c[3])
2042 mfs.setdefault(c[0], x)
2042 mfs.setdefault(c[0], x)
2043 count[0] += 1
2043 count[0] += 1
2044 progress(_bundling, count[0],
2044 progress(_bundling, count[0],
2045 unit=_changesets, total=count[1])
2045 unit=_changesets, total=count[1])
2046 return x
2046 return x
2047 elif revlog == mf:
2047 elif revlog == mf:
2048 clnode = mfs[x]
2048 clnode = mfs[x]
2049 mdata = mf.readfast(x)
2049 mdata = mf.readfast(x)
2050 for f, n in mdata.iteritems():
2050 for f, n in mdata.iteritems():
2051 if f in changedfiles:
2051 if f in changedfiles:
2052 fnodes[f].setdefault(n, clnode)
2052 fnodes[f].setdefault(n, clnode)
2053 count[0] += 1
2053 count[0] += 1
2054 progress(_bundling, count[0],
2054 progress(_bundling, count[0],
2055 unit=_manifests, total=count[1])
2055 unit=_manifests, total=count[1])
2056 return clnode
2056 return clnode
2057 else:
2057 else:
2058 progress(_bundling, count[0], item=fstate[0],
2058 progress(_bundling, count[0], item=fstate[0],
2059 unit=_files, total=count[1])
2059 unit=_files, total=count[1])
2060 return fstate[1][x]
2060 return fstate[1][x]
2061
2061
2062 bundler = changegroup.bundle10(lookup)
2062 bundler = changegroup.bundle10(lookup)
2063 reorder = self.ui.config('bundle', 'reorder', 'auto')
2063 reorder = self.ui.config('bundle', 'reorder', 'auto')
2064 if reorder == 'auto':
2064 if reorder == 'auto':
2065 reorder = None
2065 reorder = None
2066 else:
2066 else:
2067 reorder = util.parsebool(reorder)
2067 reorder = util.parsebool(reorder)
2068
2068
2069 def gengroup():
2069 def gengroup():
2070 # Create a changenode group generator that will call our functions
2070 # Create a changenode group generator that will call our functions
2071 # back to lookup the owning changenode and collect information.
2071 # back to lookup the owning changenode and collect information.
2072 count[:] = [0, len(csets)]
2072 count[:] = [0, len(csets)]
2073 for chunk in cl.group(csets, bundler, reorder=reorder):
2073 for chunk in cl.group(csets, bundler, reorder=reorder):
2074 yield chunk
2074 yield chunk
2075 progress(_bundling, None)
2075 progress(_bundling, None)
2076
2076
2077 # Create a generator for the manifestnodes that calls our lookup
2077 # Create a generator for the manifestnodes that calls our lookup
2078 # and data collection functions back.
2078 # and data collection functions back.
2079 for f in changedfiles:
2079 for f in changedfiles:
2080 fnodes[f] = {}
2080 fnodes[f] = {}
2081 count[:] = [0, len(mfs)]
2081 count[:] = [0, len(mfs)]
2082 for chunk in mf.group(prune(mf, mfs), bundler, reorder=reorder):
2082 for chunk in mf.group(prune(mf, mfs), bundler, reorder=reorder):
2083 yield chunk
2083 yield chunk
2084 progress(_bundling, None)
2084 progress(_bundling, None)
2085
2085
2086 mfs.clear()
2086 mfs.clear()
2087
2087
2088 # Go through all our files in order sorted by name.
2088 # Go through all our files in order sorted by name.
2089 count[:] = [0, len(changedfiles)]
2089 count[:] = [0, len(changedfiles)]
2090 for fname in sorted(changedfiles):
2090 for fname in sorted(changedfiles):
2091 filerevlog = self.file(fname)
2091 filerevlog = self.file(fname)
2092 if not len(filerevlog):
2092 if not len(filerevlog):
2093 raise util.Abort(_("empty or missing revlog for %s")
2093 raise util.Abort(_("empty or missing revlog for %s")
2094 % fname)
2094 % fname)
2095 fstate[0] = fname
2095 fstate[0] = fname
2096 fstate[1] = fnodes.pop(fname, {})
2096 fstate[1] = fnodes.pop(fname, {})
2097
2097
2098 nodelist = prune(filerevlog, fstate[1])
2098 nodelist = prune(filerevlog, fstate[1])
2099 if nodelist:
2099 if nodelist:
2100 count[0] += 1
2100 count[0] += 1
2101 yield bundler.fileheader(fname)
2101 yield bundler.fileheader(fname)
2102 for chunk in filerevlog.group(nodelist, bundler, reorder):
2102 for chunk in filerevlog.group(nodelist, bundler, reorder):
2103 yield chunk
2103 yield chunk
2104
2104
2105 # Signal that no more groups are left.
2105 # Signal that no more groups are left.
2106 yield bundler.close()
2106 yield bundler.close()
2107 progress(_bundling, None)
2107 progress(_bundling, None)
2108
2108
2109 if csets:
2109 if csets:
2110 self.hook('outgoing', node=hex(csets[0]), source=source)
2110 self.hook('outgoing', node=hex(csets[0]), source=source)
2111
2111
2112 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
2112 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
2113
2113
2114 def changegroup(self, basenodes, source):
2114 def changegroup(self, basenodes, source):
2115 # to avoid a race we use changegroupsubset() (issue1320)
2115 # to avoid a race we use changegroupsubset() (issue1320)
2116 return self.changegroupsubset(basenodes, self.heads(), source)
2116 return self.changegroupsubset(basenodes, self.heads(), source)
2117
2117
2118 @unfilteredmethod
2118 @unfilteredmethod
2119 def _changegroup(self, nodes, source):
2119 def _changegroup(self, nodes, source):
2120 """Compute the changegroup of all nodes that we have that a recipient
2120 """Compute the changegroup of all nodes that we have that a recipient
2121 doesn't. Return a chunkbuffer object whose read() method will return
2121 doesn't. Return a chunkbuffer object whose read() method will return
2122 successive changegroup chunks.
2122 successive changegroup chunks.
2123
2123
2124 This is much easier than the previous function as we can assume that
2124 This is much easier than the previous function as we can assume that
2125 the recipient has any changenode we aren't sending them.
2125 the recipient has any changenode we aren't sending them.
2126
2126
2127 nodes is the set of nodes to send"""
2127 nodes is the set of nodes to send"""
2128
2128
2129 cl = self.changelog
2129 cl = self.changelog
2130 mf = self.manifest
2130 mf = self.manifest
2131 mfs = {}
2131 mfs = {}
2132 changedfiles = set()
2132 changedfiles = set()
2133 fstate = ['']
2133 fstate = ['']
2134 count = [0, 0]
2134 count = [0, 0]
2135
2135
2136 self.hook('preoutgoing', throw=True, source=source)
2136 self.hook('preoutgoing', throw=True, source=source)
2137 self.changegroupinfo(nodes, source)
2137 self.changegroupinfo(nodes, source)
2138
2138
2139 revset = set([cl.rev(n) for n in nodes])
2139 revset = set([cl.rev(n) for n in nodes])
2140
2140
2141 def gennodelst(log):
2141 def gennodelst(log):
2142 ln, llr = log.node, log.linkrev
2142 ln, llr = log.node, log.linkrev
2143 return [ln(r) for r in log if llr(r) in revset]
2143 return [ln(r) for r in log if llr(r) in revset]
2144
2144
2145 progress = self.ui.progress
2145 progress = self.ui.progress
2146 _bundling = _('bundling')
2146 _bundling = _('bundling')
2147 _changesets = _('changesets')
2147 _changesets = _('changesets')
2148 _manifests = _('manifests')
2148 _manifests = _('manifests')
2149 _files = _('files')
2149 _files = _('files')
2150
2150
2151 def lookup(revlog, x):
2151 def lookup(revlog, x):
2152 if revlog == cl:
2152 if revlog == cl:
2153 c = cl.read(x)
2153 c = cl.read(x)
2154 changedfiles.update(c[3])
2154 changedfiles.update(c[3])
2155 mfs.setdefault(c[0], x)
2155 mfs.setdefault(c[0], x)
2156 count[0] += 1
2156 count[0] += 1
2157 progress(_bundling, count[0],
2157 progress(_bundling, count[0],
2158 unit=_changesets, total=count[1])
2158 unit=_changesets, total=count[1])
2159 return x
2159 return x
2160 elif revlog == mf:
2160 elif revlog == mf:
2161 count[0] += 1
2161 count[0] += 1
2162 progress(_bundling, count[0],
2162 progress(_bundling, count[0],
2163 unit=_manifests, total=count[1])
2163 unit=_manifests, total=count[1])
2164 return cl.node(revlog.linkrev(revlog.rev(x)))
2164 return cl.node(revlog.linkrev(revlog.rev(x)))
2165 else:
2165 else:
2166 progress(_bundling, count[0], item=fstate[0],
2166 progress(_bundling, count[0], item=fstate[0],
2167 total=count[1], unit=_files)
2167 total=count[1], unit=_files)
2168 return cl.node(revlog.linkrev(revlog.rev(x)))
2168 return cl.node(revlog.linkrev(revlog.rev(x)))
2169
2169
2170 bundler = changegroup.bundle10(lookup)
2170 bundler = changegroup.bundle10(lookup)
2171 reorder = self.ui.config('bundle', 'reorder', 'auto')
2171 reorder = self.ui.config('bundle', 'reorder', 'auto')
2172 if reorder == 'auto':
2172 if reorder == 'auto':
2173 reorder = None
2173 reorder = None
2174 else:
2174 else:
2175 reorder = util.parsebool(reorder)
2175 reorder = util.parsebool(reorder)
2176
2176
2177 def gengroup():
2177 def gengroup():
2178 '''yield a sequence of changegroup chunks (strings)'''
2178 '''yield a sequence of changegroup chunks (strings)'''
2179 # construct a list of all changed files
2179 # construct a list of all changed files
2180
2180
2181 count[:] = [0, len(nodes)]
2181 count[:] = [0, len(nodes)]
2182 for chunk in cl.group(nodes, bundler, reorder=reorder):
2182 for chunk in cl.group(nodes, bundler, reorder=reorder):
2183 yield chunk
2183 yield chunk
2184 progress(_bundling, None)
2184 progress(_bundling, None)
2185
2185
2186 count[:] = [0, len(mfs)]
2186 count[:] = [0, len(mfs)]
2187 for chunk in mf.group(gennodelst(mf), bundler, reorder=reorder):
2187 for chunk in mf.group(gennodelst(mf), bundler, reorder=reorder):
2188 yield chunk
2188 yield chunk
2189 progress(_bundling, None)
2189 progress(_bundling, None)
2190
2190
2191 count[:] = [0, len(changedfiles)]
2191 count[:] = [0, len(changedfiles)]
2192 for fname in sorted(changedfiles):
2192 for fname in sorted(changedfiles):
2193 filerevlog = self.file(fname)
2193 filerevlog = self.file(fname)
2194 if not len(filerevlog):
2194 if not len(filerevlog):
2195 raise util.Abort(_("empty or missing revlog for %s")
2195 raise util.Abort(_("empty or missing revlog for %s")
2196 % fname)
2196 % fname)
2197 fstate[0] = fname
2197 fstate[0] = fname
2198 nodelist = gennodelst(filerevlog)
2198 nodelist = gennodelst(filerevlog)
2199 if nodelist:
2199 if nodelist:
2200 count[0] += 1
2200 count[0] += 1
2201 yield bundler.fileheader(fname)
2201 yield bundler.fileheader(fname)
2202 for chunk in filerevlog.group(nodelist, bundler, reorder):
2202 for chunk in filerevlog.group(nodelist, bundler, reorder):
2203 yield chunk
2203 yield chunk
2204 yield bundler.close()
2204 yield bundler.close()
2205 progress(_bundling, None)
2205 progress(_bundling, None)
2206
2206
2207 if nodes:
2207 if nodes:
2208 self.hook('outgoing', node=hex(nodes[0]), source=source)
2208 self.hook('outgoing', node=hex(nodes[0]), source=source)
2209
2209
2210 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
2210 return changegroup.unbundle10(util.chunkbuffer(gengroup()), 'UN')
2211
2211
2212 @unfilteredmethod
2212 @unfilteredmethod
2213 def addchangegroup(self, source, srctype, url, emptyok=False):
2213 def addchangegroup(self, source, srctype, url, emptyok=False):
2214 """Add the changegroup returned by source.read() to this repo.
2214 """Add the changegroup returned by source.read() to this repo.
2215 srctype is a string like 'push', 'pull', or 'unbundle'. url is
2215 srctype is a string like 'push', 'pull', or 'unbundle'. url is
2216 the URL of the repo where this changegroup is coming from.
2216 the URL of the repo where this changegroup is coming from.
2217
2217
2218 Return an integer summarizing the change to this repo:
2218 Return an integer summarizing the change to this repo:
2219 - nothing changed or no source: 0
2219 - nothing changed or no source: 0
2220 - more heads than before: 1+added heads (2..n)
2220 - more heads than before: 1+added heads (2..n)
2221 - fewer heads than before: -1-removed heads (-2..-n)
2221 - fewer heads than before: -1-removed heads (-2..-n)
2222 - number of heads stays the same: 1
2222 - number of heads stays the same: 1
2223 """
2223 """
2224 def csmap(x):
2224 def csmap(x):
2225 self.ui.debug("add changeset %s\n" % short(x))
2225 self.ui.debug("add changeset %s\n" % short(x))
2226 return len(cl)
2226 return len(cl)
2227
2227
2228 def revmap(x):
2228 def revmap(x):
2229 return cl.rev(x)
2229 return cl.rev(x)
2230
2230
2231 if not source:
2231 if not source:
2232 return 0
2232 return 0
2233
2233
2234 self.hook('prechangegroup', throw=True, source=srctype, url=url)
2234 self.hook('prechangegroup', throw=True, source=srctype, url=url)
2235
2235
2236 changesets = files = revisions = 0
2236 changesets = files = revisions = 0
2237 efiles = set()
2237 efiles = set()
2238
2238
2239 # write changelog data to temp files so concurrent readers will not see
2239 # write changelog data to temp files so concurrent readers will not see
2240 # inconsistent view
2240 # inconsistent view
2241 cl = self.changelog
2241 cl = self.changelog
2242 cl.delayupdate()
2242 cl.delayupdate()
2243 oldheads = cl.heads()
2243 oldheads = cl.heads()
2244
2244
2245 tr = self.transaction("\n".join([srctype, util.hidepassword(url)]))
2245 tr = self.transaction("\n".join([srctype, util.hidepassword(url)]))
2246 try:
2246 try:
2247 trp = weakref.proxy(tr)
2247 trp = weakref.proxy(tr)
2248 # pull off the changeset group
2248 # pull off the changeset group
2249 self.ui.status(_("adding changesets\n"))
2249 self.ui.status(_("adding changesets\n"))
2250 clstart = len(cl)
2250 clstart = len(cl)
2251 class prog(object):
2251 class prog(object):
2252 step = _('changesets')
2252 step = _('changesets')
2253 count = 1
2253 count = 1
2254 ui = self.ui
2254 ui = self.ui
2255 total = None
2255 total = None
2256 def __call__(self):
2256 def __call__(self):
2257 self.ui.progress(self.step, self.count, unit=_('chunks'),
2257 self.ui.progress(self.step, self.count, unit=_('chunks'),
2258 total=self.total)
2258 total=self.total)
2259 self.count += 1
2259 self.count += 1
2260 pr = prog()
2260 pr = prog()
2261 source.callback = pr
2261 source.callback = pr
2262
2262
2263 source.changelogheader()
2263 source.changelogheader()
2264 srccontent = cl.addgroup(source, csmap, trp)
2264 srccontent = cl.addgroup(source, csmap, trp)
2265 if not (srccontent or emptyok):
2265 if not (srccontent or emptyok):
2266 raise util.Abort(_("received changelog group is empty"))
2266 raise util.Abort(_("received changelog group is empty"))
2267 clend = len(cl)
2267 clend = len(cl)
2268 changesets = clend - clstart
2268 changesets = clend - clstart
2269 for c in xrange(clstart, clend):
2269 for c in xrange(clstart, clend):
2270 efiles.update(self[c].files())
2270 efiles.update(self[c].files())
2271 efiles = len(efiles)
2271 efiles = len(efiles)
2272 self.ui.progress(_('changesets'), None)
2272 self.ui.progress(_('changesets'), None)
2273
2273
2274 # pull off the manifest group
2274 # pull off the manifest group
2275 self.ui.status(_("adding manifests\n"))
2275 self.ui.status(_("adding manifests\n"))
2276 pr.step = _('manifests')
2276 pr.step = _('manifests')
2277 pr.count = 1
2277 pr.count = 1
2278 pr.total = changesets # manifests <= changesets
2278 pr.total = changesets # manifests <= changesets
2279 # no need to check for empty manifest group here:
2279 # no need to check for empty manifest group here:
2280 # if the result of the merge of 1 and 2 is the same in 3 and 4,
2280 # if the result of the merge of 1 and 2 is the same in 3 and 4,
2281 # no new manifest will be created and the manifest group will
2281 # no new manifest will be created and the manifest group will
2282 # be empty during the pull
2282 # be empty during the pull
2283 source.manifestheader()
2283 source.manifestheader()
2284 self.manifest.addgroup(source, revmap, trp)
2284 self.manifest.addgroup(source, revmap, trp)
2285 self.ui.progress(_('manifests'), None)
2285 self.ui.progress(_('manifests'), None)
2286
2286
2287 needfiles = {}
2287 needfiles = {}
2288 if self.ui.configbool('server', 'validate', default=False):
2288 if self.ui.configbool('server', 'validate', default=False):
2289 # validate incoming csets have their manifests
2289 # validate incoming csets have their manifests
2290 for cset in xrange(clstart, clend):
2290 for cset in xrange(clstart, clend):
2291 mfest = self.changelog.read(self.changelog.node(cset))[0]
2291 mfest = self.changelog.read(self.changelog.node(cset))[0]
2292 mfest = self.manifest.readdelta(mfest)
2292 mfest = self.manifest.readdelta(mfest)
2293 # store file nodes we must see
2293 # store file nodes we must see
2294 for f, n in mfest.iteritems():
2294 for f, n in mfest.iteritems():
2295 needfiles.setdefault(f, set()).add(n)
2295 needfiles.setdefault(f, set()).add(n)
2296
2296
2297 # process the files
2297 # process the files
2298 self.ui.status(_("adding file changes\n"))
2298 self.ui.status(_("adding file changes\n"))
2299 pr.step = _('files')
2299 pr.step = _('files')
2300 pr.count = 1
2300 pr.count = 1
2301 pr.total = efiles
2301 pr.total = efiles
2302 source.callback = None
2302 source.callback = None
2303
2303
2304 while True:
2304 while True:
2305 chunkdata = source.filelogheader()
2305 chunkdata = source.filelogheader()
2306 if not chunkdata:
2306 if not chunkdata:
2307 break
2307 break
2308 f = chunkdata["filename"]
2308 f = chunkdata["filename"]
2309 self.ui.debug("adding %s revisions\n" % f)
2309 self.ui.debug("adding %s revisions\n" % f)
2310 pr()
2310 pr()
2311 fl = self.file(f)
2311 fl = self.file(f)
2312 o = len(fl)
2312 o = len(fl)
2313 if not fl.addgroup(source, revmap, trp):
2313 if not fl.addgroup(source, revmap, trp):
2314 raise util.Abort(_("received file revlog group is empty"))
2314 raise util.Abort(_("received file revlog group is empty"))
2315 revisions += len(fl) - o
2315 revisions += len(fl) - o
2316 files += 1
2316 files += 1
2317 if f in needfiles:
2317 if f in needfiles:
2318 needs = needfiles[f]
2318 needs = needfiles[f]
2319 for new in xrange(o, len(fl)):
2319 for new in xrange(o, len(fl)):
2320 n = fl.node(new)
2320 n = fl.node(new)
2321 if n in needs:
2321 if n in needs:
2322 needs.remove(n)
2322 needs.remove(n)
2323 else:
2323 else:
2324 raise util.Abort(
2324 raise util.Abort(
2325 _("received spurious file revlog entry"))
2325 _("received spurious file revlog entry"))
2326 if not needs:
2326 if not needs:
2327 del needfiles[f]
2327 del needfiles[f]
2328 self.ui.progress(_('files'), None)
2328 self.ui.progress(_('files'), None)
2329
2329
2330 for f, needs in needfiles.iteritems():
2330 for f, needs in needfiles.iteritems():
2331 fl = self.file(f)
2331 fl = self.file(f)
2332 for n in needs:
2332 for n in needs:
2333 try:
2333 try:
2334 fl.rev(n)
2334 fl.rev(n)
2335 except error.LookupError:
2335 except error.LookupError:
2336 raise util.Abort(
2336 raise util.Abort(
2337 _('missing file data for %s:%s - run hg verify') %
2337 _('missing file data for %s:%s - run hg verify') %
2338 (f, hex(n)))
2338 (f, hex(n)))
2339
2339
2340 dh = 0
2340 dh = 0
2341 if oldheads:
2341 if oldheads:
2342 heads = cl.heads()
2342 heads = cl.heads()
2343 dh = len(heads) - len(oldheads)
2343 dh = len(heads) - len(oldheads)
2344 for h in heads:
2344 for h in heads:
2345 if h not in oldheads and self[h].closesbranch():
2345 if h not in oldheads and self[h].closesbranch():
2346 dh -= 1
2346 dh -= 1
2347 htext = ""
2347 htext = ""
2348 if dh:
2348 if dh:
2349 htext = _(" (%+d heads)") % dh
2349 htext = _(" (%+d heads)") % dh
2350
2350
2351 self.ui.status(_("added %d changesets"
2351 self.ui.status(_("added %d changesets"
2352 " with %d changes to %d files%s\n")
2352 " with %d changes to %d files%s\n")
2353 % (changesets, revisions, files, htext))
2353 % (changesets, revisions, files, htext))
2354 self.invalidatevolatilesets()
2354 self.invalidatevolatilesets()
2355
2355
2356 if changesets > 0:
2356 if changesets > 0:
2357 p = lambda: cl.writepending() and self.root or ""
2357 p = lambda: cl.writepending() and self.root or ""
2358 self.hook('pretxnchangegroup', throw=True,
2358 self.hook('pretxnchangegroup', throw=True,
2359 node=hex(cl.node(clstart)), source=srctype,
2359 node=hex(cl.node(clstart)), source=srctype,
2360 url=url, pending=p)
2360 url=url, pending=p)
2361
2361
2362 added = [cl.node(r) for r in xrange(clstart, clend)]
2362 added = [cl.node(r) for r in xrange(clstart, clend)]
2363 publishing = self.ui.configbool('phases', 'publish', True)
2363 publishing = self.ui.configbool('phases', 'publish', True)
2364 if srctype == 'push':
2364 if srctype == 'push':
2365 # Old server can not push the boundary themself.
2365 # Old server can not push the boundary themself.
2366 # New server won't push the boundary if changeset already
2366 # New server won't push the boundary if changeset already
2367 # existed locally as secrete
2367 # existed locally as secrete
2368 #
2368 #
2369 # We should not use added here but the list of all change in
2369 # We should not use added here but the list of all change in
2370 # the bundle
2370 # the bundle
2371 if publishing:
2371 if publishing:
2372 phases.advanceboundary(self, phases.public, srccontent)
2372 phases.advanceboundary(self, phases.public, srccontent)
2373 else:
2373 else:
2374 phases.advanceboundary(self, phases.draft, srccontent)
2374 phases.advanceboundary(self, phases.draft, srccontent)
2375 phases.retractboundary(self, phases.draft, added)
2375 phases.retractboundary(self, phases.draft, added)
2376 elif srctype != 'strip':
2376 elif srctype != 'strip':
2377 # publishing only alter behavior during push
2377 # publishing only alter behavior during push
2378 #
2378 #
2379 # strip should not touch boundary at all
2379 # strip should not touch boundary at all
2380 phases.retractboundary(self, phases.draft, added)
2380 phases.retractboundary(self, phases.draft, added)
2381
2381
2382 # make changelog see real files again
2382 # make changelog see real files again
2383 cl.finalize(trp)
2383 cl.finalize(trp)
2384
2384
2385 tr.close()
2385 tr.close()
2386
2386
2387 if changesets > 0:
2387 if changesets > 0:
2388 if srctype != 'strip':
2388 if srctype != 'strip':
2389 # During strip, branchcache is invalid but coming call to
2389 # During strip, branchcache is invalid but coming call to
2390 # `destroyed` will repair it.
2390 # `destroyed` will repair it.
2391 # In other case we can safely update cache on disk.
2391 # In other case we can safely update cache on disk.
2392 branchmap.updatecache(self.filtered('served'))
2392 branchmap.updatecache(self.filtered('served'))
2393 def runhooks():
2393 def runhooks():
2394 # forcefully update the on-disk branch cache
2394 # forcefully update the on-disk branch cache
2395 self.ui.debug("updating the branch cache\n")
2395 self.ui.debug("updating the branch cache\n")
2396 self.hook("changegroup", node=hex(cl.node(clstart)),
2396 self.hook("changegroup", node=hex(cl.node(clstart)),
2397 source=srctype, url=url)
2397 source=srctype, url=url)
2398
2398
2399 for n in added:
2399 for n in added:
2400 self.hook("incoming", node=hex(n), source=srctype,
2400 self.hook("incoming", node=hex(n), source=srctype,
2401 url=url)
2401 url=url)
2402
2402
2403 heads = self.heads()
2403 newheads = [h for h in self.heads() if h not in oldheads]
2404 self.ui.log("incoming",
2404 self.ui.log("incoming",
2405 _("%s incoming changes - new heads: %s\n"),
2405 _("%s incoming changes - new heads: %s\n"),
2406 len(added),
2406 len(added),
2407 ', '.join([hex(c[:6]) for c in heads]))
2407 ', '.join([hex(c[:6]) for c in newheads]))
2408 self._afterlock(runhooks)
2408 self._afterlock(runhooks)
2409
2409
2410 finally:
2410 finally:
2411 tr.release()
2411 tr.release()
2412 # never return 0 here:
2412 # never return 0 here:
2413 if dh < 0:
2413 if dh < 0:
2414 return dh - 1
2414 return dh - 1
2415 else:
2415 else:
2416 return dh + 1
2416 return dh + 1
2417
2417
2418 def stream_in(self, remote, requirements):
2418 def stream_in(self, remote, requirements):
2419 lock = self.lock()
2419 lock = self.lock()
2420 try:
2420 try:
2421 # Save remote branchmap. We will use it later
2421 # Save remote branchmap. We will use it later
2422 # to speed up branchcache creation
2422 # to speed up branchcache creation
2423 rbranchmap = None
2423 rbranchmap = None
2424 if remote.capable("branchmap"):
2424 if remote.capable("branchmap"):
2425 rbranchmap = remote.branchmap()
2425 rbranchmap = remote.branchmap()
2426
2426
2427 fp = remote.stream_out()
2427 fp = remote.stream_out()
2428 l = fp.readline()
2428 l = fp.readline()
2429 try:
2429 try:
2430 resp = int(l)
2430 resp = int(l)
2431 except ValueError:
2431 except ValueError:
2432 raise error.ResponseError(
2432 raise error.ResponseError(
2433 _('unexpected response from remote server:'), l)
2433 _('unexpected response from remote server:'), l)
2434 if resp == 1:
2434 if resp == 1:
2435 raise util.Abort(_('operation forbidden by server'))
2435 raise util.Abort(_('operation forbidden by server'))
2436 elif resp == 2:
2436 elif resp == 2:
2437 raise util.Abort(_('locking the remote repository failed'))
2437 raise util.Abort(_('locking the remote repository failed'))
2438 elif resp != 0:
2438 elif resp != 0:
2439 raise util.Abort(_('the server sent an unknown error code'))
2439 raise util.Abort(_('the server sent an unknown error code'))
2440 self.ui.status(_('streaming all changes\n'))
2440 self.ui.status(_('streaming all changes\n'))
2441 l = fp.readline()
2441 l = fp.readline()
2442 try:
2442 try:
2443 total_files, total_bytes = map(int, l.split(' ', 1))
2443 total_files, total_bytes = map(int, l.split(' ', 1))
2444 except (ValueError, TypeError):
2444 except (ValueError, TypeError):
2445 raise error.ResponseError(
2445 raise error.ResponseError(
2446 _('unexpected response from remote server:'), l)
2446 _('unexpected response from remote server:'), l)
2447 self.ui.status(_('%d files to transfer, %s of data\n') %
2447 self.ui.status(_('%d files to transfer, %s of data\n') %
2448 (total_files, util.bytecount(total_bytes)))
2448 (total_files, util.bytecount(total_bytes)))
2449 handled_bytes = 0
2449 handled_bytes = 0
2450 self.ui.progress(_('clone'), 0, total=total_bytes)
2450 self.ui.progress(_('clone'), 0, total=total_bytes)
2451 start = time.time()
2451 start = time.time()
2452 for i in xrange(total_files):
2452 for i in xrange(total_files):
2453 # XXX doesn't support '\n' or '\r' in filenames
2453 # XXX doesn't support '\n' or '\r' in filenames
2454 l = fp.readline()
2454 l = fp.readline()
2455 try:
2455 try:
2456 name, size = l.split('\0', 1)
2456 name, size = l.split('\0', 1)
2457 size = int(size)
2457 size = int(size)
2458 except (ValueError, TypeError):
2458 except (ValueError, TypeError):
2459 raise error.ResponseError(
2459 raise error.ResponseError(
2460 _('unexpected response from remote server:'), l)
2460 _('unexpected response from remote server:'), l)
2461 if self.ui.debugflag:
2461 if self.ui.debugflag:
2462 self.ui.debug('adding %s (%s)\n' %
2462 self.ui.debug('adding %s (%s)\n' %
2463 (name, util.bytecount(size)))
2463 (name, util.bytecount(size)))
2464 # for backwards compat, name was partially encoded
2464 # for backwards compat, name was partially encoded
2465 ofp = self.sopener(store.decodedir(name), 'w')
2465 ofp = self.sopener(store.decodedir(name), 'w')
2466 for chunk in util.filechunkiter(fp, limit=size):
2466 for chunk in util.filechunkiter(fp, limit=size):
2467 handled_bytes += len(chunk)
2467 handled_bytes += len(chunk)
2468 self.ui.progress(_('clone'), handled_bytes,
2468 self.ui.progress(_('clone'), handled_bytes,
2469 total=total_bytes)
2469 total=total_bytes)
2470 ofp.write(chunk)
2470 ofp.write(chunk)
2471 ofp.close()
2471 ofp.close()
2472 elapsed = time.time() - start
2472 elapsed = time.time() - start
2473 if elapsed <= 0:
2473 if elapsed <= 0:
2474 elapsed = 0.001
2474 elapsed = 0.001
2475 self.ui.progress(_('clone'), None)
2475 self.ui.progress(_('clone'), None)
2476 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
2476 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
2477 (util.bytecount(total_bytes), elapsed,
2477 (util.bytecount(total_bytes), elapsed,
2478 util.bytecount(total_bytes / elapsed)))
2478 util.bytecount(total_bytes / elapsed)))
2479
2479
2480 # new requirements = old non-format requirements +
2480 # new requirements = old non-format requirements +
2481 # new format-related
2481 # new format-related
2482 # requirements from the streamed-in repository
2482 # requirements from the streamed-in repository
2483 requirements.update(set(self.requirements) - self.supportedformats)
2483 requirements.update(set(self.requirements) - self.supportedformats)
2484 self._applyrequirements(requirements)
2484 self._applyrequirements(requirements)
2485 self._writerequirements()
2485 self._writerequirements()
2486
2486
2487 if rbranchmap:
2487 if rbranchmap:
2488 rbheads = []
2488 rbheads = []
2489 for bheads in rbranchmap.itervalues():
2489 for bheads in rbranchmap.itervalues():
2490 rbheads.extend(bheads)
2490 rbheads.extend(bheads)
2491
2491
2492 if rbheads:
2492 if rbheads:
2493 rtiprev = max((int(self.changelog.rev(node))
2493 rtiprev = max((int(self.changelog.rev(node))
2494 for node in rbheads))
2494 for node in rbheads))
2495 cache = branchmap.branchcache(rbranchmap,
2495 cache = branchmap.branchcache(rbranchmap,
2496 self[rtiprev].node(),
2496 self[rtiprev].node(),
2497 rtiprev)
2497 rtiprev)
2498 # Try to stick it as low as possible
2498 # Try to stick it as low as possible
2499 # filter above served are unlikely to be fetch from a clone
2499 # filter above served are unlikely to be fetch from a clone
2500 for candidate in ('base', 'immutable', 'served'):
2500 for candidate in ('base', 'immutable', 'served'):
2501 rview = self.filtered(candidate)
2501 rview = self.filtered(candidate)
2502 if cache.validfor(rview):
2502 if cache.validfor(rview):
2503 self._branchcaches[candidate] = cache
2503 self._branchcaches[candidate] = cache
2504 cache.write(rview)
2504 cache.write(rview)
2505 break
2505 break
2506 self.invalidate()
2506 self.invalidate()
2507 return len(self.heads()) + 1
2507 return len(self.heads()) + 1
2508 finally:
2508 finally:
2509 lock.release()
2509 lock.release()
2510
2510
2511 def clone(self, remote, heads=[], stream=False):
2511 def clone(self, remote, heads=[], stream=False):
2512 '''clone remote repository.
2512 '''clone remote repository.
2513
2513
2514 keyword arguments:
2514 keyword arguments:
2515 heads: list of revs to clone (forces use of pull)
2515 heads: list of revs to clone (forces use of pull)
2516 stream: use streaming clone if possible'''
2516 stream: use streaming clone if possible'''
2517
2517
2518 # now, all clients that can request uncompressed clones can
2518 # now, all clients that can request uncompressed clones can
2519 # read repo formats supported by all servers that can serve
2519 # read repo formats supported by all servers that can serve
2520 # them.
2520 # them.
2521
2521
2522 # if revlog format changes, client will have to check version
2522 # if revlog format changes, client will have to check version
2523 # and format flags on "stream" capability, and use
2523 # and format flags on "stream" capability, and use
2524 # uncompressed only if compatible.
2524 # uncompressed only if compatible.
2525
2525
2526 if not stream:
2526 if not stream:
2527 # if the server explicitly prefers to stream (for fast LANs)
2527 # if the server explicitly prefers to stream (for fast LANs)
2528 stream = remote.capable('stream-preferred')
2528 stream = remote.capable('stream-preferred')
2529
2529
2530 if stream and not heads:
2530 if stream and not heads:
2531 # 'stream' means remote revlog format is revlogv1 only
2531 # 'stream' means remote revlog format is revlogv1 only
2532 if remote.capable('stream'):
2532 if remote.capable('stream'):
2533 return self.stream_in(remote, set(('revlogv1',)))
2533 return self.stream_in(remote, set(('revlogv1',)))
2534 # otherwise, 'streamreqs' contains the remote revlog format
2534 # otherwise, 'streamreqs' contains the remote revlog format
2535 streamreqs = remote.capable('streamreqs')
2535 streamreqs = remote.capable('streamreqs')
2536 if streamreqs:
2536 if streamreqs:
2537 streamreqs = set(streamreqs.split(','))
2537 streamreqs = set(streamreqs.split(','))
2538 # if we support it, stream in and adjust our requirements
2538 # if we support it, stream in and adjust our requirements
2539 if not streamreqs - self.supportedformats:
2539 if not streamreqs - self.supportedformats:
2540 return self.stream_in(remote, streamreqs)
2540 return self.stream_in(remote, streamreqs)
2541 return self.pull(remote, heads)
2541 return self.pull(remote, heads)
2542
2542
2543 def pushkey(self, namespace, key, old, new):
2543 def pushkey(self, namespace, key, old, new):
2544 self.hook('prepushkey', throw=True, namespace=namespace, key=key,
2544 self.hook('prepushkey', throw=True, namespace=namespace, key=key,
2545 old=old, new=new)
2545 old=old, new=new)
2546 self.ui.debug('pushing key for "%s:%s"\n' % (namespace, key))
2546 self.ui.debug('pushing key for "%s:%s"\n' % (namespace, key))
2547 ret = pushkey.push(self, namespace, key, old, new)
2547 ret = pushkey.push(self, namespace, key, old, new)
2548 self.hook('pushkey', namespace=namespace, key=key, old=old, new=new,
2548 self.hook('pushkey', namespace=namespace, key=key, old=old, new=new,
2549 ret=ret)
2549 ret=ret)
2550 return ret
2550 return ret
2551
2551
2552 def listkeys(self, namespace):
2552 def listkeys(self, namespace):
2553 self.hook('prelistkeys', throw=True, namespace=namespace)
2553 self.hook('prelistkeys', throw=True, namespace=namespace)
2554 self.ui.debug('listing keys for "%s"\n' % namespace)
2554 self.ui.debug('listing keys for "%s"\n' % namespace)
2555 values = pushkey.list(self, namespace)
2555 values = pushkey.list(self, namespace)
2556 self.hook('listkeys', namespace=namespace, values=values)
2556 self.hook('listkeys', namespace=namespace, values=values)
2557 return values
2557 return values
2558
2558
2559 def debugwireargs(self, one, two, three=None, four=None, five=None):
2559 def debugwireargs(self, one, two, three=None, four=None, five=None):
2560 '''used to test argument passing over the wire'''
2560 '''used to test argument passing over the wire'''
2561 return "%s %s %s %s %s" % (one, two, three, four, five)
2561 return "%s %s %s %s %s" % (one, two, three, four, five)
2562
2562
2563 def savecommitmessage(self, text):
2563 def savecommitmessage(self, text):
2564 fp = self.opener('last-message.txt', 'wb')
2564 fp = self.opener('last-message.txt', 'wb')
2565 try:
2565 try:
2566 fp.write(text)
2566 fp.write(text)
2567 finally:
2567 finally:
2568 fp.close()
2568 fp.close()
2569 return self.pathto(fp.name[len(self.root) + 1:])
2569 return self.pathto(fp.name[len(self.root) + 1:])
2570
2570
2571 # used to avoid circular references so destructors work
2571 # used to avoid circular references so destructors work
2572 def aftertrans(files):
2572 def aftertrans(files):
2573 renamefiles = [tuple(t) for t in files]
2573 renamefiles = [tuple(t) for t in files]
2574 def a():
2574 def a():
2575 for src, dest in renamefiles:
2575 for src, dest in renamefiles:
2576 try:
2576 try:
2577 util.rename(src, dest)
2577 util.rename(src, dest)
2578 except OSError: # journal file does not yet exist
2578 except OSError: # journal file does not yet exist
2579 pass
2579 pass
2580 return a
2580 return a
2581
2581
2582 def undoname(fn):
2582 def undoname(fn):
2583 base, name = os.path.split(fn)
2583 base, name = os.path.split(fn)
2584 assert name.startswith('journal')
2584 assert name.startswith('journal')
2585 return os.path.join(base, name.replace('journal', 'undo', 1))
2585 return os.path.join(base, name.replace('journal', 'undo', 1))
2586
2586
2587 def instance(ui, path, create):
2587 def instance(ui, path, create):
2588 return localrepository(ui, util.urllocalpath(path), create)
2588 return localrepository(ui, util.urllocalpath(path), create)
2589
2589
2590 def islocal(path):
2590 def islocal(path):
2591 return True
2591 return True
@@ -1,67 +1,79 b''
1 setup
1 setup
2 $ cat > mock.py <<EOF
2 $ cat > mock.py <<EOF
3 > from mercurial import util
3 > from mercurial import util
4 > import getpass
4 > import getpass
5 >
5 >
6 > def makedate():
6 > def makedate():
7 > return 0, 0
7 > return 0, 0
8 > def getuser():
8 > def getuser():
9 > return 'bob'
9 > return 'bob'
10 > # mock the date and user apis so the output is always the same
10 > # mock the date and user apis so the output is always the same
11 > def uisetup(ui):
11 > def uisetup(ui):
12 > util.makedate = makedate
12 > util.makedate = makedate
13 > getpass.getuser = getuser
13 > getpass.getuser = getuser
14 > EOF
14 > EOF
15 $ cat >> $HGRCPATH <<EOF
15 $ cat >> $HGRCPATH <<EOF
16 > [extensions]
16 > [extensions]
17 > blackbox=
17 > blackbox=
18 > mock=`pwd`/mock.py
18 > mock=`pwd`/mock.py
19 > EOF
19 > EOF
20 $ hg init blackboxtest
20 $ hg init blackboxtest
21 $ cd blackboxtest
21 $ cd blackboxtest
22
22
23 command, exit codes, and duration
23 command, exit codes, and duration
24
24
25 $ echo a > a
25 $ echo a > a
26 $ hg add a
26 $ hg add a
27 $ hg blackbox
27 $ hg blackbox
28 1970/01/01 00:00:00 bob> add a
28 1970/01/01 00:00:00 bob> add a
29 1970/01/01 00:00:00 bob> add exited 0 after * seconds (glob)
29 1970/01/01 00:00:00 bob> add exited 0 after * seconds (glob)
30
30
31 incoming change tracking
32
33 create two heads to verify that we only see one change in the log later
34 $ hg commit -ma
35 $ hg up null
36 0 files updated, 0 files merged, 1 files removed, 0 files unresolved
37 $ echo b > b
38 $ hg commit -Amb
39 adding b
40 created new head
41
42 clone, commit, pull
43 $ hg clone . ../blackboxtest2
44 updating to branch default
45 1 files updated, 0 files merged, 0 files removed, 0 files unresolved
46 $ echo c > c
47 $ hg commit -Amc
48 adding c
49 $ cd ../blackboxtest2
50 $ hg pull
51 pulling from $TESTTMP/blackboxtest
52 searching for changes
53 adding changesets
54 adding manifests
55 adding file changes
56 added 1 changesets with 1 changes to 1 files
57 (run 'hg update' to get a working copy)
58 $ hg blackbox -l 3
59 1970/01/01 00:00:00 bob> pull
60 1970/01/01 00:00:00 bob> 1 incoming changes - new heads: d02f48003e62 (glob)
61 1970/01/01 00:00:00 bob> pull exited None after * seconds (glob)
62
31 extension and python hooks - use the eol extension for a pythonhook
63 extension and python hooks - use the eol extension for a pythonhook
32
64
33 $ echo '[extensions]' >> .hg/hgrc
65 $ echo '[extensions]' >> .hg/hgrc
34 $ echo 'eol=' >> .hg/hgrc
66 $ echo 'eol=' >> .hg/hgrc
35 $ echo '[hooks]' >> .hg/hgrc
67 $ echo '[hooks]' >> .hg/hgrc
36 $ echo 'update = echo hooked' >> .hg/hgrc
68 $ echo 'update = echo hooked' >> .hg/hgrc
37 $ hg update
69 $ hg update
38 hooked
70 hooked
39 0 files updated, 0 files merged, 0 files removed, 0 files unresolved
71 1 files updated, 0 files merged, 0 files removed, 0 files unresolved
40 $ hg blackbox -l 4
72 $ hg blackbox -l 4
41 1970/01/01 00:00:00 bob> update
73 1970/01/01 00:00:00 bob> update
42 1970/01/01 00:00:00 bob> pythonhook-preupdate: hgext.eol.preupdate finished in * seconds (glob)
74 1970/01/01 00:00:00 bob> pythonhook-preupdate: hgext.eol.preupdate finished in * seconds (glob)
43 1970/01/01 00:00:00 bob> exthook-update: echo hooked finished in * seconds (glob)
75 1970/01/01 00:00:00 bob> exthook-update: echo hooked finished in * seconds (glob)
44 1970/01/01 00:00:00 bob> update exited False after * seconds (glob)
76 1970/01/01 00:00:00 bob> update exited False after * seconds (glob)
45
77
46 incoming change tracking
47
48 $ hg clone . ../blackboxtest2
49 updating to branch default
50 0 files updated, 0 files merged, 0 files removed, 0 files unresolved
51 $ hg commit -ma
52 $ cd ../blackboxtest2
53 $ hg pull
54 pulling from $TESTTMP/blackboxtest
55 requesting all changes
56 adding changesets
57 adding manifests
58 adding file changes
59 added 1 changesets with 1 changes to 1 files
60 (run 'hg update' to get a working copy)
61 $ hg blackbox -l 3
62 1970/01/01 00:00:00 bob> pull
63 1970/01/01 00:00:00 bob> 1 incoming changes - new heads: cb9a9f314b8b (glob)
64 1970/01/01 00:00:00 bob> pull exited None after * seconds (glob)
65
66 cleanup
78 cleanup
67 $ cd ..
79 $ cd ..
General Comments 0
You need to be logged in to leave comments. Login now