cli extension for connected kubernetes RP

.github/CODEOWNERS

@@ -70,6 +70,8 @@
 
 /src/ip-group/ @haroldrandom
 
+/src/connectedk8s/ @akashkeshari
+
 /src/storagesync/ @jsntcy
 
 /src/maintenance/ @gautamd-ms

src/connectedk8s/HISTORY.rst

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

src/connectedk8s/README.md

@@ -0,0 +1,54 @@
+# Azure CLI Connected Kubernetes Extension #
+This package is for the 'connectedk8s' extension, i.e. 'az connectedk8s'.
+
+### How to use ###
+Install this extension using the below CLI command
+```
+az extension add --name connectedk8s
+```
+
+### Included Features
+#### Connected Kubernetes Management:
+*Examples:*
+
+##### Create a connected kubernetes cluster
+```
+az connectedk8s connect \
+    --subscription subscription_id \
+    --resource-group my-rg \
+    --name my-cluster \
+    --location eastus
+```
+
+##### Show connected kubernetes cluster
+```
+az connectedk8s show \
+    --subscription subscription_id \
+    --resource-group my-rg \
+    --name my-cluster \
+```
+or
+```
+az connectedk8s show \
+    --ids "/subscriptions/subscription_id/resourceGroups/my-rg/providers/Microsoft.Kubernetes/connectedClusters/my-cluster" \
+```
+
+##### List connected kubernetes cluster in resource group
+```
+az connectedk8s list \
+    --resource-group my-rg
+```
+
+##### Delete connected kubernetes cluster
+```
+az connectedk8s delete \
+    --subscription subscription_id \
+    --resource-group my-rg \
+    --name my-cluster \
+```
+or
+```
+az connectedk8s delete \
+    --ids "/subscriptions/subscription_id/resourceGroups/my-rg/providers/Microsoft.Kubernetes/connectedClusters/my-cluster" \
+    -y
+```

src/connectedk8s/azext_connectedk8s/__init__.py

@@ -0,0 +1,31 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+from azure.cli.core import AzCommandsLoader
+
+from azext_connectedk8s._help import helps  # pylint: disable=unused-import
+
+
+class Connectedk8sCommandsLoader(AzCommandsLoader):
+
+    def __init__(self, cli_ctx=None):
+        from azure.cli.core.commands import CliCommandType
+        from azext_connectedk8s._client_factory import cf_connectedk8s
+        connectedk8s_custom = CliCommandType(
+            operations_tmpl='azext_connectedk8s.custom#{}',
+            client_factory=cf_connectedk8s)
+        super(Connectedk8sCommandsLoader, self).__init__(cli_ctx=cli_ctx, custom_command_type=connectedk8s_custom)
+
+    def load_command_table(self, args):
+        from azext_connectedk8s.commands import load_command_table
+        load_command_table(self, args)
+        return self.command_table
+
+    def load_arguments(self, command):
+        from azext_connectedk8s._params import load_arguments
+        load_arguments(self, command)
+
+
+COMMAND_LOADER_CLS = Connectedk8sCommandsLoader

src/connectedk8s/azext_connectedk8s/_client_factory.py

