##// END OF EJS Templates
localrepo: make it possible to pass multiple path elements to join and wjoin...
Angel Ezquerra -
r22362:c9b32c6a default
parent child Browse files
Show More
@@ -1,1775 +1,1775 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 urllib
9 import urllib
10 import peer, changegroup, subrepo, pushkey, obsolete, repoview
10 import peer, changegroup, subrepo, pushkey, obsolete, repoview
11 import changelog, dirstate, filelog, manifest, context, bookmarks, phases
11 import changelog, dirstate, filelog, manifest, context, bookmarks, phases
12 import lock as lockmod
12 import lock as lockmod
13 import transaction, store, encoding, exchange, bundle2
13 import transaction, store, encoding, exchange, bundle2
14 import scmutil, util, extensions, hook, error, revset
14 import scmutil, util, extensions, hook, error, revset
15 import match as matchmod
15 import match as matchmod
16 import merge as mergemod
16 import merge as mergemod
17 import tags as tagsmod
17 import tags as tagsmod
18 from lock import release
18 from lock import release
19 import weakref, errno, os, time, inspect
19 import weakref, errno, os, time, inspect
20 import branchmap, pathutil
20 import branchmap, pathutil
21 propertycache = util.propertycache
21 propertycache = util.propertycache
22 filecache = scmutil.filecache
22 filecache = scmutil.filecache
23
23
24 class repofilecache(filecache):
24 class repofilecache(filecache):
25 """All filecache usage on repo are done for logic that should be unfiltered
25 """All filecache usage on repo are done for logic that should be unfiltered
26 """
26 """
27
27
28 def __get__(self, repo, type=None):
28 def __get__(self, repo, type=None):
29 return super(repofilecache, self).__get__(repo.unfiltered(), type)
29 return super(repofilecache, self).__get__(repo.unfiltered(), type)
30 def __set__(self, repo, value):
30 def __set__(self, repo, value):
31 return super(repofilecache, self).__set__(repo.unfiltered(), value)
31 return super(repofilecache, self).__set__(repo.unfiltered(), value)
32 def __delete__(self, repo):
32 def __delete__(self, repo):
33 return super(repofilecache, self).__delete__(repo.unfiltered())
33 return super(repofilecache, self).__delete__(repo.unfiltered())
34
34
35 class storecache(repofilecache):
35 class storecache(repofilecache):
36 """filecache for files in the store"""
36 """filecache for files in the store"""
37 def join(self, obj, fname):
37 def join(self, obj, fname):
38 return obj.sjoin(fname)
38 return obj.sjoin(fname)
39
39
40 class unfilteredpropertycache(propertycache):
40 class unfilteredpropertycache(propertycache):
41 """propertycache that apply to unfiltered repo only"""
41 """propertycache that apply to unfiltered repo only"""
42
42
43 def __get__(self, repo, type=None):
43 def __get__(self, repo, type=None):
44 unfi = repo.unfiltered()
44 unfi = repo.unfiltered()
45 if unfi is repo:
45 if unfi is repo:
46 return super(unfilteredpropertycache, self).__get__(unfi)
46 return super(unfilteredpropertycache, self).__get__(unfi)
47 return getattr(unfi, self.name)
47 return getattr(unfi, self.name)
48
48
49 class filteredpropertycache(propertycache):
49 class filteredpropertycache(propertycache):
50 """propertycache that must take filtering in account"""
50 """propertycache that must take filtering in account"""
51
51
52 def cachevalue(self, obj, value):
52 def cachevalue(self, obj, value):
53 object.__setattr__(obj, self.name, value)
53 object.__setattr__(obj, self.name, value)
54
54
55
55
56 def hasunfilteredcache(repo, name):
56 def hasunfilteredcache(repo, name):
57 """check if a repo has an unfilteredpropertycache value for <name>"""
57 """check if a repo has an unfilteredpropertycache value for <name>"""
58 return name in vars(repo.unfiltered())
58 return name in vars(repo.unfiltered())
59
59
60 def unfilteredmethod(orig):
60 def unfilteredmethod(orig):
61 """decorate method that always need to be run on unfiltered version"""
61 """decorate method that always need to be run on unfiltered version"""
62 def wrapper(repo, *args, **kwargs):
62 def wrapper(repo, *args, **kwargs):
63 return orig(repo.unfiltered(), *args, **kwargs)
63 return orig(repo.unfiltered(), *args, **kwargs)
64 return wrapper
64 return wrapper
65
65
66 moderncaps = set(('lookup', 'branchmap', 'pushkey', 'known', 'getbundle',
66 moderncaps = set(('lookup', 'branchmap', 'pushkey', 'known', 'getbundle',
67 'unbundle'))
67 'unbundle'))
68 legacycaps = moderncaps.union(set(['changegroupsubset']))
68 legacycaps = moderncaps.union(set(['changegroupsubset']))
69
69
70 class localpeer(peer.peerrepository):
70 class localpeer(peer.peerrepository):
71 '''peer for a local repo; reflects only the most recent API'''
71 '''peer for a local repo; reflects only the most recent API'''
72
72
73 def __init__(self, repo, caps=moderncaps):
73 def __init__(self, repo, caps=moderncaps):
74 peer.peerrepository.__init__(self)
74 peer.peerrepository.__init__(self)
75 self._repo = repo.filtered('served')
75 self._repo = repo.filtered('served')
76 self.ui = repo.ui
76 self.ui = repo.ui
77 self._caps = repo._restrictcapabilities(caps)
77 self._caps = repo._restrictcapabilities(caps)
78 self.requirements = repo.requirements
78 self.requirements = repo.requirements
79 self.supportedformats = repo.supportedformats
79 self.supportedformats = repo.supportedformats
80
80
81 def close(self):
81 def close(self):
82 self._repo.close()
82 self._repo.close()
83
83
84 def _capabilities(self):
84 def _capabilities(self):
85 return self._caps
85 return self._caps
86
86
87 def local(self):
87 def local(self):
88 return self._repo
88 return self._repo
89
89
90 def canpush(self):
90 def canpush(self):
91 return True
91 return True
92
92
93 def url(self):
93 def url(self):
94 return self._repo.url()
94 return self._repo.url()
95
95
96 def lookup(self, key):
96 def lookup(self, key):
97 return self._repo.lookup(key)
97 return self._repo.lookup(key)
98
98
99 def branchmap(self):
99 def branchmap(self):
100 return self._repo.branchmap()
100 return self._repo.branchmap()
101
101
102 def heads(self):
102 def heads(self):
103 return self._repo.heads()
103 return self._repo.heads()
104
104
105 def known(self, nodes):
105 def known(self, nodes):
106 return self._repo.known(nodes)
106 return self._repo.known(nodes)
107
107
108 def getbundle(self, source, heads=None, common=None, bundlecaps=None,
108 def getbundle(self, source, heads=None, common=None, bundlecaps=None,
109 format='HG10', **kwargs):
109 format='HG10', **kwargs):
110 cg = exchange.getbundle(self._repo, source, heads=heads,
110 cg = exchange.getbundle(self._repo, source, heads=heads,
111 common=common, bundlecaps=bundlecaps, **kwargs)
111 common=common, bundlecaps=bundlecaps, **kwargs)
112 if bundlecaps is not None and 'HG2X' in bundlecaps:
112 if bundlecaps is not None and 'HG2X' in bundlecaps:
113 # When requesting a bundle2, getbundle returns a stream to make the
113 # When requesting a bundle2, getbundle returns a stream to make the
114 # wire level function happier. We need to build a proper object
114 # wire level function happier. We need to build a proper object
115 # from it in local peer.
115 # from it in local peer.
116 cg = bundle2.unbundle20(self.ui, cg)
116 cg = bundle2.unbundle20(self.ui, cg)
117 return cg
117 return cg
118
118
119 # TODO We might want to move the next two calls into legacypeer and add
119 # TODO We might want to move the next two calls into legacypeer and add
120 # unbundle instead.
120 # unbundle instead.
121
121
122 def unbundle(self, cg, heads, url):
122 def unbundle(self, cg, heads, url):
123 """apply a bundle on a repo
123 """apply a bundle on a repo
124
124
125 This function handles the repo locking itself."""
125 This function handles the repo locking itself."""
126 try:
126 try:
127 cg = exchange.readbundle(self.ui, cg, None)
127 cg = exchange.readbundle(self.ui, cg, None)
128 ret = exchange.unbundle(self._repo, cg, heads, 'push', url)
128 ret = exchange.unbundle(self._repo, cg, heads, 'push', url)
129 if util.safehasattr(ret, 'getchunks'):
129 if util.safehasattr(ret, 'getchunks'):
130 # This is a bundle20 object, turn it into an unbundler.
130 # This is a bundle20 object, turn it into an unbundler.
131 # This little dance should be dropped eventually when the API
131 # This little dance should be dropped eventually when the API
132 # is finally improved.
132 # is finally improved.
133 stream = util.chunkbuffer(ret.getchunks())
133 stream = util.chunkbuffer(ret.getchunks())
134 ret = bundle2.unbundle20(self.ui, stream)
134 ret = bundle2.unbundle20(self.ui, stream)
135 return ret
135 return ret
136 except error.PushRaced, exc:
136 except error.PushRaced, exc:
137 raise error.ResponseError(_('push failed:'), str(exc))
137 raise error.ResponseError(_('push failed:'), str(exc))
138
138
139 def lock(self):
139 def lock(self):
140 return self._repo.lock()
140 return self._repo.lock()
141
141
142 def addchangegroup(self, cg, source, url):
142 def addchangegroup(self, cg, source, url):
143 return changegroup.addchangegroup(self._repo, cg, source, url)
143 return changegroup.addchangegroup(self._repo, cg, source, url)
144
144
145 def pushkey(self, namespace, key, old, new):
145 def pushkey(self, namespace, key, old, new):
146 return self._repo.pushkey(namespace, key, old, new)
146 return self._repo.pushkey(namespace, key, old, new)
147
147
148 def listkeys(self, namespace):
148 def listkeys(self, namespace):
149 return self._repo.listkeys(namespace)
149 return self._repo.listkeys(namespace)
150
150
151 def debugwireargs(self, one, two, three=None, four=None, five=None):
151 def debugwireargs(self, one, two, three=None, four=None, five=None):
152 '''used to test argument passing over the wire'''
152 '''used to test argument passing over the wire'''
153 return "%s %s %s %s %s" % (one, two, three, four, five)
153 return "%s %s %s %s %s" % (one, two, three, four, five)
154
154
155 class locallegacypeer(localpeer):
155 class locallegacypeer(localpeer):
156 '''peer extension which implements legacy methods too; used for tests with
156 '''peer extension which implements legacy methods too; used for tests with
157 restricted capabilities'''
157 restricted capabilities'''
158
158
159 def __init__(self, repo):
159 def __init__(self, repo):
160 localpeer.__init__(self, repo, caps=legacycaps)
160 localpeer.__init__(self, repo, caps=legacycaps)
161
161
162 def branches(self, nodes):
162 def branches(self, nodes):
163 return self._repo.branches(nodes)
163 return self._repo.branches(nodes)
164
164
165 def between(self, pairs):
165 def between(self, pairs):
166 return self._repo.between(pairs)
166 return self._repo.between(pairs)
167
167
168 def changegroup(self, basenodes, source):
168 def changegroup(self, basenodes, source):
169 return changegroup.changegroup(self._repo, basenodes, source)
169 return changegroup.changegroup(self._repo, basenodes, source)
170
170
171 def changegroupsubset(self, bases, heads, source):
171 def changegroupsubset(self, bases, heads, source):
172 return changegroup.changegroupsubset(self._repo, bases, heads, source)
172 return changegroup.changegroupsubset(self._repo, bases, heads, source)
173
173
174 class localrepository(object):
174 class localrepository(object):
175
175
176 supportedformats = set(('revlogv1', 'generaldelta'))
176 supportedformats = set(('revlogv1', 'generaldelta'))
177 _basesupported = supportedformats | set(('store', 'fncache', 'shared',
177 _basesupported = supportedformats | set(('store', 'fncache', 'shared',
178 'dotencode'))
178 'dotencode'))
179 openerreqs = set(('revlogv1', 'generaldelta'))
179 openerreqs = set(('revlogv1', 'generaldelta'))
180 requirements = ['revlogv1']
180 requirements = ['revlogv1']
181 filtername = None
181 filtername = None
182
182
183 # a list of (ui, featureset) functions.
183 # a list of (ui, featureset) functions.
184 # only functions defined in module of enabled extensions are invoked
184 # only functions defined in module of enabled extensions are invoked
185 featuresetupfuncs = set()
185 featuresetupfuncs = set()
186
186
187 def _baserequirements(self, create):
187 def _baserequirements(self, create):
188 return self.requirements[:]
188 return self.requirements[:]
189
189
190 def __init__(self, baseui, path=None, create=False):
190 def __init__(self, baseui, path=None, create=False):
191 self.wvfs = scmutil.vfs(path, expandpath=True, realpath=True)
191 self.wvfs = scmutil.vfs(path, expandpath=True, realpath=True)
192 self.wopener = self.wvfs
192 self.wopener = self.wvfs
193 self.root = self.wvfs.base
193 self.root = self.wvfs.base
194 self.path = self.wvfs.join(".hg")
194 self.path = self.wvfs.join(".hg")
195 self.origroot = path
195 self.origroot = path
196 self.auditor = pathutil.pathauditor(self.root, self._checknested)
196 self.auditor = pathutil.pathauditor(self.root, self._checknested)
197 self.vfs = scmutil.vfs(self.path)
197 self.vfs = scmutil.vfs(self.path)
198 self.opener = self.vfs
198 self.opener = self.vfs
199 self.baseui = baseui
199 self.baseui = baseui
200 self.ui = baseui.copy()
200 self.ui = baseui.copy()
201 self.ui.copy = baseui.copy # prevent copying repo configuration
201 self.ui.copy = baseui.copy # prevent copying repo configuration
202 # A list of callback to shape the phase if no data were found.
202 # A list of callback to shape the phase if no data were found.
203 # Callback are in the form: func(repo, roots) --> processed root.
203 # Callback are in the form: func(repo, roots) --> processed root.
204 # This list it to be filled by extension during repo setup
204 # This list it to be filled by extension during repo setup
205 self._phasedefaults = []
205 self._phasedefaults = []
206 try:
206 try:
207 self.ui.readconfig(self.join("hgrc"), self.root)
207 self.ui.readconfig(self.join("hgrc"), self.root)
208 extensions.loadall(self.ui)
208 extensions.loadall(self.ui)
209 except IOError:
209 except IOError:
210 pass
210 pass
211
211
212 if self.featuresetupfuncs:
212 if self.featuresetupfuncs:
213 self.supported = set(self._basesupported) # use private copy
213 self.supported = set(self._basesupported) # use private copy
214 extmods = set(m.__name__ for n, m
214 extmods = set(m.__name__ for n, m
215 in extensions.extensions(self.ui))
215 in extensions.extensions(self.ui))
216 for setupfunc in self.featuresetupfuncs:
216 for setupfunc in self.featuresetupfuncs:
217 if setupfunc.__module__ in extmods:
217 if setupfunc.__module__ in extmods:
218 setupfunc(self.ui, self.supported)
218 setupfunc(self.ui, self.supported)
219 else:
219 else:
220 self.supported = self._basesupported
220 self.supported = self._basesupported
221
221
222 if not self.vfs.isdir():
222 if not self.vfs.isdir():
223 if create:
223 if create:
224 if not self.wvfs.exists():
224 if not self.wvfs.exists():
225 self.wvfs.makedirs()
225 self.wvfs.makedirs()
226 self.vfs.makedir(notindexed=True)
226 self.vfs.makedir(notindexed=True)
227 requirements = self._baserequirements(create)
227 requirements = self._baserequirements(create)
228 if self.ui.configbool('format', 'usestore', True):
228 if self.ui.configbool('format', 'usestore', True):
229 self.vfs.mkdir("store")
229 self.vfs.mkdir("store")
230 requirements.append("store")
230 requirements.append("store")
231 if self.ui.configbool('format', 'usefncache', True):
231 if self.ui.configbool('format', 'usefncache', True):
232 requirements.append("fncache")
232 requirements.append("fncache")
233 if self.ui.configbool('format', 'dotencode', True):
233 if self.ui.configbool('format', 'dotencode', True):
234 requirements.append('dotencode')
234 requirements.append('dotencode')
235 # create an invalid changelog
235 # create an invalid changelog
236 self.vfs.append(
236 self.vfs.append(
237 "00changelog.i",
237 "00changelog.i",
238 '\0\0\0\2' # represents revlogv2
238 '\0\0\0\2' # represents revlogv2
239 ' dummy changelog to prevent using the old repo layout'
239 ' dummy changelog to prevent using the old repo layout'
240 )
240 )
241 if self.ui.configbool('format', 'generaldelta', False):
241 if self.ui.configbool('format', 'generaldelta', False):
242 requirements.append("generaldelta")
242 requirements.append("generaldelta")
243 requirements = set(requirements)
243 requirements = set(requirements)
244 else:
244 else:
245 raise error.RepoError(_("repository %s not found") % path)
245 raise error.RepoError(_("repository %s not found") % path)
246 elif create:
246 elif create:
247 raise error.RepoError(_("repository %s already exists") % path)
247 raise error.RepoError(_("repository %s already exists") % path)
248 else:
248 else:
249 try:
249 try:
250 requirements = scmutil.readrequires(self.vfs, self.supported)
250 requirements = scmutil.readrequires(self.vfs, self.supported)
251 except IOError, inst:
251 except IOError, inst:
252 if inst.errno != errno.ENOENT:
252 if inst.errno != errno.ENOENT:
253 raise
253 raise
254 requirements = set()
254 requirements = set()
255
255
256 self.sharedpath = self.path
256 self.sharedpath = self.path
257 try:
257 try:
258 vfs = scmutil.vfs(self.vfs.read("sharedpath").rstrip('\n'),
258 vfs = scmutil.vfs(self.vfs.read("sharedpath").rstrip('\n'),
259 realpath=True)
259 realpath=True)
260 s = vfs.base
260 s = vfs.base
261 if not vfs.exists():
261 if not vfs.exists():
262 raise error.RepoError(
262 raise error.RepoError(
263 _('.hg/sharedpath points to nonexistent directory %s') % s)
263 _('.hg/sharedpath points to nonexistent directory %s') % s)
264 self.sharedpath = s
264 self.sharedpath = s
265 except IOError, inst:
265 except IOError, inst:
266 if inst.errno != errno.ENOENT:
266 if inst.errno != errno.ENOENT:
267 raise
267 raise
268
268
269 self.store = store.store(requirements, self.sharedpath, scmutil.vfs)
269 self.store = store.store(requirements, self.sharedpath, scmutil.vfs)
270 self.spath = self.store.path
270 self.spath = self.store.path
271 self.svfs = self.store.vfs
271 self.svfs = self.store.vfs
272 self.sopener = self.svfs
272 self.sopener = self.svfs
273 self.sjoin = self.store.join
273 self.sjoin = self.store.join
274 self.vfs.createmode = self.store.createmode
274 self.vfs.createmode = self.store.createmode
275 self._applyrequirements(requirements)
275 self._applyrequirements(requirements)
276 if create:
276 if create:
277 self._writerequirements()
277 self._writerequirements()
278
278
279
279
280 self._branchcaches = {}
280 self._branchcaches = {}
281 self.filterpats = {}
281 self.filterpats = {}
282 self._datafilters = {}
282 self._datafilters = {}
283 self._transref = self._lockref = self._wlockref = None
283 self._transref = self._lockref = self._wlockref = None
284
284
285 # A cache for various files under .hg/ that tracks file changes,
285 # A cache for various files under .hg/ that tracks file changes,
286 # (used by the filecache decorator)
286 # (used by the filecache decorator)
287 #
287 #
288 # Maps a property name to its util.filecacheentry
288 # Maps a property name to its util.filecacheentry
289 self._filecache = {}
289 self._filecache = {}
290
290
291 # hold sets of revision to be filtered
291 # hold sets of revision to be filtered
292 # should be cleared when something might have changed the filter value:
292 # should be cleared when something might have changed the filter value:
293 # - new changesets,
293 # - new changesets,
294 # - phase change,
294 # - phase change,
295 # - new obsolescence marker,
295 # - new obsolescence marker,
296 # - working directory parent change,
296 # - working directory parent change,
297 # - bookmark changes
297 # - bookmark changes
298 self.filteredrevcache = {}
298 self.filteredrevcache = {}
299
299
300 def close(self):
300 def close(self):
301 pass
301 pass
302
302
303 def _restrictcapabilities(self, caps):
303 def _restrictcapabilities(self, caps):
304 # bundle2 is not ready for prime time, drop it unless explicitly
304 # bundle2 is not ready for prime time, drop it unless explicitly
305 # required by the tests (or some brave tester)
305 # required by the tests (or some brave tester)
306 if self.ui.configbool('experimental', 'bundle2-exp', False):
306 if self.ui.configbool('experimental', 'bundle2-exp', False):
307 caps = set(caps)
307 caps = set(caps)
308 capsblob = bundle2.encodecaps(bundle2.getrepocaps(self))
308 capsblob = bundle2.encodecaps(bundle2.getrepocaps(self))
309 caps.add('bundle2-exp=' + urllib.quote(capsblob))
309 caps.add('bundle2-exp=' + urllib.quote(capsblob))
310 return caps
310 return caps
311
311
312 def _applyrequirements(self, requirements):
312 def _applyrequirements(self, requirements):
313 self.requirements = requirements
313 self.requirements = requirements
314 self.sopener.options = dict((r, 1) for r in requirements
314 self.sopener.options = dict((r, 1) for r in requirements
315 if r in self.openerreqs)
315 if r in self.openerreqs)
316 chunkcachesize = self.ui.configint('format', 'chunkcachesize')
316 chunkcachesize = self.ui.configint('format', 'chunkcachesize')
317 if chunkcachesize is not None:
317 if chunkcachesize is not None:
318 self.sopener.options['chunkcachesize'] = chunkcachesize
318 self.sopener.options['chunkcachesize'] = chunkcachesize
319
319
320 def _writerequirements(self):
320 def _writerequirements(self):
321 reqfile = self.opener("requires", "w")
321 reqfile = self.opener("requires", "w")
322 for r in sorted(self.requirements):
322 for r in sorted(self.requirements):
323 reqfile.write("%s\n" % r)
323 reqfile.write("%s\n" % r)
324 reqfile.close()
324 reqfile.close()
325
325
326 def _checknested(self, path):
326 def _checknested(self, path):
327 """Determine if path is a legal nested repository."""
327 """Determine if path is a legal nested repository."""
328 if not path.startswith(self.root):
328 if not path.startswith(self.root):
329 return False
329 return False
330 subpath = path[len(self.root) + 1:]
330 subpath = path[len(self.root) + 1:]
331 normsubpath = util.pconvert(subpath)
331 normsubpath = util.pconvert(subpath)
332
332
333 # XXX: Checking against the current working copy is wrong in
333 # XXX: Checking against the current working copy is wrong in
334 # the sense that it can reject things like
334 # the sense that it can reject things like
335 #
335 #
336 # $ hg cat -r 10 sub/x.txt
336 # $ hg cat -r 10 sub/x.txt
337 #
337 #
338 # if sub/ is no longer a subrepository in the working copy
338 # if sub/ is no longer a subrepository in the working copy
339 # parent revision.
339 # parent revision.
340 #
340 #
341 # However, it can of course also allow things that would have
341 # However, it can of course also allow things that would have
342 # been rejected before, such as the above cat command if sub/
342 # been rejected before, such as the above cat command if sub/
343 # is a subrepository now, but was a normal directory before.
343 # is a subrepository now, but was a normal directory before.
344 # The old path auditor would have rejected by mistake since it
344 # The old path auditor would have rejected by mistake since it
345 # panics when it sees sub/.hg/.
345 # panics when it sees sub/.hg/.
346 #
346 #
347 # All in all, checking against the working copy seems sensible
347 # All in all, checking against the working copy seems sensible
348 # since we want to prevent access to nested repositories on
348 # since we want to prevent access to nested repositories on
349 # the filesystem *now*.
349 # the filesystem *now*.
350 ctx = self[None]
350 ctx = self[None]
351 parts = util.splitpath(subpath)
351 parts = util.splitpath(subpath)
352 while parts:
352 while parts:
353 prefix = '/'.join(parts)
353 prefix = '/'.join(parts)
354 if prefix in ctx.substate:
354 if prefix in ctx.substate:
355 if prefix == normsubpath:
355 if prefix == normsubpath:
356 return True
356 return True
357 else:
357 else:
358 sub = ctx.sub(prefix)
358 sub = ctx.sub(prefix)
359 return sub.checknested(subpath[len(prefix) + 1:])
359 return sub.checknested(subpath[len(prefix) + 1:])
360 else:
360 else:
361 parts.pop()
361 parts.pop()
362 return False
362 return False
363
363
364 def peer(self):
364 def peer(self):
365 return localpeer(self) # not cached to avoid reference cycle
365 return localpeer(self) # not cached to avoid reference cycle
366
366
367 def unfiltered(self):
367 def unfiltered(self):
368 """Return unfiltered version of the repository
368 """Return unfiltered version of the repository
369
369
370 Intended to be overwritten by filtered repo."""
370 Intended to be overwritten by filtered repo."""
371 return self
371 return self
372
372
373 def filtered(self, name):
373 def filtered(self, name):
374 """Return a filtered version of a repository"""
374 """Return a filtered version of a repository"""
375 # build a new class with the mixin and the current class
375 # build a new class with the mixin and the current class
376 # (possibly subclass of the repo)
376 # (possibly subclass of the repo)
377 class proxycls(repoview.repoview, self.unfiltered().__class__):
377 class proxycls(repoview.repoview, self.unfiltered().__class__):
378 pass
378 pass
379 return proxycls(self, name)
379 return proxycls(self, name)
380
380
381 @repofilecache('bookmarks')
381 @repofilecache('bookmarks')
382 def _bookmarks(self):
382 def _bookmarks(self):
383 return bookmarks.bmstore(self)
383 return bookmarks.bmstore(self)
384
384
385 @repofilecache('bookmarks.current')
385 @repofilecache('bookmarks.current')
386 def _bookmarkcurrent(self):
386 def _bookmarkcurrent(self):
387 return bookmarks.readcurrent(self)
387 return bookmarks.readcurrent(self)
388
388
389 def bookmarkheads(self, bookmark):
389 def bookmarkheads(self, bookmark):
390 name = bookmark.split('@', 1)[0]
390 name = bookmark.split('@', 1)[0]
391 heads = []
391 heads = []
392 for mark, n in self._bookmarks.iteritems():
392 for mark, n in self._bookmarks.iteritems():
393 if mark.split('@', 1)[0] == name:
393 if mark.split('@', 1)[0] == name:
394 heads.append(n)
394 heads.append(n)
395 return heads
395 return heads
396
396
397 @storecache('phaseroots')
397 @storecache('phaseroots')
398 def _phasecache(self):
398 def _phasecache(self):
399 return phases.phasecache(self, self._phasedefaults)
399 return phases.phasecache(self, self._phasedefaults)
400
400
401 @storecache('obsstore')
401 @storecache('obsstore')
402 def obsstore(self):
402 def obsstore(self):
403 store = obsolete.obsstore(self.sopener)
403 store = obsolete.obsstore(self.sopener)
404 if store and not obsolete._enabled:
404 if store and not obsolete._enabled:
405 # message is rare enough to not be translated
405 # message is rare enough to not be translated
406 msg = 'obsolete feature not enabled but %i markers found!\n'
406 msg = 'obsolete feature not enabled but %i markers found!\n'
407 self.ui.warn(msg % len(list(store)))
407 self.ui.warn(msg % len(list(store)))
408 return store
408 return store
409
409
410 @storecache('00changelog.i')
410 @storecache('00changelog.i')
411 def changelog(self):
411 def changelog(self):
412 c = changelog.changelog(self.sopener)
412 c = changelog.changelog(self.sopener)
413 if 'HG_PENDING' in os.environ:
413 if 'HG_PENDING' in os.environ:
414 p = os.environ['HG_PENDING']
414 p = os.environ['HG_PENDING']
415 if p.startswith(self.root):
415 if p.startswith(self.root):
416 c.readpending('00changelog.i.a')
416 c.readpending('00changelog.i.a')
417 return c
417 return c
418
418
419 @storecache('00manifest.i')
419 @storecache('00manifest.i')
420 def manifest(self):
420 def manifest(self):
421 return manifest.manifest(self.sopener)
421 return manifest.manifest(self.sopener)
422
422
423 @repofilecache('dirstate')
423 @repofilecache('dirstate')
424 def dirstate(self):
424 def dirstate(self):
425 warned = [0]
425 warned = [0]
426 def validate(node):
426 def validate(node):
427 try:
427 try:
428 self.changelog.rev(node)
428 self.changelog.rev(node)
429 return node
429 return node
430 except error.LookupError:
430 except error.LookupError:
431 if not warned[0]:
431 if not warned[0]:
432 warned[0] = True
432 warned[0] = True
433 self.ui.warn(_("warning: ignoring unknown"
433 self.ui.warn(_("warning: ignoring unknown"
434 " working parent %s!\n") % short(node))
434 " working parent %s!\n") % short(node))
435 return nullid
435 return nullid
436
436
437 return dirstate.dirstate(self.opener, self.ui, self.root, validate)
437 return dirstate.dirstate(self.opener, self.ui, self.root, validate)
438
438
439 def __getitem__(self, changeid):
439 def __getitem__(self, changeid):
440 if changeid is None:
440 if changeid is None:
441 return context.workingctx(self)
441 return context.workingctx(self)
442 return context.changectx(self, changeid)
442 return context.changectx(self, changeid)
443
443
444 def __contains__(self, changeid):
444 def __contains__(self, changeid):
445 try:
445 try:
446 return bool(self.lookup(changeid))
446 return bool(self.lookup(changeid))
447 except error.RepoLookupError:
447 except error.RepoLookupError:
448 return False
448 return False
449
449
450 def __nonzero__(self):
450 def __nonzero__(self):
451 return True
451 return True
452
452
453 def __len__(self):
453 def __len__(self):
454 return len(self.changelog)
454 return len(self.changelog)
455
455
456 def __iter__(self):
456 def __iter__(self):
457 return iter(self.changelog)
457 return iter(self.changelog)
458
458
459 def revs(self, expr, *args):
459 def revs(self, expr, *args):
460 '''Return a list of revisions matching the given revset'''
460 '''Return a list of revisions matching the given revset'''
461 expr = revset.formatspec(expr, *args)
461 expr = revset.formatspec(expr, *args)
462 m = revset.match(None, expr)
462 m = revset.match(None, expr)
463 return m(self, revset.spanset(self))
463 return m(self, revset.spanset(self))
464
464
465 def set(self, expr, *args):
465 def set(self, expr, *args):
466 '''
466 '''
467 Yield a context for each matching revision, after doing arg
467 Yield a context for each matching revision, after doing arg
468 replacement via revset.formatspec
468 replacement via revset.formatspec
469 '''
469 '''
470 for r in self.revs(expr, *args):
470 for r in self.revs(expr, *args):
471 yield self[r]
471 yield self[r]
472
472
473 def url(self):
473 def url(self):
474 return 'file:' + self.root
474 return 'file:' + self.root
475
475
476 def hook(self, name, throw=False, **args):
476 def hook(self, name, throw=False, **args):
477 """Call a hook, passing this repo instance.
477 """Call a hook, passing this repo instance.
478
478
479 This a convenience method to aid invoking hooks. Extensions likely
479 This a convenience method to aid invoking hooks. Extensions likely
480 won't call this unless they have registered a custom hook or are
480 won't call this unless they have registered a custom hook or are
481 replacing code that is expected to call a hook.
481 replacing code that is expected to call a hook.
482 """
482 """
483 return hook.hook(self.ui, self, name, throw, **args)
483 return hook.hook(self.ui, self, name, throw, **args)
484
484
485 @unfilteredmethod
485 @unfilteredmethod
486 def _tag(self, names, node, message, local, user, date, extra={},
486 def _tag(self, names, node, message, local, user, date, extra={},
487 editor=False):
487 editor=False):
488 if isinstance(names, str):
488 if isinstance(names, str):
489 names = (names,)
489 names = (names,)
490
490
491 branches = self.branchmap()
491 branches = self.branchmap()
492 for name in names:
492 for name in names:
493 self.hook('pretag', throw=True, node=hex(node), tag=name,
493 self.hook('pretag', throw=True, node=hex(node), tag=name,
494 local=local)
494 local=local)
495 if name in branches:
495 if name in branches:
496 self.ui.warn(_("warning: tag %s conflicts with existing"
496 self.ui.warn(_("warning: tag %s conflicts with existing"
497 " branch name\n") % name)
497 " branch name\n") % name)
498
498
499 def writetags(fp, names, munge, prevtags):
499 def writetags(fp, names, munge, prevtags):
500 fp.seek(0, 2)
500 fp.seek(0, 2)
501 if prevtags and prevtags[-1] != '\n':
501 if prevtags and prevtags[-1] != '\n':
502 fp.write('\n')
502 fp.write('\n')
503 for name in names:
503 for name in names:
504 m = munge and munge(name) or name
504 m = munge and munge(name) or name
505 if (self._tagscache.tagtypes and
505 if (self._tagscache.tagtypes and
506 name in self._tagscache.tagtypes):
506 name in self._tagscache.tagtypes):
507 old = self.tags().get(name, nullid)
507 old = self.tags().get(name, nullid)
508 fp.write('%s %s\n' % (hex(old), m))
508 fp.write('%s %s\n' % (hex(old), m))
509 fp.write('%s %s\n' % (hex(node), m))
509 fp.write('%s %s\n' % (hex(node), m))
510 fp.close()
510 fp.close()
511
511
512 prevtags = ''
512 prevtags = ''
513 if local:
513 if local:
514 try:
514 try:
515 fp = self.opener('localtags', 'r+')
515 fp = self.opener('localtags', 'r+')
516 except IOError:
516 except IOError:
517 fp = self.opener('localtags', 'a')
517 fp = self.opener('localtags', 'a')
518 else:
518 else:
519 prevtags = fp.read()
519 prevtags = fp.read()
520
520
521 # local tags are stored in the current charset
521 # local tags are stored in the current charset
522 writetags(fp, names, None, prevtags)
522 writetags(fp, names, None, prevtags)
523 for name in names:
523 for name in names:
524 self.hook('tag', node=hex(node), tag=name, local=local)
524 self.hook('tag', node=hex(node), tag=name, local=local)
525 return
525 return
526
526
527 try:
527 try:
528 fp = self.wfile('.hgtags', 'rb+')
528 fp = self.wfile('.hgtags', 'rb+')
529 except IOError, e:
529 except IOError, e:
530 if e.errno != errno.ENOENT:
530 if e.errno != errno.ENOENT:
531 raise
531 raise
532 fp = self.wfile('.hgtags', 'ab')
532 fp = self.wfile('.hgtags', 'ab')
533 else:
533 else:
534 prevtags = fp.read()
534 prevtags = fp.read()
535
535
536 # committed tags are stored in UTF-8
536 # committed tags are stored in UTF-8
537 writetags(fp, names, encoding.fromlocal, prevtags)
537 writetags(fp, names, encoding.fromlocal, prevtags)
538
538
539 fp.close()
539 fp.close()
540
540
541 self.invalidatecaches()
541 self.invalidatecaches()
542
542
543 if '.hgtags' not in self.dirstate:
543 if '.hgtags' not in self.dirstate:
544 self[None].add(['.hgtags'])
544 self[None].add(['.hgtags'])
545
545
546 m = matchmod.exact(self.root, '', ['.hgtags'])
546 m = matchmod.exact(self.root, '', ['.hgtags'])
547 tagnode = self.commit(message, user, date, extra=extra, match=m,
547 tagnode = self.commit(message, user, date, extra=extra, match=m,
548 editor=editor)
548 editor=editor)
549
549
550 for name in names:
550 for name in names:
551 self.hook('tag', node=hex(node), tag=name, local=local)
551 self.hook('tag', node=hex(node), tag=name, local=local)
552
552
553 return tagnode
553 return tagnode
554
554
555 def tag(self, names, node, message, local, user, date, editor=False):
555 def tag(self, names, node, message, local, user, date, editor=False):
556 '''tag a revision with one or more symbolic names.
556 '''tag a revision with one or more symbolic names.
557
557
558 names is a list of strings or, when adding a single tag, names may be a
558 names is a list of strings or, when adding a single tag, names may be a
559 string.
559 string.
560
560
561 if local is True, the tags are stored in a per-repository file.
561 if local is True, the tags are stored in a per-repository file.
562 otherwise, they are stored in the .hgtags file, and a new
562 otherwise, they are stored in the .hgtags file, and a new
563 changeset is committed with the change.
563 changeset is committed with the change.
564
564
565 keyword arguments:
565 keyword arguments:
566
566
567 local: whether to store tags in non-version-controlled file
567 local: whether to store tags in non-version-controlled file
568 (default False)
568 (default False)
569
569
570 message: commit message to use if committing
570 message: commit message to use if committing
571
571
572 user: name of user to use if committing
572 user: name of user to use if committing
573
573
574 date: date tuple to use if committing'''
574 date: date tuple to use if committing'''
575
575
576 if not local:
576 if not local:
577 for x in self.status()[:5]:
577 for x in self.status()[:5]:
578 if '.hgtags' in x:
578 if '.hgtags' in x:
579 raise util.Abort(_('working copy of .hgtags is changed '
579 raise util.Abort(_('working copy of .hgtags is changed '
580 '(please commit .hgtags manually)'))
580 '(please commit .hgtags manually)'))
581
581
582 self.tags() # instantiate the cache
582 self.tags() # instantiate the cache
583 self._tag(names, node, message, local, user, date, editor=editor)
583 self._tag(names, node, message, local, user, date, editor=editor)
584
584
585 @filteredpropertycache
585 @filteredpropertycache
586 def _tagscache(self):
586 def _tagscache(self):
587 '''Returns a tagscache object that contains various tags related
587 '''Returns a tagscache object that contains various tags related
588 caches.'''
588 caches.'''
589
589
590 # This simplifies its cache management by having one decorated
590 # This simplifies its cache management by having one decorated
591 # function (this one) and the rest simply fetch things from it.
591 # function (this one) and the rest simply fetch things from it.
592 class tagscache(object):
592 class tagscache(object):
593 def __init__(self):
593 def __init__(self):
594 # These two define the set of tags for this repository. tags
594 # These two define the set of tags for this repository. tags
595 # maps tag name to node; tagtypes maps tag name to 'global' or
595 # maps tag name to node; tagtypes maps tag name to 'global' or
596 # 'local'. (Global tags are defined by .hgtags across all
596 # 'local'. (Global tags are defined by .hgtags across all
597 # heads, and local tags are defined in .hg/localtags.)
597 # heads, and local tags are defined in .hg/localtags.)
598 # They constitute the in-memory cache of tags.
598 # They constitute the in-memory cache of tags.
599 self.tags = self.tagtypes = None
599 self.tags = self.tagtypes = None
600
600
601 self.nodetagscache = self.tagslist = None
601 self.nodetagscache = self.tagslist = None
602
602
603 cache = tagscache()
603 cache = tagscache()
604 cache.tags, cache.tagtypes = self._findtags()
604 cache.tags, cache.tagtypes = self._findtags()
605
605
606 return cache
606 return cache
607
607
608 def tags(self):
608 def tags(self):
609 '''return a mapping of tag to node'''
609 '''return a mapping of tag to node'''
610 t = {}
610 t = {}
611 if self.changelog.filteredrevs:
611 if self.changelog.filteredrevs:
612 tags, tt = self._findtags()
612 tags, tt = self._findtags()
613 else:
613 else:
614 tags = self._tagscache.tags
614 tags = self._tagscache.tags
615 for k, v in tags.iteritems():
615 for k, v in tags.iteritems():
616 try:
616 try:
617 # ignore tags to unknown nodes
617 # ignore tags to unknown nodes
618 self.changelog.rev(v)
618 self.changelog.rev(v)
619 t[k] = v
619 t[k] = v
620 except (error.LookupError, ValueError):
620 except (error.LookupError, ValueError):
621 pass
621 pass
622 return t
622 return t
623
623
624 def _findtags(self):
624 def _findtags(self):
625 '''Do the hard work of finding tags. Return a pair of dicts
625 '''Do the hard work of finding tags. Return a pair of dicts
626 (tags, tagtypes) where tags maps tag name to node, and tagtypes
626 (tags, tagtypes) where tags maps tag name to node, and tagtypes
627 maps tag name to a string like \'global\' or \'local\'.
627 maps tag name to a string like \'global\' or \'local\'.
628 Subclasses or extensions are free to add their own tags, but
628 Subclasses or extensions are free to add their own tags, but
629 should be aware that the returned dicts will be retained for the
629 should be aware that the returned dicts will be retained for the
630 duration of the localrepo object.'''
630 duration of the localrepo object.'''
631
631
632 # XXX what tagtype should subclasses/extensions use? Currently
632 # XXX what tagtype should subclasses/extensions use? Currently
633 # mq and bookmarks add tags, but do not set the tagtype at all.
633 # mq and bookmarks add tags, but do not set the tagtype at all.
634 # Should each extension invent its own tag type? Should there
634 # Should each extension invent its own tag type? Should there
635 # be one tagtype for all such "virtual" tags? Or is the status
635 # be one tagtype for all such "virtual" tags? Or is the status
636 # quo fine?
636 # quo fine?
637
637
638 alltags = {} # map tag name to (node, hist)
638 alltags = {} # map tag name to (node, hist)
639 tagtypes = {}
639 tagtypes = {}
640
640
641 tagsmod.findglobaltags(self.ui, self, alltags, tagtypes)
641 tagsmod.findglobaltags(self.ui, self, alltags, tagtypes)
642 tagsmod.readlocaltags(self.ui, self, alltags, tagtypes)
642 tagsmod.readlocaltags(self.ui, self, alltags, tagtypes)
643
643
644 # Build the return dicts. Have to re-encode tag names because
644 # Build the return dicts. Have to re-encode tag names because
645 # the tags module always uses UTF-8 (in order not to lose info
645 # the tags module always uses UTF-8 (in order not to lose info
646 # writing to the cache), but the rest of Mercurial wants them in
646 # writing to the cache), but the rest of Mercurial wants them in
647 # local encoding.
647 # local encoding.
648 tags = {}
648 tags = {}
649 for (name, (node, hist)) in alltags.iteritems():
649 for (name, (node, hist)) in alltags.iteritems():
650 if node != nullid:
650 if node != nullid:
651 tags[encoding.tolocal(name)] = node
651 tags[encoding.tolocal(name)] = node
652 tags['tip'] = self.changelog.tip()
652 tags['tip'] = self.changelog.tip()
653 tagtypes = dict([(encoding.tolocal(name), value)
653 tagtypes = dict([(encoding.tolocal(name), value)
654 for (name, value) in tagtypes.iteritems()])
654 for (name, value) in tagtypes.iteritems()])
655 return (tags, tagtypes)
655 return (tags, tagtypes)
656
656
657 def tagtype(self, tagname):
657 def tagtype(self, tagname):
658 '''
658 '''
659 return the type of the given tag. result can be:
659 return the type of the given tag. result can be:
660
660
661 'local' : a local tag
661 'local' : a local tag
662 'global' : a global tag
662 'global' : a global tag
663 None : tag does not exist
663 None : tag does not exist
664 '''
664 '''
665
665
666 return self._tagscache.tagtypes.get(tagname)
666 return self._tagscache.tagtypes.get(tagname)
667
667
668 def tagslist(self):
668 def tagslist(self):
669 '''return a list of tags ordered by revision'''
669 '''return a list of tags ordered by revision'''
670 if not self._tagscache.tagslist:
670 if not self._tagscache.tagslist:
671 l = []
671 l = []
672 for t, n in self.tags().iteritems():
672 for t, n in self.tags().iteritems():
673 l.append((self.changelog.rev(n), t, n))
673 l.append((self.changelog.rev(n), t, n))
674 self._tagscache.tagslist = [(t, n) for r, t, n in sorted(l)]
674 self._tagscache.tagslist = [(t, n) for r, t, n in sorted(l)]
675
675
676 return self._tagscache.tagslist
676 return self._tagscache.tagslist
677
677
678 def nodetags(self, node):
678 def nodetags(self, node):
679 '''return the tags associated with a node'''
679 '''return the tags associated with a node'''
680 if not self._tagscache.nodetagscache:
680 if not self._tagscache.nodetagscache:
681 nodetagscache = {}
681 nodetagscache = {}
682 for t, n in self._tagscache.tags.iteritems():
682 for t, n in self._tagscache.tags.iteritems():
683 nodetagscache.setdefault(n, []).append(t)
683 nodetagscache.setdefault(n, []).append(t)
684 for tags in nodetagscache.itervalues():
684 for tags in nodetagscache.itervalues():
685 tags.sort()
685 tags.sort()
686 self._tagscache.nodetagscache = nodetagscache
686 self._tagscache.nodetagscache = nodetagscache
687 return self._tagscache.nodetagscache.get(node, [])
687 return self._tagscache.nodetagscache.get(node, [])
688
688
689 def nodebookmarks(self, node):
689 def nodebookmarks(self, node):
690 marks = []
690 marks = []
691 for bookmark, n in self._bookmarks.iteritems():
691 for bookmark, n in self._bookmarks.iteritems():
692 if n == node:
692 if n == node:
693 marks.append(bookmark)
693 marks.append(bookmark)
694 return sorted(marks)
694 return sorted(marks)
695
695
696 def branchmap(self):
696 def branchmap(self):
697 '''returns a dictionary {branch: [branchheads]} with branchheads
697 '''returns a dictionary {branch: [branchheads]} with branchheads
698 ordered by increasing revision number'''
698 ordered by increasing revision number'''
699 branchmap.updatecache(self)
699 branchmap.updatecache(self)
700 return self._branchcaches[self.filtername]
700 return self._branchcaches[self.filtername]
701
701
702 def branchtip(self, branch):
702 def branchtip(self, branch):
703 '''return the tip node for a given branch'''
703 '''return the tip node for a given branch'''
704 try:
704 try:
705 return self.branchmap().branchtip(branch)
705 return self.branchmap().branchtip(branch)
706 except KeyError:
706 except KeyError:
707 raise error.RepoLookupError(_("unknown branch '%s'") % branch)
707 raise error.RepoLookupError(_("unknown branch '%s'") % branch)
708
708
709 def lookup(self, key):
709 def lookup(self, key):
710 return self[key].node()
710 return self[key].node()
711
711
712 def lookupbranch(self, key, remote=None):
712 def lookupbranch(self, key, remote=None):
713 repo = remote or self
713 repo = remote or self
714 if key in repo.branchmap():
714 if key in repo.branchmap():
715 return key
715 return key
716
716
717 repo = (remote and remote.local()) and remote or self
717 repo = (remote and remote.local()) and remote or self
718 return repo[key].branch()
718 return repo[key].branch()
719
719
720 def known(self, nodes):
720 def known(self, nodes):
721 nm = self.changelog.nodemap
721 nm = self.changelog.nodemap
722 pc = self._phasecache
722 pc = self._phasecache
723 result = []
723 result = []
724 for n in nodes:
724 for n in nodes:
725 r = nm.get(n)
725 r = nm.get(n)
726 resp = not (r is None or pc.phase(self, r) >= phases.secret)
726 resp = not (r is None or pc.phase(self, r) >= phases.secret)
727 result.append(resp)
727 result.append(resp)
728 return result
728 return result
729
729
730 def local(self):
730 def local(self):
731 return self
731 return self
732
732
733 def cancopy(self):
733 def cancopy(self):
734 # so statichttprepo's override of local() works
734 # so statichttprepo's override of local() works
735 if not self.local():
735 if not self.local():
736 return False
736 return False
737 if not self.ui.configbool('phases', 'publish', True):
737 if not self.ui.configbool('phases', 'publish', True):
738 return True
738 return True
739 # if publishing we can't copy if there is filtered content
739 # if publishing we can't copy if there is filtered content
740 return not self.filtered('visible').changelog.filteredrevs
740 return not self.filtered('visible').changelog.filteredrevs
741
741
742 def join(self, f):
742 def join(self, f, *insidef):
743 return os.path.join(self.path, f)
743 return os.path.join(self.path, f, *insidef)
744
744
745 def wjoin(self, f):
745 def wjoin(self, f, *insidef):
746 return os.path.join(self.root, f)
746 return os.path.join(self.root, f, *insidef)
747
747
748 def file(self, f):
748 def file(self, f):
749 if f[0] == '/':
749 if f[0] == '/':
750 f = f[1:]
750 f = f[1:]
751 return filelog.filelog(self.sopener, f)
751 return filelog.filelog(self.sopener, f)
752
752
753 def changectx(self, changeid):
753 def changectx(self, changeid):
754 return self[changeid]
754 return self[changeid]
755
755
756 def parents(self, changeid=None):
756 def parents(self, changeid=None):
757 '''get list of changectxs for parents of changeid'''
757 '''get list of changectxs for parents of changeid'''
758 return self[changeid].parents()
758 return self[changeid].parents()
759
759
760 def setparents(self, p1, p2=nullid):
760 def setparents(self, p1, p2=nullid):
761 copies = self.dirstate.setparents(p1, p2)
761 copies = self.dirstate.setparents(p1, p2)
762 pctx = self[p1]
762 pctx = self[p1]
763 if copies:
763 if copies:
764 # Adjust copy records, the dirstate cannot do it, it
764 # Adjust copy records, the dirstate cannot do it, it
765 # requires access to parents manifests. Preserve them
765 # requires access to parents manifests. Preserve them
766 # only for entries added to first parent.
766 # only for entries added to first parent.
767 for f in copies:
767 for f in copies:
768 if f not in pctx and copies[f] in pctx:
768 if f not in pctx and copies[f] in pctx:
769 self.dirstate.copy(copies[f], f)
769 self.dirstate.copy(copies[f], f)
770 if p2 == nullid:
770 if p2 == nullid:
771 for f, s in sorted(self.dirstate.copies().items()):
771 for f, s in sorted(self.dirstate.copies().items()):
772 if f not in pctx and s not in pctx:
772 if f not in pctx and s not in pctx:
773 self.dirstate.copy(None, f)
773 self.dirstate.copy(None, f)
774
774
775 def filectx(self, path, changeid=None, fileid=None):
775 def filectx(self, path, changeid=None, fileid=None):
776 """changeid can be a changeset revision, node, or tag.
776 """changeid can be a changeset revision, node, or tag.
777 fileid can be a file revision or node."""
777 fileid can be a file revision or node."""
778 return context.filectx(self, path, changeid, fileid)
778 return context.filectx(self, path, changeid, fileid)
779
779
780 def getcwd(self):
780 def getcwd(self):
781 return self.dirstate.getcwd()
781 return self.dirstate.getcwd()
782
782
783 def pathto(self, f, cwd=None):
783 def pathto(self, f, cwd=None):
784 return self.dirstate.pathto(f, cwd)
784 return self.dirstate.pathto(f, cwd)
785
785
786 def wfile(self, f, mode='r'):
786 def wfile(self, f, mode='r'):
787 return self.wopener(f, mode)
787 return self.wopener(f, mode)
788
788
789 def _link(self, f):
789 def _link(self, f):
790 return self.wvfs.islink(f)
790 return self.wvfs.islink(f)
791
791
792 def _loadfilter(self, filter):
792 def _loadfilter(self, filter):
793 if filter not in self.filterpats:
793 if filter not in self.filterpats:
794 l = []
794 l = []
795 for pat, cmd in self.ui.configitems(filter):
795 for pat, cmd in self.ui.configitems(filter):
796 if cmd == '!':
796 if cmd == '!':
797 continue
797 continue
798 mf = matchmod.match(self.root, '', [pat])
798 mf = matchmod.match(self.root, '', [pat])
799 fn = None
799 fn = None
800 params = cmd
800 params = cmd
801 for name, filterfn in self._datafilters.iteritems():
801 for name, filterfn in self._datafilters.iteritems():
802 if cmd.startswith(name):
802 if cmd.startswith(name):
803 fn = filterfn
803 fn = filterfn
804 params = cmd[len(name):].lstrip()
804 params = cmd[len(name):].lstrip()
805 break
805 break
806 if not fn:
806 if not fn:
807 fn = lambda s, c, **kwargs: util.filter(s, c)
807 fn = lambda s, c, **kwargs: util.filter(s, c)
808 # Wrap old filters not supporting keyword arguments
808 # Wrap old filters not supporting keyword arguments
809 if not inspect.getargspec(fn)[2]:
809 if not inspect.getargspec(fn)[2]:
810 oldfn = fn
810 oldfn = fn
811 fn = lambda s, c, **kwargs: oldfn(s, c)
811 fn = lambda s, c, **kwargs: oldfn(s, c)
812 l.append((mf, fn, params))
812 l.append((mf, fn, params))
813 self.filterpats[filter] = l
813 self.filterpats[filter] = l
814 return self.filterpats[filter]
814 return self.filterpats[filter]
815
815
816 def _filter(self, filterpats, filename, data):
816 def _filter(self, filterpats, filename, data):
817 for mf, fn, cmd in filterpats:
817 for mf, fn, cmd in filterpats:
818 if mf(filename):
818 if mf(filename):
819 self.ui.debug("filtering %s through %s\n" % (filename, cmd))
819 self.ui.debug("filtering %s through %s\n" % (filename, cmd))
820 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
820 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
821 break
821 break
822
822
823 return data
823 return data
824
824
825 @unfilteredpropertycache
825 @unfilteredpropertycache
826 def _encodefilterpats(self):
826 def _encodefilterpats(self):
827 return self._loadfilter('encode')
827 return self._loadfilter('encode')
828
828
829 @unfilteredpropertycache
829 @unfilteredpropertycache
830 def _decodefilterpats(self):
830 def _decodefilterpats(self):
831 return self._loadfilter('decode')
831 return self._loadfilter('decode')
832
832
833 def adddatafilter(self, name, filter):
833 def adddatafilter(self, name, filter):
834 self._datafilters[name] = filter
834 self._datafilters[name] = filter
835
835
836 def wread(self, filename):
836 def wread(self, filename):
837 if self._link(filename):
837 if self._link(filename):
838 data = self.wvfs.readlink(filename)
838 data = self.wvfs.readlink(filename)
839 else:
839 else:
840 data = self.wopener.read(filename)
840 data = self.wopener.read(filename)
841 return self._filter(self._encodefilterpats, filename, data)
841 return self._filter(self._encodefilterpats, filename, data)
842
842
843 def wwrite(self, filename, data, flags):
843 def wwrite(self, filename, data, flags):
844 data = self._filter(self._decodefilterpats, filename, data)
844 data = self._filter(self._decodefilterpats, filename, data)
845 if 'l' in flags:
845 if 'l' in flags:
846 self.wopener.symlink(data, filename)
846 self.wopener.symlink(data, filename)
847 else:
847 else:
848 self.wopener.write(filename, data)
848 self.wopener.write(filename, data)
849 if 'x' in flags:
849 if 'x' in flags:
850 self.wvfs.setflags(filename, False, True)
850 self.wvfs.setflags(filename, False, True)
851
851
852 def wwritedata(self, filename, data):
852 def wwritedata(self, filename, data):
853 return self._filter(self._decodefilterpats, filename, data)
853 return self._filter(self._decodefilterpats, filename, data)
854
854
855 def transaction(self, desc, report=None):
855 def transaction(self, desc, report=None):
856 tr = self._transref and self._transref() or None
856 tr = self._transref and self._transref() or None
857 if tr and tr.running():
857 if tr and tr.running():
858 return tr.nest()
858 return tr.nest()
859
859
860 # abort here if the journal already exists
860 # abort here if the journal already exists
861 if self.svfs.exists("journal"):
861 if self.svfs.exists("journal"):
862 raise error.RepoError(
862 raise error.RepoError(
863 _("abandoned transaction found"),
863 _("abandoned transaction found"),
864 hint=_("run 'hg recover' to clean up transaction"))
864 hint=_("run 'hg recover' to clean up transaction"))
865
865
866 def onclose():
866 def onclose():
867 self.store.write(self._transref())
867 self.store.write(self._transref())
868
868
869 self._writejournal(desc)
869 self._writejournal(desc)
870 renames = [(vfs, x, undoname(x)) for vfs, x in self._journalfiles()]
870 renames = [(vfs, x, undoname(x)) for vfs, x in self._journalfiles()]
871 rp = report and report or self.ui.warn
871 rp = report and report or self.ui.warn
872 tr = transaction.transaction(rp, self.sopener,
872 tr = transaction.transaction(rp, self.sopener,
873 "journal",
873 "journal",
874 aftertrans(renames),
874 aftertrans(renames),
875 self.store.createmode,
875 self.store.createmode,
876 onclose)
876 onclose)
877 self._transref = weakref.ref(tr)
877 self._transref = weakref.ref(tr)
878 return tr
878 return tr
879
879
880 def _journalfiles(self):
880 def _journalfiles(self):
881 return ((self.svfs, 'journal'),
881 return ((self.svfs, 'journal'),
882 (self.vfs, 'journal.dirstate'),
882 (self.vfs, 'journal.dirstate'),
883 (self.vfs, 'journal.branch'),
883 (self.vfs, 'journal.branch'),
884 (self.vfs, 'journal.desc'),
884 (self.vfs, 'journal.desc'),
885 (self.vfs, 'journal.bookmarks'),
885 (self.vfs, 'journal.bookmarks'),
886 (self.svfs, 'journal.phaseroots'))
886 (self.svfs, 'journal.phaseroots'))
887
887
888 def undofiles(self):
888 def undofiles(self):
889 return [(vfs, undoname(x)) for vfs, x in self._journalfiles()]
889 return [(vfs, undoname(x)) for vfs, x in self._journalfiles()]
890
890
891 def _writejournal(self, desc):
891 def _writejournal(self, desc):
892 self.opener.write("journal.dirstate",
892 self.opener.write("journal.dirstate",
893 self.opener.tryread("dirstate"))
893 self.opener.tryread("dirstate"))
894 self.opener.write("journal.branch",
894 self.opener.write("journal.branch",
895 encoding.fromlocal(self.dirstate.branch()))
895 encoding.fromlocal(self.dirstate.branch()))
896 self.opener.write("journal.desc",
896 self.opener.write("journal.desc",
897 "%d\n%s\n" % (len(self), desc))
897 "%d\n%s\n" % (len(self), desc))
898 self.opener.write("journal.bookmarks",
898 self.opener.write("journal.bookmarks",
899 self.opener.tryread("bookmarks"))
899 self.opener.tryread("bookmarks"))
900 self.sopener.write("journal.phaseroots",
900 self.sopener.write("journal.phaseroots",
901 self.sopener.tryread("phaseroots"))
901 self.sopener.tryread("phaseroots"))
902
902
903 def recover(self):
903 def recover(self):
904 lock = self.lock()
904 lock = self.lock()
905 try:
905 try:
906 if self.svfs.exists("journal"):
906 if self.svfs.exists("journal"):
907 self.ui.status(_("rolling back interrupted transaction\n"))
907 self.ui.status(_("rolling back interrupted transaction\n"))
908 transaction.rollback(self.sopener, "journal",
908 transaction.rollback(self.sopener, "journal",
909 self.ui.warn)
909 self.ui.warn)
910 self.invalidate()
910 self.invalidate()
911 return True
911 return True
912 else:
912 else:
913 self.ui.warn(_("no interrupted transaction available\n"))
913 self.ui.warn(_("no interrupted transaction available\n"))
914 return False
914 return False
915 finally:
915 finally:
916 lock.release()
916 lock.release()
917
917
918 def rollback(self, dryrun=False, force=False):
918 def rollback(self, dryrun=False, force=False):
919 wlock = lock = None
919 wlock = lock = None
920 try:
920 try:
921 wlock = self.wlock()
921 wlock = self.wlock()
922 lock = self.lock()
922 lock = self.lock()
923 if self.svfs.exists("undo"):
923 if self.svfs.exists("undo"):
924 return self._rollback(dryrun, force)
924 return self._rollback(dryrun, force)
925 else:
925 else:
926 self.ui.warn(_("no rollback information available\n"))
926 self.ui.warn(_("no rollback information available\n"))
927 return 1
927 return 1
928 finally:
928 finally:
929 release(lock, wlock)
929 release(lock, wlock)
930
930
931 @unfilteredmethod # Until we get smarter cache management
931 @unfilteredmethod # Until we get smarter cache management
932 def _rollback(self, dryrun, force):
932 def _rollback(self, dryrun, force):
933 ui = self.ui
933 ui = self.ui
934 try:
934 try:
935 args = self.opener.read('undo.desc').splitlines()
935 args = self.opener.read('undo.desc').splitlines()
936 (oldlen, desc, detail) = (int(args[0]), args[1], None)
936 (oldlen, desc, detail) = (int(args[0]), args[1], None)
937 if len(args) >= 3:
937 if len(args) >= 3:
938 detail = args[2]
938 detail = args[2]
939 oldtip = oldlen - 1
939 oldtip = oldlen - 1
940
940
941 if detail and ui.verbose:
941 if detail and ui.verbose:
942 msg = (_('repository tip rolled back to revision %s'
942 msg = (_('repository tip rolled back to revision %s'
943 ' (undo %s: %s)\n')
943 ' (undo %s: %s)\n')
944 % (oldtip, desc, detail))
944 % (oldtip, desc, detail))
945 else:
945 else:
946 msg = (_('repository tip rolled back to revision %s'
946 msg = (_('repository tip rolled back to revision %s'
947 ' (undo %s)\n')
947 ' (undo %s)\n')
948 % (oldtip, desc))
948 % (oldtip, desc))
949 except IOError:
949 except IOError:
950 msg = _('rolling back unknown transaction\n')
950 msg = _('rolling back unknown transaction\n')
951 desc = None
951 desc = None
952
952
953 if not force and self['.'] != self['tip'] and desc == 'commit':
953 if not force and self['.'] != self['tip'] and desc == 'commit':
954 raise util.Abort(
954 raise util.Abort(
955 _('rollback of last commit while not checked out '
955 _('rollback of last commit while not checked out '
956 'may lose data'), hint=_('use -f to force'))
956 'may lose data'), hint=_('use -f to force'))
957
957
958 ui.status(msg)
958 ui.status(msg)
959 if dryrun:
959 if dryrun:
960 return 0
960 return 0
961
961
962 parents = self.dirstate.parents()
962 parents = self.dirstate.parents()
963 self.destroying()
963 self.destroying()
964 transaction.rollback(self.sopener, 'undo', ui.warn)
964 transaction.rollback(self.sopener, 'undo', ui.warn)
965 if self.vfs.exists('undo.bookmarks'):
965 if self.vfs.exists('undo.bookmarks'):
966 self.vfs.rename('undo.bookmarks', 'bookmarks')
966 self.vfs.rename('undo.bookmarks', 'bookmarks')
967 if self.svfs.exists('undo.phaseroots'):
967 if self.svfs.exists('undo.phaseroots'):
968 self.svfs.rename('undo.phaseroots', 'phaseroots')
968 self.svfs.rename('undo.phaseroots', 'phaseroots')
969 self.invalidate()
969 self.invalidate()
970
970
971 parentgone = (parents[0] not in self.changelog.nodemap or
971 parentgone = (parents[0] not in self.changelog.nodemap or
972 parents[1] not in self.changelog.nodemap)
972 parents[1] not in self.changelog.nodemap)
973 if parentgone:
973 if parentgone:
974 self.vfs.rename('undo.dirstate', 'dirstate')
974 self.vfs.rename('undo.dirstate', 'dirstate')
975 try:
975 try:
976 branch = self.opener.read('undo.branch')
976 branch = self.opener.read('undo.branch')
977 self.dirstate.setbranch(encoding.tolocal(branch))
977 self.dirstate.setbranch(encoding.tolocal(branch))
978 except IOError:
978 except IOError:
979 ui.warn(_('named branch could not be reset: '
979 ui.warn(_('named branch could not be reset: '
980 'current branch is still \'%s\'\n')
980 'current branch is still \'%s\'\n')
981 % self.dirstate.branch())
981 % self.dirstate.branch())
982
982
983 self.dirstate.invalidate()
983 self.dirstate.invalidate()
984 parents = tuple([p.rev() for p in self.parents()])
984 parents = tuple([p.rev() for p in self.parents()])
985 if len(parents) > 1:
985 if len(parents) > 1:
986 ui.status(_('working directory now based on '
986 ui.status(_('working directory now based on '
987 'revisions %d and %d\n') % parents)
987 'revisions %d and %d\n') % parents)
988 else:
988 else:
989 ui.status(_('working directory now based on '
989 ui.status(_('working directory now based on '
990 'revision %d\n') % parents)
990 'revision %d\n') % parents)
991 # TODO: if we know which new heads may result from this rollback, pass
991 # TODO: if we know which new heads may result from this rollback, pass
992 # them to destroy(), which will prevent the branchhead cache from being
992 # them to destroy(), which will prevent the branchhead cache from being
993 # invalidated.
993 # invalidated.
994 self.destroyed()
994 self.destroyed()
995 return 0
995 return 0
996
996
997 def invalidatecaches(self):
997 def invalidatecaches(self):
998
998
999 if '_tagscache' in vars(self):
999 if '_tagscache' in vars(self):
1000 # can't use delattr on proxy
1000 # can't use delattr on proxy
1001 del self.__dict__['_tagscache']
1001 del self.__dict__['_tagscache']
1002
1002
1003 self.unfiltered()._branchcaches.clear()
1003 self.unfiltered()._branchcaches.clear()
1004 self.invalidatevolatilesets()
1004 self.invalidatevolatilesets()
1005
1005
1006 def invalidatevolatilesets(self):
1006 def invalidatevolatilesets(self):
1007 self.filteredrevcache.clear()
1007 self.filteredrevcache.clear()
1008 obsolete.clearobscaches(self)
1008 obsolete.clearobscaches(self)
1009
1009
1010 def invalidatedirstate(self):
1010 def invalidatedirstate(self):
1011 '''Invalidates the dirstate, causing the next call to dirstate
1011 '''Invalidates the dirstate, causing the next call to dirstate
1012 to check if it was modified since the last time it was read,
1012 to check if it was modified since the last time it was read,
1013 rereading it if it has.
1013 rereading it if it has.
1014
1014
1015 This is different to dirstate.invalidate() that it doesn't always
1015 This is different to dirstate.invalidate() that it doesn't always
1016 rereads the dirstate. Use dirstate.invalidate() if you want to
1016 rereads the dirstate. Use dirstate.invalidate() if you want to
1017 explicitly read the dirstate again (i.e. restoring it to a previous
1017 explicitly read the dirstate again (i.e. restoring it to a previous
1018 known good state).'''
1018 known good state).'''
1019 if hasunfilteredcache(self, 'dirstate'):
1019 if hasunfilteredcache(self, 'dirstate'):
1020 for k in self.dirstate._filecache:
1020 for k in self.dirstate._filecache:
1021 try:
1021 try:
1022 delattr(self.dirstate, k)
1022 delattr(self.dirstate, k)
1023 except AttributeError:
1023 except AttributeError:
1024 pass
1024 pass
1025 delattr(self.unfiltered(), 'dirstate')
1025 delattr(self.unfiltered(), 'dirstate')
1026
1026
1027 def invalidate(self):
1027 def invalidate(self):
1028 unfiltered = self.unfiltered() # all file caches are stored unfiltered
1028 unfiltered = self.unfiltered() # all file caches are stored unfiltered
1029 for k in self._filecache:
1029 for k in self._filecache:
1030 # dirstate is invalidated separately in invalidatedirstate()
1030 # dirstate is invalidated separately in invalidatedirstate()
1031 if k == 'dirstate':
1031 if k == 'dirstate':
1032 continue
1032 continue
1033
1033
1034 try:
1034 try:
1035 delattr(unfiltered, k)
1035 delattr(unfiltered, k)
1036 except AttributeError:
1036 except AttributeError:
1037 pass
1037 pass
1038 self.invalidatecaches()
1038 self.invalidatecaches()
1039 self.store.invalidatecaches()
1039 self.store.invalidatecaches()
1040
1040
1041 def invalidateall(self):
1041 def invalidateall(self):
1042 '''Fully invalidates both store and non-store parts, causing the
1042 '''Fully invalidates both store and non-store parts, causing the
1043 subsequent operation to reread any outside changes.'''
1043 subsequent operation to reread any outside changes.'''
1044 # extension should hook this to invalidate its caches
1044 # extension should hook this to invalidate its caches
1045 self.invalidate()
1045 self.invalidate()
1046 self.invalidatedirstate()
1046 self.invalidatedirstate()
1047
1047
1048 def _lock(self, vfs, lockname, wait, releasefn, acquirefn, desc):
1048 def _lock(self, vfs, lockname, wait, releasefn, acquirefn, desc):
1049 try:
1049 try:
1050 l = lockmod.lock(vfs, lockname, 0, releasefn, desc=desc)
1050 l = lockmod.lock(vfs, lockname, 0, releasefn, desc=desc)
1051 except error.LockHeld, inst:
1051 except error.LockHeld, inst:
1052 if not wait:
1052 if not wait:
1053 raise
1053 raise
1054 self.ui.warn(_("waiting for lock on %s held by %r\n") %
1054 self.ui.warn(_("waiting for lock on %s held by %r\n") %
1055 (desc, inst.locker))
1055 (desc, inst.locker))
1056 # default to 600 seconds timeout
1056 # default to 600 seconds timeout
1057 l = lockmod.lock(vfs, lockname,
1057 l = lockmod.lock(vfs, lockname,
1058 int(self.ui.config("ui", "timeout", "600")),
1058 int(self.ui.config("ui", "timeout", "600")),
1059 releasefn, desc=desc)
1059 releasefn, desc=desc)
1060 self.ui.warn(_("got lock after %s seconds\n") % l.delay)
1060 self.ui.warn(_("got lock after %s seconds\n") % l.delay)
1061 if acquirefn:
1061 if acquirefn:
1062 acquirefn()
1062 acquirefn()
1063 return l
1063 return l
1064
1064
1065 def _afterlock(self, callback):
1065 def _afterlock(self, callback):
1066 """add a callback to the current repository lock.
1066 """add a callback to the current repository lock.
1067
1067
1068 The callback will be executed on lock release."""
1068 The callback will be executed on lock release."""
1069 l = self._lockref and self._lockref()
1069 l = self._lockref and self._lockref()
1070 if l:
1070 if l:
1071 l.postrelease.append(callback)
1071 l.postrelease.append(callback)
1072 else:
1072 else:
1073 callback()
1073 callback()
1074
1074
1075 def lock(self, wait=True):
1075 def lock(self, wait=True):
1076 '''Lock the repository store (.hg/store) and return a weak reference
1076 '''Lock the repository store (.hg/store) and return a weak reference
1077 to the lock. Use this before modifying the store (e.g. committing or
1077 to the lock. Use this before modifying the store (e.g. committing or
1078 stripping). If you are opening a transaction, get a lock as well.)'''
1078 stripping). If you are opening a transaction, get a lock as well.)'''
1079 l = self._lockref and self._lockref()
1079 l = self._lockref and self._lockref()
1080 if l is not None and l.held:
1080 if l is not None and l.held:
1081 l.lock()
1081 l.lock()
1082 return l
1082 return l
1083
1083
1084 def unlock():
1084 def unlock():
1085 for k, ce in self._filecache.items():
1085 for k, ce in self._filecache.items():
1086 if k == 'dirstate' or k not in self.__dict__:
1086 if k == 'dirstate' or k not in self.__dict__:
1087 continue
1087 continue
1088 ce.refresh()
1088 ce.refresh()
1089
1089
1090 l = self._lock(self.svfs, "lock", wait, unlock,
1090 l = self._lock(self.svfs, "lock", wait, unlock,
1091 self.invalidate, _('repository %s') % self.origroot)
1091 self.invalidate, _('repository %s') % self.origroot)
1092 self._lockref = weakref.ref(l)
1092 self._lockref = weakref.ref(l)
1093 return l
1093 return l
1094
1094
1095 def wlock(self, wait=True):
1095 def wlock(self, wait=True):
1096 '''Lock the non-store parts of the repository (everything under
1096 '''Lock the non-store parts of the repository (everything under
1097 .hg except .hg/store) and return a weak reference to the lock.
1097 .hg except .hg/store) and return a weak reference to the lock.
1098 Use this before modifying files in .hg.'''
1098 Use this before modifying files in .hg.'''
1099 l = self._wlockref and self._wlockref()
1099 l = self._wlockref and self._wlockref()
1100 if l is not None and l.held:
1100 if l is not None and l.held:
1101 l.lock()
1101 l.lock()
1102 return l
1102 return l
1103
1103
1104 def unlock():
1104 def unlock():
1105 self.dirstate.write()
1105 self.dirstate.write()
1106 self._filecache['dirstate'].refresh()
1106 self._filecache['dirstate'].refresh()
1107
1107
1108 l = self._lock(self.vfs, "wlock", wait, unlock,
1108 l = self._lock(self.vfs, "wlock", wait, unlock,
1109 self.invalidatedirstate, _('working directory of %s') %
1109 self.invalidatedirstate, _('working directory of %s') %
1110 self.origroot)
1110 self.origroot)
1111 self._wlockref = weakref.ref(l)
1111 self._wlockref = weakref.ref(l)
1112 return l
1112 return l
1113
1113
1114 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
1114 def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
1115 """
1115 """
1116 commit an individual file as part of a larger transaction
1116 commit an individual file as part of a larger transaction
1117 """
1117 """
1118
1118
1119 fname = fctx.path()
1119 fname = fctx.path()
1120 text = fctx.data()
1120 text = fctx.data()
1121 flog = self.file(fname)
1121 flog = self.file(fname)
1122 fparent1 = manifest1.get(fname, nullid)
1122 fparent1 = manifest1.get(fname, nullid)
1123 fparent2 = fparent2o = manifest2.get(fname, nullid)
1123 fparent2 = fparent2o = manifest2.get(fname, nullid)
1124
1124
1125 meta = {}
1125 meta = {}
1126 copy = fctx.renamed()
1126 copy = fctx.renamed()
1127 if copy and copy[0] != fname:
1127 if copy and copy[0] != fname:
1128 # Mark the new revision of this file as a copy of another
1128 # Mark the new revision of this file as a copy of another
1129 # file. This copy data will effectively act as a parent
1129 # file. This copy data will effectively act as a parent
1130 # of this new revision. If this is a merge, the first
1130 # of this new revision. If this is a merge, the first
1131 # parent will be the nullid (meaning "look up the copy data")
1131 # parent will be the nullid (meaning "look up the copy data")
1132 # and the second one will be the other parent. For example:
1132 # and the second one will be the other parent. For example:
1133 #
1133 #
1134 # 0 --- 1 --- 3 rev1 changes file foo
1134 # 0 --- 1 --- 3 rev1 changes file foo
1135 # \ / rev2 renames foo to bar and changes it
1135 # \ / rev2 renames foo to bar and changes it
1136 # \- 2 -/ rev3 should have bar with all changes and
1136 # \- 2 -/ rev3 should have bar with all changes and
1137 # should record that bar descends from
1137 # should record that bar descends from
1138 # bar in rev2 and foo in rev1
1138 # bar in rev2 and foo in rev1
1139 #
1139 #
1140 # this allows this merge to succeed:
1140 # this allows this merge to succeed:
1141 #
1141 #
1142 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
1142 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
1143 # \ / merging rev3 and rev4 should use bar@rev2
1143 # \ / merging rev3 and rev4 should use bar@rev2
1144 # \- 2 --- 4 as the merge base
1144 # \- 2 --- 4 as the merge base
1145 #
1145 #
1146
1146
1147 cfname = copy[0]
1147 cfname = copy[0]
1148 crev = manifest1.get(cfname)
1148 crev = manifest1.get(cfname)
1149 newfparent = fparent2
1149 newfparent = fparent2
1150
1150
1151 if manifest2: # branch merge
1151 if manifest2: # branch merge
1152 if fparent2 == nullid or crev is None: # copied on remote side
1152 if fparent2 == nullid or crev is None: # copied on remote side
1153 if cfname in manifest2:
1153 if cfname in manifest2:
1154 crev = manifest2[cfname]
1154 crev = manifest2[cfname]
1155 newfparent = fparent1
1155 newfparent = fparent1
1156
1156
1157 # find source in nearest ancestor if we've lost track
1157 # find source in nearest ancestor if we've lost track
1158 if not crev:
1158 if not crev:
1159 self.ui.debug(" %s: searching for copy revision for %s\n" %
1159 self.ui.debug(" %s: searching for copy revision for %s\n" %
1160 (fname, cfname))
1160 (fname, cfname))
1161 for ancestor in self[None].ancestors():
1161 for ancestor in self[None].ancestors():
1162 if cfname in ancestor:
1162 if cfname in ancestor:
1163 crev = ancestor[cfname].filenode()
1163 crev = ancestor[cfname].filenode()
1164 break
1164 break
1165
1165
1166 if crev:
1166 if crev:
1167 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
1167 self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
1168 meta["copy"] = cfname
1168 meta["copy"] = cfname
1169 meta["copyrev"] = hex(crev)
1169 meta["copyrev"] = hex(crev)
1170 fparent1, fparent2 = nullid, newfparent
1170 fparent1, fparent2 = nullid, newfparent
1171 else:
1171 else:
1172 self.ui.warn(_("warning: can't find ancestor for '%s' "
1172 self.ui.warn(_("warning: can't find ancestor for '%s' "
1173 "copied from '%s'!\n") % (fname, cfname))
1173 "copied from '%s'!\n") % (fname, cfname))
1174
1174
1175 elif fparent1 == nullid:
1175 elif fparent1 == nullid:
1176 fparent1, fparent2 = fparent2, nullid
1176 fparent1, fparent2 = fparent2, nullid
1177 elif fparent2 != nullid:
1177 elif fparent2 != nullid:
1178 # is one parent an ancestor of the other?
1178 # is one parent an ancestor of the other?
1179 fparentancestors = flog.commonancestorsheads(fparent1, fparent2)
1179 fparentancestors = flog.commonancestorsheads(fparent1, fparent2)
1180 if fparent1 in fparentancestors:
1180 if fparent1 in fparentancestors:
1181 fparent1, fparent2 = fparent2, nullid
1181 fparent1, fparent2 = fparent2, nullid
1182 elif fparent2 in fparentancestors:
1182 elif fparent2 in fparentancestors:
1183 fparent2 = nullid
1183 fparent2 = nullid
1184
1184
1185 # is the file changed?
1185 # is the file changed?
1186 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
1186 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
1187 changelist.append(fname)
1187 changelist.append(fname)
1188 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
1188 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
1189
1189
1190 # are just the flags changed during merge?
1190 # are just the flags changed during merge?
1191 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
1191 if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
1192 changelist.append(fname)
1192 changelist.append(fname)
1193
1193
1194 return fparent1
1194 return fparent1
1195
1195
1196 @unfilteredmethod
1196 @unfilteredmethod
1197 def commit(self, text="", user=None, date=None, match=None, force=False,
1197 def commit(self, text="", user=None, date=None, match=None, force=False,
1198 editor=False, extra={}):
1198 editor=False, extra={}):
1199 """Add a new revision to current repository.
1199 """Add a new revision to current repository.
1200
1200
1201 Revision information is gathered from the working directory,
1201 Revision information is gathered from the working directory,
1202 match can be used to filter the committed files. If editor is
1202 match can be used to filter the committed files. If editor is
1203 supplied, it is called to get a commit message.
1203 supplied, it is called to get a commit message.
1204 """
1204 """
1205
1205
1206 def fail(f, msg):
1206 def fail(f, msg):
1207 raise util.Abort('%s: %s' % (f, msg))
1207 raise util.Abort('%s: %s' % (f, msg))
1208
1208
1209 if not match:
1209 if not match:
1210 match = matchmod.always(self.root, '')
1210 match = matchmod.always(self.root, '')
1211
1211
1212 if not force:
1212 if not force:
1213 vdirs = []
1213 vdirs = []
1214 match.explicitdir = vdirs.append
1214 match.explicitdir = vdirs.append
1215 match.bad = fail
1215 match.bad = fail
1216
1216
1217 wlock = self.wlock()
1217 wlock = self.wlock()
1218 try:
1218 try:
1219 wctx = self[None]
1219 wctx = self[None]
1220 merge = len(wctx.parents()) > 1
1220 merge = len(wctx.parents()) > 1
1221
1221
1222 if (not force and merge and match and
1222 if (not force and merge and match and
1223 (match.files() or match.anypats())):
1223 (match.files() or match.anypats())):
1224 raise util.Abort(_('cannot partially commit a merge '
1224 raise util.Abort(_('cannot partially commit a merge '
1225 '(do not specify files or patterns)'))
1225 '(do not specify files or patterns)'))
1226
1226
1227 changes = self.status(match=match, clean=force)
1227 changes = self.status(match=match, clean=force)
1228 if force:
1228 if force:
1229 changes[0].extend(changes[6]) # mq may commit unchanged files
1229 changes[0].extend(changes[6]) # mq may commit unchanged files
1230
1230
1231 # check subrepos
1231 # check subrepos
1232 subs = []
1232 subs = []
1233 commitsubs = set()
1233 commitsubs = set()
1234 newstate = wctx.substate.copy()
1234 newstate = wctx.substate.copy()
1235 # only manage subrepos and .hgsubstate if .hgsub is present
1235 # only manage subrepos and .hgsubstate if .hgsub is present
1236 if '.hgsub' in wctx:
1236 if '.hgsub' in wctx:
1237 # we'll decide whether to track this ourselves, thanks
1237 # we'll decide whether to track this ourselves, thanks
1238 for c in changes[:3]:
1238 for c in changes[:3]:
1239 if '.hgsubstate' in c:
1239 if '.hgsubstate' in c:
1240 c.remove('.hgsubstate')
1240 c.remove('.hgsubstate')
1241
1241
1242 # compare current state to last committed state
1242 # compare current state to last committed state
1243 # build new substate based on last committed state
1243 # build new substate based on last committed state
1244 oldstate = wctx.p1().substate
1244 oldstate = wctx.p1().substate
1245 for s in sorted(newstate.keys()):
1245 for s in sorted(newstate.keys()):
1246 if not match(s):
1246 if not match(s):
1247 # ignore working copy, use old state if present
1247 # ignore working copy, use old state if present
1248 if s in oldstate:
1248 if s in oldstate:
1249 newstate[s] = oldstate[s]
1249 newstate[s] = oldstate[s]
1250 continue
1250 continue
1251 if not force:
1251 if not force:
1252 raise util.Abort(
1252 raise util.Abort(
1253 _("commit with new subrepo %s excluded") % s)
1253 _("commit with new subrepo %s excluded") % s)
1254 if wctx.sub(s).dirty(True):
1254 if wctx.sub(s).dirty(True):
1255 if not self.ui.configbool('ui', 'commitsubrepos'):
1255 if not self.ui.configbool('ui', 'commitsubrepos'):
1256 raise util.Abort(
1256 raise util.Abort(
1257 _("uncommitted changes in subrepo %s") % s,
1257 _("uncommitted changes in subrepo %s") % s,
1258 hint=_("use --subrepos for recursive commit"))
1258 hint=_("use --subrepos for recursive commit"))
1259 subs.append(s)
1259 subs.append(s)
1260 commitsubs.add(s)
1260 commitsubs.add(s)
1261 else:
1261 else:
1262 bs = wctx.sub(s).basestate()
1262 bs = wctx.sub(s).basestate()
1263 newstate[s] = (newstate[s][0], bs, newstate[s][2])
1263 newstate[s] = (newstate[s][0], bs, newstate[s][2])
1264 if oldstate.get(s, (None, None, None))[1] != bs:
1264 if oldstate.get(s, (None, None, None))[1] != bs:
1265 subs.append(s)
1265 subs.append(s)
1266
1266
1267 # check for removed subrepos
1267 # check for removed subrepos
1268 for p in wctx.parents():
1268 for p in wctx.parents():
1269 r = [s for s in p.substate if s not in newstate]
1269 r = [s for s in p.substate if s not in newstate]
1270 subs += [s for s in r if match(s)]
1270 subs += [s for s in r if match(s)]
1271 if subs:
1271 if subs:
1272 if (not match('.hgsub') and
1272 if (not match('.hgsub') and
1273 '.hgsub' in (wctx.modified() + wctx.added())):
1273 '.hgsub' in (wctx.modified() + wctx.added())):
1274 raise util.Abort(
1274 raise util.Abort(
1275 _("can't commit subrepos without .hgsub"))
1275 _("can't commit subrepos without .hgsub"))
1276 changes[0].insert(0, '.hgsubstate')
1276 changes[0].insert(0, '.hgsubstate')
1277
1277
1278 elif '.hgsub' in changes[2]:
1278 elif '.hgsub' in changes[2]:
1279 # clean up .hgsubstate when .hgsub is removed
1279 # clean up .hgsubstate when .hgsub is removed
1280 if ('.hgsubstate' in wctx and
1280 if ('.hgsubstate' in wctx and
1281 '.hgsubstate' not in changes[0] + changes[1] + changes[2]):
1281 '.hgsubstate' not in changes[0] + changes[1] + changes[2]):
1282 changes[2].insert(0, '.hgsubstate')
1282 changes[2].insert(0, '.hgsubstate')
1283
1283
1284 # make sure all explicit patterns are matched
1284 # make sure all explicit patterns are matched
1285 if not force and match.files():
1285 if not force and match.files():
1286 matched = set(changes[0] + changes[1] + changes[2])
1286 matched = set(changes[0] + changes[1] + changes[2])
1287
1287
1288 for f in match.files():
1288 for f in match.files():
1289 f = self.dirstate.normalize(f)
1289 f = self.dirstate.normalize(f)
1290 if f == '.' or f in matched or f in wctx.substate:
1290 if f == '.' or f in matched or f in wctx.substate:
1291 continue
1291 continue
1292 if f in changes[3]: # missing
1292 if f in changes[3]: # missing
1293 fail(f, _('file not found!'))
1293 fail(f, _('file not found!'))
1294 if f in vdirs: # visited directory
1294 if f in vdirs: # visited directory
1295 d = f + '/'
1295 d = f + '/'
1296 for mf in matched:
1296 for mf in matched:
1297 if mf.startswith(d):
1297 if mf.startswith(d):
1298 break
1298 break
1299 else:
1299 else:
1300 fail(f, _("no match under directory!"))
1300 fail(f, _("no match under directory!"))
1301 elif f not in self.dirstate:
1301 elif f not in self.dirstate:
1302 fail(f, _("file not tracked!"))
1302 fail(f, _("file not tracked!"))
1303
1303
1304 cctx = context.workingctx(self, text, user, date, extra, changes)
1304 cctx = context.workingctx(self, text, user, date, extra, changes)
1305
1305
1306 if (not force and not extra.get("close") and not merge
1306 if (not force and not extra.get("close") and not merge
1307 and not cctx.files()
1307 and not cctx.files()
1308 and wctx.branch() == wctx.p1().branch()):
1308 and wctx.branch() == wctx.p1().branch()):
1309 return None
1309 return None
1310
1310
1311 if merge and cctx.deleted():
1311 if merge and cctx.deleted():
1312 raise util.Abort(_("cannot commit merge with missing files"))
1312 raise util.Abort(_("cannot commit merge with missing files"))
1313
1313
1314 ms = mergemod.mergestate(self)
1314 ms = mergemod.mergestate(self)
1315 for f in changes[0]:
1315 for f in changes[0]:
1316 if f in ms and ms[f] == 'u':
1316 if f in ms and ms[f] == 'u':
1317 raise util.Abort(_("unresolved merge conflicts "
1317 raise util.Abort(_("unresolved merge conflicts "
1318 "(see hg help resolve)"))
1318 "(see hg help resolve)"))
1319
1319
1320 if editor:
1320 if editor:
1321 cctx._text = editor(self, cctx, subs)
1321 cctx._text = editor(self, cctx, subs)
1322 edited = (text != cctx._text)
1322 edited = (text != cctx._text)
1323
1323
1324 # Save commit message in case this transaction gets rolled back
1324 # Save commit message in case this transaction gets rolled back
1325 # (e.g. by a pretxncommit hook). Leave the content alone on
1325 # (e.g. by a pretxncommit hook). Leave the content alone on
1326 # the assumption that the user will use the same editor again.
1326 # the assumption that the user will use the same editor again.
1327 msgfn = self.savecommitmessage(cctx._text)
1327 msgfn = self.savecommitmessage(cctx._text)
1328
1328
1329 # commit subs and write new state
1329 # commit subs and write new state
1330 if subs:
1330 if subs:
1331 for s in sorted(commitsubs):
1331 for s in sorted(commitsubs):
1332 sub = wctx.sub(s)
1332 sub = wctx.sub(s)
1333 self.ui.status(_('committing subrepository %s\n') %
1333 self.ui.status(_('committing subrepository %s\n') %
1334 subrepo.subrelpath(sub))
1334 subrepo.subrelpath(sub))
1335 sr = sub.commit(cctx._text, user, date)
1335 sr = sub.commit(cctx._text, user, date)
1336 newstate[s] = (newstate[s][0], sr)
1336 newstate[s] = (newstate[s][0], sr)
1337 subrepo.writestate(self, newstate)
1337 subrepo.writestate(self, newstate)
1338
1338
1339 p1, p2 = self.dirstate.parents()
1339 p1, p2 = self.dirstate.parents()
1340 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1340 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or '')
1341 try:
1341 try:
1342 self.hook("precommit", throw=True, parent1=hookp1,
1342 self.hook("precommit", throw=True, parent1=hookp1,
1343 parent2=hookp2)
1343 parent2=hookp2)
1344 ret = self.commitctx(cctx, True)
1344 ret = self.commitctx(cctx, True)
1345 except: # re-raises
1345 except: # re-raises
1346 if edited:
1346 if edited:
1347 self.ui.write(
1347 self.ui.write(
1348 _('note: commit message saved in %s\n') % msgfn)
1348 _('note: commit message saved in %s\n') % msgfn)
1349 raise
1349 raise
1350
1350
1351 # update bookmarks, dirstate and mergestate
1351 # update bookmarks, dirstate and mergestate
1352 bookmarks.update(self, [p1, p2], ret)
1352 bookmarks.update(self, [p1, p2], ret)
1353 cctx.markcommitted(ret)
1353 cctx.markcommitted(ret)
1354 ms.reset()
1354 ms.reset()
1355 finally:
1355 finally:
1356 wlock.release()
1356 wlock.release()
1357
1357
1358 def commithook(node=hex(ret), parent1=hookp1, parent2=hookp2):
1358 def commithook(node=hex(ret), parent1=hookp1, parent2=hookp2):
1359 self.hook("commit", node=node, parent1=parent1, parent2=parent2)
1359 self.hook("commit", node=node, parent1=parent1, parent2=parent2)
1360 self._afterlock(commithook)
1360 self._afterlock(commithook)
1361 return ret
1361 return ret
1362
1362
1363 @unfilteredmethod
1363 @unfilteredmethod
1364 def commitctx(self, ctx, error=False):
1364 def commitctx(self, ctx, error=False):
1365 """Add a new revision to current repository.
1365 """Add a new revision to current repository.
1366 Revision information is passed via the context argument.
1366 Revision information is passed via the context argument.
1367 """
1367 """
1368
1368
1369 tr = lock = None
1369 tr = lock = None
1370 removed = list(ctx.removed())
1370 removed = list(ctx.removed())
1371 p1, p2 = ctx.p1(), ctx.p2()
1371 p1, p2 = ctx.p1(), ctx.p2()
1372 user = ctx.user()
1372 user = ctx.user()
1373
1373
1374 lock = self.lock()
1374 lock = self.lock()
1375 try:
1375 try:
1376 tr = self.transaction("commit")
1376 tr = self.transaction("commit")
1377 trp = weakref.proxy(tr)
1377 trp = weakref.proxy(tr)
1378
1378
1379 if ctx.files():
1379 if ctx.files():
1380 m1 = p1.manifest().copy()
1380 m1 = p1.manifest().copy()
1381 m2 = p2.manifest()
1381 m2 = p2.manifest()
1382
1382
1383 # check in files
1383 # check in files
1384 new = {}
1384 new = {}
1385 changed = []
1385 changed = []
1386 linkrev = len(self)
1386 linkrev = len(self)
1387 for f in sorted(ctx.modified() + ctx.added()):
1387 for f in sorted(ctx.modified() + ctx.added()):
1388 self.ui.note(f + "\n")
1388 self.ui.note(f + "\n")
1389 try:
1389 try:
1390 fctx = ctx[f]
1390 fctx = ctx[f]
1391 if fctx is None:
1391 if fctx is None:
1392 removed.append(f)
1392 removed.append(f)
1393 else:
1393 else:
1394 new[f] = self._filecommit(fctx, m1, m2, linkrev,
1394 new[f] = self._filecommit(fctx, m1, m2, linkrev,
1395 trp, changed)
1395 trp, changed)
1396 m1.set(f, fctx.flags())
1396 m1.set(f, fctx.flags())
1397 except OSError, inst:
1397 except OSError, inst:
1398 self.ui.warn(_("trouble committing %s!\n") % f)
1398 self.ui.warn(_("trouble committing %s!\n") % f)
1399 raise
1399 raise
1400 except IOError, inst:
1400 except IOError, inst:
1401 errcode = getattr(inst, 'errno', errno.ENOENT)
1401 errcode = getattr(inst, 'errno', errno.ENOENT)
1402 if error or errcode and errcode != errno.ENOENT:
1402 if error or errcode and errcode != errno.ENOENT:
1403 self.ui.warn(_("trouble committing %s!\n") % f)
1403 self.ui.warn(_("trouble committing %s!\n") % f)
1404 raise
1404 raise
1405
1405
1406 # update manifest
1406 # update manifest
1407 m1.update(new)
1407 m1.update(new)
1408 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1408 removed = [f for f in sorted(removed) if f in m1 or f in m2]
1409 drop = [f for f in removed if f in m1]
1409 drop = [f for f in removed if f in m1]
1410 for f in drop:
1410 for f in drop:
1411 del m1[f]
1411 del m1[f]
1412 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1412 mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
1413 p2.manifestnode(), (new, drop))
1413 p2.manifestnode(), (new, drop))
1414 files = changed + removed
1414 files = changed + removed
1415 else:
1415 else:
1416 mn = p1.manifestnode()
1416 mn = p1.manifestnode()
1417 files = []
1417 files = []
1418
1418
1419 # update changelog
1419 # update changelog
1420 self.changelog.delayupdate()
1420 self.changelog.delayupdate()
1421 n = self.changelog.add(mn, files, ctx.description(),
1421 n = self.changelog.add(mn, files, ctx.description(),
1422 trp, p1.node(), p2.node(),
1422 trp, p1.node(), p2.node(),
1423 user, ctx.date(), ctx.extra().copy())
1423 user, ctx.date(), ctx.extra().copy())
1424 p = lambda: self.changelog.writepending() and self.root or ""
1424 p = lambda: self.changelog.writepending() and self.root or ""
1425 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1425 xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
1426 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1426 self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
1427 parent2=xp2, pending=p)
1427 parent2=xp2, pending=p)
1428 self.changelog.finalize(trp)
1428 self.changelog.finalize(trp)
1429 # set the new commit is proper phase
1429 # set the new commit is proper phase
1430 targetphase = subrepo.newcommitphase(self.ui, ctx)
1430 targetphase = subrepo.newcommitphase(self.ui, ctx)
1431 if targetphase:
1431 if targetphase:
1432 # retract boundary do not alter parent changeset.
1432 # retract boundary do not alter parent changeset.
1433 # if a parent have higher the resulting phase will
1433 # if a parent have higher the resulting phase will
1434 # be compliant anyway
1434 # be compliant anyway
1435 #
1435 #
1436 # if minimal phase was 0 we don't need to retract anything
1436 # if minimal phase was 0 we don't need to retract anything
1437 phases.retractboundary(self, tr, targetphase, [n])
1437 phases.retractboundary(self, tr, targetphase, [n])
1438 tr.close()
1438 tr.close()
1439 branchmap.updatecache(self.filtered('served'))
1439 branchmap.updatecache(self.filtered('served'))
1440 return n
1440 return n
1441 finally:
1441 finally:
1442 if tr:
1442 if tr:
1443 tr.release()
1443 tr.release()
1444 lock.release()
1444 lock.release()
1445
1445
1446 @unfilteredmethod
1446 @unfilteredmethod
1447 def destroying(self):
1447 def destroying(self):
1448 '''Inform the repository that nodes are about to be destroyed.
1448 '''Inform the repository that nodes are about to be destroyed.
1449 Intended for use by strip and rollback, so there's a common
1449 Intended for use by strip and rollback, so there's a common
1450 place for anything that has to be done before destroying history.
1450 place for anything that has to be done before destroying history.
1451
1451
1452 This is mostly useful for saving state that is in memory and waiting
1452 This is mostly useful for saving state that is in memory and waiting
1453 to be flushed when the current lock is released. Because a call to
1453 to be flushed when the current lock is released. Because a call to
1454 destroyed is imminent, the repo will be invalidated causing those
1454 destroyed is imminent, the repo will be invalidated causing those
1455 changes to stay in memory (waiting for the next unlock), or vanish
1455 changes to stay in memory (waiting for the next unlock), or vanish
1456 completely.
1456 completely.
1457 '''
1457 '''
1458 # When using the same lock to commit and strip, the phasecache is left
1458 # When using the same lock to commit and strip, the phasecache is left
1459 # dirty after committing. Then when we strip, the repo is invalidated,
1459 # dirty after committing. Then when we strip, the repo is invalidated,
1460 # causing those changes to disappear.
1460 # causing those changes to disappear.
1461 if '_phasecache' in vars(self):
1461 if '_phasecache' in vars(self):
1462 self._phasecache.write()
1462 self._phasecache.write()
1463
1463
1464 @unfilteredmethod
1464 @unfilteredmethod
1465 def destroyed(self):
1465 def destroyed(self):
1466 '''Inform the repository that nodes have been destroyed.
1466 '''Inform the repository that nodes have been destroyed.
1467 Intended for use by strip and rollback, so there's a common
1467 Intended for use by strip and rollback, so there's a common
1468 place for anything that has to be done after destroying history.
1468 place for anything that has to be done after destroying history.
1469 '''
1469 '''
1470 # When one tries to:
1470 # When one tries to:
1471 # 1) destroy nodes thus calling this method (e.g. strip)
1471 # 1) destroy nodes thus calling this method (e.g. strip)
1472 # 2) use phasecache somewhere (e.g. commit)
1472 # 2) use phasecache somewhere (e.g. commit)
1473 #
1473 #
1474 # then 2) will fail because the phasecache contains nodes that were
1474 # then 2) will fail because the phasecache contains nodes that were
1475 # removed. We can either remove phasecache from the filecache,
1475 # removed. We can either remove phasecache from the filecache,
1476 # causing it to reload next time it is accessed, or simply filter
1476 # causing it to reload next time it is accessed, or simply filter
1477 # the removed nodes now and write the updated cache.
1477 # the removed nodes now and write the updated cache.
1478 self._phasecache.filterunknown(self)
1478 self._phasecache.filterunknown(self)
1479 self._phasecache.write()
1479 self._phasecache.write()
1480
1480
1481 # update the 'served' branch cache to help read only server process
1481 # update the 'served' branch cache to help read only server process
1482 # Thanks to branchcache collaboration this is done from the nearest
1482 # Thanks to branchcache collaboration this is done from the nearest
1483 # filtered subset and it is expected to be fast.
1483 # filtered subset and it is expected to be fast.
1484 branchmap.updatecache(self.filtered('served'))
1484 branchmap.updatecache(self.filtered('served'))
1485
1485
1486 # Ensure the persistent tag cache is updated. Doing it now
1486 # Ensure the persistent tag cache is updated. Doing it now
1487 # means that the tag cache only has to worry about destroyed
1487 # means that the tag cache only has to worry about destroyed
1488 # heads immediately after a strip/rollback. That in turn
1488 # heads immediately after a strip/rollback. That in turn
1489 # guarantees that "cachetip == currenttip" (comparing both rev
1489 # guarantees that "cachetip == currenttip" (comparing both rev
1490 # and node) always means no nodes have been added or destroyed.
1490 # and node) always means no nodes have been added or destroyed.
1491
1491
1492 # XXX this is suboptimal when qrefresh'ing: we strip the current
1492 # XXX this is suboptimal when qrefresh'ing: we strip the current
1493 # head, refresh the tag cache, then immediately add a new head.
1493 # head, refresh the tag cache, then immediately add a new head.
1494 # But I think doing it this way is necessary for the "instant
1494 # But I think doing it this way is necessary for the "instant
1495 # tag cache retrieval" case to work.
1495 # tag cache retrieval" case to work.
1496 self.invalidate()
1496 self.invalidate()
1497
1497
1498 def walk(self, match, node=None):
1498 def walk(self, match, node=None):
1499 '''
1499 '''
1500 walk recursively through the directory tree or a given
1500 walk recursively through the directory tree or a given
1501 changeset, finding all files matched by the match
1501 changeset, finding all files matched by the match
1502 function
1502 function
1503 '''
1503 '''
1504 return self[node].walk(match)
1504 return self[node].walk(match)
1505
1505
1506 def status(self, node1='.', node2=None, match=None,
1506 def status(self, node1='.', node2=None, match=None,
1507 ignored=False, clean=False, unknown=False,
1507 ignored=False, clean=False, unknown=False,
1508 listsubrepos=False):
1508 listsubrepos=False):
1509 '''a convenience method that calls node1.status(node2)'''
1509 '''a convenience method that calls node1.status(node2)'''
1510 return self[node1].status(node2, match, ignored, clean, unknown,
1510 return self[node1].status(node2, match, ignored, clean, unknown,
1511 listsubrepos)
1511 listsubrepos)
1512
1512
1513 def heads(self, start=None):
1513 def heads(self, start=None):
1514 heads = self.changelog.heads(start)
1514 heads = self.changelog.heads(start)
1515 # sort the output in rev descending order
1515 # sort the output in rev descending order
1516 return sorted(heads, key=self.changelog.rev, reverse=True)
1516 return sorted(heads, key=self.changelog.rev, reverse=True)
1517
1517
1518 def branchheads(self, branch=None, start=None, closed=False):
1518 def branchheads(self, branch=None, start=None, closed=False):
1519 '''return a (possibly filtered) list of heads for the given branch
1519 '''return a (possibly filtered) list of heads for the given branch
1520
1520
1521 Heads are returned in topological order, from newest to oldest.
1521 Heads are returned in topological order, from newest to oldest.
1522 If branch is None, use the dirstate branch.
1522 If branch is None, use the dirstate branch.
1523 If start is not None, return only heads reachable from start.
1523 If start is not None, return only heads reachable from start.
1524 If closed is True, return heads that are marked as closed as well.
1524 If closed is True, return heads that are marked as closed as well.
1525 '''
1525 '''
1526 if branch is None:
1526 if branch is None:
1527 branch = self[None].branch()
1527 branch = self[None].branch()
1528 branches = self.branchmap()
1528 branches = self.branchmap()
1529 if branch not in branches:
1529 if branch not in branches:
1530 return []
1530 return []
1531 # the cache returns heads ordered lowest to highest
1531 # the cache returns heads ordered lowest to highest
1532 bheads = list(reversed(branches.branchheads(branch, closed=closed)))
1532 bheads = list(reversed(branches.branchheads(branch, closed=closed)))
1533 if start is not None:
1533 if start is not None:
1534 # filter out the heads that cannot be reached from startrev
1534 # filter out the heads that cannot be reached from startrev
1535 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1535 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
1536 bheads = [h for h in bheads if h in fbheads]
1536 bheads = [h for h in bheads if h in fbheads]
1537 return bheads
1537 return bheads
1538
1538
1539 def branches(self, nodes):
1539 def branches(self, nodes):
1540 if not nodes:
1540 if not nodes:
1541 nodes = [self.changelog.tip()]
1541 nodes = [self.changelog.tip()]
1542 b = []
1542 b = []
1543 for n in nodes:
1543 for n in nodes:
1544 t = n
1544 t = n
1545 while True:
1545 while True:
1546 p = self.changelog.parents(n)
1546 p = self.changelog.parents(n)
1547 if p[1] != nullid or p[0] == nullid:
1547 if p[1] != nullid or p[0] == nullid:
1548 b.append((t, n, p[0], p[1]))
1548 b.append((t, n, p[0], p[1]))
1549 break
1549 break
1550 n = p[0]
1550 n = p[0]
1551 return b
1551 return b
1552
1552
1553 def between(self, pairs):
1553 def between(self, pairs):
1554 r = []
1554 r = []
1555
1555
1556 for top, bottom in pairs:
1556 for top, bottom in pairs:
1557 n, l, i = top, [], 0
1557 n, l, i = top, [], 0
1558 f = 1
1558 f = 1
1559
1559
1560 while n != bottom and n != nullid:
1560 while n != bottom and n != nullid:
1561 p = self.changelog.parents(n)[0]
1561 p = self.changelog.parents(n)[0]
1562 if i == f:
1562 if i == f:
1563 l.append(n)
1563 l.append(n)
1564 f = f * 2
1564 f = f * 2
1565 n = p
1565 n = p
1566 i += 1
1566 i += 1
1567
1567
1568 r.append(l)
1568 r.append(l)
1569
1569
1570 return r
1570 return r
1571
1571
1572 def pull(self, remote, heads=None, force=False):
1572 def pull(self, remote, heads=None, force=False):
1573 return exchange.pull (self, remote, heads, force)
1573 return exchange.pull (self, remote, heads, force)
1574
1574
1575 def checkpush(self, pushop):
1575 def checkpush(self, pushop):
1576 """Extensions can override this function if additional checks have
1576 """Extensions can override this function if additional checks have
1577 to be performed before pushing, or call it if they override push
1577 to be performed before pushing, or call it if they override push
1578 command.
1578 command.
1579 """
1579 """
1580 pass
1580 pass
1581
1581
1582 @unfilteredpropertycache
1582 @unfilteredpropertycache
1583 def prepushoutgoinghooks(self):
1583 def prepushoutgoinghooks(self):
1584 """Return util.hooks consists of "(repo, remote, outgoing)"
1584 """Return util.hooks consists of "(repo, remote, outgoing)"
1585 functions, which are called before pushing changesets.
1585 functions, which are called before pushing changesets.
1586 """
1586 """
1587 return util.hooks()
1587 return util.hooks()
1588
1588
1589 def push(self, remote, force=False, revs=None, newbranch=False):
1589 def push(self, remote, force=False, revs=None, newbranch=False):
1590 return exchange.push(self, remote, force, revs, newbranch)
1590 return exchange.push(self, remote, force, revs, newbranch)
1591
1591
1592 def stream_in(self, remote, requirements):
1592 def stream_in(self, remote, requirements):
1593 lock = self.lock()
1593 lock = self.lock()
1594 try:
1594 try:
1595 # Save remote branchmap. We will use it later
1595 # Save remote branchmap. We will use it later
1596 # to speed up branchcache creation
1596 # to speed up branchcache creation
1597 rbranchmap = None
1597 rbranchmap = None
1598 if remote.capable("branchmap"):
1598 if remote.capable("branchmap"):
1599 rbranchmap = remote.branchmap()
1599 rbranchmap = remote.branchmap()
1600
1600
1601 fp = remote.stream_out()
1601 fp = remote.stream_out()
1602 l = fp.readline()
1602 l = fp.readline()
1603 try:
1603 try:
1604 resp = int(l)
1604 resp = int(l)
1605 except ValueError:
1605 except ValueError:
1606 raise error.ResponseError(
1606 raise error.ResponseError(
1607 _('unexpected response from remote server:'), l)
1607 _('unexpected response from remote server:'), l)
1608 if resp == 1:
1608 if resp == 1:
1609 raise util.Abort(_('operation forbidden by server'))
1609 raise util.Abort(_('operation forbidden by server'))
1610 elif resp == 2:
1610 elif resp == 2:
1611 raise util.Abort(_('locking the remote repository failed'))
1611 raise util.Abort(_('locking the remote repository failed'))
1612 elif resp != 0:
1612 elif resp != 0:
1613 raise util.Abort(_('the server sent an unknown error code'))
1613 raise util.Abort(_('the server sent an unknown error code'))
1614 self.ui.status(_('streaming all changes\n'))
1614 self.ui.status(_('streaming all changes\n'))
1615 l = fp.readline()
1615 l = fp.readline()
1616 try:
1616 try:
1617 total_files, total_bytes = map(int, l.split(' ', 1))
1617 total_files, total_bytes = map(int, l.split(' ', 1))
1618 except (ValueError, TypeError):
1618 except (ValueError, TypeError):
1619 raise error.ResponseError(
1619 raise error.ResponseError(
1620 _('unexpected response from remote server:'), l)
1620 _('unexpected response from remote server:'), l)
1621 self.ui.status(_('%d files to transfer, %s of data\n') %
1621 self.ui.status(_('%d files to transfer, %s of data\n') %
1622 (total_files, util.bytecount(total_bytes)))
1622 (total_files, util.bytecount(total_bytes)))
1623 handled_bytes = 0
1623 handled_bytes = 0
1624 self.ui.progress(_('clone'), 0, total=total_bytes)
1624 self.ui.progress(_('clone'), 0, total=total_bytes)
1625 start = time.time()
1625 start = time.time()
1626
1626
1627 tr = self.transaction(_('clone'))
1627 tr = self.transaction(_('clone'))
1628 try:
1628 try:
1629 for i in xrange(total_files):
1629 for i in xrange(total_files):
1630 # XXX doesn't support '\n' or '\r' in filenames
1630 # XXX doesn't support '\n' or '\r' in filenames
1631 l = fp.readline()
1631 l = fp.readline()
1632 try:
1632 try:
1633 name, size = l.split('\0', 1)
1633 name, size = l.split('\0', 1)
1634 size = int(size)
1634 size = int(size)
1635 except (ValueError, TypeError):
1635 except (ValueError, TypeError):
1636 raise error.ResponseError(
1636 raise error.ResponseError(
1637 _('unexpected response from remote server:'), l)
1637 _('unexpected response from remote server:'), l)
1638 if self.ui.debugflag:
1638 if self.ui.debugflag:
1639 self.ui.debug('adding %s (%s)\n' %
1639 self.ui.debug('adding %s (%s)\n' %
1640 (name, util.bytecount(size)))
1640 (name, util.bytecount(size)))
1641 # for backwards compat, name was partially encoded
1641 # for backwards compat, name was partially encoded
1642 ofp = self.sopener(store.decodedir(name), 'w')
1642 ofp = self.sopener(store.decodedir(name), 'w')
1643 for chunk in util.filechunkiter(fp, limit=size):
1643 for chunk in util.filechunkiter(fp, limit=size):
1644 handled_bytes += len(chunk)
1644 handled_bytes += len(chunk)
1645 self.ui.progress(_('clone'), handled_bytes,
1645 self.ui.progress(_('clone'), handled_bytes,
1646 total=total_bytes)
1646 total=total_bytes)
1647 ofp.write(chunk)
1647 ofp.write(chunk)
1648 ofp.close()
1648 ofp.close()
1649 tr.close()
1649 tr.close()
1650 finally:
1650 finally:
1651 tr.release()
1651 tr.release()
1652
1652
1653 # Writing straight to files circumvented the inmemory caches
1653 # Writing straight to files circumvented the inmemory caches
1654 self.invalidate()
1654 self.invalidate()
1655
1655
1656 elapsed = time.time() - start
1656 elapsed = time.time() - start
1657 if elapsed <= 0:
1657 if elapsed <= 0:
1658 elapsed = 0.001
1658 elapsed = 0.001
1659 self.ui.progress(_('clone'), None)
1659 self.ui.progress(_('clone'), None)
1660 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
1660 self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
1661 (util.bytecount(total_bytes), elapsed,
1661 (util.bytecount(total_bytes), elapsed,
1662 util.bytecount(total_bytes / elapsed)))
1662 util.bytecount(total_bytes / elapsed)))
1663
1663
1664 # new requirements = old non-format requirements +
1664 # new requirements = old non-format requirements +
1665 # new format-related
1665 # new format-related
1666 # requirements from the streamed-in repository
1666 # requirements from the streamed-in repository
1667 requirements.update(set(self.requirements) - self.supportedformats)
1667 requirements.update(set(self.requirements) - self.supportedformats)
1668 self._applyrequirements(requirements)
1668 self._applyrequirements(requirements)
1669 self._writerequirements()
1669 self._writerequirements()
1670
1670
1671 if rbranchmap:
1671 if rbranchmap:
1672 rbheads = []
1672 rbheads = []
1673 for bheads in rbranchmap.itervalues():
1673 for bheads in rbranchmap.itervalues():
1674 rbheads.extend(bheads)
1674 rbheads.extend(bheads)
1675
1675
1676 if rbheads:
1676 if rbheads:
1677 rtiprev = max((int(self.changelog.rev(node))
1677 rtiprev = max((int(self.changelog.rev(node))
1678 for node in rbheads))
1678 for node in rbheads))
1679 cache = branchmap.branchcache(rbranchmap,
1679 cache = branchmap.branchcache(rbranchmap,
1680 self[rtiprev].node(),
1680 self[rtiprev].node(),
1681 rtiprev)
1681 rtiprev)
1682 # Try to stick it as low as possible
1682 # Try to stick it as low as possible
1683 # filter above served are unlikely to be fetch from a clone
1683 # filter above served are unlikely to be fetch from a clone
1684 for candidate in ('base', 'immutable', 'served'):
1684 for candidate in ('base', 'immutable', 'served'):
1685 rview = self.filtered(candidate)
1685 rview = self.filtered(candidate)
1686 if cache.validfor(rview):
1686 if cache.validfor(rview):
1687 self._branchcaches[candidate] = cache
1687 self._branchcaches[candidate] = cache
1688 cache.write(rview)
1688 cache.write(rview)
1689 break
1689 break
1690 self.invalidate()
1690 self.invalidate()
1691 return len(self.heads()) + 1
1691 return len(self.heads()) + 1
1692 finally:
1692 finally:
1693 lock.release()
1693 lock.release()
1694
1694
1695 def clone(self, remote, heads=[], stream=False):
1695 def clone(self, remote, heads=[], stream=False):
1696 '''clone remote repository.
1696 '''clone remote repository.
1697
1697
1698 keyword arguments:
1698 keyword arguments:
1699 heads: list of revs to clone (forces use of pull)
1699 heads: list of revs to clone (forces use of pull)
1700 stream: use streaming clone if possible'''
1700 stream: use streaming clone if possible'''
1701
1701
1702 # now, all clients that can request uncompressed clones can
1702 # now, all clients that can request uncompressed clones can
1703 # read repo formats supported by all servers that can serve
1703 # read repo formats supported by all servers that can serve
1704 # them.
1704 # them.
1705
1705
1706 # if revlog format changes, client will have to check version
1706 # if revlog format changes, client will have to check version
1707 # and format flags on "stream" capability, and use
1707 # and format flags on "stream" capability, and use
1708 # uncompressed only if compatible.
1708 # uncompressed only if compatible.
1709
1709
1710 if not stream:
1710 if not stream:
1711 # if the server explicitly prefers to stream (for fast LANs)
1711 # if the server explicitly prefers to stream (for fast LANs)
1712 stream = remote.capable('stream-preferred')
1712 stream = remote.capable('stream-preferred')
1713
1713
1714 if stream and not heads:
1714 if stream and not heads:
1715 # 'stream' means remote revlog format is revlogv1 only
1715 # 'stream' means remote revlog format is revlogv1 only
1716 if remote.capable('stream'):
1716 if remote.capable('stream'):
1717 return self.stream_in(remote, set(('revlogv1',)))
1717 return self.stream_in(remote, set(('revlogv1',)))
1718 # otherwise, 'streamreqs' contains the remote revlog format
1718 # otherwise, 'streamreqs' contains the remote revlog format
1719 streamreqs = remote.capable('streamreqs')
1719 streamreqs = remote.capable('streamreqs')
1720 if streamreqs:
1720 if streamreqs:
1721 streamreqs = set(streamreqs.split(','))
1721 streamreqs = set(streamreqs.split(','))
1722 # if we support it, stream in and adjust our requirements
1722 # if we support it, stream in and adjust our requirements
1723 if not streamreqs - self.supportedformats:
1723 if not streamreqs - self.supportedformats:
1724 return self.stream_in(remote, streamreqs)
1724 return self.stream_in(remote, streamreqs)
1725 return self.pull(remote, heads)
1725 return self.pull(remote, heads)
1726
1726
1727 def pushkey(self, namespace, key, old, new):
1727 def pushkey(self, namespace, key, old, new):
1728 self.hook('prepushkey', throw=True, namespace=namespace, key=key,
1728 self.hook('prepushkey', throw=True, namespace=namespace, key=key,
1729 old=old, new=new)
1729 old=old, new=new)
1730 self.ui.debug('pushing key for "%s:%s"\n' % (namespace, key))
1730 self.ui.debug('pushing key for "%s:%s"\n' % (namespace, key))
1731 ret = pushkey.push(self, namespace, key, old, new)
1731 ret = pushkey.push(self, namespace, key, old, new)
1732 self.hook('pushkey', namespace=namespace, key=key, old=old, new=new,
1732 self.hook('pushkey', namespace=namespace, key=key, old=old, new=new,
1733 ret=ret)
1733 ret=ret)
1734 return ret
1734 return ret
1735
1735
1736 def listkeys(self, namespace):
1736 def listkeys(self, namespace):
1737 self.hook('prelistkeys', throw=True, namespace=namespace)
1737 self.hook('prelistkeys', throw=True, namespace=namespace)
1738 self.ui.debug('listing keys for "%s"\n' % namespace)
1738 self.ui.debug('listing keys for "%s"\n' % namespace)
1739 values = pushkey.list(self, namespace)
1739 values = pushkey.list(self, namespace)
1740 self.hook('listkeys', namespace=namespace, values=values)
1740 self.hook('listkeys', namespace=namespace, values=values)
1741 return values
1741 return values
1742
1742
1743 def debugwireargs(self, one, two, three=None, four=None, five=None):
1743 def debugwireargs(self, one, two, three=None, four=None, five=None):
1744 '''used to test argument passing over the wire'''
1744 '''used to test argument passing over the wire'''
1745 return "%s %s %s %s %s" % (one, two, three, four, five)
1745 return "%s %s %s %s %s" % (one, two, three, four, five)
1746
1746
1747 def savecommitmessage(self, text):
1747 def savecommitmessage(self, text):
1748 fp = self.opener('last-message.txt', 'wb')
1748 fp = self.opener('last-message.txt', 'wb')
1749 try:
1749 try:
1750 fp.write(text)
1750 fp.write(text)
1751 finally:
1751 finally:
1752 fp.close()
1752 fp.close()
1753 return self.pathto(fp.name[len(self.root) + 1:])
1753 return self.pathto(fp.name[len(self.root) + 1:])
1754
1754
1755 # used to avoid circular references so destructors work
1755 # used to avoid circular references so destructors work
1756 def aftertrans(files):
1756 def aftertrans(files):
1757 renamefiles = [tuple(t) for t in files]
1757 renamefiles = [tuple(t) for t in files]
1758 def a():
1758 def a():
1759 for vfs, src, dest in renamefiles:
1759 for vfs, src, dest in renamefiles:
1760 try:
1760 try:
1761 vfs.rename(src, dest)
1761 vfs.rename(src, dest)
1762 except OSError: # journal file does not yet exist
1762 except OSError: # journal file does not yet exist
1763 pass
1763 pass
1764 return a
1764 return a
1765
1765
1766 def undoname(fn):
1766 def undoname(fn):
1767 base, name = os.path.split(fn)
1767 base, name = os.path.split(fn)
1768 assert name.startswith('journal')
1768 assert name.startswith('journal')
1769 return os.path.join(base, name.replace('journal', 'undo', 1))
1769 return os.path.join(base, name.replace('journal', 'undo', 1))
1770
1770
1771 def instance(ui, path, create):
1771 def instance(ui, path, create):
1772 return localrepository(ui, util.urllocalpath(path), create)
1772 return localrepository(ui, util.urllocalpath(path), create)
1773
1773
1774 def islocal(path):
1774 def islocal(path):
1775 return True
1775 return True
General Comments 0
You need to be logged in to leave comments. Login now