From 44abe3ad4aa69f886600190ef2d718d3fdd5899c Mon Sep 17 00:00:00 2001
From: rolanmen1 <rolanmen1@4bbf4f03-6f44-549e-7e59-35d4b0962838>
Date: Fri, 3 Aug 2012 06:17:21 +0000
Subject: [PATCH] LuaWriter. Now colors LongStrings.

---
 BizHawk.MultiClient/tools/LuaWriter.cs | 78 ++++++++++++++++++++++++--
 1 file changed, 74 insertions(+), 4 deletions(-)

diff --git a/BizHawk.MultiClient/tools/LuaWriter.cs b/BizHawk.MultiClient/tools/LuaWriter.cs
index 3e23d1691a..9f229aa195 100644
--- a/BizHawk.MultiClient/tools/LuaWriter.cs
+++ b/BizHawk.MultiClient/tools/LuaWriter.cs
@@ -81,12 +81,83 @@ namespace BizHawk.MultiClient
 
 			ColorReservedWords();
 			ColorLibraries();
+			ColorSymbols();
 			ColorComments();
 			ColorStrings();
-			ColorSymbols();
+			ColorLongStrings();
 			LuaText.Select(selPos, selChars);
 		}
 
+		private void ColorLongStrings()
+		{
+			int firstBracket = LuaText.Find("["), secondBracket, ending;
+			while (firstBracket >= 0)
+			{
+				ending = 0;
+				if (firstBracket > 1 && LuaText.Text.Substring(firstBracket - 2, 2) == "--")
+				{
+					firstBracket = LuaText.Find("[", firstBracket + 1, LuaText.Text.Length, RichTextBoxFinds.MatchCase);
+					if (firstBracket == LuaText.Text.Length - 1)
+						break;
+				}
+				else if (firstBracket != LuaText.Text.Length - 1)
+				{
+					secondBracket = LuaText.Find("[", firstBracket + 1, LuaText.Text.Length, RichTextBoxFinds.MatchCase);
+					if (secondBracket >= 0 && IsLongString(LuaText.Text.Substring(firstBracket, secondBracket - firstBracket)))
+					{
+						if (secondBracket + 1 == LuaText.Text.Length)
+							ending = LuaText.Text.Length - 1;
+						else
+						{
+							string temp = GetLongStringClosingBracket(LuaText.Text.Substring(firstBracket, secondBracket - firstBracket + 1));
+							ending = LuaText.Find(temp, secondBracket + 1, LuaText.Text.Length, RichTextBoxFinds.MatchCase);
+							ending += temp.Length - 1;
+						}
+						if (ending < 0)
+							ending = LuaText.Text.Length - 1;
+
+						LuaText.Select(firstBracket, ending - firstBracket + 1);
+						LuaText.SelectionColor = Color.FromArgb(Global.Config.LuaStringColor);
+
+						if(ending < LuaText.Text.Length - 1)
+							firstBracket = LuaText.Find("[", ending + 1, LuaText.Text.Length, RichTextBoxFinds.MatchCase);
+						else
+							break;
+					}
+					else
+						break;
+				}
+				else
+					break;
+			}
+		}
+
+		private string GetLongStringClosingBracket(string openingBracket)
+		{
+			string closingBrackets = "]";
+			int level = 0;
+
+			foreach (char c in openingBracket)
+				if (c == '=')
+					level++;
+
+			for (int x = 0; x < level; x++)
+				closingBrackets += '=';
+
+			return closingBrackets + ']';
+		}
+
+		private bool IsLongString(string longstring)
+		{
+			bool Validated = true;
+
+			foreach (char c in longstring)
+				if (c != ']' && c != '=')
+					break;
+
+			return Validated;
+		}
+
 		private void ColorSymbols()
 		{
 			foreach (char mark in Symbols)
@@ -94,8 +165,7 @@ namespace BizHawk.MultiClient
 				int currPos = 0;
 				while (LuaText.Find(mark.ToString(), currPos, RichTextBoxFinds.None) >= 0)
 				{
-					if (LuaText.SelectionColor.ToArgb() != Global.Config.LuaCommentColor && LuaText.SelectionColor.ToArgb() != Global.Config.LuaStringColor)
-						LuaText.SelectionColor = Color.FromArgb(Global.Config.LuaSymbolColor);
+					LuaText.SelectionColor = Color.FromArgb(Global.Config.LuaSymbolColor);
                     if (Global.Config.LuaSymbolBold)
                         LuaText.SelectionFont = new Font(LuaText.SelectionFont, FontStyle.Bold);
 					currPos = LuaText.SelectionStart + 1;
@@ -194,7 +264,7 @@ namespace BizHawk.MultiClient
 			{
 				int endComment;
 
-				if (LuaText.Text.Substring(CommentMatch.Index, 4) == "--[[")
+				if (CommentMatch.Index + 4 < LuaText.Text.Length && LuaText.Text.Substring(CommentMatch.Index, 4) == "--[[")
 				{
 					if (LuaText.Find("]]", RichTextBoxFinds.MatchCase) > 0)
 						endComment = LuaText.SelectionStart - CommentMatch.Index + 2;