Initial commit

OpenDBDiff.Schema.SQLServer.Generates/Model/Util/FormatCode.cs

@@ -0,0 +1,138 @@
+using System;
+using System.Text.RegularExpressions;
+using OpenDBDiff.Schema.Model;
+
+namespace OpenDBDiff.Schema.SQLServer.Generates.Model.Util
+{
+    internal static class FormatCode
+    {
+        private static readonly char[] TrimCharacters = { ' ', '\r', '\n', '\t' };
+
+        private class SearchItem
+        {
+            public int FindPosition;
+            public string Body = "";
+        }
+
+        /// <summary>
+        /// Clears all unnecessary characters that are after the END statement of the body, and whitespaces at the beginning.
+        /// </summary>
+        private static string CleanLast(string body)
+        {
+            if (string.IsNullOrEmpty(body))
+            {
+                return string.Empty;
+            }
+
+            return body.TrimStart().TrimEnd(TrimCharacters);
+        }
+
+        /// <summary>
+        /// Inserta la sentencia GO dentro del body
+        /// </summary>
+        private static string SmartGO(string code)
+        {
+            string prevText = code;
+            try
+            {
+                if (!prevText.Substring(prevText.Length - 2, 2).Equals("\r\n"))
+                    prevText += "\r\n";
+                return prevText + "GO\r\n";
+            }
+            catch
+            {
+                return prevText;
+            }
+        }
+
+        /// <summary>
+        /// Busca la primer entrada con el nombre completo dentro de una funcion, store, vista, trigger o rule.
+        /// Ignora los comentarios.
+        /// </summary>
+        private static SearchItem FindCreate(string ObjectType, ISchemaBase item, string prevText)
+        {
+            SearchItem sitem = new SearchItem();
+            Regex regex = new Regex(@"((/\*)(\w|\s|\d|\[|\]|\.)*(\*/))|((\-\-)(.)*)", RegexOptions.IgnoreCase);
+            Regex reg2 = new Regex(@"CREATE " + ObjectType + @"(\s|\r|\n|\t|\w|\/|\*|-|@|_|&|#)*((\[)?" + item.Owner + @"(\])?((\s)*)?\.)?((\s)*)?(\[)?" + item.Name + @"(\])?", (RegexOptions)((int)RegexOptions.IgnoreCase + (int)RegexOptions.Multiline));
+            Regex reg3 = new Regex(@"((\[)?" + item.Owner + @"(\])?\.)?((\s)+\.)?(\s)*(\[)?" + item.Name + @"(\])?", RegexOptions.IgnoreCase);
+            Regex reg4 = new Regex(@"( )*\[");
+            //Regex reg3 = new Regex(@"((\[)?" + item.Owner + @"(\])?.)?(\[)?" + item.Name + @"(\])?", RegexOptions.Multiline);
+
+            MatchCollection abiertas = regex.Matches(prevText);
+            Boolean finish = false;
+            int indexStart = 0;
+            int indexBegin = 0;
+            int iAux = -1;
+
+            while (!finish)
+            {
+                Match match = reg2.Match(prevText, indexBegin);
+                if (match.Success)
+                    iAux = match.Index;
+                else
+                    iAux = -1;
+                if ((abiertas.Count == indexStart) || (match.Success))
+                    finish = true;
+                else
+                {
+                    if ((iAux < abiertas[indexStart].Index) || (iAux > abiertas[indexStart].Index + abiertas[indexStart].Length))
+                        finish = true;
+                    else
+                    {
+                        //indexBegin = abiertas[indexStart].Index + abiertas[indexStart].Length;
+                        indexBegin = iAux + 1;
+                        indexStart++;
+                    }
+                }
+            }
+            string result = reg3.Replace(prevText, " " + item.FullName, 1, iAux + 1);
+            if (iAux != -1)
+                sitem.Body = reg4.Replace(result, " [", 1, iAux);
+            sitem.FindPosition = iAux;
+            return sitem;
+        }
+
+        public static string FormatCreate(string ObjectType, string body, ISchemaBase item)
+        {
+            try
+            {
+                string prevText = (string)body.Clone();
+                prevText = FindCreate(ObjectType, item, prevText).Body;
+                if (String.IsNullOrEmpty(prevText))
+                    prevText = body;
+                prevText = CleanLast(prevText);
+                return SmartGO(prevText);
+            }
+            catch
+            {
+                return SmartGO(CleanLast(body));
+            }
+        }
+
+        public static string FormatAlter(string ObjectType, string body, ISchemaBase item, Boolean quitSchemaBinding)
+        {
+            string prevText = null;
+            try
+            {
+                prevText = (string)body.Clone();
+                SearchItem sitem = FindCreate(ObjectType, item, prevText);
+                Regex regAlter = new Regex("CREATE");
+
+                if (!quitSchemaBinding)
+                    return regAlter.Replace(sitem.Body, "ALTER", 1, sitem.FindPosition);
+                //return prevText.Substring(0, iFind) + "ALTER " + sitem.ObjectType + " " + prevText.Substring(iFind + sitem.ObjectType.Length + 7, prevText.Length - (iFind + sitem.ObjectType.Length + 7)).TrimStart();
+                else
+                {
+                    string text = regAlter.Replace(sitem.Body, "ALTER", 1, sitem.FindPosition);
+                    Regex regex = new Regex("WITH SCHEMABINDING", RegexOptions.IgnoreCase);
+                    return regex.Replace(text, "");
+                }
+                //return "";
+            }
+            catch
+            {
+                return prevText;
+            }
+        }
+    }
+}