mirror of
https://github.com/NoiTheCat/BirthdayBot.git
synced 2024-11-25 01:44:12 +00:00
64 lines
2.3 KiB
C#
64 lines
2.3 KiB
C#
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
|
|
}.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);
|
|
});
|
|
}
|
|
}
|