Add actor conditions lint rule.

OpenRA.Mods.Common/Lint/CheckConditions.cs

@@ -0,0 +1,72 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2016 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, either version 3 of
+ * the License, or (at your option) any later version. For more
+ * information, see COPYING.
+ */
+#endregion
+
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using OpenRA.Mods.Common.Traits;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Lint
+{
+	public class CheckConditions : ILintRulesPass
+	{
+		public void Run(Action<string> emitError, Action<string> emitWarning, Ruleset rules)
+		{
+			foreach (var actorInfo in rules.Actors)
+			{
+				if (actorInfo.Key.StartsWith("^", StringComparison.Ordinal))
+					continue;
+
+				var granted = new HashSet<string>();
+				var consumed = new HashSet<string>();
+
+				foreach (var trait in actorInfo.Value.TraitInfos<ITraitInfo>())
+				{
+					var fieldConsumed = trait.GetType().GetFields()
+						.Where(x => x.HasAttribute<ConsumedConditionReferenceAttribute>())
+						.SelectMany(f => LintExts.GetFieldValues(trait, f, emitError));
+
+					var propertyConsumed = trait.GetType().GetProperties()
+						.Where(x => x.HasAttribute<ConsumedConditionReferenceAttribute>())
+						.SelectMany(p => LintExts.GetPropertyValues(trait, p, emitError));
+
+					var fieldGranted = trait.GetType().GetFields()
+                        .Where(x => x.HasAttribute<GrantedConditionReferenceAttribute>())
+	  					.SelectMany(f => LintExts.GetFieldValues(trait, f, emitError));
+
+					var propertyGranted = trait.GetType().GetProperties()
+                        .Where(x => x.HasAttribute<GrantedConditionReferenceAttribute>())
+	  					.SelectMany(f => LintExts.GetPropertyValues(trait, f, emitError));
+
+					foreach (var c in fieldConsumed.Concat(propertyConsumed))
+						if (!string.IsNullOrEmpty(c))
+							consumed.Add(c);
+
+					foreach (var g in fieldGranted.Concat(propertyGranted))
+						if (!string.IsNullOrEmpty(g))
+							granted.Add(g);
+				}
+
+				var unconsumed = granted.Except(consumed);
+				if (unconsumed.Any())
+					emitWarning("Actor type `{0}` grants conditions that are not consumed: {1}".F(actorInfo.Key, unconsumed.JoinWith(", ")));
+
+				var ungranted = consumed.Except(granted);
+				if (ungranted.Any())
+					emitError("Actor type `{0}` consumes conditions that are not granted: {1}".F(actorInfo.Key, ungranted.JoinWith(", ")));
+
+				if ((consumed.Any() || granted.Any()) && actorInfo.Value.TraitInfoOrDefault<UpgradeManagerInfo>() == null)
+					emitError("Actor type `{0}` defines conditions but does not include an UpgradeManager".F(actorInfo.Key));
+			}
+		}
+	}
+}