diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/auth.py | 44 | ||||
| -rw-r--r-- | src/db.py | 32 | ||||
| -rw-r--r-- | src/main.py | 21 |
3 files changed, 95 insertions, 2 deletions
diff --git a/src/auth.py b/src/auth.py new file mode 100644 index 0000000..3f8ff02 --- /dev/null +++ b/src/auth.py @@ -0,0 +1,44 @@ +from fastapi import FastAPI, Request, Query +from fastapi.responses import RedirectResponse +import httpx +import os +from .db import add_user +from dotenv import load_dotenv + +load_dotenv() + +app = FastAPI() + +CLIENT_ID = os.environ.get("SLAKC_CLIENT_ID") +CLIENT_SECRET = os.environ.get("SLAKC_CLIENT_SECRET") +REDIRECT_URI = os.environ.get("SLAKC_REDIRECT_URI") + [email protected]("/slack/oauth/install") +async def install(): + slack_auth_url = ( + "https://slack.com/oauth/v2/authorize" + f"?client_id={CLIENT_ID}" + f"&redirect_uri={REDIRECT_URI}" + "&scope=" + "&user_scope=channels:write,channels:history,channels:read,chat:write,im:history,users:read" + ) + return RedirectResponse(slack_auth_url) + [email protected]("/slack/oauth/redirect") +async def oauth_redirect(request: Request, code: str = Query(...), state: str = Query(None)): + async with httpx.AsyncClient() as client: + response = await client.post("https://slack.com/api/oauth.v2.access", data={"client_id": CLIENT_ID, "client_secret": CLIENT_SECRET, "code": code, "redirect_uri": REDIRECT_URI}) + data = response.json() + + if not data.get("ok"): + return {"error": data.get("error", "Unknown error")} + + user_id = data.get("authed_user", {}).get("id") + access_token = data.get("authed_user", {}).get("access_token") + + if not user_id or not access_token: + return {"error": "Could not get user info from OAuth response"} + + await add_user(user_id, access_token) + + return {"message": "Successfully joined Botnet!", "user_id": user_id} diff --git a/src/db.py b/src/db.py new file mode 100644 index 0000000..6dd5cad --- /dev/null +++ b/src/db.py @@ -0,0 +1,32 @@ +from slack_sdk.web.async_client import AsyncWebClient +import aiosqlite + +DB_PATH = "botnet.db" + +async def init_db(): + async with aiosqlite.connect(DB_PATH) as db: + await db.execute(""" + CREATE TABLE IF NOT EXISTS Users ( + slack_id TEXT PRIMARY KEY, + oauth_token TEXT NOT NULL, + joined_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP + ) + """) + await db.commit() + +async def add_user(slack_id: str, oauth_token: str): + async with aiosqlite.connect(DB_PATH) as db: + await db.execute( + "INSERT OR REPLACE INTO Users (slack_id, oauth_token) VALUES (?, ?)", + (slack_id, oauth_token) + ) + await db.commit() + client = AsyncWebClient(token=oauth_token) + try: + await client.chat_postMessage( + channel="C0BGMV1CAQG", + text="I've joined the botnet!", + as_user=True + ) + except Exception as e: + print(f"Failed to send welcome message: {e}") diff --git a/src/main.py b/src/main.py index 6f5e2d6..8b0e692 100644 --- a/src/main.py +++ b/src/main.py @@ -2,7 +2,11 @@ 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() @@ -19,11 +23,16 @@ async def handle_messages(message, say): 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"): - return + # yay more scuffed arbitrary stuff + text = "botnet help" parts = text.split(" ") if len(parts) == 1: @@ -41,7 +50,7 @@ async def handle_messages(message, say): elif command == "approve": await say("Not implemented yet") else: - await say("Unknown command: {command}") + await say(f"Unknown command: {command}") @app.event("message") async def shut_up_slack_bolt(): @@ -49,6 +58,14 @@ async def shut_up_slack_bolt(): 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() |