AEM: initial change for Azure Enhanced Monitoring Extension for SAP (#41)

* initial code

* copy code over

* wip

* fix a few lint errors

* address review feedback

* add code owner

* fix lint error

* revert unnecessary script change

* fix an incorrect usage of get_models

Author: yugangw-msft

.github/CODEOWNERS

@@ -8,4 +8,6 @@
 
 /src/eventhubs/ @v-ajnava
 
-/src/webapps/ @panchagnula
+/src/webapps/ @panchagnula
+
+/src/aem/ @yugangw-msft

src/aem/azext_aem/__init__.py

@@ -0,0 +1,48 @@
+# --------------------------------------------------------------------------------------------
+# 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
+
+import azext_aem._help  # pylint: disable=unused-import
+
+
+class AEMCommandsLoader(AzCommandsLoader):
+
+    def __init__(self, cli_ctx=None):
+        from azure.cli.core.commands import CliCommandType
+        aem_custom = CliCommandType(
+            operations_tmpl='azext_aem.custom#{}')
+        super(AEMCommandsLoader, self).__init__(cli_ctx=cli_ctx,
+                                                custom_command_type=aem_custom)
+
+    def load_command_table(self, _):
+        with self.command_group('vm aem', min_api='2016-04-30-preview') as g:
+            g.custom_command('set', 'set_aem')
+            g.custom_command('delete', 'delete_aem')
+            g.custom_command('verify', 'verify_aem')
+
+        return self.command_table
+
+    def load_arguments(self, _):
+        # pylint: disable=line-too-long
+        from knack.arguments import CLIArgumentType
+        from azure.cli.core.commands.parameters import get_resource_name_completion_list
+        name_arg_type = CLIArgumentType(options_list=['--name', '-n'], metavar='NAME')
+        existing_vm_name = CLIArgumentType(overrides=name_arg_type,
+                                           configured_default='vm',
+                                           help="The name of the Virtual Machine. You can configure the default using `az configure --defaults vm=<name>`",
+                                           completer=get_resource_name_completion_list('Microsoft.Compute/virtualMachines'), id_part='name')
+
+        with self.argument_context('vm aem') as c:
+            c.argument('vm_name', existing_vm_name)
+            c.argument('skip_storage_check', action='store_true',
+                       help='Disables the test for table content')
+            c.argument('skip_storage_analytics', action='store_true',
+                       help='skip enabling analytics on storage accounts')
+            c.argument('wait_time_in_minutes', type=int,
+                       help='Maximum minutes to wait for the storage metrics to be available')
+
+
+COMMAND_LOADER_CLS = AEMCommandsLoader

src/aem/azext_aem/_help.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.
+# --------------------------------------------------------------------------------------------
+
+from knack.help_files import helps
+
+helps['vm aem'] = """
+    type: group
+    short-summary: Manage Azure Enhanced Monitoring Extension for SAP
+"""
+
+helps['vm aem set'] = """
+    type: command
+    short-summary: Configure Azure Enhanced Monitoring Extension
+    long-summary: It can take up to 15 minutes for the monitoring data to appear in the SAP system
+"""
+
+helps['vm aem delete'] = """
+    type: command
+    short-summary: Remove Azure Enhanced Monitoring Extension
+"""
+
+helps['vm aem verify'] = """
+    type: command
+    short-summary: Verify Azure Enhanced Monitoring Extensions configured correctly
+"""

src/aem/azext_aem/azext_metadata.json

@@ -0,0 +1,3 @@
+{
+    "azext.minCliCoreVersion": "2.0.24"     
+}

src/aem/azext_aem/custom.py

@@ -0,0 +1,4 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------

src/aem/azext_aem/tests/__init__.py

@@ -0,0 +1,22 @@
+# --------------------------------------------------------------------------------------------
+# 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.testsdk import (ScenarioTest, ResourceGroupPreparer)
+
+# pylint: disable=unused-argument,too-few-public-methods
+
+
+class VMAEM(ScenarioTest):
+
+    @ResourceGroupPreparer()
+    def test_vm_aem_configure(self, resource_group):
+        self.kwargs.update({
+            'vm': 'vm1',
+        })
+        self.cmd('vm create -g {rg} -n {vm} --image centos')
+        self.cmd('vm aem set -g {rg} -n {vm}')
+        self.cmd('vm aem verify -g {rg} -n {vm}')
+        self.cmd('vm aem delete -g {rg} -n {vm}')
+        self.cmd('vm aem verify -g {rg} -n {vm}')

src/aem/azext_aem/tests/recordings/latest/test_vm_aem_configure.yaml

@@ -0,0 +1,13 @@
+# Azure CLI Enhanced Monitoring Extension #
+This is an extension to azure cli which provides commands to configure, verify and remove Azure Enhanced Monitoring Extension for SAP 
+
+## How to use ##
+First, install the extension:
+```
+az extension add --name aem
+```
+
+Then, call it as you would any other az command:
+```
+az vm aem set --resource-group rg --name vm1
+```

src/aem/azext_aem/tests/test_aem_commands.py

@@ -0,0 +1,2 @@
+[bdist_wheel]
+universal=1