Generate Asp.Net extensions (#3109)

* first version of generating aspdotnet extensions

* regenerate

* add new methods

* forgot the this keyword

* move asp writer files to common directory for anyone to use

* update xml docs

* regen after update

* only create for top level clients

* regen after changes

* refactor and fix test cases

* a few refactor

---------

Co-authored-by: m-nash <[email protected]>

Author: ArcturusZhang

samples/AnomalyDetector/Generated/AnomalyDetectorClientBuilderExtensions.cs

@@ -0,0 +1,105 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See License.txt in the project root for license information.
+
+using System;
+using System.Linq;
+using AutoRest.CSharp.Common.Output.Models.Types;
+using AutoRest.CSharp.Generation.Writers;
+using AutoRest.CSharp.Output.Models;
+using Azure.Core;
+using Azure.Core.Extensions;
+
+namespace AutoRest.CSharp.Common.Generation.Writers
+{
+    internal class AspDotNetExtensionWriter
+    {
+        private CodeWriter _writer;
+
+        private AspDotNetExtensionTypeProvider This { get; }
+
+        public AspDotNetExtensionWriter(AspDotNetExtensionTypeProvider aspDotNetExtension)
+        {
+            _writer = new CodeWriter();
+            This = aspDotNetExtension;
+        }
+
+        public void Write()
+        {
+            using (_writer.Namespace(This.Declaration.Namespace))
+            {
+                WriteClassDeclaration();
+                using (_writer.Scope())
+                {
+                    WriteImplementations();
+                }
+            }
+        }
+
+        private void WriteClassDeclaration()
+        {
+            _writer.WriteXmlDocumentationSummary(This.Description);
+            _writer.AppendRaw(This.Declaration.Accessibility)
+                .AppendRaw(" static")
+                .Append($" partial class {This.Type.Name}");
+        }
+
+        private void WriteImplementations()
+        {
+            foreach (var (signature, (declarations, values)) in This.ExtesnsionMethods)
+            {
+                using (_writer.WriteCommonMethodWithoutValidation(signature, null, false, false))
+                {
+                    var builder = signature.Parameters.First();
+                    var arguments = signature.ReturnType!.Arguments;
+                    var clientType = arguments.First();
+                    _writer.Append($"return {builder.Name:I}.RegisterClientFactory")
+                        .AppendRaw("<");
+                    foreach (var argument in arguments)
+                    {
+                        _writer.Append($"{argument},");
+                    }
+                    _writer.RemoveTrailingComma();
+                    _writer.AppendRaw(">((");
+                    foreach (var declaration in declarations)
+                    {
+                        _writer.Append($"{declaration},");
+                    }
+                    _writer.RemoveTrailingComma();
+                    _writer.Append($") => new {clientType}(");
+                    foreach (var value in values)
+                    {
+                        _writer.Append($"{value},");
+                    }
+                    _writer.RemoveTrailingComma();
+                    _writer.LineRaw("));");
+                }
+                _writer.Line();
+            }
+
+            foreach (var signature in This.ExtensionMethodsWithoutCallback)
+            {
+                using (_writer.WriteCommonMethodWithoutValidation(signature, null, false, false))
+                {
+                    var builder = signature.Parameters.First();
+                    var otherParameters = signature.Parameters.Skip(1);
+                    _writer.Append($"return {builder.Name:I}.RegisterClientFactory")
+                        .AppendRaw("<");
+                    foreach (var argument in signature.ReturnType!.Arguments)
+                    {
+                        _writer.Append($"{argument},");
+                    }
+                    _writer.RemoveTrailingComma();
+                    _writer.AppendRaw(">(");
+                    foreach (var parameter in otherParameters)
+                    {
+                        _writer.Append($"{parameter.Name:I},");
+                    }
+                    _writer.RemoveTrailingComma();
+                    _writer.LineRaw(");");
+                }
+            }
+        }
+
+        public override string ToString() => _writer.ToString();
+    }
+}

samples/Azure.AI.DocumentTranslation/Generated/AzureAIDocumentTranslationClientBuilderExtensions.cs

@@ -176,8 +176,8 @@ public static IDisposable WriteMethodDeclaration(this CodeWriter writer, MethodS
 
             writer.AppendRawIf("new ", methodBase.Modifiers.HasFlag(New));
 
-
-            if (methodBase is MethodSignature method)
+            var method = methodBase as MethodSignature;
+            if (method != null)
             {
                 writer
                     .AppendRawIf("virtual ", methodBase.Modifiers.HasFlag(Virtual))
@@ -195,8 +195,21 @@ public static IDisposable WriteMethodDeclaration(this CodeWriter writer, MethodS
                 }
             }
 
+            writer.Append($"{methodBase.Name}");
+
+            if (method?.GenericArguments != null)
+            {
+                writer.AppendRaw("<");
+                foreach (var argument in method.GenericArguments)
+                {
+                    writer.Append($"{argument:D},");
+                }
+                writer.RemoveTrailingComma();
+                writer.AppendRaw(">");
+            }
+
             writer
-                .Append($"{methodBase.Name}(")
+                .AppendRaw("(")
                 .AppendRawIf("this ", methodBase.Modifiers.HasFlag(Extension));
 
             var outerScope = writer.AmbientScope();
@@ -209,6 +222,15 @@ public static IDisposable WriteMethodDeclaration(this CodeWriter writer, MethodS
             writer.RemoveTrailingComma();
             writer.Append($")");
 
+            if (method?.GenericParameterConstraints != null)
+            {
+                writer.Line();
+                foreach (var (argument, constraint) in method.GenericParameterConstraints)
+                {
+                    writer.Append($"where {argument:I}: {constraint}");
+                }
+            }
+
             if (methodBase is ConstructorSignature { Initializer: { } } constructor)
             {
                 var (isBase, arguments) = constructor.Initializer;

samples/Azure.Analytics.Purview.Account/Generated/PurviewAccountClientBuilderExtensions.cs

@@ -12,7 +12,7 @@
 
 namespace AutoRest.CSharp.Output.Models
 {
-    internal record MethodSignature(string Name, string? Summary, string? Description, MethodSignatureModifiers Modifiers, CSharpType? ReturnType, FormattableString? ReturnDescription, IReadOnlyList<Parameter> Parameters, IReadOnlyList<CSharpAttribute>? Attributes = null)
+    internal record MethodSignature(string Name, string? Summary, string? Description, MethodSignatureModifiers Modifiers, CSharpType? ReturnType, FormattableString? ReturnDescription, IReadOnlyList<Parameter> Parameters, IReadOnlyList<CSharpAttribute>? Attributes = null, IReadOnlyList<CSharpType>? GenericArguments = null, IReadOnlyDictionary<CSharpType, FormattableString>? GenericParameterConstraints = null)
         : MethodSignatureBase(Name, Summary, Description, Modifiers, Parameters, Attributes ?? Array.Empty<CSharpAttribute>())
     {
         public FormattableString? FormattableDescription => Description is null ? (FormattableString?)null : $"{Description}";