@@ -0,0 +1,37 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+from azure.cli.core.commands.client_factory import get_mgmt_service_client
+from azure.cli.core.profiles import ResourceType
+from azure.cli.core._profile import Profile
+from azure.cli.core.commands.client_factory import configure_common_settings
+from azure.graphrbac import GraphRbacManagementClient
+
+
+def cf_connectedk8s(cli_ctx, *_):
+    from azext_connectedk8s.vendored_sdks import KubernetesConnectRPClient
+    return get_mgmt_service_client(cli_ctx, KubernetesConnectRPClient)
+
+
+def cf_connected_cluster(cli_ctx, _):
+    return cf_connectedk8s(cli_ctx).connected_cluster
+
+
+def cf_resource_groups(cli_ctx, subscription_id=None):
+    return get_mgmt_service_client(cli_ctx, ResourceType.MGMT_RESOURCE_RESOURCES,
+                                   subscription_id=subscription_id).resource_groups
+
+
+def _resource_client_factory(cli_ctx, subscription_id=None):
+    return get_mgmt_service_client(cli_ctx, ResourceType.MGMT_RESOURCE_RESOURCES, subscription_id=subscription_id)
+
+
+def _graph_client_factory(cli_ctx, **_):
+    profile = Profile(cli_ctx=cli_ctx)
+    cred, _, tenant_id = profile.get_login_credentials(
+        resource=cli_ctx.cloud.endpoints.active_directory_graph_resource_id)
+    client = GraphRbacManagementClient(cred, tenant_id,
+                                       base_url=cli_ctx.cloud.endpoints.active_directory_graph_resource_id)
+    configure_common_settings(cli_ctx, client)
+    return client

src/connectedk8s/azext_connectedk8s/_format.py

@@ -0,0 +1,37 @@
+# --------------------------------------------------------------------------------------------
+# 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  # pylint: disable=import-error
+
+
+def connectedk8s_show_table_format(result):
+    """Format a connected cluster as summary results for display with "-o table"."""
+    return [_connectedk8s_table_format(result)]
+
+
+def connectedk8s_list_table_format(results):
+    """Format an connected cluster list for display with "-o table"."""
+    return [_connectedk8s_list_table_format(r) for r in results]
+
+
+def _connectedk8s_table_format(result):
+    parsed = compile_jmes("""{
+        name: name,
+        location: location,
+        resourceGroup: resourceGroup
+    }""")
+    # use ordered dicts so headers are predictable
+    return parsed.search(result, Options(dict_cls=OrderedDict))
+
+
+def _connectedk8s_list_table_format(result):
+    parsed = compile_jmes("""{
+        name: name,
+        location: location,
+        resourceGroup: resourceGroup
+    }""")
+    # use ordered dicts so headers are predictable
+    return parsed.search(result, Options(dict_cls=OrderedDict))

src/connectedk8s/azext_connectedk8s/_help.py

@@ -0,0 +1,52 @@
+# coding=utf-8
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+from knack.help_files import helps  # pylint: disable=unused-import
+
+
+helps['connectedk8s'] = """
+    type: group
+    short-summary: Commands to manage connected kubernetes clusters.
+"""
+
+helps['connectedk8s connect'] = """
+    type: command
+    short-summary: Onboard a connected kubernetes cluster to azure.
+    examples:
+    - name: Onboard a connected kubernetes cluster with default kube config and kube context.
+      text: az connectedk8s connect -g resourceGroupName -n connectedClusterName
+    - name: Onboard a connected kubernetes cluster by specifying the kubeconfig and kubecontext.
+      text: az connectedk8s connect -g resourceGroupName -n connectedClusterName --kube-config /path/to/kubeconfig --kube-context kubeContextName
+"""
+
+helps['connectedk8s list'] = """
+    type: command
+    short-summary: List connected kubernetes clusters.
+    examples:
+    - name: List all connected kubernetes clusters in a resource group.
+      text: az connectedk8s list -g resourceGroupName --subscription subscriptionName
+    - name: List all connected kubernetes clusters in a subscription.
+      text: az connectedk8s list --subscription subscriptionName
+
+"""
+
+helps['connectedk8s delete'] = """
+    type: command
+    short-summary: Delete a connected kubernetes cluster along with connected cluster agents.
+    examples:
+    - name: Delete a connected kubernetes cluster and connected cluster agents with default kubeconfig and kubecontext.
+      text: az connectedk8s delete -g resourceGroupName -n connectedClusterName
+    - name: Delete a connected kubernetes cluster by specifying the kubeconfig and kubecontext for connected cluster agents deletion.
+      text: az connectedk8s delete -g resourceGroupName -n connectedClusterName --kube-config /path/to/kubeconfig --kube-context kubeContextName
+"""
+
+helps['connectedk8s show'] = """
+    type: command
+    short-summary: Show details of a connected kubernetes cluster.
+    examples:
+    - name: Show the details for a connected kubernetes cluster
+      text: az connectedk8s show -g resourceGroupName -n connectedClusterName
+"""

