public static void AddFieldDefinition(this ClassWeavingContext context)
{
context.WriteDebug($"Add lazy loader field to {context.TypeDefinition.Name}");
context.LazyLoaderField = new FieldDefinition("lazyLoader", FieldAttributes.Private | FieldAttributes.InitOnly,
context.References.LazyLoaderType);
context.TypeDefinition.Fields.Add(context.LazyLoaderField);
}
public static void AddConstructorOverloads(this ClassWeavingContext context)
{
foreach (MethodDefinition constructor in context.TypeDefinition.GetConstructors().ToList())
{
context.WriteDebug(
$"Add constructor overload for {context.TypeDefinition.Name}({string.Join(", ", constructor.Parameters.Select(x => $"{x.ParameterType.Name} {x.Name}"))})");
MethodDefinition method = new MethodDefinition(constructor.Name, constructor.Attributes,
context.TypeDefinition.Module.TypeSystem.Void)
{
IsFamily = true,
};
foreach (ParameterDefinition parameterDefinition in constructor.Parameters)
{
method.Parameters.Add(new ParameterDefinition(parameterDefinition.Name, parameterDefinition.Attributes,
parameterDefinition.ParameterType));
}
ParameterDefinition lazyLoaderParameter =
new ParameterDefinition("lazyLoader", ParameterAttributes.None, context.References.LazyLoaderType);
method.Parameters.Add(lazyLoaderParameter);
context.TypeDefinition.Methods.Add(method);
ILProcessorContext processor = method.Body.GetILProcessor().Start();
processor = processor.Append(x => x.Create(OpCodes.Ldarg_0));
foreach (ParameterDefinition parameterDefinition in constructor.Parameters)
{
processor = processor.Append(x => x.Create(OpCodes.Ldarg, parameterDefinition.Index + 1));
}
processor = processor.Append(x => x.Create(OpCodes.Call, constructor));
processor = processor.Append(x => x.Create(OpCodes.Ldarg_0))
.Append(x => x.Create(OpCodes.Ldarg, lazyLoaderParameter))
.Append(x => x.Create(OpCodes.Stfld, context.LazyLoaderField))
.Append(x => x.Create(OpCodes.Ret));
}
}
public virtual void WriteDebug(string message) => ClassWeavingContext.WriteDebug(message);