[Service Fabric] add sf cluster update command #4328

src/azure-cli/azure/cli/command_modules/servicefabric/_help.py

@@ -280,6 +280,32 @@
 
 """
 
+helps['sf cluster update'] = """
+type: command
+short-summary: Manage cluster updates.
+examples:
+  - name: Change the cluster durability level to 'Silver'.
+    text: >
+        az sf cluster update -g group-name -c cluster1 --durability-level Silver --node-type nt1
+
+  - name: Change the cluster reliability level to 'Silver'.
+    text: >
+        az sf cluster update -g group-name -c cluster1 --reliability-level Silver --auto-add-node
+
+  - name: Set the `MaxFileOperationTimeout` setting for a cluster to 5 seconds.
+    text: >
+        az sf cluster update -g group-name -c cluster1 --settings-section-set --section 'NamingService' --parameter 'MaxFileOperationTimeout' --value 5000
+
+  - name: Remove the `MaxFileOperationTimeout` setting from a cluster.
+    text: >
+        az sf cluster update -g group-name -c cluster1 --settings-section-rem --section 'NamingService' --parameter 'MaxFileOperationTimeout'
+
+  - name: Set a cluster to use the 'Automatic' upgrade mode.
+    text: >
+        az sf cluster update -g group-name -c cluster1 --upgrade-mode Automatic
+
+"""
+
 helps['sf cluster list'] = """
 type: command
 short-summary: List cluster resources.
@@ -306,7 +332,7 @@
 examples:
   - name: Remove 2 'nt1' nodes from a cluster.
     text: >
-        az sf cluster node remove -g group-name -c cluster1 --node-type 'nt1' --number-of-nodes-to-remove 2
+        az sf cluster node remove -g group-name -c cluster1 --number-of-nodes-to-rem 2 --node-type 'nt1'
 
 """
 
@@ -321,7 +347,7 @@
 examples:
   - name: Add a new node type to a cluster.
     text: >
-        az sf cluster node-type add -g group-name -c cluster1 --node-type 'n2' --capacity 5 --vm-user-name 'adminName' --vm-password testPassword0
+        az sf cluster node-type add -g group-name -c cluster1 --node-type 'n2' --capacity 5 --vm-user-name 'adminName' --vm-password testPassword0 --durability-level Bronze
 
 """
 

src/azure-cli/azure/cli/command_modules/servicefabric/_params.py

@@ -112,14 +112,6 @@ def load_arguments(self, _):  # pylint: disable=too-many-statements
     with self.argument_context('sf cluster client-certificate remove') as c:
         c.argument('thumbprints', nargs='+', help='A single or Space-separated list of client certificate thumbprint(s) to be remove.')
 
-    with self.argument_context('sf cluster node') as c:
-        c.argument('number_of_nodes_to_add', options_list=['--number-of-nodes-to-add', '--nodes-to-add'], help='number of nodes to add.')
-        c.argument('number_of_nodes_to_remove', options_list=['--number-of-nodes-to-remove', '--nodes-to-remove'], help='number of nodes to remove.')
-
-    with self.argument_context('sf cluster node-type') as c:
-        c.argument('capacity', help='The capacity tag applied to nodes in the node type. The cluster resource manager uses these tags to understand how much capacity a node has.')
-        c.argument('vm_tier', help='VM tier.')
-
     with self.argument_context('sf cluster') as c:
         c.argument('durability_level', arg_type=get_enum_type(['Bronze', 'Silver', 'Gold']), help='durability level.')
 
@@ -147,6 +139,29 @@ def load_arguments(self, _):  # pylint: disable=too-many-statements
                    help='JSON encoded parameters configuration. Use @{file} to load from a file. '
                         'For example: [{"section": "NamingService","parameter": "MaxOperationTimeout"}]')
 
