using DominionBase.Cards;
using DominionBase.Enums;
using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;

namespace DominionBase.Piles
{
	public class DeckCollection : IPile<Card>, IEnumerable<Card>
	{
		private readonly List<Deck> _decks = new List<Deck>();

		public DeckCollection(Deck deck)
		{
			Add(deck);
		}

		public DeckCollection(IEnumerable<Deck> collection)
		{
			_decks.AddRange(collection);
		}

		public DeckCollection(params Deck[] collection)
		{
			_decks.AddRange(collection);
		}

		public void Add(Deck deck)
		{
			_decks.Add(deck);
		}

		public ItemCollection<Card> this[Categories type]
		{
			get
			{
				return new ItemCollection<Card>(_decks.SelectMany(d => d[type]));
			}
		}

		public ItemCollection<Card> this[Type type]
		{
			get
			{
				return new ItemCollection<Card>(_decks.SelectMany(d => d[type]));
			}
		}

		public ItemCollection<Card> this[string name]
		{
			get
			{
				return new ItemCollection<Card>(_decks.SelectMany(d => d[name]));
			}
		}

		public ItemCollection<Card> this[Predicate<Card> predicate]
		{
			get
			{
				return new ItemCollection<Card>(_decks.SelectMany(d => d[predicate]));
			}
		}

		public int Count
		{
			get { return _decks.Sum(d => d.Count); }
		}

		public IEnumerator<Card> GetEnumerator()
		{
			foreach (var deck in _decks)
			{
				var enumerator = deck.GetEnumerator();
				while (enumerator.MoveNext())
					yield return enumerator.Current;
                enumerator.Dispose();
			}
		}

		IEnumerator IEnumerable.GetEnumerator()
		{
			return GetEnumerator();
		}

		public bool Contains(Card card)
		{
			return this.Any(c => c == card);
		}
	}
}