from discord import Embed

from db import Base
from sqlalchemy import Column, Integer, Text, Boolean, BigInteger


class Greetings(Base):
    __tablename__ = "greetings"
    id = Column(Integer, primary_key=True)
    join_message = Column(Text, nullable=False, default="")
    join_enable = Column(Boolean, nullable=False, default=False)
    leave_message = Column(Text, nullable=False, default="")
    leave_enable = Column(Boolean, nullable=False, default=False)
    guild = Column(BigInteger, nullable=False, unique=True)

    def __init__(self, guild: int):
        self.guild = guild

    def join_embed(self, guild_name: str,  user: str):
        embed = Embed()
        embed.add_field(name=f"Welcome to {guild_name} !", value=self.join_message.format(user))
        return embed

    def leave_msg(self, user: str):
        return self.leave_message.format(user)