Commit c9741149 authored by Maxime Buquet's avatar Maxime Buquet

Fix #2280: Allow /affiliation to list all affiliations

Signed-off-by: Maxime Buquet's avatarMaxime “pep” Buquet <pep@bouah.net>
parent 854bb0ce
......@@ -7,6 +7,7 @@ It keeps track of many things such as part/joins, maintains an
user list, and updates private tabs when necessary.
"""
import asyncio
import bisect
import curses
import logging
......@@ -20,6 +21,7 @@ from datetime import datetime
from typing import Dict, Callable, List, Optional, Union, Set
from slixmpp import InvalidJID, JID
from slixmpp.exceptions import IqError, IqTimeout
from poezio.tabs import ChatTab, Tab, SHOW_NAME
from poezio import common
......@@ -1596,24 +1598,55 @@ class MucTab(ChatTab):
nick, role, reason = args[0], args[1].lower(), args[2]
self.change_role(nick, role, reason)
@command_args_parser.quoted(2)
def command_affiliation(self, args):
@command_args_parser.quoted(0, 2)
def command_affiliation(self, args) -> None:
"""
/affiliation <nick or jid> <affiliation>
/affiliation [<nick or jid> [<affiliation>]]
Changes the affiliation of a user
affiliations can be: outcast, none, member, admin, owner
"""
def callback(iq):
if iq['type'] == 'error':
self.core.room_error(iq, self.jid.bare)
room = JID(self.name)
if not room:
self.core.information('affiliation: requires a valid chat address', 'Error')
return
if args is None:
# List affiliations
if not args:
asyncio.ensure_future(self.get_users_affiliations(room))
return None
if len(args) != 2:
return self.core.command.help('affiliation')
nick, affiliation = args[0], args[1].lower()
# Set affiliation
self.change_affiliation(nick, affiliation)
async def get_users_affiliations(self, jid: JID) -> None:
MUC_ADMIN_NS = 'http://jabber.org/protocol/muc#admin'
try:
iqs = await asyncio.gather(
self.core.xmpp['xep_0045'].get_users_by_affiliation(jid, 'owner'),
self.core.xmpp['xep_0045'].get_users_by_affiliation(jid, 'admin'),
self.core.xmpp['xep_0045'].get_users_by_affiliation(jid, 'member'),
self.core.xmpp['xep_0045'].get_users_by_affiliation(jid, 'outcast'),
)
except (IqError, IqTimeout) as exn:
self.core.room_error(exn.iq, jid)
return None
self._text_buffer.add_message('Affiliations:')
for iq in iqs:
query = iq.xml.find('{%s}query' % MUC_ADMIN_NS)
for item in query.findall('{%s}item' % MUC_ADMIN_NS):
self._text_buffer.add_message(
'%s: %s' % (item.get('jid'), item.get('affiliation'))
)
self.refresh()
return None
@command_args_parser.raw
def command_say(self, line, correct=False):
"""
......@@ -1936,7 +1969,7 @@ class MucTab(ChatTab):
'func':
self.command_affiliation,
'usage':
'<nick or jid> <affiliation>',
'[<nick or jid> [<affiliation>]]',
'desc': ('Set the affiliation of a user. Affiliations can be:'
' outcast, none, member, admin, owner.'),
'shortdesc':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment