public void TestCSharpClassWithIndexerGeneration()
{
// Arrange
var definition = new CSharpClassDefinition
{
Namespaces = new List <string>
{
"System",
"System.Collections.Generic"
},
Namespace = "Mapping",
Name = "DataSet"
};
definition.Indexers.Add(new IndexerDefinition
{
AccessModifier = AccessModifier.Public,
Type = "string",
Parameters = new List <ParameterDefinition>
{
new ParameterDefinition
{
Type = "int",
Name = "index"
}
},
GetBody = new List <ILine>
{
new CodeLine("return m_columns[index];")
},
SetBody = new List <ILine>
{
new CodeLine("m_columns = value;")
}
});
definition.AddPropertyWithField("List<string>", "Columns");
// Act
CSharpCodeBuilder.CreateFiles("C:\\Temp\\CatFactory.DotNetCore", string.Empty, true, definition);
}
public static CSharpClassDefinition CreateView(this DapperProject project, IView view)
{
var definition = new CSharpClassDefinition();
definition.Namespaces.Add("System");
definition.Namespace = view.HasDefaultSchema() ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(view.Schema);
definition.Name = view.GetEntityName();
definition.Constructors.Add(new ClassConstructorDefinition());
if (!string.IsNullOrEmpty(view.Description))
{
definition.Documentation.Summary = view.Description;
}
var selection = project.GetSelection(view);
foreach (var column in view.Columns)
{
if (selection.Settings.UseAutomaticPropertiesForEntities)
{
definition.Properties.Add(new PropertyDefinition(project.Database.ResolveType(column), column.GetPropertyName()));
}
else
{
definition.AddPropertyWithField(project.Database.ResolveType(column), column.GetPropertyName());
}
}
definition.Implements.Add("IEntity");
if (selection.Settings.SimplifyDataTypes)
{
definition.SimplifyDataTypes();
}
return(definition);
}
public static CSharpClassDefinition GetEntityClassDefinition(this EntityFrameworkCoreProject project, ITable table)
{
var classDefinition = new CSharpClassDefinition();
classDefinition.Namespaces.Add("System");
if (project.Settings.UseDataAnnotations)
{
classDefinition.Namespaces.Add("System.ComponentModel.DataAnnotations");
classDefinition.Namespaces.Add("System.ComponentModel.DataAnnotations.Schema");
}
if (project.Settings.EnableDataBindings)
{
classDefinition.Namespaces.Add("System.ComponentModel");
classDefinition.Implements.Add("INotifyPropertyChanged");
classDefinition.Events.Add(new EventDefinition("PropertyChangedEventHandler", "PropertyChanged"));
}
classDefinition.Namespace = table.HasDefaultSchema() ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(table.Schema);
classDefinition.Name = table.GetSingularName();
classDefinition.IsPartial = true;
classDefinition.Constructors.Add(new ClassConstructorDefinition());
var columns = table.Columns;
if (table.PrimaryKey?.Key.Count == 1)
{
var column = table.PrimaryKey.GetColumns(table).First();
classDefinition.Constructors.Add(new ClassConstructorDefinition(new ParameterDefinition(column.GetClrType(), column.GetParameterName()))
{
Lines = new List <ILine>
{
new CodeLine("{0} = {1};", column.GetPropertyName(), column.GetParameterName())
}
});
}
if (!string.IsNullOrEmpty(table.Description))
{
classDefinition.Documentation.Summary = table.Description;
}
foreach (var column in columns)
{
if (project.Settings.EnableDataBindings)
{
classDefinition.AddViewModelProperty(column.GetClrType(), column.HasSameNameEnclosingType(table) ? column.GetNameForEnclosing() : column.GetPropertyName());
}
else
{
if (project.Settings.BackingFields.Contains(table.GetFullColumnName(column)))
{
classDefinition.AddPropertyWithField(column.GetClrType(), column.HasSameNameEnclosingType(table) ? column.GetNameForEnclosing() : column.GetPropertyName());
}
else if (project.Settings.UseAutomaticPropertiesForEntities)
{
classDefinition.Properties.Add(new PropertyDefinition(column.GetClrType(), column.HasSameNameEnclosingType(table) ? column.GetNameForEnclosing() : column.GetPropertyName()));
}
else
{
classDefinition.AddPropertyWithField(column.GetClrType(), column.HasSameNameEnclosingType(table) ? column.GetNameForEnclosing() : column.GetPropertyName());
}
}
}
if (project.Settings.AuditEntity == null)
{
classDefinition.Implements.Add("IEntity");
}
else
{
var count = 0;
foreach (var column in columns)
{
if (project.Settings.AuditEntity.Names.Contains(column.Name))
{
count += 1;
}
}
if (count == project.Settings.AuditEntity.Names.Length)
{
classDefinition.Implements.Add(project.Settings.EntityInterfaceName);
}
else
{
classDefinition.Implements.Add("IEntity");
}
}
foreach (var foreignKey in table.ForeignKeys)
{
var foreignTable = project.Database.FindTableByFullName(foreignKey.References);
if (foreignTable == null)
{
continue;
}
classDefinition.Namespaces.AddUnique(foreignTable.HasDefaultSchema() ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(foreignTable.Schema));
classDefinition.Namespace = table.HasDefaultSchema() ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(table.Schema);
classDefinition.Properties.Add(foreignKey.GetParentNavigationProperty(project, foreignTable));
}
foreach (var child in project.Database.Tables)
{
if (table.FullName == child.FullName)
{
continue;
}
foreach (var foreignKey in child.ForeignKeys)
{
if (foreignKey.References.EndsWith(table.FullName))
{
classDefinition.Namespaces.AddUnique(project.Settings.NavigationPropertyEnumerableNamespace);
classDefinition.Namespaces.AddUnique(child.HasDefaultSchema() ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(child.Schema));
var navigationProperty = project.GetChildNavigationProperty(child, foreignKey);
if (classDefinition.Properties.FirstOrDefault(item => item.Name == navigationProperty.Name) == null)
{
classDefinition.Properties.Add(navigationProperty);
}
}
}
}
if (project.Settings.SimplifyDataTypes)
{
classDefinition.SimplifyDataTypes();
}
return(classDefinition);
}
public static CSharpClassDefinition CreateEntity(this DapperProject project, ITable table)
{
var classDefinition = new CSharpClassDefinition();
classDefinition.Namespaces.Add("System");
var selection = project.GetSelection(table);
if (selection.Settings.EnableDataBindings)
{
classDefinition.Namespaces.Add("System.ComponentModel");
classDefinition.Implements.Add("INotifyPropertyChanged");
classDefinition.Events.Add(new EventDefinition("PropertyChangedEventHandler", "PropertyChanged"));
}
classDefinition.Namespace = table.HasDefaultSchema() ? project.GetEntityLayerNamespace() : project.GetEntityLayerNamespace(table.Schema);
classDefinition.Name = table.GetEntityName();
classDefinition.Constructors.Add(new ClassConstructorDefinition());
if (table.PrimaryKey != null && table.PrimaryKey.Key.Count == 1)
{
var column = table.GetColumnsFromConstraint(table.PrimaryKey).First();
classDefinition.Constructors.Add(new ClassConstructorDefinition(new ParameterDefinition(project.Database.ResolveType(column), column.GetParameterName()))
{
Lines = new List <ILine>()
{
new CodeLine("{0} = {1};", column.GetPropertyName(), column.GetParameterName())
}
});
}
if (!string.IsNullOrEmpty(table.Description))
{
classDefinition.Documentation.Summary = table.Description;
}
foreach (var column in table.Columns)
{
if (selection.Settings.EnableDataBindings)
{
classDefinition.AddViewModelProperty(project.Database.ResolveType(column), column.GetPropertyName());
}
else
{
if (selection.Settings.UseAutomaticPropertiesForEntities)
{
classDefinition.Properties.Add(new PropertyDefinition(project.Database.ResolveType(column), column.GetPropertyName()));
}
else
{
classDefinition.AddPropertyWithField(project.Database.ResolveType(column), column.GetPropertyName());
}
}
}
classDefinition.Implements.Add("IEntity");
if (selection.Settings.SimplifyDataTypes)
{
classDefinition.SimplifyDataTypes();
}
return(classDefinition);
}