// This file (UnitRequirementMinNumber.cs) is a part of the IBBoard.WarFoundry.API project and is copyright 2008, 2009 IBBoard.
//
// The file and the library/program it is in are licensed and distributed, without warranty, under the GNU Affero GPL license, either version 3 of the License or (at your option) any later version. Please see COPYING for more information and the full license.

using System;
using IBBoard.WarFoundry.API.Objects;

namespace IBBoard.WarFoundry.API.Requirements
{
	public class UnitRequirementMinNumber : UnitRequirement
	{
		private int minUnitCount;
		
		public UnitRequirementMinNumber(UnitType type, int minNumber) : base(type)
		{
			minUnitCount = minNumber;
		}
		
		public override string Description
		{
			get { return "You must include at least "+minUnitCount+" "+unitType.Name+" units in an army"; }
		}
		
		protected override AbstractFailedRequirement CanAddToArmy(Army army, UnitType type)
		{
			return null;
		}

		protected override AbstractFailedRequirement CanRemoveFromArmy (Army army, UnitType type)
		{
			FailedUnitRequirement failed = null;
			
			if (army.GetUnitTypeCount(type) <= minUnitCount)
			{
				failed = new FailedUnitRequirement(this);
			}
			
			return failed;
		}
	}
}