Aaru.Server/DiscImageChef.Server/Models/Context.cs

// /***************************************************************************
// The Disc Image Chef
// ----------------------------------------------------------------------------
//
// Filename       : Context.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : DiscImageChef 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 <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2011-2019 Natalia Portillo
// ****************************************************************************/

using System.Data.Common;
using Microsoft.AspNetCore.Identity;
using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;

namespace DiscImageChef.Server.Models
{
    public sealed class DicServerContext : IdentityDbContext<IdentityUser>
    {
        public DicServerContext() { }

        public DicServerContext(DbContextOptions<DicServerContext> options) : base(options) { }

        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<DiscImageChef.CommonTypes.Metadata.Ata> Ata { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.BlockDescriptor> BlockDescriptor { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.Chs> Chs { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.DensityCode> DensityCode { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.FireWire> FireWire { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.Mmc> Mmc { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.MmcSd> MmcSd { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.MmcFeatures> MmcFeatures { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.Pcmcia> Pcmcia { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.Scsi> Scsi { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.ScsiMode> ScsiMode { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.ScsiPage> ScsiPage { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.Ssc> Ssc { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.SupportedDensity> SupportedDensity { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.TestedMedia> TestedMedia { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.TestedSequentialMedia> TestedSequentialMedia { get; set; }
        public DbSet<DiscImageChef.CommonTypes.Metadata.Usb> Usb { 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"));
        }

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

            modelBuilder.Entity<CompactDiscOffset>().HasIndex(b => b.ModifiedWhen);

            modelBuilder.Entity<Device>().HasIndex(b => b.ModifiedWhen);

            modelBuilder.Entity<UsbProduct>().HasIndex(b => b.ModifiedWhen);
            modelBuilder.Entity<UsbProduct>().HasIndex(b => b.ProductId);
            modelBuilder.Entity<UsbProduct>().HasIndex(b => b.VendorId);

            modelBuilder.Entity<UsbVendor>().HasIndex(b => b.ModifiedWhen);
            modelBuilder.Entity<UsbVendor>().HasIndex(b => b.VendorId).IsUnique();
        }

        internal static bool TableExists(string tableName)
        {
            using var db = new DicServerContext();

            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}\"";

            long result = (long)command.ExecuteScalar();

            return result != 0;
        }
    }
}