src/connectedk8s/azext_connectedk8s/_params.py

@@ -0,0 +1,31 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+# pylint: disable=line-too-long
+
+from azure.cli.core.commands.parameters import get_location_type
+from azure.cli.core.commands.validators import get_default_location_from_resource_group
+
+
+def load_arguments(self, _):
+
+    from azure.cli.core.commands.parameters import tags_type
+
+    with self.argument_context('connectedk8s connect') as c:
+        c.argument('tags', tags_type)
+        c.argument('location', arg_type=get_location_type(self.cli_ctx), validator=get_default_location_from_resource_group)
+        c.argument('cluster_name', options_list=['--name', '-n'], help='The name of the connected cluster.')
+        c.argument('kube_config', options_list=['--kube-config'], help='Path to the kube config file.')
+        c.argument('kube_context', options_list=['--kube-context'], help='Kubconfig context from current machine.')
+
+    with self.argument_context('connectedk8s list') as c:
+        pass
+
+    with self.argument_context('connectedk8s show') as c:
+        c.argument('cluster_name', options_list=['--name', '-n'], id_part='name', help='The name of the connected cluster.')
+
+    with self.argument_context('connectedk8s delete') as c:
+        c.argument('cluster_name', options_list=['--name', '-n'], id_part='name', help='The name of the connected cluster.')
+        c.argument('kube_config', options_list=['--kube-config'], help='Path to the kube config file.')
+        c.argument('kube_context', options_list=['--kube-context'], help='Kubconfig context from current machine.')

src/connectedk8s/azext_connectedk8s/_validators.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.
+# --------------------------------------------------------------------------------------------
+
+
+def example_name_or_id_validator(cmd, namespace):
+    # Example of a storage account name or ID validator.
+    from azure.cli.core.commands.client_factory import get_subscription_id
+    from msrestazure.tools import is_valid_resource_id, resource_id
+    if namespace.storage_account:
+        if not is_valid_resource_id(namespace.RESOURCE):
+            namespace.storage_account = resource_id(
+                subscription=get_subscription_id(cmd.cli_ctx),
+                resource_group=namespace.resource_group_name,
+                namespace='Microsoft.Storage',
+                type='storageAccounts',
+                name=namespace.storage_account
+            )

src/connectedk8s/azext_connectedk8s/azext_metadata.json

@@ -0,0 +1,4 @@
+{
+    "azext.isPreview": true,
+    "azext.minCliCoreVersion": "2.0.67"
+}

src/connectedk8s/azext_connectedk8s/commands.py

@@ -0,0 +1,27 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+# pylint: disable=line-too-long
+from azure.cli.core.commands import CliCommandType
+from azext_connectedk8s._client_factory import (cf_connectedk8s, cf_connected_cluster)
+from ._format import connectedk8s_show_table_format
+from ._format import connectedk8s_list_table_format
+
+
+def load_command_table(self, _):
+
+    connectedk8s_sdk = CliCommandType(
+        operations_tmpl='azext_connectedk8s.vendored_sdks.operations#ConnectedClusterOperations.{}',
+        client_factory=cf_connectedk8s
+    )
+
+    with self.command_group('connectedk8s', connectedk8s_sdk, client_factory=cf_connected_cluster) as g:
+        g.custom_command('connect', 'create_connectedk8s', supports_no_wait=True)
+        g.custom_command('delete', 'delete_connectedk8s', confirmation=True, supports_no_wait=True)
+        g.custom_command('list', 'list_connectedk8s', table_transformer=connectedk8s_list_table_format)
+        g.custom_show_command('show', 'get_connectedk8s', table_transformer=connectedk8s_show_table_format)
+
+    with self.command_group('connectedk8s', is_preview=True):
+        pass