refactor some code

2021-07-26 15:09:07 +00:00
parent d7cb238cb9
commit 5d48b67ba3
2 changed files with 77 additions and 173 deletions
@@ -1,63 +1,78 @@
-using System.Text;
-using Microsoft.CodeAnalysis;
-using Microsoft.CodeAnalysis.Text;
-using ProxyInterfaceSourceGenerator.FileGenerators;
-using ProxyInterfaceSourceGenerator.SyntaxReceiver;
-
-namespace ProxyInterfaceSourceGenerator
-{
-    [Generator]
-    public class ProxyInterfaceCodeGenerator : ISourceGenerator
-    {
-        private readonly ProxyAttributeGenerator _proxyAttributeGenerator = new ProxyAttributeGenerator();
-
-        public void Initialize(GeneratorInitializationContext context)
-        {
-            //if (!System.Diagnostics.Debugger.IsAttached)
-            //{
-            //    System.Diagnostics.Debugger.Launch();
-            //}
-
-            context.RegisterForSyntaxNotifications(() => new ProxySyntaxReceiver());
-        }
-
-        public void Execute(GeneratorExecutionContext ctx)
-        {
-            if (ctx.SyntaxReceiver is not ProxySyntaxReceiver receiver)
-            {
-                return;
-            }
-
-            var context1 = new Context
-            {
-                GeneratorExecutionContext = ctx,
-                CandidateInterfaces = receiver.CandidateInterfaces
-            };
-
-            var attributeData = _proxyAttributeGenerator.GenerateFile();
-            context1.GeneratorExecutionContext.AddSource(attributeData.FileName, SourceText.From(attributeData.Text, Encoding.UTF8));
-
-            var context2 = new Context
-            {
-                GeneratorExecutionContext = ctx,
-                CandidateInterfaces = receiver.CandidateInterfaces
-            };
-            var partialInterfacesGenerator = new PartialInterfacesGenerator(context2);
-            foreach (var data in partialInterfacesGenerator.GenerateFiles())
-            {
-                context2.GeneratorExecutionContext.AddSource(data.FileName, SourceText.From(data.Text, Encoding.UTF8));
-            }
-
-            var context3 = new Context
-            {
-                GeneratorExecutionContext = ctx,
-                CandidateInterfaces = receiver.CandidateInterfaces
-            };
-            var proxyClassesGenerator = new ProxyClassesGenerator(context3);
-            foreach (var data in proxyClassesGenerator.GenerateFiles())
-            {
-                context3.GeneratorExecutionContext.AddSource(data.FileName, SourceText.From(data.Text, Encoding.UTF8));
-            }
-        }
-    }
+using System.Text;
+using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.Text;
+using ProxyInterfaceSourceGenerator.FileGenerators;
+using ProxyInterfaceSourceGenerator.SyntaxReceiver;
+
+namespace ProxyInterfaceSourceGenerator
+{
+    [Generator]
+    public class ProxyInterfaceCodeGenerator : ISourceGenerator
+    {
+        private readonly ProxyAttributeGenerator _proxyAttributeGenerator = new ProxyAttributeGenerator();
+
+        public void Initialize(GeneratorInitializationContext context)
+        {
+            //if (!System.Diagnostics.Debugger.IsAttached)
+            //{
+            //    System.Diagnostics.Debugger.Launch();
+            //}
+
+            context.RegisterForSyntaxNotifications(() => new ProxySyntaxReceiver());
+        }
+
+        public void Execute(GeneratorExecutionContext context)
+        {
+            if (context.SyntaxReceiver is not ProxySyntaxReceiver receiver)
+            {
+                return;
+            }
+
+            GenerateProxyAttribute(context, receiver);
+            GeneratePartialInterfaces(context, receiver);
+            GenerateProxyClasses(context, receiver);
+        }
+
+        private void GenerateProxyAttribute(GeneratorExecutionContext ctx, ProxySyntaxReceiver receiver)
+        {
+            var context = new Context
+            {
+                GeneratorExecutionContext = ctx,
+                CandidateInterfaces = receiver.CandidateInterfaces
+            };
+
+            var attributeData = _proxyAttributeGenerator.GenerateFile();
+            context.GeneratorExecutionContext.AddSource(attributeData.FileName, SourceText.From(attributeData.Text, Encoding.UTF8));
+        }
+
+        private static void GeneratePartialInterfaces(GeneratorExecutionContext ctx, ProxySyntaxReceiver receiver)
+        {
+            var context = new Context
+            {
+                GeneratorExecutionContext = ctx,
+                CandidateInterfaces = receiver.CandidateInterfaces
+            };
+
+            var partialInterfacesGenerator = new PartialInterfacesGenerator(context);
+            foreach (var data in partialInterfacesGenerator.GenerateFiles())
+            {
+                context.GeneratorExecutionContext.AddSource(data.FileName, SourceText.From(data.Text, Encoding.UTF8));
+            }
+        }
+
+        private static void GenerateProxyClasses(GeneratorExecutionContext ctx, ProxySyntaxReceiver receiver)
+        {
+            var context = new Context
+            {
+                GeneratorExecutionContext = ctx,
+                CandidateInterfaces = receiver.CandidateInterfaces
+            };
+
+            var proxyClassesGenerator = new ProxyClassesGenerator(context);
+            foreach (var data in proxyClassesGenerator.GenerateFiles())
+            {
+                context.GeneratorExecutionContext.AddSource(data.FileName, SourceText.From(data.Text, Encoding.UTF8));
+            }
+        }
+    }
 }