private void AddProxyInfo(ClassNotation @class, ProxyGeneratorContext context, ITypeSymbolInfo type)
{
@class.Members.Add(new FieldNotation()
{
Accessibility = AccessibilityInfo.Public, Type = type.FullName, Name = context.GetProxyFieldName()
});
var setProxyNode = new MethodNotation()
{
Accessibility = AccessibilityInfo.Public, ReturnType = "void", Name = "SetProxy"
};
setProxyNode.Parameters.Add(new ParameterNotation()
{
Type = "object", Name = "instance"
});
setProxyNode.Parameters.Add(new ParameterNotation()
{
Type = "System.IServiceProvider", Name = "serviceProvider"
});
setProxyNode.Body.AddRange(Notation.Create(context.GetProxyFieldName(), " = instance as ", type.FullName, ";"));
@class.Members.Add(setProxyNode);
@class.Inherits.Add("Norns.Destiny.AOP.IInterceptProxy".ToNotation());
foreach (var f in @class.Members.Select(i => i as FieldNotation).Where(i => i != null && i.IsFromDI))
{
setProxyNode.Body.AddRange(Notation.Create(f.Name, " = serviceProvider.GetService(typeof(", f.Type, ")) as ", f.Type, ";"));
}
}
private INotation CreateProxyMethod(IMethodSymbolInfo method, ProxyGeneratorContext typeContext)
{
var context = new ProxyGeneratorContext()
{
Parent = typeContext,
Symbol = method
};
var notation = method.ToNotationDefinition();
context.SetCurrentMethodNotation(notation);
var isInterface = method.ContainingType.IsInterface;
notation.IsOverride = !isInterface && method.CanOverride();
var returnValueParameterName = context.GetReturnValueParameterName();
if (method.HasReturnValue)
{
notation.Body.AddRange(Notation.Create("var ", returnValueParameterName, " = default(", method.IsAsync ? method.ReturnType.TypeArguments.First().FullName : method.ReturnType.FullName, ");"));
}
notation.Body.Add(method.Parameters.Where(i => i.RefKind == RefKindInfo.Out).Select(i => $"{i.Name} = default;".ToNotation()).Combine());
notation.Body.AddRange(interceptors.SelectMany(i => i.BeforeMethod(context)));
if (isInterface || (!method.IsAbstract && (method.Accessibility == AccessibilityInfo.Internal || method.Accessibility == AccessibilityInfo.Public)))
{
if (method.HasReturnValue)
{
notation.Body.AddRange(Notation.Create(returnValueParameterName, " = "));
}
notation.Body.AddRange(Notation.Create(method.IsAsync ? "await " : string.Empty, method.ContainingType.IsInterface && !method.IsAbstract ? $"({context.GetProxyFieldName()} as {method.ContainingType.FullName})" : context.GetProxyFieldName(), ".", method.Name));
notation.Body.Add(ConstNotations.OpenParen);
notation.Body.Add(notation.Parameters.ToCallParameters());
notation.Body.Add(ConstNotations.CloseParen);
notation.Body.Add(ConstNotations.Semicolon);
}
notation.Body.AddRange(interceptors.SelectMany(i => i.AfterMethod(context)));
if (method.HasReturnValue)
{
notation.Body.AddRange(Notation.Create("return ", returnValueParameterName, ";"));
}
return(notation);
}
private INotation CreateProxyProperty(IPropertySymbolInfo property, ProxyGeneratorContext typeContext)
{
var context = new ProxyGeneratorContext()
{
Parent = typeContext,
Symbol = property
};
PropertyNotation notation;
List <INotation> callName = new List <INotation>();
if (property.IsIndexer)
{
var indexer = new IndexerPropertyNotation();
indexer.Parameters.AddRange(property.Parameters.Select(i => new ParameterNotation()
{
Type = i.Type.FullName,
Name = i.Name
}));
notation = indexer;
callName.Add(ConstNotations.OpenBracket);
callName.Add(indexer.Parameters.ToCallParameters());
callName.Add(ConstNotations.CloseBracket);
}
else
{
callName.Add(ConstNotations.Dot);
callName.Add(property.Name.ToNotation());
notation = new PropertyNotation();
}
var isInterface = property.ContainingType.IsInterface;
notation.IsOverride = !isInterface && property.CanOverride();
notation.Accessibility = property.Accessibility;
notation.Name = property.Name;
notation.Type = property.Type.FullName;
if (property.CanRead)
{
context.SetCurrentPropertyMethod(property.GetMethod);
var getter = PropertyMethodNotation.Create(true);
getter.Accessibility = property.GetMethod.Accessibility;
var returnValueParameterName = context.GetReturnValueParameterName();
getter.Body.AddRange(Notation.Create("var ", returnValueParameterName, " = default(", property.Type.FullName, ");"));
getter.Body.AddRange(interceptors.SelectMany(i => i.BeforeMethod(context)));
if (isInterface || (getter.Accessibility == AccessibilityInfo.Public || getter.Accessibility == AccessibilityInfo.Internal))
{
getter.Body.AddRange(Notation.Create(returnValueParameterName, " = ", context.GetProxyFieldName()));
getter.Body.AddRange(callName);
getter.Body.Add(ConstNotations.Semicolon);
}
getter.Body.AddRange(interceptors.SelectMany(i => i.AfterMethod(context)));
getter.Body.AddRange(Notation.Create("return ", returnValueParameterName, ";"));
notation.Accessers.Add(getter);
}
if (property.CanWrite)
{
context.SetCurrentPropertyMethod(property.SetMethod);
var setter = PropertyMethodNotation.Create(false);
setter.Accessibility = property.SetMethod.Accessibility;
var returnValueParameterName = context.GetReturnValueParameterName();
setter.Body.AddRange(Notation.Create("var ", returnValueParameterName, " = value;"));
setter.Body.AddRange(interceptors.SelectMany(i => i.BeforeMethod(context)));
if (isInterface || (setter.Accessibility == AccessibilityInfo.Public || setter.Accessibility == AccessibilityInfo.Internal))
{
setter.Body.Add(context.GetProxyFieldName().ToNotation());
setter.Body.AddRange(callName);
setter.Body.AddRange(Notation.Create(" = ", returnValueParameterName, ";"));
}
setter.Body.AddRange(interceptors.SelectMany(i => i.AfterMethod(context)));
notation.Accessers.Add(setter);
}
return(notation);
}
