MigrationProgram.cs

/**
Copyright (c) 2016 Foundation.IO (https://github.com/foundationio). All rights reserved.

This work is licensed under the terms of the BSD license.
For a copy, see <https://opensource.org/licenses/BSD-3-Clause>.
**/
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Text;
using Framework.Infrastructure.Attributes;

namespace CodeGenerator
{
    public static class MigrationProgram
    {
        public static void MainApp()
        {
            var fileName = Directory.GetCurrentDirectory() + "\\..\\..\\..\\Backend\\Shared\\Repository\\Migration\\AppDBMigrationSteps\\Mig_002.cs";

            if (File.Exists(fileName))
            {
                File.Delete(fileName);
            }

            var sb = new StringBuilder();

            var t = typeof(LogR.Common.Constants.ConfigurationConstants);

            var assembly = t.Assembly;
            if (assembly == null)
            {
                Console.WriteLine("Unabel to find the LogR.Common assembly");
                return;
            }

            WriteHeader(ref sb);
            var start = true;

            foreach (var aClass in assembly.GetTypes())
            {
                if (!aClass.FullName.StartsWith("LogR.Common.Enums"))
                    continue;

                if (!aClass.IsEnum)
                {
                    sb.AppendLine(aClass.FullName + " is not enum");
                    continue;
                }

                var alst = aClass.GetCustomAttributes(typeof(TypeTableAttribute), true);
                if (alst == null || alst.Length == 0)
                {
                    sb.AppendLine(aClass.FullName + " does not have TypeTable attribute");
                    continue;
                }

                var aitem = alst[0] as TypeTableAttribute;
                var tableName = aitem.TableName;

                //Console.WriteLine(aClass.FullName);
                var camelCaseClass = (aClass.Name.Substring(0, 1).ToLower() + aClass.Name.Substring(1));
                camelCaseClass = camelCaseClass.Substring(0, camelCaseClass.Length - 1);

                if (!start)
                    sb.AppendLine();

                start = false;

                sb.AppendLine($"            dynamic[] {camelCaseClass}Params =");
                sb.AppendLine("            {");
                foreach (var enumName in System.Enum.GetNames(aClass))
                {
                    var memberInfos = aClass.GetMember(enumName);
                    var enumValueMemberInfo = memberInfos.FirstOrDefault(m => m.DeclaringType == aClass);
                    var pKeyValue = "";
                    var pkeyLst = enumValueMemberInfo.GetCustomAttributes(typeof(ParentKeyAttribute), true);
                    if (pkeyLst != null && pkeyLst.Length > 0)
                    {
                        pKeyValue = GetPKeyValue(pkeyLst.Select(x => x as ParentKeyAttribute));
                    }

                    sb.AppendLine($"                new {{ {tableName}Id = (int){aClass}.{enumName},  {tableName}Guid = Guid.NewGuid(), {tableName}Name = \"{enumName}\", {pKeyValue} CreatedDate = DateTime.UtcNow, CreatedBy = \"\" ,  ModifiedDate = DateTime.UtcNow , ModifiedBy = \"\" }} , ");
                }
                sb.AppendLine("            };");
                sb.AppendLine($"");
                sb.AppendLine($"                InsertData(\"{tableName}\", {camelCaseClass}Params);");
            }


            WriteFooter(ref sb);

            File.AppendAllText(fileName, sb.ToString());

            Process.Start("notepad.exe", fileName);

        }

        static string GetPKeyValue(IEnumerable<ParentKeyAttribute> pkAttList)
        {
            var result = string.Join(",", pkAttList.Select(x => x.ParentKeyColumn + " = " + x.ParentKeyValue).ToArray());
            return result + " , ";
        }

        static void WriteHeader(ref StringBuilder sb)
        {
            sb.AppendLine("using System;");
            sb.AppendLine("using FluentMigrator;");
            sb.AppendLine("using LogR.Common.Enums;");
            sb.AppendLine();
            sb.AppendLine("#pragma warning disable SA1028");
            sb.AppendLine("#pragma warning disable S101 // Types should be named in PascalCase");
            sb.AppendLine("#pragma warning disable SA1137");
            sb.AppendLine("namespace LogR.Repository.Migration.Application");
            sb.AppendLine("{");
            sb.AppendLine("    [Migration(002)]");
            sb.AppendLine("    public class Mig_002 : BaseMigration");
            sb.AppendLine("    {");
            sb.AppendLine("        public override void Up()");
            sb.AppendLine("        {");
        }

        static void WriteFooter(ref StringBuilder sb)
        {
            sb.AppendLine("        }");
            sb.AppendLine("    }");
            sb.AppendLine("}");
            sb.AppendLine("#pragma warning restore SA1137");
            sb.AppendLine("#pragma warning restore S101 // Types should be named in PascalCase");
            sb.AppendLine("#pragma warning restore SA1028");
        }
    }
}