Generate parameters syntax using Roslyn API

Author: kant2002

SqlMarshal/AbstractGenerator.cs

@@ -11,7 +11,6 @@ namespace SqlMarshal;
 using System.Text;
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp;
-using Microsoft.CodeAnalysis.CSharp.Syntax;
 using Microsoft.CodeAnalysis.Text;
 using static Microsoft.CodeAnalysis.CSharp.SyntaxFactory;
 using static SqlMarshal.Extensions;
@@ -101,6 +100,15 @@ internal static IEnumerable<string> GetUsings(ClassGenerationContext classGenera
         }
     }
 
+    /// <summary>
+    /// Gets code for parameter declaration.
+    /// </summary>
+    /// <param name="methodSymbol">Method symbol from which we copy parameter.</param>
+    /// <param name="parameter">Parameter to copy.</param>
+    /// <param name="index">Index of parameter to copy.</param>
+    /// <returns>Generated syntax node for the parameter.</returns>
+    protected abstract SyntaxNode GetParameterDeclaration(IMethodSymbol methodSymbol, IParameterSymbol parameter, int index);
+
     private static string GetAccessibility(Accessibility a)
     {
         return a switch
@@ -138,26 +146,6 @@ private static string GetAccessibility(Accessibility a)
         return null;
     }
 
