patreonfix.py

from asyncio import sleep
from os import getenv
from typing import Dict, List, Final

from dotenv import load_dotenv
from disnake import (
    Intents,
    Activity,
    Status,
    AuditLogEntry,
    AuditLogAction,
    Role,
    Client,
    Member,
)

load_dotenv()


if getenv("PROXY_URL"):
    from disnake.http import Route
    Route.BASE = getenv("PROXY_URL") 


ROLE_UPDATE: Dict[int, List[Role]] = {}
PATREON_DISCORD_BOT_ID: Final[int] = int(getenv("PATREON_DISCORD_BOT_ID"))
PATREON_ROLE_ID: Final[int] = int(getenv("PATREON_ROLE_ID"))
SLEEP_DURATION: Final[int] = int(getenv("SLEEP_DURATION"))
INTENTS: Final[Intents] = Intents(
    members=True,
    moderation=True,
)

client: Client = Client(
    intents=INTENTS,
    status=Status.dnd,
    activity=Activity(name=f"Patreons bot make mistakes.", type=3),
)


async def wait_and_check(entry: AuditLogEntry) -> None:
    # Wait for all roles to be removed by the Patreon bot.
    await sleep(SLEEP_DURATION)

    targeted_member: Member = (
        entry.guild.get_member(entry.target.id) # Get member obj from cache.
        or await entry.guild.fetch_member(entry.target.id) # Fetch new member obj if not in cache.
    )

    removed_roles: List[Role] = [role for role in ROLE_UPDATE[entry.target.id]]
    removed_role_ids: List[int] = [role.id for role in removed_roles]

    if PATREON_ROLE_ID in removed_role_ids:
        # Patreon is removing roles from someone it should be removing roles from.
        return
    
    for role in removed_roles:
        # Add roles back that patreon mistakenly removed.
        await targeted_member.add_roles(
            role,
            reason="Adding role back that Patreon removed.",
        )
    
    # Remove member removed role information from cache.
    del ROLE_UPDATE[entry.target.id]

@client.event
async def on_connect() -> None:
    print("Bot has connected to the Discord gateway")

@client.event
async def on_audit_log_entry_create(entry: AuditLogEntry) -> None:
    if (
        entry.user.id == PATREON_DISCORD_BOT_ID
        and entry.action == AuditLogAction.member_role_update
        and len(entry.changes.before.roles) == 1 # Role was removed rather than added.
    ):
        removed_role: Role = entry.changes.before.roles[0]
        targeted_user: User = entry.target

        if ROLE_UPDATE.get(targeted_user.id, None) is None:
            # The Patreon bot hasn't removed any roles yet.
            ROLE_UPDATE[targeted_user.id] = [removed_role]
            return await wait_and_check(entry)

        # Store addtional roles that the Patreon bot removes.
        ROLE_UPDATE[targeted_user.id].append(removed_role)


# Run the bot
client.run(getenv("DISCORD_BOT_TOKEN"))