view API/Factories/Requirement/UnitRequiresAtLeastNUnitsRequirementFactory.cs @ 374:13793f3a2a2e

Re #351: Add extensible requirement handling method * Add initial conditional to fetching of factory * Exception if content isn't correct
author IBBoard <dev@ibboard.co.uk>
date Sat, 02 Jul 2011 14:59:45 +0000
parents e10630f39ec2
children d9bf78a8f517
line wrap: on
line source

// This file (UnitRequiresAtLeastNUnitsRequirementFactory.cs) is a part of the IBBoard.WarFoundry.API project and is copyright 2011 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;
using IBBoard.WarFoundry.API.Objects.Requirement;

namespace IBBoard.WarFoundry.API.Factories.Requirement
{
	public class UnitRequiresAtLeastNUnitsRequirementFactory
	{
		public UnitRequiresAtLeastNUnitsRequirementFactory()
		{
			//Do nothing special
		}

		public UnitRequiresAtLeastNUnitsRequirement CreateRequirement(UnitType type, string data)
		{
			UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(type);
			Race race = type.Race;
			AddRequirements(req, race, data);
			return req;
		}

		private void AddRequirements(UnitRequiresAtLeastNUnitsRequirement req, Race race, string data)
		{
			foreach (string requirement in data.Split('|'))
			{
				string[] requirementParts = requirement.Split(':');
				UnitType unitType = race.GetUnitType(requirementParts[0]);

				if (unitType == null)
				{
					throw new InvalidRequirementException(String.Format("Invalid unit type '{0}' for 'Requires at least N units' requirement", requirementParts[0]));
				}

				if (requirementParts.Length == 2)
				{
					req.AddUnitTypeRequirement(unitType, Int32.Parse(requirementParts[1]));
				}
				else
				{
					req.AddUnitTypeRequirement(unitType);
				}
			}
		}
	}
}