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 Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;

namespace DiscImageChef.Server.Models
{
    public sealed class DicServerContext : DbContext
    {
        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; }

        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
        {
            if (optionsBuilder.IsConfigured) return;


            var builder = new ConfigurationBuilder().AddJsonFile("appsettings.json");
            var configuration = builder.Build();
            optionsBuilder.UseMySql(configuration.GetConnectionString("DefaultConnection"));
        }

        protected override void OnModelCreating(ModelBuilder 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())
            {
                var connection = db.Database.GetDbConnection();
                connection.Open();

                var 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;
            }
        }
    }
}