[EventHubs] Migrate to TypeSpec (#48673)

Author: weidongxu-microsoft

eng/lintingconfigs/revapi/track2/revapi.json

@@ -769,6 +769,38 @@
           },
           "justification": "Pageable models moved to implementation package. Unlikely used by user."
         },
+        {
+          "code": "java.method.visibilityReduced",
+          "old" : {
+            "matcher": "regex",
+            "match": "method void com\\.azure\\.resourcemanager\\.eventhubs\\.models\\..*\\:\\:\\<init\\>\\(\\)"
+          },
+          "justification": "Output-only immutable models' constructors are now private."
+        },
+        {
+          "code": "java.method.visibilityReduced",
+          "old" : {
+            "matcher": "regex",
+            "match": "method .* com\\.azure\\.resourcemanager\\.eventhubs\\.models\\..*\\:\\:with.*\\(.*\\).*"
+          },
+          "justification": "Output-only immutable models' setters are now package-private if it's being used by child class."
+        },
+        {
+          "code": "java.method.removed",
+          "old" : {
+            "matcher": "regex",
+            "match": "method .* com\\.azure\\.resourcemanager\\.eventhubs\\.models\\..*\\:\\:with.*\\(.*\\).*"
+          },
+          "justification": "Output-only immutable models' setters are removed if no explicit usage."
+        },
+        {
+          "code": "java.class.removed",
+          "old" : {
+            "matcher": "regex",
+            "match": "class com\\.azure\\.resourcemanager\\.eventhubs\\.models\\..*(ListResult|List)"
+          },
+          "justification": "Pageable models moved to implementation package. Unlikely used by user."
+        },
         {
           "ignore": true,
           "code": "java.method.removed",

sdk/cosmos/azure-resourcemanager-cosmos/src/main/java/com/azure/resourcemanager/cosmos/implementation/CosmosDBAccountImpl.java

@@ -806,6 +806,12 @@ public CosmosDBAccountImpl disableLocalAuth() {
         return this;
     }
 
+    @Override
+    public CosmosDBAccountImpl enableLocalAuth() {
+        this.innerModel().withDisableLocalAuth(false);
+        return this;
+    }
+
     @Override
     public CosmosDBAccountImpl enableAutomaticFailover() {
         this.innerModel().withEnableAutomaticFailover(true);

sdk/cosmos/azure-resourcemanager-cosmos/src/main/java/com/azure/resourcemanager/cosmos/models/CosmosDBAccount.java

@@ -592,7 +592,7 @@ interface WithOptionals extends Resource.UpdateWithTags<WithOptionals>, Appliabl
             UpdateStages.WithConsistencyPolicy, UpdateStages.WithVirtualNetworkRule, UpdateStages.WithMultipleLocations,
             UpdateStages.WithConnector, UpdateStages.WithKeyBasedMetadataWriteAccess,
             UpdateStages.WithPrivateEndpointConnection, UpdateStages.WithIpRules, UpdateStages.WithPublicNetworkAccess,
-            UpdateStages.WithAutomaticFailover {
+            UpdateStages.WithAutomaticFailover, UpdateStages.WithLocalAuth {
         }
 
         /** The stage of the cosmos db definition allowing the definition of a write location. */
@@ -816,5 +816,28 @@ default Update disableAutomaticFailover() {
                 throw new UnsupportedOperationException("[disableAutomaticFailover] is not supported in " + getClass());
             }
         }
+
+        /**
+         * The stage of CosmosDB account update allowing to configure local auth settings.
+         */
+        interface WithLocalAuth {
+            /**
+             * Disables local auth for the CosmosDB account.
+             *
+             * @return the next stage of the update
+             */
+            default Update disableLocalAuth() {
+                throw new UnsupportedOperationException("[disableLocalAuth] is not supported in " + getClass());
+            }
+
+            /**
+             * Enables local auth for the CosmosDB account.
+             *
+             * @return the next stage of the update
+             */
+            default Update enableLocalAuth() {
+                throw new UnsupportedOperationException("[enableLocalAuth] is not supported in " + getClass());
+            }
+        }
     }
 }

