[SQL] az sql midb log-replay: Support for log replay service on managed database

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

@@ -895,6 +895,45 @@
 short-summary: Place the CLI in a waiting state until a condition of the managed database is met.
 """
 
+helps['sql midb log-replay'] = """
+type: group
+short-summary: SQL Managed Instance database Log Replay service commands.
+"""
+
+helps['sql midb log-replay start'] = """
+type: command
+short-summary: Starts Log Replay service on specified database.
+examples:
+  - name: Log Replay with auto complete option.
+    text: az sql midb log-replay start -g mygroup --mi myinstance -n mymanageddb -a --last-bn "backup.bak" --storage-uri "https://test.blob.core.windows.net/testing" --storage-sas "sv=2019-02-02&ss=b&srt=sco&sp=rl&se=2023-12-02T00:09:14Z&st=2019-11-25T16:09:14Z&spr=https&sig=92kAe4QYmXaht%2Fgjocqwerqwer41s%3D"
+  - name: Log Replay without auto complete option.
+    text: az sql midb log-replay start -g mygroup --mi myinstance -n mymanageddb --storage-uri "https://test.blob.core.windows.net/testing" --storage-sas "sv=2019-02-02&ss=b&srt=sco&sp=rl&se=2023-12-02T00:09:14Z&st=2019-11-25T16:09:14Z&spr=https&sig=92kAe4QYmXaht%2Fgjocqwerqwer41s%3D"
+"""
+
+helps['sql midb log-replay complete'] = """
+type: command
+short-summary: Completes Log Replay service on specified database.
+examples:
+  - name: Completes log replay service.
+    text: az sql midb log-replay complete -g mygroup --mi myinstance -n mymanageddb --last-backup-name "backup.bak"
+"""
+
+helps['sql midb log-replay show'] = """
+type: command
+short-summary: Gets status of Log Replay service.
+examples:
+  - name: Gets status of the ongoing log replay service.
+    text: az sql midb log-replay show -g mygroup --mi myinstance -n mymanageddb
+"""
+
+helps['sql midb log-replay stop'] = """
+type: command
+short-summary: Stops Log Replay service.
+examples:
+  - name: Stops ongoing log replay service by deleting database.
+    text: az sql midb log-replay stop -g mygroup --mi myinstance -n mymanageddb
+"""
+
 helps['sql server'] = """
 type: group
 short-summary: Manage SQL servers.

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

@@ -58,8 +58,7 @@
     ElasticPoolCapabilitiesAdditionalDetails,
     FailoverPolicyType,
     SqlServerMinimalTlsVersionType,
-    SqlManagedInstanceMinimalTlsVersionType,
-    AuthenticationType
+    SqlManagedInstanceMinimalTlsVersionType
 )
 
 from ._validators import (
@@ -654,8 +653,7 @@ def load_arguments(self, _):
                    options_list=['--admin-password', '-p'])
 
         c.argument('authentication_type',
-                   options_list=['--auth-type', '-a'],
-                   arg_type=get_enum_type(AuthenticationType))
+                   options_list=['--auth-type', '-a'])
 
         c.argument('storage_key_type',
                    arg_type=get_enum_type(StorageKeyType))
@@ -678,8 +676,7 @@ def load_arguments(self, _):
                    options_list=['--admin-password', '-p'])
 
         c.argument('authentication_type',
-                   options_list=['--auth-type', '-a'],
-                   arg_type=get_enum_type(AuthenticationType))
+                   options_list=['--auth-type', '-a'])
 
         c.argument('storage_key_type',
                    arg_type=get_enum_type(StorageKeyType))
@@ -1839,6 +1836,47 @@ def _configure_security_policy_storage_params(arg_ctx):
                    help='The resource id of the long term retention backup to be restored. '
                    'Use \'az sql midb ltr-backup show\' or \'az sql midb ltr-backup list\' for backup id.')
 
+    with self.argument_context('sql midb log-replay start') as c:
+        create_args_for_complex_type(
+            c, 'parameters', ManagedDatabase, [
+                'auto_complete',
+                'last_backup_name',
+                'storage_container_uri',
+                'storage_container_sas_token'
+            ])
+
+        c.argument('auto_complete',
+                   required=False,
+                   options_list=['--auto-complete', '-a'],
+                   action='store_true',
+                   help='The flag that in usage with last_backup_name automatically completes log replay servise.')
+
+        c.argument('last_backup_name',
+                   required=False,
+                   options_list=['--last-backup-name', '--last-bn'],
+                   help='The name of the last backup to restore.')
+
+        c.argument('storage_container_uri',
+                   required=True,
+                   options_list=['--storage-uri', '--su'],
+                   help='The URI of the storage container where backups are.')
+
+        c.argument('storage_container_sas_token',
+                   required=True,
+                   options_list=['--storage-sas', '--ss'],
+                   help='The authorization Sas token to access storage container where backups are.')
+
+    with self.argument_context('sql midb log-replay complete') as c:
+        create_args_for_complex_type(
+            c, 'parameters', ManagedDatabase, [
+                'last_backup_name'
+            ])
+
+        c.argument('last_backup_name',
+                   required=False,
+                   options_list=['--last-backup-name', '--last-bn'],
+                   help='The name of the last backup to restore.')
+
     ###############################################
     #                sql virtual cluster          #
     ###############################################

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

