mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 17:27:53 +03:00
9dc21f8d0b
Summary: D13853115 adds `edenscm/` to `sys.path` and code still uses `import mercurial`. That has nasty problems if both `import mercurial` and `import edenscm.mercurial` are used, because Python would think `mercurial.foo` and `edenscm.mercurial.foo` are different modules so code like `try: ... except mercurial.error.Foo: ...`, or `isinstance(x, mercurial.foo.Bar)` would fail to handle the `edenscm.mercurial` version. There are also some module-level states (ex. `extensions._extensions`) that would cause trouble if they have multiple versions in a single process. Change imports to use the `edenscm` so ideally the `mercurial` is no longer imported at all. Add checks in extensions.py to catch unexpected extensions importing modules from the old (wrong) locations when running tests. Reviewed By: phillco Differential Revision: D13868981 fbshipit-source-id: f4e2513766957fd81d85407994f7521a08e4de48
86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
# grpcheck.py - check if the user is in specified groups
|
|
#
|
|
# Copyright 2015 Facebook, Inc.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
|
|
"""check if the user is in specified groups
|
|
|
|
If the user is not a member of specified groups, optionally show a warning
|
|
message and override some other config items.
|
|
|
|
::
|
|
[grpcheck]
|
|
# the user is expected to be a member of devs and users group
|
|
groups = devs, users
|
|
# warning to show if the user is not a member of any of those groups
|
|
warning = You are not a member of %s group. Consult IT department for help.
|
|
# if the user is not a member of any of those groups, override chgserver
|
|
# config to make chgserver exit earlier
|
|
overrides.chgserver.idletimeout = 2
|
|
"""
|
|
|
|
import grp
|
|
import os
|
|
|
|
from edenscm.mercurial import registrar
|
|
|
|
|
|
testedwith = "ships-with-fb-hgext"
|
|
|
|
_missinggroup = None
|
|
|
|
configtable = {}
|
|
configitem = registrar.configitem(configtable)
|
|
|
|
configitem("grpcheck", "groups", default=[])
|
|
|
|
|
|
def _grpname2gid(name):
|
|
try:
|
|
return grp.getgrnam(name).gr_gid
|
|
except KeyError:
|
|
return None
|
|
|
|
|
|
def _firstmissinggroup(groupnames):
|
|
usergids = set(os.getgroups())
|
|
for name in groupnames:
|
|
expectedgid = _grpname2gid(name)
|
|
# ignore unknown groups
|
|
if expectedgid is not None and expectedgid not in usergids:
|
|
return name
|
|
|
|
|
|
def _overrideconfigs(ui):
|
|
for k, v in ui.configitems("grpcheck"):
|
|
if not k.startswith("overrides."):
|
|
continue
|
|
section, name = k[len("overrides.") :].split(".", 1)
|
|
ui.setconfig(section, name, v)
|
|
|
|
|
|
def extsetup(ui):
|
|
groupnames = ui.configlist("grpcheck", "groups")
|
|
if not groupnames:
|
|
return
|
|
missing = _firstmissinggroup(groupnames)
|
|
if not missing:
|
|
return
|
|
message = ui.config("grpcheck", "warning")
|
|
if message and not ui.plain():
|
|
if "%s" in message:
|
|
message = message % missing
|
|
ui.warn(message + "\n")
|
|
_overrideconfigs(ui)
|
|
# re-used by reposetup. groups information is immutable for a process,
|
|
# so we can re-use the "missing" calculation result safely.
|
|
global _missinggroup
|
|
_missinggroup = missing
|
|
|
|
|
|
def reposetup(ui, repo):
|
|
if _missinggroup:
|
|
_overrideconfigs(ui)
|