From 2fcab65c22db4435ac95d51790bf1151206410b1 Mon Sep 17 00:00:00 2001
From: arookas <arookas@outlook.com>
Date: Sat, 12 Dec 2015 16:51:31 -0500
Subject: [PATCH] Added error-line preview to frontend.

---
 frontend/main.cs | 40 +++++++++++++++++++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

diff --git a/frontend/main.cs b/frontend/main.cs
index 1a5e836..31b2ad5 100644
--- a/frontend/main.cs
+++ b/frontend/main.cs
@@ -1,6 +1,7 @@
 using System;
 using System.Diagnostics;
 using System.IO;
+using System.Text;
 
 namespace arookas
 {
@@ -37,7 +38,7 @@ namespace arookas
 					if (results.Error is sunScriptException)
 					{
 						var error = results.Error as sunScriptException;
-						Error("  \"{0}\"\n  pos ({1}, {2})\n{3}", error.Location.File, error.Location.Line, error.Location.Column, error.Message);
+						Error("  in file \"{0}\"\n  at line {1}, col {2}\n\n{3}{4}", error.Location.File, error.Location.Line, error.Location.Column, GetErrorPreview(error.Location), error.Message);
 						exitCode = 1;
 					}
 					else
@@ -123,6 +124,43 @@ namespace arookas
 			outputFile = prm[0];
 		}
 
+		static string GetErrorPreview(sunSourceLocation location)
+		{
+			Stream file;
+			try
+			{
+				file = File.OpenRead(location.File);
+			}
+			catch
+			{
+				// simply don't do a preview if opening a file fails
+				return "";
+			}
+			using (var reader = new StreamReader(file))
+			{
+				// skip to line
+				for (var line = 1; line < location.Line; ++line)
+				{
+					reader.ReadLine();
+				}
+				// generate column string
+				var sb = new StringBuilder();
+				var preview = reader.ReadLine();
+				sb.AppendLine(preview);
+				for (var column = 1; column < location.Column; ++column)
+				{
+					switch (preview[column - 1])
+					{
+						case '\t': sb.Append('\t'); break;
+						default: sb.Append(" "); break;
+					}
+				}
+				sb.Append("^");
+				sb.Append("\n");
+				return sb.ToString();
+			}
+		}
+
 		static void Message(string format, params object[] args) { Console.Write(format, args); }
 		static void Warning(string format, params object[] args)
 		{