@@ -194,3 +194,7 @@ def get_sql_database_long_term_retention_policies_operations(cli_ctx, _):
 
 def get_sql_database_long_term_retention_backups_operations(cli_ctx, _):
     return get_sql_management_client(cli_ctx).long_term_retention_backups
+
+
+def get_sql_managed_database_restore_details_operations(cli_ctx, _):
+    return get_sql_management_client(cli_ctx).managed_database_restore_details

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

@@ -42,6 +42,7 @@
     get_sql_firewall_rules_operations,
     get_sql_instance_pools_operations,
     get_sql_managed_databases_operations,
+    get_sql_managed_database_restore_details_operations,
     get_sql_managed_backup_short_term_retention_policies_operations,
     get_sql_managed_database_long_term_retention_policies_operations,
     get_sql_managed_database_long_term_retention_backups_operations,
@@ -703,6 +704,23 @@ def load_command_table(self, _):
             is_preview=True)
         g.wait_command('wait')
 
+    with self.command_group('sql midb log-replay',
+                            managed_databases_operations,
+                            client_factory=get_sql_managed_databases_operations) as g:
+        g.custom_command('start', 'managed_db_log_replay_start', supports_no_wait=True)
+        g.command('stop', 'delete', confirmation=True, supports_no_wait=True)
+        g.command('complete', 'complete_restore')
+
+    managed_database_restore_details_operations = CliCommandType(
+        operations_tmpl='azure.mgmt.sql.operations#ManagedDatabaseRestoreDetailsOperations.{}',
+        client_factory=get_sql_managed_database_restore_details_operations)
+
+    with self.command_group('sql midb log-replay',
+                            managed_database_restore_details_operations,
+                            client_factory=get_sql_managed_database_restore_details_operations) as g:
+
+        g.show_command('show', 'get')
+
     ###############################################
     #                sql virtual cluster         #
     ###############################################

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

@@ -1276,7 +1276,7 @@ def db_update(
     Applies requested parameters to a db resource instance for a DB update.
     '''
     # Verify edition
-    if instance.sku.tier.lower() == DatabaseEdition.data_warehouse.value.lower():  # pylint: disable=no-member
+    if instance.sku.tier.lower() == "datawarehouse":  # pylint: disable=no-member
         raise CLIError('Azure SQL Data Warehouse can be updated with the command'
                        ' `az sql dw update`.')
 
@@ -1953,7 +1953,7 @@ def dw_create(
     '''
 
     # Set edition
-    kwargs['sku'].tier = DatabaseEdition.data_warehouse.value
+    kwargs['sku'].tier = "DataWarehouse"
 
     # Create
     return _db_dw_create(
@@ -1978,7 +1978,7 @@ def dw_list(
         server_name=server_name)
 
     # Include only DW's
-    return [db for db in dbs if db.sku.tier == DatabaseEdition.data_warehouse.value]
+    return [db for db in dbs if db.sku.tier == "DataWarehouse"]
 
 
 def dw_update(
@@ -3359,6 +3359,43 @@ def restore_long_term_retention_mi_backup(
         resource_group_name=target_resource_group_name,
         parameters=kwargs)
 
+
+def managed_db_log_replay_start(
+        cmd,
+        client,
+        database_name,
+        managed_instance_name,
+        resource_group_name,
+        auto_complete,
+        last_backup_name,
+        storage_container_uri,
+        storage_container_sas_token,
+        **kwargs):
+
+    # Determine managed instance location
+    kwargs['location'] = _get_managed_instance_location(
+        cmd.cli_ctx,
+        managed_instance_name=managed_instance_name,
+        resource_group_name=resource_group_name)
+
+    kwargs['create_mode'] = CreateMode.restore_external_backup.value
+
+    if auto_complete and not last_backup_name:
+        raise CLIError('Please specify a last backup name when using auto complete flag.')
+
+    kwargs['auto_complete'] = auto_complete
+    kwargs['last_backup_name'] = last_backup_name
+
+    kwargs['storageContainerUri'] = storage_container_uri
+    kwargs['storageContainerSasToken'] = storage_container_sas_token
+
+    # Create
+    return client.create_or_update(
+        database_name=database_name,
+        managed_instance_name=managed_instance_name,
+        resource_group_name=resource_group_name,
+        parameters=kwargs)
+
 ###############################################
 #              sql failover-group             #
 ###############################################