2020-04-01 07:53:40 +03:00
|
|
|
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)
|
|
|
|
|
2020-04-01 07:54:03 +03:00
|
|
|
if args[0] == 'servers':
|
|
|
|
await MatrixModule.servers_in_room(bot, room)
|
2020-04-01 07:54:47 +03:00
|
|
|
elif args[0] == 'joined':
|
|
|
|
await MatrixModule.joined_members(bot, room)
|
2020-04-01 07:56:48 +03:00
|
|
|
elif args[0] == 'banned':
|
|
|
|
await MatrixModule.banned_members(bot, room)
|
2020-04-01 07:58:04 +03:00
|
|
|
elif args[0] == 'kicked':
|
|
|
|
await MatrixModule.kicked_members(bot, room)
|
2020-04-01 07:54:03 +03:00
|
|
|
|
|
|
|
@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}"
|
2020-04-01 07:54:47 +03:00
|
|
|
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}")
|
2020-04-01 07:56:48 +03:00
|
|
|
|
|
|
|
@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)
|
2020-04-01 07:58:04 +03:00
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
async def kicked_members(bot, room: nio.MatrixRoom):
|
|
|
|
kicked_members = "Kicked 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.get("reason") is not None:
|
|
|
|
name = state["state_key"]
|
|
|
|
reason = content.get("reason")
|
|
|
|
kicked_members += f" - {name}: \"{reason}\"\n"
|
|
|
|
count += 1
|
|
|
|
kicked_members = f"({count}) {kicked_members}"
|
|
|
|
await bot.send_text(room, kicked_members)
|