using System;
using System.Collections.Generic;
using System.Linq;
using iTextSharp.text.pdf;
using PDFPatcher.Common;
using PDFPatcher.Processor;

namespace PDFPatcher.Model
{
	[System.Diagnostics.DebuggerDisplay("Name = {Name}({FriendlyName}); Value = {Value}; {HasChildren}")]
	public sealed class DocumentObject : IHierarchicalObject<DocumentObject>
	{
		static readonly string[] __ReversalRefNames = new string[] { "Parent", "Prev", "First", "Last", "P" };
		static readonly int[] __CompoundTypes = new int[] { PdfObject.DICTIONARY, PdfObject.ARRAY, PdfObject.STREAM };
		static readonly DocumentObject[] __Leaf = new DocumentObject[0];

		internal PdfPathDocument OwnerDocument { get; private set; }
		internal DocumentObject Parent { get; private set; }
		internal string Name { get; set; }
		internal PdfObject Value { get; set; }
		internal string Description { get; set; }
		internal object ExtensiveObject { get; set; }
		internal PdfObjectType Type { get; private set; }
		internal bool IsKeyObject { get; set; }
		internal object ImageKey { get; set; }
		/// <summary>
		/// 获取友好形式的名称。
		/// </summary>
		internal string FriendlyName { get; set; }
		/// <summary>
		/// 获取友好形式的值。
		/// </summary>
		internal string FriendlyValue { get; set; }

		public string LiteralValue => GetItemValueText(Value, ExtensiveObject as PdfObject);
		public bool HasChildren {
			get {
				if (Type != PdfObjectType.Normal
					&& (Type == PdfObjectType.Trailer || Type == PdfObjectType.Pages || Type == PdfObjectType.Page || Type == PdfObjectType.PageCommands || Type == PdfObjectType.Hidden || Type == PdfObjectType.PageCommand && Children.Count > 0)) {
					return true;
				}
				var po = Value ?? ExtensiveObject as PdfObject;
				if (po == null) {
					return false;
				}
				if (PdfHelper.CompoundTypes.Contains(po.Type)) {
					return true;
				}
				else if (po.Type == PdfObject.INDIRECT) {
					if (Type == PdfObjectType.GoToPage) {
						return false;
					}
					var r = ExtensiveObject as PdfObject;
					if (r == null) {
						return false;
					}
					if (r.Type == PdfObject.DICTIONARY && Parent.Type == PdfObjectType.Outline && Name == "Next") {
						return false;
					}
					return (r.Type == PdfObject.DICTIONARY && __ReversalRefNames.Contains(Name) == false
						|| r.Type == PdfObject.ARRAY
						|| r.Type == PdfObject.STREAM
						);
				}
				return false;
			}
		}
		IList<DocumentObject> _Children;
		public ICollection<DocumentObject> Children {
			get {
				if (_Children == null) {
					PopulateChildren(false);
					if (_Children == null) {
						_Children = __Leaf;
					}
				}
				return _Children;
			}
		}

		internal DocumentObject(PdfPathDocument ownerDocument, DocumentObject parent, string name, PdfObject value) :
			this(ownerDocument, parent, name, value, PdfObjectType.Normal) {
		}
		internal DocumentObject(PdfPathDocument ownerDocument, DocumentObject parent, string name, PdfObject value, PdfObjectType type) {
			OwnerDocument = ownerDocument;
			Parent = parent;
			if (value != null && value.Type == PdfObject.INDIRECT) {
				var r = PdfReader.GetPdfObjectRelease(value);
				if (r != null) {
					ExtensiveObject = r;
					if (r.Type == PdfObject.DICTIONARY) {
						int page = ownerDocument.GetPageNumber(value as PdfIndirectReference);
						if (page > 0) {
							Description = String.Concat("指向第 ", page, " 页");
							type = PdfObjectType.GoToPage;
						}
					}
					else if (r.Type == PdfObject.STREAM && PdfName.IMAGE.Equals(((PdfDictionary)r).GetAsName(PdfName.SUBTYPE))) {
						type = PdfObjectType.Image;
					}
				}
			}
			Name = name; Value = value; Type = type;
		}

