feat: add appnet-preview extension

Author: German Robayo Paz

src/appnet-preview/HISTORY.rst

@@ -0,0 +1,8 @@
+.. :changelog:
+
+Release History
+===============
+
+1.0.0b1
+++++++
+* Initial release.

src/appnet-preview/README.md

@@ -0,0 +1,54 @@
+# Azure CLI AppnetPreview Extension #
+This is an extension to Azure CLI to manage Application Network resources.
+
+## How to use ##
+
+### Install the extension
+```bash
+az extension add --name appnet-preview
+```
+
+### List available Application Network versions
+```bash
+az appnet list-versions --location <location>
+```
+
+### Create an Application Network
+```bash
+az appnet create --resource-group <resource-group> --appnet-name <name> --location <location>
+```
+
+### Join an AKS cluster to an Application Network
+
+**Prerequisites:** You must have an existing AKS cluster before running `az appnet member join`.
+
+```bash
+# Create an AKS cluster first (if you don't have one)
+az aks create --resource-group <resource-group> --name <aks-name> --location <location>
+
+# Get the AKS cluster resource ID
+AKS_ID=$(az aks show --resource-group <resource-group> --name <aks-name> --query id -o tsv)
+
+# Join the AKS cluster to the Application Network
+az appnet member join --resource-group <resource-group> --appnet-name <appnet-name> \
+    --member-name <member-name> --cluster-type AKS --member-resource-id $AKS_ID \
+    --upgrade-mode FullyManaged --release-channel Stable
+```
+
+### Other commands
+```bash
+# List Application Networks
+az appnet list --resource-group <resource-group>
+
+# Show an Application Network
+az appnet show --resource-group <resource-group> --name <appnet-name>
+
+# List members in an Application Network
+az appnet member list --resource-group <resource-group> --appnet-name <appnet-name>
+
+# Remove a member from an Application Network
+az appnet member remove --resource-group <resource-group> --appnet-name <appnet-name> --member-name <member-name>
+
+# Delete an Application Network
+az appnet delete --resource-group <resource-group> --name <appnet-name>
+```

src/appnet-preview/azext_appnet_preview/__init__.py

@@ -0,0 +1,66 @@
+# --------------------------------------------------------------------------------------------
+# 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_appnet_preview._help import helps  # pylint: disable=unused-import
+
+
+class AppnetPreviewCommandsLoader(AzCommandsLoader):
+
+    def __init__(self, cli_ctx=None):
+        from azure.cli.core.commands import CliCommandType
+        custom_command_type = CliCommandType(
+            operations_tmpl='azext_appnet_preview.custom#{}')
+        super().__init__(cli_ctx=cli_ctx,
+                         custom_command_type=custom_command_type)
+
+    def load_command_table(self, args):
+        from azext_appnet_preview.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)
+
+        # Add table transformers to AAZ commands
+        self._add_table_transformers()
+
+        return self.command_table
+
+    def _add_table_transformers(self):
+        """Add table transformers to AAZ-generated commands."""
+        from azext_appnet_preview._format import (
+            appnet_list_versions_table_format,
+            appnet_member_list_table_format,
+            appnet_member_upgrade_history_table_format
+        )
+
+        # Map command names to their table transformers
+        table_transformers = {
+            'appnet list-versions': appnet_list_versions_table_format,
+            'appnet member list': appnet_member_list_table_format,
+            'appnet member upgrade-history list': appnet_member_upgrade_history_table_format,
+        }
+
+        # Apply table transformers to commands
+        for command_name, transformer in table_transformers.items():
+            if command_name in self.command_table:
+                self.command_table[command_name].table_transformer = transformer
+
+    def load_arguments(self, command):
+        from azext_appnet_preview._params import load_arguments
+        load_arguments(self, command)
+
+
+COMMAND_LOADER_CLS = AppnetPreviewCommandsLoader

src/appnet-preview/azext_appnet_preview/_format.py

