Make OAuth work and send a welcome message!

1 files changed, 44 insertions, 0 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}