-    private static string GetParameterDeclaration(IMethodSymbol methodSymbol, IParameterSymbol parameter, int index)
-    {
-        if (parameter.RefKind == RefKind.Out)
-        {
-            return $"out {parameter.Type.ToDisplayString()} {parameter.Name}";
-        }
-
-        if (parameter.RefKind == RefKind.Ref)
-        {
-            return $"ref {parameter.Type.ToDisplayString()} {parameter.Name}";
-        }
-
-        if (methodSymbol.IsExtensionMethod && index == 0)
-        {
-            return $"this {parameter.Type.ToDisplayString()} {parameter.Name}";
-        }
-
-        return $"{parameter.Type.ToDisplayString()} {parameter.Name}";
-    }
-
     private static string GetParameterPassing(IParameterSymbol parameter)
     {
         var parameterName = NameMapper.MapName(parameter.Name);
@@ -929,7 +917,7 @@ private void ProcessMethod(
         var originalParameters = methodSymbol.Parameters;
 
         bool hasCustomSql = methodGenerationContext.CustomSqlParameter != null;
-        var signature = $"({string.Join(", ", originalParameters.Select((parameterSymbol, index) => GetParameterDeclaration(methodSymbol, parameterSymbol, index)))})";
+        var signature = $"({string.Join(", ", originalParameters.Select((parameterSymbol, index) => this.GetParameterDeclaration(methodSymbol, parameterSymbol, index)))})";
         var itemType = methodGenerationContext.ItemType;
         var getConnection = this.GetConnectionStatement(methodGenerationContext);
         var isList = methodGenerationContext.IsList || methodGenerationContext.IsEnumerable;

SqlMarshal/CSharpGenerator.cs

@@ -7,9 +7,11 @@
 namespace SqlMarshal;
 
 using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.CSharp;
 using Microsoft.CodeAnalysis.CSharp.Syntax;
 using System.Collections.Generic;
 using System.Linq;
+using static Microsoft.CodeAnalysis.CSharp.SyntaxFactory;
 
 /// <summary>
 /// Stored procedures generator for C#.
@@ -64,6 +66,32 @@ public override void Initialize(GeneratorInitializationContext context)
         context.RegisterForSyntaxNotifications(() => new SyntaxReceiver());
     }
 
+    /// <inheritdoc/>
+    protected override SyntaxNode GetParameterDeclaration(IMethodSymbol methodSymbol, IParameterSymbol parameter, int index)
+    {
+        var typeAsClause = ParseTypeName(parameter.Type.ToDisplayString()).WithTrailingTrivia(Whitespace(" "));
+        if (parameter.RefKind == RefKind.Out)
+        {
+            var parameterSyntax = Parameter(default, SyntaxTokenList.Create(Token(SyntaxKind.OutKeyword).WithTrailingTrivia(Whitespace(" "))), typeAsClause, Identifier(parameter.Name), @default: null);
+            return parameterSyntax;
+        }
+        else if (parameter.RefKind == RefKind.Ref)
+        {
+            var parameterSyntax = Parameter(default, SyntaxTokenList.Create(Token(SyntaxKind.RefKeyword).WithTrailingTrivia(Whitespace(" "))), typeAsClause, Identifier(parameter.Name), @default: null);
+            return parameterSyntax;
+        }
+        else if (methodSymbol.IsExtensionMethod && index == 0)
+        {
+            var parameterSyntax = Parameter(default, SyntaxTokenList.Create(Token(SyntaxKind.ThisKeyword).WithTrailingTrivia(Whitespace(" "))), typeAsClause, Identifier(parameter.Name), @default: null);
+            return parameterSyntax;
+        }
+        else
+        {
+            var parameterSyntax = Parameter(default, default, typeAsClause, Identifier(parameter.Name), @default: null);
+            return parameterSyntax;
+        }
+    }
+
     internal class SyntaxReceiver : ISqlMarshalSyntaxReceiver
     {
         public List<IMethodSymbol> Methods { get; } = new List<IMethodSymbol>();

SqlMarshal/VisualBasicGenerator.cs

@@ -7,9 +7,11 @@
 namespace SqlMarshal;
 
 using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.VisualBasic;
 using Microsoft.CodeAnalysis.VisualBasic.Syntax;
 using System.Collections.Generic;
 using System.Linq;
+using static Microsoft.CodeAnalysis.VisualBasic.SyntaxFactory;
 
 /// <summary>
 /// Stored procedures generator for C#.
@@ -23,6 +25,8 @@ public class VisualBasicGenerator : AbstractGenerator
 ' regenerated.
 ' </auto-generated>
 
+Namespace SqlMarshal.Annotations
+
 <System.AttributeUsage(System.AttributeTargets.Method, AllowMultiple:=True)>
 Friend NotInheritable Class SqlMarshalAttribute
     Inherits System.Attribute
@@ -57,6 +61,7 @@ End Sub
     Public Property EntityType As System.Type
 End Class
 
+End Namespace
 ";
 
     /// <inheritdoc/>
@@ -69,6 +74,27 @@ public override void Initialize(GeneratorInitializationContext context)
         context.RegisterForSyntaxNotifications(() => new SyntaxReceiver());
     }
 
+    /// <inheritdoc/>
+    protected override SyntaxNode GetParameterDeclaration(IMethodSymbol methodSymbol, IParameterSymbol parameter, int index)
+    {
+        var typeAsClause = SimpleAsClause(ParseTypeName(parameter.Type.ToDisplayString()).WithLeadingTrivia(Whitespace(" ")));
+        if (parameter.RefKind == RefKind.Out)
+        {
+            var parameterSyntax = Parameter(default, SyntaxTokenList.Create(Token(SyntaxKind.ByRefKeyword).WithTrailingTrivia(Whitespace(" "))), ModifiedIdentifier(parameter.Name).WithTrailingTrivia(Whitespace(" ")), typeAsClause, @default: null);
+            return parameterSyntax;
+        }
+        else if (parameter.RefKind == RefKind.Ref)
+        {
+            var parameterSyntax = Parameter(default, SyntaxTokenList.Create(Token(SyntaxKind.ByRefKeyword).WithTrailingTrivia(Whitespace(" "))), ModifiedIdentifier(parameter.Name).WithTrailingTrivia(Whitespace(" ")), typeAsClause, @default: null);
+            return parameterSyntax;
+        }
+        else
+        {
+            var parameterSyntax = Parameter(default, default, ModifiedIdentifier(parameter.Name).WithTrailingTrivia(Whitespace(" ")), typeAsClause, @default: null);
+            return parameterSyntax;
+        }
+    }
+
     internal class SyntaxReceiver : ISqlMarshalSyntaxReceiver
     {
         public List<IMethodSymbol> Methods { get; } = new List<IMethodSymbol>();