@@ -0,0 +1,96 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+from collections import OrderedDict
+from jmespath import compile as compile_jmes, Options
+
+
+def appnet_list_versions_table_format(results):
+    """Format appnet list-versions results for display with "-o table"."""
+    formatted_results = []
+    for result in results:
+        # Extract the main info
+        base_info = {
+            'kubernetesVersion': result.get('properties', {}).get('kubernetesVersion', 'N/A'),
+        }
+
+        # Get fully managed versions
+        fully_managed = result.get('properties', {}).get('fullyManagedVersions', {})
+        if fully_managed and fully_managed.get('releaseChannels'):
+            for channel in fully_managed['releaseChannels']:
+                formatted_results.append({
+                    'kubernetesVersion': base_info['kubernetesVersion'],
+                    'mode': 'FullyManaged',
+                    'releaseChannel': channel.get('releaseChannel', 'N/A'),
+                    'version': channel.get('version', 'N/A'),
+                })
+
+        # Get self managed versions
+        self_managed = result.get('properties', {}).get('selfManagedVersions', {})
+        if self_managed and self_managed.get('versions'):
+            for version_info in self_managed['versions']:
+                upgrades_str = ', '.join(version_info.get('upgrades', []))
+                formatted_results.append({
+                    'kubernetesVersion': base_info['kubernetesVersion'],
+                    'mode': 'SelfManaged',
+                    'releaseChannel': 'N/A',
+                    'version': version_info.get('version', 'N/A'),
+                    'availableUpgrades': upgrades_str if upgrades_str else 'None',
+                })
+
+    return formatted_results
+
+
+def appnet_member_list_table_format(results):
+    """Format appnet member list results for display with "-o table"."""
+    return [_appnet_member_show_table_format(r) for r in results]
+
+
+def _appnet_member_show_table_format(result):
+    """Format a single appnet member as summary results for display with "-o table"."""
+    # Extract resource group from ID
+    resource_id = result.get('id', '')
+    resource_group = 'N/A'
+    if resource_id:
+        parts = resource_id.split('/resourceGroups/')
+        if len(parts) > 1:
+            resource_group = parts[1].split('/')[0]
+
+    parsed = compile_jmes("""{
+        name: name,
+        location: location,
+        clusterType: properties.clusterType,
+        mode: properties.mode,
+        provisioningState: properties.provisioningState,
+        clusterResourceId: properties.metadata.resourceId
+    }""")
+
+    result_dict = parsed.search(result, Options(dict_cls=OrderedDict))
+    result_dict['resourceGroup'] = resource_group
+
+    # Shorten cluster resource ID for readability
+    if result_dict.get('clusterResourceId'):
+        result_dict['clusterName'] = result_dict['clusterResourceId'].split('/')[-1]
+        del result_dict['clusterResourceId']
+
+    return result_dict
+
+
+def appnet_member_upgrade_history_table_format(results):
+    """Format appnet member upgrade history results for display with "-o table"."""
+    formatted_results = []
+    for result in results:
+        parsed = compile_jmes("""{
+            name: name,
+            upgradeState: properties.upgradeState,
+            fromVersion: properties.fromVersion,
+            toVersion: properties.toVersion,
+            startTime: properties.startTime,
+            endTime: properties.endTime,
+            upgradeType: properties.upgradeType
+        }""")
+        formatted_results.append(parsed.search(result, Options(dict_cls=OrderedDict)))
+
+    return formatted_results

src/appnet-preview/azext_appnet_preview/_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/appnet-preview/azext_appnet_preview/_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/appnet-preview/azext_appnet_preview/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/appnet-preview/azext_appnet_preview/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/appnet-preview/azext_appnet_preview/aaz/latest/appnet/__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(
+    "appnet",
+    is_preview=True,
+)
+class __CMDGroup(AAZCommandGroup):
+    """Azure Kubernetes Application Network
+    """
+    pass
+
+
+__all__ = ["__CMDGroup"]

src/appnet-preview/azext_appnet_preview/aaz/latest/appnet/__init__.py

@@ -0,0 +1,18 @@
+# --------------------------------------------------------------------------------------------
+# 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 ._create import *
+from ._delete import *
+from ._list import *
+from ._list_versions import *
+from ._show import *
+from ._update import *
+from ._wait import *