Convention-CSharp/Convention/[Runtime]/EasySave/EasySaveSpreadsheet.cs

using System.Collections;
using System.Collections.Generic;
using System.IO;
using Convention.EasySave.Internal;

namespace Convention.EasySave
{
	public class EasySaveSpreadsheet
	{
		private int cols = 0;
		private int rows = 0;
		private Dictionary<Index, string> cells = new Dictionary<Index, string>();

		private const string QUOTE = "\"";
		private const char QUOTE_CHAR = '"';
		private const char COMMA_CHAR = ',';
		private const char NEWLINE_CHAR = '\n';
		private const string ESCAPED_QUOTE = "\"\"";
		private static char[] CHARS_TO_ESCAPE = { ',', '"', '\n', ' ' };

		public int ColumnCount
		{
			get { return cols; }
		}

		public int RowCount
		{
			get { return rows; }
		}

		public int GetColumnLength(int col)
		{
			if (col >= cols)
				return 0;

			int maxRow = -1;

			foreach (var index in cells.Keys)
				if (index.col == col && index.row > maxRow)
					maxRow = index.row;

			return maxRow + 1;
		}

		public int GetRowLength(int row)
		{
			if (row >= rows)
				return 0;

			int maxCol = -1;

			foreach (var index in cells.Keys)
				if (index.row == row && index.col > maxCol)
					maxCol = index.col;

			return maxCol + 1;
		}

		public void SetCell(int col, int row, object value)
		{
			var type = value.GetType();

			// If we're writing a string, add it without formatting.
			if (type == typeof(string))
			{
				SetCellString(col, row, (string)value);
				return;
			}

			var settings = new EasySaveSettings();
			if (EasySaveReflection.IsPrimitive(type))
				SetCellString(col, row, value.ToString());
			else
				SetCellString(col, row, settings.encoding.GetString(EasySave.Serialize(value, EasySaveTypeMgr.GetOrCreateEasySaveType(type))));

			// Expand the spreadsheet if necessary.
			if (col >= cols)
				cols = (col + 1);
			if (row >= rows)
				rows = (row + 1);
		}

		private void SetCellString(int col, int row, string value)
		{
			cells[new Index(col, row)] = value;

			// Expand the spreadsheet if necessary.
			if (col >= cols)
				cols = (col + 1);
			if (row >= rows)
				rows = (row + 1);
		}


		// Don't create non-generic version of this. Generic parameter is necessary as no type data is stored in the CSV file.
		public T GetCell<T>(int col, int row)
		{
			var val = GetCell(typeof(T), col, row);

			if (val == null)
				return default(T);
			return (T)val;
		}

		public object GetCell(System.Type type, int col, int row)
		{
			string value;

			if (col >= cols || row >= rows)
				throw new System.IndexOutOfRangeException("Cell (" + col + ", " + row + ") is out of bounds of spreadsheet (" + cols + ", " + rows + ").");

			if (!cells.TryGetValue(new Index(col, row), out value) || value == null)
				return null;

			// If we're loading a string, simply return the string value.
			if (type == typeof(string))
			{
				var str = (object)value;
				return str;
			}

			var settings = new EasySaveSettings();
			return EasySave.Deserialize(EasySaveTypeMgr.GetOrCreateEasySaveType(type, true), settings.encoding.GetBytes(value), settings);
		}

		public void Load(string filePath)
		{
			Load(new EasySaveSettings(filePath));
		}

		public void Load(string filePath, EasySaveSettings settings)
		{
			Load(new EasySaveSettings(filePath, settings));
		}

		public void Load(EasySaveSettings settings)
		{
			Load(EasySaveStream.CreateStream(settings, EasySaveFileMode.Read), settings);
		}

		public void LoadRaw(string str)
		{
			Load(new MemoryStream(((new EasySaveSettings()).encoding).GetBytes(str)), new EasySaveSettings());
		}

