import disnake
from disnake.ext import commands
import os

from logger import Logger

logger = Logger('Liza Events')


class EventsCog(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.JOIN_MESSAGE = os.getenv('JOIN_MESSAGE')
        self.LEAVE_MESSAGE = os.getenv('LEAVE_MESSAGE')
        self.CHANNEL = os.getenv('WELCOME_LEAVE_CHANNEL_ID')

    @commands.Cog.listener()
    async def on_member_join(self, member: disnake.Member):
        c: disnake.guild = self.bot.get_channel(self.CHANNEL)

        await c.send_message(self.JOIN_MESSAGE.format(member.mention))


    @commands.Cog.listener()
    async def on_member_update(self, before: disnake.Member, after: disnake.Member):
        pass

    @commands.Cog.listener()
    async def on_member_remove(self, member: disnake.Member):
        c: disnake.guild = self.bot.get_channel(self.CHANNEL)

        await c.send_message(self.LEAVE_MESSAGE.format(member.mention))

    @commands.Cog.listener()
    async def on_member_ban(self, guild: disnake.Guild, member: disnake.Member or disnake.User):
        pass


def setup(bot):
    bot.add_cog(EventsCog(bot))