rollbot4/repl_driver.py

from datetime import datetime
import asyncio

import rollbot


class MyBot(rollbot.Rollbot[str]):
    def read_config(self, key):
        return key

    def parse(self, raw):
        return rollbot.Message(
            origin_id="REPL",
            channel_id=".",
            sender_id=".",
            timestamp=datetime.now(),
            origin_admin=True,
            channel_admin=True,
            text=raw,
            attachments=[],
        )

    async def respond(self, res):
        print(res, flush=True)


async def goodbye_command(message, context):
    await context.respond(rollbot.Response.from_message(message, "Goodbye!"))


async def count_command(message, context):
    args = message.text.split("count", maxsplit=1)[1].strip().split()
    name = args[0] if len(args) > 0 else "main"
    async with context.database() as db:
        await db.execute("INSERT INTO counter VALUES (?, 1) \
                          ON CONFLICT (name) DO \
                          UPDATE SET count=count + 1", (name,))
        await db.commit()
        async with db.execute("SELECT count FROM counter WHERE name = ?", (name,)) as cursor:
            res = (await cursor.fetchone())[0]
    await context.respond(rollbot.Response.from_message(message, f"{name} = {res}"))


async def make_table(context):
    async with context.database() as db:
        await db.execute("CREATE TABLE IF NOT EXISTS counter ( \
                            name TEXT NOT NULL PRIMARY KEY, \
                            count INTEGER NOT NULL DEFAULT 0 \
                        );")
        await db.commit()


config = rollbot.CommandConfiguration(
    commands={
        "goodbye": goodbye_command,
        "count": count_command,
    },
    call_and_response={
        "hello": "Hello!",
    },
    aliases={
        "hi": "hello",
        "bye": "goodbye",
    },
    bangs=("/",),
    startup=[make_table],
)

bot = MyBot(config, "/tmp/my.db")

async def run():
    await bot.on_startup()
    try:
        while True:
            await bot.on_message(input("> "))
    except EOFError:
        pass
    await bot.on_shutdown()


asyncio.run(run())