Allow ActorReferenceAttribute to list required trait types enforced by --check-yaml

This commit is contained in:
Taryn Hill
2015-12-27 10:26:49 -06:00
parent feba44cd6d
commit a7a6f41ef9
2 changed files with 81 additions and 11 deletions

View File

@@ -15,7 +15,14 @@ namespace OpenRA.Traits
/* attributes used by OpenRA.Lint to understand the rules */
[AttributeUsage(AttributeTargets.Field)]
public sealed class ActorReferenceAttribute : Attribute { }
public sealed class ActorReferenceAttribute : Attribute
{
public Type[] RequiredTraits;
public ActorReferenceAttribute(params Type[] requiredTraits)
{
RequiredTraits = requiredTraits;
}
}
[AttributeUsage(AttributeTargets.Field)]
public sealed class WeaponReferenceAttribute : Attribute { }

View File

@@ -11,6 +11,7 @@
using System;
using System.Linq;
using System.Reflection;
using OpenRA.GameRules;
using OpenRA.Traits;
namespace OpenRA.Mods.Common.Lint
@@ -34,22 +35,84 @@ namespace OpenRA.Mods.Common.Lint
foreach (var field in actualType.GetFields())
{
if (field.HasAttribute<ActorReferenceAttribute>())
CheckReference(actorInfo, traitInfo, field, rules.Actors, "actor");
CheckActorReference(actorInfo, traitInfo, field, rules.Actors,
field.GetCustomAttributes<ActorReferenceAttribute>(true)[0]);
if (field.HasAttribute<WeaponReferenceAttribute>())
CheckReference(actorInfo, traitInfo, field, rules.Weapons, "weapon");
CheckWeaponReference(actorInfo, traitInfo, field, rules.Weapons,
field.GetCustomAttributes<WeaponReferenceAttribute>(true)[0]);
if (field.HasAttribute<VoiceSetReferenceAttribute>())
CheckReference(actorInfo, traitInfo, field, rules.Voices, "voice");
CheckVoiceReference(actorInfo, traitInfo, field, rules.Voices,
field.GetCustomAttributes<VoiceSetReferenceAttribute>(true)[0]);
}
}
void CheckReference<T>(ActorInfo actorInfo, ITraitInfo traitInfo, FieldInfo fieldInfo,
IReadOnlyDictionary<string, T> dict, string type)
void CheckActorReference(ActorInfo actorInfo,
ITraitInfo traitInfo,
FieldInfo fieldInfo,
IReadOnlyDictionary<string, ActorInfo> dict,
ActorReferenceAttribute attribute)
{
var values = LintExts.GetFieldValues(traitInfo, fieldInfo, emitError);
foreach (var v in values)
if (v != null && !dict.ContainsKey(v.ToLowerInvariant()))
emitError("{0}.{1}.{2}: Missing {3} `{4}`."
.F(actorInfo.Name, traitInfo.GetType().Name, fieldInfo.Name, type, v));
foreach (var value in values)
{
if (value == null)
continue;
// NOTE: Once https://github.com/OpenRA/OpenRA/issues/4124 is resolved we won't
// have to .ToLower* anything here.
var v = value.ToLowerInvariant();
if (!dict.ContainsKey(v))
{
emitError("{0}.{1}.{2}: Missing actor `{3}`."
.F(actorInfo.Name, traitInfo.GetType().Name, fieldInfo.Name, value));
continue;
}
foreach (var requiredTrait in attribute.RequiredTraits)
if (!dict[v].TraitsInConstructOrder().Any(t => t.GetType() == requiredTrait || t.GetType().IsSubclassOf(requiredTrait)))
emitError("Actor type {0} does not have trait {1} which is required by {2}.{3}."
.F(value, requiredTrait.Name, traitInfo.GetType().Name, fieldInfo.Name));
}
}
void CheckWeaponReference(ActorInfo actorInfo,
ITraitInfo traitInfo,
FieldInfo fieldInfo,
IReadOnlyDictionary<string, WeaponInfo> dict,
WeaponReferenceAttribute attribute)
{
var values = LintExts.GetFieldValues(traitInfo, fieldInfo, emitError);
foreach (var value in values)
{
if (value == null)
continue;
if (!dict.ContainsKey(value.ToLower()))
emitError("{0}.{1}.{2}: Missing weapon `{3}`."
.F(actorInfo.Name, traitInfo.GetType().Name, fieldInfo.Name, value));
}
}
void CheckVoiceReference(ActorInfo actorInfo,
ITraitInfo traitInfo,
FieldInfo fieldInfo,
IReadOnlyDictionary<string, SoundInfo> dict,
VoiceSetReferenceAttribute attribute)
{
var values = LintExts.GetFieldValues(traitInfo, fieldInfo, emitError);
foreach (var value in values)
{
if (value == null)
continue;
if (!dict.ContainsKey(value.ToLower()))
emitError("{0}.{1}.{2}: Missing voice `{3}`."
.F(actorInfo.Name, traitInfo.GetType().Name, fieldInfo.Name, value));
}
}
}
}
}