private static void DoScript(
string pathOrNull,
bool naive,
ConnectionStringSettings connectionStringSettings,
DashingSettings dashingSettings,
ReverseEngineerSettings reverseEngineerSettings)
{
DisplayMigrationHeader(naive, dashingSettings);
// fetch the to state
var config = (IConfiguration)configObject;
IEnumerable <string> warnings, errors;
var migrationScript = GenerateMigrationScript(connectionStringSettings, reverseEngineerSettings, config, naive, out warnings, out errors);
// report errors
DisplayMigrationWarningsAndErrors(errors, warnings);
if (string.IsNullOrWhiteSpace(migrationScript))
{
migrationScript = "-- Nothing to be migrated";
}
// write it
using (var writer = string.IsNullOrEmpty(pathOrNull) ? Console.Out : new StreamWriter(File.OpenWrite(pathOrNull))) {
writer.WriteLine(migrationScript);
}
}
private static object LoadConfiguration(
Assembly configAssembly,
DashingSettings dashingSettings,
ConnectionStringSettings connectionStringSettings)
{
// fetch the to state
var configType = configAssembly.GetLoadableTypes().SingleOrDefault(t => t.FullName == dashingSettings.ConfigurationName);
if (configType == null)
{
using (Color(ConsoleColor.Red)) {
var candidates = configAssembly.GetLoadableTypes().Where(t => t.GetInterface(typeof(IConfiguration).FullName) != null).ToArray();
if (candidates.Any())
{
throw new CatchyException(
"Could not locate {0}, but found candidates: {1}",
dashingSettings.ConfigurationName,
string.Join(", ", candidates.Select(c => c.FullName)));
}
throw new CatchyException("Could not locate {0}, and found no candidate configurations", dashingSettings.ConfigurationName);
}
}
// attempt to find the call to ConfigurationManager and overwrite the connection string
InjectConnectionString(dashingSettings, connectionStringSettings);
// TODO add in a factory way of generating the config for cases where constructor not empty
return(Activator.CreateInstance(configType));
}
private static void InjectConnectionString(DashingSettings dashingSettings, ConnectionStringSettings connectionStringSettings)
{
var assemblyDefinition = AssemblyDefinition.ReadAssembly(dashingSettings.PathToDll);
var cecilConfigType = assemblyDefinition.MainModule.Types.Single(t => t.FullName == dashingSettings.ConfigurationName);
var constructor = cecilConfigType.Methods.FirstOrDefault(m => m.IsConstructor && !m.HasParameters); // default constructor
if (constructor == null)
{
using (Color(ConsoleColor.Red)) {
throw new CatchyException("Unable to find a Default Constructor on the Configuration");
}
}
var getConnectionStringCall =
constructor.Body.Instructions.FirstOrDefault(
i =>
i.OpCode.Code == Code.Call &&
i.Operand.ToString()
== "System.Configuration.ConnectionStringSettingsCollection System.Configuration.ConfigurationManager::get_ConnectionStrings()");
if (getConnectionStringCall == null)
{
using (Color(ConsoleColor.Red)) {
throw new CatchyException("Unable to find the ConnectionStrings call in the constructor");
}
}
var connectionStringKey = getConnectionStringCall.Next.Operand.ToString();
// override readonly property of connectionstrings
var readOnlyField = typeof(ConfigurationElementCollection).GetField("bReadOnly", BindingFlags.Instance | BindingFlags.NonPublic);
if (readOnlyField != null)
{
readOnlyField.SetValue(ConfigurationManager.ConnectionStrings, false);
}
// remove any existing
if (ConfigurationManager.ConnectionStrings[connectionStringKey] != null)
{
ConfigurationManager.ConnectionStrings.Remove(connectionStringKey);
}
// and add in the one from our ini
ConfigurationManager.ConnectionStrings.Add(
new System.Configuration.ConnectionStringSettings(
connectionStringKey,
connectionStringSettings.ConnectionString,
connectionStringSettings.ProviderName));
}
private static void DoReverseEngineer(
CommandLineOptions options,
DashingSettings dashingSettings,
ReverseEngineerSettings reverseEngineerSettings,
ConnectionStringSettings connectionString)
{
// overwrite the path with the default if necessary
if (string.IsNullOrEmpty(options.Location))
{
options.Location = dashingSettings.DefaultSavePath;
}
// if it is still empty, ...
if (string.IsNullOrEmpty(options.Location) && options.ReverseEngineer)
{
throw new CatchyException("You must specify a location for generated files to be saved");
}
// require a generated namespace
if (string.IsNullOrEmpty(reverseEngineerSettings.GeneratedNamespace))
{
throw new CatchyException("You must specify a GeneratedNamespace in the Project ini file");
}
DatabaseSchema schema;
var engineer = new Engineer(reverseEngineerSettings.ExtraPluralizationWords);
var databaseReader = new DatabaseReader(connectionString.ConnectionString, connectionString.ProviderName);
schema = databaseReader.ReadAll();
var maps = engineer.ReverseEngineer(
schema,
new DialectFactory().Create(connectionString.ToSystem()),
reverseEngineerSettings.GetTablesToIgnore(),
consoleAnswerProvider,
true);
var reverseEngineer = new ModelGenerator();
var sources = reverseEngineer.GenerateFiles(maps, schema, reverseEngineerSettings.GeneratedNamespace, consoleAnswerProvider);
foreach (var source in sources)
{
File.WriteAllText(options.Location + "\\" + source.Key + ".cs", source.Value);
}
}
private static void DisplayMigrationHeader(bool naive, DashingSettings dashingSettings)
{
using (Color(ConsoleColor.Yellow)) {
Console.WriteLine("-- Dashing: Migration Script");
}
Console.WriteLine("-- -------------------------------");
Console.WriteLine("-- Assembly: {0}", dashingSettings.PathToDll);
Console.WriteLine("-- Class: {0}", dashingSettings.ConfigurationName);
Console.WriteLine("-- ");
if (!naive)
{
using (Color(ConsoleColor.Yellow)) {
Console.WriteLine("-- -------------------------------");
Console.WriteLine("-- Migration is experimental:");
Console.WriteLine("-- Please check the output!");
Console.WriteLine("-- -------------------------------");
}
}
}
private static void ParseIni(
CommandLineOptions options,
out ConnectionStringSettings connectionStringSettings,
out DashingSettings dashingSettings,
out ReverseEngineerSettings reverseEngineerSettings)
{
var config = IniParser.Parse(options.ConfigPath);
connectionStringSettings = new ConnectionStringSettings();
connectionStringSettings = IniParser.AssignTo(config["Database"], connectionStringSettings);
dashingSettings = new DashingSettings();
dashingSettings = IniParser.AssignTo(config["Dashing"], dashingSettings);
// fix path to dll to be avsolute path
if (!Path.IsPathRooted(dashingSettings.PathToDll))
{
dashingSettings.PathToDll = Path.Combine(Path.GetDirectoryName(options.ConfigPath), dashingSettings.PathToDll);
}
reverseEngineerSettings = new ReverseEngineerSettings();
reverseEngineerSettings = IniParser.AssignTo(config["ReverseEngineer"], reverseEngineerSettings);
}
private static void DoMigrate(
bool naive,
ConnectionStringSettings connectionStringSettings,
DashingSettings dashingSettings,
ReverseEngineerSettings reverseEngineerSettings)
{
DisplayMigrationHeader(naive, dashingSettings);
// fetch the to state
var config = (IConfiguration)configObject;
IEnumerable <string> warnings, errors;
var script = GenerateMigrationScript(connectionStringSettings, reverseEngineerSettings, config, naive, out warnings, out errors);
if (DisplayMigrationWarningsAndErrors(errors, warnings))
{
using (Color(ConsoleColor.Red)) {
Console.WriteLine("-- Fatal errors encountered: aborting migration. Please review the output.");
}
}
else
{
// migrate it
var factory = DbProviderFactories.GetFactory(connectionStringSettings.ProviderName);
using (var connection = factory.CreateConnection()) {
if (connection == null)
{
throw new Exception("Could not connect to database");
}
connection.ConnectionString = connectionStringSettings.ConnectionString;
connection.Open();
if (string.IsNullOrWhiteSpace(script))
{
using (Color(ConsoleColor.Green)) {
Console.WriteLine("-- No migration script to run");
}
}
else
{
using (new TimedOperation("-- Executing migration script on {0}", connection.ConnectionString)) {
using (var command = connection.CreateCommand()) {
using (Color(ConsoleColor.DarkGray)) {
Console.WriteLine();
Console.WriteLine(script);
Console.WriteLine();
}
command.CommandText = script;
command.ExecuteNonQuery();
}
}
}
// now let's call Seed
var seederConfig = config as ISeeder;
if (seederConfig != null)
{
using (new TimedOperation("-- Executing seeds")) {
using (var session = config.BeginSession(connection)) {
seederConfig.Seed(session);
session.Complete();
}
}
}
}
}
}
