bot.py

import html, json, re, requests, traceback

from flask import Flask, request
from flask_cors import CORS

app = Flask(__name__)

from chatbot import *

with open("../configurations/vybot.txt", "r") as f:
    secret = f.read().strip()

hooks = {}


def handler(room):
    def _inner(activity):
        if room != rid:
            return
        if "e" in activity:
            for x in activity["e"]:
                print(x)
                if (
                    x["user_id"] == 281362
                    and x.get("content") == "!!/swap-rooms"
                ):
                    send("I am now leaving this room. Goodbye!")
                    swap()
                    send("Hello! I am now operating in this room.")

                if x["event_type"] == 1 and x["room_id"] == room:
                    if x["user_id"] == 296403:
                        continue
                    try:
                        r = requests.post(
                            "http://localhost:5666/msg",
                            headers={"Content-Type": "application/json"},
                            json={"secret": secret, "message": x},
                        )
                        if r.status_code == 200 and r.text:
                            hooks[x["message_id"]] = send(r.text)
                    except:
                        traceback.print_exc()

                if x["event_type"] == 2 and x["room_id"] == room:
                    if x["user_id"] == 296403:
                        continue
                    try:
                        r = requests.post(
                            "http://localhost:5666/msg",
                            headers={"Content-Type": "application/json"},
                            json={"secret": secret, "message": x},
                        )
                        if r.status_code == 200 and r.text:
                            if x["message_id"] in hooks:
                                rooms[rid].editMessage(
                                    r.text, hooks[x["message_id"]]
                                )
                            else:
                                hooks[x["message_id"]] = send(r.text)
                        elif x["message_id"] in hooks:
                            rooms[rid].editMessage(
                                "(message response failed or the new message is"
                                " not a command)",
                                hooks[x["message_id"]],
                            )
                    except:
                        traceback.print_exc()

                if x["event_type"] == 3 and x["room_id"] == room:
                    if x["user_id"] == 296403:
                        continue
                    r = requests.post(
                        "http://localhost:5666/join",
                        headers={"Content-Type": "application/json"},
                        json={"secret": secret, "data": x},
                    )
                    if r.status_code == 200 and r.text:
                        send(r.text)

                if x["event_type"] == 10 and x["room_id"] == room:
                    if x["user_id"] == 296403:
                        continue
                    if x["message_id"] in hooks:
                        rooms[rid].deleteMessage(hooks[x["message_id"]])

    return _inner


chatbot = Chatbot()
chatbot.login()

rid = 106764


def swap():
    global rid
    rid = 106765 - rid


rooms = {k: chatbot.joinRoom(k, handler(k)) for k in [1, 106764]}


def send(message):
    return rooms[rid].sendMessage(message)


@app.route("/", methods=["POST"])
def post_message():
    data = request.json
    msg = send(data["message"])
    if data.get("pin"):
        chatbot.sendRequest(
            f"http://chat.stackexchange.com/messages/{msg}/owner-star",
            "post",
            {"fkey": chatbot.fkey},
            headers={
                "Referer": f"http://chat.stackexchange.com/rooms/{rid}",
                "Origin": "http://chat.stackexchange.com",
            },
        )
    return str(msg)


@app.route("/kill")
def kill():
    request.environ.get("werkzeug.server.shutdown")()


@app.route("/edit", methods=["POST"])
def edit_message():
    data = request.json
    rooms[rid].editMessage(data["message"], data["id"])
    return ""


if __name__ == "__main__":
    app.run(host="localhost", port=5888)