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