		internal bool RemoveChildByName(string name) {
			if (HasChildren == false) {
				return false;
			}
			for (int i = _Children.Count - 1; i >= 0; i--) {
				if (_Children[i].Name == name) {
					if (_Children is Array) {
						_Children = new List<DocumentObject>(_Children);
					}
					_Children.RemoveAt(i);
					var po = Value;
					if (po != null) {
						while (po.Type == PdfObject.INDIRECT) {
							po = PdfReader.GetPdfObject(po);
						}
						if (po.Type == PdfObject.ARRAY) {
							((PdfArray)po).Remove(i);
						}
						else if (po.Type == PdfObject.DICTIONARY || po.Type == PdfObject.STREAM) {
							((PdfDictionary)po).Remove(new PdfName(name));
						}
					}
					return true;
				}
			}
			return false;
		}

		internal DocumentObject FindReferenceAncestor() {
			var d = this;
			do {
				if (d.Value != null && d.Value.Type == PdfObject.INDIRECT) {
					return d;
				}
			} while ((d = d.Parent) != null);
			return null;
		}

		internal bool UpdateDocumentObject(object value) {
			var po = Value as PdfObject;
			if (po == null) {
				return false;
			}
			switch (po.Type) {
				case PdfObject.STRING:
					var s = value as string;
					if (s == (po as PdfString).ToUnicodeString()) {
						break;
					}
					Value = s.ToPdfString();
					break;
				case PdfObject.NUMBER:
					double n;
					if (((string)value).TryParse(out n)) {
						Value = new PdfNumber(n);
						break;
					}
					return false;
				case PdfObject.NAME:
					Value = new PdfName((string)value); break;
				case PdfObject.BOOLEAN:
					Value = new PdfBoolean((bool)value); break;
			}
			if (Parent != null) {
				var pd = ((Parent.ExtensiveObject ?? Parent.Value) as PdfDictionary);
				if (pd != null) {
					pd.Put(new PdfName(Name), Value);
					_Children = null;
					return true;
				}
				var pa = ((Parent.ExtensiveObject ?? Parent.Value) as PdfArray);
				if (pa != null) {
					pa.ArrayList[Int32.Parse(Name) - 1] = Value;
					_Children = null;
					return true;
				}
			}
			return false;
		}

		private static string GetItemValueText(PdfObject po, PdfObject eo) {
			if (po == null && eo == null) {
				goto Exit;
			}
			if (po == null) {
				po = eo;
				eo = null;
			}
			switch (po.Type) {
				case PdfObject.DICTIONARY: return String.Concat("<<", (po as PdfDictionary).Size, " 子项>>");
				case PdfObject.INDIRECT:
					if (eo == null || __CompoundTypes.Contains(eo.Type)) {
						return (po as PdfIndirectReference).ToString();
					}
					else {
						return String.Concat((po as PdfIndirectReference).ToString(), "→", GetItemValueText(null, eo));
					}
				case PdfObject.NAME: return PdfHelper.DecodeKeyName(po);
				case PdfObject.NUMBER: return (po as PdfNumber).DoubleValue.ToText();
				case PdfObject.STRING: return (po as PdfString).Decode(null);
				case PdfObject.STREAM: goto case PdfObject.DICTIONARY;
				case PdfObject.ARRAY: return PdfHelper.GetArrayString(po as PdfArray);
				case PdfObject.BOOLEAN: return (po as PdfBoolean).ToString();
				case PdfObject.NULL: return "Null";
			}
		Exit:
			return null;
		}

		internal string GetContextName() {
			var d = this;
			string contextName = null;
			if (d.Type != PdfObjectType.Normal) {
				if (d.Type == PdfObjectType.Page) {
					return "Page";
				}
				else if (d.Type == PdfObjectType.Image) {
					return "Image";
				}
			}
			while ((d.IsKeyObject == false || String.IsNullOrEmpty(contextName = d.Name)) && (d = d.Parent) != null) {
			}
			return contextName;
		}

