*feature* repeat type arguments of generics

Author: Colin Wilmans

src/SourceDepend.Tests/SourceDependTests.cs

@@ -262,4 +262,98 @@ public ExampleService(ConsoleApp.IAnotherService anotherService)
         //Assert
         Assert.Equal(expec, actual.Item2);
     }
+
+    [Fact]
+    public void SourceDepend_WithGenerics_ShouldRepeatTheseGenerics()
+    {
+        //Arrange
+        var source = """
+                     public partial class ExampleService<TKey,TValue>
+                     {
+
+                         [Dependency]
+                         private readonly IAnotherService anotherService;
+                     }
+
+                     public interface IAnotherService {}
+                     """;
+
+        var expec = """
+                     // <auto-generated/>
+                     #pragma warning disable
+                     #nullable enable
+                     /// <inheritdoc/>
+                     public partial class ExampleService<TKey,TValue>
+                     {
+                         public ExampleService(IAnotherService anotherService)
+                         {
+
+                     #if NET6_0_OR_GREATER
+                             ArgumentNullException.ThrowIfNull(anotherService);
+                     #endif
+                             PreConstruct();
+
+                             this.anotherService = anotherService;
+
+                             PostConstruct();
+                         }
+                     
+                         partial void PreConstruct();
+                         partial void PostConstruct();
+                     }
+
+                     """;
+        //Act
+        var actual = SourceCompiler.GetGeneratedOutput(source, _output);
+
+        //Assert
+        Assert.Equal(expec, actual.Item2);
+    }
+
+    [Fact]
+    public void SourceDepend_WithGeneric_ShouldRepeatTheseGenerics()
+    {
+        //Arrange
+        var source = """
+                     public partial class ExampleService<T>
+                     {
+
+                         [Dependency]
+                         private readonly IAnotherService anotherService;
+                     }
+
+                     public interface IAnotherService {}
+                     """;
+
+        var expec = """
+                     // <auto-generated/>
+                     #pragma warning disable
+                     #nullable enable
+                     /// <inheritdoc/>
+                     public partial class ExampleService<T>
+                     {
+                         public ExampleService(IAnotherService anotherService)
+                         {
+
+                     #if NET6_0_OR_GREATER
+                             ArgumentNullException.ThrowIfNull(anotherService);
+                     #endif
+                             PreConstruct();
+
+                             this.anotherService = anotherService;
+
+                             PostConstruct();
+                         }
+                     
+                         partial void PreConstruct();
+                         partial void PostConstruct();
+                     }
+
+                     """;
+        //Act
+        var actual = SourceCompiler.GetGeneratedOutput(source, _output);
+
+        //Assert
+        Assert.Equal(expec, actual.Item2);
+    }
 }

src/SourceDepend/CodeGenerators/DependencyClassCodeGenerator.cs

@@ -20,6 +20,10 @@ internal static string Generate(INamedTypeSymbol classSymbol, List<ISymbol> symb
             var abstractString = classSymbol.IsAbstract ? "abstract " : "";
             var accessibilityKeyword = SyntaxFacts.GetText(classSymbol.DeclaredAccessibility);
 
+            string typeArgumentsString = classSymbol.IsGenericType ?
+                $"<{string.Join(",", classSymbol.TypeArguments.Select(t => t.ToDisplayString(SymbolDisplayFormat.MinimallyQualifiedFormat)))}>" :
+                "";
+
             var spaces = hasNameSpace ? "    " : "";
 
             // begin building the generated source
@@ -35,7 +39,7 @@ internal static string Generate(INamedTypeSymbol classSymbol, List<ISymbol> symb
             }
 
             source.AppendLine($"{spaces}/// <inheritdoc/>");
-            source.AppendLine($"{spaces}{accessibilityKeyword} {abstractString}{sealedString}partial class {classSymbol.Name}");
+            source.AppendLine($"{spaces}{accessibilityKeyword} {abstractString}{sealedString}partial class {classSymbol.Name}{typeArgumentsString}");
             source.AppendLine($"{spaces}{{");
 
             _ = source.Append($"{spaces}    public {classSymbol.Name}(");