private static MethodDefinition GetOnModelCreatingMethod(EntityFrameworkCoreProject project) { var lines = new List <ILine>(); var selection = project.GlobalSelection(); if (selection.Settings.UseDataAnnotations) { var primaryKeys = project .Database .Tables .Where(item => item.PrimaryKey != null) .Select(item => item.GetColumnsFromConstraint(item.PrimaryKey).Select(key => key.Name).First()) .ToList(); foreach (var view in project.Database.Views) { var result = view.Columns.Where(item => primaryKeys.Contains(item.Name)).ToList(); if (result.Count == 0) { lines.Add( new CodeLine("modelBuilder.Entity<{0}>().HasKey(e => new {{ {1} }});", project.GetEntityName(view), string.Join(", ", view.Columns.Select(item => string.Format("e.{0}", project.GetPropertyName(view, item)))))); lines.Add(new EmptyLine()); } else { lines.Add( new CodeLine("modelBuilder.Entity<{0}>().HasKey(e => new {{ {1} }});", project.GetEntityName(view), string.Join(", ", result.Select(item => string.Format("e.{0}", project.GetPropertyName(view, item)))))); lines.Add(new EmptyLine()); } } } else { if (project.Database.Tables.Count > 0) { lines.Add(new CommentLine(" Apply all configurations for tables")); lines.Add(new EmptyLine()); lines.Add(new CodeLine("modelBuilder")); foreach (var table in project.Database.Tables) { var existingViews = project.Database.Views.Count(item => item.Name == table.Name); var genericTypeName = existingViews == 0 ? project.GetEntityName(table) : project.GetFullEntityName(table); var name = existingViews == 0 ? project.GetEntityConfigurationName(table) : project.GetFullEntityConfigurationName(table); lines.Add(new CodeLine(1, ".ApplyConfiguration(new {0}())", name)); } lines.Add(new CodeLine(";")); lines.Add(new EmptyLine()); } if (project.Database.Views.Count > 0) { lines.Add(new CommentLine(" Apply all configurations for views")); lines.Add(new EmptyLine()); lines.Add(new CodeLine("modelBuilder")); foreach (var view in project.Database.Views) { var existingTables = project.Database.Tables.Count(item => item.Name == view.Name); var genericTypeName = existingTables == 0 ? project.GetEntityName(view) : project.GetFullEntityName(view); var name = existingTables == 0 ? project.GetEntityConfigurationName(view) : project.GetFullEntityConfigurationName(view); lines.Add(new CodeLine(1, ".ApplyConfiguration(new {0}())", name)); } lines.Add(new CodeLine(";")); lines.Add(new EmptyLine()); } if (project.Database.TableFunctions.Count > 0) { lines.Add(new CommentLine(" Register query types for table functions")); lines.Add(new EmptyLine()); foreach (var view in project.Database.TableFunctions) { lines.Add(new CodeLine("modelBuilder.Query<{0}>();", project.GetEntityResultName(view))); } lines.Add(new EmptyLine()); } if (project.Database.StoredProcedures.Count > 0) { lines.Add(new CommentLine(" Register query types for stored procedures")); lines.Add(new EmptyLine()); foreach (var view in project.Database.StoredProcedures) { lines.Add(new CodeLine("modelBuilder.Query<{0}>();", project.GetEntityResultName(view))); } lines.Add(new EmptyLine()); } } lines.Add(new CodeLine("base.OnModelCreating(modelBuilder);")); return(new MethodDefinition { AccessModifier = AccessModifier.Protected, Type = "void", Name = "OnModelCreating", Parameters = { new ParameterDefinition("ModelBuilder", "modelBuilder") }, IsOverride = true, Lines = lines }); }