PickBanStore

src/MysqlMigrations/Migrations/20240608183114_DsPickBans.cs

@@ -0,0 +1,42 @@
+using System;
+using Microsoft.EntityFrameworkCore.Metadata;
+using Microsoft.EntityFrameworkCore.Migrations;
+
+#nullable disable
+
+namespace MysqlMigrations.Migrations
+{
+    /// <inheritdoc />
+    public partial class DsPickBans : Migration
+    {
+        /// <inheritdoc />
+        protected override void Up(MigrationBuilder migrationBuilder)
+        {
+            migrationBuilder.CreateTable(
+                name: "DsPickBans",
+                columns: table => new
+                {
+                    DsPickBanId = table.Column<int>(type: "int", nullable: false)
+                        .Annotation("MySql:ValueGenerationStrategy", MySqlValueGenerationStrategy.IdentityColumn),
+                    PickBanMode = table.Column<int>(type: "int", nullable: false),
+                    Time = table.Column<DateTime>(type: "datetime(0)", precision: 0, nullable: false),
+                    Bans = table.Column<string>(type: "longtext", nullable: false)
+                        .Annotation("MySql:CharSet", "utf8mb4"),
+                    Picks = table.Column<string>(type: "longtext", nullable: false)
+                        .Annotation("MySql:CharSet", "utf8mb4")
+                },
+                constraints: table =>
+                {
+                    table.PrimaryKey("PK_DsPickBans", x => x.DsPickBanId);
+                })
+                .Annotation("MySql:CharSet", "utf8mb4");
+        }
+
+        /// <inheritdoc />
+        protected override void Down(MigrationBuilder migrationBuilder)
+        {
+            migrationBuilder.DropTable(
+                name: "DsPickBans");
+        }
+    }
+}

src/MysqlMigrations/Migrations/ReplayContextModelSnapshot.cs

@@ -17,7 +17,7 @@ protected override void BuildModel(ModelBuilder modelBuilder)
         {
 #pragma warning disable 612, 618
             modelBuilder
-                .HasAnnotation("ProductVersion", "8.0.5")
+                .HasAnnotation("ProductVersion", "8.0.6")
                 .HasAnnotation("Relational:MaxIdentifierLength", 64);
 
             MySqlModelBuilderExtensions.AutoIncrementColumns(modelBuilder);
@@ -586,6 +586,34 @@ protected override void BuildModel(ModelBuilder modelBuilder)
                     b.ToTable("DsAbilities");
                 });
 
+            modelBuilder.Entity("dsstats.db8.DsPickBan", b =>
+                {
+                    b.Property<int>("DsPickBanId")
+                        .ValueGeneratedOnAdd()
+                        .HasColumnType("int");
+
+                    MySqlPropertyBuilderExtensions.UseMySqlIdentityColumn(b.Property<int>("DsPickBanId"));
+
+                    b.Property<string>("Bans")
+                        .IsRequired()
+                        .HasColumnType("longtext");
+
+                    b.Property<int>("PickBanMode")
+                        .HasColumnType("int");
+
+                    b.Property<string>("Picks")
+                        .IsRequired()
+                        .HasColumnType("longtext");
+
+                    b.Property<DateTime>("Time")
+                        .HasPrecision(0)
+                        .HasColumnType("datetime(0)");
+
+                    b.HasKey("DsPickBanId");
+
+                    b.ToTable("DsPickBans");
+                });
+
             modelBuilder.Entity("dsstats.db8.DsUnit", b =>
                 {
                     b.Property<int>("DsUnitId")

src/dsstats.db8/PickBan.cs

@@ -0,0 +1,14 @@
+using dsstats.shared;
+using Microsoft.EntityFrameworkCore;
+
+namespace dsstats.db8;
+
+public class DsPickBan
+{
+    public int DsPickBanId { get; set; }
+    public PickBanMode PickBanMode { get; set; }
+    [Precision(0)]
+    public DateTime Time {  get; set; }
+    public List<Commander> Bans { get; set; } = [];
+    public List<Commander> Picks { get; set; } = [];
+}

src/dsstats.db8/ReplayContext.cs

@@ -57,6 +57,7 @@ public class ReplayContext : DbContext
     public virtual DbSet<FaqVote> FaqVotes { get; set; } = null!;
     public virtual DbSet<IhSession> IhSessions { get; set; } = null!;
     public virtual DbSet<IhSessionPlayer> IhSessionPlayers { get; set; } = null!;
+    public virtual DbSet<DsPickBan> DsPickBans { get; set; } = null!;
     public int Week(DateTime date) => throw new InvalidOperationException($"{nameof(Week)} cannot be called client side.");
     public int Strftime(string arg, DateTime date) => throw new InvalidOperationException($"{nameof(Strftime)} cannot be called client side.");
 

src/dsstats.pickban/PickBanHub.cs

@@ -54,7 +54,10 @@ public async Task SetBan(PickBan pickBan)
                 return;
             }
             await Clients.Group(guid.ToString()).SendAsync("Bans", resultDto.Bans);
-
+            if (resultDto.TotalPicks == 0 && resultDto.Bans.Count == resultDto.TotalBans)
+            {
+                await pickBanRepository.SavePickBan(resultDto);
+            }
         }
     }
 
@@ -69,6 +72,10 @@ public async Task SetPick(PickBan pickBan)
                 return;
             }
             await Clients.Group(guid.ToString()).SendAsync("Picks", resultDto.Picks);
+            if (resultDto.Picks.Count == resultDto.TotalPicks)
+            {
+                await pickBanRepository.SavePickBan(resultDto);
+            }
         }
     }
 

src/dsstats.pickban/PickBanService.cs

@@ -1,8 +1,10 @@
-using dsstats.shared;
+using dsstats.db8;
+using dsstats.shared;
+using Microsoft.Extensions.DependencyInjection;
 
 namespace dsstats.pickban;
 
-public class PickBanRepository
+public class PickBanRepository(IServiceScopeFactory scopeFactory)
 {
     private Dictionary<Guid, PickBanState> pickbanStates = [];
 
@@ -63,4 +65,20 @@ public int SetVisitor(Guid guid, bool joined)
         }
         return null;
     }
+
+    public async Task SavePickBan(PickBanStateDto pickBan)
+    {
+        using var scope = scopeFactory.CreateScope();
+        var context = scope.ServiceProvider.GetRequiredService<ReplayContext>();
+
+        DsPickBan dsPickBan = new()
+        {
+            PickBanMode = pickBan.PickBanMode,
+            Time = DateTime.UtcNow,
+            Bans = pickBan.Bans.Select(s => s.Commander).ToList(),
+            Picks = pickBan.Picks.Select(s => s.Commander).ToList()
+        };
+        context.DsPickBans.Add(dsPickBan);
+        await context.SaveChangesAsync();
+    }
 }

src/dsstats.pickban/dsstats.pickban.csproj

@@ -11,6 +11,7 @@
   </ItemGroup>
 
   <ItemGroup>
+    <ProjectReference Include="..\dsstats.db8\dsstats.db8.csproj" />
     <ProjectReference Include="..\dsstats.shared\dsstats.shared.csproj" />
   </ItemGroup>