forked from Georg/limnoria-keycloak
Generic user group query function
Signed-off-by: Georg <georg@lysergic.dev>
This commit is contained in:
parent
cdd640ab9e
commit
4f9b155d66
44
plugin.py
44
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
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user