Fix useDI issue

Author: Jack251970

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

@@ -72,9 +72,22 @@ private void Execute(SourceProductionContext spc,
             var assemblyNamespace = compilation.AssemblyName ?? Constants.DefaultNamespace;
             var useDI = configOptions.GetFLLUseDependencyInjection();
 
-            var pluginInfo = PluginInfoHelper.GetValidPluginInfoAndReportDiagnostic(pluginClasses, spc, useDI);
-
-            if (pluginInfo == null) return;
+            PluginClassInfo pluginInfo;
+            if (useDI)
+            {
+                // 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
+                pluginInfo = null;
+            }
+            else
+            {
+                pluginInfo = PluginInfoHelper.GetValidPluginInfoAndReportDiagnostic(pluginClasses, spc);
+                if (pluginInfo == null)
+                {
+                    // If we cannot find a valid plugin info, we do not need to generate the source
+                    return;
+                }
+            }
 
             foreach (var enumDeclaration in enumsDeclarations.Distinct())
             {

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

@@ -100,9 +100,22 @@ private void Execute(SourceProductionContext spc,
             var assemblyNamespace = compilation.AssemblyName ?? Constants.DefaultNamespace;
             var useDI = configOptions.GetFLLUseDependencyInjection();
 
-            var pluginInfo = PluginInfoHelper.GetValidPluginInfoAndReportDiagnostic(pluginClasses, spc, useDI);
-
-            if (pluginInfo == null) return;
+            PluginClassInfo pluginInfo;
+            if (useDI)
+            {
+                // 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
+                pluginInfo = null;
+            }
+            else
+            {
+                pluginInfo = PluginInfoHelper.GetValidPluginInfoAndReportDiagnostic(pluginClasses, spc);
+                if (pluginInfo == null)
+                {
+                    // If we cannot find a valid plugin info, we do not need to generate the source
+                    return;
+                }
+            }
 
             GenerateSource(
                 spc,

Flow.Launcher.Localization.SourceGenerators/PluginInfoHelper.cs

@@ -9,8 +9,7 @@ internal class PluginInfoHelper
     {
         public static PluginClassInfo GetValidPluginInfoAndReportDiagnostic(
             ImmutableArray<PluginClassInfo> pluginClasses,
-            SourceProductionContext context,
-            bool useDI)
+            SourceProductionContext context)
         {
             // If p is null, this class does not implement IPluginI18n
             var iPluginI18nClasses = pluginClasses.Where(p => p != null).ToArray();
@@ -23,13 +22,6 @@ public static PluginClassInfo GetValidPluginInfoAndReportDiagnostic(
                 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)