public void Build()
{
var result = Builder.CreateFile(f =>
{
f.SetTargetFolder(_config.TargetFolder)
.SetFilename(_config.Filename);
f.AddUsing("System");
f.AddUsing("WhizzBase.Attributes");
if (_config.BaseEntityClassNamespace != null)
{
f.AddUsing(_config.BaseEntityClassNamespace);
}
foreach (var usingNamespace in _schema.UsingNamespaces)
{
f.AddUsing(usingNamespace);
}
f.AddNamespace(n =>
{
n.Name(_config.Namespace);
foreach (var relation in _schema.RelationEntities)
{
n.AddClass(c =>
{
if (relation.Kind == "v" || relation.Kind == "m")
{
c.AddViewAttribute(_schema.EscapedQuotedRelationName(relation.RelationName, relation.SchemaName));
}
else
{
c.AddTableAttribute(_schema.EscapedQuotedRelationName(relation.RelationName, relation.SchemaName));
}
c.Name(relation.SchemaName == DbSchema.DefaultSchema
? $"{relation.RelationName.ToPascalCase()}"
: $"{relation.SchemaName.ToPascalCase()}{relation.RelationName.ToPascalCase()}");
if (_config.AddPartialModifier)
{
c.ClassModifiers(m => m.Partial);
}
if (_config.BaseEntityClass != null)
{
c.AddBaseClass(_config.BaseEntityClass);
}
foreach (var column in _schema.GetColumns(relation.RelationName, relation.SchemaName))
{
c.AddProperty(p =>
{
if (column.IsPrimaryKey)
{
p.AddPrimaryKeyAttribute();
}
if (column.IsRequired && relation.Kind != "v" && relation.Kind != "m")
{
p.AddRequiredAttribute();
}
else if (column.IsReadonly || relation.Kind == "v" || relation.Kind == "m")
{
p.AddReadonlyAttribute();
}
if (column.CharacterMaximumLength != null && relation.Kind != "v" && relation.Kind != "m")
{
p.AddLengthAttribute((int)column.CharacterMaximumLength);
}
foreach (var foreignKey in _schema.GetForeignKeys(relation.RelationName, relation.SchemaName))
{
p.AddForeignKeyAttribute(
_schema.EscapedQuotedRelationName(foreignKey.PrimaryKeyTableName, foreignKey.PrimaryKeySchemaName),
_schema.EscapedQuote(foreignKey.PrimaryKeyColumnName));
}
var uniqueIndexes = _schema.GetUniqueIndexes(relation.RelationName, relation.SchemaName);
foreach (var uniqueIndex in _schema.GetUniqueIndexes(relation.RelationName, relation.SchemaName).Where(q => q.ColumnNames.Contains(column.ColumnName)))
{
p.AddUniqueIndexAttribute(_schema.EscapedQuote(uniqueIndex.IndexName));
}
p.AddColumnAttribute(_schema.EscapedQuote(column.ColumnName), column.Position);
p.Name(column.ColumnName.ToPascalCase());
p.Type(_schema.GetTypeName(column));
if (_config.MarkDirty && relation.Kind != "v" && relation.Kind != "m")
{
p.MarkDirty();
}
});
}
});
}
});
});
if (!string.IsNullOrEmpty(_config.TargetFolder) &&
!string.IsNullOrEmpty(_config.Filename) &&
System.IO.Directory.Exists(_config.TargetFolder))
{
System.IO.File.WriteAllText($"{_config.TargetFolder}/{_config.Filename}", result, Encoding.UTF8);
}
}
