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