##// END OF EJS Templates
cmdutil: don't use repr on paths in pathauditor - it looks strange on windows
Mads Kiilerich -
r15436:18f1bb70 default
parent child Browse files
Show More
@@ -1,803 +1,803
1 # scmutil.py - Mercurial core utility functions
1 # scmutil.py - Mercurial core utility functions
2 #
2 #
3 # Copyright Matt Mackall <mpm@selenic.com>
3 # Copyright Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 from i18n import _
8 from i18n import _
9 import util, error, osutil, revset, similar, encoding
9 import util, error, osutil, revset, similar, encoding
10 import match as matchmod
10 import match as matchmod
11 import os, errno, re, stat, sys, glob
11 import os, errno, re, stat, sys, glob
12
12
13 def checkfilename(f):
13 def checkfilename(f):
14 '''Check that the filename f is an acceptable filename for a tracked file'''
14 '''Check that the filename f is an acceptable filename for a tracked file'''
15 if '\r' in f or '\n' in f:
15 if '\r' in f or '\n' in f:
16 raise util.Abort(_("'\\n' and '\\r' disallowed in filenames: %r") % f)
16 raise util.Abort(_("'\\n' and '\\r' disallowed in filenames: %r") % f)
17
17
18 def checkportable(ui, f):
18 def checkportable(ui, f):
19 '''Check if filename f is portable and warn or abort depending on config'''
19 '''Check if filename f is portable and warn or abort depending on config'''
20 checkfilename(f)
20 checkfilename(f)
21 abort, warn = checkportabilityalert(ui)
21 abort, warn = checkportabilityalert(ui)
22 if abort or warn:
22 if abort or warn:
23 msg = util.checkwinfilename(f)
23 msg = util.checkwinfilename(f)
24 if msg:
24 if msg:
25 msg = "%s: %r" % (msg, f)
25 msg = "%s: %r" % (msg, f)
26 if abort:
26 if abort:
27 raise util.Abort(msg)
27 raise util.Abort(msg)
28 ui.warn(_("warning: %s\n") % msg)
28 ui.warn(_("warning: %s\n") % msg)
29
29
30 def checkportabilityalert(ui):
30 def checkportabilityalert(ui):
31 '''check if the user's config requests nothing, a warning, or abort for
31 '''check if the user's config requests nothing, a warning, or abort for
32 non-portable filenames'''
32 non-portable filenames'''
33 val = ui.config('ui', 'portablefilenames', 'warn')
33 val = ui.config('ui', 'portablefilenames', 'warn')
34 lval = val.lower()
34 lval = val.lower()
35 bval = util.parsebool(val)
35 bval = util.parsebool(val)
36 abort = os.name == 'nt' or lval == 'abort'
36 abort = os.name == 'nt' or lval == 'abort'
37 warn = bval or lval == 'warn'
37 warn = bval or lval == 'warn'
38 if bval is None and not (warn or abort or lval == 'ignore'):
38 if bval is None and not (warn or abort or lval == 'ignore'):
39 raise error.ConfigError(
39 raise error.ConfigError(
40 _("ui.portablefilenames value is invalid ('%s')") % val)
40 _("ui.portablefilenames value is invalid ('%s')") % val)
41 return abort, warn
41 return abort, warn
42
42
43 class casecollisionauditor(object):
43 class casecollisionauditor(object):
44 def __init__(self, ui, abort, existingiter):
44 def __init__(self, ui, abort, existingiter):
45 self._ui = ui
45 self._ui = ui
46 self._abort = abort
46 self._abort = abort
47 self._map = {}
47 self._map = {}
48 for f in existingiter:
48 for f in existingiter:
49 self._map[encoding.lower(f)] = f
49 self._map[encoding.lower(f)] = f
50
50
51 def __call__(self, f):
51 def __call__(self, f):
52 fl = encoding.lower(f)
52 fl = encoding.lower(f)
53 map = self._map
53 map = self._map
54 if fl in map and map[fl] != f:
54 if fl in map and map[fl] != f:
55 msg = _('possible case-folding collision for %s') % f
55 msg = _('possible case-folding collision for %s') % f
56 if self._abort:
56 if self._abort:
57 raise util.Abort(msg)
57 raise util.Abort(msg)
58 self._ui.warn(_("warning: %s\n") % msg)
58 self._ui.warn(_("warning: %s\n") % msg)
59 map[fl] = f
59 map[fl] = f
60
60
61 class pathauditor(object):
61 class pathauditor(object):
62 '''ensure that a filesystem path contains no banned components.
62 '''ensure that a filesystem path contains no banned components.
63 the following properties of a path are checked:
63 the following properties of a path are checked:
64
64
65 - ends with a directory separator
65 - ends with a directory separator
66 - under top-level .hg
66 - under top-level .hg
67 - starts at the root of a windows drive
67 - starts at the root of a windows drive
68 - contains ".."
68 - contains ".."
69 - traverses a symlink (e.g. a/symlink_here/b)
69 - traverses a symlink (e.g. a/symlink_here/b)
70 - inside a nested repository (a callback can be used to approve
70 - inside a nested repository (a callback can be used to approve
71 some nested repositories, e.g., subrepositories)
71 some nested repositories, e.g., subrepositories)
72 '''
72 '''
73
73
74 def __init__(self, root, callback=None):
74 def __init__(self, root, callback=None):
75 self.audited = set()
75 self.audited = set()
76 self.auditeddir = set()
76 self.auditeddir = set()
77 self.root = root
77 self.root = root
78 self.callback = callback
78 self.callback = callback
79
79
80 def __call__(self, path):
80 def __call__(self, path):
81 '''Check the relative path.
81 '''Check the relative path.
82 path may contain a pattern (e.g. foodir/**.txt)'''
82 path may contain a pattern (e.g. foodir/**.txt)'''
83
83
84 if path in self.audited:
84 if path in self.audited:
85 return
85 return
86 # AIX ignores "/" at end of path, others raise EISDIR.
86 # AIX ignores "/" at end of path, others raise EISDIR.
87 if util.endswithsep(path):
87 if util.endswithsep(path):
88 raise util.Abort(_("path ends in directory separator: %s") % path)
88 raise util.Abort(_("path ends in directory separator: %s") % path)
89 normpath = os.path.normcase(path)
89 normpath = os.path.normcase(path)
90 parts = util.splitpath(normpath)
90 parts = util.splitpath(normpath)
91 if (os.path.splitdrive(path)[0]
91 if (os.path.splitdrive(path)[0]
92 or parts[0].lower() in ('.hg', '.hg.', '')
92 or parts[0].lower() in ('.hg', '.hg.', '')
93 or os.pardir in parts):
93 or os.pardir in parts):
94 raise util.Abort(_("path contains illegal component: %s") % path)
94 raise util.Abort(_("path contains illegal component: %s") % path)
95 if '.hg' in path.lower():
95 if '.hg' in path.lower():
96 lparts = [p.lower() for p in parts]
96 lparts = [p.lower() for p in parts]
97 for p in '.hg', '.hg.':
97 for p in '.hg', '.hg.':
98 if p in lparts[1:]:
98 if p in lparts[1:]:
99 pos = lparts.index(p)
99 pos = lparts.index(p)
100 base = os.path.join(*parts[:pos])
100 base = os.path.join(*parts[:pos])
101 raise util.Abort(_('path %r is inside nested repo %r')
101 raise util.Abort(_("path '%s' is inside nested repo %r")
102 % (path, base))
102 % (path, base))
103
103
104 parts.pop()
104 parts.pop()
105 prefixes = []
105 prefixes = []
106 while parts:
106 while parts:
107 prefix = os.sep.join(parts)
107 prefix = os.sep.join(parts)
108 if prefix in self.auditeddir:
108 if prefix in self.auditeddir:
109 break
109 break
110 curpath = os.path.join(self.root, prefix)
110 curpath = os.path.join(self.root, prefix)
111 try:
111 try:
112 st = os.lstat(curpath)
112 st = os.lstat(curpath)
113 except OSError, err:
113 except OSError, err:
114 # EINVAL can be raised as invalid path syntax under win32.
114 # EINVAL can be raised as invalid path syntax under win32.
115 # They must be ignored for patterns can be checked too.
115 # They must be ignored for patterns can be checked too.
116 if err.errno not in (errno.ENOENT, errno.ENOTDIR, errno.EINVAL):
116 if err.errno not in (errno.ENOENT, errno.ENOTDIR, errno.EINVAL):
117 raise
117 raise
118 else:
118 else:
119 if stat.S_ISLNK(st.st_mode):
119 if stat.S_ISLNK(st.st_mode):
120 raise util.Abort(
120 raise util.Abort(
121 _('path %r traverses symbolic link %r')
121 _('path %r traverses symbolic link %r')
122 % (path, prefix))
122 % (path, prefix))
123 elif (stat.S_ISDIR(st.st_mode) and
123 elif (stat.S_ISDIR(st.st_mode) and
124 os.path.isdir(os.path.join(curpath, '.hg'))):
124 os.path.isdir(os.path.join(curpath, '.hg'))):
125 if not self.callback or not self.callback(curpath):
125 if not self.callback or not self.callback(curpath):
126 raise util.Abort(_('path %r is inside nested repo %r') %
126 raise util.Abort(_("path '%s' is inside nested repo %r") %
127 (path, prefix))
127 (path, prefix))
128 prefixes.append(prefix)
128 prefixes.append(prefix)
129 parts.pop()
129 parts.pop()
130
130
131 self.audited.add(path)
131 self.audited.add(path)
132 # only add prefixes to the cache after checking everything: we don't
132 # only add prefixes to the cache after checking everything: we don't
133 # want to add "foo/bar/baz" before checking if there's a "foo/.hg"
133 # want to add "foo/bar/baz" before checking if there's a "foo/.hg"
134 self.auditeddir.update(prefixes)
134 self.auditeddir.update(prefixes)
135
135
136 class abstractopener(object):
136 class abstractopener(object):
137 """Abstract base class; cannot be instantiated"""
137 """Abstract base class; cannot be instantiated"""
138
138
139 def __init__(self, *args, **kwargs):
139 def __init__(self, *args, **kwargs):
140 '''Prevent instantiation; don't call this from subclasses.'''
140 '''Prevent instantiation; don't call this from subclasses.'''
141 raise NotImplementedError('attempted instantiating ' + str(type(self)))
141 raise NotImplementedError('attempted instantiating ' + str(type(self)))
142
142
143 def read(self, path):
143 def read(self, path):
144 fp = self(path, 'rb')
144 fp = self(path, 'rb')
145 try:
145 try:
146 return fp.read()
146 return fp.read()
147 finally:
147 finally:
148 fp.close()
148 fp.close()
149
149
150 def write(self, path, data):
150 def write(self, path, data):
151 fp = self(path, 'wb')
151 fp = self(path, 'wb')
152 try:
152 try:
153 return fp.write(data)
153 return fp.write(data)
154 finally:
154 finally:
155 fp.close()
155 fp.close()
156
156
157 def append(self, path, data):
157 def append(self, path, data):
158 fp = self(path, 'ab')
158 fp = self(path, 'ab')
159 try:
159 try:
160 return fp.write(data)
160 return fp.write(data)
161 finally:
161 finally:
162 fp.close()
162 fp.close()
163
163
164 class opener(abstractopener):
164 class opener(abstractopener):
165 '''Open files relative to a base directory
165 '''Open files relative to a base directory
166
166
167 This class is used to hide the details of COW semantics and
167 This class is used to hide the details of COW semantics and
168 remote file access from higher level code.
168 remote file access from higher level code.
169 '''
169 '''
170 def __init__(self, base, audit=True):
170 def __init__(self, base, audit=True):
171 self.base = base
171 self.base = base
172 self._audit = audit
172 self._audit = audit
173 if audit:
173 if audit:
174 self.auditor = pathauditor(base)
174 self.auditor = pathauditor(base)
175 else:
175 else:
176 self.auditor = util.always
176 self.auditor = util.always
177 self.createmode = None
177 self.createmode = None
178 self._trustnlink = None
178 self._trustnlink = None
179
179
180 @util.propertycache
180 @util.propertycache
181 def _cansymlink(self):
181 def _cansymlink(self):
182 return util.checklink(self.base)
182 return util.checklink(self.base)
183
183
184 def _fixfilemode(self, name):
184 def _fixfilemode(self, name):
185 if self.createmode is None:
185 if self.createmode is None:
186 return
186 return
187 os.chmod(name, self.createmode & 0666)
187 os.chmod(name, self.createmode & 0666)
188
188
189 def __call__(self, path, mode="r", text=False, atomictemp=False):
189 def __call__(self, path, mode="r", text=False, atomictemp=False):
190 if self._audit:
190 if self._audit:
191 r = util.checkosfilename(path)
191 r = util.checkosfilename(path)
192 if r:
192 if r:
193 raise util.Abort("%s: %r" % (r, path))
193 raise util.Abort("%s: %r" % (r, path))
194 self.auditor(path)
194 self.auditor(path)
195 f = os.path.join(self.base, path)
195 f = os.path.join(self.base, path)
196
196
197 if not text and "b" not in mode:
197 if not text and "b" not in mode:
198 mode += "b" # for that other OS
198 mode += "b" # for that other OS
199
199
200 nlink = -1
200 nlink = -1
201 dirname, basename = os.path.split(f)
201 dirname, basename = os.path.split(f)
202 # If basename is empty, then the path is malformed because it points
202 # If basename is empty, then the path is malformed because it points
203 # to a directory. Let the posixfile() call below raise IOError.
203 # to a directory. Let the posixfile() call below raise IOError.
204 if basename and mode not in ('r', 'rb'):
204 if basename and mode not in ('r', 'rb'):
205 if atomictemp:
205 if atomictemp:
206 if not os.path.isdir(dirname):
206 if not os.path.isdir(dirname):
207 util.makedirs(dirname, self.createmode)
207 util.makedirs(dirname, self.createmode)
208 return util.atomictempfile(f, mode, self.createmode)
208 return util.atomictempfile(f, mode, self.createmode)
209 try:
209 try:
210 if 'w' in mode:
210 if 'w' in mode:
211 util.unlink(f)
211 util.unlink(f)
212 nlink = 0
212 nlink = 0
213 else:
213 else:
214 # nlinks() may behave differently for files on Windows
214 # nlinks() may behave differently for files on Windows
215 # shares if the file is open.
215 # shares if the file is open.
216 fd = util.posixfile(f)
216 fd = util.posixfile(f)
217 nlink = util.nlinks(f)
217 nlink = util.nlinks(f)
218 if nlink < 1:
218 if nlink < 1:
219 nlink = 2 # force mktempcopy (issue1922)
219 nlink = 2 # force mktempcopy (issue1922)
220 fd.close()
220 fd.close()
221 except (OSError, IOError), e:
221 except (OSError, IOError), e:
222 if e.errno != errno.ENOENT:
222 if e.errno != errno.ENOENT:
223 raise
223 raise
224 nlink = 0
224 nlink = 0
225 if not os.path.isdir(dirname):
225 if not os.path.isdir(dirname):
226 util.makedirs(dirname, self.createmode)
226 util.makedirs(dirname, self.createmode)
227 if nlink > 0:
227 if nlink > 0:
228 if self._trustnlink is None:
228 if self._trustnlink is None:
229 self._trustnlink = nlink > 1 or util.checknlink(f)
229 self._trustnlink = nlink > 1 or util.checknlink(f)
230 if nlink > 1 or not self._trustnlink:
230 if nlink > 1 or not self._trustnlink:
231 util.rename(util.mktempcopy(f), f)
231 util.rename(util.mktempcopy(f), f)
232 fp = util.posixfile(f, mode)
232 fp = util.posixfile(f, mode)
233 if nlink == 0:
233 if nlink == 0:
234 self._fixfilemode(f)
234 self._fixfilemode(f)
235 return fp
235 return fp
236
236
237 def symlink(self, src, dst):
237 def symlink(self, src, dst):
238 self.auditor(dst)
238 self.auditor(dst)
239 linkname = os.path.join(self.base, dst)
239 linkname = os.path.join(self.base, dst)
240 try:
240 try:
241 os.unlink(linkname)
241 os.unlink(linkname)
242 except OSError:
242 except OSError:
243 pass
243 pass
244
244
245 dirname = os.path.dirname(linkname)
245 dirname = os.path.dirname(linkname)
246 if not os.path.exists(dirname):
246 if not os.path.exists(dirname):
247 util.makedirs(dirname, self.createmode)
247 util.makedirs(dirname, self.createmode)
248
248
249 if self._cansymlink:
249 if self._cansymlink:
250 try:
250 try:
251 os.symlink(src, linkname)
251 os.symlink(src, linkname)
252 except OSError, err:
252 except OSError, err:
253 raise OSError(err.errno, _('could not symlink to %r: %s') %
253 raise OSError(err.errno, _('could not symlink to %r: %s') %
254 (src, err.strerror), linkname)
254 (src, err.strerror), linkname)
255 else:
255 else:
256 f = self(dst, "w")
256 f = self(dst, "w")
257 f.write(src)
257 f.write(src)
258 f.close()
258 f.close()
259 self._fixfilemode(dst)
259 self._fixfilemode(dst)
260
260
261 def audit(self, path):
261 def audit(self, path):
262 self.auditor(path)
262 self.auditor(path)
263
263
264 class filteropener(abstractopener):
264 class filteropener(abstractopener):
265 '''Wrapper opener for filtering filenames with a function.'''
265 '''Wrapper opener for filtering filenames with a function.'''
266
266
267 def __init__(self, opener, filter):
267 def __init__(self, opener, filter):
268 self._filter = filter
268 self._filter = filter
269 self._orig = opener
269 self._orig = opener
270
270
271 def __call__(self, path, *args, **kwargs):
271 def __call__(self, path, *args, **kwargs):
272 return self._orig(self._filter(path), *args, **kwargs)
272 return self._orig(self._filter(path), *args, **kwargs)
273
273
274 def canonpath(root, cwd, myname, auditor=None):
274 def canonpath(root, cwd, myname, auditor=None):
275 '''return the canonical path of myname, given cwd and root'''
275 '''return the canonical path of myname, given cwd and root'''
276 if util.endswithsep(root):
276 if util.endswithsep(root):
277 rootsep = root
277 rootsep = root
278 else:
278 else:
279 rootsep = root + os.sep
279 rootsep = root + os.sep
280 name = myname
280 name = myname
281 if not os.path.isabs(name):
281 if not os.path.isabs(name):
282 name = os.path.join(root, cwd, name)
282 name = os.path.join(root, cwd, name)
283 name = os.path.normpath(name)
283 name = os.path.normpath(name)
284 if auditor is None:
284 if auditor is None:
285 auditor = pathauditor(root)
285 auditor = pathauditor(root)
286 if name != rootsep and name.startswith(rootsep):
286 if name != rootsep and name.startswith(rootsep):
287 name = name[len(rootsep):]
287 name = name[len(rootsep):]
288 auditor(name)
288 auditor(name)
289 return util.pconvert(name)
289 return util.pconvert(name)
290 elif name == root:
290 elif name == root:
291 return ''
291 return ''
292 else:
292 else:
293 # Determine whether `name' is in the hierarchy at or beneath `root',
293 # Determine whether `name' is in the hierarchy at or beneath `root',
294 # by iterating name=dirname(name) until that causes no change (can't
294 # by iterating name=dirname(name) until that causes no change (can't
295 # check name == '/', because that doesn't work on windows). For each
295 # check name == '/', because that doesn't work on windows). For each
296 # `name', compare dev/inode numbers. If they match, the list `rel'
296 # `name', compare dev/inode numbers. If they match, the list `rel'
297 # holds the reversed list of components making up the relative file
297 # holds the reversed list of components making up the relative file
298 # name we want.
298 # name we want.
299 root_st = os.stat(root)
299 root_st = os.stat(root)
300 rel = []
300 rel = []
301 while True:
301 while True:
302 try:
302 try:
303 name_st = os.stat(name)
303 name_st = os.stat(name)
304 except OSError:
304 except OSError:
305 break
305 break
306 if util.samestat(name_st, root_st):
306 if util.samestat(name_st, root_st):
307 if not rel:
307 if not rel:
308 # name was actually the same as root (maybe a symlink)
308 # name was actually the same as root (maybe a symlink)
309 return ''
309 return ''
310 rel.reverse()
310 rel.reverse()
311 name = os.path.join(*rel)
311 name = os.path.join(*rel)
312 auditor(name)
312 auditor(name)
313 return util.pconvert(name)
313 return util.pconvert(name)
314 dirname, basename = os.path.split(name)
314 dirname, basename = os.path.split(name)
315 rel.append(basename)
315 rel.append(basename)
316 if dirname == name:
316 if dirname == name:
317 break
317 break
318 name = dirname
318 name = dirname
319
319
320 raise util.Abort('%s not under root' % myname)
320 raise util.Abort('%s not under root' % myname)
321
321
322 def walkrepos(path, followsym=False, seen_dirs=None, recurse=False):
322 def walkrepos(path, followsym=False, seen_dirs=None, recurse=False):
323 '''yield every hg repository under path, recursively.'''
323 '''yield every hg repository under path, recursively.'''
324 def errhandler(err):
324 def errhandler(err):
325 if err.filename == path:
325 if err.filename == path:
326 raise err
326 raise err
327 samestat = getattr(os.path, 'samestat', None)
327 samestat = getattr(os.path, 'samestat', None)
328 if followsym and samestat is not None:
328 if followsym and samestat is not None:
329 def adddir(dirlst, dirname):
329 def adddir(dirlst, dirname):
330 match = False
330 match = False
331 dirstat = os.stat(dirname)
331 dirstat = os.stat(dirname)
332 for lstdirstat in dirlst:
332 for lstdirstat in dirlst:
333 if samestat(dirstat, lstdirstat):
333 if samestat(dirstat, lstdirstat):
334 match = True
334 match = True
335 break
335 break
336 if not match:
336 if not match:
337 dirlst.append(dirstat)
337 dirlst.append(dirstat)
338 return not match
338 return not match
339 else:
339 else:
340 followsym = False
340 followsym = False
341
341
342 if (seen_dirs is None) and followsym:
342 if (seen_dirs is None) and followsym:
343 seen_dirs = []
343 seen_dirs = []
344 adddir(seen_dirs, path)
344 adddir(seen_dirs, path)
345 for root, dirs, files in os.walk(path, topdown=True, onerror=errhandler):
345 for root, dirs, files in os.walk(path, topdown=True, onerror=errhandler):
346 dirs.sort()
346 dirs.sort()
347 if '.hg' in dirs:
347 if '.hg' in dirs:
348 yield root # found a repository
348 yield root # found a repository
349 qroot = os.path.join(root, '.hg', 'patches')
349 qroot = os.path.join(root, '.hg', 'patches')
350 if os.path.isdir(os.path.join(qroot, '.hg')):
350 if os.path.isdir(os.path.join(qroot, '.hg')):
351 yield qroot # we have a patch queue repo here
351 yield qroot # we have a patch queue repo here
352 if recurse:
352 if recurse:
353 # avoid recursing inside the .hg directory
353 # avoid recursing inside the .hg directory
354 dirs.remove('.hg')
354 dirs.remove('.hg')
355 else:
355 else:
356 dirs[:] = [] # don't descend further
356 dirs[:] = [] # don't descend further
357 elif followsym:
357 elif followsym:
358 newdirs = []
358 newdirs = []
359 for d in dirs:
359 for d in dirs:
360 fname = os.path.join(root, d)
360 fname = os.path.join(root, d)
361 if adddir(seen_dirs, fname):
361 if adddir(seen_dirs, fname):
362 if os.path.islink(fname):
362 if os.path.islink(fname):
363 for hgname in walkrepos(fname, True, seen_dirs):
363 for hgname in walkrepos(fname, True, seen_dirs):
364 yield hgname
364 yield hgname
365 else:
365 else:
366 newdirs.append(d)
366 newdirs.append(d)
367 dirs[:] = newdirs
367 dirs[:] = newdirs
368
368
369 def osrcpath():
369 def osrcpath():
370 '''return default os-specific hgrc search path'''
370 '''return default os-specific hgrc search path'''
371 path = systemrcpath()
371 path = systemrcpath()
372 path.extend(userrcpath())
372 path.extend(userrcpath())
373 path = [os.path.normpath(f) for f in path]
373 path = [os.path.normpath(f) for f in path]
374 return path
374 return path
375
375
376 _rcpath = None
376 _rcpath = None
377
377
378 def rcpath():
378 def rcpath():
379 '''return hgrc search path. if env var HGRCPATH is set, use it.
379 '''return hgrc search path. if env var HGRCPATH is set, use it.
380 for each item in path, if directory, use files ending in .rc,
380 for each item in path, if directory, use files ending in .rc,
381 else use item.
381 else use item.
382 make HGRCPATH empty to only look in .hg/hgrc of current repo.
382 make HGRCPATH empty to only look in .hg/hgrc of current repo.
383 if no HGRCPATH, use default os-specific path.'''
383 if no HGRCPATH, use default os-specific path.'''
384 global _rcpath
384 global _rcpath
385 if _rcpath is None:
385 if _rcpath is None:
386 if 'HGRCPATH' in os.environ:
386 if 'HGRCPATH' in os.environ:
387 _rcpath = []
387 _rcpath = []
388 for p in os.environ['HGRCPATH'].split(os.pathsep):
388 for p in os.environ['HGRCPATH'].split(os.pathsep):
389 if not p:
389 if not p:
390 continue
390 continue
391 p = util.expandpath(p)
391 p = util.expandpath(p)
392 if os.path.isdir(p):
392 if os.path.isdir(p):
393 for f, kind in osutil.listdir(p):
393 for f, kind in osutil.listdir(p):
394 if f.endswith('.rc'):
394 if f.endswith('.rc'):
395 _rcpath.append(os.path.join(p, f))
395 _rcpath.append(os.path.join(p, f))
396 else:
396 else:
397 _rcpath.append(p)
397 _rcpath.append(p)
398 else:
398 else:
399 _rcpath = osrcpath()
399 _rcpath = osrcpath()
400 return _rcpath
400 return _rcpath
401
401
402 if os.name != 'nt':
402 if os.name != 'nt':
403
403
404 def rcfiles(path):
404 def rcfiles(path):
405 rcs = [os.path.join(path, 'hgrc')]
405 rcs = [os.path.join(path, 'hgrc')]
406 rcdir = os.path.join(path, 'hgrc.d')
406 rcdir = os.path.join(path, 'hgrc.d')
407 try:
407 try:
408 rcs.extend([os.path.join(rcdir, f)
408 rcs.extend([os.path.join(rcdir, f)
409 for f, kind in osutil.listdir(rcdir)
409 for f, kind in osutil.listdir(rcdir)
410 if f.endswith(".rc")])
410 if f.endswith(".rc")])
411 except OSError:
411 except OSError:
412 pass
412 pass
413 return rcs
413 return rcs
414
414
415 def systemrcpath():
415 def systemrcpath():
416 path = []
416 path = []
417 # old mod_python does not set sys.argv
417 # old mod_python does not set sys.argv
418 if len(getattr(sys, 'argv', [])) > 0:
418 if len(getattr(sys, 'argv', [])) > 0:
419 p = os.path.dirname(os.path.dirname(sys.argv[0]))
419 p = os.path.dirname(os.path.dirname(sys.argv[0]))
420 path.extend(rcfiles(os.path.join(p, 'etc/mercurial')))
420 path.extend(rcfiles(os.path.join(p, 'etc/mercurial')))
421 path.extend(rcfiles('/etc/mercurial'))
421 path.extend(rcfiles('/etc/mercurial'))
422 return path
422 return path
423
423
424 def userrcpath():
424 def userrcpath():
425 return [os.path.expanduser('~/.hgrc')]
425 return [os.path.expanduser('~/.hgrc')]
426
426
427 else:
427 else:
428
428
429 _HKEY_LOCAL_MACHINE = 0x80000002L
429 _HKEY_LOCAL_MACHINE = 0x80000002L
430
430
431 def systemrcpath():
431 def systemrcpath():
432 '''return default os-specific hgrc search path'''
432 '''return default os-specific hgrc search path'''
433 rcpath = []
433 rcpath = []
434 filename = util.executablepath()
434 filename = util.executablepath()
435 # Use mercurial.ini found in directory with hg.exe
435 # Use mercurial.ini found in directory with hg.exe
436 progrc = os.path.join(os.path.dirname(filename), 'mercurial.ini')
436 progrc = os.path.join(os.path.dirname(filename), 'mercurial.ini')
437 if os.path.isfile(progrc):
437 if os.path.isfile(progrc):
438 rcpath.append(progrc)
438 rcpath.append(progrc)
439 return rcpath
439 return rcpath
440 # Use hgrc.d found in directory with hg.exe
440 # Use hgrc.d found in directory with hg.exe
441 progrcd = os.path.join(os.path.dirname(filename), 'hgrc.d')
441 progrcd = os.path.join(os.path.dirname(filename), 'hgrc.d')
442 if os.path.isdir(progrcd):
442 if os.path.isdir(progrcd):
443 for f, kind in osutil.listdir(progrcd):
443 for f, kind in osutil.listdir(progrcd):
444 if f.endswith('.rc'):
444 if f.endswith('.rc'):
445 rcpath.append(os.path.join(progrcd, f))
445 rcpath.append(os.path.join(progrcd, f))
446 return rcpath
446 return rcpath
447 # else look for a system rcpath in the registry
447 # else look for a system rcpath in the registry
448 value = util.lookupreg('SOFTWARE\\Mercurial', None,
448 value = util.lookupreg('SOFTWARE\\Mercurial', None,
449 _HKEY_LOCAL_MACHINE)
449 _HKEY_LOCAL_MACHINE)
450 if not isinstance(value, str) or not value:
450 if not isinstance(value, str) or not value:
451 return rcpath
451 return rcpath
452 value = value.replace('/', os.sep)
452 value = value.replace('/', os.sep)
453 for p in value.split(os.pathsep):
453 for p in value.split(os.pathsep):
454 if p.lower().endswith('mercurial.ini'):
454 if p.lower().endswith('mercurial.ini'):
455 rcpath.append(p)
455 rcpath.append(p)
456 elif os.path.isdir(p):
456 elif os.path.isdir(p):
457 for f, kind in osutil.listdir(p):
457 for f, kind in osutil.listdir(p):
458 if f.endswith('.rc'):
458 if f.endswith('.rc'):
459 rcpath.append(os.path.join(p, f))
459 rcpath.append(os.path.join(p, f))
460 return rcpath
460 return rcpath
461
461
462 def userrcpath():
462 def userrcpath():
463 '''return os-specific hgrc search path to the user dir'''
463 '''return os-specific hgrc search path to the user dir'''
464 home = os.path.expanduser('~')
464 home = os.path.expanduser('~')
465 path = [os.path.join(home, 'mercurial.ini'),
465 path = [os.path.join(home, 'mercurial.ini'),
466 os.path.join(home, '.hgrc')]
466 os.path.join(home, '.hgrc')]
467 userprofile = os.environ.get('USERPROFILE')
467 userprofile = os.environ.get('USERPROFILE')
468 if userprofile:
468 if userprofile:
469 path.append(os.path.join(userprofile, 'mercurial.ini'))
469 path.append(os.path.join(userprofile, 'mercurial.ini'))
470 path.append(os.path.join(userprofile, '.hgrc'))
470 path.append(os.path.join(userprofile, '.hgrc'))
471 return path
471 return path
472
472
473 def revsingle(repo, revspec, default='.'):
473 def revsingle(repo, revspec, default='.'):
474 if not revspec:
474 if not revspec:
475 return repo[default]
475 return repo[default]
476
476
477 l = revrange(repo, [revspec])
477 l = revrange(repo, [revspec])
478 if len(l) < 1:
478 if len(l) < 1:
479 raise util.Abort(_('empty revision set'))
479 raise util.Abort(_('empty revision set'))
480 return repo[l[-1]]
480 return repo[l[-1]]
481
481
482 def revpair(repo, revs):
482 def revpair(repo, revs):
483 if not revs:
483 if not revs:
484 return repo.dirstate.p1(), None
484 return repo.dirstate.p1(), None
485
485
486 l = revrange(repo, revs)
486 l = revrange(repo, revs)
487
487
488 if len(l) == 0:
488 if len(l) == 0:
489 return repo.dirstate.p1(), None
489 return repo.dirstate.p1(), None
490
490
491 if len(l) == 1:
491 if len(l) == 1:
492 return repo.lookup(l[0]), None
492 return repo.lookup(l[0]), None
493
493
494 return repo.lookup(l[0]), repo.lookup(l[-1])
494 return repo.lookup(l[0]), repo.lookup(l[-1])
495
495
496 _revrangesep = ':'
496 _revrangesep = ':'
497
497
498 def revrange(repo, revs):
498 def revrange(repo, revs):
499 """Yield revision as strings from a list of revision specifications."""
499 """Yield revision as strings from a list of revision specifications."""
500
500
501 def revfix(repo, val, defval):
501 def revfix(repo, val, defval):
502 if not val and val != 0 and defval is not None:
502 if not val and val != 0 and defval is not None:
503 return defval
503 return defval
504 return repo.changelog.rev(repo.lookup(val))
504 return repo.changelog.rev(repo.lookup(val))
505
505
506 seen, l = set(), []
506 seen, l = set(), []
507 for spec in revs:
507 for spec in revs:
508 # attempt to parse old-style ranges first to deal with
508 # attempt to parse old-style ranges first to deal with
509 # things like old-tag which contain query metacharacters
509 # things like old-tag which contain query metacharacters
510 try:
510 try:
511 if isinstance(spec, int):
511 if isinstance(spec, int):
512 seen.add(spec)
512 seen.add(spec)
513 l.append(spec)
513 l.append(spec)
514 continue
514 continue
515
515
516 if _revrangesep in spec:
516 if _revrangesep in spec:
517 start, end = spec.split(_revrangesep, 1)
517 start, end = spec.split(_revrangesep, 1)
518 start = revfix(repo, start, 0)
518 start = revfix(repo, start, 0)
519 end = revfix(repo, end, len(repo) - 1)
519 end = revfix(repo, end, len(repo) - 1)
520 step = start > end and -1 or 1
520 step = start > end and -1 or 1
521 for rev in xrange(start, end + step, step):
521 for rev in xrange(start, end + step, step):
522 if rev in seen:
522 if rev in seen:
523 continue
523 continue
524 seen.add(rev)
524 seen.add(rev)
525 l.append(rev)
525 l.append(rev)
526 continue
526 continue
527 elif spec and spec in repo: # single unquoted rev
527 elif spec and spec in repo: # single unquoted rev
528 rev = revfix(repo, spec, None)
528 rev = revfix(repo, spec, None)
529 if rev in seen:
529 if rev in seen:
530 continue
530 continue
531 seen.add(rev)
531 seen.add(rev)
532 l.append(rev)
532 l.append(rev)
533 continue
533 continue
534 except error.RepoLookupError:
534 except error.RepoLookupError:
535 pass
535 pass
536
536
537 # fall through to new-style queries if old-style fails
537 # fall through to new-style queries if old-style fails
538 m = revset.match(repo.ui, spec)
538 m = revset.match(repo.ui, spec)
539 for r in m(repo, range(len(repo))):
539 for r in m(repo, range(len(repo))):
540 if r not in seen:
540 if r not in seen:
541 l.append(r)
541 l.append(r)
542 seen.update(l)
542 seen.update(l)
543
543
544 return l
544 return l
545
545
546 def expandpats(pats):
546 def expandpats(pats):
547 if not util.expandglobs:
547 if not util.expandglobs:
548 return list(pats)
548 return list(pats)
549 ret = []
549 ret = []
550 for p in pats:
550 for p in pats:
551 kind, name = matchmod._patsplit(p, None)
551 kind, name = matchmod._patsplit(p, None)
552 if kind is None:
552 if kind is None:
553 try:
553 try:
554 globbed = glob.glob(name)
554 globbed = glob.glob(name)
555 except re.error:
555 except re.error:
556 globbed = [name]
556 globbed = [name]
557 if globbed:
557 if globbed:
558 ret.extend(globbed)
558 ret.extend(globbed)
559 continue
559 continue
560 ret.append(p)
560 ret.append(p)
561 return ret
561 return ret
562
562
563 def match(ctx, pats=[], opts={}, globbed=False, default='relpath'):
563 def match(ctx, pats=[], opts={}, globbed=False, default='relpath'):
564 if pats == ("",):
564 if pats == ("",):
565 pats = []
565 pats = []
566 if not globbed and default == 'relpath':
566 if not globbed and default == 'relpath':
567 pats = expandpats(pats or [])
567 pats = expandpats(pats or [])
568
568
569 m = ctx.match(pats, opts.get('include'), opts.get('exclude'),
569 m = ctx.match(pats, opts.get('include'), opts.get('exclude'),
570 default)
570 default)
571 def badfn(f, msg):
571 def badfn(f, msg):
572 ctx._repo.ui.warn("%s: %s\n" % (m.rel(f), msg))
572 ctx._repo.ui.warn("%s: %s\n" % (m.rel(f), msg))
573 m.bad = badfn
573 m.bad = badfn
574 return m
574 return m
575
575
576 def matchall(repo):
576 def matchall(repo):
577 return matchmod.always(repo.root, repo.getcwd())
577 return matchmod.always(repo.root, repo.getcwd())
578
578
579 def matchfiles(repo, files):
579 def matchfiles(repo, files):
580 return matchmod.exact(repo.root, repo.getcwd(), files)
580 return matchmod.exact(repo.root, repo.getcwd(), files)
581
581
582 def addremove(repo, pats=[], opts={}, dry_run=None, similarity=None):
582 def addremove(repo, pats=[], opts={}, dry_run=None, similarity=None):
583 if dry_run is None:
583 if dry_run is None:
584 dry_run = opts.get('dry_run')
584 dry_run = opts.get('dry_run')
585 if similarity is None:
585 if similarity is None:
586 similarity = float(opts.get('similarity') or 0)
586 similarity = float(opts.get('similarity') or 0)
587 # we'd use status here, except handling of symlinks and ignore is tricky
587 # we'd use status here, except handling of symlinks and ignore is tricky
588 added, unknown, deleted, removed = [], [], [], []
588 added, unknown, deleted, removed = [], [], [], []
589 audit_path = pathauditor(repo.root)
589 audit_path = pathauditor(repo.root)
590 m = match(repo[None], pats, opts)
590 m = match(repo[None], pats, opts)
591 for abs in repo.walk(m):
591 for abs in repo.walk(m):
592 target = repo.wjoin(abs)
592 target = repo.wjoin(abs)
593 good = True
593 good = True
594 try:
594 try:
595 audit_path(abs)
595 audit_path(abs)
596 except (OSError, util.Abort):
596 except (OSError, util.Abort):
597 good = False
597 good = False
598 rel = m.rel(abs)
598 rel = m.rel(abs)
599 exact = m.exact(abs)
599 exact = m.exact(abs)
600 if good and abs not in repo.dirstate:
600 if good and abs not in repo.dirstate:
601 unknown.append(abs)
601 unknown.append(abs)
602 if repo.ui.verbose or not exact:
602 if repo.ui.verbose or not exact:
603 repo.ui.status(_('adding %s\n') % ((pats and rel) or abs))
603 repo.ui.status(_('adding %s\n') % ((pats and rel) or abs))
604 elif repo.dirstate[abs] != 'r' and (not good or not os.path.lexists(target)
604 elif repo.dirstate[abs] != 'r' and (not good or not os.path.lexists(target)
605 or (os.path.isdir(target) and not os.path.islink(target))):
605 or (os.path.isdir(target) and not os.path.islink(target))):
606 deleted.append(abs)
606 deleted.append(abs)
607 if repo.ui.verbose or not exact:
607 if repo.ui.verbose or not exact:
608 repo.ui.status(_('removing %s\n') % ((pats and rel) or abs))
608 repo.ui.status(_('removing %s\n') % ((pats and rel) or abs))
609 # for finding renames
609 # for finding renames
610 elif repo.dirstate[abs] == 'r':
610 elif repo.dirstate[abs] == 'r':
611 removed.append(abs)
611 removed.append(abs)
612 elif repo.dirstate[abs] == 'a':
612 elif repo.dirstate[abs] == 'a':
613 added.append(abs)
613 added.append(abs)
614 copies = {}
614 copies = {}
615 if similarity > 0:
615 if similarity > 0:
616 for old, new, score in similar.findrenames(repo,
616 for old, new, score in similar.findrenames(repo,
617 added + unknown, removed + deleted, similarity):
617 added + unknown, removed + deleted, similarity):
618 if repo.ui.verbose or not m.exact(old) or not m.exact(new):
618 if repo.ui.verbose or not m.exact(old) or not m.exact(new):
619 repo.ui.status(_('recording removal of %s as rename to %s '
619 repo.ui.status(_('recording removal of %s as rename to %s '
620 '(%d%% similar)\n') %
620 '(%d%% similar)\n') %
621 (m.rel(old), m.rel(new), score * 100))
621 (m.rel(old), m.rel(new), score * 100))
622 copies[new] = old
622 copies[new] = old
623
623
624 if not dry_run:
624 if not dry_run:
625 wctx = repo[None]
625 wctx = repo[None]
626 wlock = repo.wlock()
626 wlock = repo.wlock()
627 try:
627 try:
628 wctx.forget(deleted)
628 wctx.forget(deleted)
629 wctx.add(unknown)
629 wctx.add(unknown)
630 for new, old in copies.iteritems():
630 for new, old in copies.iteritems():
631 wctx.copy(old, new)
631 wctx.copy(old, new)
632 finally:
632 finally:
633 wlock.release()
633 wlock.release()
634
634
635 def updatedir(ui, repo, patches, similarity=0):
635 def updatedir(ui, repo, patches, similarity=0):
636 '''Update dirstate after patch application according to metadata'''
636 '''Update dirstate after patch application according to metadata'''
637 if not patches:
637 if not patches:
638 return []
638 return []
639 copies = []
639 copies = []
640 removes = set()
640 removes = set()
641 cfiles = patches.keys()
641 cfiles = patches.keys()
642 cwd = repo.getcwd()
642 cwd = repo.getcwd()
643 if cwd:
643 if cwd:
644 cfiles = [util.pathto(repo.root, cwd, f) for f in patches.keys()]
644 cfiles = [util.pathto(repo.root, cwd, f) for f in patches.keys()]
645 for f in patches:
645 for f in patches:
646 gp = patches[f]
646 gp = patches[f]
647 if not gp:
647 if not gp:
648 continue
648 continue
649 if gp.op == 'RENAME':
649 if gp.op == 'RENAME':
650 copies.append((gp.oldpath, gp.path))
650 copies.append((gp.oldpath, gp.path))
651 removes.add(gp.oldpath)
651 removes.add(gp.oldpath)
652 elif gp.op == 'COPY':
652 elif gp.op == 'COPY':
653 copies.append((gp.oldpath, gp.path))
653 copies.append((gp.oldpath, gp.path))
654 elif gp.op == 'DELETE':
654 elif gp.op == 'DELETE':
655 removes.add(gp.path)
655 removes.add(gp.path)
656
656
657 wctx = repo[None]
657 wctx = repo[None]
658 for src, dst in copies:
658 for src, dst in copies:
659 dirstatecopy(ui, repo, wctx, src, dst, cwd=cwd)
659 dirstatecopy(ui, repo, wctx, src, dst, cwd=cwd)
660 if (not similarity) and removes:
660 if (not similarity) and removes:
661 wctx.remove(sorted(removes), True)
661 wctx.remove(sorted(removes), True)
662
662
663 for f in patches:
663 for f in patches:
664 gp = patches[f]
664 gp = patches[f]
665 if gp and gp.mode:
665 if gp and gp.mode:
666 islink, isexec = gp.mode
666 islink, isexec = gp.mode
667 dst = repo.wjoin(gp.path)
667 dst = repo.wjoin(gp.path)
668 # patch won't create empty files
668 # patch won't create empty files
669 if gp.op == 'ADD' and not os.path.lexists(dst):
669 if gp.op == 'ADD' and not os.path.lexists(dst):
670 flags = (isexec and 'x' or '') + (islink and 'l' or '')
670 flags = (isexec and 'x' or '') + (islink and 'l' or '')
671 repo.wwrite(gp.path, '', flags)
671 repo.wwrite(gp.path, '', flags)
672 util.setflags(dst, islink, isexec)
672 util.setflags(dst, islink, isexec)
673 addremove(repo, cfiles, similarity=similarity)
673 addremove(repo, cfiles, similarity=similarity)
674 files = patches.keys()
674 files = patches.keys()
675 files.extend([r for r in removes if r not in files])
675 files.extend([r for r in removes if r not in files])
676 return sorted(files)
676 return sorted(files)
677
677
678 def dirstatecopy(ui, repo, wctx, src, dst, dryrun=False, cwd=None):
678 def dirstatecopy(ui, repo, wctx, src, dst, dryrun=False, cwd=None):
679 """Update the dirstate to reflect the intent of copying src to dst. For
679 """Update the dirstate to reflect the intent of copying src to dst. For
680 different reasons it might not end with dst being marked as copied from src.
680 different reasons it might not end with dst being marked as copied from src.
681 """
681 """
682 origsrc = repo.dirstate.copied(src) or src
682 origsrc = repo.dirstate.copied(src) or src
683 if dst == origsrc: # copying back a copy?
683 if dst == origsrc: # copying back a copy?
684 if repo.dirstate[dst] not in 'mn' and not dryrun:
684 if repo.dirstate[dst] not in 'mn' and not dryrun:
685 repo.dirstate.normallookup(dst)
685 repo.dirstate.normallookup(dst)
686 else:
686 else:
687 if repo.dirstate[origsrc] == 'a' and origsrc == src:
687 if repo.dirstate[origsrc] == 'a' and origsrc == src:
688 if not ui.quiet:
688 if not ui.quiet:
689 ui.warn(_("%s has not been committed yet, so no copy "
689 ui.warn(_("%s has not been committed yet, so no copy "
690 "data will be stored for %s.\n")
690 "data will be stored for %s.\n")
691 % (repo.pathto(origsrc, cwd), repo.pathto(dst, cwd)))
691 % (repo.pathto(origsrc, cwd), repo.pathto(dst, cwd)))
692 if repo.dirstate[dst] in '?r' and not dryrun:
692 if repo.dirstate[dst] in '?r' and not dryrun:
693 wctx.add([dst])
693 wctx.add([dst])
694 elif not dryrun:
694 elif not dryrun:
695 wctx.copy(origsrc, dst)
695 wctx.copy(origsrc, dst)
696
696
697 def readrequires(opener, supported):
697 def readrequires(opener, supported):
698 '''Reads and parses .hg/requires and checks if all entries found
698 '''Reads and parses .hg/requires and checks if all entries found
699 are in the list of supported features.'''
699 are in the list of supported features.'''
700 requirements = set(opener.read("requires").splitlines())
700 requirements = set(opener.read("requires").splitlines())
701 missings = []
701 missings = []
702 for r in requirements:
702 for r in requirements:
703 if r not in supported:
703 if r not in supported:
704 if not r or not r[0].isalnum():
704 if not r or not r[0].isalnum():
705 raise error.RequirementError(_(".hg/requires file is corrupt"))
705 raise error.RequirementError(_(".hg/requires file is corrupt"))
706 missings.append(r)
706 missings.append(r)
707 missings.sort()
707 missings.sort()
708 if missings:
708 if missings:
709 raise error.RequirementError(_("unknown repository format: "
709 raise error.RequirementError(_("unknown repository format: "
710 "requires features '%s' (upgrade Mercurial)") % "', '".join(missings))
710 "requires features '%s' (upgrade Mercurial)") % "', '".join(missings))
711 return requirements
711 return requirements
712
712
713 class filecacheentry(object):
713 class filecacheentry(object):
714 def __init__(self, path):
714 def __init__(self, path):
715 self.path = path
715 self.path = path
716 self.cachestat = filecacheentry.stat(self.path)
716 self.cachestat = filecacheentry.stat(self.path)
717
717
718 if self.cachestat:
718 if self.cachestat:
719 self._cacheable = self.cachestat.cacheable()
719 self._cacheable = self.cachestat.cacheable()
720 else:
720 else:
721 # None means we don't know yet
721 # None means we don't know yet
722 self._cacheable = None
722 self._cacheable = None
723
723
724 def refresh(self):
724 def refresh(self):
725 if self.cacheable():
725 if self.cacheable():
726 self.cachestat = filecacheentry.stat(self.path)
726 self.cachestat = filecacheentry.stat(self.path)
727
727
728 def cacheable(self):
728 def cacheable(self):
729 if self._cacheable is not None:
729 if self._cacheable is not None:
730 return self._cacheable
730 return self._cacheable
731
731
732 # we don't know yet, assume it is for now
732 # we don't know yet, assume it is for now
733 return True
733 return True
734
734
735 def changed(self):
735 def changed(self):
736 # no point in going further if we can't cache it
736 # no point in going further if we can't cache it
737 if not self.cacheable():
737 if not self.cacheable():
738 return True
738 return True
739
739
740 newstat = filecacheentry.stat(self.path)
740 newstat = filecacheentry.stat(self.path)
741
741
742 # we may not know if it's cacheable yet, check again now
742 # we may not know if it's cacheable yet, check again now
743 if newstat and self._cacheable is None:
743 if newstat and self._cacheable is None:
744 self._cacheable = newstat.cacheable()
744 self._cacheable = newstat.cacheable()
745
745
746 # check again
746 # check again
747 if not self._cacheable:
747 if not self._cacheable:
748 return True
748 return True
749
749
750 if self.cachestat != newstat:
750 if self.cachestat != newstat:
751 self.cachestat = newstat
751 self.cachestat = newstat
752 return True
752 return True
753 else:
753 else:
754 return False
754 return False
755
755
756 @staticmethod
756 @staticmethod
757 def stat(path):
757 def stat(path):
758 try:
758 try:
759 return util.cachestat(path)
759 return util.cachestat(path)
760 except OSError, e:
760 except OSError, e:
761 if e.errno != errno.ENOENT:
761 if e.errno != errno.ENOENT:
762 raise
762 raise
763
763
764 class filecache(object):
764 class filecache(object):
765 '''A property like decorator that tracks a file under .hg/ for updates.
765 '''A property like decorator that tracks a file under .hg/ for updates.
766
766
767 Records stat info when called in _filecache.
767 Records stat info when called in _filecache.
768
768
769 On subsequent calls, compares old stat info with new info, and recreates
769 On subsequent calls, compares old stat info with new info, and recreates
770 the object when needed, updating the new stat info in _filecache.
770 the object when needed, updating the new stat info in _filecache.
771
771
772 Mercurial either atomic renames or appends for files under .hg,
772 Mercurial either atomic renames or appends for files under .hg,
773 so to ensure the cache is reliable we need the filesystem to be able
773 so to ensure the cache is reliable we need the filesystem to be able
774 to tell us if a file has been replaced. If it can't, we fallback to
774 to tell us if a file has been replaced. If it can't, we fallback to
775 recreating the object on every call (essentially the same behaviour as
775 recreating the object on every call (essentially the same behaviour as
776 propertycache).'''
776 propertycache).'''
777 def __init__(self, path, instore=False):
777 def __init__(self, path, instore=False):
778 self.path = path
778 self.path = path
779 self.instore = instore
779 self.instore = instore
780
780
781 def __call__(self, func):
781 def __call__(self, func):
782 self.func = func
782 self.func = func
783 self.name = func.__name__
783 self.name = func.__name__
784 return self
784 return self
785
785
786 def __get__(self, obj, type=None):
786 def __get__(self, obj, type=None):
787 entry = obj._filecache.get(self.name)
787 entry = obj._filecache.get(self.name)
788
788
789 if entry:
789 if entry:
790 if entry.changed():
790 if entry.changed():
791 entry.obj = self.func(obj)
791 entry.obj = self.func(obj)
792 else:
792 else:
793 path = self.instore and obj.sjoin(self.path) or obj.join(self.path)
793 path = self.instore and obj.sjoin(self.path) or obj.join(self.path)
794
794
795 # We stat -before- creating the object so our cache doesn't lie if
795 # We stat -before- creating the object so our cache doesn't lie if
796 # a writer modified between the time we read and stat
796 # a writer modified between the time we read and stat
797 entry = filecacheentry(path)
797 entry = filecacheentry(path)
798 entry.obj = self.func(obj)
798 entry.obj = self.func(obj)
799
799
800 obj._filecache[self.name] = entry
800 obj._filecache[self.name] = entry
801
801
802 setattr(obj, self.name, entry.obj)
802 setattr(obj, self.name, entry.obj)
803 return entry.obj
803 return entry.obj
General Comments 0
You need to be logged in to leave comments. Login now