91 lines
2.4 KiB
Python
91 lines
2.4 KiB
Python
|
import logging
|
||
|
|
from smal.simple.bot import SMALBot
|
||
|
|
|
||
|
|
# setup logging, we want timestamps
|
||
|
|
logging.basicConfig(
|
||
|
|
level=logging.INFO,
|
||
|
|
format="%(asctime)s.%(msecs)03d %(levelname)s %(name)s - %(funcName)s: %(message)s",
|
||
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
||
|
|
)
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
logger.setLevel(level=logging.INFO)
|
||
|
|
|
||
|
|
|
||
|
|
DEFAULT_PREFIX = "!"
|
||
|
|
|
||
|
|
|
||
|
|
class SimpleDemoBot(SMALBot):
|
||
|
|
|
||
|
|
def on_sys(self, ntf):
|
||
|
|
print("Got a system notification: ", ntf)
|
||
|
|
|
||
|
|
def on_event(self, evt):
|
||
|
|
print("Got an event: ", evt)
|
||
|
|
|
||
|
|
def on_message(self, msg):
|
||
|
|
|
||
|
|
if msg["type"] != "m.room.message":
|
||
|
|
# not a room message
|
||
|
|
logger.error(f"not a room message: {msg}")
|
||
|
|
return
|
||
|
|
|
||
|
|
if msg["sender"] == self.UserID:
|
||
|
|
# ignore own messages
|
||
|
|
logger.info(f"ignore own message: {msg}")
|
||
|
|
return
|
||
|
|
|
||
|
|
if "msgtype" in msg["content"].keys() and msg["content"]["msgtype"] != "m.text":
|
||
|
|
# only react to messages, not emotes
|
||
|
|
logger.debug(f"ignore unknown message type: {msg}")
|
||
|
|
return
|
||
|
|
|
||
|
|
if msg["content"]["body"] == "!stop":
|
||
|
|
logger.info("stopping the bot")
|
||
|
|
self.stop()
|
||
|
|
return
|
||
|
|
|
||
|
|
if msg["content"]["body"] == "!leave":
|
||
|
|
logger.info(f"leaving room {msg['roomid']}")
|
||
|
|
self.leaveroom(msg["roomid"])
|
||
|
|
return
|
||
|
|
|
||
|
|
if msg["content"]["body"].startswith("!echo"):
|
||
|
|
|
||
|
|
txt = msg["content"]["body"][5:].strip()
|
||
|
|
|
||
|
|
if txt == "":
|
||
|
|
txt = "Empty text? Are you kidding me?"
|
||
|
|
|
||
|
|
if msg["is_direct"]:
|
||
|
|
self.sendmessage(msg["roomid"], txt)
|
||
|
|
else:
|
||
|
|
self.sendmessagereply(msg["roomid"], msg["id"], msg["sender"], txt)
|
||
|
|
return
|
||
|
|
|
||
|
|
logger.info(f"ignored a message: {msg}")
|
||
|
|
|
||
|
|
def listjoinedrooms(self):
|
||
|
|
roomlist = self.joinedrooms()
|
||
|
|
for room in roomlist:
|
||
|
|
if room["is_direct"]:
|
||
|
|
txt = "Hey, I'm back for secret talk :)"
|
||
|
|
else:
|
||
|
|
txt = "I'm back online."
|
||
|
|
self.sendnotice(room["roomid"], txt)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
# create and initialize the bot
|
||
|
|
bot = SimpleDemoBot(DEFAULT_PREFIX)
|
||
|
|
|
||
|
|
# the bot's matrix client is ready to use now
|
||
|
|
# request the list of joined rooms
|
||
|
|
bot.listjoinedrooms()
|
||
|
|
|
||
|
|
# start syncing forever (listen for incommmig messages/events)
|
||
|
|
bot.run()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|