ProxyBaseClasses (#27)

2022-02-01 18:49:01 +01:00
parent 649ed89bb6
commit f9664e0564
45 changed files with 1305 additions and 836 deletions
@@ -1,47 +1,47 @@
-using System.Collections.Generic;
-using System.Linq;
 using System.Text;
-using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp.Syntax;
 using ProxyInterfaceSourceGenerator.Enums;
 using ProxyInterfaceSourceGenerator.Extensions;
+using ProxyInterfaceSourceGenerator.Model;
 using ProxyInterfaceSourceGenerator.SyntaxReceiver;
 using ProxyInterfaceSourceGenerator.Utils;

-namespace ProxyInterfaceSourceGenerator.FileGenerators
+namespace ProxyInterfaceSourceGenerator.FileGenerators;
+
+internal class PartialInterfacesGenerator : BaseGenerator, IFilesGenerator
 {
-    internal class PartialInterfacesGenerator : BaseGenerator, IFilesGenerator
+    public PartialInterfacesGenerator(Context context, bool supportsNullable) :
+        base(context, supportsNullable)
    {
-        public PartialInterfacesGenerator(Context context, bool supportsNullable) :
-            base(context, supportsNullable)
+    }
+
+    public IEnumerable<FileData> GenerateFiles()
+    {
+        foreach (var ci in Context.CandidateInterfaces)
        {
+            yield return GenerateFile(ci.Key, ci.Value);
        }
+    }

-        public IEnumerable<FileData> GenerateFiles()
-        {
-            foreach (var ci in _context.CandidateInterfaces)
-            {
-                yield return GenerateFile(ci.Key, ci.Value);
-            }
-        }
+    private FileData GenerateFile(InterfaceDeclarationSyntax ci, ProxyData pd)
+    {
+        var sourceInterfaceSymbol = GetNamedTypeSymbolByFullName(ci.Identifier.ToString(), pd.Usings);
+        var targetClassSymbol = GetNamedTypeSymbolByFullName(pd.TypeName, pd.Usings);
+        var interfaceName = targetClassSymbol.Symbol.ResolveInterfaceNameWithOptionalTypeConstraints(pd.InterfaceName);

-        private FileData GenerateFile(InterfaceDeclarationSyntax ci, ProxyData pd)
-        {
-            var sourceInterfaceSymbol = GetNamedTypeSymbolByFullName(ci.Identifier.ToString(), pd.Usings);
-            var targetClassSymbol = GetNamedTypeSymbolByFullName(pd.TypeName, pd.Usings);
-            var interfaceName = targetClassSymbol.ResolveInterfaceNameWithOptionalTypeConstraints(pd.InterfaceName);
+        var file = new FileData(
+            $"{sourceInterfaceSymbol.Symbol.GetFileName()}.g.cs",
+            CreatePartialInterfaceCode(pd.Namespace, targetClassSymbol, interfaceName, pd.ProxyBaseClasses)
+        );

-            var file = new FileData(
-                $"{sourceInterfaceSymbol.GetFileName()}.g.cs",
-                CreatePartialInterfaceCode(pd.Namespace, targetClassSymbol, interfaceName, pd.ProxyAll)
-            );
+        return file;
+    }

-            // _context.GeneratedData.Add(new() { InterfaceName = interfaceName, ClassName = null, FileData = file });
-
-            return file;
-        }
-
-        private string CreatePartialInterfaceCode(string ns, INamedTypeSymbol targetClassSymbol, string interfaceName, bool proxyAll) => $@"//----------------------------------------------------------------------------------------
+    private string CreatePartialInterfaceCode(
+        string ns,
+        ClassSymbol classSymbol,
+        string interfaceName,
+        bool proxyBaseClasses) => $@"//----------------------------------------------------------------------------------------
 // <auto-generated>
 //     This code was generated by https://github.com/StefH/ProxyInterfaceSourceGenerator.
 //
@@ -50,74 +50,73 @@ namespace ProxyInterfaceSourceGenerator.FileGenerators
 // </auto-generated>
 //----------------------------------------------------------------------------------------

-{(_supportsNullable ? "#nullable enable" : string.Empty)}
+{(SupportsNullable ? "#nullable enable" : string.Empty)}
 using System;

 namespace {ns}
 {{
    public partial interface {interfaceName}
    {{
-{GenerateProperties(targetClassSymbol, proxyAll)}
+{GenerateProperties(classSymbol, proxyBaseClasses)}

-{GenerateMethods(targetClassSymbol)}
+{GenerateMethods(classSymbol, proxyBaseClasses)}

-{GenerateEvents(targetClassSymbol)}
+{GenerateEvents(classSymbol, proxyBaseClasses)}
    }}
 }}
-{(_supportsNullable ? "#nullable disable" : string.Empty)}";
+{(SupportsNullable ? "#nullable disable" : string.Empty)}";

-        private string GenerateProperties(INamedTypeSymbol targetClassSymbol, bool proxyAll)
+    private string GenerateProperties(ClassSymbol targetClassSymbol, bool proxyBaseClasses)
+    {
+        var str = new StringBuilder();
+
+        foreach (var property in MemberHelper.GetPublicProperties(targetClassSymbol, proxyBaseClasses))
        {
-            var str = new StringBuilder();
-
-            foreach (var property in MemberHelper.GetPublicProperties(targetClassSymbol))
+            var type = GetPropertyType(property, out var isReplaced);
+            if (isReplaced)
            {
-                var type = GetPropertyType(property, out var isReplaced);
-                if (isReplaced)
-                {
-                    str.AppendLine($"        {property.ToPropertyText(type)}");
-                }
-                else
-                {
-                    str.AppendLine($"        {property.ToPropertyText()}");
-                }
-                str.AppendLine();
+                str.AppendLine($"        {property.ToPropertyText(type)}");
            }
-
-            return str.ToString();
+            else
+            {
+                str.AppendLine($"        {property.ToPropertyText()}");
+            }
+            str.AppendLine();
        }

-        private string GenerateMethods(INamedTypeSymbol targetClassSymbol)
+        return str.ToString();
+    }
+
+    private string GenerateMethods(ClassSymbol targetClassSymbol, bool proxyBaseClasses)
+    {
+        var str = new StringBuilder();
+        foreach (var method in MemberHelper.GetPublicMethods(targetClassSymbol, proxyBaseClasses))
        {
-            var str = new StringBuilder();
-            foreach (var method in MemberHelper.GetPublicMethods(targetClassSymbol))
+            var methodParameters = new List<string>();
+            foreach (var ps in method.Parameters)
            {
-                var methodParameters = new List<string>();
-                foreach (var ps in method.Parameters)
-                {
-                    var type = ps.GetTypeEnum() == TypeEnum.Complex ? GetParameterType(ps, out _) : ps.Type.ToString();
-                    methodParameters.Add($"{ps.GetParamsPrefix()}{ps.GetRefPrefix()}{type} {ps.GetSanitizedName()}{ps.GetDefaultValue()}");
-                }
-
-                str.AppendLine($"        {GetReplacedType(method.ReturnType, out _)} {method.GetMethodNameWithOptionalTypeParameters()}({string.Join(", ", methodParameters)}){method.GetWhereStatement()};");
-                str.AppendLine();
-            }
-
-            return str.ToString();
-        }
-
-        private string GenerateEvents(INamedTypeSymbol targetClassSymbol)
-        {
-            var str = new StringBuilder();
-            foreach (var @event in MemberHelper.GetPublicEvents(targetClassSymbol))
-            {
-                var ps = @event.First().Parameters.First();
                var type = ps.GetTypeEnum() == TypeEnum.Complex ? GetParameterType(ps, out _) : ps.Type.ToString();
-                str.AppendLine($"        event {type} {@event.Key.GetSanitizedName()};");
-                str.AppendLine();
+                methodParameters.Add($"{ps.GetParamsPrefix()}{ps.GetRefPrefix()}{type} {ps.GetSanitizedName()}{ps.GetDefaultValue()}");
            }

-            return str.ToString();
+            str.AppendLine($"        {GetReplacedType(method.ReturnType, out _)} {method.GetMethodNameWithOptionalTypeParameters()}({string.Join(", ", methodParameters)}){method.GetWhereStatement()};");
+            str.AppendLine();
        }
+
+        return str.ToString();
+    }
+
+    private string GenerateEvents(ClassSymbol targetClassSymbol, bool proxyBaseClasses)
+    {
+        var str = new StringBuilder();
+        foreach (var @event in MemberHelper.GetPublicEvents(targetClassSymbol, proxyBaseClasses))
+        {
+            var ps = @event.First().Parameters.First();
+            var type = ps.GetTypeEnum() == TypeEnum.Complex ? GetParameterType(ps, out _) : ps.Type.ToString();
+            str.AppendLine($"        event {type} {@event.Key.GetSanitizedName()};");
+            str.AppendLine();
+        }
+
+        return str.ToString();
    }
 }