DigitalData.StaffDBServer/DAL/WebApiContext.cs

using DAL._Shared.SharedModels;
using DAL.Models.Entities;
using HRD.WebApi.DAL;

using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Metadata;
using System.Collections;
using System.Collections.Generic;
using System.Linq;

namespace DAL
{
    public partial class WebApiContext : WebApiBaseContext
    {
        public WebApiContext(DbContextOptions<WebApiContext> options) : base(options)
        {
        }

        public virtual DbSet<WebAppUser> WebAppUserSet { get; set; }
        public virtual DbSet<WebAppEmployeeInfo> WebAppEmployeeInfo { get; set; }

        public virtual DbSet<AdWebAppToWebAppRole> AdWebAppToWebAppRoleSet { get; set; }
        public virtual DbSet<CostCentre> CostCentreSet { get; set; }
        public virtual DbSet<Department> DepartmentSet { get; set; }
        public virtual DbSet<DocumentArt> DocumentArtSet { get; set; }
        public virtual DbSet<EmployeeAttribute> EmployeeAttributeSet { get; set; }
        public virtual DbSet<EmployeeStatus> EmployeeStatusSet { get; set; }
        public virtual DbSet<Project> ProjectSet { get; set; }
        public virtual DbSet<Rang> RangSet { get; set; }
        public virtual DbSet<WebApp> WebAppSet { get; set; }
        public virtual DbSet<WebAppRole> WebAppRoleSet { get; set; }
        public virtual DbSet<DepartmentToWebAppToEmployeeForWindream> DepartmentToWebAppToEmployeeForWindreamSet { get; set; }
        public virtual DbSet<DocumentArtToDepartment> DocumentArtToDepartmentSet { get; set; }
        public virtual DbSet<Employee> EmployeeSet { get; set; }
        public virtual DbSet<EmployeeToAttribute> EmployeeToAttributeSet { get; set; }
        public virtual DbSet<EmployeeToDepartment> EmployeeToDepartmentSet { get; set; }
        public virtual DbSet<EmployeeToWebApp> EmployeeToWebAppSet { get; set; }
        public virtual DbSet<WebAppAdditionalRole> WebAppAdditionalRoleSet { get; set; }
        public virtual DbSet<WebAppToDepartment> WebAppToDepartmentSet { get; set; }
        public virtual DbSet<WebAppToWebAppAdditionalRole> WebAppToWebAppAdditionalRoleSet { get; set; }
        public virtual DbSet<WebAppToWebAppRole> WebAppToWebAppRoleSet { get; set; }
        public virtual DbSet<WindreamColumnsToDepartment> WindreamColumnsToDepartmentSet { get; set; }
        public virtual DbSet<WindreamIndex> WindreamIndexSet { get; set; }
        public virtual DbSet<WindreamIndexToWindreamSearchToDepartment> WindreamIndexToWindreamSearchToDepartmentSet { get; set; }
        public virtual DbSet<WindreamSearch> WindreamSearchSet { get; set; }
        public virtual DbSet<WindreamSearchItem> WindreamSearchItemSet { get; set; }
        public virtual DbSet<WindreamSearchItemToWindreamSearchToDepartment> WindreamSearchItemToWindreamSearchToDepartmentSet { get; set; }
        public virtual DbSet<WindreamSearchToDepartment> WindreamSearchToDepartmentSet { get; set; }
        public virtual DbSet<WindreamInputFolder> WindreamInputFolderSet { get; set; }
        public virtual DbSet<Subsidiary> SubsidiarySet { get; set; }
        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            modelBuilder.Entity<Subsidiary>(entity =>
            {
                entity.ToTable("TBSTF_SUBSIDIARY", "dbo");
            });

            modelBuilder.Entity<WindreamInputFolder>(entity =>
            {
                entity.ToTable("TBSTF_WINDREAM_INPUT_FOLDER", "dbo");
            });

            modelBuilder.Entity<WindreamSearchToDepartment>(entity =>
            {
                entity.ToTable("TBSTF_WINDREAM_SEARCH_TO_DEPARTMENT", "dbo");
            });

            modelBuilder.Entity<WindreamSearchItemToWindreamSearchToDepartment>(entity =>
            {
                entity.ToTable("TBSTF_WINDREAM_SEARCH_ITEM_TO_WINDREAM_SEARCH_TO_DEPARTMENT", "dbo");
            });

            modelBuilder.Entity<WindreamSearchItem>(entity =>
            {
                entity.ToTable("TBSTF_WINDREAM_SEARCH_ITEM", "dbo");
            });

            modelBuilder.Entity<WindreamSearch>(entity =>
            {
                entity.ToTable("TBSTF_WINDREAM_SEARCH", "dbo");
            });

            modelBuilder.Entity<WindreamIndexToWindreamSearchToDepartment>(entity =>
            {
                entity.ToTable("TBSTF_WINDREAM_INDEX_TO_WINDREAM_SEARCH_TO_DEPARTMENT", "dbo");
            });

            modelBuilder.Entity<WindreamIndex>(entity =>
            {
                entity.ToTable("TBSTF_WINDREAM_INDEX", "dbo");
            });

