Add initial Mongo cluster CLI extension

src/docdb/HISTORY.rst

@@ -0,0 +1,13 @@
+.. :changelog:
+
+Release History
+===============
+
+1.0.0b1
+++++++
+* Initial preview version of the Document DB extension for Azure CLI for API version 2024-07-01.
+* Adds commands to
+  * CRUD for clusters.
+  * CRUD for cluster firewall rules.
+  * Create, list and promote replica clusters.
+  * Point in time restore of a cluster.

src/docdb/README.md

@@ -0,0 +1,49 @@
+# Azure CLI Database for Document DB Extension
+This is an extension to Azure CLI to manage Database for Document DB resources.
+
+This includes:
+- Create/update cluster instances.
+- Add/remove firewall rules to a cluster.
+- Restoring cluster to a point in time from backups.
+- Promoting a replica cluster to be a primary cluster.
+
+## How to use
+
+Install the extension with this command:
+
+```bash
+az extension add --name docdb
+```
+
+Use help `-h` to get a list of available commands:
+```bash
+az docdb cluster -h
+```
+
+And to get a description of a command and usage examples:
+```bash
+az docdb cluster <command> -h
+```
+
+### Create a new cluster
+
+```bash
+az docdb cluster create --resource-group TestResourceGroup --cluster-name myCluster \
+    --location westus2 --administrator-name myAdmin --administrator-password password231 \
+    --server-version 5.0 --storage-size-gb 128 --compute-tier M30 \
+    --shard-count 1 --high-availability-mode ZoneRedundantPreferred
+```
+
+### Create a replica cluster
+```bash
+az docdb cluster replica create --resource-group TestResourceGroup \
+    --cluster-name myCluster --location eastus2 --source-location westus3 \
+    --source-resource mySourceCluster
+```
+
+### Promote a replica cluster
+```bash
+az docdb cluster replica promote --resource-group TestGroup \
+    --cluster-name replicaCluster --promote-option Forced \
+    --promote-mode Switchover
+```

src/docdb/azext_docdb/__init__.py

@@ -0,0 +1,42 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+#
+# Code generated by aaz-dev-tools
+# --------------------------------------------------------------------------------------------
+
+from azure.cli.core import AzCommandsLoader
+from azext_docdb._help import helps  # pylint: disable=unused-import
+
+
+class DocumentDbCommandsLoader(AzCommandsLoader):
+
+    def __init__(self, cli_ctx=None):
+        from azure.cli.core.commands import CliCommandType
+        custom_command_type = CliCommandType(
+            operations_tmpl='azext_docdb.custom#{}')
+        super().__init__(cli_ctx=cli_ctx,
+                         custom_command_type=custom_command_type)
+
+    def load_command_table(self, args):
+        from azext_docdb.commands import load_command_table
+        from azure.cli.core.aaz import load_aaz_command_table
+        try:
+            from . import aaz
+        except ImportError:
+            aaz = None
+        if aaz:
+            load_aaz_command_table(
+                loader=self,
+                aaz_pkg_name=aaz.__name__,
+                args=args
+            )
+        load_command_table(self, args)
+        return self.command_table
+
+    def load_arguments(self, command):
+        from azext_docdb._params import load_arguments
+        load_arguments(self, command)
+
+
+COMMAND_LOADER_CLS = DocumentDbCommandsLoader

src/docdb/azext_docdb/_help.py

@@ -0,0 +1,11 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+#
+# Code generated by aaz-dev-tools
+# --------------------------------------------------------------------------------------------
+
+# pylint: disable=line-too-long
+# pylint: disable=too-many-lines
+
+from knack.help_files import helps  # pylint: disable=unused-import

src/docdb/azext_docdb/_params.py

@@ -0,0 +1,13 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+#
+# Code generated by aaz-dev-tools
+# --------------------------------------------------------------------------------------------
+
+# pylint: disable=too-many-lines
+# pylint: disable=too-many-statements
+
+
+def load_arguments(self, _):  # pylint: disable=unused-argument
+    pass

src/docdb/azext_docdb/aaz/__init__.py

@@ -0,0 +1,6 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+#
+# Code generated by aaz-dev-tools
+# --------------------------------------------------------------------------------------------

src/docdb/azext_docdb/aaz/latest/__init__.py

@@ -0,0 +1,10 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+#
+# Code generated by aaz-dev-tools
+# --------------------------------------------------------------------------------------------
+
+# pylint: skip-file
+# flake8: noqa
+

src/docdb/azext_docdb/aaz/latest/docdb/__cmd_group.py

@@ -0,0 +1,24 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+#
+# Code generated by aaz-dev-tools
+# --------------------------------------------------------------------------------------------
+
+# pylint: skip-file
+# flake8: noqa
+
+from azure.cli.core.aaz import *
+
+
+@register_command_group(
+    "docdb",
+    is_preview=True,
+)
+class __CMDGroup(AAZCommandGroup):
+    """Manage Azure Database for Document DB resources.
+    """
+    pass
+
+
+__all__ = ["__CMDGroup"]

src/docdb/azext_docdb/aaz/latest/docdb/__init__.py

@@ -0,0 +1,11 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+#
+# Code generated by aaz-dev-tools
+# --------------------------------------------------------------------------------------------
+
+# pylint: skip-file
+# flake8: noqa
+
+from .__cmd_group import *

src/docdb/azext_docdb/aaz/latest/docdb/cluster/__cmd_group.py

@@ -0,0 +1,24 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+#
+# Code generated by aaz-dev-tools
+# --------------------------------------------------------------------------------------------
+
+# pylint: skip-file
+# flake8: noqa
+
+from azure.cli.core.aaz import *
+
+
+@register_command_group(
+    "docdb cluster",
+    is_preview=True,
+)
+class __CMDGroup(AAZCommandGroup):
+    """Manage Document DB cluster resources.
+    """
+    pass
+
+
+__all__ = ["__CMDGroup"]

src/docdb/azext_docdb/aaz/latest/docdb/cluster/__init__.py

@@ -0,0 +1,19 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+#
+# Code generated by aaz-dev-tools
+# --------------------------------------------------------------------------------------------
+
+# pylint: skip-file
+# flake8: noqa
+
+from .__cmd_group import *
+from ._check_name_availability import *
+from ._create import *
+from ._delete import *
+from ._list import *
+from ._list_connection_strings import *
+from ._show import *
+from ._update import *
+from ._wait import *