BirthdayBot/Data/BotDatabaseContext.cs

using Microsoft.EntityFrameworkCore;
using Npgsql;

namespace BirthdayBot.Data;

public class BotDatabaseContext : DbContext {
    private static readonly string _connectionString;

    static BotDatabaseContext() {
        // Get our own config loaded just for the SQL stuff
        var conf = new Configuration();
        _connectionString = new NpgsqlConnectionStringBuilder() {
            Host = conf.SqlHost ?? "localhost", // default to localhost
            Database = conf.SqlDatabase,
            Username = conf.SqlUsername,
            Password = conf.SqlPassword,
            ApplicationName = conf.SqlApplicationName,

            // Let's see if this works?
            ConnectionIdleLifetime = 60,
            MaxPoolSize = Math.Max(
                (int)Math.Ceiling(conf.ShardAmount * 2 * 0.75),
                (int)Math.Ceiling(ShardManager.MaxConcurrentOperations * 2.5))
        }.ToString();
    }

    public DbSet<BlocklistEntry> BlocklistEntries { get; set; } = null!;
    public DbSet<GuildConfig> GuildConfigurations { get; set; } = null!;
    public DbSet<UserEntry> UserEntries { get; set; } = null!;

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
        => optionsBuilder
            .UseNpgsql(_connectionString)
            .UseSnakeCaseNamingConvention();

    protected override void OnModelCreating(ModelBuilder modelBuilder) {
        modelBuilder.Entity<BlocklistEntry>(entity => {
            entity.HasKey(e => new { e.GuildId, e.UserId })
                .HasName("banned_users_pkey");

            entity.HasOne(d => d.Guild)
                .WithMany(p => p.BlockedUsers)
                .HasForeignKey(d => d.GuildId)
                .HasConstraintName("banned_users_guild_id_fkey")
                .OnDelete(DeleteBehavior.Cascade);
        });

        modelBuilder.Entity<GuildConfig>(entity => {
            entity.HasKey(e => e.GuildId)
                .HasName("settings_pkey");

            entity.Property(e => e.GuildId).ValueGeneratedNever();

            entity.Property(e => e.LastSeen).HasDefaultValueSql("now()");
        });

        modelBuilder.Entity<UserEntry>(entity => {
            entity.HasKey(e => new { e.GuildId, e.UserId })
                .HasName("user_birthdays_pkey");

            entity.Property(e => e.LastSeen).HasDefaultValueSql("now()");

            entity.HasOne(d => d.Guild)
                .WithMany(p => p.UserEntries)
                .HasForeignKey(d => d.GuildId)
                .HasConstraintName("user_birthdays_guild_id_fkey")
                .OnDelete(DeleteBehavior.Cascade);
        });
    }
}