// -----------------------------------------------------------------------
// <copyright company="Fireasy"
//      email="faib920@126.com"
//      qq="55570729">
//   (c) Copyright Fireasy. All rights reserved.
// </copyright>
// -----------------------------------------------------------------------
using CodeBuilder.Core.Source;
using CodeBuilder.Core.Variable;
using System.Collections.Generic;
using System.Linq;

namespace CodeBuilder.Core
{
    /// <summary>
    /// 架构辅助类。
    /// </summary>
    public class SchemaUnity
    {
        /// <summary>
        /// 重构所有表。
        /// </summary>
        /// <param name="tables"></param>
        /// <returns></returns>
        public static List<Table> Refactoring(List<Table> tables)
        {
            var newTables = new List<Table>();

            foreach (var table in tables)
            {
                var newTable = SchemaExtensionManager.Build<Table>();
                newTable.Refactoring(table, () => SchemaExtensionManager.Build<Column>());
                newTables.Add(newTable);
            }

            foreach (var table in tables)
            {
                var newTable = newTables.FirstOrDefault(s => s.Name == table.Name);

                foreach (var column in table.Columns.Where(s => s.ForeignKey != null))
                {
                    var newColumn = newTable.Columns.FirstOrDefault(s => s.Name == column.Name);

                    Table pkTable = null, fkTable = null;
                    Column pkColumn = null, fkColumn = null;
                    var fk = column.ForeignKey;
                    if (fk.PkTable != null)
                    {
                        pkTable = newTables.FirstOrDefault(s => s.Name == fk.PkTable.Name);

                        if (fk.PkColumn != null && pkTable != null)
                        {
                            pkColumn = pkTable.Columns.FirstOrDefault(s => s.Name == fk.PkColumn.Name);
                        }
                    }

                    if (fk.FkTable != null)
                    {
                        fkTable = newTables.FirstOrDefault(s => s.Name == fk.FkTable.Name);

                        if (fk.FkColumn != null && fkTable != null)
                        {
                            fkColumn = fkTable.Columns.FirstOrDefault(s => s.Name == fk.FkColumn.Name);
                        }
                    }

                    var newRefer = SchemaExtensionManager.Build<Reference>(pkTable, pkColumn, fkTable, fkColumn);
                    newRefer.Name = fk.Name;

                    newColumn.BindForeignKey(newRefer);
                }
            }

            return newTables;
        }
    }
}