rollbot3/src/app.py

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

from flask import Flask, request, render_template

import db
from config import BOTS_LOOKUP, get_config, get_secret
from command_system import RollbotMessage, RollbotPlugin
from rollbot import Rollbot
from util import post_groupme_message, find_all_subclasses
import plugins

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"]
    }
})

app = Flask(__name__)
app.config["PROPAGATE_EXCEPTIONS"] = True
rollbot = Rollbot(
    logger=app.logger,
    plugin_classes=find_all_subclasses(RollbotPlugin),
    aliases=get_config("aliases"),
    responses=get_config("responses"),
    sleep_time=float(get_config("sleep_time")),
    callback=lambda msg, group_id: post_groupme_message(msg, BOTS_LOOKUP[group_id]),
    session_factory=db.session_scope
)
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("/teamspeak", methods=["GET"])
def teamspeak():
    response = rollbot.run_command(RollbotMessage("MANUAL", None, None, None, None, "!teamspeak", False))
    if response.is_success:
        response = response.txt
    else:
        response = response.failure_msg
    return render_template("teamspeak.html", r=response, scrolling=get_config("teamspeak.scrolling"))


@app.route("/rollbot", 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 in BOTS_LOOKUP:
        self.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


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