sdk/eventhubs/azure-resourcemanager-eventhubs/CHANGELOG.md

@@ -4,6 +4,8 @@
 
 ### Features Added
 
+- Supported `disableLocalAuth()` and `enableLocalAuth()` in `EventHubNamespace`.
+
 ### Breaking Changes
 
 ### Bugs Fixed

sdk/eventhubs/azure-resourcemanager-eventhubs/README.md

@@ -18,7 +18,7 @@ For documentation on how to use this package, please see [Azure Management Libra
 <dependency>
     <groupId>com.azure.resourcemanager</groupId>
     <artifactId>azure-resourcemanager-eventhubs</artifactId>
-    <version>2.53.1</version>
+    <version>2.54.0-beta.1</version>
 </dependency>
 ```
 [//]: # ({x-version-update-end})

sdk/eventhubs/azure-resourcemanager-eventhubs/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "java",
   "TagPrefix": "java/eventhubs/azure-resourcemanager-eventhubs",
-  "Tag": "java/eventhubs/azure-resourcemanager-eventhubs_7f9757c7bd"
+  "Tag": "java/eventhubs/azure-resourcemanager-eventhubs_aef499556f"
 }

sdk/eventhubs/azure-resourcemanager-eventhubs/customization/pom.xml

@@ -0,0 +1,17 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+
+  <parent>
+    <groupId>com.azure</groupId>
+    <artifactId>azure-code-customization-parent</artifactId>
+    <version>1.0.0-beta.1</version> <!-- {x-version-update;com.azure:azure-code-customization-parent;current} -->
+    <relativePath>../../../parents/azure-code-customization-parent</relativePath>
+  </parent>
+
+  <groupId>com.azure.resourcemanager</groupId>
+  <artifactId>azure-resourcemanager-eventhubs-customization</artifactId>
+  <version>1.0.0-beta.1</version>
+  <modelVersion>4.0.0</modelVersion>
+</project>

sdk/eventhubs/azure-resourcemanager-eventhubs/customization/src/main/java/EventHubsCustomization.java

@@ -0,0 +1,38 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+import com.azure.autorest.customization.ClassCustomization;
+import com.azure.autorest.customization.Customization;
+import com.azure.autorest.customization.LibraryCustomization;
+import com.azure.autorest.customization.PackageCustomization;
+import com.github.javaparser.ast.type.ClassOrInterfaceType;
+import org.slf4j.Logger;
+
+/**
+ * Code customization after code generation.
+ */
+public class EventHubsCustomization extends Customization {
+    @Override
+    public void customize(LibraryCustomization customization, Logger logger) {
+        PackageCustomization fluentModelsPackage = customization.getPackage("com.azure.resourcemanager.eventhubs.fluent.models");
+        // change base class from `ProxyResource` to `Resource`, to avoid breaking changes and compilation errors
+        customizeResourceBaseClass(fluentModelsPackage.getClass("EHNamespaceInner"));
+        customizeResourceBaseClass(fluentModelsPackage.getClass("ClusterInner"));
+    }
+
+    /**
+     * Customize the base class to be "com.azure.core.management.Resource".
+     *
+     * @param customization the customization for class
+     */
+    private static void customizeResourceBaseClass(ClassCustomization customization) {
+        customization.customizeAst(ast -> {
+            ast.getClassByName(customization.getClassName()).ifPresent(clazz -> {
+                String resourceClassName = "com.azure.core.management.Resource";
+                ast.addImport(resourceClassName);
+                clazz.getExtendedTypes().clear();
+                clazz.addExtendedType(new ClassOrInterfaceType(null, "Resource"));
+            });
+        });
+    }
+}