		internal IList<DocumentObject> PopulateChildren(bool refresh) {
			if (refresh) {
				_Children = null;
			}
			if (_Children == null) {
				if (Type == PdfObjectType.Page && Value == null) {
					Value = OwnerDocument.Document.GetPageN((int)ExtensiveObject);
				}
				else if (Type != PdfObjectType.Normal) {
					PopulateChildrenForSpecialObject();
				}
				if (_Children == null) {
					PopulateChildrenForNormalObject();
				}
			}
			return _Children;
		}

		private void PopulateChildrenForNormalObject() {
			var po = (ExtensiveObject as PdfObject) ?? Value;
			_Children = __Leaf;
			if (po == null) {
				return;
			}
			if (po.Type == PdfObject.DICTIONARY || po.Type == PdfObject.STREAM) {
				var pd = po as PdfDictionary;
				var r = new DocumentObject[pd.Size + (Type == PdfObjectType.Page ? 1 : 0)];
				var n = 0;
				foreach (var item in pd) {
					var d = new DocumentObject(OwnerDocument, this, PdfHelper.DecodeKeyName(item.Key), item.Value);
					r[n++] = d;
					var i = Model.PdfStructInfo.GetInfo(GetContextName(), d.Name);
					if (i.Name != null && i.IsKeyObject) {
						d.IsKeyObject = true;
					}
					if (String.IsNullOrEmpty(i.ImageKey) == false) {
						d.ImageKey = i.ImageKey;
					}
				}
				if (Type != PdfObjectType.Normal) {
					if (Type == PdfObjectType.Page) {
						r[n++] = new DocumentObject(OwnerDocument, this, Constants.Content.Operators, null, PdfObjectType.PageCommands) { IsKeyObject = true };
					}
					else if (Type == PdfObjectType.Trailer) {
						var d = Array.Find(r, (o) => { return o.Name == "Root"; });
						if (d != null) {
							d.Type = PdfObjectType.Root;
						}
					}
					else if (Type == PdfObjectType.Root) {
						var d = Array.Find(r, (o) => { return o.Name == "Outlines"; });
						if (d != null) {
							d.Type = PdfObjectType.Outline;
						}
					}
					else if (Type == PdfObjectType.Outline) {
						var o = new List<DocumentObject>(r);
						var or = pd.Get(PdfName.FIRST);
						pd = PdfReader.GetPdfObject(or) as PdfDictionary;
						if (pd != null) {
							o.Add(new DocumentObject(OwnerDocument, this, Constants.Bookmark, or, PdfObjectType.Outline) {
								Description = pd.Contains(PdfName.TITLE) ? pd.GetAsString(PdfName.TITLE).ToUnicodeString() : null
							});
							while ((or = pd.Get(PdfName.NEXT)) != null && (pd = PdfReader.GetPdfObject(or) as PdfDictionary) != null) {
								o.Add(new DocumentObject(OwnerDocument, this, Constants.Bookmark, or, PdfObjectType.Outline) {
									Description = pd.Contains(PdfName.TITLE) ? pd.GetAsString(PdfName.TITLE).ToUnicodeString() : null
								});
							}
						}
						_Children = o;
						return;
					}
				}
				_Children = r;
			}
			else if (po.Type == PdfObject.ARRAY) {
				var pd = po as PdfArray;
				var r = new DocumentObject[pd.Size];
				var n = 0;
				foreach (var item in pd.ArrayList) {
					var d = new DocumentObject(OwnerDocument, this, (++n).ToText(), item);
					r[n - 1] = d;
					var i = Model.PdfStructInfo.GetInfo(GetContextName(), d.Name);
					if (i.Name != null && i.IsKeyObject) {
						d.IsKeyObject = true;
					}
					if (String.IsNullOrEmpty(i.ImageKey) == false) {
						d.ImageKey = i.ImageKey;
					}
				}
				_Children = r;
			}
		}

