rollbot3/src/groupme_bot.py

import atexit
from logging.config import dictConfig
from threading import Thread

from flask import Flask, request
import requests

from rollbot import Rollbot, RollbotMessage, RollbotPlugin

import plugins

import db
from config import BOTS_LOOKUP, get_config, get_secret

GLOBAL_ADMINS = get_secret("auths.global")
GROUP_ADMINS = get_secret("auths.group")

dictConfig({
    "version": 1,
    "formatters": {"default": {
        "format": "[%(asctime)s] %(levelname)s in %(module)s: %(message)s",
    }},
    "handlers": {"wsgi": {
        "class": "logging.StreamHandler",
        "stream": "ext://flask.logging.wsgi_errors_stream",
        "formatter": "default"
    }},
    "root": {
        "level": "INFO",
        "handlers": ["wsgi"]
    }
})


def post_groupme_message(msg, group_id):
    bot_id = BOTS_LOOKUP[group_id]
    requests.post(
        "https://api.groupme.com/v3/bots/post",
        json={
            "bot_id": bot_id,
            "text": msg
        },
        timeout=10
    )


app = Flask(__name__)
app.config["PROPAGATE_EXCEPTIONS"] = True
rollbot = Rollbot(
    logger=app.logger,
    plugin_classes=RollbotPlugin.find_all_plugins(),
    aliases=get_config("aliases"),
    responses=get_config("responses"),
    sleep_time=float(get_config("sleep_time")),
    callback=post_groupme_message,
    session_factory=db.Session
)
app.logger.info("Initializing database tables")
db.init_db()
app.logger.info("Finished initializing database")
rollbot.start_plugins()
atexit.register(rollbot.shutdown_plugins)


@app.route("/", methods=["POST"])
def execute():
    json = request.get_json()
    msg = RollbotMessage.from_groupme(json, global_admins=GLOBAL_ADMINS, group_admins=GROUP_ADMINS)
    if msg.group_id not in BOTS_LOOKUP:
        app.logger.warning(f"Received message from unknown group ID {msg.group_id}")
        return "", 400
    t = Thread(target=lambda: rollbot.handle_command(msg))
    t.start()
    return "", 204


@app.route("/health")
def health():
    return "Rollbot healthy!", 200


if __name__ == "__main__":
    app.run(host="0.0.0.0", port=6070)