using System.Data;
using ChaosBot.Attribute;

namespace ChaosBot.Database.Entity
{
    [DBEntity("RaffleTable")]
    public class Raffle : BaseEntity
    {
        [DBPrimaryKey]
        [DBAutoIncrement]
        [DBNotNull]
        [DBUnique]
        public int id { get; private set; }
        public string userId { get; private set; }
        public string guildId { get; private set; }

        public Raffle() {}

        public Raffle(int id, string userId, string guildId)
        {
            this.id = id;
            this.userId = userId;
            this.guildId = guildId;
        }

        public Raffle(string userId, string guildId)
        {
            this.userId = userId;
            this.guildId = guildId;
        }

        public static QueryBuilder<Raffle> Query()
        {
            return BaseEntity.Query<Raffle>();
        }

        public override void SetFromRow(DataRow row)
        {
            id = (int)row["id"];
            userId = (string)row["userid"];
            guildId = (string) row["guildid"];
        }
    }
}