using Microsoft.EntityFrameworkCore; using Npgsql; namespace RegexBot.Data; /// /// Represents a database connection using the settings defined in the bot's global configuration. /// public class BotDatabaseContext : DbContext { private static readonly string _connectionString; static BotDatabaseContext() { // Get our own config loaded just for the SQL stuff var conf = new InstanceConfig(); _connectionString = new NpgsqlConnectionStringBuilder() { Host = conf.SqlHost ?? "localhost", // default to localhost Database = conf.SqlDatabase, Username = conf.SqlUsername, Password = conf.SqlPassword }.ToString(); } /// /// Retrieves the user cache. /// public DbSet UserCache { get; set; } = null!; /// /// Retrieves the guild user cache. /// public DbSet GuildUserCache { get; set; } = null!; /// /// Retrieves the guild message cache. /// public DbSet GuildMessageCache { get; set; } = null!; /// /// Retrieves the moderator logs. /// public DbSet ModLogs { get; set; } = null!; /// protected sealed override void OnConfiguring(DbContextOptionsBuilder optionsBuilder) => optionsBuilder .UseNpgsql(_connectionString) .UseSnakeCaseNamingConvention(); /// protected override void OnModelCreating(ModelBuilder modelBuilder) { modelBuilder.Entity(entity => entity.Property(e => e.Discriminator).HasMaxLength(4).IsFixedLength()); modelBuilder.Entity(e => { e.HasKey(p => new { p.UserId, p.GuildId }); e.Property(p => p.FirstSeenTime).HasDefaultValueSql("now()"); }); modelBuilder.Entity(e => e.Property(p => p.CreatedAt).HasDefaultValueSql("now()")); modelBuilder.HasPostgresEnum(); modelBuilder.Entity(e => e.Property(p => p.Timestamp).HasDefaultValueSql("now()")); } }