view API/Commands/RemoveUnitCommand.cs @ 473:0d032c04210e

Re #359: Add "only contained" attribute to unit types * Make sure remove handles sub-units as well Note: warning the user that the unit has sub-units rather than just removing them is left to the UI
author IBBoard <dev@ibboard.co.uk>
date Mon, 16 Apr 2012 20:47:14 +0100
parents 3c4a6403a88c
children
line wrap: on
line source

// This file (RemoveUnitCommand.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;
using System.Collections.Generic;

namespace IBBoard.WarFoundry.API.Commands
{
	/// <summary>
	/// Summary description for RemoveUnitCommand.
	/// </summary>
	public class RemoveUnitCommand : Command
	{
		private Unit unit;
		private Unit parentUnit;
		private ArmyCategory cat;
		private List<RemoveUnitCommand> subCommands;

		public RemoveUnitCommand(Unit toRemove)
		{
			unit = toRemove;
			cat = unit.Category;
			parentUnit = unit.ParentUnit;
		}

		public override bool CanExecute()
		{
			return (unit != null);
		}

		public override string Description
		{
			get
			{
				return Translation.GetTranslation("removeUnitCommandDescription", "remove {0} from the army", unit.Name);
			}
		}

		public override string UndoDescription
		{
			get
			{
				return Translation.GetTranslation("removeUnitCommandUndoDescription", "re-add {0} to the army", unit.Name);
			}
		}

		private void Prepare()
		{
			subCommands = new List<RemoveUnitCommand>();
			foreach (Unit containedUnit in unit.ContainedUnits)
			{
				RemoveUnitCommand removeUnitCommand = new RemoveUnitCommand(containedUnit);
				removeUnitCommand.Prepare();
				subCommands.Add(removeUnitCommand);
			}
		}
		public override bool Execute()
		{
			Prepare();
			Redo();
			return true;
		}

		public override void Redo()
		{
			unit.ParentUnit = null;
			cat.RemoveUnit(unit);

			foreach (RemoveUnitCommand subCommand in subCommands)
			{
				subCommand.Redo();
			}
		}

		public override void Undo()
		{
			unit.ParentUnit = parentUnit;
			cat.AddUnit(unit);

			foreach (RemoveUnitCommand subCommand in subCommands)
			{
				subCommand.Undo();
			}
		}

		public override string Name
		{
			get { return "Remove unit"; }
		}
	}
}