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

ProxyAll for interface part 1...

Browse Source
This commit is contained in:
Stef Heyenrath
2021-07-24 10:05:54 +02:00
parent afe1710816
commit 97fddc2f3f
9 changed files with 127 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src-examples/ProxyInterfaceConsumer/IPerson.cs
+1 -1
View File
@@ -1,6 +1,6 @@
namespace SourceGeneratorInterface
{
[ProxyInterfaceGenerator.Proxy(typeof(SourceGeneratorInterface.Person), false)]
[ProxyInterfaceGenerator.Proxy(typeof(SourceGeneratorInterface.Person), true)]
public partial interface IPerson
{
}
src-examples/ProxyInterfaceConsumer/Program.cs
+2 -2
View File
@@ -8,8 +8,8 @@ namespace SourceGeneratorInterface
{
IPerson p = new PersonProxy(new Person());
p.Name = "test";
p.MyNamedTypeSymbol = null;
p.Compilation = null;
//p.MyNamedTypeSymbol = null;
//p.Compilation = null;
//p.Add("x");
//p.Void();
Console.WriteLine(System.Text.Json.JsonSerializer.Serialize(p));
src/ProxyInterfaceSourceGenerator/Context.cs
+6 -2
View File
@@ -1,8 +1,12 @@
using Microsoft.CodeAnalysis;
using System.Collections.Generic;
using Microsoft.CodeAnalysis;
namespace ProxyInterfaceSourceGenerator
{
internal record Context(GeneratorExecutionContext GeneratorExecutionContext)
internal record Context
{
public GeneratorExecutionContext GeneratorExecutionContext { get; init; }
public List<ContextData> GeneratedData { get; } = new List<ContextData>();
}
}
src/ProxyInterfaceSourceGenerator/ContextData.cs
+13
View File
@@ -0,0 +1,13 @@
using ProxyInterfaceSourceGenerator.FileGenerators;
namespace ProxyInterfaceSourceGenerator
{
internal record ContextData
{
public string? InterfaceName { get; init; }
public string? ClassName { get; init; }
public FileData FileData { get; init; }
}
}
src/ProxyInterfaceSourceGenerator/Extensions/SymbolExtensions.cs
+8 -6
View File
@@ -5,18 +5,20 @@ namespace ProxyInterfaceSourceGenerator.Extensions
{
internal static class SymbolExtensions
{
public static string ToPropertyText(this IPropertySymbol property)
public static string ToPropertyText(this IPropertySymbol property, string? overrideType = null)
{
string get = property.GetMethod != null ? "get; " : string.Empty;
string set = property.SetMethod != null ? "set; " : string.Empty;
var get = property.GetMethod != null ? "get; " : string.Empty;
var set = property.SetMethod != null ? "set; " : string.Empty;
return $"{property.Type} {property.Name} {{ {get}{set}}}";
var type = !string.IsNullOrEmpty(overrideType) ? overrideType : $"{property.Type}";
return $"{type} {property.Name} {{ {get}{set}}}";
}
public static string ToPropertyTextForClass(this IPropertySymbol property)
{
string get = property.GetMethod != null ? $"get => _instance.{property.Name}; " : string.Empty;
string set = property.SetMethod != null ? $"set => _instance.{property.Name} = value; " : string.Empty;
var get = property.GetMethod != null ? $"get => _instance.{property.Name}; " : string.Empty;
var set = property.SetMethod != null ? $"set => _instance.{property.Name} = value; " : string.Empty;
return $"{property.Type} {property.Name} {{ {get}{set}}}";
}
src/ProxyInterfaceSourceGenerator/FileGenerators/BaseGenerator.cs
+31
View File
@@ -0,0 +1,31 @@
using System;
using System.Collections.Generic;
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp.Syntax;
using ProxyInterfaceSourceGenerator.SyntaxReceiver;
namespace ProxyInterfaceSourceGenerator.FileGenerators
{
internal abstract class BaseGenerator
{
protected readonly Context _context;
protected readonly IDictionary<InterfaceDeclarationSyntax, ProxyData> _candidateInterfaces;
public BaseGenerator(Context context, IDictionary<InterfaceDeclarationSyntax, ProxyData> candidateInterfaces)
{
_context = context;
_candidateInterfaces = candidateInterfaces;
}
protected INamedTypeSymbol GetType(string name)
{
var symbol = _context.GeneratorExecutionContext.Compilation.GetTypeByMetadataName(name);
if (symbol is null)
{
throw new Exception($"The type '{name}' is not found.");
}
return symbol;
}
}
}
src/ProxyInterfaceSourceGenerator/FileGenerators/PartialInterfacesGenerator.cs
+43 -22
View File
@@ -10,32 +10,38 @@ using ProxyInterfaceSourceGenerator.Utils;
namespace ProxyInterfaceSourceGenerator.FileGenerators
{
internal class PartialInterfacesGenerator : IFilesGenerator
internal class PartialInterfacesGenerator : BaseGenerator, IFilesGenerator
{
private readonly Context _context;
private readonly IDictionary<InterfaceDeclarationSyntax, ProxyData> _candidateInterfaces;
private readonly List<FileData> files = new List<FileData>();
public PartialInterfacesGenerator(Context context, IDictionary<InterfaceDeclarationSyntax, ProxyData> candidateInterfaces)
public PartialInterfacesGenerator(Context context, IDictionary<InterfaceDeclarationSyntax, ProxyData> candidateInterfaces) :
base(context, candidateInterfaces)
{
_context = context;
_candidateInterfaces = candidateInterfaces;
}
public IEnumerable<FileData> GenerateFiles()
{
foreach (var ci in _candidateInterfaces)
{
var symbol = _context.GeneratorExecutionContext.Compilation.GetTypeByMetadataName(ci.Value.TypeName);
if (symbol is null)
{
throw new Exception($"The type '{ci.Value.TypeName}' is not found.");
}
yield return new FileData(
$"{ci.Value.InterfaceName}.cs",
CreatePartialInterfaceCode(symbol, ci.Value.InterfaceName, ci.Value.ProxyAll)
);
var file = GenerateFile(ci.Value.InterfaceName, ci.Value.TypeName, ci.Value.ProxyAll);
files.Add(file);
}
return files;
}
private FileData GenerateFile(string interfaceName, string typeName, bool proxyAll)
{
var symbol = GetType(typeName);
var file = new FileData(
$"{interfaceName}.cs",
CreatePartialInterfaceCode(symbol, interfaceName, proxyAll)
);
_context.GeneratedData.Add(new() { InterfaceName = interfaceName, ClassName = null, FileData = file });
return file;
}
private string CreatePartialInterfaceCode(INamedTypeSymbol symbol, string interfaceName, bool proxyAll) => $@"using System;
@@ -78,14 +84,29 @@ namespace {symbol.ContainingNamespace}
p => p.Type.TypeKind != TypeKind.Interface
};
if (proxyAll)
{
}
foreach (var property in MemberHelper.GetPublicProperties(symbol, complexFilters.ToArray()))
{
str.AppendLine($" {property.ToPropertyText()}");
if (proxyAll)
{
var existing = _context.GeneratedData
.FirstOrDefault(x => x.ClassName == $"{property.Name}Proxy" || x.InterfaceName == $"I{property.Name}");
if (existing is not null)
{
str.AppendLine($" {property.ToPropertyText(existing.InterfaceName)}");
}
else
{
// Create new
var typeName = $"{property.Type}";
var file = GenerateFile($"I{property.Name}", typeName, false);
str.AppendLine($" // {property.ToPropertyText($"I{property.Name}")}");
}
}
else
{
str.AppendLine($" {property.ToPropertyText()}");
}
str.AppendLine();
}
src/ProxyInterfaceSourceGenerator/FileGenerators/ProxyClassesGenerator.cs
+15 -18
View File
@@ -9,26 +9,18 @@ using ProxyInterfaceSourceGenerator.Utils;
namespace ProxyInterfaceSourceGenerator.FileGenerators
{
internal class ProxyClassesGenerator : IFilesGenerator
internal class ProxyClassesGenerator : BaseGenerator, IFilesGenerator
{
private readonly Context _context;
private readonly IDictionary<InterfaceDeclarationSyntax, ProxyData> _candidateInterfaces;
public ProxyClassesGenerator(Context context, IDictionary<InterfaceDeclarationSyntax, ProxyData> candidateInterfaces)
public ProxyClassesGenerator(Context context, IDictionary<InterfaceDeclarationSyntax, ProxyData> candidateInterfaces) :
base(context, candidateInterfaces)
{
_context = context;
_candidateInterfaces = candidateInterfaces;
}
public IEnumerable<FileData> GenerateFiles()
{
foreach (var ci in _candidateInterfaces)
{
var symbol = _context.GeneratorExecutionContext.Compilation.GetTypeByMetadataName(ci.Value.TypeName);
if (symbol is null)
{
throw new Exception($"The type '{ci.Value.TypeName}' is not found.");
}
var symbol = GetType(ci.Value.TypeName);
yield return new FileData(
$"{ci.Value.ClassName}Proxy.cs",
@@ -44,20 +36,25 @@ namespace {symbol.ContainingNamespace}
public class {className}Proxy : {interfaceName}
{{
private {className} _instance;
{GenerateComplexFields(symbol,proxyAll)}
{GeneratePrivateComplexInterfaceFields(symbol,proxyAll)}
public {className}Proxy({className} instance)
{{
_instance = instance;
}}
{GenerateProperties(symbol, proxyAll)}
{GeneratePublicProperties(symbol, proxyAll)}
{GenerateMethods(symbol)}
{GeneratePublicMethods(symbol)}
}}
}}";
private string GenerateComplexFields(INamedTypeSymbol symbol, bool proxyAll)
private string GeneratePrivateComplexInterfaceFields(INamedTypeSymbol symbol, bool proxyAll)
{
if (!proxyAll)
{
return string.Empty;
}
var str = new StringBuilder();
foreach (var property in GetComplexProperties(symbol, proxyAll))
@@ -68,7 +65,7 @@ namespace {symbol.ContainingNamespace}
return str.ToString();
}
private string GenerateProperties(INamedTypeSymbol symbol, bool proxyAll)
private string GeneratePublicProperties(INamedTypeSymbol symbol, bool proxyAll)
{
var str = new StringBuilder();
@@ -115,7 +112,7 @@ namespace {symbol.ContainingNamespace}
return MemberHelper.GetPublicProperties(symbol, complexFilters.ToArray());
}
private string GenerateMethods(INamedTypeSymbol symbol)
private string GeneratePublicMethods(INamedTypeSymbol symbol)
{
var str = new StringBuilder();
foreach (var method in MemberHelper.GetPublicMethods(symbol))
src/ProxyInterfaceSourceGenerator/ProxyInterfaceCodeGenerator.cs
+8 -5
View File
@@ -13,17 +13,20 @@ namespace ProxyInterfaceSourceGenerator
public void Initialize(GeneratorInitializationContext context)
{
//if (!System.Diagnostics.Debugger.IsAttached)
//{
// System.Diagnostics.Debugger.Launch();
//}
if (!System.Diagnostics.Debugger.IsAttached)
{
System.Diagnostics.Debugger.Launch();
}
context.RegisterForSyntaxNotifications(() => new ProxySyntaxReceiver());
}
public void Execute(GeneratorExecutionContext ctx)
{
var context = new Context(ctx);
var context = new Context
{
GeneratorExecutionContext = ctx
};
var attributeData = _proxyAttributeGenerator.GenerateFile();
context.GeneratorExecutionContext.AddSource(attributeData.FileName, SourceText.From(attributeData.Text, Encoding.UTF8));