mirror of
https://github.com/Kareadita/Kavita.git
synced 2025-05-24 00:52:23 -04:00
56 lines
2.0 KiB
C#
56 lines
2.0 KiB
C#
using System;
|
|
using API.Entities;
|
|
using API.Entities.Interfaces;
|
|
using Microsoft.AspNetCore.Identity;
|
|
using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.EntityFrameworkCore.ChangeTracking;
|
|
|
|
namespace API.Data
|
|
{
|
|
public sealed class DataContext : IdentityDbContext<AppUser, AppRole, int,
|
|
IdentityUserClaim<int>, AppUserRole, IdentityUserLogin<int>,
|
|
IdentityRoleClaim<int>, IdentityUserToken<int>>
|
|
{
|
|
public DataContext(DbContextOptions options) : base(options)
|
|
{
|
|
ChangeTracker.Tracked += OnEntityTracked;
|
|
ChangeTracker.StateChanged += OnEntityStateChanged;
|
|
|
|
}
|
|
public DbSet<Library> Library { get; set; }
|
|
public DbSet<Series> Series { get; set; }
|
|
public DbSet<Volume> Volume { get; set; }
|
|
public DbSet<AppUser> AppUser { get; set; }
|
|
public DbSet<AppUserProgress> AppUserProgresses { get; set; }
|
|
|
|
protected override void OnModelCreating(ModelBuilder builder)
|
|
{
|
|
base.OnModelCreating(builder);
|
|
|
|
builder.Entity<AppUser>()
|
|
.HasMany(ur => ur.UserRoles)
|
|
.WithOne(u => u.User)
|
|
.HasForeignKey(ur => ur.UserId)
|
|
.IsRequired();
|
|
|
|
builder.Entity<AppRole>()
|
|
.HasMany(ur => ur.UserRoles)
|
|
.WithOne(u => u.Role)
|
|
.HasForeignKey(ur => ur.RoleId)
|
|
.IsRequired();
|
|
}
|
|
|
|
void OnEntityTracked(object sender, EntityTrackedEventArgs e)
|
|
{
|
|
if (!e.FromQuery && e.Entry.State == EntityState.Added && e.Entry.Entity is IEntityDate entity)
|
|
entity.Created = DateTime.Now;
|
|
}
|
|
|
|
void OnEntityStateChanged(object sender, EntityStateChangedEventArgs e)
|
|
{
|
|
if (e.NewState == EntityState.Modified && e.Entry.Entity is IEntityDate entity)
|
|
entity.LastModified = DateTime.Now;
|
|
}
|
|
}
|
|
} |