Discord Bot集成指南：通过OAuth2授权将机器人添加到服务器

from urllib.parse import urlencode

# 替换为你的机器人Client ID
CLIENT_ID = "YOUR_BOT_CLIENT_ID" 

# 替换为你在Discord开发者门户中配置的回调URI
# 例如: "http://localhost:8000/bot-added" 或你的生产环境URL
REDIRECT_URI = "YOUR_REDIRECT_URI" 

# 计算所需权限的整数值。
# 建议使用Discord开发者门户的权限计算器来获取准确值。
# 例如：8 (管理员权限), 2048 (发送消息), 1024 (读取消息历史) 等。
# 这里以一个示例权限值 (如发送消息和读取消息历史) 为例
PERMISSIONS = 2048 | 1024 # 发送消息 和 读取消息历史

def generate_authorization_url(client_id: str, permissions: int, redirect_uri: str):
    params = {
        "client_id": client_id,
        "scope": "bot",
        "permissions": permissions,
        "redirect_uri": redirect_uri,
        "response_type": "code",
        # "prompt": "consent" # 可选：每次都强制显示授权页面
    }
    return f"https://discord.com/oauth2/authorize?{urlencode(params)}"

# 生成并打印授权URL
authorization_url = generate_authorization_url(CLIENT_ID, PERMISSIONS, REDIRECT_URI)
print(f"请将此链接提供给服务器管理员进行授权：\n{authorization_url}")

# 示例输出 (YOUR_BOT_CLIENT_ID 和 YOUR_REDIRECT_URI 会被替换)
# 请将此链接提供给服务器管理员进行授权：
# https://discord.com/oauth2/authorize?client_id=YOUR_BOT_CLIENT_ID&scope=bot&permissions=3072&redirect_uri=YOUR_REDIRECT_URI&response_type=code

from fastapi import FastAPI, Request, HTTPException
from urllib.parse import urlencode
import os

app = FastAPI()

# 替换为你的机器人Client ID和Secret
# 建议通过环境变量管理敏感信息
CLIENT_ID = os.getenv("DISCORD_CLIENT_ID", "YOUR_BOT_CLIENT_ID")
CLIENT_SECRET = os.getenv("DISCORD_CLIENT_SECRET", "YOUR_BOT_CLIENT_SECRET")
# 替换为你在Discord开发者门户中配置的回调URI
REDIRECT_URI = os.getenv("DISCORD_REDIRECT_URI", "http://localhost:8000/bot-added") 

# --- 辅助函数：生成授权URL (与上文相同) ---
def generate_authorization_url(permissions: int):
    params = {
        "client_id": CLIENT_ID,
        "scope": "bot",
        "permissions": permissions,
        "redirect_uri": REDIRECT_URI,
        "response_type": "code",
    }
    return f"https://discord.com/oauth2/authorize?{urlencode(params)}"

# --- 示例端点：提供授权链接 ---
@app.get("/invite-bot")
async def invite_bot_page():
    # 示例权限：发送消息 (2048) 和 读取消息历史 (1024)
    # 实际应用中应根据机器人功能设置
    example_permissions = 2048 | 1024 
    auth_url = generate_authorization_url(example_permissions)
    return {"message": "请将此链接提供给服务器管理员进行授权", "authorization_url": auth_url}

# --- 核心：处理OAuth2回调的端点 ---
@app.get("/bot-added")
async def bot_added_callback(request: Request):
    code = request.query_params.get("code")
    guild_id = request.query_params.get("guild_id")
    permissions_granted = request.query_params.get("permissions")
    error = request.query_params.get("error")
    error_description = request.query_params.get("error_description")

    if error:
        print(f"授权失败: {error} - {error_description}")
        raise HTTPException(status_code=400, detail=f"授权失败: {error_description or error}")

    if not code: