summaryrefslogtreecommitdiff
path: root/HBContext.cs
diff options
context:
space:
mode:
Diffstat (limited to 'HBContext.cs')
-rw-r--r--HBContext.cs85
1 files changed, 0 insertions, 85 deletions
diff --git a/HBContext.cs b/HBContext.cs
deleted file mode 100644
index 766c0a3..0000000
--- a/HBContext.cs
+++ /dev/null
@@ -1,85 +0,0 @@
-using HyperBooru.ApiModels;
-using HyperBooru.Services;
-using Microsoft.EntityFrameworkCore;
-
-namespace HyperBooru;
-
-enum HBObjectId {
- NsfwTag = -1,
- IngestTag = -2,
- AdminUser = -3
-}
-
-public class HBContext : DbContext {
- public static readonly Guid NsfwTag = new("EBDAD4F8-455A-4351-8017-1D4854D6FA38");
- public static readonly Guid IngestTag = new("EA212801-5BCC-4C0E-814F-FB9D30DB58BC");
- public static readonly Guid AdminUser = new("4FA948F4-7C45-4F81-BB6B-E417491E6C96");
-
- public DbSet<HBObject> Objects { get; set; }
- public DbSet<User> Users { get; set; }
- public DbSet<TagDefinition> TagDefinitions { get; set; }
- public DbSet<Tag> Tags { get; set; }
- public DbSet<Media> Media { get; set; }
- public DbSet<UploadedFile> UploadedFiles { get; set; }
- public DbSet<OcrData> OcrData { get; set; }
-
- private IConfigService config;
-
- public HBContext(DbContextOptions<HBContext> options, IConfigService config) : base(options) =>
- this.config = config;
-
- protected override void OnConfiguring(DbContextOptionsBuilder options) {
- options.UseNpgsql(config.DbConnectionString);
-
- #if DEBUG
- options.EnableSensitiveDataLogging();
- #endif
- }
-
- protected override void OnModelCreating(ModelBuilder modelBuilder) {
- // Don't use shared tables for inherited types
- modelBuilder.Entity<HBObject>().ToTable("Objects");
- modelBuilder.Entity<TagDefinition>().ToTable("TagDefinitions");
- modelBuilder.Entity<Tag>().ToTable("Tags");
- modelBuilder.Entity<Media>().ToTable("Media");
- modelBuilder.Entity<UploadedFile>().ToTable("UploadedFiles");
-
- // Seed internal tag definitions
- // These should NEVER change
- modelBuilder.Entity<TagDefinition>().HasData(new TagDefinition[] {
- new() {
- ObjectId = (int) HBObjectId.NsfwTag,
- Guid = NsfwTag,
- Source = TagSource.Internal,
- Name = "nsfw"
- },
- new() {
- ObjectId = (int) HBObjectId.IngestTag,
- Guid = IngestTag,
- Source = TagSource.Internal,
- Name = "ingest"
- }
- });
-
- // Seed initial admin user
- modelBuilder.Entity<User>().HasData(new User[] {
- new() {
- ObjectId = (int) HBObjectId.AdminUser,
- Guid = AdminUser,
- Username = "admin",
- PasswordHash = UserService.HashPassword("admin")
- }
- });
-
- // Some complex relationships cannot be inferred and require
- // additional configuration, as seen below.
- modelBuilder.Entity<TagDefinition>()
- .HasMany(e => e.ImplicitTags)
- .WithMany();
-
- modelBuilder.Entity<Media>()
- .HasOne(m => m.CurrentUploadedFile)
- .WithOne()
- .HasForeignKey<Media>("CurrentUploadedFileId");
- }
-} \ No newline at end of file