mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 15:27:13 +03:00
f08e17d3ed
Summary: Using `testedwith = 'internal'` is not a good habit [1]. Having it auto-updated in batch would also introduce a lot of churn. This diff makes them "ships-with-fb-hgext". If we do want to fill the ideal "testedwith" information, we could put it in a centric place, like a "fbtestedwith" extension rewriting those "ships-with-fb-hgext" on the fly. Maybe having in-repo tags for tested Mercurial releases is also a good idea. [1]: www.mercurial-scm.org/repo/hg/rev/2af1014c2534 Test Plan: `arc lint` Reviewers: #sourcecontrol, rmcelroy Reviewed By: rmcelroy Subscribers: rmcelroy, mjpieters Differential Revision: https://phabricator.intern.facebook.com/D4244689 Signature: t1:4244689:1480440027:3dc18d017b48beba1176fbfd120351889259eb4b
73 lines
2.1 KiB
Python
73 lines
2.1 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 os
|
|
import grp
|
|
|
|
testedwith = 'ships-with-fb-hgext'
|
|
|
|
_missinggroup = None
|
|
|
|
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)
|