Allow ActorReferenceAttribute to list required trait types enforced by --check-yaml
This commit is contained in:
@@ -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 { }
|
||||
|
||||
@@ -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));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user