using OpenRa.Traits;
using OpenRa.Orders;
using System.Collections.Generic;
using System.Linq;

namespace OpenRa.Traits
{
	class ChronoshiftableInfo : ITraitInfo
	{
		public object Create(Actor self) { return new Chronoshiftable(self); }
	}

	public class Chronoshiftable : IResolveOrder, ITick
	{
		// Return-to-sender logic
		[Sync]
		int2 chronoshiftOrigin;
		[Sync]
		int chronoshiftReturnTicks = 0;

		public Chronoshiftable(Actor self) { }

		public void Tick(Actor self)
		{
			if (chronoshiftReturnTicks <= 0)
				return;

			if (chronoshiftReturnTicks > 0)
				chronoshiftReturnTicks--;

			// Return to original location
			if (chronoshiftReturnTicks == 0)
			{
				self.CancelActivity();
				// Todo: need a new Teleport method that will move to the closest available cell
				self.QueueActivity(new Activities.Teleport(chronoshiftOrigin));
			}
		}

		public void ResolveOrder(Actor self, Order order)
		{
			var movement = self.traits.GetOrDefault<IMovement>();
			if (order.OrderString == "Chronoshift" && movement.CanEnterCell(order.TargetLocation))
			{
				// Cannot chronoshift into unexplored location
				if (!self.Owner.Shroud.IsExplored(order.TargetLocation))
					return;

				var info = self.Owner.PlayerActor.Info.Traits.Get<ChronoshiftPowerInfo>();
				
				// Set up return-to-sender info
				chronoshiftOrigin = self.Location;
				chronoshiftReturnTicks = (int)(info.Duration * 60 * 25);

				// Kill cargo
				if (info.KillCargo && self.traits.Contains<Cargo>())
				{
					var cargo = self.traits.Get<Cargo>();
					while (!cargo.IsEmpty(self))
					{
						order.Player.Kills++;
						cargo.Unload(self);
					}
				}
				
				// Set up the teleport
				self.CancelActivity();
				self.QueueActivity(new Activities.Teleport(order.TargetLocation));
			}
		}
	}

}