Implement mgmt resource

eng/Packages.Data.props

@@ -276,6 +276,7 @@
     <PackageReference Update="Azure.Messaging.EventGrid" Version="4.17.0" />
     <PackageReference Update="Azure.Messaging.EventHubs.Processor" Version="5.11.6" />
     <PackageReference Update="Azure.Messaging.ServiceBus" Version="7.18.2" />
+    <PackageReference Update="Azure.ResourceManager" Version="1.13.0" />
     <PackageReference Update="Azure.ResourceManager.Compute" Version="1.7.0-beta.1" />
     <PackageReference Update="Azure.ResourceManager.CognitiveServices" Version="1.3.0" />
     <PackageReference Update="Azure.ResourceManager.KeyVault" Version="1.1.0" />

eng/packages/http-client-csharp/emitter/src/sdk-context-options.ts

@@ -9,6 +9,14 @@ export const azureSDKContextOptions: CreateSdkContextOptions = {
         // https://github.com/Azure/typespec-azure/blob/main/packages/typespec-client-generator-core/README.md#usesystemtextjsonconverter
         "Azure\\.ClientGenerator\\.Core\\.@useSystemTextJsonConverter",
         // https://github.com/Azure/typespec-azure/blob/main/packages/typespec-azure-resource-manager/README.md#armprovidernamespace
-        "Azure\\.ResourceManager\\.@armProviderNamespace"
+        "Azure\\.ResourceManager\\.@armProviderNamespace",
+        // https://github.com/Azure/typespec-azure/blob/main/packages/typespec-azure-resource-manager/README.md#armresourceoperations
+        "Azure\\.ResourceManager\\.@armResourceOperations",
+        // https://github.com/Azure/typespec-azure/blob/main/packages/typespec-azure-resource-manager/README.md#armResourceRead
+        "Azure\\.ResourceManager\\.@armResourceRead",
+        // https://github.com/microsoft/typespec/blob/main/packages/rest/README.md#parentresource
+        "TypeSpec\\.Rest\\.parentResource",
+        // https://github.com/Azure/typespec-azure/blob/main/packages/typespec-azure-resource-manager/README.md#singleton
+        "Azure\\.ResourceManager\\.@singleton"
     ]
 };

eng/packages/http-client-csharp/eng/scripts/Generate.ps1

@@ -18,10 +18,9 @@ if (-not $LaunchOnly) {
         Write-Host "Generating BasicTypeSpec" -ForegroundColor Cyan
         $testProjectsLocalDir = Join-Path $packageRoot 'generator' 'TestProjects' 'Local'
 
-        $unbrandedTypespecTestProject = Join-Path $testProjectsLocalDir "Basic-TypeSpec"
-        $unbrandedTypespecTestProject = $unbrandedTypespecTestProject
+        $basicTypespecTestProject = Join-Path $testProjectsLocalDir "Basic-TypeSpec"
 
-        Invoke (Get-TspCommand "$unbrandedTypespecTestProject/Basic-TypeSpec.tsp" $unbrandedTypespecTestProject -forceNewProject $ForceNewProject)
+        Invoke (Get-TspCommand "$basicTypespecTestProject/Basic-TypeSpec.tsp" $basicTypespecTestProject -forceNewProject $ForceNewProject)
 
         # exit if the generation failed
         if ($LASTEXITCODE -ne 0) {
@@ -36,6 +35,28 @@ if (-not $LaunchOnly) {
             exit $LASTEXITCODE
         }
     }
+
+    if ($null -eq $filter -or $filter -eq "Mgmt-TypeSpec") {
+        Write-Host "Generating MgmtTypeSpec" -ForegroundColor Cyan
+        $testProjectsLocalDir = Join-Path $packageRoot 'generator' 'TestProjects' 'Local'
+
+        $mgmtTypespecTestProject = Join-Path $testProjectsLocalDir "Mgmt-TypeSpec"
+
+        Invoke (Get-TspCommand "$mgmtTypespecTestProject/main.tsp" $mgmtTypespecTestProject -forceNewProject $ForceNewProject)
+
+        # exit if the generation failed
+        if ($LASTEXITCODE -ne 0) {
+            exit $LASTEXITCODE
+        }
+
+        Write-Host "Building MgmtTypeSpec" -ForegroundColor Cyan
+        Invoke "dotnet build $packageRoot/generator/TestProjects/Local/Mgmt-TypeSpec/src/MgmtTypeSpec.csproj"
+
+        # exit if the generation failed
+        if ($LASTEXITCODE -ne 0) {
+            exit $LASTEXITCODE
+        }
+    }
 }
 
 $specsDirectory = "$packageRoot/node_modules/@typespec/http-specs"
