mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 23:38:50 +03:00
371257fd73
readconfig now reads only single files readconfig takes an optional list of sections readconfig trusts files we're looking for sections in
410 lines
15 KiB
Python
410 lines
15 KiB
Python
# ui.py - user interface bits for mercurial
|
|
#
|
|
# Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
|
|
#
|
|
# This software may be used and distributed according to the terms
|
|
# of the GNU General Public License, incorporated herein by reference.
|
|
|
|
from i18n import _
|
|
import errno, getpass, os, re, socket, sys, tempfile
|
|
import ConfigParser, traceback, util
|
|
|
|
def dupconfig(orig):
|
|
new = util.configparser(orig.defaults())
|
|
updateconfig(orig, new)
|
|
return new
|
|
|
|
def updateconfig(source, dest, sections=None):
|
|
if not sections:
|
|
sections = source.sections()
|
|
for section in sections:
|
|
if not dest.has_section(section):
|
|
dest.add_section(section)
|
|
if not source.has_section(section):
|
|
continue
|
|
for name, value in source.items(section, raw=True):
|
|
dest.set(section, name, value)
|
|
|
|
class ui(object):
|
|
def __init__(self, parentui=None):
|
|
self.buffers = []
|
|
self.quiet = self.verbose = self.debugflag = self.traceback = False
|
|
self.interactive = self.report_untrusted = True
|
|
|
|
if parentui is None:
|
|
# this is the parent of all ui children
|
|
self.parentui = None
|
|
self.trusted_users = {}
|
|
self.trusted_groups = {}
|
|
self.overlay = util.configparser()
|
|
self.cdata = util.configparser()
|
|
self.ucdata = util.configparser()
|
|
|
|
# we always trust global config files
|
|
for f in util.rcpath():
|
|
self.readconfig(f, assumetrusted=True)
|
|
else:
|
|
# parentui may point to an ui object which is already a child
|
|
self.parentui = parentui.parentui or parentui
|
|
self.buffers = parentui.buffers
|
|
self.trusted_users = parentui.trusted_users.copy()
|
|
self.trusted_groups = parentui.trusted_groups.copy()
|
|
self.cdata = dupconfig(self.parentui.cdata)
|
|
self.ucdata = dupconfig(self.parentui.ucdata)
|
|
|
|
# we want the overlay from the parent, not the root
|
|
self.overlay = dupconfig(parentui.overlay)
|
|
self.fixconfig()
|
|
|
|
def __getattr__(self, key):
|
|
return getattr(self.parentui, key)
|
|
|
|
_isatty = None
|
|
def isatty(self):
|
|
if ui._isatty is None:
|
|
try:
|
|
ui._isatty = sys.stdin.isatty()
|
|
except AttributeError: # not a real file object
|
|
ui._isatty = False
|
|
return ui._isatty
|
|
|
|
def _is_trusted(self, fp, f):
|
|
st = util.fstat(fp)
|
|
if util.isowner(fp, st):
|
|
return True
|
|
|
|
tusers = self.trusted_users
|
|
tgroups = self.trusted_groups
|
|
if '*' in tusers or '*' in tgroups:
|
|
return True
|
|
|
|
user = util.username(st.st_uid)
|
|
group = util.groupname(st.st_gid)
|
|
if user in tusers or group in tgroups or user == util.username():
|
|
return True
|
|
|
|
if self.report_untrusted:
|
|
self.warn(_('Not trusting file %s from untrusted '
|
|
'user %s, group %s\n') % (f, user, group))
|
|
return False
|
|
|
|
def readconfig(self, filename, root=None, assumetrusted=False,
|
|
sections = None):
|
|
try:
|
|
fp = open(filename)
|
|
except IOError:
|
|
if not sections: # ignore unless we were looking for something
|
|
return
|
|
raise
|
|
|
|
cdata = util.configparser()
|
|
trusted = sections or assumetrusted or self._is_trusted(fp, filename)
|
|
|
|
try:
|
|
cdata.readfp(fp, filename)
|
|
except ConfigParser.ParsingError, inst:
|
|
msg = _("Failed to parse %s\n%s") % (filename, inst)
|
|
if trusted:
|
|
raise util.Abort(msg)
|
|
self.warn(_("Ignored: %s\n") % msg)
|
|
|
|
if trusted:
|
|
updateconfig(cdata, self.cdata, sections)
|
|
updateconfig(self.overlay, self.cdata, sections)
|
|
updateconfig(cdata, self.ucdata, sections)
|
|
updateconfig(self.overlay, self.ucdata, sections)
|
|
|
|
if root is None:
|
|
root = os.path.expanduser('~')
|
|
self.fixconfig(root=root)
|
|
|
|
def fixconfig(self, section=None, name=None, value=None, root=None):
|
|
# translate paths relative to root (or home) into absolute paths
|
|
if section is None or section == 'paths':
|
|
if root is None:
|
|
root = os.getcwd()
|
|
items = section and [(name, value)] or []
|
|
for cdata in self.cdata, self.ucdata, self.overlay:
|
|
if not items and cdata.has_section('paths'):
|
|
pathsitems = cdata.items('paths')
|
|
else:
|
|
pathsitems = items
|
|
for n, path in pathsitems:
|
|
if path and "://" not in path and not os.path.isabs(path):
|
|
cdata.set("paths", n,
|
|
os.path.normpath(os.path.join(root, path)))
|
|
|
|
# update ui options
|
|
if section is None or section == 'ui':
|
|
self.debugflag = self.configbool('ui', 'debug')
|
|
self.verbose = self.debugflag or self.configbool('ui', 'verbose')
|
|
self.quiet = not self.debugflag and self.configbool('ui', 'quiet')
|
|
if self.verbose and self.quiet:
|
|
self.quiet = self.verbose = False
|
|
|
|
self.report_untrusted = self.configbool("ui", "report_untrusted",
|
|
True)
|
|
self.interactive = self.configbool("ui", "interactive",
|
|
self.isatty())
|
|
self.traceback = self.configbool('ui', 'traceback', False)
|
|
|
|
# update trust information
|
|
if section is None or section == 'trusted':
|
|
for user in self.configlist('trusted', 'users'):
|
|
self.trusted_users[user] = 1
|
|
for group in self.configlist('trusted', 'groups'):
|
|
self.trusted_groups[group] = 1
|
|
|
|
def setconfig(self, section, name, value):
|
|
for cdata in (self.overlay, self.cdata, self.ucdata):
|
|
if not cdata.has_section(section):
|
|
cdata.add_section(section)
|
|
cdata.set(section, name, value)
|
|
self.fixconfig(section, name, value)
|
|
|
|
def _get_cdata(self, untrusted):
|
|
if untrusted:
|
|
return self.ucdata
|
|
return self.cdata
|
|
|
|
def _config(self, section, name, default, funcname, untrusted, abort):
|
|
cdata = self._get_cdata(untrusted)
|
|
if cdata.has_option(section, name):
|
|
try:
|
|
func = getattr(cdata, funcname)
|
|
return func(section, name)
|
|
except (ConfigParser.InterpolationError, ValueError), inst:
|
|
msg = _("Error in configuration section [%s] "
|
|
"parameter '%s':\n%s") % (section, name, inst)
|
|
if abort:
|
|
raise util.Abort(msg)
|
|
self.warn(_("Ignored: %s\n") % msg)
|
|
return default
|
|
|
|
def _configcommon(self, section, name, default, funcname, untrusted):
|
|
value = self._config(section, name, default, funcname,
|
|
untrusted, abort=True)
|
|
if self.debugflag and not untrusted:
|
|
uvalue = self._config(section, name, None, funcname,
|
|
untrusted=True, abort=False)
|
|
if uvalue is not None and uvalue != value:
|
|
self.warn(_("Ignoring untrusted configuration option "
|
|
"%s.%s = %s\n") % (section, name, uvalue))
|
|
return value
|
|
|
|
def config(self, section, name, default=None, untrusted=False):
|
|
return self._configcommon(section, name, default, 'get', untrusted)
|
|
|
|
def configbool(self, section, name, default=False, untrusted=False):
|
|
return self._configcommon(section, name, default, 'getboolean',
|
|
untrusted)
|
|
|
|
def configlist(self, section, name, default=None, untrusted=False):
|
|
"""Return a list of comma/space separated strings"""
|
|
result = self.config(section, name, untrusted=untrusted)
|
|
if result is None:
|
|
result = default or []
|
|
if isinstance(result, basestring):
|
|
result = result.replace(",", " ").split()
|
|
return result
|
|
|
|
def has_section(self, section, untrusted=False):
|
|
'''tell whether section exists in config.'''
|
|
cdata = self._get_cdata(untrusted)
|
|
return cdata.has_section(section)
|
|
|
|
def _configitems(self, section, untrusted, abort):
|
|
items = {}
|
|
cdata = self._get_cdata(untrusted)
|
|
if cdata.has_section(section):
|
|
try:
|
|
items.update(dict(cdata.items(section)))
|
|
except ConfigParser.InterpolationError, inst:
|
|
msg = _("Error in configuration section [%s]:\n"
|
|
"%s") % (section, inst)
|
|
if abort:
|
|
raise util.Abort(msg)
|
|
self.warn(_("Ignored: %s\n") % msg)
|
|
return items
|
|
|
|
def configitems(self, section, untrusted=False):
|
|
items = self._configitems(section, untrusted=untrusted, abort=True)
|
|
if self.debugflag and not untrusted:
|
|
uitems = self._configitems(section, untrusted=True, abort=False)
|
|
for k in util.sort(uitems):
|
|
if uitems[k] != items.get(k):
|
|
self.warn(_("Ignoring untrusted configuration option "
|
|
"%s.%s = %s\n") % (section, k, uitems[k]))
|
|
return util.sort(items.items())
|
|
|
|
def walkconfig(self, untrusted=False):
|
|
cdata = self._get_cdata(untrusted)
|
|
sections = cdata.sections()
|
|
sections.sort()
|
|
for section in sections:
|
|
for name, value in self.configitems(section, untrusted):
|
|
yield section, name, str(value).replace('\n', '\\n')
|
|
|
|
def username(self):
|
|
"""Return default username to be used in commits.
|
|
|
|
Searched in this order: $HGUSER, [ui] section of hgrcs, $EMAIL
|
|
and stop searching if one of these is set.
|
|
If not found and ui.askusername is True, ask the user, else use
|
|
($LOGNAME or $USER or $LNAME or $USERNAME) + "@full.hostname".
|
|
"""
|
|
user = os.environ.get("HGUSER")
|
|
if user is None:
|
|
user = self.config("ui", "username")
|
|
if user is None:
|
|
user = os.environ.get("EMAIL")
|
|
if user is None and self.configbool("ui", "askusername"):
|
|
user = self.prompt(_("enter a commit username:"), default=None)
|
|
if user is None:
|
|
try:
|
|
user = '%s@%s' % (util.getuser(), socket.getfqdn())
|
|
self.warn(_("No username found, using '%s' instead\n") % user)
|
|
except KeyError:
|
|
pass
|
|
if not user:
|
|
raise util.Abort(_("Please specify a username."))
|
|
if "\n" in user:
|
|
raise util.Abort(_("username %s contains a newline\n") % repr(user))
|
|
return user
|
|
|
|
def shortuser(self, user):
|
|
"""Return a short representation of a user name or email address."""
|
|
if not self.verbose: user = util.shortuser(user)
|
|
return user
|
|
|
|
def expandpath(self, loc, default=None):
|
|
"""Return repository location relative to cwd or from [paths]"""
|
|
if "://" in loc or os.path.isdir(os.path.join(loc, '.hg')):
|
|
return loc
|
|
|
|
path = self.config("paths", loc)
|
|
if not path and default is not None:
|
|
path = self.config("paths", default)
|
|
return path or loc
|
|
|
|
def pushbuffer(self):
|
|
self.buffers.append([])
|
|
|
|
def popbuffer(self):
|
|
return "".join(self.buffers.pop())
|
|
|
|
def write(self, *args):
|
|
if self.buffers:
|
|
self.buffers[-1].extend([str(a) for a in args])
|
|
else:
|
|
for a in args:
|
|
sys.stdout.write(str(a))
|
|
|
|
def write_err(self, *args):
|
|
try:
|
|
if not sys.stdout.closed: sys.stdout.flush()
|
|
for a in args:
|
|
sys.stderr.write(str(a))
|
|
# stderr may be buffered under win32 when redirected to files,
|
|
# including stdout.
|
|
if not sys.stderr.closed: sys.stderr.flush()
|
|
except IOError, inst:
|
|
if inst.errno != errno.EPIPE:
|
|
raise
|
|
|
|
def flush(self):
|
|
try: sys.stdout.flush()
|
|
except: pass
|
|
try: sys.stderr.flush()
|
|
except: pass
|
|
|
|
def _readline(self, prompt=''):
|
|
if self.isatty():
|
|
try:
|
|
# magically add command line editing support, where
|
|
# available
|
|
import readline
|
|
# force demandimport to really load the module
|
|
readline.read_history_file
|
|
# windows sometimes raises something other than ImportError
|
|
except Exception:
|
|
pass
|
|
line = raw_input(prompt)
|
|
# When stdin is in binary mode on Windows, it can cause
|
|
# raw_input() to emit an extra trailing carriage return
|
|
if os.linesep == '\r\n' and line and line[-1] == '\r':
|
|
line = line[:-1]
|
|
return line
|
|
|
|
def prompt(self, msg, pat=None, default="y"):
|
|
"""Prompt user with msg, read response, and ensure it matches pat
|
|
|
|
If not interactive -- the default is returned
|
|
"""
|
|
if not self.interactive:
|
|
self.note(msg, ' ', default, "\n")
|
|
return default
|
|
while True:
|
|
try:
|
|
r = self._readline(msg + ' ')
|
|
if not r:
|
|
return default
|
|
if not pat or re.match(pat, r):
|
|
return r
|
|
else:
|
|
self.write(_("unrecognized response\n"))
|
|
except EOFError:
|
|
raise util.Abort(_('response expected'))
|
|
|
|
def getpass(self, prompt=None, default=None):
|
|
if not self.interactive: return default
|
|
try:
|
|
return getpass.getpass(prompt or _('password: '))
|
|
except EOFError:
|
|
raise util.Abort(_('response expected'))
|
|
def status(self, *msg):
|
|
if not self.quiet: self.write(*msg)
|
|
def warn(self, *msg):
|
|
self.write_err(*msg)
|
|
def note(self, *msg):
|
|
if self.verbose: self.write(*msg)
|
|
def debug(self, *msg):
|
|
if self.debugflag: self.write(*msg)
|
|
def edit(self, text, user):
|
|
(fd, name) = tempfile.mkstemp(prefix="hg-editor-", suffix=".txt",
|
|
text=True)
|
|
try:
|
|
f = os.fdopen(fd, "w")
|
|
f.write(text)
|
|
f.close()
|
|
|
|
editor = self.geteditor()
|
|
|
|
util.system("%s \"%s\"" % (editor, name),
|
|
environ={'HGUSER': user},
|
|
onerr=util.Abort, errprefix=_("edit failed"))
|
|
|
|
f = open(name)
|
|
t = f.read()
|
|
f.close()
|
|
t = re.sub("(?m)^HG:.*\n", "", t)
|
|
finally:
|
|
os.unlink(name)
|
|
|
|
return t
|
|
|
|
def print_exc(self):
|
|
'''print exception traceback if traceback printing enabled.
|
|
only to call in exception handler. returns true if traceback
|
|
printed.'''
|
|
if self.traceback:
|
|
traceback.print_exc()
|
|
return self.traceback
|
|
|
|
def geteditor(self):
|
|
'''return editor to use'''
|
|
return (os.environ.get("HGEDITOR") or
|
|
self.config("ui", "editor") or
|
|
os.environ.get("VISUAL") or
|
|
os.environ.get("EDITOR", "vi"))
|