Updated TDE NuGet and contracts (#2242)

Packages.props

@@ -31,7 +31,7 @@
 		<PackageReference Update="Microsoft.SqlServer.Migration.Logins" Version="1.0.20230407.56" />
 		<PackageReference Update="Microsoft.SqlServer.Management.QueryStoreModel" Version="163.26.1" />
 		<PackageReference Update="Microsoft.SqlServer.Management.SqlParser" Version="170.9.0" />
-		<PackageReference Update="Microsoft.SqlServer.Migration.Tde" Version="1.0.0-preview.1.0.20230720.98" />
+		<PackageReference Update="Microsoft.SqlServer.Migration.Tde" Version="1.0.0-preview.1.0.20230914.107" />
 		<PackageReference Update="Microsoft.Azure.OperationalInsights" Version="1.0.0" />
 		<PackageReference Update="Microsoft.CodeAnalysis.CSharp" Version="3.10.0" />
 		<PackageReference Update="Microsoft.CodeAnalysis.CSharp.Workspaces" Version="3.10.0" />

src/Microsoft.SqlTools.Migration/Contracts/TdeValidationRequest.cs

@@ -0,0 +1,29 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+using Microsoft.SqlServer.Migration.Tde.Validations;
+using Microsoft.SqlTools.Hosting.Protocol.Contracts;
+
+namespace Microsoft.SqlTools.Migration.Contracts
+{
+    public class TdeValidationParams
+    {
+        /// <summary>
+        /// Source connection string for SQL Server.
+        /// </summary>
+        public string? SourceSqlConnectionString { get; set; }
+
+        /// <summary>
+        /// Location where certificates will be exported.
+        /// </summary>
+        public string? NetworkSharePath { get; set; }
+    }
+
+    public class TdeValidationRequest
+    {
+        public static readonly RequestType<TdeValidationParams, TdeValidationResult[]> Type =
+            RequestType<TdeValidationParams, TdeValidationResult[]>.Create("migration/tdevalidation");
+    }
+}

src/Microsoft.SqlTools.Migration/Contracts/TdeValidationTitlesRequest.cs

@@ -0,0 +1,19 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+using Microsoft.SqlTools.Hosting.Protocol.Contracts;
+
+namespace Microsoft.SqlTools.Migration.Contracts
+{
+    public class TdeValidationTitlesParams
+    {
+    }
+
+    public class TdeValidationTitlesRequest
+    {
+        public static readonly RequestType<TdeValidationTitlesParams, string[]> Type =
+            RequestType<TdeValidationTitlesParams, string[]>.Create("migration/tdevalidationtitles");
+    }
+}

src/Microsoft.SqlTools.Migration/MigrationService.cs

@@ -47,6 +47,7 @@ using Microsoft.SqlServer.Migration.SkuRecommendation.Models.Sql;
 using Microsoft.SqlServer.Migration.SkuRecommendation.Utils;
 using Microsoft.SqlServer.Migration.Tde;
 using Microsoft.SqlServer.Migration.Tde.Common;
+using Microsoft.SqlServer.Migration.Tde.Validations;
 using Microsoft.SqlTools.Hosting;
 using Microsoft.SqlTools.Hosting.Protocol;
 using Microsoft.SqlTools.Migration.Contracts;
@@ -114,10 +115,11 @@ namespace Microsoft.SqlTools.Migration
             this.ServiceHost.SetRequestHandler(EstablishUserMappingRequest.Type, HandleEstablishUserMapping, true);
             this.ServiceHost.SetRequestHandler(MigrateServerRolesAndSetPermissionsRequest.Type, HandleMigrateServerRolesAndSetPermissions, true);
             this.ServiceHost.SetRequestHandler(CertificateMigrationRequest.Type, HandleTdeCertificateMigrationRequest);
+            this.ServiceHost.SetRequestHandler(TdeValidationRequest.Type, HandleTdeValidationRequest);
+            this.ServiceHost.SetRequestHandler(TdeValidationTitlesRequest.Type, HandleTdeValidationTitlesRequest);
             Logger.Verbose("Migration Service initialized");
         }
 
-
         /// <summary>
         /// Handle request to start a migration session
         /// </summary>
@@ -977,6 +979,24 @@ namespace Microsoft.SqlTools.Migration
             await requestContext.SendResult(result);
         }
 
+        internal async Task HandleTdeValidationRequest(
+            TdeValidationParams parameters,
+            RequestContext<TdeValidationResult[]> requestContext)
+        {
+            TdeValidationResult[] result = 
+                await TdeMigration.RunTdeValidation(
+                   parameters.SourceSqlConnectionString,
+                   parameters.NetworkSharePath);
+            await requestContext.SendResult(result);
+        }
+
+        internal async Task HandleTdeValidationTitlesRequest(
+            TdeValidationTitlesParams parameters,
+            RequestContext<string[]> requestContext)
+        {
+            await requestContext.SendResult(TdeMigration.TdeValidationTitles);
+        }
+
         /// <summary>
         /// Individual certificate migration operation
         /// </summary>