Rework UpgradeActorsNear -> ProximityExternalCondition.

OpenRA.Mods.Common/OpenRA.Mods.Common.csproj

@@ -493,7 +493,7 @@
     <Compile Include="Traits\Upgrades\DeployToUpgrade.cs" />
     <Compile Include="Traits\Upgrades\DisableOnUpgrade.cs" />
     <Compile Include="Traits\Upgrades\UpgradableTrait.cs" />
-    <Compile Include="Traits\Upgrades\UpgradeActorsNear.cs" />
+    <Compile Include="Traits\Upgrades\ProximityExternalCondition.cs" />
     <Compile Include="Traits\Upgrades\UpgradeOnDamageState.cs" />
     <Compile Include="Traits\Upgrades\UpgradeOnTerrain.cs" />
     <Compile Include="Traits\Upgrades\UpgradeOnMovement.cs" />

OpenRA.Mods.Common/Traits/Upgrades/ProximityExternalCondition.cs

@@ -9,16 +9,17 @@
  */
 #endregion
 
+using System.Collections.Generic;
 using OpenRA.Traits;
 
 namespace OpenRA.Mods.Common.Traits
 {
 	[Desc("Applies an upgrade to actors within a specified range.")]
-	public class UpgradeActorsNearInfo : ITraitInfo
+	public class ProximityExternalConditionInfo : ITraitInfo
 	{
-		[UpgradeGrantedReference, FieldLoader.Require]
-		[Desc("The upgrades to grant.")]
-		public readonly string[] Upgrades = { };
+		[FieldLoader.Require]
+		[Desc("The condition to apply. Must be included in the target actor's ExternalConditions list.")]
+		public readonly string Condition = null;
 
 		[Desc("The range to search for actors to upgrade.")]
 		public readonly WDist Range = WDist.FromCells(3);
@@ -36,14 +37,16 @@ namespace OpenRA.Mods.Common.Traits
 		public readonly string EnableSound = null;
 		public readonly string DisableSound = null;
 
-		public object Create(ActorInitializer init) { return new UpgradeActorsNear(init.Self, this); }
+		public object Create(ActorInitializer init) { return new ProximityExternalCondition(init.Self, this); }
 	}
 
-	public class UpgradeActorsNear : ITick, INotifyAddedToWorld, INotifyRemovedFromWorld, INotifyOtherProduction
+	public class ProximityExternalCondition : ITick, INotifyAddedToWorld, INotifyRemovedFromWorld, INotifyOtherProduction
 	{
-		readonly UpgradeActorsNearInfo info;
+		readonly ProximityExternalConditionInfo info;
 		readonly Actor self;
 
+		readonly Dictionary<Actor, int> tokens = new Dictionary<Actor, int>();
+
 		int proximityTrigger;
 		WPos cachedPosition;
 		WDist cachedRange;
@@ -53,7 +56,7 @@ namespace OpenRA.Mods.Common.Traits
 
 		bool cachedDisabled = true;
 
-		public UpgradeActorsNear(Actor self, UpgradeActorsNearInfo info)
+		public ProximityExternalCondition(Actor self, ProximityExternalConditionInfo info)
 		{
 			this.info = info;
 			this.self = self;
@@ -106,9 +109,8 @@ namespace OpenRA.Mods.Common.Traits
 				return;
 
 			var um = a.TraitOrDefault<UpgradeManager>();
-			if (um != null)
-				foreach (var u in info.Upgrades)
-					um.GrantUpgrade(a, u, this);
+			if (um != null && !tokens.ContainsKey(a) && um.AcceptsExternalCondition(a, info.Condition))
+				tokens[a] = um.GrantCondition(a, info.Condition, true);
 		}
 
 		public void UnitProducedByOther(Actor self, Actor producer, Actor produced)
@@ -129,26 +131,24 @@ namespace OpenRA.Mods.Common.Traits
 					return;
 
 				var um = produced.TraitOrDefault<UpgradeManager>();
-				if (um != null)
-					foreach (var u in info.Upgrades)
-						if (um.AcknowledgesUpgrade(produced, u))
-							um.GrantTimedUpgrade(produced, u, 1);
+				if (um != null && um.AcceptsExternalCondition(produced, info.Condition))
+					tokens[produced] = um.GrantCondition(produced, info.Condition, true);
 			}
 		}
 
 		void ActorExited(Actor a)
 		{
-			if (a == self || a.Disposed || self.Disposed)
+			if (a.Disposed)
 				return;
 
-			var stance = self.Owner.Stances[a.Owner];
-			if (!info.ValidStances.HasStance(stance))
+			int token;
+			if (!tokens.TryGetValue(a, out token))
 				return;
 
+			tokens.Remove(a);
 			var um = a.TraitOrDefault<UpgradeManager>();
 			if (um != null)
-				foreach (var u in info.Upgrades)
-					um.RevokeUpgrade(a, u, this);
+				um.RevokeCondition(a, token);
 		}
 	}
 }

OpenRA.Mods.Common/UtilityCommands/UpgradeRules.cs

@@ -567,6 +567,15 @@ namespace OpenRA.Mods.Common.UtilityCommands
 					}
 				}
 
+				if (engineVersion < 20161212)
+				{
+					if (node.Key.StartsWith("UpgradeActorsNear", StringComparison.Ordinal))
+					{
+						RenameNodeKey(node, "ProximityExternalCondition");
+						ConvertUpgradesToCondition(parent, node, "Upgrades", "Condition");
+					}
+				}
+
 				UpgradeActorRules(modData, engineVersion, ref node.Value.Nodes, node, depth + 1);
 			}