huanld/ProxyGenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

AutoMapper

Browse Source
This commit is contained in:
Stef Heyenrath
2021-07-25 15:26:50 +02:00
parent 3c9a292a0c
commit 2531e8e688
9 changed files with 264 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ProxyInterfaceSourceGenerator/Extensions/SymbolExtensions.cs
+13 -5
View File
@@ -39,15 +39,23 @@ namespace ProxyInterfaceSourceGenerator.Extensions
return $"{property.Type} {property.Name} {{ {get}{set}}}";
}
public static string ToPropertyTextForClass(this IPropertySymbol property, string interfaceName, string className)
public static string ToPropertyTextForClass(this IPropertySymbol property, string overrideType)
{
var classNameProxy = $"{className}Proxy";
var get = property.GetMethod != null ? $"get => new {classNameProxy}(_Instance.{property.Name}); " : string.Empty;
var set = property.SetMethod != null ? $"set => _Instance.{property.Name} = (({classNameProxy}) value)._Instance; " : string.Empty;
var get = property.GetMethod != null ? $"get => _mapper.Map<{overrideType}>(_Instance.{property.Name}); " : string.Empty;
var set = property.SetMethod != null ? $"set => _Instance.{property.Name} = _mapper.Map<{property.Type}>(value); " : string.Empty;
return $"{interfaceName} {property.Name} {{ {get}{set}}}";
return $"{overrideType} {property.Name} {{ {get}{set}}}";
}
//public static string ToPropertyTextForClass(this IPropertySymbol property, string overrideType, string className)
//{
// var classNameProxy = $"{className}Proxy";
// var get = property.GetMethod != null ? $"get => new {classNameProxy}(_Instance.{property.Name}); " : string.Empty;
// var set = property.SetMethod != null ? $"set => _Instance.{property.Name} = (({classNameProxy}) value)._Instance; " : string.Empty;
// return $"{overrideType} {property.Name} {{ {get}{set}}}";
//}
public static string ToMethodText(this IMethodSymbol method)
{
var parameters = new List<string>();
src/ProxyInterfaceSourceGenerator/FileGenerators/BaseGenerator.cs
+37
View File
@@ -1,4 +1,6 @@
using System;
using System.Collections.Generic;
using System.Linq;
using Microsoft.CodeAnalysis;
namespace ProxyInterfaceSourceGenerator.FileGenerators
@@ -12,6 +14,41 @@ namespace ProxyInterfaceSourceGenerator.FileGenerators
_context = context;
}
protected string GetPropertyType(IPropertySymbol property, out Dictionary<string, string> differs)
{
differs = new Dictionary<string, string>();
var existing = _context.CandidateInterfaces.Values.FirstOrDefault(x => x.TypeName == property.Type.ToString());
if (existing is not null)
{
differs.Add(property.Type.ToString(), existing.InterfaceName);
return existing.InterfaceName;
}
if (property.Type is INamedTypeSymbol namedTypedSymbol)
{
var type = property.Type.ToString();
foreach (var typeArgument in namedTypedSymbol.TypeArguments)
{
var exist = _context.CandidateInterfaces.Values.FirstOrDefault(x => x.TypeName == typeArgument.ToString());
if (exist is not null)
{
if (!differs.ContainsKey(typeArgument.ToString()))
{
differs.Add(typeArgument.ToString(), exist.InterfaceName);
}
type = type.Replace(typeArgument.ToString(), exist.InterfaceName);
}
}
return type;
}
return property.Type.ToString();
}
protected INamedTypeSymbol GetType(string name)
{
var symbol = _context.GeneratorExecutionContext.Compilation.GetTypeByMetadataName(name);
src/ProxyInterfaceSourceGenerator/FileGenerators/PartialInterfacesGenerator.cs
+13 -10
View File
@@ -94,16 +94,19 @@ namespace {symbol.ContainingNamespace}
//}
//else
var existing = _context.CandidateInterfaces.Values.FirstOrDefault(x => x.TypeName == property.Type.ToString());
if (existing is not null)
{
str.AppendLine($" {property.ToPropertyText(existing.InterfaceName)}");
}
else
{
str.AppendLine($" {property.ToPropertyText()}");
}
var type = GetPropertyType(property, out _);
str.AppendLine($" {property.ToPropertyText(type)}");
//var existing = _context.CandidateInterfaces.Values.FirstOrDefault(x => x.TypeName == property.Type.ToString());
//if (existing is not null)
//{
// str.AppendLine($" {property.ToPropertyText(existing.InterfaceName)}");
//}
//else
//{
// str.AppendLine($" {property.ToPropertyText()}");
//}
str.AppendLine();
}
src/ProxyInterfaceSourceGenerator/FileGenerators/ProxyClassesGenerator.cs
+63 -6
View File
@@ -43,16 +43,21 @@ namespace ProxyInterfaceSourceGenerator.FileGenerators
}
private string CreateProxyClassCode(INamedTypeSymbol symbol, string interfaceName, string className, bool proxyAll) => $@"using System;
using AutoMapper;
namespace {symbol.ContainingNamespace}
{{
public class {className}Proxy : {interfaceName}
{{
private readonly IMapper _mapper;
public {className} _Instance {{ get; }}
public {className}Proxy({className} instance)
{{
_Instance = instance;
{GenerateAutoMapper()}
}}
{GeneratePublicProperties(symbol, proxyAll)}
@@ -61,6 +66,22 @@ namespace {symbol.ContainingNamespace}
}}
}}";
private string GenerateAutoMapper()
{
var str = new StringBuilder();
str.AppendLine(" _mapper = new MapperConfiguration(cfg =>");
str.AppendLine(" {");
foreach (var x in _context.CandidateInterfaces)
{
str.AppendLine($" cfg.CreateMap<{x.Value.InterfaceName}, {x.Value.ClassName}>();");
str.AppendLine($" cfg.CreateMap<{x.Value.ClassName}, {x.Value.InterfaceName}>();");
}
str.AppendLine(" }).CreateMapper();");
return str.ToString();
}
private string GeneratePublicProperties(INamedTypeSymbol symbol, bool proxyAll)
{
var str = new StringBuilder();
@@ -82,15 +103,51 @@ namespace {symbol.ContainingNamespace}
// ComplexProperties
foreach (var property in MemberHelper.GetPublicProperties(symbol, p => p.GetTypeEnum() == TypeEnum.Complex))
{
var existing = _context.CandidateInterfaces.Values.FirstOrDefault(x => x.TypeName == property.Type.ToString());
if (existing is not null)
{
str.AppendLine($" public {property.ToPropertyTextForClass(existing.InterfaceName, existing.ClassName)}");
}
else
var type = GetPropertyType(property, out var differs);
if (!differs.Any())
{
str.AppendLine($" public {property.ToPropertyTextForClass()}");
}
else
{
str.AppendLine($" public {property.ToPropertyTextForClass(type)}");
// var get = property.GetMethod != null ? $"get => _mapper.Map<{type}>(_Instance.{property.Name}); " : string.Empty;
// var set = property.SetMethod != null ? $"set => _Instance.{property.Name} = _mapper.Map<{property.Type}>(value);" : string.Empty;
//var p = $"{type} {property.Name} {{ {get}{set}}}";
//str.AppendLine($" public {type} {property.Name} {{ {get}{set}}}");
}
/*
public IList<IClazz> Cs
{
get => _mapper.Map<IList<IClazz>>(_instance.Cs);
set => _instance.Cs = _mapper.Map<IList<Clazz>>(value);
}
}*/
//public static string ToPropertyTextForClass(this IPropertySymbol property, string overrideType)
//{
// // var classNameProxy = $"Proxy";
// var get = property.GetMethod != null ? $"get => _mapper.Map<{overrideType}>(_Instance.{property.Name}); " : string.Empty;
// var set = property.SetMethod != null ? $"set => _mapper.Map<{property.Type}>( = (({classNameProxy}) value)._Instance; " : string.Empty;
// return $"{overrideType} {property.Name} {{ {get}{set}}}";
//}
//str.AppendLine($" public {property.ToPropertyTextForClass(type)}");
//var existing = _context.CandidateInterfaces.Values.FirstOrDefault(x => x.TypeName == property.Type.ToString());
//if (existing is not null)
//{
// str.AppendLine($" public {property.ToPropertyTextForClass(existing.InterfaceName, existing.ClassName)}");
//}
//else
//{
// str.AppendLine($" public {property.ToPropertyTextForClass()}");
//}
str.AppendLine();
}
src/ProxyInterfaceSourceGenerator/ProxyInterfaceCodeGenerator.cs
+3 -70
View File
@@ -23,9 +23,6 @@ namespace ProxyInterfaceSourceGenerator
public void Execute(GeneratorExecutionContext ctx)
{
var attributeData = _proxyAttributeGenerator.GenerateFile();
ctx.AddSource(attributeData.FileName, SourceText.From(attributeData.Text, Encoding.UTF8));
if (ctx.SyntaxReceiver is not ProxySyntaxReceiver receiver)
{
return;
@@ -37,6 +34,9 @@ namespace ProxyInterfaceSourceGenerator
CandidateInterfaces = receiver.CandidateInterfaces
};
var attributeData = _proxyAttributeGenerator.GenerateFile();
context.GeneratorExecutionContext.AddSource(attributeData.FileName, SourceText.From(attributeData.Text, Encoding.UTF8));
var partialInterfacesGenerator = new PartialInterfacesGenerator(context);
foreach (var data in partialInterfacesGenerator.GenerateFiles())
{
@@ -50,71 +50,4 @@ namespace ProxyInterfaceSourceGenerator
}
}
}
public struct Test
{
public int Id { get; set; }
public Clazz C { get; }
public int Add(string s)
{
return 600;
}
}
public sealed class Clazz
{
public string Name { get; set; }
}
public interface ITest
{
int Id { get; set; }
IClazz C { get; }
int Add(string s);
}
public interface IClazz
{
string Name { get; set; }
}
public class TestMock : ITest
{
private Test _instance;
private IClazz _clazz;
public TestMock(Test instance)
{
_instance = instance;
_clazz = new ClazzMock(_instance.C);
}
public int Id
{
get => _instance.Id;
set => _instance.Id = value;
}
public IClazz C => _clazz;
public int Add(string s) => _instance.Add(s);
}
public class ClazzMock : IClazz
{
private Clazz _instance;
public ClazzMock(Clazz instance)
{
_instance = instance;
}
public string Name { get => _instance.Name; set => _instance.Name = value; }
}
}
src/ProxyInterfaceSourceGenerator/ProxyInterfaceSourceGenerator.csproj
+1
View File
@@ -31,6 +31,7 @@
</PropertyGroup>
<ItemGroup>
<PackageReference Include="AutoMapper" Version="10.1.1" />
<PackageReference Include="Microsoft.CodeAnalysis.Analyzers" Version="3.3.2">
<PrivateAssets>all</PrivateAssets>
<IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>