decentral1se/xbotlib
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Support @ commands for direct messages

Browse Source
This commit is contained in:
Luke Murphy 2021-01-16 13:05:53 +01:00
parent fd2134f914
commit aa64a1a68e
No known key found for this signature in database
GPG Key ID: 5E2EF5A63E3718CC
1 changed files with 12 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
xbotlib.py
View File

@ -217,11 +217,8 @@ class Bot(ClientXMPP):
if message["type"] in ("chat", "normal"):
_message = SimpleMessage(message)
if _message.body.startswith("/"):
self.meta(_message, to=_message.sender)
if _message.body.startswith("!"):
self.command(_message, to=_message.sender)
if "@" in _message.body:
self.direct_command(_message)
try:
self.direct(_message)
@ -264,7 +261,7 @@ class Bot(ClientXMPP):
if message["mucnick"] != self.config.nick:
_message = SimpleMessage(message)
if _message.body.startswith("/"):
if _message.body.startswith("@"):
self.meta(_message, room=_message.room)
if f"{self.nick}:!" in _message.body:
@ -327,24 +324,17 @@ class Bot(ClientXMPP):
"""Time since the bot came up."""
return naturaldelta(self.start - dt.now())
def command(self, message, **kwargs):
"""Handle "!" style commands with built-in responses."""
command = message.body.split("!")[-1]
if command == "uptime":
self.reply(self.uptime, **kwargs)
elif command == "help":
def direct_command(self, message):
"""Handle commands in direct messages."""
if "@uptime" in message.body:
self.reply(self.uptime, to=message.sender)
elif "@help" in message.body:
try:
self.reply(self.help, **kwargs)
self.reply(cleandoc(self.help), to=message.sender)
except AttributeError:
self.reply("No help found 🤔️", **kwargs)
def meta(self, message, **kwargs):
"""Handle "/" style commands with built-in responses."""
command = message.body.split("/")[-1]
if command == "bots":
self.reply("🖐️", **kwargs)
self.reply("No help found 🤔️", to=message.sender)
else:
self.log.info(f"'{message.body}' not handled")
class EchoBot(Bot):