jellyfin/Jellyfin.Server.Implementations/JellyfinDb.cs

226 lines
7.5 KiB
C#
Raw Normal View History

#nullable disable
2020-05-02 22:32:22 +00:00
#pragma warning disable CS1591
using System;
2020-05-02 22:32:22 +00:00
using System.Linq;
using Jellyfin.Data.Entities;
using Jellyfin.Data.Interfaces;
2020-05-02 22:32:22 +00:00
using Microsoft.EntityFrameworkCore;
namespace Jellyfin.Server.Implementations
{
/// <inheritdoc/>
public class JellyfinDb : DbContext
2020-05-02 22:32:22 +00:00
{
2020-05-23 00:20:18 +00:00
/// <summary>
/// Initializes a new instance of the <see cref="JellyfinDb"/> class.
/// </summary>
/// <param name="options">The database context options.</param>
public JellyfinDb(DbContextOptions<JellyfinDb> options) : base(options)
{
}
/// <summary>
/// Gets or sets the default connection string.
/// </summary>
public static string ConnectionString { get; set; } = @"Data Source=jellyfin.db";
public virtual DbSet<AccessSchedule> AccessSchedules { get; set; }
2020-05-02 22:32:22 +00:00
public virtual DbSet<ActivityLog> ActivityLogs { get; set; }
2020-05-15 21:24:01 +00:00
2020-07-01 01:44:41 +00:00
public virtual DbSet<DisplayPreferences> DisplayPreferences { get; set; }
public virtual DbSet<ImageInfo> ImageInfos { get; set; }
public virtual DbSet<ItemDisplayPreferences> ItemDisplayPreferences { get; set; }
public virtual DbSet<CustomItemDisplayPreferences> CustomItemDisplayPreferences { get; set; }
2020-05-15 21:24:01 +00:00
public virtual DbSet<Permission> Permissions { get; set; }
public virtual DbSet<Preference> Preferences { get; set; }
2020-05-21 04:22:43 +00:00
public virtual DbSet<User> Users { get; set; }
2020-06-15 21:43:52 +00:00
/*public virtual DbSet<Artwork> Artwork { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Book> Books { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<BookMetadata> BookMetadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Chapter> Chapters { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Collection> Collections { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<CollectionItem> CollectionItems { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Company> Companies { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<CompanyMetadata> CompanyMetadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<CustomItem> CustomItems { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<CustomItemMetadata> CustomItemMetadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Episode> Episodes { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<EpisodeMetadata> EpisodeMetadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Genre> Genres { get; set; }
2020-06-15 21:43:52 +00:00
public virtual DbSet<Group> Groups { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Library> Libraries { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<LibraryItem> LibraryItems { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<LibraryRoot> LibraryRoot { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<MediaFile> MediaFiles { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<MediaFileStream> MediaFileStream { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Metadata> Metadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<MetadataProvider> MetadataProviders { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<MetadataProviderId> MetadataProviderIds { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Movie> Movies { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<MovieMetadata> MovieMetadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<MusicAlbum> MusicAlbums { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<MusicAlbumMetadata> MusicAlbumMetadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Person> People { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<PersonRole> PersonRoles { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Photo> Photo { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<PhotoMetadata> PhotoMetadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<ProviderMapping> ProviderMappings { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Rating> Ratings { get; set; }
/// <summary>
/// Repository for global::Jellyfin.Data.Entities.RatingSource - This is the entity to
/// store review ratings, not age ratings.
2020-05-02 22:32:22 +00:00
/// </summary>
public virtual DbSet<RatingSource> RatingSources { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Release> Releases { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Season> Seasons { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<SeasonMetadata> SeasonMetadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Series> Series { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<SeriesMetadata> SeriesMetadata { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-02 22:32:22 +00:00
public virtual DbSet<Track> Tracks { get; set; }
2020-06-15 21:43:52 +00:00
2020-05-13 02:25:45 +00:00
public virtual DbSet<TrackMetadata> TrackMetadata { get; set; }*/
2020-05-02 22:32:22 +00:00
2020-05-23 00:20:18 +00:00
/// <inheritdoc/>
public override int SaveChanges()
2020-05-02 22:32:22 +00:00
{
2020-05-23 00:20:18 +00:00
foreach (var saveEntity in ChangeTracker.Entries()
.Where(e => e.State == EntityState.Modified)
.Select(entry => entry.Entity)
.OfType<IHasConcurrencyToken>())
2020-05-23 00:20:18 +00:00
{
saveEntity.OnSavingChanges();
}
2020-05-02 22:32:22 +00:00
2020-05-23 00:20:18 +00:00
return base.SaveChanges();
}
2020-05-02 22:32:22 +00:00
/// <inheritdoc />
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
modelBuilder.SetDefaultDateTimeKind(DateTimeKind.Utc);
2020-05-02 22:32:22 +00:00
base.OnModelCreating(modelBuilder);
modelBuilder.HasDefaultSchema("jellyfin");
// Collations
modelBuilder.Entity<User>()
.Property(user => user.Username)
.UseCollation("NOCASE");
// Delete behavior
2021-03-18 01:41:52 +00:00
modelBuilder.Entity<User>()
.HasOne(u => u.ProfileImage)
.WithOne()
.OnDelete(DeleteBehavior.Cascade);
modelBuilder.Entity<User>()
.HasMany(u => u.Permissions)
.WithOne()
2021-03-19 04:26:00 +00:00
.HasForeignKey(p => p.UserId)
2021-03-18 01:41:52 +00:00
.OnDelete(DeleteBehavior.Cascade);
modelBuilder.Entity<User>()
.HasMany(u => u.Preferences)
.WithOne()
2021-03-19 04:26:00 +00:00
.HasForeignKey(p => p.UserId)
2021-03-18 01:41:52 +00:00
.OnDelete(DeleteBehavior.Cascade);
modelBuilder.Entity<User>()
.HasMany(u => u.AccessSchedules)
.WithOne()
.OnDelete(DeleteBehavior.Cascade);
modelBuilder.Entity<User>()
.HasMany(u => u.DisplayPreferences)
.WithOne()
.OnDelete(DeleteBehavior.Cascade);
2021-03-19 04:26:07 +00:00
modelBuilder.Entity<User>()
.HasMany(u => u.ItemDisplayPreferences)
.WithOne()
.OnDelete(DeleteBehavior.Cascade);
modelBuilder.Entity<DisplayPreferences>()
.HasMany(d => d.HomeSections)
.WithOne()
.OnDelete(DeleteBehavior.Cascade);
// Indexes
modelBuilder.Entity<User>()
.HasIndex(entity => entity.Username)
.IsUnique();
modelBuilder.Entity<DisplayPreferences>()
2020-12-04 23:00:11 +00:00
.HasIndex(entity => new { entity.UserId, entity.ItemId, entity.Client })
.IsUnique();
modelBuilder.Entity<CustomItemDisplayPreferences>()
2020-12-04 23:00:11 +00:00
.HasIndex(entity => new { entity.UserId, entity.ItemId, entity.Client, entity.Key })
.IsUnique();
// Used to get a user's permissions or a specific permission for a user.
// Also prevents multiple values being created for a user.
// Filtered over non-null user ids for when other entities (groups, API keys) get permissions
modelBuilder.Entity<Permission>()
.HasIndex(p => new { p.UserId, p.Kind })
.HasFilter("[UserId] IS NOT NULL")
.IsUnique();
modelBuilder.Entity<Preference>()
.HasIndex(p => new { p.UserId, p.Kind })
.HasFilter("[UserId] IS NOT NULL")
.IsUnique();
2020-05-02 22:32:22 +00:00
}
}
}