SCHALE.GameServer/SCHALE.Common/Database/SCHALEContext.cs

using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.ChangeTracking;
using Microsoft.EntityFrameworkCore.ValueGeneration;
using MongoDB.EntityFrameworkCore.Extensions;
using SCHALE.Common.Database.Models;
using SCHALE.Common.Database.Models.Game;

namespace SCHALE.Common.Database
{
    public class SCHALEContext : DbContext
    {
        public DbSet<GuestAccount> GuestAccounts { get; set; }
        public DbSet<Account> Accounts { get; set; }
        public DbSet<Counter> Counters { get; set; }
        public DbSet<Player> Players {  get; set; }

        public Player CurrentPlayer { get { return Players.FirstOrDefault();} } // temp

        public SCHALEContext(DbContextOptions<SCHALEContext> options) : base(options)
        {
        }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            base.OnModelCreating(modelBuilder);

            modelBuilder.Entity<GuestAccount>().Property(x => x.Uid).HasValueGenerator<GuestAccountAutoIncrementValueGenerator>();
            modelBuilder.Entity<GuestAccount>().ToCollection("guest_accounts");

            modelBuilder.Entity<Account>().ToCollection("accounts");
            modelBuilder.Entity<Player>().ToCollection("players");

            // attempt to fix MissionProgressDB.Dictionary<long, long> ProgressParameters serialization
            modelBuilder.Entity<Player>().Property(e => e.MissionProgressDBs).HasJsonConversion<List<MissionProgressDB>>();

            modelBuilder.Entity<Counter>().ToCollection("counters");
        }

        private class GuestAccountAutoIncrementValueGenerator : AutoIncrementValueGenerator
        {
            protected override string Collection => "guest_account";
        }

        private abstract class AutoIncrementValueGenerator : ValueGenerator<uint>
        {
            protected abstract string Collection { get; }
            public override bool GeneratesTemporaryValues => false;

            public override uint Next(EntityEntry entry)
            {
                if (entry.Context is not SCHALEContext)
                {
                    throw new ArgumentNullException($"{nameof(AutoIncrementValueGenerator)} is only implemented for {nameof(SCHALEContext)}");
                }

                var context = ((SCHALEContext)entry.Context);
                var counter = context.Counters.SingleOrDefault(x => x.Id == Collection);

                if (counter is null)
                {
                    counter = new Counter() { Id = Collection, Seq = 0 };
                    context.Add(counter);
                }

                counter.Seq++;
                context.SaveChanges();

                return counter.Seq;
            }
        }
    }
}