Implement mgmt resource

eng/Packages.Data.props

@@ -280,6 +280,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/emitter.ts

@@ -2,17 +2,91 @@
 // Licensed under the MIT License. See License.txt in the project root for license information.
 
 import { EmitContext } from "@typespec/compiler";
+import { DecoratorInfo } from "@azure-tools/typespec-client-generator-core";
 
 import {
-  $onEmit as $OnMGCEmit,
+  $onEmit as $onMTGEmit,
+  CodeModel,
   CSharpEmitterOptions,
+  InputModelType,
   setSDKContextOptions
 } from "@typespec/http-client-csharp";
 import { azureSDKContextOptions } from "./sdk-context-options.js";
+import { calculateResourceTypeFromPath } from "./resource-type.js";
+
+const armResourceOperations = "Azure.ResourceManager.@armResourceOperations";
+const armResourceRead = "Azure.ResourceManager.@armResourceRead";
+const armResourceCreateOrUpdate =
+  "Azure.ResourceManager.@armResourceCreateOrUpdate";
+const singleton = "Azure.ResourceManager.@singleton";
+const resourceMetadata = "Azure.ClientGenerator.Core.@resourceSchema";
 
 export async function $onEmit(context: EmitContext<CSharpEmitterOptions>) {
   context.options["plugin-name"] ??= "AzureClientPlugin";
   context.options["emitter-extension-path"] = import.meta.url;
+  context.options["update-code-model"] = updateCodeModel;
   setSDKContextOptions(azureSDKContextOptions);
-  await $OnMGCEmit(context);
+  await $onMTGEmit(context);
+}
+
+function updateCodeModel(codeModel: CodeModel): CodeModel {
+  for (const client of codeModel.Clients) {
+    // TODO: we can implement this decorator in TCGC until we meet the corner case
+    // if the client has resourceMetadata decorator, it is a resource client and we don't need to add it again
+    if (client.Decorators?.some((d) => d.name == resourceMetadata)) {
+      continue;
+    }
+
+    // TODO: Once we have the ability to get resource hierarchy from TCGC directly, we can remove this implementation
+    // A resource client should have decorator armResourceOperations and contains either a get operation(containing armResourceRead deocrator) or a put operation(containing armResourceCreateOrUpdate decorator)
+    if (
+      client.Decorators?.some((d) => d.name == armResourceOperations) &&
+      client.Operations.some(
+        (op) =>
+          op.Decorators?.some(
+            (d) => d.name == armResourceRead || armResourceCreateOrUpdate
+          )
+      )
+    ) {
+      let resourceModel: InputModelType | undefined = undefined;
+      let isSingleton: boolean = false;
+      let resourceType: string | undefined = undefined;
+      // We will try to get resource metadata from put operation firstly, if not found, we will try to get it from get operation
+      const putOperation = client.Operations.find(
+        (op) => op.Decorators?.some((d) => d.name == armResourceCreateOrUpdate)
+      );
+      if (putOperation) {
+        const path = putOperation.Path;
+        resourceType = calculateResourceTypeFromPath(path);
+        resourceModel = putOperation.Responses.filter((r) => r.BodyType)[0]
+          .BodyType as InputModelType;
+        isSingleton =
+          resourceModel.decorators?.some((d) => d.name == singleton) ?? false;
+      } else {
+        const getOperation = client.Operations.find(
+          (op) => op.Decorators?.some((d) => d.name == armResourceRead)
+        );
+        if (getOperation) {
+          const path = getOperation.Path;
+          resourceType = calculateResourceTypeFromPath(path);
+          resourceModel = getOperation.Responses.filter((r) => r.BodyType)[0]
+            .BodyType as InputModelType;
+          isSingleton =
+            resourceModel.decorators?.some((d) => d.name == singleton) ?? false;
+        }
+      }
+
+      const resourceMetadataDecorator: DecoratorInfo = {
+        name: resourceMetadata,
+        arguments: {}
+      };
+      resourceMetadataDecorator.arguments["resourceModel"] =
+        resourceModel?.crossLanguageDefinitionId;
+      resourceMetadataDecorator.arguments["isSingleton"] =
+        isSingleton.toString();
+      resourceMetadataDecorator.arguments["resourceType"] = resourceType;
+      client.Decorators.push(resourceMetadataDecorator);
+    }
+  }
+  return codeModel;
 }

eng/packages/http-client-csharp/emitter/src/resource-type.ts

@@ -0,0 +1,28 @@
+const ResourceGroupScopePrefix =
+  "/subscriptions/{subscriptionId}/resourceGroups";
+const SubscriptionScopePrefix = "/subscriptions";
+const TenantScopePrefix = "/tenants";
+const Providers = "/providers";
+
+export function calculateResourceTypeFromPath(path: string): string {
+  const providerIndex = path.indexOf(Providers);
+  if (providerIndex === -1) {
+    if (path.startsWith(ResourceGroupScopePrefix)) {
+      return "Microsoft.Resources/resourceGroups";
+    } else if (path.startsWith(SubscriptionScopePrefix)) {
+      return "Microsoft.Resources/subscriptions";
+    } else if (path.startsWith(TenantScopePrefix)) {
+      return "Microsoft.Resources/tenants";
+    }
+    throw `Path ${path} doesn't have resource type`;
+  }
+
+  return path
+    .substring(providerIndex + Providers.length)
+    .split("/")
+    .reduce((result, current, index) => {
+      if (index === 1 || index % 2 === 0)
+        return result === "" ? current : `${result}/${current}`;
+      else return result;
+    }, "");
+}

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

