using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Data;

using Uber.Collections;
using Uber.Storage;
using Uber.Messages;

namespace Uber.HabboHotel.Users.Badges
{
    class BadgeComponent
    {
        private SafeList<Badge> Badges;
        private uint UserId;

        public int Count
        {
            get
            {
                return Badges.Count;
            }
        }

        public SafeList<Badge> BadgeList
        {
            get
            {
                return Badges;
            }
        }

        public int EquippedCount
        {
            get
            {
                int i = 0;

                foreach (Badge Badge in Badges)
                {
                    if (Badge.Slot <= 0)
                    {
                        continue;
                    }

                    i++;
                }

                return i;
            }
        }

        public BadgeComponent(uint mUserId)
        {
            this.Badges = new SafeList<Badge>();

            this.UserId = mUserId;
        }

        public Badge GetBadge(string Badge)
        {
            foreach (Badge B in Badges)
            {
                if (Badge.ToLower() == B.Code.ToLower())
                {
                    return B;
                }
            }

            return null;
        }

        public Boolean HasBadge(string Badge)
        {
            if (GetBadge(Badge) != null)
            {
                return true;
            }

            return false;
        }

        public void GiveBadge(string Badge, Boolean InDatabase)
        {
            GiveBadge(Badge, 0, InDatabase);
        }

        public void GiveBadge(string Badge, int Slot, Boolean InDatabase)
        {
            if (HasBadge(Badge))
            {
                return;
            }

            if (InDatabase)
            {
                using (DatabaseClient dbClient = UberEnvironment.GetDatabase().GetClient())
                {
                    dbClient.ExecuteQuery("INSERT INTO user_badges (user_id,badge_id,badge_slot) VALUES ('" + UserId + "','" + Badge + "','" + Slot + "')");
                }
            }

            Badges.Add(new Badge(Badge, Slot));
        }

        public void SetBadgeSlot(string Badge, int Slot)
        {
            Badge B = GetBadge(Badge);

            if (B == null)
            {
                return;
            }

            B.Slot = Slot;
        }

        public void ResetSlots()
        {
            foreach (Badge Badge in Badges)
            {
                Badge.Slot = 0;
            }
        }

        public void RemoveBadge(string Badge)
        {
            if (!HasBadge(Badge))
            {
                return;
            }

            using (DatabaseClient dbClient = UberEnvironment.GetDatabase().GetClient())
            {
                dbClient.ExecuteQuery("DELETE FROM user_badges WHERE badge_id = '" + Badge + "' AND user_id = '" + UserId + "' LIMIT 1");
            }

            Badges.Remove(GetBadge(Badge));
        }

        public void LoadBadges()
        {
            this.Badges = null;
            this.Badges = new SafeList<Badge>();

            DataTable Data = null;

            using (DatabaseClient dbClient = UberEnvironment.GetDatabase().GetClient())
            {
                Data = dbClient.ReadDataTable("SELECT badge_id,badge_slot FROM user_badges WHERE user_id = '" + UserId + "'");
            }

            if (Data == null)
            {
                return;
            }

            foreach (DataRow Row in Data.Rows)
            {
                GiveBadge((string)Row["badge_id"], (int)Row["badge_slot"], false);
            }
        }

        public ServerMessage Serialize()
        {
            var EquippedBadges = new List<Badge>();

            ServerMessage Message = new ServerMessage(229);
            Message.AppendInt32(Count);

            foreach (Badge Badge in Badges)
            {
                Message.AppendStringWithBreak(Badge.Code);

                if (Badge.Slot > 0)
                {
                    EquippedBadges.Add(Badge);
                }
            }

            Message.AppendInt32(EquippedBadges.Count);

            foreach (Badge Badge in EquippedBadges)
            {
                Message.AppendInt32(Badge.Slot);
                Message.AppendStringWithBreak(Badge.Code);
            }

            return Message;
        }
    }
}