// 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) : this(toAdd, armyCatTo, null)
		{
			//Do nothing special
		}

		public CreateAndAddUnitCommand(UnitType toAdd, ArmyCategory armyCatTo, Unit parentUnit)
		{
			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);
			Redo();
			return true;
		}

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

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

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

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