- don't generate multiple files, fix distinction of typeSymbols

LightweightIocContainer.FactoryGenerator/FactoryGenerator.cs

@@ -101,13 +101,8 @@ public class FactoryGenerator : IIncrementalGenerator
     
     private void GenerateFactory(SourceProductionContext context, ImmutableArray<ITypeSymbol?> types)
     {
-        foreach (ISymbol? symbol in types.Distinct(SymbolEqualityComparer.IncludeNullability))
+        foreach (ITypeSymbol typeSymbol in GetDistinctTypes(types)) 
-        {
+            context.AddSource($"Generated{typeSymbol.Name}.g.cs", GenerateFactorySourceCode(typeSymbol));
-            if (symbol is not ITypeSymbol typeSymbol)
-                continue;
-        
-            context.AddSource($"Generated{GetGenericFileName(typeSymbol)}.g.cs", GenerateFactorySourceCode(typeSymbol));
-        }
     }
 
     private string GenerateBuilderClassSourceCode(string? classNamespace, ImmutableArray<ITypeSymbol?> types)
@@ -135,11 +130,8 @@ public class FactoryGenerator : IIncrementalGenerator
         stringBuilder.AppendLine($"{INDENT}public TFactory Create<TFactory>(IocContainer container)");
         stringBuilder.AppendLine($"{INDENT}{{");
 
-        foreach (ISymbol? symbol in types.Distinct(SymbolEqualityComparer.IncludeNullability))
+        foreach (ITypeSymbol type in GetDistinctTypes(types))
         {
-            if (symbol is not ITypeSymbol type)
-                continue;
-            
             stringBuilder.AppendLine($"{INDENT}{INDENT}if (typeof(TFactory) == typeof({GetTypeText(type, false)}))");
             stringBuilder.AppendLine($"{INDENT}{INDENT}{{");
             stringBuilder.AppendLine($"{INDENT}{INDENT}{INDENT}return (TFactory) (object) new Generated{GetTypeText(type, false)}(container);");
@@ -263,15 +255,24 @@ public class FactoryGenerator : IIncrementalGenerator
         return stringBuilder.ToString();
     }
     
-    private string GetGenericFileName(ITypeSymbol typeSymbol)
+    private IEnumerable<ITypeSymbol> GetDistinctTypes(ImmutableArray<ITypeSymbol?> types)
     {
-        StringBuilder stringBuilder = new();
+        List<ITypeSymbol> distinctTypes = [];
-        stringBuilder.Append(typeSymbol.Name);
+        foreach (ITypeSymbol? typeSymbol in types)
+        {
+            if (typeSymbol is null)
+                continue;
 
-        if (typeSymbol is INamedTypeSymbol { IsGenericType: true } namedTypeSymbol)
+            if (distinctTypes.Contains(typeSymbol, SymbolEqualityComparer.IncludeNullability))
-            stringBuilder.Append(string.Join("", namedTypeSymbol.TypeArguments.Select(t => CapitalizeFirstLetter(t.Name))));
+                continue;
             
-        return stringBuilder.ToString();
+            if (distinctTypes.Any(t => t.Name == typeSymbol.Name))
+                continue;
+            
+            distinctTypes.Add(typeSymbol);
+        }
+
+        return distinctTypes;
     }
 
     private string? GetNamespaceOfType(ITypeSymbol typeSymbol) => typeSymbol.ContainingNamespace.IsGlobalNamespace ? null : typeSymbol.ContainingNamespace.ToString();
@@ -378,12 +379,4 @@ public class FactoryGenerator : IIncrementalGenerator
         
         return constraints;
     }
-    
-    private string CapitalizeFirstLetter(string input)
-    {
-        if (string.IsNullOrEmpty(input))
-            return input;
-        
-        return char.ToUpper(input[0]) + input.Substring(1);
-    }
 }