		public void LoadRaw(string str, EasySaveSettings settings)
		{
			Load(new MemoryStream((settings.encoding).GetBytes(str)), settings);
		}

		private void Load(Stream stream, EasySaveSettings settings)
		{
			using (var reader = new StreamReader(stream))
			{
				int c_int;
				char c;
				string value = "";
				int col = 0;
				int row = 0;

				// Read until the end of the stream.
				while (true)
				{
					c_int = reader.Read();
					c = (char)c_int;
					if (c == QUOTE_CHAR)
					{
						while (true)
						{
							c = (char)reader.Read();

							if (c == QUOTE_CHAR)
							{
								// If this quote isn't escaped by another, it is the last quote, so we should stop parsing this value.
								if (((char)reader.Peek()) != QUOTE_CHAR)
									break;
								else
									c = (char)reader.Read();
							}
							value += c;
						}
					}
					// If this is the end of a column, row, or the stream, add the value to the spreadsheet.
					else if (c == COMMA_CHAR || c == NEWLINE_CHAR || c_int == -1)
					{
						SetCell(col, row, value);
						value = "";
						if (c == COMMA_CHAR)
							col++;
						else if (c == NEWLINE_CHAR)
						{
							col = 0;
							row++;
						}
						else
							break;
					}
					else
						value += c;
				}
			}
		}

		public void Save(string filePath)
		{
			Save(new EasySaveSettings(filePath), false);
		}

		public void Save(string filePath, EasySaveSettings settings)
		{
			Save(new EasySaveSettings(filePath, settings), false);
		}

		public void Save(EasySaveSettings settings)
		{
			Save(settings, false);
		}

		public void Save(string filePath, bool append)
		{
			Save(new EasySaveSettings(filePath), append);
		}

		public void Save(string filePath, EasySaveSettings settings, bool append)
		{
			Save(new EasySaveSettings(filePath, settings), append);
		}

		public void Save(EasySaveSettings settings, bool append)
		{
			using (var writer = new StreamWriter(EasySaveStream.CreateStream(settings, append ? EasySaveFileMode.Append : EasySaveFileMode.Write)))
			{
				// If data already exists and we're appending, we need to prepend a newline.
				if (append && EasySave.FileExists(settings))
					writer.Write(NEWLINE_CHAR);

				var array = ToArray();
				for (int row = 0; row < rows; row++)
				{
					if (row != 0)
						writer.Write(NEWLINE_CHAR);

					for (int col = 0; col < cols; col++)
					{
						if (col != 0)
							writer.Write(COMMA_CHAR);

						writer.Write(Escape(array[col, row]));
					}
				}
			}
			if (!append)
				EasySaveIO.CommitBackup(settings);
		}

		private static string Escape(string str, bool isAlreadyWrappedInQuotes = false)
		{
			if (str == "")
				return "\"\"";
			else if (str == null)
				return null;

			// Now escape any other quotes.
			if (str.Contains(QUOTE))
				str = str.Replace(QUOTE, ESCAPED_QUOTE);

			// If there's chars to escape, wrap the value in quotes.
			if (str.IndexOfAny(CHARS_TO_ESCAPE) > -1)
				str = QUOTE + str + QUOTE;
			return str;
		}

		private static string Unescape(string str)
		{
			if (str.StartsWith(QUOTE) && str.EndsWith(QUOTE))
			{
				str = str.Substring(1, str.Length - 2);
				if (str.Contains(ESCAPED_QUOTE))
					str = str.Replace(ESCAPED_QUOTE, QUOTE);
			}
			return str;
		}

		private string[,] ToArray()
		{
			var array = new string[cols, rows];
			foreach (var cell in cells)
				array[cell.Key.col, cell.Key.row] = cell.Value;
			return array;
		}

		protected struct Index
		{
			public int col;
			public int row;

			public Index(int col, int row)
			{
				this.col = col;
				this.row = row;
			}
		}
	}
}