            modelBuilder.Entity<WindreamColumnsToDepartment>(entity =>
            {
                entity.ToTable("TBSTF_WINDREAM_COLUMNS_TO_DEPARTMENT", "dbo");
            });

            modelBuilder.Entity<WebAppToWebAppRole>(entity =>
            {
                entity.ToTable("TBSTF_WEB_APP_TO_WEB_APP_ROLE", "dbo");
            });

            modelBuilder.Entity<WebAppToWebAppAdditionalRole>(entity =>
            {
                entity.ToTable("TBSTF_WEB_APP_TO_WEB_APP_ADDITIONAL_ROLE", "dbo");
            });

            modelBuilder.Entity<WebAppToDepartment>(entity =>
            {
                entity.ToTable("TBSTF_WEB_APP_TO_DEPARTMENT", "dbo");
            });

            modelBuilder.Entity<WebAppAdditionalRole>(entity =>
            {
                entity.ToTable("TBSTF_WEB_APP_ADDITIONAL_ROLE", "dbo");
            });

            modelBuilder.Entity<EmployeeToWebApp>(entity =>
            {
                entity.ToTable("TBSTF_EMPLOYEE_TO_WEB_APP", "dbo");
            });

            modelBuilder.Entity<EmployeeToDepartment>(entity =>
            {
                entity.ToTable("TBSTF_EMPLOYEE_TO_DEPARTMENT", "dbo");
            });

            modelBuilder.Entity<EmployeeToAttribute>(entity =>
            {
                entity.ToTable("TBSTF_EMPLOYEE_TO_ATTRIBUTE", "dbo");
            });

            modelBuilder.Entity<Employee>(entity =>
            {
                entity.ToTable("TBSTF_EMPLOYEE", "dbo");
            });

            modelBuilder.Entity<DocumentArtToDepartment>(entity =>
            {
                entity.ToTable("TBSTF_DOCUMENT_ART_TO_DEPARTMENT", "dbo");
            });

            modelBuilder.Entity<DepartmentToWebAppToEmployeeForWindream>(entity =>
            {
                entity.ToTable("TBSTF_DEPARTMENT_TO_WEB_APP_TO_EMPLOYEE_FOR_WINDREAM", "dbo");
            });

            modelBuilder.Entity<WebAppRole>(entity =>
            {
                entity.ToTable("TBSTF_WEB_APP_ROLE", "dbo");
            });

            modelBuilder.Entity<WebApp>(entity =>
            {
                entity.ToTable("TBSTF_WEB_APP", "dbo");
            });

            modelBuilder.Entity<Rang>(entity =>
            {
                entity.ToTable("TBSTF_RANG", "dbo");
            });

            modelBuilder.Entity<Project>(entity =>
            {
                entity.ToTable("TBSTF_PROJECT", "dbo");
            });

            modelBuilder.Entity<EmployeeStatus>(entity =>
            {
                entity.ToTable("TBSTF_EMPLOYEE_STATUS", "dbo");
            });

            modelBuilder.Entity<EmployeeAttribute>(entity =>
            {
                entity.ToTable("TBSTF_EMPLOYEE_ATTRIBUTE", "dbo");
            });

            modelBuilder.Entity<DocumentArt>(entity =>
            {
                entity.ToTable("TBSTF_DOCUMENT_ART", "dbo");
            });

            modelBuilder.Entity<Department>(entity =>
            {
                entity.ToTable("TBSTF_DEPARTMENT", "dbo");
            });

            modelBuilder.Entity<CostCentre>(entity =>
            {
                entity.ToTable("TBSTF_COST_CENTRE", "dbo");
            });

            modelBuilder.Entity<AdWebAppToWebAppRole>(entity =>
            {
                entity.ToTable("TBSTF_AD_WEB_APP_TO_WEB_APP_ROLE", "dbo");
            });

            modelBuilder.Entity<WebAppUser>(entity =>
            {
                entity.ToTable("TBSTF_WEB_APP_USER", "dbo");
            });

            modelBuilder.Entity<WebAppEmployeeInfo>(entity =>
            {
                entity.ToTable("TBSTF_WEB_APP_EMPLOYEE_INFO", "dbo");
            });

            Configure4DDNaming(modelBuilder.Model.GetEntityTypes().ToArray());

            base.OnModelCreating(modelBuilder);
        }

        public static void Configure4DDNaming(params IMutableEntityType[] entityTypes)
        {
            foreach (var entityType in entityTypes)
            {
                // Configure column names
                foreach (var property in entityType.GetProperties())
                {
                    var columnName = ToSnakeCase(property.Name);
                    property.SetColumnName(columnName);
                }
            }
        }

        private static string ToSnakeCase(string input)
        {
            if (string.IsNullOrEmpty(input))
            {
                return input;
            }

            var startUnderscores = System.Text.RegularExpressions.Regex.Match(input, @"^_+");
            return startUnderscores + System.Text.RegularExpressions.Regex.Replace(input, @"([a-z0-9])([A-Z])", "$1_$2").ToUpper();
        }
    }
}