server.py

import asyncio
import os
from contextlib import asynccontextmanager

import discord
import uvicorn
from fastapi import FastAPI, Request

bot: discord.Bot = None

token = os.getenv("PINGME_BOT_TOKEN")
address = os.getenv("PINGME_ADDRESS")
port = os.getenv("PINGME_PORT")

if not token:
    raise RuntimeError("Missing necessary environment variable: PINGME_BOT_TOKEN")
if not address:
    raise RuntimeError("Missing necessary environment variable: PINGME_ADDRESS")
if not port:
    raise RuntimeError("Missing necessary environment variable: PINGME_PORT")

@asynccontextmanager
async def lifespan(app: FastAPI):
    global token
    global bot

    bot = discord.Bot()

    @bot.event
    async def on_ready():
        print(f"Logged in as {bot.user.name}.")

    bot_task = asyncio.create_task(bot.start(token))
    try:
        yield
    finally:
        await bot.close()
        try:
            await bot_task
        except asyncio.CancelledError:
            pass


server = FastAPI(lifespan=lifespan)


@server.post("/notify")
async def notify(request: Request) -> dict:
    try:
        global user_id
        payload = await request.json()

        if not payload["content"] or not payload["user_id"]:
            return {"error": "Malformed JSON contents."}
        if not str.isdigit(payload["user_id"]):
            return {"error": "user_id must be an integer."}
        if len(payload["content"]) > 2000:
            return {"error": "content must be less than 2001 characters."}

        user = await bot.fetch_user(int(payload["user_id"]))

        await user.send(payload["content"])
        return {"content": "you got it boss"}
    except Exception as e:
        return {"exception": e}


uvicorn.run(server, host=address, port=int(port))