public void ProvideAdvices(Weaver codeWeaver)
{
// Gets the dictionary of custom attributes.
AnnotationRepositoryTask customAttributeDictionary =
AnnotationRepositoryTask.GetTask(this.Project);
// Requests an enumerator of all instances of our Singleton.
IEnumerator<IAnnotationInstance> customAttributeEnumerator =
customAttributeDictionary.GetAnnotationsOfType(typeof (SingletonAttribute), true);
ICollection<TypeDefDeclaration> singletons = new HashSet<TypeDefDeclaration>();
// For each instance of our Singleton.
while (customAttributeEnumerator.MoveNext())
{
// Gets the type to which it applies.
TypeDefDeclaration typeDef = customAttributeEnumerator.Current.TargetElement
as TypeDefDeclaration;
if (typeDef != null && !singletons.Contains(typeDef))
{
singletons.Add(typeDef);
codeWeaver.AddTypeLevelAdvice(new SingletonAccessorAdvice(typeDef),
JoinPointKinds.BeforeStaticConstructor,
new Singleton<TypeDefDeclaration>(typeDef));
codeWeaver.AddMethodLevelAdvice(new SingletonAdvice(typeDef),
null,
JoinPointKinds.InsteadOfNewObject,
new Singleton<MetadataDeclaration>(
typeDef.Methods.GetOneByName(".ctor")));
}
}
singletons.Clear();
foreach (AssemblyRefDeclaration assembly in this.Project.Module.AssemblyRefs)
{
foreach (TypeRefDeclaration type in assembly.TypeRefs)
{
TypeDefDeclaration def = type.GetTypeDefinition();
foreach (CustomAttributeDeclaration att in def.CustomAttributes)
{
if (Equals(att.Constructor.DeclaringType.GetSystemType(new Type[] {}, new Type[] {}),
typeof (SingletonAttribute)))
{
singletons.Add(def);
}
}
}
}
foreach (TypeDefDeclaration type in singletons)
{
codeWeaver.AddMethodLevelAdvice(new SingletonAdvice(type),
null,
JoinPointKinds.InsteadOfNewObject,
new Singleton<MetadataDeclaration>(type.Methods.GetOneByName(".ctor")));
}
}