Generic user group query function

Signed-off-by: Georg <georg@lysergic.dev>

1 files changed, 43 insertions, 1 deletions

diff --git a/plugin.py b/plugin.py
index ec247a7..23a7f82 100644
--- a/plugin.py
+++ b/plugin.py
@@ -128,7 +128,7 @@ class Keycloak(callbacks.Plugin):
     register = wrap(register, ['anything'])
 
     def ircprom(self, irc, msg, args, option):
-        """<status>
+        """<option>
         true/on = enable authentication to your IRC account with an SSO account going by the same username --
         false/off = allow authentication to your IRC account ONLY with internal IRC credentials (NickServ) --
         Warning: Enabling this without having an SSO account with the same username as your IRC nickname is a security risk."""
@@ -218,6 +218,48 @@ class Keycloak(callbacks.Plugin):
 
     ircprom = wrap(ircprom, ['anything'])
 
+    def user(self, irc, msg, args, option):
+        """<option>
+        groups = dumps the groups you are joined to."""
+
+        user = msg.nick
+        server = self.registryValue('backend.server')
+        realm = self.registryValue('backend.realm')
+        tokenurl = self.registryValue('backend.token')
+        usererr = self.registryValue('replies.error')
+        try:
+            tokendl = requests.get(tokenurl)
+            tokendata = tokendl.json()
+            token = tokendata['access_token']
+            url = server + '/auth/admin/realms/' + realm + '/users'
+            userdata = requests.get(url, params = {'username': user}, headers = {'Content-Type': 'application/json', 'Authorization': 'Bearer ' + token})
+            userresp = userdata.json()
+            uid = userresp[0]['id']
+            print(user, uid)
+        except:
+            print("ERROR: Keycloak token could not be installed.")
+            irc.error(usererr)
+        if option == 'groups':
+            try:
+                url = server + '/auth/admin/realms/' + realm + '/users/' + uid + '/groups'
+                response = requests.get(
+                url,
+                headers = {'Content-Type': 'application/json', 'Authorization': 'Bearer ' + token})
+                test = "{}"
+                print(url)
+                usergroups = response.json()
+                if usergroups:
+                    for group in usergroups:
+                        groupname = usergroups[0]['name']
+                        irc.reply(groupname)
+                else:
+                    irc.reply("No groups.")
+            except:
+                print('Operation failed.')
+                irc.reply(usererr)
+
+    user = wrap(user, ['anything'])
+
 Class = Keycloak