Move everything ExternalCapture-related to Mods.Common

OpenRA.Mods.Common/Activities/ExternalCaptureActor.cs

@@ -0,0 +1,84 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2014 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 OpenRA.Activities;
+using OpenRA.Effects;
+using OpenRA.Mods.Common.Traits;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Activities
+{
+	class ExternalCaptureActor : Activity
+	{
+		Target target;
+
+		public ExternalCaptureActor(Target target) { this.target = target; }
+
+		public override Activity Tick(Actor self)
+		{
+			if (target.Type != TargetType.Actor)
+				return NextActivity;
+
+			var capturable = target.Actor.Trait<ExternalCapturable>();
+
+			if (IsCanceled || !self.IsInWorld || self.IsDead || !target.IsValidFor(self))
+			{
+				if (capturable.CaptureInProgress)
+					capturable.EndCapture();
+
+				return NextActivity;
+			}
+
+			var mobile = self.Trait<Mobile>();
+			var nearest = target.Actor.OccupiesSpace.NearestCellTo(mobile.ToCell);
+
+			if ((nearest - mobile.ToCell).LengthSquared > 2)
+				return Util.SequenceActivities(new MoveAdjacentTo(self, target), this);
+
+			if (!capturable.CaptureInProgress)
+				capturable.BeginCapture(self);
+			else
+			{
+				if (capturable.Captor != self) return NextActivity;
+
+				if (capturable.CaptureProgressTime % 25 == 0)
+				{
+					self.World.Add(new FlashTarget(target.Actor, self.Owner));
+					self.World.Add(new FlashTarget(self));
+				}
+
+				if (capturable.CaptureProgressTime == capturable.Info.CaptureCompleteTime * 25)
+				{
+					var capturesInfo = self.Info.Traits.Get<ExternalCapturesInfo>();
+
+					self.World.AddFrameEndTask(w =>
+					{
+						if (target.Actor.IsDead)
+							return;
+
+						var oldOwner = target.Actor.Owner;
+
+						target.Actor.ChangeOwner(self.Owner);
+
+						foreach (var t in target.Actor.TraitsImplementing<INotifyCapture>())
+							t.OnCapture(target.Actor, self, oldOwner, self.Owner);
+
+						capturable.EndCapture();
+
+						if (capturesInfo != null && capturesInfo.ConsumeActor)
+							self.Destroy();
+					});
+				}
+			}
+
+			return this;
+		}
+	}
+}