@@ -174,14 +195,20 @@ if ($null -eq $filter) {
     Write-Host "Writing new launch settings" -ForegroundColor Cyan
     $mgcExe = "`$(SolutionDir)/../dist/generator/Microsoft.Generator.CSharp.exe"
     $sampleExe = "`$(SolutionDir)/../generator/artifacts/bin/SamplePlugin/Debug/net8.0/Microsoft.Generator.CSharp.exe"
-    $unbrandedSpec = "TestProjects/Local/Basic-TypeSpec"
+    $basicSpec = "TestProjects/Local/Basic-TypeSpec"
+    $mgmtSpec = "TestProjects/Local/Mgmt-TypeSpec"
 
     $launchSettings = @{}
     $launchSettings.Add("profiles", @{})
     $launchSettings["profiles"].Add("Basic-TypeSpec", @{})
-    $launchSettings["profiles"]["Basic-TypeSpec"].Add("commandLineArgs", "`$(SolutionDir)/../dist/generator/Microsoft.Generator.CSharp.dll `$(SolutionDir)/$unbrandedSpec -p AzureClientPlugin")
+    $launchSettings["profiles"]["Basic-TypeSpec"].Add("commandLineArgs", "`$(SolutionDir)/../dist/generator/Microsoft.Generator.CSharp.dll `$(SolutionDir)/$basicSpec -p AzureClientPlugin")
     $launchSettings["profiles"]["Basic-TypeSpec"].Add("commandName", "Executable")
     $launchSettings["profiles"]["Basic-TypeSpec"].Add("executablePath", "dotnet")
+
+    $launchSettings["profiles"].Add("Mgmt-TypeSpec", @{})
+    $launchSettings["profiles"]["Mgmt-TypeSpec"].Add("commandLineArgs", "`$(SolutionDir)/../dist/generator/Microsoft.Generator.CSharp.dll `$(SolutionDir)/$mgmtSpec -p AzureClientPlugin")
+    $launchSettings["profiles"]["Mgmt-TypeSpec"].Add("commandName", "Executable")
+    $launchSettings["profiles"]["Mgmt-TypeSpec"].Add("executablePath", "dotnet")
 
     foreach ($kvp in $spectorLaunchProjects.GetEnumerator()) {
         $launchSettings["profiles"].Add($kvp.Key, @{})

eng/packages/http-client-csharp/generator/Azure.Generator/src/AzureClientPlugin.cs

@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
+using Azure.ResourceManager;
 using Azure.Generator.Utilities;
 using Microsoft.CodeAnalysis;
 using Microsoft.TypeSpec.Generator;
@@ -9,6 +10,7 @@
 using System.ComponentModel.Composition;
 using System.IO;
 using System.Linq;
+using Azure.Generator.Primitives;
 
 namespace Azure.Generator;
 
@@ -29,8 +31,6 @@ public class AzureClientPlugin : ScmCodeModelPlugin
     /// <inheritdoc/>
     public override AzureOutputLibrary OutputLibrary => _azureOutputLibrary ??= new();
 
-    internal ResourceDetection ResourceDetection { get; } = new();
-
     /// <summary>
     /// The Azure client plugin to generate the Azure client SDK.
     /// </summary>
@@ -55,6 +55,8 @@ public override void Configure()
         AddVisitor(new NamespaceVisitor());
         if (IsAzureArm.Value)
         {
+            // Include Azure.ResourceManager
+            AddMetadataReference(MetadataReference.CreateFromFile(typeof(ArmClient).Assembly.Location));
             AddVisitor(new RestClientVisitor());
             AddVisitor(new ResourceVisitor());
         }
@@ -71,5 +73,5 @@ public override void Configure()
     /// <summary>
     /// Identify if the input is generated for Azure ARM.
     /// </summary>
-    internal Lazy<bool> IsAzureArm => new Lazy<bool>(() => InputLibrary.InputNamespace.Clients.Any(c => c.Decorators.Any(d => d.Name.Equals("Azure.ResourceManager.@armProviderNamespace"))));
+    internal Lazy<bool> IsAzureArm => new Lazy<bool>(() => InputLibrary.InputNamespace.Clients.Any(c => c.Decorators.Any(d => d.Name.Equals(KnownDecorators.ArmProviderNamespace))));
 }

eng/packages/http-client-csharp/generator/Azure.Generator/src/AzureOutputLibrary.cs

@@ -1,96 +1,61 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
-using Azure.Generator.Mgmt.Models;
+using Azure.Generator.Primitives;
 using Azure.Generator.Providers;
-using Azure.Generator.Utilities;
 using Microsoft.TypeSpec.Generator.ClientModel;
+using Microsoft.TypeSpec.Generator.Input;
 using Microsoft.TypeSpec.Generator.Providers;
 using System.Collections.Generic;
+using System.Linq;
 
 namespace Azure.Generator
 {
     /// <inheritdoc/>
     public class AzureOutputLibrary : ScmOutputLibrary
     {
-        // TODO: categorize clients into operationSets, which contains operations sharing the same Path
-        private Dictionary<string, OperationSet> _pathToOperationSetMap;
-        private Dictionary<string, HashSet<OperationSet>> _resourceDataBySpecNameMap;
+        private LongRunningOperationProvider? _armOperation;
+        internal LongRunningOperationProvider ArmOperation => _armOperation ??= new LongRunningOperationProvider(false);
 
-        /// <inheritdoc/>
-        public AzureOutputLibrary()
-        {
-            _pathToOperationSetMap = CategorizeClients();
-            _resourceDataBySpecNameMap = EnsureResourceDataMap();
-        }
+        private LongRunningOperationProvider? _genericArmOperation;
+        internal LongRunningOperationProvider GenericArmOperation => _genericArmOperation ??= new LongRunningOperationProvider(true);
 
-        private Dictionary<string, HashSet<OperationSet>> EnsureResourceDataMap()
+        private IReadOnlyList<TypeProvider> BuildResources()
         {
-            var result = new Dictionary<string, HashSet<OperationSet>>();
-            foreach (var operationSet in _pathToOperationSetMap.Values)
+            var result = new List<TypeProvider>();
+            foreach (var client in AzureClientPlugin.Instance.InputLibrary.InputNamespace.Clients)
             {
-                if (AzureClientPlugin.Instance.ResourceDetection.TryGetResourceDataSchema(operationSet, out var resourceSpecName, out var resourceSchema))
+                // A resource client should contain the decorator "Azure.ResourceManager.@armResourceOperations"
+                // and it should contain a get operation, which contains the decorator "Azure.ResourceManager.@armResourceRead"
+                if (client.Decorators.Any(d => d.Name.Equals(KnownDecorators.ArmResourceOperations))
+                    && client.Operations.Any(operation => operation.Decorators.Any(d => d.Name.Equals(KnownDecorators.ArmResourceRead))))
                 {
-                    // if this operation set corresponds to a SDK resource, we add it to the map
-                    if (!result.TryGetValue(resourceSpecName!, out HashSet<OperationSet>? value))
-                    {
-                        value = new HashSet<OperationSet>();
-                        result.Add(resourceSpecName!, value);
-                    }
-                    value.Add(operationSet);
+                    var resource = CreateResourceCore(client);
+                    AzureClientPlugin.Instance.AddTypeToKeep(resource.Name);
+                    result.Add(CreateResourceCore(client));
                 }
             }
-
             return result;
         }
 
-        private Dictionary<string, OperationSet> CategorizeClients()
-        {
-            var result = new Dictionary<string, OperationSet>();
-            foreach (var inputClient in AzureClientPlugin.Instance.InputLibrary.InputNamespace.Clients)
-            {
-                var requestPathList = new HashSet<string>();
-                foreach (var operation in inputClient.Operations)
-                {
-                    var path = operation.GetHttpPath();
-                    requestPathList.Add(path);
-                    if (result.TryGetValue(path, out var operationSet))
-                    {
-                        operationSet.Add(operation);
-                    }
-                    else
-                    {
-                        operationSet = new OperationSet(path, inputClient)
-                        {
-                            operation
-                        };
-                        result.Add(path, operationSet);
-                    }
-                }
-            }
-
-            // TODO: add operation set for the partial resources here
-
-            return result;
-        }
+        /// <summary>
+        /// Create a resource client provider
+        /// </summary>
+        /// <param name="inputClient"></param>
+        /// <returns></returns>
+        public virtual TypeProvider CreateResourceCore(InputClient inputClient) => new ResourceClientProvider(inputClient);
 
         /// <inheritdoc/>
-        // TODO: generate resources and collections
+        // TODO: generate collections
         protected override TypeProvider[] BuildTypeProviders()
         {
+            var baseProviders = base.BuildTypeProviders();
             if (AzureClientPlugin.Instance.IsAzureArm.Value == true)
             {
-                var armOperation = new MgmtLongRunningOperationProvider(false);
-                var genericArmOperation = new MgmtLongRunningOperationProvider(true);
-
-                // TODO: remove them once they are referenced in Resource operation implementation
-                AzureClientPlugin.Instance.AddTypeToKeep(armOperation.Name);
-                AzureClientPlugin.Instance.AddTypeToKeep(genericArmOperation.Name);
-                return [.. base.BuildTypeProviders(), new RequestContextExtensionsDefinition(), armOperation, genericArmOperation];
+                var resources = BuildResources();
+                return [.. baseProviders, new RequestContextExtensionsDefinition(), ArmOperation, GenericArmOperation, .. resources, .. resources.Select(r => ((ResourceClientProvider)r).Source)];
             }
-            return [.. base.BuildTypeProviders(), new RequestContextExtensionsDefinition()];
+            return [.. baseProviders, new RequestContextExtensionsDefinition()];
         }
-
-        internal bool IsResource(string name) => _resourceDataBySpecNameMap.ContainsKey(name);
     }
 }