initial code

2021-07-23 16:28:11 +02:00
parent af0a3e0a5d
commit 7c68455a8e
15 changed files with 527 additions and 0 deletions
@@ -0,0 +1,9 @@
+namespace ProxyInterfaceSourceGenerator.FileGenerators
+{
+    internal record Data
+    {
+        public string FileName { get; set; }
+
+        public string Text { get; set; }
+    }
+}
@@ -0,0 +1,7 @@
+namespace ProxyInterfaceSourceGenerator.FileGenerators
+{
+    internal interface IFileGenerator
+    {
+        Data GenerateFile();
+    }
+}
@@ -0,0 +1,9 @@
+using System.Collections.Generic;
+
+namespace ProxyInterfaceSourceGenerator.FileGenerators
+{
+    internal interface IFilesGenerator
+    {
+        IEnumerable<Data> GenerateFiles();
+    }
+}
@@ -0,0 +1,72 @@
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.CSharp.Syntax;
+using ProxyInterfaceSourceGenerator.Extensions;
+using ProxyInterfaceSourceGenerator.FileGenerators;
+using ProxyInterfaceSourceGenerator.Utils;
+
+namespace ClassLibrarySourceGen
+{
+    internal class PartialInterfacesGenerator : IFilesGenerator
+    {
+        private readonly GeneratorExecutionContext _context;
+        private readonly IDictionary<InterfaceDeclarationSyntax, string> _candidateInterfaces;
+
+        public PartialInterfacesGenerator(GeneratorExecutionContext context, IDictionary<InterfaceDeclarationSyntax, string> candidateInterfaces)
+        {
+            _context = context;
+            _candidateInterfaces = candidateInterfaces;
+        }
+
+        public IEnumerable<Data> GenerateFiles()
+        {
+            foreach (var x in _candidateInterfaces)
+            {
+                string interfaceName = $"I{x.Value.Split('.').Last()}";
+                yield return new Data
+                {
+                    FileName = $"I{interfaceName}.cs",
+                    Text = CreatePartialInterfaceCode(_context.Compilation.GetTypeByMetadataName(x.Value), interfaceName)
+                };
+            }
+        }
+
+        private string CreatePartialInterfaceCode(INamedTypeSymbol symbol, string interfaceName) => $@"using System;
+
+namespace {symbol.ContainingNamespace}
+{{
+    public partial interface {interfaceName}
+    {{
+{GenerateSimpleProperties(symbol)}
+
+{GenerateMethods(symbol)}
+    }}
+}}";
+
+        private string GenerateSimpleProperties(INamedTypeSymbol symbol)
+        {
+            var str = new StringBuilder();
+            foreach (IPropertySymbol property in MemberHelper.GetPublicProperties(symbol, p => p.Type.IsValueType || p.Type.ToString() == "string"))
+            {
+                str.AppendLine($"        {property.ToCode()}");
+                str.AppendLine();
+            }
+
+            return str.ToString();
+        }
+
+        private string GenerateMethods(INamedTypeSymbol symbol)
+        {
+            var str = new StringBuilder();
+            foreach (IMethodSymbol method in MemberHelper.GetPublicMethods(symbol))
+            {
+                str.AppendLine($"        {method.ToCode()}");
+                str.AppendLine();
+            }
+
+            return str.ToString();
+        }
+    }
+}
@@ -0,0 +1,30 @@
+namespace ProxyInterfaceSourceGenerator.FileGenerators
+{
+    internal class ProxyAttributeGenerator : IFileGenerator
+    {
+        private const string ClassName = "ProxyAttribute";
+
+        public Data GenerateFile()
+        {
+            return new Data
+            {
+                FileName = $"{ClassName}.cs",
+                Text = $@"using System;
+
+namespace ProxyInterfaceGenerator
+{{
+    [AttributeUsage(AttributeTargets.Interface)]
+    public class {ClassName} : Attribute
+    {{
+        public Type Type {{ get; }}
+
+        public {ClassName}(Type type)
+        {{
+            Type = type;
+        }}
+    }}
+}}"
+            };
+        }
+    }
+}