CodeLiturgy.Dashboard/BlueWest.Api/Extensions/ModelBuilderExtensions.cs

using BlueWest.Data;
using Microsoft.EntityFrameworkCore;

namespace BlueWest.WebApi.EF.Model
{
    /// <summary>
    /// Database model configuration extensions
    /// </summary>
    public static class ModelBuilderExtensions
    {
        #region Initialization
        
        /// <summary>
        ///  Setup the database model
        /// </summary>
        /// <param name="modelBuilder"></param>
        public static void ConfigureCurrentDbModel(this ModelBuilder modelBuilder)
        {
            modelBuilder
                .ConfigureDatabaseKeys()
                .CurrencyModel()
                .ConfigureUserModel();
        }
        
        #endregion

        #region DatabasePK
        
        private static ModelBuilder ConfigureDatabaseKeys(this ModelBuilder modelBuilder)
        {
            // User
            modelBuilder
                .Entity<User>(builder => builder
                .HasKey(user => user.Id))
                .Entity<User>().Property(user => user.Id)
                .ValueGeneratedOnAdd();
            
            // Country
            modelBuilder
                .Entity<Country>(builder => builder
                .HasKey(country => country.Id))
                .Entity<Country>().Property(country =>  country.Id)
                .ValueGeneratedOnAdd();
            
            // Currency
            modelBuilder
                .Entity<Currency>(builder => builder
                .HasKey(currency => currency.Id))
                .Entity<Currency>()
                .Property(currency =>  currency.Id)
                .ValueGeneratedOnAdd();
            
            // Company PK
            modelBuilder
                .Entity<Company>(builder => builder
                .HasKey(company => company.Id))
                .Entity<Company>().Property(company =>  company.Id)
                .ValueGeneratedOnAdd();
             
            // Industry PK
            modelBuilder
                .Entity<Industry>(builder => builder
                .HasKey(industry => industry.Id))
                .Entity<Industry>().Property(industry =>  industry.Id)
                .ValueGeneratedOnAdd();
            
             
            // Product
            modelBuilder
                .Entity<Industry>(builder => builder
                .HasKey(industry => industry.Id));
            
            modelBuilder
                .Entity<Industry>()
                .Property(industry =>  industry.Id)
                .ValueGeneratedOnAdd();

            // FinanceOp
            return 
                modelBuilder.Entity<FinanceOp>(builder => builder
                .HasKey(financeOp => financeOp.Id));
            
            
        }
        #endregion

        #region Database Models

        private static ModelBuilder CurrencyModel(this ModelBuilder modelBuilder)
        {

            modelBuilder
                .Entity<Currency>()
                .HasMany(ub => ub.Countries)
                .WithMany(au => au.Currencies);
        
            modelBuilder.Entity<Country>()
                .HasMany(ub => ub.Currencies)
                .WithMany(au => au.Countries);

            return modelBuilder;

        }

        private static ModelBuilder ConfigureUserModel(this ModelBuilder modelBuilder)
        {

            // User => Country
            modelBuilder
                .Entity<User>(builder => builder
                    .HasOne<Country>()
                    .WithMany(co => co.Users));
            
            
            // Country => User
            modelBuilder
                .Entity<Country>(builder => builder
                    .HasMany<User>()
                    .WithOne(u => u.Country)
                    .HasForeignKey(x => x.CountryId));
            
            
            // FinanceOp => User
            modelBuilder
                .Entity<FinanceOp>(builder => builder
                    .HasOne<User>()
                    .WithMany(ft => ft.FinanceTransactions)
                    .HasForeignKey(x => x.UserId));
            
            
            return modelBuilder;
        }
        
        #endregion
    }
}