public static DataContractClassDefinition GetDataContractClassDefinition(this EntityFrameworkCoreProject project, ITable table, bool isDomainDrivenDesign) { var definition = new DataContractClassDefinition { Namespaces = { "System" }, Namespace = isDomainDrivenDesign ? project.GetDomainQueryModelsNamespace() : project.GetDataLayerDataContractsNamespace(), AccessModifier = AccessModifier.Public, Name = project.GetDataContractName(table), DbObject = table }; foreach (var column in table.Columns) { definition.Properties.Add(new PropertyDefinition(AccessModifier.Public, project.Database.ResolveDatabaseType(column), project.GetPropertyName(table, column)) { IsAutomatic = true }); } foreach (var foreignKey in table.ForeignKeys) { var foreignTable = project.Database.FindTable(foreignKey.References); if (foreignTable == null) { continue; } var foreignKeyAlias = NamingConvention.GetCamelCase(project.GetEntityName(foreignTable)); foreach (var column in foreignTable?.GetColumnsWithNoPrimaryKey()) { var col = (Column)column; var propertyName = project.GetPropertyName(foreignTable, col); var target = string.Format("{0}{1}", project.GetEntityName(foreignTable), propertyName); if (definition.Properties.Count(item => item.Name == propertyName) == 0) { definition.Properties.Add(new PropertyDefinition(AccessModifier.Public, project.Database.ResolveDatabaseType(col), target) { IsAutomatic = true }); } } } definition.SimplifyDataTypes(); return(definition); }
public static CSharpClassDefinition GetEntityClassDefinition(this EntityFrameworkCoreProject project, IView view) { var definition = new CSharpClassDefinition { Namespaces = { "System" }, Namespace = project.Database.HasDefaultSchema(view) ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(view.Schema), Name = project.GetEntityName(view), IsPartial = true, Constructors = { new ClassConstructorDefinition() } }; if (!string.IsNullOrEmpty(view.Description)) { definition.Documentation.Summary = view.Description; } var projectSelection = project.GetSelection(view); if (projectSelection.Settings.UseDataAnnotations) { definition.Namespaces.Add("System.ComponentModel.DataAnnotations"); definition.Namespaces.Add("System.ComponentModel.DataAnnotations.Schema"); } foreach (var column in view.Columns) { var propertyType = string.Empty; if (project.Database.HasTypeMappedToClr(column)) { var clrType = project.Database.GetClrMapForType(column); propertyType = clrType.AllowClrNullable ? string.Format("{0}?", clrType.GetClrType().Name) : clrType.GetClrType().Name; } else { propertyType = "object"; } definition.Properties.Add(new PropertyDefinition(propertyType, column.HasSameNameEnclosingType(view) ? column.GetNameForEnclosing() : view.GetPropertyNameHack(column))); } if (projectSelection.Settings.SimplifyDataTypes) { definition.SimplifyDataTypes(); } return(definition); }
public static EntityClassDefinition GetEntityClassDefinition(this EntityFrameworkCoreProject project, IView view) { var definition = new EntityClassDefinition { Namespaces = { "System" }, Namespace = project.Database.HasDefaultSchema(view) ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(view.Schema), AccessModifier = AccessModifier.Public, Name = project.GetEntityName(view), IsPartial = true, Constructors = { new ClassConstructorDefinition { AccessModifier = AccessModifier.Public } }, DbObject = view }; if (!string.IsNullOrEmpty(view.Description)) { definition.Documentation.Summary = view.Description; } var projectSelection = project.GetSelection(view); if (projectSelection.Settings.UseDataAnnotations) { definition.Namespaces.Add("System.ComponentModel.DataAnnotations"); definition.Namespaces.Add("System.ComponentModel.DataAnnotations.Schema"); } foreach (var column in view.Columns) { var propertyType = project.Database.ResolveDatabaseType(column); definition.Properties.Add(new PropertyDefinition { AccessModifier = AccessModifier.Public, Type = propertyType, Name = project.GetPropertyName(view, column), IsAutomatic = true }); } if (projectSelection.Settings.SimplifyDataTypes) { definition.SimplifyDataTypes(); } return(definition); }
public static DataContractClassDefinition GetDataContractClassDefinition(this EntityFrameworkCoreProject project, ITable table) { var definition = new DataContractClassDefinition { Namespaces = { "System" }, Namespace = project.GetDataLayerDataContractsNamespace(), Name = project.GetDataContractName(table) }; foreach (var column in table.Columns) { definition.Properties.Add(new PropertyDefinition(project.Database.ResolveDatabaseType(column), column.GetPropertyName())); } foreach (var foreignKey in table.ForeignKeys) { var foreignTable = project.Database.FindTable(foreignKey.References); if (foreignTable == null) { continue; } var foreignKeyAlias = NamingConvention.GetCamelCase(project.GetEntityName(foreignTable)); foreach (var column in foreignTable?.GetColumnsWithNoPrimaryKey()) { var target = string.Format("{0}{1}", project.GetEntityName(foreignTable), column.GetPropertyName()); if (definition.Properties.Count(item => item.Name == column.GetPropertyName()) == 0) { definition.Properties.Add(new PropertyDefinition(project.Database.ResolveDatabaseType(column), target)); } } } return(definition); }
public static EntityConfigurationClassDefinition GetEntityConfigurationClassDefinition(this EntityFrameworkCoreProject project, IView view) { var definition = new EntityConfigurationClassDefinition { Namespaces = { "Microsoft.EntityFrameworkCore", "Microsoft.EntityFrameworkCore.Metadata.Builders" }, Namespace = project.Database.HasDefaultSchema(view) ? project.GetDataLayerConfigurationsNamespace() : project.GetDataLayerConfigurationsNamespace(view.Schema), AccessModifier = AccessModifier.Public, Name = project.GetEntityConfigurationName(view), Implements = { string.Format("IEntityTypeConfiguration<{0}>", project.GetEntityName(view)) }, DbObject = view }; definition.Namespaces.AddUnique(project.GetEntityLayerNamespace(project.Database.HasDefaultSchema(view) ? string.Empty : view.Schema)); var configLines = new List <ILine> { new CommentLine(" Set configuration for entity") }; if (string.IsNullOrEmpty(view.Schema)) { configLines.Add(new CodeLine("builder.ToTable(\"{0}\");", view.Name)); } else { configLines.Add(new CodeLine("builder.ToTable(\"{0}\", \"{1}\");", view.Name, view.Schema)); } configLines.Add(new EmptyLine()); var primaryKeys = project .Database .Tables .Where(item => item.PrimaryKey != null) .Select(item => item.GetColumnsFromConstraint(item.PrimaryKey).Select(c => c.Name).First()) .ToList(); var result = view.Columns.Where(item => primaryKeys.Contains(item.Name)).ToList(); if (result.Count == 0) { result = view.Columns.Where(item => !item.Nullable).ToList(); } configLines.Add(new CommentLine(" Add configuration for entity's key")); if (result.Count == 1) { configLines.Add(new CodeLine("builder.HasKey(p => {0});", string.Format("p.{0}", project.CodeNamingConvention.GetPropertyName(result.First().Name)))); } else { configLines.Add(new CodeLine("builder.HasKey(p => new {{ {0} }});", string.Join(", ", result.Select(item => string.Format("p.{0}", project.CodeNamingConvention.GetPropertyName(item.Name)))))); } configLines.Add(new EmptyLine()); configLines.Add(new CommentLine(" Set configuration for columns")); for (var i = 0; i < view.Columns.Count; i++) { var column = view.Columns[i]; var valueConversion = default(Type); if (project.Database.HasTypeMappedToClr(column)) { var lines = new List <string> { string.Format("Property(p => p.{0})", project.GetPropertyName(view, column)) }; if (string.Compare(column.Name, project.GetPropertyName(view, column)) != 0) { lines.Add(string.Format("HasColumnName(\"{0}\")", column.Name)); } else if (project.Database.ColumnIsDecimal(column)) { lines.Add(string.Format("HasColumnType(\"{0}({1}, {2})\")", column.Type, column.Prec, column.Scale)); } else if (project.Database.ColumnIsDouble(column) || project.Database.ColumnIsSingle(column)) { lines.Add(string.Format("HasColumnType(\"{0}({1})\")", column.Type, column.Prec)); } if (project.Database.ColumnIsString(column)) { lines.Add(column.Length <= 0 ? string.Format("HasColumnType(\"{0}(max)\")", column.Type) : string.Format("HasColumnType(\"{0}({1})\")", column.Type, column.Length)); } else { lines.Add(string.Format("HasColumnType(\"{0}\")", column.Type)); } // Use ValueConversionMaps to detect and apply ValueConversion Type based on Type if (project.ValueConversionMaps?.TryGetValue(column.Type, out valueConversion) == true) { lines.Add($"HasConversion(typeof({valueConversion?.FullName}))"); } configLines.Add(new CodeLine("builder")); foreach (var line in lines) { configLines.Add(new CodeLine(1, ".{0}", line)); } configLines.Add(new CodeLine(1, ";")); configLines.Add(new EmptyLine()); } else { var lines = new List <string> { string.Format("builder.Ignore(p => p.{0})", project.GetPropertyName(view, column)) }; configLines.Add(new CodeLine("{0};", string.Join(".", lines))); } } definition.Methods.Add(new MethodDefinition { AccessModifier = AccessModifier.Public, Type = "void", Name = "Configure", Parameters = { new ParameterDefinition(string.Format("EntityTypeBuilder<{0}>", project.GetEntityName(view)), "builder") }, Lines = configLines }); return(definition); }
public static EntityConfigurationClassDefinition GetEntityConfigurationClassDefinition(this EntityFrameworkCoreProject project, ITable table) { var definition = new EntityConfigurationClassDefinition { Namespaces = { "Microsoft.EntityFrameworkCore", "Microsoft.EntityFrameworkCore.Metadata.Builders" }, Namespace = project.Database.HasDefaultSchema(table) ? project.GetDataLayerConfigurationsNamespace() : project.GetDataLayerConfigurationsNamespace(table.Schema), AccessModifier = AccessModifier.Public, Name = project.GetEntityConfigurationName(table), DbObject = table }; definition.Namespaces.AddUnique(project.GetEntityLayerNamespace(project.Database.HasDefaultSchema(table) ? string.Empty : table.Schema)); // todo: Check logic to build property's name var propertyType = ""; if (table.HasSameEnclosingName()) { propertyType = string.Join(".", (new string[] { project.ProjectNamespaces.EntityLayer, project.Database.HasDefaultSchema(table) ? string.Empty : table.Schema, project.GetEntityName(table) }).Where(item => !string.IsNullOrEmpty(item))); } else { propertyType = project.GetEntityName(table); } definition.Implements.Add(string.Format("IEntityTypeConfiguration<{0}>", propertyType)); var configLines = new List <ILine> { new CommentLine(" Set configuration for entity") }; if (string.IsNullOrEmpty(table.Schema)) { configLines.Add(new CodeLine("builder.ToTable(\"{0}\");", table.Name)); } else { configLines.Add(new CodeLine("builder.ToTable(\"{0}\", \"{1}\");", table.Name, table.Schema)); } configLines.Add(new EmptyLine()); var columns = default(List <Column>); if (table.PrimaryKey == null || table.PrimaryKey.Key.Count == 0) { configLines.Add(LineHelper.Warning("Add configuration for entity's key")); configLines.Add(new EmptyLine()); } else { configLines.Add(new CommentLine(" Set key for entity")); if (table.PrimaryKey.Key.Count == 1) { configLines.Add(new CodeLine("builder.HasKey(p => p.{0});", project.CodeNamingConvention.GetPropertyName(table.PrimaryKey.Key[0]))); configLines.Add(new EmptyLine()); } else if (table.PrimaryKey.Key.Count > 1) { configLines.Add(new CodeLine("builder.HasKey(p => new {{ {0} }});", string.Join(", ", table.PrimaryKey.Key.Select(item => string.Format("p.{0}", project.CodeNamingConvention.GetPropertyName(item)))))); configLines.Add(new EmptyLine()); } } if (table.Identity != null) { configLines.Add(new CommentLine(" Set identity for entity (auto increment)")); configLines.Add(new CodeLine("builder.Property(p => p.{0}).UseSqlServerIdentityColumn();", project.CodeNamingConvention.GetPropertyName(table.Identity.Name))); configLines.Add(new EmptyLine()); } columns = table.Columns; configLines.Add(new CommentLine(" Set configuration for columns")); for (var i = 0; i < columns.Count; i++) { var column = columns[i]; var valueConversion = default(Type); if (project.Database.HasTypeMappedToClr(column)) { var lines = new List <string> { string.Format("Property(p => p.{0})", project.GetPropertyName(table, column)) }; if (string.Compare(column.Name, project.GetPropertyName(table, column)) != 0) { lines.Add(string.Format("HasColumnName(\"{0}\")", column.Name)); } if (project.Database.ColumnIsByteArray(column)) { lines.Add(string.Format("HasColumnType(\"{0}({1})\")", column.Type, column.Length)); } else if (project.Database.ColumnIsDecimal(column)) { lines.Add(string.Format("HasColumnType(\"{0}({1}, {2})\")", column.Type, column.Prec, column.Scale)); } else if (project.Database.ColumnIsDouble(column) || project.Database.ColumnIsSingle(column)) { lines.Add(string.Format("HasColumnType(\"{0}({1})\")", column.Type, column.Prec)); } else if (project.Database.ColumnIsString(column)) { if (column.Length <= 0) { lines.Add(string.Format("HasColumnType(\"{0}(max)\")", column.Type)); } else { lines.Add(string.Format("HasColumnType(\"{0}\")", column.Type)); lines.Add(string.Format("HasMaxLength({0})", column.Length)); } } else { lines.Add(string.Format("HasColumnType(\"{0}\")", column.Type)); } // Use ValueConversionMaps to detect and apply ValueConversion Type based on Type if (project.ValueConversionMaps.TryGetValue(column.Type, out valueConversion) == true) { lines.Add($".HasConversion(typeof({valueConversion?.FullName}))"); } if (!column.Nullable) { lines.Add("IsRequired()"); } configLines.Add(new CodeLine("builder")); foreach (var line in lines) { configLines.Add(new CodeLine(1, ".{0}", line)); } configLines.Add(new CodeLine(1, ";")); configLines.Add(new EmptyLine()); } else { configLines.Add(new CodeLine("builder.Ignore(p => p.{0});", project.GetPropertyName(table, column))); configLines.Add(new EmptyLine()); } } var projectSelection = project.GetSelection(table); for (var i = 0; i < columns.Count; i++) { var column = columns[i]; if (!string.IsNullOrEmpty(projectSelection.Settings.ConcurrencyToken) && string.Compare(column.Name, projectSelection.Settings.ConcurrencyToken) == 0) { configLines.Add(new CommentLine(" Set concurrency token for entity")); configLines.Add(new CodeLine("builder")); configLines.Add(new CodeLine(1, ".Property(p => p.{0})", project.GetPropertyName(table, column))); configLines.Add(new CodeLine(1, ".ValueGeneratedOnAddOrUpdate()")); configLines.Add(new CodeLine(1, ".IsConcurrencyToken();")); configLines.Add(new EmptyLine()); } } if (projectSelection.Settings.AddConfigurationForUniquesInFluentAPI && table.Uniques.Count > 0) { configLines.Add(new CommentLine(" Add configuration for uniques")); configLines.Add(new EmptyLine()); foreach (var unique in table.Uniques) { configLines.Add(new CodeLine("builder")); if (unique.Key.Count == 1) { configLines.Add(new CodeLine(1, ".HasIndex(p => p.{0})", project.CodeNamingConvention.GetPropertyName(unique.Key.First()))); configLines.Add(new CodeLine(1, ".IsUnique()")); } else { configLines.Add(new CodeLine(1, ".HasIndex(p => new {{ {0} }})", string.Join(", ", unique.Key.Select(item => string.Format("p.{0}", project.CodeNamingConvention.GetPropertyName(item)))))); configLines.Add(new CodeLine(1, ".IsUnique()")); } configLines.Add(new CodeLine(1, ".HasName(\"{0}\");", unique.ConstraintName)); configLines.Add(new EmptyLine()); } } if (projectSelection.Settings.AddConfigurationForForeignKeysInFluentAPI && projectSelection.Settings.DeclareNavigationProperties && table.ForeignKeys.Count > 0) { configLines.Add(new CommentLine(" Add configuration for foreign keys")); configLines.Add(new EmptyLine()); foreach (var foreignKey in table.ForeignKeys) { var foreignTable = project.Database.FindTable(foreignKey.References); if (foreignTable == null || foreignKey.Key.Count == 0) { continue; } if (foreignKey.Key.Count == 1) { var foreignProperty = foreignKey.GetParentNavigationProperty(foreignTable, project); configLines.Add(new CodeLine("builder")); configLines.Add(new CodeLine(1, ".HasOne(p => p.{0})", foreignProperty.Name)); configLines.Add(new CodeLine(1, ".WithMany(b => b.{0})", project.GetNavigationPropertyName(table))); configLines.Add(new CodeLine(1, ".HasForeignKey(p => {0})", string.Format("p.{0}", project.CodeNamingConvention.GetPropertyName(foreignKey.Key.First())))); configLines.Add(new CodeLine(1, ".HasConstraintName(\"{0}\");", foreignKey.ConstraintName)); configLines.Add(new EmptyLine()); } else { configLines.Add(LineHelper.Warning(" Add logic for foreign key with multiple key")); } } } if (projectSelection.Settings.AddConfigurationForDefaultsInFluentAPI && table.Defaults.Count > 0) { configLines.Add(new CommentLine(" Add configuration for defaults")); configLines.Add(new EmptyLine()); foreach (var def in table.Defaults) { var propertyName = def.Key.First(); configLines.Add(new CodeLine("builder")); configLines.Add(new CodeLine(1, ".Property(p => p.{0})", project.GetPropertyName(propertyName))); configLines.Add(new CodeLine(1, ".HasDefaultValueSql(\"{0}\");", def.Value)); configLines.Add(new EmptyLine()); } } definition.Methods.Add(new MethodDefinition { AccessModifier = AccessModifier.Public, Type = "void", Name = "Configure", Parameters = { new ParameterDefinition(string.Format("EntityTypeBuilder<{0}>", propertyType), "builder") }, Lines = configLines }); return(definition); }
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 }); }
public static DbContextClassDefinition GetDbContextClassDefinition(this EntityFrameworkCoreProject project, ProjectSelection <EntityFrameworkCoreProjectSettings> projectSelection) { var definition = new DbContextClassDefinition { Namespaces = { "System", "Microsoft.EntityFrameworkCore", project.GetEntityLayerNamespace() }, Namespace = project.GetDataLayerNamespace(), AccessModifier = AccessModifier.Public, Name = project.GetDbContextName(project.Database), BaseClass = "DbContext" }; if (!projectSelection.Settings.UseDataAnnotations) { definition.Namespaces.Add(project.GetDataLayerConfigurationsNamespace()); } definition.Constructors.Add(new ClassConstructorDefinition { AccessModifier = AccessModifier.Public, Parameters = { new ParameterDefinition(string.Format("DbContextOptions<{0}>", definition.Name), "options") }, Invocation = "base(options)" }); definition.Methods.Add(GetOnModelCreatingMethod(project)); foreach (var table in project.Database.Tables) { if (!project.Database.HasDefaultSchema(table)) { definition.Namespaces.AddUnique(project.GetEntityLayerNamespace(table.Schema)); } 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.GetDbSetPropertyName(table) : project.GetFullDbSetPropertyName(table); definition.Properties.Add( new PropertyDefinition { AccessModifier = AccessModifier.Public, Type = string.Format("DbSet<{0}>", genericTypeName), Name = name, IsAutomatic = true } ); } foreach (var view in project.Database.Views) { if (!project.Database.HasDefaultSchema(view)) { definition.Namespaces.AddUnique(project.GetEntityLayerNamespace(view.Schema)); } 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.GetDbSetPropertyName(view) : project.GetFullDbSetPropertyName(view); definition.Properties.Add( new PropertyDefinition { AccessModifier = AccessModifier.Public, Type = string.Format("DbSet<{0}>", genericTypeName), Name = name, IsAutomatic = true } ); } foreach (var table in project.Database.Tables) { if (!projectSelection.Settings.UseDataAnnotations && !project.Database.HasDefaultSchema(table)) { definition.Namespaces.AddUnique(project.GetDataLayerConfigurationsNamespace(table.Schema)); } } foreach (var view in project.Database.Views) { if (!projectSelection.Settings.UseDataAnnotations && !project.Database.HasDefaultSchema(view)) { definition.Namespaces.AddUnique(project.GetDataLayerConfigurationsNamespace(view.Schema)); } } foreach (var scalarFunction in project.Database.ScalarFunctions) { var parameterType = string.Empty; if (project.Database.HasTypeMappedToClr(scalarFunction.Parameters[0])) { var clrType = project.Database.GetClrMapForType(scalarFunction.Parameters[0]); parameterType = clrType.AllowClrNullable ? string.Format("{0}?", clrType.GetClrType().Name) : clrType.GetClrType().Name; } else { parameterType = "object"; } var method = new MethodDefinition { Attributes = { new MetadataAttribute("DbFunction") { Sets = { new MetadataAttributeSet("FunctionName", string.Format("\"{0}\"", scalarFunction.Name)), new MetadataAttributeSet("Schema", string.Format("\"{0}\"", scalarFunction.Schema)) } } }, IsStatic = true, Type = parameterType, AccessModifier = AccessModifier.Public, Name = project.GetScalarFunctionMethodName(scalarFunction), Lines = { new CodeLine("throw new Exception();") } }; var parameters = scalarFunction.Parameters.Where(item => !string.IsNullOrEmpty(item.Name)).ToList(); foreach (var parameter in parameters) { var propertyType = project.Database.ResolveDatabaseType(parameter); method.Parameters.Add(new ParameterDefinition(parameterType, project.GetPropertyName(parameter))); } definition.Methods.Add(method); } if (projectSelection.Settings.SimplifyDataTypes) { definition.SimplifyDataTypes(); } return(definition); }
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(c => c.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}", item.GetPropertyName()))))); lines.Add(new CodeLine()); } else { lines.Add(new CodeLine("modelBuilder.Entity<{0}>().HasKey(e => new {{ {1} }});", project.GetEntityName(view), string.Join(", ", result.Select(item => string.Format("e.{0}", item.GetPropertyName()))))); lines.Add(new CodeLine()); } } } else { if (project.Database.Tables.Count > 0) { lines.Add(new CommentLine(" Apply all configurations for tables")); lines.Add(new CodeLine()); lines.Add(new CodeLine("modelBuilder")); foreach (var table in project.Database.Tables) { lines.Add(new CodeLine(1, ".ApplyConfiguration(new {0}())", project.GetEntityConfigurationName(table))); } lines.Add(new CodeLine(";")); lines.Add(new CodeLine()); } if (project.Database.Views.Count > 0) { lines.Add(new CommentLine(" Apply all configurations for views")); lines.Add(new CodeLine()); lines.Add(new CodeLine("modelBuilder")); foreach (var view in project.Database.Views) { lines.Add(new CodeLine(1, ".ApplyConfiguration(new {0}())", project.GetEntityConfigurationName(view))); } lines.Add(new CodeLine(";")); lines.Add(new CodeLine()); } } lines.Add(new CodeLine("base.OnModelCreating(modelBuilder);")); return(new MethodDefinition(AccessModifier.Protected, "void", "OnModelCreating", new ParameterDefinition("ModelBuilder", "modelBuilder")) { IsOverride = true, Lines = lines }); }
public static EntityClassDefinition GetEntityClassDefinition(this EntityFrameworkCoreProject project, ITable table) { var definition = new EntityClassDefinition { Namespaces = { "System" }, Namespace = project.Database.HasDefaultSchema(table) ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(table.Schema), Name = project.GetEntityName(table), IsPartial = true, Constructors = { new ClassConstructorDefinition() } }; if (!string.IsNullOrEmpty(table.Description)) { definition.Documentation.Summary = table.Description; } var projectSelection = project.GetSelection(table); if (projectSelection.Settings.UseDataAnnotations) { definition.Namespaces.Add("System.ComponentModel.DataAnnotations"); definition.Namespaces.Add("System.ComponentModel.DataAnnotations.Schema"); } if (projectSelection.Settings.EnableDataBindings) { definition.Namespaces.Add("System.ComponentModel"); definition.Implements.Add("INotifyPropertyChanged"); definition.Events.Add(new EventDefinition("PropertyChangedEventHandler", "PropertyChanged")); } if (table.PrimaryKey != null) { var constructor = new ClassConstructorDefinition(); foreach (var key in table.GetColumnsFromConstraint(table.PrimaryKey)) { var propertyType = string.Empty; if (project.Database.HasTypeMappedToClr(key)) { var clrType = project.Database.GetClrMapForType(key); propertyType = clrType.AllowClrNullable ? string.Format("{0}?", clrType.GetClrType().Name) : clrType.GetClrType().Name; } else { propertyType = "object"; } constructor.Parameters.Add(new ParameterDefinition(propertyType, project.GetParameterName(key))); constructor.Lines.Add(new CodeLine("{0} = {1};", key.GetPropertyName(), project.GetParameterName(key))); } definition.Constructors.Add(constructor); } var columns = table.Columns; foreach (var column in columns) { var propertyType = string.Empty; if (project.Database.HasTypeMappedToClr(column)) { var clrType = project.Database.GetClrMapForType(column); propertyType = clrType.AllowClrNullable ? string.Format("{0}?", clrType.GetClrType().Name) : clrType.GetClrType().Name; } else { propertyType = "object"; } if (projectSelection.Settings.EnableDataBindings) { definition.AddViewModelProperty(propertyType, column.HasSameNameEnclosingType(table) ? column.GetNameForEnclosing() : table.GetPropertyNameHack(column)); } else { if (projectSelection.Settings.BackingFields.Contains(table.GetFullColumnName(column))) { definition.AddPropertyWithField(propertyType, column.HasSameNameEnclosingType(table) ? column.GetNameForEnclosing() : table.GetPropertyNameHack(column)); } else if (projectSelection.Settings.UseAutomaticPropertiesForEntities) { definition.Properties.Add(new PropertyDefinition(propertyType, column.HasSameNameEnclosingType(table) ? column.GetNameForEnclosing() : table.GetPropertyNameHack(column))); } else { definition.AddPropertyWithField(propertyType, column.HasSameNameEnclosingType(table) ? column.GetNameForEnclosing() : table.GetPropertyNameHack(column)); } } } if (projectSelection.Settings.AuditEntity == null) { definition.Implements.Add(projectSelection.Settings.EntityInterfaceName); } else { var count = 0; foreach (var column in columns) { if (projectSelection.Settings.AuditEntity.Names.Contains(column.Name)) { count += 1; } } if (count == projectSelection.Settings.AuditEntity.Names.Length) { definition.Implements.Add("IAuditEntity"); } else { definition.Implements.Add("IEntity"); } } if (projectSelection.Settings.SimplifyDataTypes) { definition.SimplifyDataTypes(); } if (projectSelection.Settings.DeclareNavigationProperties) { foreach (var foreignKey in table.ForeignKeys) { var foreignTable = project.Database.FindTable(foreignKey.References); if (foreignTable == null) { continue; } definition.Namespaces .AddUnique(project.Database.HasDefaultSchema(foreignTable) ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(foreignTable.Schema)); definition.Namespace = project.Database.HasDefaultSchema(table) ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(table.Schema); var fkProperty = foreignKey.GetParentNavigationProperty(foreignTable, project); if (definition.Properties.FirstOrDefault(item => item.Name == fkProperty.Name) == null) { definition.Properties.Add(fkProperty); } } foreach (var child in project.Database.Tables) { foreach (var foreignKey in child.ForeignKeys) { if (foreignKey.References.EndsWith(table.FullName)) { definition.Namespaces .AddUnique(projectSelection.Settings.NavigationPropertyEnumerableNamespace); definition.Namespaces .AddUnique(project.Database.HasDefaultSchema(child) ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(child.Schema)); var navigationProperty = project.GetChildNavigationProperty(projectSelection, child, foreignKey); if (definition.Properties.FirstOrDefault(item => item.Name == navigationProperty.Name) == null) { definition.Properties.Add(navigationProperty); } } } } } return(definition); }
public static CSharpClassDefinition GetEntityTypeConfigurationClassDefinition(this EntityFrameworkCoreProject project, IView view) { var definition = new CSharpClassDefinition { Namespaces = { "Microsoft.EntityFrameworkCore", "Microsoft.EntityFrameworkCore.Metadata.Builders" }, Namespace = project.GetDataLayerConfigurationsNamespace(), Name = project.GetEntityConfigurationName(view), Implements = { string.Format("IEntityTypeConfiguration<{0}>", project.GetEntityName(view)) } }; definition.Namespaces.AddUnique(project.GetEntityLayerNamespace(project.Database.HasDefaultSchema(view) ? string.Empty : view.Schema)); var configLines = new List <ILine> { new CommentLine(" Set configuration for entity") }; if (string.IsNullOrEmpty(view.Schema)) { configLines.Add(new CodeLine("builder.ToTable(\"{0}\");", view.Name)); } else { configLines.Add(new CodeLine("builder.ToTable(\"{0}\", \"{1}\");", view.Name, view.Schema)); } configLines.Add(new CodeLine()); var primaryKeys = project.Database.Tables.Where(item => item.PrimaryKey != null).Select(item => item.GetColumnsFromConstraint(item.PrimaryKey).Select(c => c.Name).First()).ToList(); var result = view.Columns.Where(item => primaryKeys.Contains(item.Name)).ToList(); if (result.Count == 0) { result = view.Columns.Where(item => !item.Nullable).ToList(); } configLines.Add(new CommentLine(" Add configuration for entity's key")); configLines.Add(new CodeLine("builder.HasKey(p => new {{ {0} }});", string.Join(", ", result.Select(item => string.Format("p.{0}", definition.NamingConvention.GetPropertyName(item.Name)))))); configLines.Add(new CodeLine()); configLines.Add(new CommentLine(" Set configuration for columns")); for (var i = 0; i < view.Columns.Count; i++) { var column = view.Columns[i]; if (project.Database.HasTypeMappedToClr(column)) { var lines = new List <string> { string.Format("builder.Property(p => p.{0})", view.GetPropertyNameHack(column)) }; if (string.Compare(column.Name, column.GetPropertyName()) != 0) { lines.Add(string.Format("HasColumnName(\"{0}\")", column.Name)); } if (project.Database.ColumnIsString(column)) { lines.Add(column.Length <= 0 ? string.Format("HasColumnType(\"{0}(max)\")", column.Type) : string.Format("HasColumnType(\"{0}({1})\")", column.Type, column.Length)); } else if (project.Database.ColumnIsDecimal(column)) { lines.Add(string.Format("HasColumnType(\"{0}({1}, {2})\")", column.Type, column.Prec, column.Scale)); } else if (project.Database.ColumnIsDouble(column) || project.Database.ColumnIsSingle(column)) { lines.Add(string.Format("HasColumnType(\"{0}({1})\")", column.Type, column.Prec)); } else { lines.Add(string.Format("HasColumnType(\"{0}\")", column.Type)); } configLines.Add(new CodeLine("{0};", string.Join(".", lines))); } else { var lines = new List <string> { string.Format("builder.Ignore(p => p.{0})", view.GetPropertyNameHack(column)) }; configLines.Add(new CodeLine("{0};", string.Join(".", lines))); } } definition.Methods.Add(new MethodDefinition("void", "Configure", new ParameterDefinition(string.Format("EntityTypeBuilder<{0}>", project.GetEntityName(view)), "builder")) { Lines = configLines }); return(definition); }
public static EntityClassDefinition GetEntityClassDefinition(this EntityFrameworkCoreProject project, ITable table) { var definition = new EntityClassDefinition { Namespaces = { "System" }, Namespace = project.Database.HasDefaultSchema(table) ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(table.Schema), AccessModifier = AccessModifier.Public, Name = project.GetEntityName(table), IsPartial = true, Constructors = { new ClassConstructorDefinition(AccessModifier.Public) }, DbObject = table }; if (!string.IsNullOrEmpty(table.Description)) { definition.Documentation.Summary = table.Description; } var projectSelection = project.GetSelection(table); if (projectSelection.Settings.UseDataAnnotations) { definition.Namespaces.Add("System.ComponentModel.DataAnnotations"); definition.Namespaces.Add("System.ComponentModel.DataAnnotations.Schema"); } if (projectSelection.Settings.EnableDataBindings) { definition.Namespaces.Add("System.ComponentModel"); definition.Implements.Add("INotifyPropertyChanged"); definition.Events.Add(new EventDefinition("PropertyChangedEventHandler", "PropertyChanged")); } if (table.PrimaryKey != null) { var constructor = new ClassConstructorDefinition { AccessModifier = AccessModifier.Public }; foreach (var key in table.GetColumnsFromConstraint(table.PrimaryKey)) { var col = (Column)key; var propertyType = project.Database.ResolveDatabaseType(col); constructor.Parameters.Add(new ParameterDefinition(propertyType, project.GetParameterName(col))); constructor.Lines.Add(new CodeLine("{0} = {1};", project.GetPropertyName(key.Name), project.GetParameterName(col))); } definition.Constructors.Add(constructor); } var columns = table.Columns; foreach (var column in columns) { var propertyType = project.Database.ResolveDatabaseType(column); if (projectSelection.Settings.EnableDataBindings) { definition.AddViewModelProperty(propertyType, project.GetPropertyName(table, column)); } else { if (projectSelection.Settings.BackingFields.Contains(table.GetFullColumnName(column))) { definition.AddPropertyWithField(propertyType, project.GetPropertyName(table, column)); } else if (projectSelection.Settings.UseAutomaticPropertiesForEntities) { definition.Properties.Add(new PropertyDefinition { AccessModifier = AccessModifier.Public, Type = propertyType, Name = project.GetPropertyName(table, column), IsAutomatic = true }); } else { definition.AddPropertyWithField(propertyType, project.GetPropertyName(table, column)); } } } if (projectSelection.Settings.AuditEntity == null) { definition.Implements.Add(projectSelection.Settings.EntityInterfaceName); } else { var count = 0; foreach (var column in columns) { if (projectSelection.Settings.AuditEntity.Names.Contains(column.Name)) { count += 1; } } if (count == projectSelection.Settings.AuditEntity.Names.Count()) { definition.Implements.Add("IAuditEntity"); } else { definition.Implements.Add("IEntity"); } } if (projectSelection.Settings.SimplifyDataTypes) { definition.SimplifyDataTypes(); } if (projectSelection.Settings.DeclareNavigationProperties) { foreach (var foreignKey in table.ForeignKeys) { var foreignTable = project.Database.FindTable(foreignKey.References); if (foreignTable == null) { continue; } if (definition.Namespace != project.GetEntityLayerNamespace(foreignTable.Schema)) { definition.Namespaces .AddUnique(project.Database.HasDefaultSchema(foreignTable) ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(foreignTable.Schema)); } var fkProperty = foreignKey.GetParentNavigationProperty(foreignTable, project); if (!definition.Properties.Any(item => item.Name == fkProperty.Name)) { definition.Properties.Add(fkProperty); } } foreach (var child in project.Database.Tables) { foreach (var foreignKey in child.ForeignKeys) { if (foreignKey.References.EndsWith(table.FullName)) { definition.Namespaces .AddUnique(projectSelection.Settings.NavigationPropertyEnumerableNamespace); if (definition.Namespace != project.GetEntityLayerNamespace(child.Schema)) { definition.Namespaces .AddUnique(project.Database.HasDefaultSchema(child) ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(child.Schema)); } var navigationProperty = project.GetChildNavigationProperty(projectSelection, child, foreignKey); if (!definition.Properties.Any(item => item.Name == navigationProperty.Name)) { definition.Properties.Add(navigationProperty); } } } } } return(definition); }