100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
from discord.ext.commands import Cog
|
|
|
|
from robocop_ng.helpers.checks import check_if_staff
|
|
|
|
|
|
class RyujinxVerification(Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
# Export reset channel functions
|
|
self.bot.do_reset = self.do_reset
|
|
self.bot.do_resetalgo = self.do_resetalgo
|
|
|
|
@Cog.listener()
|
|
async def on_member_join(self, member):
|
|
await self.bot.wait_until_ready()
|
|
|
|
if member.guild.id not in self.bot.config.guild_whitelist:
|
|
return
|
|
|
|
join_channel = self.bot.get_channel(self.bot.config.welcome_channel)
|
|
|
|
if join_channel is not None:
|
|
await join_channel.send(
|
|
"Hello {0.mention}! Welcome to Ryujinx! Please read the <#411271165429022730>, and then type the verifying command here to gain access to the rest of the channels.\n\nIf you need help with basic common questions, visit the <#585288848704143371> channel after joining.\n\nIf you need help with Animal Crossing visit the <#692104087889641472> channel for common issues and solutions. If you need help that is not Animal Crossing related, please visit the <#410208610455519243> channel after verifying.".format(
|
|
member
|
|
)
|
|
)
|
|
|
|
async def process_message(self, message):
|
|
"""Process the verification process"""
|
|
if message.channel.id == self.bot.config.welcome_channel:
|
|
# Assign common stuff into variables to make stuff less of a mess
|
|
mcl = message.content.lower()
|
|
|
|
# Get the role we will give in case of success
|
|
success_role = message.guild.get_role(self.bot.config.participant_role)
|
|
|
|
if self.bot.config.verification_string == mcl:
|
|
await message.author.add_roles(success_role)
|
|
await message.delete()
|
|
|
|
@Cog.listener()
|
|
async def on_message(self, message):
|
|
if message.author.bot:
|
|
return
|
|
|
|
try:
|
|
await self.process_message(message)
|
|
except discord.errors.Forbidden:
|
|
chan = self.bot.get_channel(message.channel)
|
|
await chan.send("💢 I don't have permission to do this.")
|
|
|
|
@Cog.listener()
|
|
async def on_message_edit(self, before, after):
|
|
if after.author.bot:
|
|
return
|
|
|
|
try:
|
|
await self.process_message(after)
|
|
except discord.errors.Forbidden:
|
|
chan = self.bot.get_channel(after.channel)
|
|
await chan.send("💢 I don't have permission to do this.")
|
|
|
|
@Cog.listener()
|
|
async def on_member_join(self, member):
|
|
await self.bot.wait_until_ready()
|
|
|
|
if member.guild.id not in self.bot.config.guild_whitelist:
|
|
return
|
|
|
|
join_channel = self.bot.get_channel(self.bot.config.welcome_channel)
|
|
|
|
if join_channel is not None:
|
|
await join_channel.send(self.bot.config.join_message.format(member))
|
|
|
|
@commands.check(check_if_staff)
|
|
@commands.command()
|
|
async def reset(self, ctx, limit: int = 100, force: bool = False):
|
|
"""Wipes messages and pastes the welcome message again. Staff only."""
|
|
if ctx.message.channel.id != self.bot.config.welcome_channel and not force:
|
|
await ctx.send(
|
|
f"This command is limited to"
|
|
f" <#{self.bot.config.welcome_channel}>, unless forced."
|
|
)
|
|
return
|
|
await self.do_reset(ctx.channel, ctx.author.mention, limit)
|
|
|
|
async def do_reset(self, channel, author, limit: int = 100):
|
|
await channel.purge(limit=limit)
|
|
|
|
async def do_resetalgo(self, channel, author, limit: int = 100):
|
|
# We only auto clear the channel daily
|
|
await self.do_reset(channel, author)
|
|
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(RyujinxVerification(bot))
|