Dissolve Crates namespace, move CrateActions and Crate trait to Traits/Crates

OpenRA.Mods.RA/Traits/Crates/GrantUpgradeCrateAction.cs

@@ -0,0 +1,90 @@
+#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 System.Linq;
+using OpenRA.Mods.Common.Traits;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.RA.Traits
+{
+	[Desc("Grants an upgrade to the collector.")]
+	public class GrantUpgradeCrateActionInfo : CrateActionInfo
+	{
+		[Desc("The upgrades to apply.")]
+		public readonly string[] Upgrades = { };
+
+		[Desc("Duration of the upgrade (in ticks). Set to 0 for a permanent upgrade.")]
+		public readonly int Duration = 0;
+
+		[Desc("The range to search for extra collectors in.", "Extra collectors will also be granted the crate action.")]
+		public readonly WRange Range = new WRange(3);
+
+		[Desc("The maximum number of extra collectors to grant the crate action to.", "-1 = no limit")]
+		public readonly int MaxExtraCollectors = 4;
+
+		public override object Create(ActorInitializer init) { return new GrantUpgradeCrateAction(init.self, this); }
+	}
+
+	public class GrantUpgradeCrateAction : CrateAction
+	{
+		readonly Actor self;
+		readonly GrantUpgradeCrateActionInfo info;
+
+		public GrantUpgradeCrateAction(Actor self, GrantUpgradeCrateActionInfo info)
+			: base(self, info) 
+		{
+			this.self = self;
+			this.info = info;
+		}
+
+		bool AcceptsUpgrade(Actor a)
+		{
+			var um = a.TraitOrDefault<UpgradeManager>();
+			return um != null && info.Upgrades.Any(u => um.AcceptsUpgrade(a, u));
+		}
+
+		public override int GetSelectionShares(Actor collector)
+		{
+			return AcceptsUpgrade(collector) ? info.SelectionShares : 0;
+		}
+
+		public override void Activate(Actor collector)
+		{
+			var actorsInRange = self.World.FindActorsInCircle(self.CenterPosition, info.Range)
+				.Where(a => a != self && a != collector && a.Owner == collector.Owner && AcceptsUpgrade(a));
+
+			if (info.MaxExtraCollectors > -1)
+				actorsInRange = actorsInRange.Take(info.MaxExtraCollectors);
+
+			collector.World.AddFrameEndTask(w =>
+			{
+				foreach (var a in actorsInRange.Append(collector))
+				{
+					if (!a.IsInWorld || a.IsDead)
+						continue;
+
+					var um = a.TraitOrDefault<UpgradeManager>();
+					foreach (var u in info.Upgrades)
+					{
+						if (!um.AcceptsUpgrade(a, u))
+							continue;
+
+						if (info.Duration > 0)
+							um.GrantTimedUpgrade(a, u, info.Duration);
+						else
+							um.GrantUpgrade(a, u, this);
+					}
+				}
+			});
+
+			base.Activate(collector);
+		}
+	}
+}