		private void PopulateChildrenForSpecialObject() {
			var pdf = OwnerDocument.Document;
			if (Type == PdfObjectType.Pages) {
				if (pdf.NumberOfPages == 0) {
					return;
				}
				var r = PageRangeCollection.Parse(ExtensiveObject as string, 1, pdf.NumberOfPages, true);
				var pn = new DocumentObject[r.TotalPages];
				var i = 0;
				foreach (var item in r) {
					foreach (var p in item) {
						pn[i++] = new DocumentObject(OwnerDocument, this, "第" + p + "页", null, PdfObjectType.Page) { ExtensiveObject = p };
					}
				}
				_Children = pn;
			}
			else if (Type == PdfObjectType.PageCommands) {
				// 解释页面指令
				var pn = (int)Parent.ExtensiveObject;
				var cp = new PdfPageCommandProcessor();
				cp.ProcessContent(pdf.GetPageContent(pn), pdf.GetPageN(pn).GetAsDict(PdfName.RESOURCES));
				foreach (var item in cp.Commands) {
					PopulatePageCommand(item);
				}
			}
			else if (Type == PdfObjectType.PageCommand) {
				_Children = __Leaf;
			}
			else if (Type == PdfObjectType.Hidden) {
				var ul = PdfHelper.ListUnusedObjects(pdf, AppContext.LoadPartialPdfFile);
				ExtensiveObject = ul;
				var uo = new List<DocumentObject>();
				foreach (var item in ul) {
					var u = pdf.GetPdfObjectRelease(item);
					if (u != null) {
						uo.Add(new DocumentObject(OwnerDocument, this, item.ToText(), u));
					}
				}
				_Children = uo;
			}
		}

		private void PopulatePageCommand(PdfPageCommand item) {
			string fn;
			var op = item.Name.ToString();
			if (PdfPageCommand.GetFriendlyCommandName(op, out fn) == false) {
				fn = "未知操作符";
			}
			var o = new DocumentObject(OwnerDocument, this, fn, null, PdfObjectType.PageCommand) {
				FriendlyName = String.Concat(fn, "(", op, ")"),
				ExtensiveObject = op
			};
			if (item.Type == PdfPageCommandType.Text) {
				var t = item as TextCommand;
				o.FriendlyValue = t.TextInfo.PdfString.GetOriginalBytes().ToHexBinString();
				o.Description = t.TextInfo.Text;
				if (item.Name.ToString() == "TJ") {
					var a = item.Operands[0] as PdfArray;
					if (a.Size > 0) {
						var pt = item as PaceAndTextCommand;
						var i = 0;
						CreateChildrenList(ref o._Children);
						foreach (var ti in a.ArrayList) {
							var d = new DocumentObject(OwnerDocument, o, (++i).ToText(), ti);
							if (ti.Type == PdfObject.STRING) {
								d.FriendlyValue = (ti as PdfString).GetOriginalBytes().ToHexBinString();
								d.Description = pt.DecodedTexts[i - 1];
							}
							o._Children.Add(d);
						}
					}
				}
			}
			else if (item.Type == PdfPageCommandType.Font) {
				var f = item as FontCommand;
				o.FriendlyValue = String.Concat(
					f.FontName,
					" (", Constants.Content.OperandNames.ResourceName, "：", f.ResourceName.ToString(), "); ",
					Constants.Content.OperandNames.Size, "：", f.FontSize.DoubleValue.ToText()
					);
			}
			else if (item.Type == PdfPageCommandType.Enclosure) {
				if (item.Operands.HasContent()) {
					var i = 0;
					CreateChildrenList(ref o._Children);
					foreach (var t in item.Operands) {
						o._Children.Add(new DocumentObject(OwnerDocument, o, (++i).ToText(), t));
					}
				}
				var e = item as EnclosingCommand;
				if (e.HasCommand == false) {
					return;
				}
				foreach (var cmd in e.Commands) {
					o.PopulatePageCommand(cmd);
				}
			}
			else if (item.Type == PdfPageCommandType.InlineImage) {
				var s = item.Operands[0] as PdfImageData;
				CreateChildrenList(ref o._Children);
				foreach (var ii in s) {
					o._Children.Add(new DocumentObject(OwnerDocument, o, PdfHelper.DecodeKeyName(ii.Key), ii.Value));
				}
			}
			else {
				o.FriendlyValue = item.GetOperandsText();
			}
			CreateChildrenList(ref _Children);
			_Children.Add(o);
		}

		static void CreateChildrenList(ref IList<DocumentObject> list) {
			if (list == null || list == __Leaf) {
				list = new List<DocumentObject>();
			}
		}

	}
}