@@ -4,11 +4,22 @@
 import { CreateSdkContextOptions } from "@azure-tools/typespec-client-generator-core";
 
 export const azureSDKContextOptions: CreateSdkContextOptions = {
-    versioning: {},
-    additionalDecorators: [
-        // 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"
-    ]
+  versioning: {},
+  additionalDecorators: [
+    // https://github.com/Azure/typespec-azure/blob/main/packages/typespec-client-generator-core/README.md#usesystemtextjsonconverter
+    "Azure\\.ClientGenerator\\.Core\\.@useSystemTextJsonConverter",
+    // TODO: add this decorator to TCGC
+    "Azure\\.ClientGenerator\\.Core\\.@resourceSchema",
+    // https://github.com/Azure/typespec-azure/blob/main/packages/typespec-azure-resource-manager/README.md#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",
+    "Azure\\.ResourceManager\\.Private\\.@armResourceInternal"
+  ]
 };

eng/packages/http-client-csharp/emitter/test/Unit/property-type.test.ts

@@ -3,15 +3,21 @@ import { UsageFlags } from "@azure-tools/typespec-client-generator-core";
 import { strictEqual } from "assert";
 import { beforeEach, describe, it } from "vitest";
 import { createModel } from "@typespec/http-client-csharp";
-import { createCSharpSdkContext, createEmitterContext, createEmitterTestHost, typeSpecCompile, } from "./test-util.js";
+import {
+  createCSharpSdkContext,
+  createEmitterContext,
+  createEmitterTestHost,
+  typeSpecCompile
+} from "./test-util.js";
 
 describe("Test GetInputType for enum", () => {
-    let runner: TestHost;
-    beforeEach(async () => {
-        runner = await createEmitterTestHost();
-    });
-    it("Fixed string enum", async () => {
-        const program = await typeSpecCompile(`
+  let runner: TestHost;
+  beforeEach(async () => {
+    runner = await createEmitterTestHost();
+  });
+  it("Fixed string enum", async () => {
+    const program = await typeSpecCompile(
+      `
         #suppress "@azure-tools/typespec-azure-core/use-extensible-enum" "Enums should be defined without the @fixed decorator."
         @doc("fixed string enum")
         @fixed
@@ -26,31 +32,40 @@ describe("Test GetInputType for enum", () => {
         #suppress "@azure-tools/typespec-azure-core/use-standard-operations" "Operation 'test' should be defined using a signature from the Azure.Core namespace."
         @doc("test fixed enum.")
         op test(@doc("fixed enum as input.")@body input: SimpleEnum): string[];
-      `, runner, { IsNamespaceNeeded: true});
-        const context = createEmitterContext(program);
-        const sdkContext = await createCSharpSdkContext(context);
-        const root = createModel(sdkContext);
-        const inputParamArray = root.Clients[0].Operations[0].Parameters.filter((p) => p.Name === "input");
-        strictEqual(1, inputParamArray.length);
-        const type = inputParamArray[0].Type;
-        strictEqual(type.kind, "enum");
-        strictEqual(type.name, "SimpleEnum");
-        strictEqual(type.isFixed, true);
-        strictEqual(type.doc, "fixed string enum");
-        strictEqual(type.crossLanguageDefinitionId, "Azure.Csharp.Testing.SimpleEnum");
-        strictEqual(type.access, undefined);
-        strictEqual(type.valueType.kind, "string");
-        strictEqual(type.values.length, 3);
-        strictEqual(type.values[0].name, "One");
-        strictEqual(type.values[0].value, "1");
-        strictEqual(type.values[1].name, "Two");
-        strictEqual(type.values[1].value, "2");
-        strictEqual(type.values[2].name, "Four");
-        strictEqual(type.values[2].value, "4");
-        strictEqual(type.usage, UsageFlags.Input | UsageFlags.Json);
-    });
-    it("Fixed int enum", async () => {
-        const program = await typeSpecCompile(`
+      `,
+      runner,
+      { IsNamespaceNeeded: true }
+    );
+    const context = createEmitterContext(program);
+    const sdkContext = await createCSharpSdkContext(context);
+    const root = createModel(sdkContext);
+    const inputParamArray = root.Clients[0].Operations[0].Parameters.filter(
+      (p) => p.Name === "input"
+    );
+    strictEqual(1, inputParamArray.length);
+    const type = inputParamArray[0].Type;
+    strictEqual(type.kind, "enum");
+    strictEqual(type.name, "SimpleEnum");
+    strictEqual(type.isFixed, true);
+    strictEqual(type.doc, "fixed string enum");
+    strictEqual(
+      type.crossLanguageDefinitionId,
+      "Azure.Csharp.Testing.SimpleEnum"
+    );
+    strictEqual(type.access, undefined);
+    strictEqual(type.valueType.kind, "string");
+    strictEqual(type.values.length, 3);
+    strictEqual(type.values[0].name, "One");
+    strictEqual(type.values[0].value, "1");
+    strictEqual(type.values[1].name, "Two");
+    strictEqual(type.values[1].value, "2");
+    strictEqual(type.values[2].name, "Four");
+    strictEqual(type.values[2].value, "4");
+    strictEqual(type.usage, UsageFlags.Input | UsageFlags.Json);
+  });
+  it("Fixed int enum", async () => {
+    const program = await typeSpecCompile(
+      `
       #suppress "@azure-tools/typespec-azure-core/use-extensible-enum" "Enums should be defined without the @fixed decorator."
       @doc("Fixed int enum")
       @fixed
@@ -65,29 +80,37 @@ describe("Test GetInputType for enum", () => {
       #suppress "@azure-tools/typespec-azure-core/use-standard-operations" "Operation 'test' should be defined using a signature from the Azure.Core namespace."
       @doc("test fixed enum.")
       op test(@doc("fixed enum as input.")@body input: FixedIntEnum): string[];
-    `, runner, { IsNamespaceNeeded: true });
-        const context = createEmitterContext(program);
-        const sdkContext = await createCSharpSdkContext(context);
-        const root = createModel(sdkContext);
-        const inputParamArray = root.Clients[0].Operations[0].Parameters.filter((p) => p.Name === "input");
-        strictEqual(1, inputParamArray.length);
-        const type = inputParamArray[0].Type;
-        strictEqual(type.kind, "enum");
-        strictEqual(type.name, "FixedIntEnum");
-        strictEqual(type.crossLanguageDefinitionId, "Azure.Csharp.Testing.FixedIntEnum");
-        strictEqual(type.access, undefined);
-        strictEqual(type.doc, "Fixed int enum");
-        strictEqual(type.valueType.crossLanguageDefinitionId, "TypeSpec.int32");
-        strictEqual(type.valueType.kind, "int32");
-        strictEqual(type.values.length, 3);
-        strictEqual(type.values[0].name, "One");
-        strictEqual(type.values[0].value, 1);
-        strictEqual(type.values[1].name, "Two");
-        strictEqual(type.values[1].value, 2);
-        strictEqual(type.values[2].name, "Four");
-        strictEqual(type.values[2].value, 4);
-        strictEqual(type.isFixed, true);
-        strictEqual(type.usage, UsageFlags.Input | UsageFlags.Json);
-    });
+    `,
+      runner,
+      { IsNamespaceNeeded: true }
+    );
+    const context = createEmitterContext(program);
+    const sdkContext = await createCSharpSdkContext(context);
+    const root = createModel(sdkContext);
+    const inputParamArray = root.Clients[0].Operations[0].Parameters.filter(
+      (p) => p.Name === "input"
+    );
+    strictEqual(1, inputParamArray.length);
+    const type = inputParamArray[0].Type;
+    strictEqual(type.kind, "enum");
+    strictEqual(type.name, "FixedIntEnum");
+    strictEqual(
+      type.crossLanguageDefinitionId,
+      "Azure.Csharp.Testing.FixedIntEnum"
+    );
+    strictEqual(type.access, undefined);
+    strictEqual(type.doc, "Fixed int enum");
+    strictEqual(type.valueType.crossLanguageDefinitionId, "TypeSpec.int32");
+    strictEqual(type.valueType.kind, "int32");
+    strictEqual(type.values.length, 3);
+    strictEqual(type.values[0].name, "One");
+    strictEqual(type.values[0].value, 1);
+    strictEqual(type.values[1].name, "Two");
+    strictEqual(type.values[1].value, 2);
+    strictEqual(type.values[2].name, "Four");
+    strictEqual(type.values[2].value, 4);
+    strictEqual(type.isFixed, true);
+    strictEqual(type.usage, UsageFlags.Input | UsageFlags.Json);
+  });
 });
-//# sourceMappingURL=property-type.test.js.map
+//# sourceMappingURL=property-type.test.js.map

eng/packages/http-client-csharp/emitter/test/Unit/scalar.test.ts

@@ -6,7 +6,7 @@ import {
   createCSharpSdkContext,
   createEmitterContext,
   createEmitterTestHost,
-  typeSpecCompile,
+  typeSpecCompile
 } from "./test-util.js";
 
 describe("Test GetInputType for scalar", () => {
@@ -21,13 +21,13 @@ describe("Test GetInputType for scalar", () => {
       `
         op test(@query location: azureLocation): void;
       `,
-      runner,
+      runner
     );
     const context = await createCSharpSdkContext(createEmitterContext(program));
     const model = createModel(context);
 
     const inputParamArray = model.Clients[0].Operations[0].Parameters.filter(
-      (p) => p.Name === "location",
+      (p) => p.Name === "location"
     );
     strictEqual(1, inputParamArray.length);
     const type = inputParamArray[0].Type;