sodoff/src/Model/DBContext.cs

using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Options;
using sodoff.Configuration;

namespace sodoff.Model;
public class DBContext : DbContext {
    public DbSet<User> Users { get; set; } = null!;
    public DbSet<Viking> Vikings { get; set; } = null!;
    public DbSet<Dragon> Dragons { get; set; } = null!;
    public DbSet<Image> Images { get; set; } = null!;
    public DbSet<Session> Sessions { get; set; } = null!;
    public DbSet<Pair> Pairs { get; set; } = null!;
    public DbSet<PairData> PairData { get; set; } = null!;
    public DbSet<TaskStatus> TaskStatuses { get; set; } = null!;
    public DbSet<InventoryItem> InventoryItems { get; set; } = null!;
    public DbSet<MissionState> MissionStates { get; set; } = null!;
    public DbSet<Room> Rooms { get; set; } = null!;
    public DbSet<SceneData> SceneData { get; set; } = null!;
    public DbSet<RoomItem> RoomItems { get; set; } = null!;
    public DbSet<GameData> GameData { get; set; } = null!;
    public DbSet<GameDataPair> GameDataPairs { get; set; } = null!;
    public DbSet<AchievementPoints> AchievementPoints { get; set; } = null!;
    public DbSet<ProfileAnswer> ProfileAnswers { get; set; } = null!;
    public DbSet<MMORole> MMORoles { get; set; } = null!;
    public DbSet<Party> Parties { get; set; } = null!;
    public DbSet<AchievementTaskState> AchievementTaskState { get; set; } = null!;
    public DbSet<SavedData> SavedData { get; set; } = null!;
    public DbSet<UserMissionData> UserMissions { get; set; } = null!;
    public DbSet<Neighborhood> Neighborhoods { get; set; } = null!;
    // we had a brief debate on whether it's neighborhoods or neighborheed
    public DbSet<Group> Groups { get; set; } = null!;
    public DbSet<Rating> Ratings { get; set; } = null!;
    public DbSet<RatingRank> RatingRanks { get; set; } = null!;
    public DbSet<UserMissionData> UserMissionData { get; set; } = null!;
    public DbSet<UserBadgeCompleteData> UserBadgesCompleted { get; set; } = null!;

    private readonly IOptions<ApiServerConfig> config;

