OpenRA/OpenRA.Mods.RA/UnitStances/UnitStanceGuard.cs

using System;
using System.Drawing;
using OpenRA.Mods.RA.Move;
using OpenRA.Traits;

namespace OpenRA.Mods.RA
{
	public class UnitStanceGuardInfo : UnitStanceInfo
	{
		public override object Create(ActorInitializer init) { return new UnitStanceGuard(init.self, this); }
	}

	public class UnitStanceGuard : UnitStance, INotifyDamage, ISelectionColorModifier
	{
		public UnitStanceGuard(Actor self, UnitStanceGuardInfo info)
		{
			Info = info;
			Active = (self.World.LocalPlayer == self.Owner || (self.Owner.IsBot && Game.IsHost)) ? Info.Default : false;
		}

		protected override void OnScan(Actor self)
		{
			if (!self.IsIdle) return;
			if (!self.HasTrait<AttackBase>()) return;

			var target = ScanForTarget(self);
			if (target == null)
				return;

			AttackTarget(self, target, true);
		}

		protected override void OnFirstTick(Actor self)
		{
			if (!self.HasTrait<AttackBase>()) return;

			if (self.Trait<AttackBase>().IsAttacking)
				StopAttack(self);
		}

		public void Damaged(Actor self, AttackInfo e)
		{
			if (!Active) return;
			if (!self.HasTrait<AttackBase>()) return;

			ReturnFire(self, e, false, false, true); // only triggers when standing still
		}

		public virtual Color GetSelectionColorModifier(Actor self, Color defaultColor)
		{
			return Active ? Color.Yellow : defaultColor;
		}
	}
}