view API/Commands/CreateAndAddUnitCommand.cs @ 471:0a2068897793

Re #359: Add "only contained" attribute to unit types * Add "main units" methods to get only top-level units * Add nesting to unit creation command
author IBBoard <dev@ibboard.co.uk>
date Sun, 15 Apr 2012 20:52:32 +0100
parents 3c4a6403a88c
children e6c93ceba119
line wrap: on
line source

// This file (CreateAndAddUnitCommand.cs) is a part of the IBBoard.WarFoundry.API project and is copyright 2007, 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.Commands;
using IBBoard.Lang;
using IBBoard.WarFoundry.API.Objects;

namespace IBBoard.WarFoundry.API.Commands
{
	public class CreateAndAddUnitCommand : Command
	{
		private UnitType addedUnitType;
		private ArmyCategory armyCat;
		private Unit addedUnit;
		private Unit parentUnit;

		public CreateAndAddUnitCommand(UnitType toAdd, ArmyCategory armyCatTo, Unit parentUnit = null)
		{
			addedUnitType = toAdd;
			armyCat = armyCatTo;
			this.parentUnit = parentUnit;
		}

		public override bool CanExecute()
		{
			return (addedUnitType != null && armyCat != null);
		}

		public override string Description
		{
			get
			{
				return Translation.GetTranslation("createAndAddUnitCommandDescription", "add unit of {0} to the army", addedUnitType.Name);
			}
		}

		public override string UndoDescription
		{
			get
			{
				return Translation.GetTranslation("createAndAddUnitCommandUndoDescription", "remove unit of {0} from army", addedUnitType.Name);
			}
		}

		public override bool Execute()
		{
			addedUnit = new Unit(addedUnitType, armyCat);
			addedUnit.ParentUnit = parentUnit;
			this.Redo();
			return true;
		}

		public override void Redo()
		{
			armyCat.AddUnit(addedUnit);
		}

		public override void Undo()
		{
			armyCat.RemoveUnit(addedUnit);
		}

		public override string Name
		{
			get { return "Add new unit"; }
		}

		public Unit Unit
		{
			get { return addedUnit; }
		}
	}
}