✨ Added command to drop database on demand (#411)

Author: danielmackay

tools/AppHost/AppHost.csproj

@@ -11,6 +11,7 @@
   <ItemGroup>
     <PackageReference Include="Aspire.Hosting.AppHost" Version="9.0.0-rc.1.24511.1"/>
     <PackageReference Include="Aspire.Hosting.SqlServer" Version="9.0.0-rc.1.24511.1"/>
+    <PackageReference Include="Microsoft.EntityFrameworkCore.SqlServer" Version="9.0.0-rc.2.24474.1" />
   </ItemGroup>
 
   <ItemGroup>

tools/AppHost/Commands/SqlServerCommandExt.cs

@@ -0,0 +1,37 @@
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.Diagnostics.HealthChecks;
+
+namespace AppHost.Commands;
+
+public static class SqlServerCommandExt
+{
+    public static IResourceBuilder<SqlServerServerResource> WithDropDatabaseCommand(
+        this IResourceBuilder<SqlServerServerResource> builder)
+    {
+        builder.WithCommand(
+            "drop-database",
+            "Drop Database",
+            async context =>
+            {
+                var connectionString = await builder.Resource.GetConnectionStringAsync();
+                ArgumentException.ThrowIfNullOrWhiteSpace(connectionString);
+
+                var optionsBuilder = new DbContextOptionsBuilder<DbContext>();
+                optionsBuilder.UseSqlServer(connectionString);
+                var db = new DbContext(optionsBuilder.Options);
+                await db.Database.EnsureDeletedAsync();
+
+                return CommandResults.Success();
+            },
+            context =>
+            {
+                if (context.ResourceSnapshot.HealthStatus is HealthStatus.Healthy)
+                {
+                    return ResourceCommandState.Enabled;
+                }
+
+                return ResourceCommandState.Disabled;
+            });
+        return builder;
+    }
+}

tools/AppHost/Program.cs

@@ -1,10 +1,12 @@
+using AppHost.Commands;
 using Projects;
 
 var builder = DistributedApplication.CreateBuilder(args);
 
 // Ensure the port doesn't conflict with other docker containers (or remove it altogether)
 var sqlServer = builder
     .AddSqlServer("sql", port: 1800)
+    .WithDropDatabaseCommand()
     .WithLifetime(ContainerLifetime.Persistent);
 
 var db = sqlServer

tools/MigrationService/Initializers/DbContextInitializerBase.cs

@@ -21,10 +21,10 @@ public async Task EnsureDatabaseAsync(CancellationToken cancellationToken)
         var strategy = DbContext.Database.CreateExecutionStrategy();
         await strategy.ExecuteAsync(async () =>
         {
+            // Create the database if it does not exist.
+            // Do this first so there is then a database to start a transaction against.
             if (!await dbCreator.ExistsAsync(cancellationToken))
             {
-                // Create the database if it does not exist.
-                // Do this first so there is then a database to start a transaction against.
                 await dbCreator.CreateAsync(cancellationToken);
             }
         });