Move IniFile to Mods.Common.

OpenRA.Mods.Common/FileFormats/IniFile.cs

@@ -0,0 +1,152 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2015 The OpenRA Developers (see AUTHORS)
+ * This file is part of OpenRA, which is free software. It is made
+ * available to you under the terms of the GNU General Public License
+ * as published by the Free Software Foundation. For more information,
+ * see COPYING.
+ */
+#endregion
+
+using System;
+using System.Collections;
+using System.Collections.Generic;
+using System.IO;
+using System.Text.RegularExpressions;
+
+namespace OpenRA.Mods.Common.FileFormats
+{
+	public class IniFile
+	{
+		Dictionary<string, IniSection> sections = new Dictionary<string, IniSection>();
+
+		public IniFile(Stream s)
+		{
+			using (s)
+				Load(s);
+		}
+
+		public IniFile(params Stream[] streams)
+		{
+			foreach (var s in streams)
+				Load(s);
+		}
+
+		public void Load(Stream s)
+		{
+			var reader = new StreamReader(s);
+			IniSection currentSection = null;
+
+			while (!reader.EndOfStream)
+			{
+				var line = reader.ReadLine();
+
+				if (line.Length == 0) continue;
+
+				switch (line[0])
+				{
+					case ';': break;
+					case '[': currentSection = ProcessSection(line); break;
+					default: ProcessEntry(line, currentSection); break;
+				}
+			}
+		}
+
+		Regex sectionPattern = new Regex(@"^\[([^]]*)\]");
+
+		IniSection ProcessSection(string line)
+		{
+			var m = sectionPattern.Match(line);
+			if (!m.Success)
+				return null;
+			var sectionName = m.Groups[1].Value.ToLowerInvariant();
+
+			IniSection ret;
+			if (!sections.TryGetValue(sectionName, out ret))
+				sections.Add(sectionName, ret = new IniSection(sectionName));
+			return ret;
+		}
+
+		static bool ProcessEntry(string line, IniSection currentSection)
+		{
+			var comment = line.IndexOf(';');
+			if (comment >= 0)
+				line = line.Substring(0, comment);
+
+			line = line.Trim();
+			if (line.Length == 0)
+				return false;
+
+			var key = line;
+			var value = "";
+			var eq = line.IndexOf('=');
+			if (eq >= 0)
+			{
+				key = line.Substring(0, eq).Trim();
+				value = line.Substring(eq + 1, line.Length - eq - 1).Trim();
+			}
+
+			if (currentSection == null)
+				throw new InvalidOperationException("No current INI section");
+
+			if (!currentSection.Contains(key))
+				currentSection.Add(key, value);
+			return true;
+		}
+
+		public IniSection GetSection(string s)
+		{
+			return GetSection(s, false);
+		}
+
+		public IniSection GetSection(string s, bool allowFail)
+		{
+			IniSection section;
+			if (sections.TryGetValue(s.ToLowerInvariant(), out section))
+				return section;
+
+			if (allowFail)
+				return new IniSection(s);
+			throw new InvalidOperationException("Section does not exist in map or rules: " + s);
+		}
+
+		public IEnumerable<IniSection> Sections { get { return sections.Values; } }
+	}
+
+	public class IniSection : IEnumerable<KeyValuePair<string, string>>
+	{
+		public string Name { get; private set; }
+		Dictionary<string, string> values = new Dictionary<string, string>();
+
+		public IniSection(string name)
+		{
+			Name = name;
+		}
+
+		public void Add(string key, string value)
+		{
+			values[key] = value;
+		}
+
+		public bool Contains(string key)
+		{
+			return values.ContainsKey(key);
+		}
+
+		public string GetValue(string key, string defaultValue)
+		{
+			string s;
+			return values.TryGetValue(key, out s) ? s : defaultValue;
+		}
+
+		public IEnumerator<KeyValuePair<string, string>> GetEnumerator()
+		{
+			return values.GetEnumerator();
+		}
+
+		IEnumerator IEnumerable.GetEnumerator()
+		{
+			return GetEnumerator();
+		}
+	}
+}