ryuko-ng/robocop_ng/cogs/ryujinx_reactionroles.py

267 lines
9 KiB
Python
Raw Normal View History

2022-09-05 17:30:16 +00:00
import collections
import json
import os
import discord
2022-09-05 17:30:16 +00:00
from discord.ext import commands
from discord.ext.commands import Cog
from robocop_ng.helpers.checks import check_if_staff
2022-11-10 14:09:57 +00:00
class RyujinxReactionRoles(Cog):
def __init__(self, bot):
self.bot = bot
self.channel_id = (
self.bot.config.reaction_roles_channel_id
) # The channel to send the reaction role message. (self-roles channel)
self.file = os.path.join(
self.bot.state_dir, "data/reactionroles.json"
) # the file to store the required reaction role data. (message id of the RR message.)
self.msg_id = None
self.m = None # the msg object
2022-09-05 17:30:16 +00:00
@commands.guild_only()
@commands.check(check_if_staff)
@commands.command()
async def register_reaction_role(self, ctx, target_role_id: int, emoji_name: str):
"""Register a reaction role, staff only."""
await self.bot.wait_until_ready()
2022-11-10 14:09:57 +00:00
if emoji_name[0] == "<":
2022-09-05 17:30:16 +00:00
emoji_name = emoji_name[1:-1]
if target_role_id in self.bot.config.staff_role_ids:
2022-09-05 17:30:16 +00:00
return await ctx.send("Error: Dangerous role found!")
target_role = ctx.guild.get_role(target_role_id)
if target_role is None:
return await ctx.send("Error: Role not found!")
target_role_name = target_role.name
for key in self.reaction_config["reaction_roles_emoji_map"]:
value = self.reaction_config["reaction_roles_emoji_map"][key]
if type(value) is str and target_role_name == value:
return await ctx.send(f"Error: {target_role_name}: already registered.")
self.reaction_config["reaction_roles_emoji_map"][emoji_name] = target_role_name
self.save_reaction_config(self.reaction_config)
await self.reload_reaction_message(False)
await ctx.send(f"{target_role_name}: registered.")
def get_emoji_full_name(self, emoji):
emoji_name = emoji.name
if emoji_name is not None and emoji.id is not None:
emoji_name = f":{emoji_name}:{emoji.id}"
return emoji_name
def get_role(self, key):
return discord.utils.get(
self.bot.guilds[0].roles,
2022-09-05 17:30:16 +00:00
name=self.get_role_from_emoji(key),
)
2022-09-05 17:30:16 +00:00
def get_role_from_emoji(self, key):
value = self.emoji_map.get(key)
if value is not None and type(value) is not str:
return value.get("role")
return value
async def generate_embed(self):
2022-09-05 17:30:16 +00:00
last_descrption = []
2022-11-10 14:09:57 +00:00
description = [
"React to this message with the emojis given below to get your 'Looking for LDN game' roles. \n"
]
2022-09-05 17:30:16 +00:00
for x in self.emoji_map:
value = self.emoji_map[x]
emoji = x
2022-11-10 14:09:57 +00:00
if len(emoji.split(":")) == 3:
2022-09-05 17:30:16 +00:00
emoji = f"<{emoji}>"
2022-09-05 17:30:16 +00:00
if type(value) is str:
2022-11-10 14:09:57 +00:00
description.append(
f"{emoji} for __{self.emoji_map.get(x).split('(')[1].split(')')[0]}__"
)
else:
2022-09-05 17:30:16 +00:00
role_name = value["role"]
line_fmt = value["fmt"]
if value.get("should_be_last", False):
last_descrption.append(line_fmt.format(emoji, role_name))
else:
description.append(line_fmt.format(emoji, role_name))
embed = discord.Embed(
2022-11-10 14:09:57 +00:00
title="**Select your roles**",
description="\n".join(description) + "\n" + "\n".join(last_descrption),
color=420420,
)
embed.set_footer(
text="To remove a role, simply remove the corresponding reaction."
)
return embed
async def handle_offline_reaction_add(self):
await self.bot.wait_until_ready()
for reaction in self.m.reactions:
2023-03-03 21:20:05 +00:00
reactions_users = []
async for user in reaction.users():
reactions_users.append(user)
for user in reactions_users:
2022-09-05 17:30:16 +00:00
emoji_name = str(reaction.emoji)
2022-11-10 14:09:57 +00:00
if emoji_name[0] == "<":
2022-09-05 17:30:16 +00:00
emoji_name = emoji_name[1:-1]
if self.get_role_from_emoji(emoji_name) is not None:
role = self.get_role(emoji_name)
if (
not user in role.members
and not user.bot
and type(user) is discord.Member
):
await user.add_roles(role)
else:
await self.m.clear_reaction(reaction.emoji)
async def handle_offline_reaction_remove(self):
await self.bot.wait_until_ready()
for emoji in self.emoji_map:
for reaction in self.m.reactions:
2022-09-05 17:30:16 +00:00
emoji_name = str(reaction.emoji)
2022-11-10 14:09:57 +00:00
if emoji_name[0] == "<":
2022-09-05 17:30:16 +00:00
emoji_name = emoji_name[1:-1]
2023-03-03 21:20:05 +00:00
reactions_users = []
async for user in reaction.users():
reactions_users.append(user)
2022-09-05 17:30:16 +00:00
role = self.get_role(emoji_name)
for user in role.members:
2023-03-03 21:20:05 +00:00
if user not in reactions_users:
member = self.m.guild.get_member(user.id)
if member is not None:
await member.remove_roles(role)
2022-09-05 17:30:16 +00:00
def load_reaction_config(self):
if not os.path.exists(self.file):
with open(self.file, "w") as f:
2022-09-05 17:30:16 +00:00
json.dump({}, f)
with open(self.file, "r") as f:
2022-09-05 17:30:16 +00:00
return json.load(f)
def save_reaction_config(self, value):
with open(self.file, "w") as f:
json.dump(value, f)
2022-11-10 14:09:57 +00:00
async def reload_reaction_message(self, should_handle_offline=True):
await self.bot.wait_until_ready()
2022-11-10 14:09:57 +00:00
self.emoji_map = collections.OrderedDict(
sorted(
self.reaction_config["reaction_roles_emoji_map"].items(),
key=lambda x: str(x[1]),
)
)
2022-09-05 17:30:16 +00:00
guild = self.bot.guilds[0] # The ryu guild in which the bot is.
channel = guild.get_channel(self.channel_id)
2023-03-03 21:20:05 +00:00
if channel is None:
channel = await guild.fetch_channel(self.channel_id)
history = []
async for msg in channel.history():
history.append(msg)
m = discord.utils.get(history, id=self.reaction_config["id"])
if m is None:
2022-09-05 17:30:16 +00:00
self.reaction_config["id"] = None
embed = await self.generate_embed()
self.m = await channel.send(embed=embed)
self.msg_id = self.m.id
for x in self.emoji_map:
await self.m.add_reaction(x)
2022-09-05 17:30:16 +00:00
self.reaction_config["id"] = self.m.id
self.save_reaction_config(self.reaction_config)
await self.handle_offline_reaction_remove()
else:
2023-03-03 21:20:05 +00:00
self.m = m
self.msg_id = self.m.id
await self.m.edit(embed=await self.generate_embed())
for x in self.emoji_map:
if not x in self.m.reactions:
await self.m.add_reaction(x)
2022-09-05 17:30:16 +00:00
if should_handle_offline:
await self.handle_offline_reaction_add()
await self.handle_offline_reaction_remove()
@Cog.listener()
async def on_ready(self):
2023-03-03 21:20:05 +00:00
await self.bot.wait_until_ready()
2022-09-05 17:30:16 +00:00
self.reaction_config = self.load_reaction_config()
await self.reload_reaction_message()
@Cog.listener()
async def on_raw_reaction_add(self, payload):
await self.bot.wait_until_ready()
if payload.member.bot:
pass
else:
if payload.message_id == self.msg_id:
2022-09-05 17:30:16 +00:00
emoji_name = self.get_emoji_full_name(payload.emoji)
if self.get_role_from_emoji(emoji_name) is not None:
target_role = self.get_role(emoji_name)
if target_role is not None:
2022-11-10 14:09:57 +00:00
await payload.member.add_roles(target_role)
else:
2022-11-10 14:09:57 +00:00
self.bot.log.error(
f"Role {self.emoji_map[emoji_name]} not found."
)
2022-09-05 17:30:16 +00:00
await self.m.clear_reaction(payload.emoji)
else:
2022-09-05 17:30:16 +00:00
await self.m.clear_reaction(payload.emoji)
@Cog.listener()
async def on_raw_reaction_remove(self, payload):
await self.bot.wait_until_ready()
if payload.message_id == self.msg_id:
2022-09-05 17:30:16 +00:00
emoji_name = self.get_emoji_full_name(payload.emoji)
if self.get_role_from_emoji(emoji_name) is not None:
guild = discord.utils.find(
lambda guild: guild.id == payload.guild_id, self.bot.guilds
)
2022-09-05 17:30:16 +00:00
target_role = self.get_role(emoji_name)
if target_role is not None:
await guild.get_member(payload.user_id).remove_roles(
self.get_role(emoji_name)
) # payload.member.remove_roles will throw error
async def setup(bot):
await bot.add_cog(RyujinxReactionRoles(bot))