pygomx/smal/src/demobot/simple/demobot.py

# Copyright (C) 2026 saces@c-base.org
# SPDX-License-Identifier: AGPL-3.0-only
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()