102 lines
3.1 KiB
Python
102 lines
3.1 KiB
Python
from ipsilon.info.common import InfoProviderBase, InfoProviderInstaller
|
|
from ipsilon.util.plugin import PluginObject
|
|
from ipsilon.util.policy import Policy
|
|
from ipsilon.util import config as pconfig
|
|
|
|
import fasjson_client
|
|
|
|
|
|
# For OpenID Connect: https://openid.net/specs/openid-connect-basic-1_0.html#StandardClaims
|
|
# For OpenID: https://openid.net/specs/openid-simple-registration-extension-1_0.html#response_format
|
|
fasjson_mapping = [
|
|
['email', 'email'],
|
|
['username', 'nickname'],
|
|
['username', 'preferred_username'],
|
|
['timezone', 'zoneinfo'],
|
|
['locale', 'locale'],
|
|
['human_name', 'name'],
|
|
['givenname', 'given_name'],
|
|
['surname', 'family_name'],
|
|
['ircnicks', 'ircnicks'],
|
|
['gpgkeyids', 'gpg_keyids'],
|
|
['groups', 'groups'],
|
|
['agreements', 'agreements'],
|
|
# Plain OpenID (sreg)
|
|
['timezone', 'timezone'],
|
|
['human_name', 'fullname'],
|
|
]
|
|
|
|
|
|
class InfoProvider(InfoProviderBase):
|
|
|
|
def __init__(self, *args):
|
|
super(InfoProvider, self).__init__(*args)
|
|
self.mapper = Policy(fasjson_mapping)
|
|
|
|
self.name = 'fasjson'
|
|
self.description = """
|
|
Info plugin that retrieves user data from FASJSON. """
|
|
|
|
self.new_config(
|
|
self.name,
|
|
pconfig.String(
|
|
'FASJSON url',
|
|
'The FASJSON Url.',
|
|
'http://fasjson.tinystage.test/fasjson/'),
|
|
)
|
|
|
|
@property
|
|
def fasjson_url(self):
|
|
return self.get_config_value('FASJSON url')
|
|
|
|
|
|
def get_user_attrs(self, user):
|
|
try:
|
|
client = fasjson_client.Client(url=self.fasjson_url)
|
|
user_data = client.get_user(username=user).result
|
|
user_group_data = client.list_user_groups(username=user).result
|
|
user_agreements_data = client.list_user_agreements(username=user).result
|
|
except Exception as e:
|
|
self.error(f'FASJSON error: {e}')
|
|
return {}
|
|
|
|
# assumption that first email is the default
|
|
user_data['email'] = user_data['emails'][0] if user_data['emails'] else None
|
|
|
|
# add the groups to the user_data
|
|
user_data['groups'] = [ g['groupname'] for g in user_group_data]
|
|
|
|
# add the agreements to the user_data
|
|
user_data['agreements'] = [ g['name'] for g in user_agreements_data]
|
|
|
|
userattrs, extras = self.mapper.map_attributes(user_data)
|
|
self.debug(f'user_data: {user_data}')
|
|
self.debug(f'Userattrs: {userattrs}')
|
|
|
|
return userattrs
|
|
|
|
|
|
class Installer(InfoProviderInstaller):
|
|
|
|
def __init__(self, *pargs):
|
|
super(Installer, self).__init__()
|
|
self.name = 'fasjson'
|
|
self.pargs = pargs
|
|
|
|
def install_args(self, group):
|
|
group.add_argument('--info-fasjson', choices=['yes', 'no'], default='no',
|
|
help='Configure FAS info')
|
|
|
|
def configure(self, opts, changes):
|
|
if opts['info_fasjson'] != 'yes':
|
|
return
|
|
|
|
# Add configuration data to database
|
|
po = PluginObject(*self.pargs)
|
|
po.name = 'fasjson'
|
|
po.wipe_data()
|
|
po.wipe_config_values()
|
|
|
|
# Update global config to add login plugin
|
|
po.is_enabled = True
|
|
po.save_enabled_state()
|