Implementation for CLI for Server DNS Aliasing (#4949)

Author: maboja-msft

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

@@ -3,6 +3,10 @@
 Release History
 ===============
 
+2.0.21
+++++++
+* Added az sql server dns-alias commands.
+
 2.0.20
 ++++++
 * Added az sql db rename

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

@@ -300,6 +300,14 @@
     type: command
     short-summary: Updates a server's secure connection policy.
     """
+helps['sql server dns-alias'] = """
+    type: group
+    short-summary: Manage a server's DNS aliases.
+    """
+helps['sql server dns-alias set'] = """
+    type: command
+    short-summary: Sets a server to which DNS alias should point
+    """
 helps['sql server firewall-rule'] = """
     type: group
     short-summary: Manage a server's firewall rules.

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

@@ -654,6 +654,17 @@ def _configure_security_policy_storage_params(arg_ctx):
         c.argument('connection_type', options_list=['--connection-type', '-t'],
                    arg_type=get_enum_type(ServerConnectionType))
 
+    #####
+    #           sql server dns-alias
+    #####
+    with self.argument_context('sql server dns-alias') as c:
+        c.argument('server_name', arg_type=server_param_type)
+        c.argument('dns_alias_name', options_list=('--name', '-n'))
+        c.argument('original_server_name', options_list=('--original-server'),
+                   help='The name of the server to which alias is currently pointing')
+        c.argument('original_resource_group_name', options_list=('--original-resource-group'))
+        c.argument('original_subscription_id', options_list=('--original-subscription-id'))
+
     #####
     #           sql server firewall-rule
     #####

src/command_modules/azure-cli-sql/azure/cli/command_modules/sql/_util.py

@@ -88,6 +88,10 @@ def get_sql_server_connection_policies_operations(cli_ctx, _):
     return get_sql_management_client(cli_ctx).server_connection_policies
 
 
+def get_sql_server_dns_aliases_operations(cli_ctx, _):
+    return get_sql_management_client(cli_ctx).server_dns_aliases
+
+
 def get_sql_server_keys_operations(cli_ctx, _):
     return get_sql_management_client(cli_ctx).server_keys
 

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

@@ -20,6 +20,7 @@
     get_sql_replication_links_operations,
     get_sql_restorable_dropped_databases_operations,
     get_sql_server_connection_policies_operations,
+    get_sql_server_dns_aliases_operations,
     get_sql_server_keys_operations,
     get_sql_servers_operations,
     get_sql_server_usages_operations,
@@ -230,3 +231,13 @@ def load_command_table(self, _):
     with self.command_group('sql server conn-policy', server_connection_policies_operations, client_factory=get_sql_server_connection_policies_operations) as c:
         c.command('show', 'get')
         c.generic_update_command('update')
+
+    server_dns_aliases_operations = CliCommandType(
+        operations_tmpl='azure.mgmt.sql.operations.server_dns_aliases_operations#ServerDnsAliasesOperations.{}',
+        client_factory=get_sql_server_dns_aliases_operations)
+    with self.command_group('sql server dns-alias', server_dns_aliases_operations, client_factory=get_sql_server_dns_aliases_operations) as c:
+        c.command('show', 'get')
+        c.command('list', 'list_by_server')
+        c.command('create', 'create_or_update')
+        c.command('delete', 'delete')
+        c.custom_command('set', 'server_dns_alias_set')

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

@@ -1214,6 +1214,35 @@ def _get_server_key_name_from_uri(uri):
     return '{}_{}_{}'.format(vault, key, version)
 
 
+#####
+#           sql server dns-alias
+#####
+
+
+def server_dns_alias_set(
+        cmd,
+        client,
+        resource_group_name,
+        server_name,
+        dns_alias_name,
+        original_server_name,
+        original_subscription_id=None,
+        original_resource_group_name=None):
+
+    # Build the old alias id
+    old_alias_id = "/subscriptions/{}/resourceGroups/{}/providers/Microsoft.Sql/servers/{}/dnsAliases/{}".format(
+        quote(original_subscription_id or get_subscription_id(cmd.cli_ctx)),
+        quote(original_resource_group_name or resource_group_name),
+        quote(original_server_name),
+        quote(dns_alias_name))
+
+    return client.acquire(
+        resource_group_name=resource_group_name,
+        server_name=server_name,
+        dns_alias_name=dns_alias_name,
+        old_server_dns_alias_id=old_alias_id
+    )
+
 #####
 #           sql server encryption-protector
 #####