    public DBContext(IOptions<ApiServerConfig> config) {
        this.config = config;
    }

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder) {
        #if USE_POSTGRESQL
            if (config.Value.DbProvider == DbProviders.PostgreSQL) {
                optionsBuilder.UseNpgsql(config.Value.DbConnection).UseLazyLoadingProxies();
                return;
            }
        #endif
        #if USE_MYSQL
            if (config.Value.DbProvider == DbProviders.MySQL) {
                optionsBuilder.UseMySQL(config.Value.DbConnection).UseLazyLoadingProxies();
                return;
            }
        #endif
        #if USE_SQLITE
            if (config.Value.DbProvider == DbProviders.SQLite) {
                string DbPath;
                if (String.IsNullOrEmpty(config.Value.DbPath)) {
                    DbPath = Path.Join(Directory.GetCurrentDirectory(), "sodoff.db");
                } else {
                    DbPath = config.Value.DbPath;
                }
                optionsBuilder.UseSqlite($"Data Source={DbPath}").UseLazyLoadingProxies();
                return;
            }
        #endif
        throw new Exception($"Unsupported DbProvider {config.Value.DbProvider}");
    }

    protected override void OnModelCreating(ModelBuilder builder) {
        // Sessions
        builder.Entity<Session>().HasOne(s => s.User)
            .WithMany(e => e.Sessions)
            .HasForeignKey(e => e.UserId)
            .OnDelete(DeleteBehavior.Cascade);

        builder.Entity<Session>().HasOne(s => s.Viking)
            .WithMany(e => e.Sessions)
            .HasForeignKey(e => e.VikingId)
            .OnDelete(DeleteBehavior.Cascade);

        // Users
        builder.Entity<User>().HasMany(u => u.Sessions)
            .WithOne(e => e.User);

        builder.Entity<User>().HasMany(u => u.PairData)
            .WithOne(e => e.User);

        builder.Entity<User>().HasMany(u => u.Vikings)
            .WithOne(e => e.User);

        // Vikings
        builder.Entity<Viking>().HasOne(v => v.User)
            .WithMany(e => e.Vikings)
            .HasForeignKey(e => e.UserId);

        builder.Entity<Viking>().HasMany(v => v.Dragons)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.Sessions)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.MissionStates)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.Rooms)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.SceneData)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.AchievementPoints)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.PairData)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.Images)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.TaskStatuses)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.AchievementTaskStates)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasOne(v => v.SelectedDragon)
            .WithOne()
            .HasForeignKey<Viking>(e => e.SelectedDragonId);

        builder.Entity<Viking>().HasMany(v => v.GameData)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.SavedData)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.ProfileAnswers)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.Parties)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.MMORoles)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasOne(v => v.Neighborhood)
            .WithOne(e => e.Viking);

        builder.Entity<Viking>().HasMany(v => v.Groups)
            .WithMany(e => e.Vikings);

        builder.Entity<Viking>().HasMany(v => v.Ratings)
            .WithOne(r => r.Viking);

        builder.Entity<Viking>().HasMany(v => v.UserMissions)
            .WithOne(r => r.Viking);

        builder.Entity<Viking>().HasMany(v => v.UserBadgesCompleted)
            .WithOne(r => r.Viking);

        // Dragons
        builder.Entity<Dragon>().HasOne(d => d.Viking)
            .WithMany(e => e.Dragons)
            .HasForeignKey(e => e.VikingId);

        builder.Entity<Dragon>().HasMany(d => d.PairData)
            .WithOne(e => e.Dragon);

        // PairData & Pair
        builder.Entity<PairData>().HasMany(p => p.Pairs)
            .WithOne(e => e.PairData);

        builder.Entity<PairData>().HasOne(p => p.Viking)
            .WithMany(e => e.PairData)
            .HasForeignKey(e => e.VikingId)
            .OnDelete(DeleteBehavior.Cascade);

        builder.Entity<PairData>().HasOne(p => p.User)
            .WithMany(e => e.PairData)
            .HasForeignKey(e => e.UserId)
            .OnDelete(DeleteBehavior.Cascade);

        builder.Entity<PairData>().HasOne(p => p.Dragon)
            .WithMany(e => e.PairData)
            .HasForeignKey(e => e.DragonId)
            .OnDelete(DeleteBehavior.Cascade);

        builder.Entity<Pair>()
            .HasOne(p => p.PairData)
            .WithMany(pd => pd.Pairs)
            .HasForeignKey(p => p.MasterId)
            .HasPrincipalKey(e => e.Id);

        // Inventory & InventoryItem
        builder.Entity<Viking>()
            .HasMany(v => v.InventoryItems)
            .WithOne(i => i.Viking);

        builder.Entity<InventoryItem>()
            .HasOne(e => e.Viking)
            .WithMany(e => e.InventoryItems)
            .HasForeignKey(e => e.VikingId);

        // Room & RoomItem
        builder.Entity<Room>().HasOne(r => r.Viking)
            .WithMany(e => e.Rooms)
            .HasForeignKey(e => e.VikingId);

        builder.Entity<Room>().HasMany(r => r.Items)
            .WithOne(e => e.Room);

        builder.Entity<RoomItem>().HasOne(i => i.Room)
            .WithMany(r => r.Items)
            .HasForeignKey(e => e.RoomId);

        // GameData

        builder.Entity<GameData>().HasOne(e => e.Viking)
            .WithMany(e => e.GameData);

        builder.Entity<GameData>().HasMany(e => e.GameDataPairs)
            .WithOne(e => e.GameData);

        builder.Entity<GameDataPair>().HasOne(e => e.GameData)
            .WithMany(e => e.GameDataPairs);

        // Others ..
        builder.Entity<Image>().HasOne(s => s.Viking)
            .WithMany(e => e.Images)
            .HasForeignKey(e => e.VikingId);

        builder.Entity<TaskStatus>().HasKey(e => new { e.Id, e.VikingId, e.MissionId });

        builder.Entity<TaskStatus>()
            .HasOne(t => t.Viking)
            .WithMany(v => v.TaskStatuses)
            .HasForeignKey(t => t.VikingId);

        builder.Entity<MissionState>().HasOne(m => m.Viking)
            .WithMany(e => e.MissionStates)
            .HasForeignKey(e => e.VikingId);

        builder.Entity<AchievementTaskState>().HasOne(m => m.Viking)
            .WithMany(e => e.AchievementTaskStates)
            .HasForeignKey(e => e.VikingId);

        builder.Entity<AchievementPoints>().HasKey(e => new { e.VikingId, e.Type });

        builder.Entity<AchievementPoints>()
            .HasOne(e => e.Viking)
            .WithMany(e => e.AchievementPoints)
            .HasForeignKey(e => e.VikingId);

        builder.Entity<SavedData>().HasKey(e => new { e.VikingId, e.SaveId });

        builder.Entity<SavedData>()
            .HasOne(e => e.Viking)
            .WithMany(v => v.SavedData)
            .HasForeignKey(e => e.VikingId);

        builder.Entity<Party>().HasOne(i => i.Viking)
            .WithMany(i => i.Parties);

        builder.Entity<ProfileAnswer>().HasOne(i => i.Viking)
            .WithMany(i => i.ProfileAnswers)
            .HasForeignKey(e => e.VikingId);

        builder.Entity<SceneData>().HasOne(i => i.Viking)
            .WithMany(i => i.SceneData)
            .HasForeignKey(e => e.VikingId);

        // MMO Roles
        builder.Entity<MMORole>().HasOne(r => r.Viking)
            .WithMany(e => e.MMORoles)
            .HasForeignKey(e => e.VikingId);

        // Neighborhoods
        builder.Entity<Neighborhood>().HasOne(r => r.Viking)
            .WithOne(e => e.Neighborhood)
            .HasForeignKey<Neighborhood>(e => e.VikingId);

        // Groups
        builder.Entity<Group>().HasMany(r => r.Vikings)
            .WithMany(e => e.Groups);

        // Rating
        builder.Entity<Rating>().HasOne(r => r.Viking)
            .WithMany(v => v.Ratings)
            .HasForeignKey(r => r.VikingId);

        builder.Entity<Rating>().HasOne(r => r.Rank)
            .WithMany(rr => rr.Ratings)
            .HasForeignKey(r => r.RankId);

        builder.Entity<RatingRank>().HasMany(rr => rr.Ratings)
            .WithOne(r => r.Rank);

        // old ("step") missions
        builder.Entity<UserMissionData>().HasOne(r => r.Viking)
            .WithMany(v => v.UserMissions)
            .HasForeignKey(r => r.VikingId);

        builder.Entity<UserBadgeCompleteData>().HasOne(r => r.Viking)
            .WithMany(v => v.UserBadgesCompleted)
            .HasForeignKey(r => r.VikingId);
    }
}