private void GenerateDataContext()
{
String appStartPath = System.IO.Path.GetDirectoryName(Process.GetCurrentProcess().MainModule.FileName);
String resourcesPath = Path.Combine(appStartPath, "Resources");
string templatesPath = Path.Combine(resourcesPath, "Templates");
string templateFileName = Path.Combine(templatesPath, "EntityDataContext.tpl");
String fileName = Path.Combine(this.DataPath, "DataContext.cs");
TemplateFileWriter writer = new TemplateFileWriter(fileName, templateFileName);
writer.ReplaceTag("@NameSpaceInfrastructure", "");
writer.ReplaceTag("@NameSpace", this.DataNamespace);
foreach (DbSchema schema in this._model.Schemas)
{
foreach (DbTable table in schema.Tables)
{
String className = NameConverter.TableNameToEntityName(table.Name)+"";
writer.AddProperty("public DbSet<" + className + "> ", className + "s");
//
//public InfAccident CreateInfAccident()
//
foreach (DbForeignKey fk in table.ForeignKeys)
{
DbTable parentTable = (from t in schema.Tables where t.Id.Equals (fk.ParentTableId ) select t).FirstOrDefault();
String parentEntityName = NameConverter.TableNameToEntityName(parentTable.Name);
writer.ModelBuilders.Add("modelBuilder.Entity<" + className + ">().HasRequired<" + parentEntityName + ">(c => c." + parentEntityName + ").WithMany(t => t." + className + "s);");
}
//
writer.ModelBuilders.Add("modelBuilder.Entity<"+className+">().ToTable(\""+table.Name.ToLower ()+"\", \""+schema.Name.ToLower ()+"\");");
foreach (DbColumn column in table.Columns)
{
if (column.Name.StartsWith (table.Name +"__"))
{
String propertyName = NameConverter.ColumnNameToPropertyName(column.Name.Substring(column.Name.LastIndexOf("__")));
writer.ModelBuilders.Add("modelBuilder.Entity<" + className + ">().Property(t => t." + propertyName + ") .HasColumnName(\"" + column.Name.ToLower () + "\");");
if (!column.AllowNull)
{ writer.ModelBuilders.Add("modelBuilder.Entity<" + className + ">().Property(t => t." + propertyName + ").IsRequired();"); }
if (column.DataType.StartsWith("VARCHAR"))
{
writer.ModelBuilders.Add("modelBuilder.Entity<" + className + ">().Property(t => t." + propertyName + ").HasMaxLength(" + column.Length + ");");
}
}
else
{
String propertyName = NameConverter.ColumnNameToPropertyName(column.Name);
writer.ModelBuilders.Add("modelBuilder.Entity<" + className + ">().Property(t => t." + propertyName + ") .HasColumnName(\"" + column.Name.ToLower() + "\");");
if (!column.AllowNull)
{ writer.ModelBuilders.Add("modelBuilder.Entity<" + className + ">().Property(t => t." + propertyName + ").IsRequired();"); }
if (column.DataType.StartsWith("VARCHAR"))
{
writer.ModelBuilders.Add("modelBuilder.Entity<" + className + ">().Property(t => t." + propertyName + ").HasMaxLength("+column.Length+");");
}
}
}
if (table.PrimaryKey.ColumnIds.Count == 1)
{
DbColumn pkColumn = (from c in table.Columns where c.Id.Equals (table.PrimaryKey.ColumnIds.First ()) select c).FirstOrDefault();
String propertyName = NameConverter.ColumnNameToPropertyName(pkColumn.Name.Substring(table.Name.Length));
writer.ModelBuilders.Add("modelBuilder.Entity<" + className + ">().HasKey(t => t." + propertyName + ");");
writer.ModelBuilders.Add("modelBuilder.Entity<" + className + ">().Property(t => t." + propertyName + ").HasDatabaseGeneratedOption(DatabaseGeneratedOption.Identity);");
//
}
//modelBuilder.Entity<InfAccident>().HasKey(t => t.Id);
}
}
writer.WriteContent();
}
private void GenerateData()
{
string modelPath = Path.Combine(this.DataPath, "Models");
String[] files = Directory.GetFiles(modelPath);
foreach (String file in files)
{ File.Delete(file); }
String appStartPath = System.IO.Path.GetDirectoryName(Process.GetCurrentProcess().MainModule.FileName);
String resourcesPath = Path.Combine(appStartPath, "Resources");
string templatesPath = Path.Combine(resourcesPath, "Templates");
string templateFileName = Path.Combine(templatesPath, "EntityImplementation.tpl");
foreach (DbSchema schema in this._model.Schemas)
{
List<DbForeignKey> allFks = new List<DbForeignKey>();
foreach (DbTable table in schema.Tables)
{
foreach (DbForeignKey fk in table.ForeignKeys)
{
allFks.Add(fk);
}
}
allFks = (from fk in allFks select fk).Distinct().ToList();
foreach (DbTable table in schema.Tables)
{
List<String> constructors = new List<string>();
String className = NameConverter.TableNameToEntityName(table.Name)+"";
String modelFileName = Path.Combine(modelPath, className + ".cs");
TemplateFileWriter writer = new TemplateFileWriter(modelFileName, templateFileName);
writer.ClassAttributes.Add("[TableName(\"" + table.Name + "\")]");
writer.ClassAttributes.Add("[SchemaName(\"" + schema.Name + "\")]");
List<DbForeignKey> childFks = (from fk in allFks where fk.ParentTableId.Equals(table.Id) select fk).ToList();
foreach (DbForeignKey childFk in childFks)
{
DbTable childTable = (from t in schema.Tables where t.Id.Equals(childFk.ChildTableId) select t).FirstOrDefault();
String childEntityName = NameConverter.TableNameToEntityName(childTable.Name);
TemplateProperty prop = writer.AddProperty("public virtual ObservableCollection<" + childEntityName + ">", childEntityName + "s");
constructors.Add (" this."+childEntityName + "s = new ObservableCollection<" + childEntityName + ">();");
prop.Attributes.Add("[DisplayName(\"" + childTable.DisplayName + "s\")]");
}
List<DbForeignKey> parentFks = (from fk in allFks where fk.ChildTableId.Equals(table.Id) select fk).ToList();
foreach (DbForeignKey parentFk in parentFks)
{
DbTable parentTable = (from t in schema.Tables where t.Id.Equals(parentFk.ParentTableId) select t).FirstOrDefault();
String parentEntityName = NameConverter.TableNameToEntityName(parentTable.Name);
DbColumn childColumn = (from c in table.Columns where (from j in parentFk.Joins select j.ChildColumnId ).Contains (c.Id ) select c).FirstOrDefault();
TemplateProperty prop = writer.AddProperty("public virtual " + parentEntityName, parentEntityName);
prop.Attributes.Add("[DisplayName(\"" + parentTable.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + childColumn.Name + "\")]");
if (childColumn.AllowNull)
{ prop.Attributes.Add("[AllowNull(true)]"); }
else
{ prop.Attributes.Add("[AllowNull(false)]"); }
prop.Attributes.Add("[ControlType(ControlType.Combo)]");
prop.Attributes.Add("[ForeignKey(\"" + parentFk.Name + "\",null)]");
List<DbUniqueKey> uks = (from uk in table.UniqueKeys where uk.ColumnIds.Contains(childColumn.Id) select uk).ToList();
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
}
writer.ReplaceTag("@DisplayName", table.DisplayName);
writer.ReplaceTag("@NameSpaceInfrastructure", "");
writer.ReplaceTag("@NameSpace", this.DataNamespace);
writer.ReplaceTag("@EntityName", className);
foreach (DbColumn column in table.Columns)
{
DbForeignKey fkTable = (from fk in table.ForeignKeys where (from j in fk.Joins select j.ChildColumnId ).Contains (column.Id ) select fk).FirstOrDefault();
DbForeignKeyJoin fkJoinColumn = null;
List<DbUniqueKey> uks = (from uk in table.UniqueKeys where uk.ColumnIds.Contains (column.Id ) select uk).ToList();
if (fkTable != null)
{
fkJoinColumn = (from j in fkTable.Joins where j.ChildColumnId.Equals(column.Id) select j).FirstOrDefault();
}
String propertyName = NameConverter.ColumnNameToPropertyName(column.Name.Replace(table.Name, ""));
if (column.DataType.Equals("INT4"))
{
DbColumn childColumn = null;
foreach (DbForeignKey parentFk in parentFks)
{
if ((from c in table.Columns where (from j in parentFk.Joins select j.ChildColumnId).Contains(c.Id) select c).FirstOrDefault() != null)
{ childColumn = (from c in table.Columns where (from j in parentFk.Joins select j.ChildColumnId).Contains(c.Id) select c).FirstOrDefault(); }
}
DbRuleLocationRef ruleLocationRef = (from r in column.Rules where r is DbRuleLocationRef select r as DbRuleLocationRef).FirstOrDefault();
DbRuleLocationDeb ruleLocationDeb = (from r in column.Rules where r is DbRuleLocationDeb select r as DbRuleLocationDeb).FirstOrDefault();
DbRuleLocationFin ruleLocationFin = (from r in column.Rules where r is DbRuleLocationFin select r as DbRuleLocationFin).FirstOrDefault();
DbRulePr rulePr = (from r in column.Rules where r is DbRulePr select r as DbRulePr).FirstOrDefault();
DbRuleEmprise ruleEmprise = (from r in column.Rules where r is DbRuleEmprise select r as DbRuleEmprise).FirstOrDefault();
bool isFk = column.Name.ToLower().EndsWith("id") && column.Name.Length > 2;
if (column.AllowNull)
{
TemplateProperty prop = writer.AddProperty("public Nullable<Int64>", propertyName);
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (ruleLocationRef != null)
{
prop.Attributes.Add("[LocationAttribute(LocationAttributeType.ReferenceId)]");
}
if (ruleLocationDeb != null)
{
prop.Attributes.Add("[LocationAttribute(LocationAttributeType.ReferenceDeb)]");
}
if (ruleLocationFin != null)
{
prop.Attributes.Add("[LocationAttribute(LocationAttributeType.ReferenceFin)]");
}
// sauf que les colonne qui ne sont pas des id parent ne sont pas non plus ajouté à la unique key
if (!isFk)
{
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
}
else
{ Console.WriteLine("Exclude from uk"); }
prop.Attributes.Add("[RangeValue(-999999999999,999999999999)]");
if (fkJoinColumn != null)
{ prop.Attributes.Add("[ControlType(ControlType.None)]"); }
else
{
if (rulePr != null)
{
prop.Attributes.Add("[RulePr(\"" + rulePr.ChausseIdColumnName + "\")]");
prop.Attributes.Add("[ControlType(ControlType.Pr)]");
if (ruleEmprise != null)
{
prop.Attributes.Add("[RuleEmprise(\"" + rulePr.ChausseIdColumnName + "\")]");
}
}
else
{ prop.Attributes.Add("[ControlType(ControlType.Integer)]"); }
}
prop.Attributes.Add("[AllowNull(true)]");
}
else
{
TemplateProperty prop = writer.AddProperty("public Int64", propertyName);
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (ruleLocationRef != null)
{
prop.Attributes.Add("[LocationAttribute(LocationAttributeType.ReferenceId)]");
}
if (ruleLocationDeb != null)
{
prop.Attributes.Add("[LocationAttribute(LocationAttributeType.ReferenceDeb)]");
}
if (ruleLocationFin != null)
{
prop.Attributes.Add("[LocationAttribute(LocationAttributeType.ReferenceFin)]");
}
if (!isFk)
{
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
}
else
{ Console.WriteLine("Exclude from uk"); }
prop.Attributes.Add("[RangeValue(-999999999999,999999999999)]");
if (fkJoinColumn != null)
{prop.Attributes.Add("[ControlType(ControlType.None)]"); }
else
{
if (rulePr != null)
{
prop.Attributes.Add("[RulePr(\"" + rulePr.ChausseIdColumnName + "\")]");
prop.Attributes.Add("[ControlType(ControlType.Pr)]");
if (ruleEmprise != null)
{
prop.Attributes.Add("[RuleEmprise(\"" + rulePr.ChausseIdColumnName + "\")]");
}
}
else
{ prop.Attributes.Add("[ControlType(ControlType.Integer)]"); }
}
prop.Attributes.Add("[AllowNull(false)]");
}
}
else if (column.DataType.Equals("SERIAL"))
{
TemplateProperty prop = writer.AddProperty("public Int64", propertyName);
prop.Attributes.Add("[Browsable(false)]");
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (table.PrimaryKey.ColumnIds.Contains(column.Id))
{prop.Attributes.Add("[PrimaryKey(\"" + table.PrimaryKey.Name + "\")]");}
foreach (DbForeignKey fk in childFks)
{
foreach (DbForeignKeyJoin j in fk.Joins)
{
if (j.ParentColumnId.Equals(column.Id))
{
prop.Attributes.Add("[ForeignKeyAttribute(\"" + fk.Name + "\",\"JOIN_" + j.Id + "\")]");
}
}
}
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
prop.Attributes.Add("[ControlType(ControlType.None)]");
prop.Attributes.Add("[AllowNull(false)]");
}
else if (column.DataType.StartsWith("VBIN"))
{
DbRuleLocationRefGeom ruleLocationRefGeom = (from r in column.Rules where r is DbRuleLocationRefGeom select r as DbRuleLocationRefGeom).FirstOrDefault();
TemplateProperty prop = writer.AddProperty("public String", propertyName);
if (ruleLocationRefGeom != null)
{
prop.Attributes.Add("[LocationAttribute(LocationAttributeType.ReferenceGeometry)]");
}
prop.Attributes.Add("[ControlType(ControlType.None)]");
prop.Attributes.Add("[AllowNull(true)]");
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
}
else if (column.DataType.StartsWith("VARCHAR"))
{
DbRuleColor ruleColor = (from r in column.Rules where r is DbRuleColor select r as DbRuleColor).FirstOrDefault();
TemplateProperty prop = writer.AddProperty("public String", propertyName);
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (fkJoinColumn != null)
{ prop.Attributes.Add("[ForeignKey(\"" + fkTable.Name + "\",\"JOIN_" + fkJoinColumn.Id + "\")]"); }
foreach (DbUniqueKey uk in uks)
{
prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]");
}
prop.Attributes.Add("[MaxCharLength(" + column.Length + ")]");
if (ruleColor != null)
{ prop.Attributes.Add("[ControlType(ControlType.Color)]"); }
else
{ prop.Attributes.Add("[ControlType(ControlType.Text)]"); }
//MaxCharLengthAttribute
if (column.AllowNull)
{ prop.Attributes.Add("[AllowNull(true)]"); }
else
{ prop.Attributes.Add("[AllowNull(false)]"); }
}
else if (column.DataType.StartsWith("FLOAT8"))
{
//RangeValueAttribute
if (column.AllowNull)
{
TemplateProperty prop = writer.AddProperty("public Nullable<Double>", propertyName);
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (fkJoinColumn != null)
{ prop.Attributes.Add("[ForeignKey(\"" + fkTable.Name + "\",\"JOIN_" + fkJoinColumn.Id + "\")]"); }
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
prop.Attributes.Add("[RangeValue(-999999999999,999999999999)]");
prop.Attributes.Add("[ControlType(ControlType.Decimal)]");
prop.Attributes.Add("[AllowNull(true)]");
}
else
{
TemplateProperty prop = writer.AddProperty("public Double", propertyName);
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (fkJoinColumn != null)
{ prop.Attributes.Add("[ForeignKey(\"" + fkTable.Name + "\",\"JOIN_" + fkJoinColumn.Id + "\")]"); }
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
prop.Attributes.Add("[RangeValue(-999999999999,999999999999)]");
prop.Attributes.Add("[ControlType(ControlType.Decimal)]");
prop.Attributes.Add("[AllowNull(false)]");
}
}
else if (column.DataType.StartsWith("DATE"))
{
if (column.AllowNull)
{
TemplateProperty prop = writer.AddProperty("public Nullable<DateTime>", propertyName);
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (fkJoinColumn != null)
{ prop.Attributes.Add("[ForeignKey(\"" + fkTable.Name + "\",\"JOIN_" + fkJoinColumn.Id + "\")]"); }
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
prop.Attributes.Add("[ControlType(ControlType.Date)]");
prop.Attributes.Add("[AllowNull(true)]");
}
else
{
TemplateProperty prop = writer.AddProperty("public DateTime", propertyName);
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (fkJoinColumn != null)
{ prop.Attributes.Add("[ForeignKey(\"" + fkTable.Name + "\",\"JOIN_" + fkJoinColumn.Id + "\")]"); }
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
prop.Attributes.Add("[ControlType(ControlType.Date)]");
prop.Attributes.Add("[AllowNull(false)]");
}
}
else if (column.DataType.StartsWith("BOOL"))
{
if (column.AllowNull)
{
TemplateProperty prop = writer.AddProperty("public Nullable<Boolean>", propertyName);
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (fkJoinColumn != null)
{ prop.Attributes.Add("[ForeignKey(\"" + fkTable.Name + "\",\"JOIN_" + fkJoinColumn.Id + "\")]"); }
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
prop.Attributes.Add("[ControlType(ControlType.Check)]");
prop.Attributes.Add("[AllowNull(true)]");
}
else
{
TemplateProperty prop = writer.AddProperty("public Boolean", propertyName);
prop.Attributes.Add("[DisplayName(\"" + column.DisplayName + "\")]");
prop.Attributes.Add("[ColumnName(\"" + column.Name + "\")]");
if (fkJoinColumn != null)
{ prop.Attributes.Add("[ForeignKey(\"" + fkTable.Name + "\",\"JOIN_" + fkJoinColumn.Id + "\")]"); }
foreach (DbUniqueKey uk in uks)
{ prop.Attributes.Add("[UniqueKey(\"" + uk.Name + "\")]"); }
prop.Attributes.Add("[ControlType(ControlType.Check)]");
prop.Attributes.Add("[AllowNull(false)]");
}
}
else Console.WriteLine(column.DataType);
}
writer.ReplaceTag("@Constructors", String.Join("\r\n", constructors));
writer.WriteContent();
}
}
}
