[DevTestLabs] Exposing commands to manage secrets in the lab

src/command_modules/azure-cli-lab/HISTORY.rst

@@ -3,6 +3,10 @@
 Release History
 ===============
 
++++++++++++++++++++++
+
+* Adding commands to manage secrets in the Azure DevTest Lab.
+
 0.0.1 (2017-04-03)
 +++++++++++++++++++++
 

src/command_modules/azure-cli-lab/azure/cli/command_modules/lab/_client_factory.py

@@ -41,3 +41,7 @@ def get_devtestlabs_virtual_network_operation(kwargs):
 
 def get_devtestlabs_formula_operation(kwargs):
     return get_devtestlabs_management_client(kwargs).formula
+
+
+def get_devtestlabs_secret_operation(kwargs):
+    return get_devtestlabs_management_client(kwargs).secret

src/command_modules/azure-cli-lab/azure/cli/command_modules/lab/commands.py

@@ -10,7 +10,8 @@
                               get_devtestlabs_artifact_source_operation,
                               get_devtestlabs_lab_operation,
                               get_devtestlabs_virtual_network_operation,
-                              get_devtestlabs_formula_operation)
+                              get_devtestlabs_formula_operation,
+                              get_devtestlabs_secret_operation)
 from azure.cli.core.sdk.util import (ServiceGroup, create_service_adapter)
 
 
@@ -123,3 +124,17 @@
         c.command('get', 'get_resource')
         c.command('list', 'list')
         c.command('delete', 'delete_resource')
+
+
+# Secret Operations Commands
+secret_operations = create_service_adapter(
+    mgmt_operations_path.format('secret_operations'),
+    'SecretOperations')
+
+with ServiceGroup(__name__, get_devtestlabs_secret_operation,
+                  secret_operations) as s:
+    with s.group('lab secret') as c:
+        c.command('set', 'create_or_update_resource')
+        c.command('show', 'get_resource')
+        c.command('list', 'list')
+        c.command('delete', 'delete_resource')

src/command_modules/azure-cli-lab/azure/cli/command_modules/lab/help.py

@@ -147,3 +147,18 @@
             type: group
             short-summary: Commands to manage formulas of a DevTest Lab.
             """
+helps['lab secret'] = """
+            type: group
+            short-summary: Commands to manage secrets of a DevTest Lab.
+            """
+helps['lab secret set'] = """
+            type: command
+            short-summary: Sets a secret in the DevTest Lab.
+            parameters:
+                - name: --lab-name
+                  short-summary: Name of the Lab
+                - name: --name -n
+                  short-summary: Name of the secret
+                - name: --value
+                  short-summary: Value of the secret
+            """

src/command_modules/azure-cli-lab/azure/cli/command_modules/lab/params.py

@@ -5,7 +5,8 @@
 
 import json
 from azure.cli.command_modules.lab.validators import (validate_lab_vm_create,
-                                                      validate_lab_vm_list)
+                                                      validate_lab_vm_list,
+                                                      validate_user_name)
 from azure.cli.core.commands.parameters import resource_group_name_type
 from azure.cli.core.sdk.util import ParametersContext
 
@@ -69,3 +70,12 @@
 
 with ParametersContext(command='lab formula') as c:
     c.register_alias('name', ('--name', '-n'))
+
+
+with ParametersContext(command='lab secret') as c:
+    from .sdk.devtestlabs.models.secret import Secret
+
+    c.register_alias('name', ('--name', '-n'))
+    c.register_alias('secret', ('--value', ), type=lambda x: Secret(value=x))
+    c.ignore('user_name')
+    c.argument('lab_name', validator=validate_user_name)

src/command_modules/azure-cli-lab/azure/cli/command_modules/lab/validators.py

@@ -80,6 +80,10 @@ def validate_lab_vm_list(namespace):
         namespace.filters = "Properties/ownerObjectId eq '{}'".format(object_id)
 
 
+def validate_user_name(namespace):
+    namespace.user_name = "@me"
+
+
 # pylint: disable=no-member
 def _validate_location(namespace):
     """