From 3f68e3b5359abed276e7a7faca6bc78a4164dac4 Mon Sep 17 00:00:00 2001
From: Ryan Lerch <rlerch@redhat.com>
Date: Mon, 14 Sep 2020 22:34:44 +1000
Subject: [PATCH] Add infofasjson info plugin

This adds the initial implementation of the infofasjson plugin, with
support for profile information and groups information. Agreemeents info
coming soon.

Signed-off-by: Ryan Lerch <rlerch@redhat.com>
---
 ipsilon/info/infofasjson.py | 89 +++++++++++++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 ipsilon/info/infofasjson.py

diff --git a/ipsilon/info/infofasjson.py b/ipsilon/info/infofasjson.py
new file mode 100644
index 0000000..1acf90f
--- /dev/null
+++ b/ipsilon/info/infofasjson.py
@@ -0,0 +1,89 @@
+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
+
+
+fasjson_mapping = [
+    ['email', 'email'],
+    ['username', 'username'],
+    ['timezone', 'timezone'],
+    ['locale', 'locale'],
+    ['human_name', 'human_name'],
+    ['groups', 'groups'],
+]
+
+
+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
+        except Exception as e:
+            self.error(f'FASJSON error: {e}')
+            self.error(f'FASJSON error: {e}')
+            return
+
+        # assumption that first email is the default
+        user_data['email'] = user_data['emails'][0]
+
+        # add the groups to the user_data
+        user_data['groups'] = [ g['groupname'] for g in user_group_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()