hemppa/modules/room.py

from modules.common.module import BotModule
import nio


class MatrixModule(BotModule):
    def __init__(self, name):
        super().__init__(name)

    def help(self):
        pass

    async def matrix_message(self, bot, room, event):
        args = event.body.split()
        args.pop(0)

        if args[0] == 'servers':
            await MatrixModule.servers_in_room(bot, room)
        elif args[0] == 'joined':
            await MatrixModule.joined_members(bot, room)
        elif args[0] == 'banned':
            await MatrixModule.banned_members(bot, room)

    @staticmethod
    async def servers_in_room(bot, room: nio.MatrixRoom):
        servers_in_room = "Servers in room:\n"
        count = 0
        response = await bot.client.joined_members(room.room_id)

        if isinstance(response, nio.JoinedMembersError):
            raise response
        else:
            for member in response.members:
                assert isinstance(member, nio.RoomMember)
                server = member.user_id.split(':')[1]
                if server not in servers_in_room:
                    count += 1
                    servers_in_room += f" - {server}\n"
            servers_in_room = f"({count}) {servers_in_room}"
            await bot.send_text(room, servers_in_room)

    @staticmethod
    async def joined_members(bot, room: nio.MatrixRoom):
        await bot.send_text(room, f"Member count: {room.member_count}")

    @staticmethod
    async def banned_members(bot, room: nio.MatrixRoom):
        banned_members = "Banned members:\n"
        count = 0
        res = await bot.client.room_get_state(room.room_id)

        if isinstance(res, nio.RoomGetStateError):
            raise res
        else:
            for state in res.events:
                if state["type"] == "m.room.member":
                    content = state["content"]
                    if content["membership"] == "ban":
                        name = state["state_key"]
                        reason = content.get("reason") or "No reason"
                        banned_members += f" - {name}: \"{reason}\"\n"
                        count += 1
            banned_members = f"({count}) {banned_members}"
            await bot.send_text(room, banned_members)