#region Copyright & License Information
/*
 * Copyright 2007-2015 The OpenRA Developers (see AUTHORS)
 * This file is part of OpenRA, which is free software. It is made
 * available to you under the terms of the GNU General Public License
 * as published by the Free Software Foundation. For more information,
 * see COPYING.
 */
#endregion

using System.Collections.Generic;
using System.Linq;
using OpenRA.GameRules;
using OpenRA.Mods.Common.Traits;
using OpenRA.Traits;

namespace OpenRA.Mods.Cnc.Traits
{
	class PoisonedByTiberiumInfo : UpgradableTraitInfo
	{
		[WeaponReference] public readonly string Weapon = "Tiberium";
		public readonly HashSet<string> Resources = new HashSet<string> { "Tiberium", "BlueTiberium" };

		public override object Create(ActorInitializer init) { return new PoisonedByTiberium(this); }
	}

	class PoisonedByTiberium : UpgradableTrait<PoisonedByTiberiumInfo>, ITick, ISync
	{
		[Sync] int poisonTicks;

		public PoisonedByTiberium(PoisonedByTiberiumInfo info)
			: base(info) { }

		public void Tick(Actor self)
		{
			if (IsTraitDisabled || --poisonTicks > 0)
				return;

			// Prevents harming infantry in cargo.
			if (!self.IsInWorld)
				return;

			var rl = self.World.WorldActor.Trait<ResourceLayer>();
			var r = rl.GetResource(self.Location);
			if (r == null || !Info.Resources.Contains(r.Info.Name))
				return;

			var weapon = self.World.Map.Rules.Weapons[Info.Weapon.ToLowerInvariant()];

			weapon.Impact(Target.FromActor(self), self.World.WorldActor, Enumerable.Empty<int>());
			poisonTicks = weapon.ReloadDelay;
		}
	}
}