using System.Collections.Generic;
using System.ComponentModel;
using MongoDB.Bson;
using MongoDB.Bson.Serialization.Attributes;

namespace Component
{
	public class BuffManager: ISupportInitialize
	{
		public HashSet<Buff> Buffs { get; private set; }

		[BsonIgnore]
		public Dictionary<ObjectId, Buff> BuffGuidDict { get; private set; }

		[BsonIgnore]
		public Dictionary<int, Buff> BuffTypeDict { get; private set; }

		public BuffManager()
		{
			this.Buffs = new HashSet<Buff>();
			this.BuffGuidDict = new Dictionary<ObjectId, Buff>();
			this.BuffTypeDict = new Dictionary<int, Buff>();
		}

		void ISupportInitialize.BeginInit()
		{
		}

		void ISupportInitialize.EndInit()
		{
			foreach (var buff in Buffs)
			{
				this.BuffGuidDict.Add(buff.Id, buff);
			}

			foreach (var buff in Buffs)
			{
				this.BuffTypeDict.Add(buff.Type, buff);
			}
		}

		public bool Add(Buff buff)
		{
			if (this.Buffs.Contains(buff))
			{
				return false;
			}

			if (this.BuffGuidDict.ContainsKey(buff.Id))
			{
				return false;
			}

			if (this.BuffTypeDict.ContainsKey(buff.Type))
			{
				return false;
			}

			this.Buffs.Add(buff);
			this.BuffGuidDict.Add(buff.Id, buff);
			this.BuffTypeDict.Add(buff.Type, buff);

			return true;
		}

		public Buff Get(ObjectId guid)
		{
			if (!this.BuffGuidDict.ContainsKey(guid))
			{
				return null;
			}

			return this.BuffGuidDict[guid];
		}

		public Buff Get(int type)
		{
			if (!this.BuffTypeDict.ContainsKey(type))
			{
				return null;
			}

			return this.BuffTypeDict[type];
		}

		private bool Remove(Buff buff)
		{
			if (buff == null)
			{
				return false;
			}

			this.Buffs.Remove(buff);
			this.BuffGuidDict.Remove(buff.Id);
			this.BuffTypeDict.Remove(buff.Type);

			return true;
		}

		public bool Remove(ObjectId guid)
		{
			var buff = this.Get(guid);
			return this.Remove(buff);
		}

		public bool Remove(int type)
		{
			var buff = this.Get(type);
			return this.Remove(buff);
		}
	}
}