aboutsummaryrefslogtreecommitdiff
path: root/src/main.py
blob: 8b0e6926cf35cd480767481297d5adc2c373b9c9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import os
from slack_bolt.app.async_app import AsyncApp
from slack_bolt.adapter.socket_mode.async_handler import AsyncSocketModeHandler
from dotenv import load_dotenv
from .db import init_db
from .auth import app as fastapi_app
import threading
import asyncio
import uvicorn

load_dotenv()

selfbot_token = os.environ.get("SLACK_SELFBOT_OAUTH_XOXP")
app_token = os.environ.get("SLACK_APP_TOKEN")
selfbot_user_id = os.environ.get("SELFBOT_USER_ID")

app = AsyncApp(token=selfbot_token)

@app.message("")
async def handle_messages(message, say):
    if message.get("channel_type") != "im":
        return
    
    if message.get("user") == selfbot_user_id:
        return

    if message.get("user") != "U0AKCBZHHMH":
        await say("Botnet is temporarily disabled for users other than <@U0AKCBZHHMH> for debugging and the safety of all OAuth-authorized users. Please DM <@U0AKCBZHHMH> if this is not resolved within a few days.")
        return
    
    text = message.get("text", "").lower().strip()
    
    if not text.startswith("botnet"):
        # yay more scuffed arbitrary stuff
        text = "botnet help"
    
    parts = text.split(" ")
    if len(parts) == 1:
        # scuffed as hell, just sets it to help arbitrarily so it gets caught later LMAO
        # too bad, if it works dont touch it
        parts = ['botnet', 'help']
    
    command = parts[1]
    args = parts[2:]
    
    if command == "help":
        await say("*Botnet Commands*:\n---\n*botnet help*: shows this message\n*botnet deploy <message>*: sends a message into the deploy queue for review/voting\n*botnet approve <channel>*: approve Botnet to function in a channel (must be CM)", mrkdwn=True)
    elif command == "deploy":
        await say("Not implemented yet")
    elif command == "approve":
        await say("Not implemented yet")
    else:
        await say(f"Unknown command: {command}")

@app.event("message")
async def shut_up_slack_bolt():
    # shut up that annoying debug warning that this event is unhandled
    pass

async def main():
    await init_db()

    def run_uvicorn():
        uvicorn.run(fastapi_app, host="0.0.0.0", port=8000)

    thread = threading.Thread(target=run_uvicorn, daemon=True)
    thread.start()

    handler = AsyncSocketModeHandler(app, app_token=app_token)
    await handler.start_async()

if __name__ == "__main__":
    # handler = SocketModeHandler(app, app_token=app_token)
    # handler.start()
    asyncio.run(main())