65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
import html
|
|
|
|
from modules.common.module import BotModule
|
|
|
|
|
|
class MatrixModule(BotModule):
|
|
"""
|
|
This is a substitute for matrix' (element's?) missing user status feature.
|
|
Save a custom (status) message for users and allows to query them.
|
|
"""
|
|
|
|
def __init__(self, name):
|
|
super().__init__(name)
|
|
self.status = dict()
|
|
|
|
async def matrix_message(self, bot, room, event):
|
|
args = event.body.split()
|
|
args.pop(0)
|
|
if len(args) < 1 or args[0] == "help":
|
|
await bot.send_text(room, self.help())
|
|
elif args[0] == "show":
|
|
if len(args) > 1:
|
|
await self.send_status(bot=bot, room=room, user=args[1])
|
|
else:
|
|
await self.send_status(bot=bot, room=room)
|
|
elif args[0] == "clear":
|
|
self.status.pop(event.sender)
|
|
bot.save_settings()
|
|
await bot.send_text(room, f"Cleared status of {event.sender}")
|
|
else:
|
|
self.status[event.sender] = " ".join(args)
|
|
bot.save_settings()
|
|
await self.send_status(bot=bot, room=room, user=event.sender)
|
|
|
|
async def send_status(self, bot, room, user=None):
|
|
if user:
|
|
if user in self.status:
|
|
await bot.send_text(room, f"Status message of {user}: {self.status[user]}")
|
|
else:
|
|
await bot.send_text(room, f"No status known for {user}")
|
|
else:
|
|
await bot.send_html(room, "<b>All status messages:</b><br/><ul><li>" +
|
|
"</li><li>".join([f"<b>{html.escape(key)}:</b> {html.escape(value)}" for key, value in self.status.items()]) +
|
|
"</li></ul>", f"All status messages:\n{self.status}")
|
|
|
|
def get_settings(self):
|
|
data = super().get_settings()
|
|
data["user_status_list"] = self.status
|
|
return data
|
|
|
|
def set_settings(self, data):
|
|
super().set_settings(data)
|
|
if data.get("user_status_list"):
|
|
self.status = data["user_status_list"]
|
|
|
|
def help(self):
|
|
return """
|
|
Store a status message per user and display them.
|
|
Usage:
|
|
!status clear - clear my status
|
|
!status show [user] - show the status of user. If no user is given, show all status messages
|
|
!status help - show this text
|
|
!status [status] - set your status
|
|
"""
|