Generate matching IServiceProvider parameter name

ServiceScan.SourceGenerator.Tests/GeneratedMethodTests.cs

@@ -219,6 +219,46 @@ public static partial IServiceCollection AddServices(this IServiceCollection ser
         Assert.Equal(expected, results.GeneratedTrees[1].ToString());
     }
 
+    [Fact]
+    public void MethodWithCustomParameterName()
+    {
+        var compilation = CreateCompilation(Services,
+            """
+            using ServiceScan.SourceGenerator;
+            using Microsoft.Extensions.DependencyInjection;
+
+            namespace GeneratorTests;
+
+            public static partial class ServicesExtensions
+            {
+                [GenerateServiceRegistrations(AssignableTo = typeof(IService))]
+                public static partial IServiceCollection AddServices(this IServiceCollection strangeServices);
+            }
+            """);
+
+        var results = CSharpGeneratorDriver
+            .Create(_generator)
+            .RunGenerators(compilation)
+            .GetRunResult();
+
+        var expected = """
+            using Microsoft.Extensions.DependencyInjection;
+
+            namespace GeneratorTests;
+
+            public static partial class ServicesExtensions
+            {
+                public static partial IServiceCollection AddServices(this IServiceCollection strangeServices)
+                {
+                    return strangeServices
+                        .AddTransient<GeneratorTests.IService, GeneratorTests.MyService>();
+                }
+            }
+            """;
+
+        Assert.Equal(expected, results.GeneratedTrees[1].ToString());
+    }
+
     private static Compilation CreateCompilation(params string[] source)
     {
         var path = Path.GetDirectoryName(typeof(object).Assembly.Location)!;

ServiceScan.SourceGenerator/DependencyInjectionGenerator.cs

@@ -68,9 +68,9 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
 
                 {{method.TypeModifiers}} class {{method.TypeName}}
                 {
-                    {{method.MethodModifiers}} {{returnType}} {{method.MethodName}}({{(method.IsExtensionMethod ? "this" : "")}} IServiceCollection services)
+                    {{method.MethodModifiers}} {{returnType}} {{method.MethodName}}({{(method.IsExtensionMethod ? "this" : "")}} IServiceCollection {{method.ParameterName}})
                     {
-                        {{(method.ReturnsVoid ? "" : "return ")}}services
+                        {{(method.ReturnsVoid ? "" : "return ")}}{{method.ParameterName}}
                             {{sb.ToString().Trim()}};
                     }
                 }

ServiceScan.SourceGenerator/Model/MethodModel.cs

@@ -10,18 +10,22 @@ record MethodModel(
     string TypeModifiers,
     string MethodName,
     string MethodModifiers,
+    string ParameterName,
     bool IsExtensionMethod,
     bool ReturnsVoid)
 {
     public static MethodModel Create(IMethodSymbol method, SyntaxNode syntax)
     {
+        var parameterName = method.Parameters[0].Name;
+
         return new MethodModel(
             Namespace: method.ContainingNamespace.IsGlobalNamespace ? null : method.ContainingNamespace.ToDisplayString(),
             TypeName: method.ContainingType.Name,
             TypeMetadataName: method.ContainingType.ToFullMetadataName(),
             TypeModifiers: GetModifiers(GetTypeSyntax(syntax)),
             MethodName: method.Name,
             MethodModifiers: GetModifiers(syntax),
+            ParameterName: parameterName,
             IsExtensionMethod: method.IsExtensionMethod,
             ReturnsVoid: method.ReturnsVoid);
     }