CTT/Unity/Assets/ThirdParty/FairyGUI/Scripts/Utils/XML.cs

using System;
using System.Collections.Generic;
using UnityEngine;

namespace FairyGUI.Utils
{
	/// <summary>
	/// A simplest and readonly XML class
	/// </summary>
	public class XML
	{
		public string name { get; private set; }
		public string text { get; private set; }

		Dictionary<string, string> _attributes;
		XMLList _children;

		public XML(string text)
		{
			Parse(text);
		}

		private XML()
		{
		}

		public bool HasAttribute(string attrName)
		{
			if (_attributes == null)
				return false;

			return _attributes.ContainsKey(attrName);
		}

		public string GetAttribute(string attrName)
		{
			return GetAttribute(attrName, null);
		}

		public string GetAttribute(string attrName, string defValue)
		{
			if (_attributes == null)
				return defValue;

			string ret;
			if (_attributes.TryGetValue(attrName, out ret))
				return ret;
			else
				return defValue;
		}

		public int GetAttributeInt(string attrName)
		{
			return GetAttributeInt(attrName, 0);
		}

		public int GetAttributeInt(string attrName, int defValue)
		{
			string value = GetAttribute(attrName);
			if (value == null || value.Length == 0)
				return defValue;

			int ret;
			if (int.TryParse(value, out ret))
				return ret;
			else
				return defValue;
		}

		public float GetAttributeFloat(string attrName)
		{
			return GetAttributeFloat(attrName, 0);
		}

		public float GetAttributeFloat(string attrName, float defValue)
		{
			string value = GetAttribute(attrName);
			if (value == null || value.Length == 0)
				return defValue;

			float ret;
			if (float.TryParse(value, out ret))
				return ret;
			else
				return defValue;
		}

		public bool GetAttributeBool(string attrName)
		{
			return GetAttributeBool(attrName, false);
		}

		public bool GetAttributeBool(string attrName, bool defValue)
		{
			string value = GetAttribute(attrName);
			if (value == null || value.Length == 0)
				return defValue;

			bool ret;
			if (bool.TryParse(value, out ret))
				return ret;
			else
				return defValue;
		}

		public string[] GetAttributeArray(string attrName)
		{
			string value = GetAttribute(attrName);
			if (value != null)
			{
				if (value.Length == 0)
					return new string[] { };
				else
					return value.Split(',');
			}
			else
				return null;
		}

		public string[] GetAttributeArray(string attrName, char seperator)
		{
			string value = GetAttribute(attrName);
			if (value != null)
			{
				if (value.Length == 0)
					return new string[] { };
				else
					return value.Split(seperator);
			}
			else
				return null;
		}

		public Color GetAttributeColor(string attrName, Color defValue)
		{
			string value = GetAttribute(attrName);
			if (value == null || value.Length == 0)
				return defValue;

			return ToolSet.ConvertFromHtmlColor(value);
		}

		public Vector2 GetAttributeVector(string attrName)
		{
			string value = GetAttribute(attrName);
			if (value != null)
			{
				string[] arr = value.Split(',');
				return new Vector2(float.Parse(arr[0]), float.Parse(arr[1]));
			}
			else
				return Vector2.zero;
		}

		public void SetAttribute(string attrName, string attrValue)
		{
			if (_attributes == null)
				_attributes = new Dictionary<string, string>();

			_attributes[attrName] = attrValue;
		}

		public XML GetNode(string selector)
		{
			if (_children == null)
				return null;
			else
				return _children.Find(selector);
		}

		public XMLList Elements()
		{
			if (_children == null)
				_children = new XMLList();
			return _children;
		}

		public XMLList Elements(string selector)
		{
			if (_children == null)
				_children = new XMLList();
			return _children.Filter(selector);
		}

		public XMLList.Enumerator GetEnumerator()
		{
			if (_children == null)
				return new XMLList.Enumerator(null, null);
			else
				return new XMLList.Enumerator(_children.rawList, null);
		}

		public XMLList.Enumerator GetEnumerator(string selector)
		{
			if (_children == null)
				return new XMLList.Enumerator(null, selector);
			else
				return new XMLList.Enumerator(_children.rawList, selector);
		}

		static Stack<XML> sNodeStack = new Stack<XML>();
		void Parse(string aSource)
		{
			XML lastOpenNode = null;
			sNodeStack.Clear();

			XMLIterator.Begin(aSource);
			while (XMLIterator.NextTag())
			{
				if (XMLIterator.tagType == XMLTagType.Start || XMLIterator.tagType == XMLTagType.Void)
				{
					XML childNode;
					if (lastOpenNode != null)
						childNode = new XML();
					else
					{
						if (this.name != null)
						{
							Cleanup();
							throw new Exception("Invalid xml format - no root node.");
						}
						childNode = this;
					}

					childNode.name = XMLIterator.tagName;
					childNode._attributes = XMLIterator.GetAttributes(childNode._attributes);

					if (lastOpenNode != null)
					{
						if (XMLIterator.tagType != XMLTagType.Void)
							sNodeStack.Push(lastOpenNode);
						if (lastOpenNode._children == null)
							lastOpenNode._children = new XMLList();
						lastOpenNode._children.Add(childNode);
					}
					if (XMLIterator.tagType != XMLTagType.Void)
						lastOpenNode = childNode;
				}
				else if (XMLIterator.tagType == XMLTagType.End)
				{
					if (lastOpenNode == null || lastOpenNode.name != XMLIterator.tagName)
					{
						Cleanup();
						throw new Exception("Invalid xml format - <" + XMLIterator.tagName + "> dismatched.");
					}

					if (lastOpenNode._children == null || lastOpenNode._children.Count == 0)
					{
						lastOpenNode.text = XMLIterator.GetText();
					}

					if (sNodeStack.Count > 0)
						lastOpenNode = sNodeStack.Pop();
					else
						lastOpenNode = null;
				}
			}
		}

		void Cleanup()
		{
			this.name = null;
			if (this._attributes != null)
				this._attributes.Clear();
			if (this._children != null)
				this._children.Clear();
			this.text = null;
		}
	}
}