Delete modules/reaction_role/reaction_role.py
This commit is contained in:
parent
5b2aa79b9f
commit
c88edf1fdd
@ -1,219 +0,0 @@
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
from modules.common.emoji_accept import is_accept
|
||||
|
||||
CHECKMARK = '✅'
|
||||
|
||||
class ReactionRoleCog(commands.Cog):
|
||||
"""
|
||||
Records agreements and manages Full Access.
|
||||
Nickname flow:
|
||||
• Add accept on nickname message -> mark agreed + pending (idempotent) and open ONE review
|
||||
• Remove accept on nickname message -> clear only if user has no accept reactions left
|
||||
Full Access: granted when Rules ✅ + RoE ✅ + Nickname *claimed* (pending or verified).
|
||||
Revoked when any of the three is missing.
|
||||
"""
|
||||
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
cfg = bot.config['DEFAULT']
|
||||
|
||||
def _i(key):
|
||||
try:
|
||||
v = cfg.get(key)
|
||||
return int(v) if v else 0
|
||||
except Exception:
|
||||
return 0
|
||||
|
||||
self.rules_msg_id = _i('rules_message_id')
|
||||
self.engage_msg_id = _i('engagement_message_id')
|
||||
self.nick_msg_id = _i('nickname_message_id')
|
||||
self.rules_role = _i('rules_role_id')
|
||||
self.engage_role = _i('engagement_role_id')
|
||||
self.full_access_role = _i('full_access_role_id')
|
||||
|
||||
# ---- helpers ----
|
||||
def _has_rules(self, member_id: int) -> bool:
|
||||
return member_id in self.bot.data_manager.get('agreed_rules')
|
||||
|
||||
def _has_engage(self, member_id: int) -> bool:
|
||||
return member_id in self.bot.data_manager.get('agreed_engagement')
|
||||
|
||||
def _has_nick_claim(self, member_id: int) -> bool:
|
||||
"""Claimed = agreed_nickname; pending/verified tracked separately."""
|
||||
return member_id in self.bot.data_manager.get('agreed_nickname')
|
||||
|
||||
async def maybe_apply_full_access(self, member: discord.Member):
|
||||
"""Grant when Rules+RoE+Nickname *claimed*; revoke when any missing."""
|
||||
guild = member.guild
|
||||
role = guild.get_role(self.full_access_role) if self.full_access_role else None
|
||||
if not role:
|
||||
return
|
||||
|
||||
has_all = self._has_rules(member.id) and self._has_engage(member.id) and self._has_nick_claim(member.id)
|
||||
try:
|
||||
if has_all and role not in member.roles:
|
||||
await member.add_roles(role, reason="All agreements completed (nickname may be pending)")
|
||||
elif not has_all and role in member.roles:
|
||||
await member.remove_roles(role, reason="Agreements incomplete or nickname unclaimed")
|
||||
except discord.Forbidden:
|
||||
pass
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Best-effort: refresh user card
|
||||
cards = self.bot.get_cog('UserCardsCog')
|
||||
if cards:
|
||||
try:
|
||||
await cards.refresh_card(member)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
async def _get_member(self, guild: discord.Guild, user_id: int):
|
||||
if not guild:
|
||||
return None
|
||||
m = guild.get_member(user_id)
|
||||
if m is None:
|
||||
try:
|
||||
m = await guild.fetch_member(user_id)
|
||||
except Exception:
|
||||
return None
|
||||
return m
|
||||
|
||||
async def _user_has_any_accept(self, guild: discord.Guild, channel_id: int, message_id: int, user_id: int) -> bool:
|
||||
"""Return True if the user still has at least one 'accept' reaction on the message."""
|
||||
try:
|
||||
ch = guild.get_channel(channel_id)
|
||||
if not ch:
|
||||
return False
|
||||
msg = await ch.fetch_message(message_id)
|
||||
for rxn in msg.reactions:
|
||||
if is_accept(rxn.emoji):
|
||||
async for u in rxn.users(limit=None):
|
||||
if u.id == user_id:
|
||||
return True
|
||||
return False
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
# ---- commands (hybrid = prefix + slash) ----
|
||||
@commands.hybrid_command(name='nick_same', description='Claim that your global display name matches your in-game name (triggers mod review)')
|
||||
async def nick_same(self, ctx: commands.Context):
|
||||
member = ctx.author if isinstance(ctx.author, discord.Member) else None
|
||||
if not member or not ctx.guild:
|
||||
return await ctx.reply("Use this in a server.", ephemeral=True)
|
||||
|
||||
dm = self.bot.data_manager
|
||||
if member.id not in dm.get('agreed_nickname'):
|
||||
dm.add('agreed_nickname', int(member.id))
|
||||
# Mark pending (clear verified if present)
|
||||
dm.remove('nick_verified', lambda x: x == member.id)
|
||||
newly_pending = False
|
||||
if member.id not in dm.get('nick_claim_pending'):
|
||||
dm.add('nick_claim_pending', int(member.id))
|
||||
newly_pending = True
|
||||
|
||||
# Open/refresh a review with NickNudge (only on first transition to pending)
|
||||
if newly_pending:
|
||||
nn = self.bot.get_cog('NickNudgeCog')
|
||||
if nn and hasattr(nn, 'start_nick_review'):
|
||||
try:
|
||||
await nn.start_nick_review(ctx.guild, member, source="nick_same")
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
await self.maybe_apply_full_access(member)
|
||||
await ctx.reply("Thanks — your nickname claim was sent for moderator review.", ephemeral=True)
|
||||
|
||||
# ---- listeners ----
|
||||
@commands.Cog.listener()
|
||||
async def on_raw_reaction_add(self, payload: discord.RawReactionActionEvent):
|
||||
if not payload.guild_id or not is_accept(payload.emoji):
|
||||
return
|
||||
guild = self.bot.get_guild(payload.guild_id)
|
||||
member = await self._get_member(guild, payload.user_id)
|
||||
if not member or member.bot:
|
||||
return
|
||||
|
||||
dm = self.bot.data_manager
|
||||
try:
|
||||
if payload.message_id == self.rules_msg_id:
|
||||
role = guild.get_role(self.rules_role)
|
||||
if role:
|
||||
await member.add_roles(role, reason="Agreed to rules")
|
||||
if member.id not in dm.get('agreed_rules'):
|
||||
dm.add('agreed_rules', int(member.id))
|
||||
|
||||
elif payload.message_id == self.engage_msg_id:
|
||||
role = guild.get_role(self.engage_role)
|
||||
if role:
|
||||
await member.add_roles(role, reason="Agreed to engagement")
|
||||
if member.id not in dm.get('agreed_engagement'):
|
||||
dm.add('agreed_engagement', int(member.id))
|
||||
|
||||
elif payload.message_id == self.nick_msg_id:
|
||||
# Claim nickname via reaction -> mark agreed + pending (idempotent)
|
||||
newly_pending = False
|
||||
if member.id not in dm.get('agreed_nickname'):
|
||||
dm.add('agreed_nickname', int(member.id))
|
||||
dm.remove('nick_verified', lambda x: x == member.id)
|
||||
if member.id not in dm.get('nick_claim_pending'):
|
||||
dm.add('nick_claim_pending', int(member.id))
|
||||
newly_pending = True
|
||||
|
||||
# Only open a review when we just transitioned to pending
|
||||
if newly_pending:
|
||||
nn = self.bot.get_cog('NickNudgeCog')
|
||||
if nn and hasattr(nn, 'start_nick_review'):
|
||||
try:
|
||||
await nn.start_nick_review(guild, member, source="claim")
|
||||
except Exception:
|
||||
pass
|
||||
else:
|
||||
return
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
await self.maybe_apply_full_access(member)
|
||||
|
||||
@commands.Cog.listener()
|
||||
async def on_raw_reaction_remove(self, payload: discord.RawReactionActionEvent):
|
||||
if not payload.guild_id or not is_accept(payload.emoji):
|
||||
return
|
||||
guild = self.bot.get_guild(payload.guild_id)
|
||||
member = await self._get_member(guild, payload.user_id)
|
||||
if not member or member.bot:
|
||||
return
|
||||
|
||||
dm = self.bot.data_manager
|
||||
try:
|
||||
if payload.message_id == self.rules_msg_id:
|
||||
dm.remove('agreed_rules', lambda x: x == member.id)
|
||||
role = guild.get_role(self.rules_role)
|
||||
if role:
|
||||
await member.remove_roles(role, reason="Rules un-ticked")
|
||||
|
||||
elif payload.message_id == self.engage_msg_id:
|
||||
dm.remove('agreed_engagement', lambda x: x == member.id)
|
||||
role = guild.get_role(self.engage_role)
|
||||
if role:
|
||||
await member.remove_roles(role, reason="Engagement un-ticked")
|
||||
|
||||
elif payload.message_id == self.nick_msg_id:
|
||||
# Clear only if the user has NO accept reactions left on the message
|
||||
still_has_accept = await self._user_has_any_accept(
|
||||
guild, payload.channel_id, payload.message_id, member.id
|
||||
)
|
||||
if not still_has_accept:
|
||||
dm.remove('agreed_nickname', lambda x: x == member.id)
|
||||
dm.remove('nick_claim_pending', lambda x: x == member.id)
|
||||
dm.remove('nick_verified', lambda x: x == member.id)
|
||||
else:
|
||||
return
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
await self.maybe_apply_full_access(member)
|
||||
|
||||
async def setup(bot):
|
||||
await bot.add_cog(ReactionRoleCog(bot))
|
||||
Loading…
Reference in New Issue
Block a user