romrepomgr/RomRepoMgr.Database/Context.cs

/******************************************************************************
// RomRepoMgr - ROM repository manager
// ----------------------------------------------------------------------------
//
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// --[ License ] --------------------------------------------------------------
//
//     This program is free software: you can redistribute it and/or modify
//     it under the terms of the GNU General Public License as
//     published by the Free Software Foundation, either version 3 of the
//     License, or (at your option) any later version.
//
//     This program is distributed in the hope that it will be useful,
//     but WITHOUT ANY WARRANTY; without even the implied warranty of
//     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//     GNU General Public License for more details.
//
//     You should have received a copy of the GNU General Public License
//     along with this program.  If not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2020-2024 Natalia Portillo
*******************************************************************************/

using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using RomRepoMgr.Database.Models;

namespace RomRepoMgr.Database;

public sealed class Context(DbContextOptions options) : DbContext(options)
{
    public DbSet<DbFile>         Files            { get; set; }
    public DbSet<RomSet>         RomSets          { get; set; }
    public DbSet<Machine>        Machines         { get; set; }
    public DbSet<FileByMachine>  FilesByMachines  { get; set; }
    public DbSet<DbDisk>         Disks            { get; set; }
    public DbSet<DiskByMachine>  DisksByMachines  { get; set; }
    public DbSet<DbMedia>        Medias           { get; set; }
    public DbSet<MediaByMachine> MediasByMachines { get; set; }
    public DbSet<RomSetStat>     RomSetStats      { get; set; }

    public static Context Create(string dbPath, ILoggerFactory loggerFactory)
    {
        var optionsBuilder = new DbContextOptionsBuilder();

        optionsBuilder.UseLazyLoadingProxies()
#if DEBUG
                      .UseLoggerFactory(loggerFactory)
#endif
                      .UseSqlite($"Data Source={dbPath}");

        return new Context(optionsBuilder.Options);
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);

        modelBuilder.Entity<DbFile>(entity =>
        {
            entity.HasIndex(e => e.Crc32);

            entity.HasIndex(e => e.Md5);

            entity.HasIndex(e => e.Sha1);

            entity.HasIndex(e => e.Sha256);

            entity.HasIndex(e => e.Sha384);

            entity.HasIndex(e => e.Sha512);

            entity.HasIndex(e => e.Size);

            entity.HasIndex(e => e.IsInRepo);

            entity.HasIndex(e => new
            {
                e.Crc32,
                e.Size
            });

            entity.HasIndex(e => new
            {
                e.Md5,
                e.Size
            });

            entity.HasIndex(e => new
            {
                e.Sha1,
                e.Size
            });

            entity.HasIndex(e => new
            {
                e.Sha256,
                e.Size
            });

            entity.HasIndex(e => new
            {
                e.Sha384,
                e.Size
            });

            entity.HasIndex(e => new
            {
                e.Sha512,
                e.Size
            });
        });

        modelBuilder.Entity<RomSet>(entity =>
        {
            entity.HasIndex(e => e.Author);

            entity.HasIndex(e => e.Comment);

            entity.HasIndex(e => e.Date);

            entity.HasIndex(e => e.Description);

            entity.HasIndex(e => e.Homepage);

            entity.HasIndex(e => e.Name);

            entity.HasIndex(e => e.Version);

            entity.HasIndex(e => e.Filename);

            entity.HasIndex(e => e.Sha384);

            entity.HasIndex(e => e.Category);
        });

        modelBuilder.Entity<Machine>(entity =>
        {
            entity.HasIndex(e => e.Name);

            entity.HasOne(e => e.RomSet).WithMany(e => e.Machines).OnDelete(DeleteBehavior.Cascade);
        });

        modelBuilder.Entity<FileByMachine>(entity =>
        {
            entity.HasIndex(e => e.Name);

            entity.HasOne(e => e.Machine).WithMany(e => e.Files).OnDelete(DeleteBehavior.Cascade);

            entity.HasOne(e => e.File).WithMany(e => e.Machines).OnDelete(DeleteBehavior.Cascade);
        });

        modelBuilder.Entity<DbDisk>(entity =>
        {
            entity.HasIndex(e => e.Md5);

            entity.HasIndex(e => e.Sha1);

            entity.HasIndex(e => e.Size);

            entity.HasIndex(e => e.IsInRepo);
        });

        modelBuilder.Entity<DiskByMachine>(entity =>
        {
            entity.HasIndex(e => e.Name);

            entity.HasOne(e => e.Machine).WithMany(e => e.Disks).OnDelete(DeleteBehavior.Cascade);

            entity.HasOne(e => e.Disk).WithMany(e => e.Machines).OnDelete(DeleteBehavior.Cascade);
        });

        modelBuilder.Entity<DbMedia>(entity =>
        {
            entity.HasIndex(e => e.Md5);

            entity.HasIndex(e => e.Sha1);

            entity.HasIndex(e => e.Sha256);

            entity.HasIndex(e => e.SpamSum);

            entity.HasIndex(e => e.Size);

            entity.HasIndex(e => e.IsInRepo);
        });

        modelBuilder.Entity<MediaByMachine>(entity =>
        {
            entity.HasIndex(e => e.Name);

            entity.HasOne(e => e.Machine).WithMany(e => e.Medias).OnDelete(DeleteBehavior.Cascade);

            entity.HasOne(e => e.Media).WithMany(e => e.Machines).OnDelete(DeleteBehavior.Cascade);
        });

        modelBuilder.Entity<RomSetStat>(entity => entity.HasOne(e => e.RomSet).WithOne(e => e.Statistics));
    }
}