2024-04-26 11:13:58 +00:00
|
|
|
# noinspection PyInterpreter
|
2024-04-05 12:05:36 +00:00
|
|
|
import logging
|
|
|
|
|
|
|
|
from discord.ext.commands import Cog
|
|
|
|
|
2024-04-26 11:30:44 +00:00
|
|
|
import lib.embeds.boost
|
2024-06-20 17:48:49 +00:00
|
|
|
from lib.embeds.greet import Greet
|
2024-06-20 10:34:03 +00:00
|
|
|
from services.config_service import GuildConfig
|
2024-04-05 12:05:36 +00:00
|
|
|
|
2024-06-15 22:45:24 +00:00
|
|
|
_logs = logging.getLogger('Lumi.Core')
|
2024-04-05 12:05:36 +00:00
|
|
|
|
|
|
|
|
|
|
|
class EventHandler(Cog):
|
|
|
|
def __init__(self, client):
|
|
|
|
self.client = client
|
|
|
|
|
|
|
|
@Cog.listener()
|
|
|
|
async def on_member_join(self, member):
|
|
|
|
config = GuildConfig(member.guild.id)
|
|
|
|
|
|
|
|
if not config.welcome_channel_id:
|
|
|
|
return
|
|
|
|
|
|
|
|
embed = Greet.message(member, config.welcome_message)
|
|
|
|
|
|
|
|
try:
|
|
|
|
await member.guild.get_channel(config.welcome_channel_id).send(embed=embed, content=member.mention)
|
|
|
|
except Exception as e:
|
|
|
|
_logs.info(f"[GreetingHandler] Message not sent in '{member.guild.name}'. Channel ID may be invalid. {e}")
|
|
|
|
|
2024-04-26 11:30:44 +00:00
|
|
|
@Cog.listener()
|
|
|
|
async def on_member_update(self, before, after):
|
|
|
|
if before.premium_since is None and after.premium_since is not None:
|
|
|
|
await self.on_nitro_boost(after)
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
async def on_nitro_boost(member):
|
|
|
|
config = GuildConfig(member.guild.id)
|
|
|
|
|
|
|
|
if not config.boost_channel_id:
|
|
|
|
return
|
|
|
|
|
2024-04-26 12:54:08 +00:00
|
|
|
embed = lib.embeds.boost.Boost.message(member, config.boost_message, config.boost_image_url)
|
2024-04-26 11:30:44 +00:00
|
|
|
|
|
|
|
try:
|
|
|
|
await member.guild.get_channel(config.boost_channel_id).send(embed=embed, content=member.mention)
|
|
|
|
except Exception as e:
|
|
|
|
_logs.info(f"[BoostHandler] Message not sent in '{member.guild.name}'. Channel ID may be invalid. {e}")
|
2024-04-26 11:13:58 +00:00
|
|
|
|
2024-04-05 12:05:36 +00:00
|
|
|
@Cog.listener()
|
|
|
|
async def on_command_completion(self, ctx) -> None:
|
|
|
|
log_msg = '[CommandHandler] %s executed .%s | PREFIX' % (ctx.author.name, ctx.command.qualified_name)
|
|
|
|
|
|
|
|
if ctx.guild is not None:
|
|
|
|
_logs.info(f"{log_msg} | guild: {ctx.guild.name} ")
|
|
|
|
else:
|
|
|
|
_logs.info(f"{log_msg} | in DMs")
|
|
|
|
|
|
|
|
@Cog.listener()
|
|
|
|
async def on_application_command_completion(self, ctx) -> None:
|
|
|
|
log_msg = '[CommandHandler] %s executed /%s | SLASH' % (ctx.author.name, ctx.command.qualified_name)
|
|
|
|
|
|
|
|
if ctx.guild is not None:
|
|
|
|
_logs.info(f"{log_msg} | guild: {ctx.guild.name} ")
|
|
|
|
else:
|
|
|
|
_logs.info(f"{log_msg} | in DMs")
|
|
|
|
|
|
|
|
|
|
|
|
def setup(client):
|
|
|
|
client.add_cog(EventHandler(client))
|