// /*************************************************************************** // Aaru Data Preservation Suite // ---------------------------------------------------------------------------- // // Filename : Context.cs // Author(s) : Natalia Portillo // // Component : Aaru Server. // // --[ Description ] ---------------------------------------------------------- // // Entity framework database context. // // --[ License ] -------------------------------------------------------------- // // This library is free software; you can redistribute it and/or modify // it under the terms of the GNU Lesser General Public License as // published by the Free Software Foundation; either version 2.1 of the // License, or (at your option) any later version. // // This library 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 // Lesser General Public License for more details. // // You should have received a copy of the GNU Lesser General Public // License along with this library; if not, see . // // ---------------------------------------------------------------------------- // Copyright © 2011-2024 Natalia Portillo // ****************************************************************************/ using System.Data.Common; using Aaru.CommonTypes.Metadata; using Aaru.Server.Database.Models; using Microsoft.AspNetCore.Identity; using Microsoft.AspNetCore.Identity.EntityFrameworkCore; using Microsoft.EntityFrameworkCore; using Microsoft.Extensions.Configuration; using OperatingSystem = Aaru.Server.Database.Models.OperatingSystem; using Version = Aaru.Server.Database.Models.Version; namespace Aaru.Server.Database; public sealed class DbContext : IdentityDbContext { public DbContext() {} public DbContext(DbContextOptions options) : base(options) {} public DbSet Devices { get; set; } public DbSet Reports { get; set; } public DbSet Commands { get; set; } public DbSet DeviceStats { get; set; } public DbSet Filesystems { get; set; } public DbSet Filters { get; set; } public DbSet Medias { get; set; } public DbSet MediaFormats { get; set; } public DbSet OperatingSystems { get; set; } public DbSet Partitions { get; set; } public DbSet Versions { get; set; } public DbSet UsbVendors { get; set; } public DbSet UsbProducts { get; set; } public DbSet CdOffsets { get; set; } public DbSet Ata { get; set; } public DbSet BlockDescriptor { get; set; } public DbSet Chs { get; set; } public DbSet FireWire { get; set; } public DbSet Mmc { get; set; } public DbSet MmcSd { get; set; } public DbSet MmcFeatures { get; set; } public DbSet Pcmcia { get; set; } public DbSet Scsi { get; set; } public DbSet ScsiMode { get; set; } public DbSet ScsiPage { get; set; } public DbSet Ssc { get; set; } public DbSet SupportedDensity { get; set; } public DbSet TestedMedia { get; set; } public DbSet TestedSequentialMedia { get; set; } public DbSet Usb { get; set; } public DbSet RemoteApplications { get; set; } public DbSet RemoteArchitectures { get; set; } public DbSet RemoteOperatingSystems { get; set; } public DbSet GdRomSwapDiscCapabilities { get; set; } public DbSet NesHeaders { get; set; } protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder) { if(optionsBuilder.IsConfigured) return; IConfigurationBuilder builder = new ConfigurationBuilder().AddJsonFile("appsettings.json"); IConfigurationRoot configuration = builder.Build(); optionsBuilder .UseMySql(configuration.GetConnectionString("DefaultConnection"), new MariaDbServerVersion(new System.Version(10, 4, 0))) .UseLazyLoadingProxies(); } protected override void OnModelCreating(ModelBuilder modelBuilder) { base.OnModelCreating(modelBuilder); modelBuilder.Entity("Aaru.CommonTypes.Metadata.Ata", b => b.HasOne("Aaru.CommonTypes.Metadata.TestedMedia", "ReadCapabilities") .WithMany() .HasForeignKey("ReadCapabilitiesId") .OnDelete(DeleteBehavior.SetNull)); modelBuilder.Entity("Aaru.CommonTypes.Metadata.BlockDescriptor", b => b.HasOne("Aaru.CommonTypes.Metadata.ScsiMode", null) .WithMany("BlockDescriptors") .HasForeignKey("ScsiModeId") .OnDelete(DeleteBehavior.Cascade)); modelBuilder.Entity("Aaru.CommonTypes.Metadata.DensityCode", b => b.HasOne("Aaru.CommonTypes.Metadata.SscSupportedMedia", null) .WithMany("DensityCodes") .HasForeignKey("SscSupportedMediaId") .OnDelete(DeleteBehavior.Cascade)); modelBuilder.Entity("Aaru.CommonTypes.Metadata.Mmc", b => b.HasOne("Aaru.CommonTypes.Metadata.MmcFeatures", "Features") .WithMany() .HasForeignKey("FeaturesId") .OnDelete(DeleteBehavior.SetNull)); modelBuilder.Entity("Aaru.CommonTypes.Metadata.Scsi", b => { b.HasOne("Aaru.CommonTypes.Metadata.ScsiMode", "ModeSense") .WithMany() .HasForeignKey("ModeSenseId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Mmc", "MultiMediaDevice") .WithMany() .HasForeignKey("MultiMediaDeviceId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.TestedMedia", "ReadCapabilities") .WithMany() .HasForeignKey("ReadCapabilitiesId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Ssc", "SequentialDevice") .WithMany() .HasForeignKey("SequentialDeviceId") .OnDelete(DeleteBehavior.SetNull); }); modelBuilder.Entity("Aaru.CommonTypes.Metadata.ScsiPage", b => { b.HasOne("Aaru.CommonTypes.Metadata.Scsi", null) .WithMany("EVPDPages") .HasForeignKey("ScsiId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.ScsiMode", null) .WithMany("ModePages") .HasForeignKey("ScsiModeId") .OnDelete(DeleteBehavior.SetNull); }); modelBuilder.Entity("Aaru.CommonTypes.Metadata.SscSupportedMedia", b => { b.HasOne("Aaru.CommonTypes.Metadata.Ssc", null) .WithMany("SupportedMediaTypes") .HasForeignKey("SscId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.TestedSequentialMedia", null) .WithMany("SupportedMediaTypes") .HasForeignKey("TestedSequentialMediaId") .OnDelete(DeleteBehavior.SetNull); }); modelBuilder.Entity("Aaru.CommonTypes.Metadata.SupportedDensity", b => { b.HasOne("Aaru.CommonTypes.Metadata.Ssc", null) .WithMany("SupportedDensities") .HasForeignKey("SscId") .OnDelete(DeleteBehavior.Cascade); b.HasOne("Aaru.CommonTypes.Metadata.TestedSequentialMedia", null) .WithMany("SupportedDensities") .HasForeignKey("TestedSequentialMediaId") .OnDelete(DeleteBehavior.Cascade); }); modelBuilder.Entity("Aaru.CommonTypes.Metadata.TestedMedia", b => { b.HasOne("Aaru.CommonTypes.Metadata.Ata", null) .WithMany("RemovableMedias") .HasForeignKey("AtaId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Chs", "CHS") .WithMany() .HasForeignKey("CHSId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Chs", "CurrentCHS") .WithMany() .HasForeignKey("CurrentCHSId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Mmc", null) .WithMany("TestedMedia") .HasForeignKey("MmcId") .OnDelete(DeleteBehavior.Cascade); b.HasOne("Aaru.CommonTypes.Metadata.Scsi", null) .WithMany("RemovableMedias") .HasForeignKey("ScsiId") .OnDelete(DeleteBehavior.SetNull); }); modelBuilder.Entity("Aaru.CommonTypes.Metadata.TestedSequentialMedia", b => b.HasOne("Aaru.CommonTypes.Metadata.Ssc", null) .WithMany("TestedMedia") .HasForeignKey("SscId") .OnDelete(DeleteBehavior.SetNull)); modelBuilder.Entity("Aaru.Server.Database.Models.Device", b => { b.HasOne("Aaru.CommonTypes.Metadata.Ata", "ATA") .WithMany() .HasForeignKey("ATAId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Ata", "ATAPI") .WithMany() .HasForeignKey("ATAPIId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.Server.Database.Models.CompactDiscOffset", "CdOffset") .WithMany("Devices") .HasForeignKey("CdOffsetId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.FireWire", "FireWire") .WithMany() .HasForeignKey("FireWireId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.MmcSd", "MultiMediaCard") .WithMany() .HasForeignKey("MultiMediaCardId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Pcmcia", "PCMCIA") .WithMany() .HasForeignKey("PCMCIAId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Scsi", "SCSI") .WithMany() .HasForeignKey("SCSIId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.MmcSd", "SecureDigital") .WithMany() .HasForeignKey("SecureDigitalId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Usb", "USB") .WithMany() .HasForeignKey("USBId") .OnDelete(DeleteBehavior.SetNull); }); modelBuilder.Entity("Aaru.Server.Database.Models.DeviceStat", b => b.HasOne("Aaru.Server.Database.Models.Device", "Report") .WithMany() .HasForeignKey("ReportId") .OnDelete(DeleteBehavior.SetNull)); modelBuilder.Entity("Aaru.Server.Database.Models.UploadedReport", b => { b.HasOne("Aaru.CommonTypes.Metadata.Ata", "ATA") .WithMany() .HasForeignKey("ATAId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Ata", "ATAPI") .WithMany() .HasForeignKey("ATAPIId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.FireWire", "FireWire") .WithMany() .HasForeignKey("FireWireId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.MmcSd", "MultiMediaCard") .WithMany() .HasForeignKey("MultiMediaCardId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Pcmcia", "PCMCIA") .WithMany() .HasForeignKey("PCMCIAId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Scsi", "SCSI") .WithMany() .HasForeignKey("SCSIId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.MmcSd", "SecureDigital") .WithMany() .HasForeignKey("SecureDigitalId") .OnDelete(DeleteBehavior.SetNull); b.HasOne("Aaru.CommonTypes.Metadata.Usb", "USB") .WithMany() .HasForeignKey("USBId") .OnDelete(DeleteBehavior.SetNull); }); modelBuilder.Entity().HasIndex(b => b.ModifiedWhen); modelBuilder.Entity().HasIndex(b => b.ModifiedWhen); modelBuilder.Entity().HasIndex(b => b.ModifiedWhen); modelBuilder.Entity().HasIndex(b => b.ProductId); modelBuilder.Entity().HasIndex(b => b.VendorId); modelBuilder.Entity().HasIndex(b => b.ModifiedWhen); modelBuilder.Entity().HasIndex(b => b.VendorId).IsUnique(); modelBuilder.Entity().HasIndex(b => b.ModifiedWhen); modelBuilder.Entity().HasIndex(b => b.Sha256).IsUnique(); } internal static bool TableExists(string tableName) { using var db = new DbContext(); DbConnection connection = db.Database.GetDbConnection(); connection.Open(); DbCommand command = connection.CreateCommand(); command.CommandText = $"SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA=DATABASE() AND TABLE_NAME=\"{tableName}\""; var result = (long)command.ExecuteScalar(); return result != 0; } }