+    with self.argument_context('sf cluster node') as c:
+        c.argument('number_of_nodes_to_add', options_list=['--number-of-nodes-to-add', '--nodes-to-add'], help='number of nodes to add.')
+        c.argument('number_of_nodes_to_remove', options_list=['--number-of-nodes-to-rem', '--nodes-to-remove'], help='number of nodes to remove.')
+
+    with self.argument_context('sf cluster node-type') as c:
+        c.argument('capacity', help='The capacity tag applied to nodes in the node type. The cluster resource manager uses these tags to understand how much capacity a node has.')
+        c.argument('vm_tier', help='VM tier.')
+
+    with self.argument_context('sf cluster update') as c:
+        c.argument('durability_level', arg_type=get_enum_type(['Bronze', 'Silver', 'Gold']), help='durability level.')
+        c.argument('node_type', help='Nodetype name')
+        c.argument('parameter', help='parameter name')
+        c.argument('section', help='section name')
+        c.argument('value', help='Specify the value')
+        c.argument('version', help='cluster code version')
+        c.argument('upgrade_mode', arg_type=get_enum_type(['manual', 'automatic']), help='cluster upgrade mode')
+        c.argument('reliability_level', arg_type=get_enum_type(['Bronze', 'Silver', 'Gold', 'Platinum']), help='Reliability level.')
+        c.argument('auto_add_node', arg_type=get_three_state_flag(), help='Add node count automatically when changing reliability.')
+        c.argument('settings_section_set', arg_type=get_three_state_flag(),
+                   help='Flag to set setting')
+        c.argument('settings_section_rem', arg_type=get_three_state_flag(),
+                   help='Flag to remove setting')
+
     with self.argument_context('sf cluster client-certificate remove') as c:
         c.argument('client_certificate_common_names', options_list=['--client-certificate-common-names', '--client-cert-cn'], type=get_json_object,
                    help='JSON encoded parameters configuration. Use @{file} to load from a file. '

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

@@ -59,6 +59,7 @@ def load_command_table(self, _):
         g.custom_command('create', 'new_cluster')
         g.custom_command('client-certificate add', 'add_client_cert')
         g.custom_command('client-certificate remove', 'remove_client_cert')
+        g.custom_command('update', 'update_cluster')
         g.custom_command('setting set', 'set_cluster_setting')
         g.custom_command('setting remove', 'remove_cluster_setting')
         g.custom_command('reliability update', 'update_cluster_reliability_level')

src/azure-cli/azure/cli/command_modules/servicefabric/custom.py

@@ -577,22 +577,27 @@ def set_cluster_setting(cmd,
                         parameter=None,
                         value=None,
                         settings_section_description=None):
+
     cli_ctx = cmd.cli_ctx
 
     def _set(setting_dict, section, parameter, value):
         if section not in setting_dict:
             setting_dict[section] = {}
         setting_dict[section][parameter] = value
         return setting_dict
+
     if settings_section_description and (section or parameter or value):
         raise CLIError(
             'Only can use either \'--settings-section-description\' or \'--section\', \'--parameter\' and \'--value\' to set the settings')
+
     if section or parameter or value:
         if section is None or parameter is None or value is None:
             raise CLIError(
                 '\'--section\' , \'--parameter\' and \'--value\' can not be None')
+
     cluster = client.get(resource_group_name, cluster_name)
     setting_dict = _fabric_settings_to_dict(cluster.fabric_settings)
+
     if settings_section_description:
         for setting in settings_section_description:
             if 'section' in setting and 'parameter' in setting and 'value' in setting:
@@ -602,6 +607,7 @@ def _set(setting_dict, section, parameter, value):
                 raise CLIError('settings_section_description is invalid')
     else:
         setting_dict = _set(setting_dict, section, parameter, value)
+
     settings = _dict_to_fabric_settings(setting_dict)
     patch_request = ClusterUpdateParameters(fabric_settings=settings)
     update_cluster_poll = client.begin_update(resource_group_name, cluster_name, patch_request)
@@ -630,8 +636,10 @@ def _remove(setting_dict, section, parameter):
     if settings_section_description and (section or parameter):
         raise CLIError(
             'Only can use either \'--settings-section-description\' or \'--section\' and \'--parameter \' to set the settings')
+
     cluster = client.get(resource_group_name, cluster_name)
     setting_dict = _fabric_settings_to_dict(cluster.fabric_settings)
+
     if settings_section_description:
         for setting in settings_section_description:
             if 'section' in setting and 'parameter' in setting:
@@ -682,6 +690,71 @@ def update_cluster_reliability_level(cmd,
     return LongRunningOperation(cli_ctx)(update_cluster_poll)
 
 
+def validate_arguments(*args):
+    for arg in args:
+        if arg is None or arg is False:
+            raise CLIError(
+                'Invalid argument {}, Please run command: `az sf cluster update -h` for help message and examples of how to use the command'.
+                format(arg))
+
+
+def validate_none_arguments(*args):
+    for arg in args:
+        if arg is not None and arg is True:
+            raise CLIError(
+                'Invalid argument {}, Please run command: `az sf cluster update -h` for help message and examples of how to use the command'.
+                format(arg))
+
+
+def update_cluster(cmd,
+                   client,
+                   resource_group_name,
+                   cluster_name,
+                   node_type=None,
+                   reliability_level=None,
+                   auto_add_node=False,
+                   section=None,
+                   parameter=None,
+                   settings_section_set=False,
+                   settings_section_rem=False,
+                   value=None,
+                   upgrade_mode=None,
+                   version=None,
+                   durability_level=None):
+
+    if settings_section_set is True:
+        validate_none_arguments(node_type, reliability_level, auto_add_node,
+                                settings_section_rem, upgrade_mode, version, durability_level)
+        validate_arguments(section, parameter, value)
+        return set_cluster_setting(cmd, client, resource_group_name, cluster_name, section, parameter, value)
+
+    if settings_section_rem is True:
+        validate_none_arguments(node_type, reliability_level, auto_add_node, value,
+                                settings_section_set, upgrade_mode, version, durability_level)
+        validate_arguments(section, parameter)
+        return remove_cluster_setting(cmd, client, resource_group_name, cluster_name, section, parameter)
+
+    if reliability_level is not None:
+        validate_none_arguments(node_type, section, parameter, settings_section_set,
+                                settings_section_rem, value, upgrade_mode, version, durability_level)
+        return update_cluster_reliability_level(cmd, client, resource_group_name,
+                                                cluster_name, reliability_level, auto_add_node)
+
+    if durability_level is not None:
+        validate_none_arguments(auto_add_node, reliability_level, section, parameter,
+                                settings_section_set, settings_section_rem, value, upgrade_mode, version)
+        validate_arguments(node_type)
+        return update_cluster_durability(cmd, client, resource_group_name, cluster_name, node_type, durability_level)
+
+    if upgrade_mode is not None:
+        validate_none_arguments(node_type, reliability_level, auto_add_node, section, parameter,
+                                settings_section_set, settings_section_rem, value, durability_level)
+        return update_cluster_upgrade_type(cmd, client, resource_group_name, cluster_name, upgrade_mode, version)
+
+    raise CLIError(
+        'Invalid arguments. Please run command: `az sf cluster update -h` for help message and examples of how to use the command')
+
+
 def add_cluster_node_type(cmd,
                           client,
                           resource_group_name,