Move get valid plugin info to shared space

Author: Jack251970

Flow.Launcher.Localization.SourceGenerators/Localize/LocalizeSourceGenerator.cs

@@ -100,7 +100,9 @@ private void Execute(SourceProductionContext spc,
             var assemblyName = compilation.AssemblyName ?? Constants.DefaultNamespace;
             var useDI = configOptions.GetFLLUseDependencyInjection();
 
-            var pluginInfo = GetValidPluginInfo(pluginClasses, spc, useDI);
+            var pluginInfo = PluginInfoHelper.GetValidPluginInfoAndReportDiagnostic(pluginClasses, spc, useDI);
+
+            if (pluginInfo == null) return;
 
             GenerateSource(
                 spc,
@@ -421,88 +423,6 @@ private static string GetLocalizationKeyFromInvocation(GeneratorSyntaxContext co
 
         #endregion
 
-        #region Get Plugin Class Info
-
-        private static PluginClassInfo GetValidPluginInfo(
-            ImmutableArray<PluginClassInfo> pluginClasses,
-            SourceProductionContext context,
-            bool useDI)
-        {
-            // If p is null, this class does not implement IPluginI18n
-            var iPluginI18nClasses = pluginClasses.Where(p => p != null).ToArray();
-            if (iPluginI18nClasses.Length == 0)
-            {
-                context.ReportDiagnostic(Diagnostic.Create(
-                    SourceGeneratorDiagnostics.CouldNotFindPluginEntryClass,
-                    Location.None
-                ));
-                return null;
-            }
-
-            // If we use dependency injection, we do not need to check if there is a valid plugin context
-            // Also we do not need to return the plugin info
-            if (useDI)
-            {
-                return null;
-            }
-
-            // If p.PropertyName is null, this class does not have PluginInitContext property
-            var iPluginI18nClassesWithContext = iPluginI18nClasses.Where(p => p.PropertyName != null).ToArray();
-            if (iPluginI18nClassesWithContext.Length == 0)
-            {
-                foreach (var pluginClass in iPluginI18nClasses)
-                {
-                    context.ReportDiagnostic(Diagnostic.Create(
-                        SourceGeneratorDiagnostics.CouldNotFindContextProperty,
-                        pluginClass.Location,
-                        pluginClass.ClassName
-                    ));
-                }
-                return null;
-            }
-
-            // Rest classes have implemented IPluginI18n and have PluginInitContext property
-            // Check if the property is valid
-            foreach (var pluginClass in iPluginI18nClassesWithContext)
-            {
-                if (pluginClass.IsValid == true)
-                {
-                    return pluginClass;
-                }
-
-                if (!pluginClass.IsStatic)
-                {
-                    context.ReportDiagnostic(Diagnostic.Create(
-                        SourceGeneratorDiagnostics.ContextPropertyNotStatic,
-                        pluginClass.Location,
-                        pluginClass.PropertyName
-                    ));
-                }
-
-                if (pluginClass.IsPrivate)
-                {
-                    context.ReportDiagnostic(Diagnostic.Create(
-                        SourceGeneratorDiagnostics.ContextPropertyIsPrivate,
-                        pluginClass.Location,
-                        pluginClass.PropertyName
-                    ));
-                }
-
-                if (pluginClass.IsProtected)
-                {
-                    context.ReportDiagnostic(Diagnostic.Create(
-                        SourceGeneratorDiagnostics.ContextPropertyIsProtected,
-                        pluginClass.Location,
-                        pluginClass.PropertyName
-                    ));
-                }
-            }
-
-            return null;
-        }
-
-        #endregion
-
         #region Generate Source
 
         private static void GenerateSource(

Flow.Launcher.Localization.SourceGenerators/PluginInfoHelper.cs

@@ -0,0 +1,88 @@
+using System.Collections.Immutable;
+using System.Linq;
+using Flow.Launcher.Localization.Shared;
+using Microsoft.CodeAnalysis;
+
+namespace Flow.Launcher.Localization.SourceGenerators
+{
+    internal class PluginInfoHelper
+    {
+        public static PluginClassInfo GetValidPluginInfoAndReportDiagnostic(
+            ImmutableArray<PluginClassInfo> pluginClasses,
+            SourceProductionContext context,
+            bool useDI)
+        {
+            // If p is null, this class does not implement IPluginI18n
+            var iPluginI18nClasses = pluginClasses.Where(p => p != null).ToArray();
+            if (iPluginI18nClasses.Length == 0)
+            {
+                context.ReportDiagnostic(Diagnostic.Create(
+                    SourceGeneratorDiagnostics.CouldNotFindPluginEntryClass,
+                    Location.None
+                ));
+                return null;
+            }
+
+            // If we use dependency injection, we do not need to check if there is a valid plugin context
+            // Also we do not need to return the plugin info
+            if (useDI)
+            {
+                return null;
+            }
+
+            // If p.PropertyName is null, this class does not have PluginInitContext property
+            var iPluginI18nClassesWithContext = iPluginI18nClasses.Where(p => p.PropertyName != null).ToArray();
+            if (iPluginI18nClassesWithContext.Length == 0)
+            {
+                foreach (var pluginClass in iPluginI18nClasses)
+                {
+                    context.ReportDiagnostic(Diagnostic.Create(
+                        SourceGeneratorDiagnostics.CouldNotFindContextProperty,
+                        pluginClass.Location,
+                        pluginClass.ClassName
+                    ));
+                }
+                return null;
+            }
+
+            // Rest classes have implemented IPluginI18n and have PluginInitContext property
+            // Check if the property is valid
+            foreach (var pluginClass in iPluginI18nClassesWithContext)
+            {
+                if (pluginClass.IsValid == true)
+                {
+                    return pluginClass;
+                }
+
+                if (!pluginClass.IsStatic)
+                {
+                    context.ReportDiagnostic(Diagnostic.Create(
+                        SourceGeneratorDiagnostics.ContextPropertyNotStatic,
+                        pluginClass.Location,
+                        pluginClass.PropertyName
+                    ));
+                }
+
+                if (pluginClass.IsPrivate)
+                {
+                    context.ReportDiagnostic(Diagnostic.Create(
+                        SourceGeneratorDiagnostics.ContextPropertyIsPrivate,
+                        pluginClass.Location,
+                        pluginClass.PropertyName
+                    ));
+                }
+
+                if (pluginClass.IsProtected)
+                {
+                    context.ReportDiagnostic(Diagnostic.Create(
+                        SourceGeneratorDiagnostics.ContextPropertyIsProtected,
+                        pluginClass.Location,
+                        pluginClass.PropertyName
+                    ));
+                }
+            }
+
+            return null;
+        }
+    }
+}