[PM-3797 Part 4] Add Sends to new Key Rotation (#3442)

* add send validation

* add send repo methods

* add send rotation to delegate list

* add success test

src/Infrastructure.Dapper/Tools/Repositories/SendRepository.cs

@@ -1,10 +1,12 @@
 #nullable enable
 
 using System.Data;
+using Bit.Core.Auth.UserFeatures.UserKey;
 using Bit.Core.Settings;
 using Bit.Core.Tools.Entities;
 using Bit.Core.Tools.Repositories;
 using Bit.Infrastructure.Dapper.Repositories;
+using Bit.Infrastructure.Dapper.Tools.Helpers;
 using Dapper;
 using Microsoft.Data.SqlClient;
 
@@ -48,4 +50,57 @@ public class SendRepository : Repository<Send, Guid>, ISendRepository
             return results.ToList();
         }
     }
+
+    /// <inheritdoc />
+    public UpdateEncryptedDataForKeyRotation UpdateForKeyRotation(Guid userId, IEnumerable<Send> sends)
+    {
+        return async (connection, transaction) =>
+        {
+            // Create temp table
+            var sqlCreateTemp = @"
+                            SELECT TOP 0 *
+                            INTO #TempSend
+                            FROM [dbo].[Send]";
+
+            await using (var cmd = new SqlCommand(sqlCreateTemp, connection, transaction))
+            {
+                cmd.ExecuteNonQuery();
+            }
+
+            // Bulk copy data into temp table
+            using (var bulkCopy = new SqlBulkCopy(connection, SqlBulkCopyOptions.KeepIdentity, transaction))
+            {
+                bulkCopy.DestinationTableName = "#TempSend";
+                var sendsTable = sends.ToDataTable();
+                foreach (DataColumn col in sendsTable.Columns)
+                {
+                    bulkCopy.ColumnMappings.Add(col.ColumnName, col.ColumnName);
+                }
+
+                sendsTable.PrimaryKey = new DataColumn[] { sendsTable.Columns[0] };
+                await bulkCopy.WriteToServerAsync(sendsTable);
+            }
+
+            // Update send table from temp table
+            var sql = @"
+                UPDATE
+                    [dbo].[Send]
+                SET
+                    [Key] = TS.[Key],
+                    [RevisionDate] = TS.[RevisionDate]
+                FROM
+                    [dbo].[Send] S
+                INNER JOIN
+                    #TempSend TS ON S.Id = TS.Id
+                WHERE
+                    S.[UserId] = @UserId
+                DROP TABLE #TempSend";
+
+            await using (var cmd = new SqlCommand(sql, connection, transaction))
+            {
+                cmd.Parameters.Add("@UserId", SqlDbType.UniqueIdentifier).Value = userId;
+                cmd.ExecuteNonQuery();
+            }
+        };
+    }
 }