Files
Aaru.Server/Aaru.Server.Database/Context.cs

359 lines
18 KiB
C#
Raw Normal View History

2019-11-02 01:40:41 +00:00
// /***************************************************************************
2020-02-29 19:22:23 +00:00
// Aaru Data Preservation Suite
2019-11-02 01:40:41 +00:00
// ----------------------------------------------------------------------------
//
// Filename : Context.cs
// Author(s) : Natalia Portillo <claunia@claunia.com>
//
2020-02-29 19:24:13 +00:00
// Component : Aaru Server.
2019-11-02 01:40:41 +00:00
//
// --[ 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 <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
2024-04-30 05:11:05 +01:00
// Copyright © 2011-2024 Natalia Portillo
2019-11-02 01:40:41 +00:00
// ****************************************************************************/
2019-11-02 23:52:33 +00:00
using System.Data.Common;
2020-02-29 21:04:02 +00:00
using Aaru.CommonTypes.Metadata;
using Aaru.Server.Database.Models;
2019-11-07 00:25:43 +00:00
using Microsoft.AspNetCore.Identity;
using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore;
2019-11-02 21:01:25 +00:00
using Microsoft.Extensions.Configuration;
using OperatingSystem = Aaru.Server.Database.Models.OperatingSystem;
using Version = Aaru.Server.Database.Models.Version;
2019-11-02 01:40:41 +00:00
namespace Aaru.Server.Database;
2021-12-08 17:57:50 +00:00
public sealed class DbContext : IdentityDbContext<IdentityUser>
2019-11-02 01:40:41 +00:00
{
public DbContext() {}
2021-12-08 17:57:50 +00:00
public DbContext(DbContextOptions<DbContext> options) : base(options) {}
2021-12-08 17:57:50 +00:00
public DbSet<Device> Devices { get; set; }
public DbSet<UploadedReport> Reports { get; set; }
public DbSet<Command> Commands { get; set; }
public DbSet<DeviceStat> DeviceStats { get; set; }
public DbSet<Filesystem> Filesystems { get; set; }
public DbSet<Filter> Filters { get; set; }
public DbSet<Media> Medias { get; set; }
public DbSet<MediaFormat> MediaFormats { get; set; }
public DbSet<OperatingSystem> OperatingSystems { get; set; }
public DbSet<Partition> Partitions { get; set; }
public DbSet<Version> Versions { get; set; }
public DbSet<UsbVendor> UsbVendors { get; set; }
public DbSet<UsbProduct> UsbProducts { get; set; }
public DbSet<CompactDiscOffset> CdOffsets { get; set; }
public DbSet<Ata> Ata { get; set; }
public DbSet<BlockDescriptor> BlockDescriptor { get; set; }
public DbSet<Chs> Chs { get; set; }
public DbSet<FireWire> FireWire { get; set; }
public DbSet<Mmc> Mmc { get; set; }
public DbSet<MmcSd> MmcSd { get; set; }
public DbSet<MmcFeatures> MmcFeatures { get; set; }
public DbSet<Pcmcia> Pcmcia { get; set; }
public DbSet<Scsi> Scsi { get; set; }
public DbSet<ScsiMode> ScsiMode { get; set; }
public DbSet<ScsiPage> ScsiPage { get; set; }
public DbSet<Ssc> Ssc { get; set; }
public DbSet<SupportedDensity> SupportedDensity { get; set; }
public DbSet<TestedMedia> TestedMedia { get; set; }
public DbSet<TestedSequentialMedia> TestedSequentialMedia { get; set; }
public DbSet<Usb> Usb { get; set; }
public DbSet<RemoteApplication> RemoteApplications { get; set; }
public DbSet<RemoteArchitecture> RemoteArchitectures { get; set; }
public DbSet<RemoteOperatingSystem> RemoteOperatingSystems { get; set; }
public DbSet<GdRomSwapDiscCapabilities> GdRomSwapDiscCapabilities { get; set; }
2021-12-09 00:06:48 +00:00
public DbSet<NesHeaderInfo> NesHeaders { get; set; }
2021-12-08 17:57:50 +00:00
protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
2019-11-02 01:40:41 +00:00
{
2024-04-30 05:30:44 +01:00
if(optionsBuilder.IsConfigured) return;
2019-11-02 21:01:25 +00:00
2021-12-08 17:57:50 +00:00
IConfigurationBuilder builder = new ConfigurationBuilder().AddJsonFile("appsettings.json");
IConfigurationRoot configuration = builder.Build();
2020-12-21 02:31:12 +00:00
2024-04-30 05:30:44 +01:00
optionsBuilder
.UseMySql(configuration.GetConnectionString("DefaultConnection"),
new MariaDbServerVersion(new System.Version(10, 4, 0)))
.UseLazyLoadingProxies();
2021-12-08 17:57:50 +00:00
}
2019-11-02 21:01:25 +00:00
2021-12-08 17:57:50 +00:00
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
base.OnModelCreating(modelBuilder);
2019-11-07 00:25:43 +00:00
2021-12-08 17:57:50 +00:00
modelBuilder.Entity("Aaru.CommonTypes.Metadata.Ata",
2024-04-30 05:30:44 +01:00
b => b.HasOne("Aaru.CommonTypes.Metadata.TestedMedia", "ReadCapabilities")
.WithMany()
.HasForeignKey("ReadCapabilitiesId")
.OnDelete(DeleteBehavior.SetNull));
2021-12-08 17:57:50 +00:00
modelBuilder.Entity("Aaru.CommonTypes.Metadata.BlockDescriptor",
2024-04-30 05:30:44 +01:00
b => b.HasOne("Aaru.CommonTypes.Metadata.ScsiMode", null)
.WithMany("BlockDescriptors")
.HasForeignKey("ScsiModeId")
.OnDelete(DeleteBehavior.Cascade));
2021-12-08 17:57:50 +00:00
modelBuilder.Entity("Aaru.CommonTypes.Metadata.DensityCode",
2024-04-30 05:30:44 +01:00
b => b.HasOne("Aaru.CommonTypes.Metadata.SscSupportedMedia", null)
.WithMany("DensityCodes")
.HasForeignKey("SscSupportedMediaId")
.OnDelete(DeleteBehavior.Cascade));
2021-12-08 17:57:50 +00:00
modelBuilder.Entity("Aaru.CommonTypes.Metadata.Mmc",
2024-04-30 05:30:44 +01:00
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);
});
2021-12-08 17:57:50 +00:00
modelBuilder.Entity("Aaru.CommonTypes.Metadata.TestedSequentialMedia",
2024-04-30 05:30:44 +01:00
b => b.HasOne("Aaru.CommonTypes.Metadata.Ssc", null)
.WithMany("TestedMedia")
.HasForeignKey("SscId")
.OnDelete(DeleteBehavior.SetNull));
2024-05-04 02:27:21 +01:00
modelBuilder.Entity("Aaru.Server.Database.Models.Device",
2024-04-30 05:30:44 +01:00
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);
2024-05-04 02:27:21 +01:00
b.HasOne("Aaru.Server.Database.Models.CompactDiscOffset", "CdOffset")
2024-04-30 05:30:44 +01:00
.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);
});
2024-05-04 02:27:21 +01:00
modelBuilder.Entity("Aaru.Server.Database.Models.DeviceStat",
b => b.HasOne("Aaru.Server.Database.Models.Device", "Report")
2024-04-30 05:30:44 +01:00
.WithMany()
.HasForeignKey("ReportId")
.OnDelete(DeleteBehavior.SetNull));
2024-05-04 02:27:21 +01:00
modelBuilder.Entity("Aaru.Server.Database.Models.UploadedReport",
2024-04-30 05:30:44 +01:00
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);
});
2021-12-08 17:57:50 +00:00
modelBuilder.Entity<CompactDiscOffset>().HasIndex(b => b.ModifiedWhen);
2019-11-02 21:01:25 +00:00
2021-12-08 17:57:50 +00:00
modelBuilder.Entity<Device>().HasIndex(b => b.ModifiedWhen);
2019-11-02 21:01:25 +00:00
2021-12-08 17:57:50 +00:00
modelBuilder.Entity<UsbProduct>().HasIndex(b => b.ModifiedWhen);
modelBuilder.Entity<UsbProduct>().HasIndex(b => b.ProductId);
modelBuilder.Entity<UsbProduct>().HasIndex(b => b.VendorId);
2019-11-02 21:01:25 +00:00
2021-12-08 17:57:50 +00:00
modelBuilder.Entity<UsbVendor>().HasIndex(b => b.ModifiedWhen);
modelBuilder.Entity<UsbVendor>().HasIndex(b => b.VendorId).IsUnique();
2021-12-09 00:06:48 +00:00
modelBuilder.Entity<NesHeaderInfo>().HasIndex(b => b.ModifiedWhen);
modelBuilder.Entity<NesHeaderInfo>().HasIndex(b => b.Sha256).IsUnique();
2021-12-08 17:57:50 +00:00
}
2019-11-02 21:01:25 +00:00
2021-12-08 17:57:50 +00:00
internal static bool TableExists(string tableName)
{
using var db = new DbContext();
2019-11-07 00:25:43 +00:00
2021-12-08 17:57:50 +00:00
DbConnection connection = db.Database.GetDbConnection();
connection.Open();
2019-11-02 21:01:25 +00:00
2021-12-08 17:57:50 +00:00
DbCommand command = connection.CreateCommand();
2019-11-02 23:52:33 +00:00
2021-12-08 17:57:50 +00:00
command.CommandText =
$"SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA=DATABASE() AND TABLE_NAME=\"{tableName}\"";
2019-11-02 21:01:25 +00:00
2023-10-04 00:09:19 +01:00
var result = (long)command.ExecuteScalar();
2019-11-02 21:01:25 +00:00
2021-12-08 17:57:50 +00:00
return result != 0;
2019-11-02 01:40:41 +00:00
}
}