From 8e7de1a4fa0ae39dd43b4a92b0bf84ec24e0d63b Mon Sep 17 00:00:00 2001
From: atlimit8 <atlimit8-vcs@gmx.com>
Date: Tue, 14 Mar 2017 22:37:43 -0500
Subject: [PATCH] Make UncloakOn Damage check sign with added Heal value for
 UncloakOn.

---
 OpenRA.Mods.Common/Traits/Cloak.cs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/OpenRA.Mods.Common/Traits/Cloak.cs b/OpenRA.Mods.Common/Traits/Cloak.cs
index 2facb3f929..baa8e57633 100644
--- a/OpenRA.Mods.Common/Traits/Cloak.cs
+++ b/OpenRA.Mods.Common/Traits/Cloak.cs
@@ -29,7 +29,8 @@ namespace OpenRA.Mods.Common.Traits
 		Infiltrate = 8,
 		Demolish = 16,
 		Damage = 32,
-		Dock = 64
+		Heal = 64,
+		Dock = 128
 	}
 
 	[Desc("This unit can cloak and uncloak in specific situations.")]
@@ -101,7 +102,10 @@ namespace OpenRA.Mods.Common.Traits
 
 		void INotifyDamage.Damaged(Actor self, AttackInfo e)
 		{
-			if (Info.UncloakOn.HasFlag(UncloakType.Damage))
+			if (e.Damage.Value == 0)
+				return;
+
+			if (Info.UncloakOn.HasFlag(e.Damage.Value > 0 ? UncloakType.Damage : UncloakType.Heal))
 				Uncloak();
 		}