public static Task <GenerateCodeResponse> Generate(GenerateCodeRequest request)
{
return(Task.Run(() =>
{
using (var schemaReader = SchemaReaderProvider.GetReader(request.ServerType))
{
var connectionString = request.ConnectionString;
if (schemaReader == null || string.IsNullOrEmpty(connectionString))
{
return new GenerateCodeResponse();
}
var tables = schemaReader.ReadSchema(connectionString);
if (tables == null || tables.Count <= 0)
{
throw new Exception(string.Format("Empty database or invalid connection string: {0}", connectionString));
}
var response = new GenerateCodeResponse {
Filenames = new List <string>()
};
var types = AppDomain.CurrentDomain.GetAssemblies().SelectMany(s => s.GetTypes()).Where(p => typeof(ITemplate).IsAssignableFrom(p));
foreach (var type in types)
{
var attr = (OutputFileNameAttribute)type.GetCustomAttributes(typeof(OutputFileNameAttribute), true).FirstOrDefault();
if (attr != null)
{
typeof(GenerateCodeProcessor).GetMethod("GenerateCode")
.MakeGenericMethod(type)
.Invoke(null, new object[]
{
tables, request, response, type.Name, attr.FileName
});
}
}
return response;
}
}));
}
public static GenerateCodeResponse GenerateCode <T>(Tables tables, GenerateCodeRequest request, GenerateCodeResponse response
, string subFolder, string fileNameTemplate)
where T : ITemplate, new()
{
foreach (var table in tables)
{
try
{
var model = new T
{
IncludeRelationships = request.IncludeRelationship,
Table = table,
Tables = tables
};
string pageContent = model.TransformText();
var folderLocation = Path.Combine(request.FolderLocation, subFolder);
if (!Directory.Exists(folderLocation))
{
Directory.CreateDirectory(folderLocation);
}
string fileName = string.Format(fileNameTemplate, table.ClassName);
File.WriteAllText(Path.Combine(folderLocation, fileName), pageContent, Encoding.UTF8);
fileName = subFolder + "/" + fileName;
if (response.Filenames.IndexOf(fileName) < 0)
{
response.Filenames.Add(fileName);
}
}
catch (Exception e)
{
MessageBox.Show(string.Format("{0} {1} Failed!", table.ClassName, subFolder));
}
}
return(response);
}
