module pixelperfectengine.system.lang.textparser;

public import pixelperfectengine.graphics.fontsets;
public import pixelperfectengine.graphics.bitmap;

public import pixelperfectengine.graphics.text;
public import pixelperfectengine.system.exc;
//public import pixelperfectengine.system.etc : isInteger;


import newxml.interfaces : XMLException;
import std.utf : toUTF32, toUTF8;
import std.conv : to;
import std.algorithm : countUntil;
import std.exception : enforce;
import std.typecons : BitFlags;
import core.exception : RangeError;

/**
 * Parses text from XML/ETML
 *
 * See "ETML.md" for further info on how the ETML format works.
 */
public class TextParserTempl(BitmapType = Bitmap8Bit)
		if (is(BitmapType == Bitmap8Bit) || is(BitmapType == Bitmap16Bit) || is(BitmapType Bitmap32Bit)) {
	alias TextType = TextTempl!BitmapType;
	alias ChrFormat = CharacterFormattingInfo!BitmapType;
	public TextType[string]			output;		///All texts found within the ETML file
	private TextType				chunkRoot;
	private TextType				currTextBlock;
	//private ChrFormat				currFrmt;///currently parsed formatting
	public ChrFormat[]				chrFrmt;///All character format, that has been parsed so far
	public ChrFormat[]				frmtStack;	///Character formatting stack.
	public ChrFormat[string]		namedFormats;///Named format lookup table.
	public Fontset!BitmapType[string]	fontsets;///Fontset name association table
	public BitmapType[string]		icons;		///Icon name association table
	public dstring[dstring]			customEntities;///Custom entities that are loaded during parsing.
	private dstring					_input;		///The source XML/ETML document
	private bool					inTextChunk;///Set to true if currently in a text chunk to allow detection of cascading 
	
	///Creates a new instance with a select string input.
	///Params:
	///  _input = the input text containing the xml file.
	///  _defFrmt = the default character formatting
	public this(dstring _input, CharacterFormattingInfo!BitmapType _defFrmt) @safe pure nothrow {
		this._input = _input;
		chrFrmt = [_defFrmt];
	}
	///Gets the default formatting
	public CharacterFormattingInfo!BitmapType defaultFormatting() @property @safe pure nothrow @nogc {
		return defFrmt;
	}
	private ChrFormat currFrmt() @property @safe pure nothrow @nogc {
		return frmtStack[$ - 1];
	}
	private ref ChrFormat defFrmt() @property @safe pure nothrow @nogc {
		return chrFrmt[0];
	}
	///Sets/gets the input
	public dstring input() @property @safe @nogc pure nothrow inout {
		return _input;
	}
	/**
	 * Parses the formatted text, then sets the output values.
	 */
	public void parse()  {
		auto createParser() {
			import newxml;
			return _input.lexer.parser.cursor.saxParser;
		}
		auto parser0 = createParser;
		parser0.setSource(_input);
		//add whitespace entities
		//parser0.cursor.parser.chrEntities["spc"] = " ";
		parser0.onElementStart = &onElementStart;
		parser0.onElementEnd = &onElementEnd;
		parser0.onElementEmpty = &onElementEmpty;
		parser0.onText = &onText;
		try {
			parser0.processDocument();
		} catch (XMLException e) {	//XML formatting issue
			throw new XMLTextParsingException("XML file is badly formatted!", e);
		} catch (RangeError e) {	//Missing mandatory attributes
			throw new XMLTextParsingException("Missing mandatory attribute found!", e);
		} finally {
			_input.length = 0;
		}
	}
	protected void onText(dstring content) @safe {
		if (currTextBlock.charLength) {
			currTextBlock.next = new TextType(content, currFrmt);
			currTextBlock = currTextBlock.next;
		} else {
			currTextBlock.text = content;
		}
	}
	protected void onElementStart(dstring name, dstring[dstring] attr) @safe {
		switch (name) {
			case "text":
				onTextElementStart(attr);
				break;
			case "font":
				onFontElementStart(attr);
				break;
			case "p":
				onPElementStart(attr);
				break;
			case "u":
				onLineFormatElementStart!"u"(attr);
				break;
			case "s":
				onLineFormatElementStart!"s"(attr);
				break;
			case "o":
				onLineFormatElementStart!"o"(attr);
				break;
			case "i":
				onIElementStart(attr);
				break;
			default:
				break;
		}
	}
	protected void onElementEmpty(dstring name, dstring[dstring] attr) @safe {
		switch (name) {
			case "br":
				onBrElement();
				break;
			case "image":
				onImageElement(attr);
				break;
			case "frontTab", "ft":
				onFrontTabElement(attr);
				break;
			case "formatDef":
				onFormatDefElement(attr);
				break;
			default:
				break;
		}
	}
	protected void onElementEnd(dstring name) @safe {
		switch (name) {
			case "p":
			case "u":
			case "s":
			case "o":
			case "i":
			case "font":
			case "format":
				closeFormattingTag();
				break;
			default:
				break;
		}
	}
	/** 
	 * Begins a new text chain, and flushes the font stack.
	 * Params:
	 *   attributes = Attributes are received here from textparser.
	 */
	protected void onTextElementStart(dstring[dstring] attributes) @safe {
		string textID = toUTF8(attributes["id"]);
		currTextBlock = new TextType(null, defFrmt);
		//currFrmt = defFrmt;
		frmtStack = [defFrmt];
		//Add first chunk to the output, so it can be recalled later on.
		output[textID] = currTextBlock;
	}
	protected void onPElementStart(dstring[dstring] attributes) @safe {
		currTextBlock.next = new TextType(null, currFrmt);
		currTextBlock = currTextBlock.next;
		ChrFormat newFrmt = new ChrFormat(currFrmt);
		dstring paragraphSpaceStr = attributes.get("paragraphSpace", null);
		try {
			newFrmt.paragraphSpace = paragraphSpaceStr.to!ushort;
		} catch (Exception e) {
			
		}
		dstring rowHeightStr = attributes.get("rowHeight", null);
		try {
			newFrmt.rowHeight = rowHeightStr.to!short;
		} catch (Exception e) {
			
		}
		testFormatting(newFrmt);
		currTextBlock.formatting = currFrmt;
		currTextBlock.flags |= TextType.Flags.newParagraph;
	}
	protected void onLineFormatElementStart(string Type)(dstring[dstring] attributes) @safe {
		closeTextBlockIfNotEmpty();
		ChrFormat newFrmt = new ChrFormat(currFrmt);
		static if (Type == "u") {
			dstring style = attributes.get("style", null);
			newFrmt.formatFlags |= FormattingFlags.underline;
			if (style.length)
				newFrmt.formatFlags &= ~FormattingFlags.ulLineMultiplier;
			switch (style) {
				case "double":
					newFrmt.formatFlags |= FormattingFlags.underlineDouble;
					break;
				case "triple":
					newFrmt.formatFlags |= FormattingFlags.underlineTriple;
					break;
				case "quad":
					newFrmt.formatFlags |= FormattingFlags.underlineQuadruple;
					break;
				default:
					break;
			}
			dstring lines = attributes.get("lines", null);
			if (lines.length)
				newFrmt.formatFlags &= ~FormattingFlags.ulLineStyle;
			switch (lines) {
				case "dotted":
					newFrmt.formatFlags |= FormattingFlags.underlineDotted;
					break;
				case "wavy":
					newFrmt.formatFlags |= FormattingFlags.underlineWavy;
					break;
				case "wavySoft":
					newFrmt.formatFlags |= FormattingFlags.underlineWavySoft;
					break;
				case "stripes":
					newFrmt.formatFlags |= FormattingFlags.underlineStripes;
					break;
				default:
					break;
			}
			dstring perWord = attributes.get("perWord", null);
			if (perWord == "true" || perWord == "yes") {
				newFrmt.formatFlags |= FormattingFlags.underlinePerWord;
			} else {
				newFrmt.formatFlags &= ~FormattingFlags.underlinePerWord;
			}
		} else static if (Type == "s") {
			newFrmt.formatFlags |= FormattingFlags.strikeThrough;
		} else static if (Type == "o") {
			newFrmt.formatFlags |= FormattingFlags.overline;
		}
		testFormatting(newFrmt);
		currTextBlock.formatting = currFrmt;
	}
	protected void onIElementStart(dstring[dstring] attributes) @safe {
		closeTextBlockIfNotEmpty();
		ChrFormat newFrmt = new ChrFormat(currFrmt);
		dstring amount = attributes.get("amount", null);
		newFrmt.formatFlags &= ~FormattingFlags.forceItalicsMask;
		switch (amount) {
			case "1/3":
				newFrmt.formatFlags |= FormattingFlags.forceItalics1per3;
				break;
			case "1/2":
				newFrmt.formatFlags |= FormattingFlags.forceItalics1per2;
				break;
			default:	//case "1/4":
				newFrmt.formatFlags |= FormattingFlags.forceItalics1per4;
				break;
		}
		testFormatting(newFrmt);
		currTextBlock.formatting = currFrmt;
	}
	protected void onFontElementStart(dstring[dstring] attributes) @safe {
		closeTextBlockIfNotEmpty();
		ChrFormat newFrmt = new ChrFormat(currFrmt);
		string type = toUTF8(attributes.get("font", null));
		if (type.length) {
			newFrmt.font = fontsets[type];
		}
		string color = toUTF8(attributes.get("color", null));
		if (color.length) {
			static if (is(BitmapType == Bitmap8Bit)) {
				newFrmt.color = color.to!ubyte();
			} else static if (is(BitmapType == Bitmap16Bit)) {
				newFrmt.color = color.to!ushort();
			} else {

			}
		}
		testFormatting(newFrmt);
		currTextBlock.formatting = currFrmt;
	}
	protected void onFormatElementStart(dstring[dstring] attributes) @safe {
		closeTextBlockIfNotEmpty();
		currTextBlock.formatting = namedFormats[toUTF8(attributes["id"])];
	}
	protected void onFormatDefElement(dstring[dstring] attributes) @safe {
		bool checkBoolean(dstring defFrmt) @safe {
			if (defFrmt == "true" || defFrmt == "yes")
				return true;
			else
				return false;
		}
		closeTextBlockIfNotEmpty();
		ChrFormat newFrmt = new ChrFormat(currFrmt);
		foreach (dstring key, dstring elem; attributes) {
			switch (key) {
				case "u":
					if (checkBoolean(elem))
						newFrmt.formatFlags &= ~FormattingFlags.underline;
					else
						newFrmt.formatFlags |= FormattingFlags.underline;
					break;
				case "s":
					if (checkBoolean(elem))
						newFrmt.formatFlags &= ~FormattingFlags.strikeThrough;
					else
						newFrmt.formatFlags |= FormattingFlags.strikeThrough;
					break;
				case "o":
					if (checkBoolean(elem))
						newFrmt.formatFlags &= ~FormattingFlags.overline;
					else
						newFrmt.formatFlags |= FormattingFlags.overline;
					break;
				case "i":
					if (checkBoolean(elem)) {
						newFrmt.formatFlags &= ~FormattingFlags.forceItalicsMask;
					} else {
						dstring i_amount = attributes.get("i_amount", null);
						switch (i_amount) {
							case "1/2":
								newFrmt.formatFlags |= FormattingFlags.forceItalics1per2;
								break;
							case "1/3":
								newFrmt.formatFlags |= FormattingFlags.forceItalics1per3;
								break;
							default:
								newFrmt.formatFlags |= FormattingFlags.forceItalics1per4;
								break;
						}
					}
					break;
				default:
					break;
			}
		}
		const sizediff_t frmtNum = hasChrFormatting(newFrmt);
		if (frmtNum == -1) {
			namedFormats[toUTF8(attributes["id"])] = newFrmt;
			chrFrmt ~= newFrmt;
		} else {
			namedFormats[toUTF8(attributes["id"])] = chrFrmt[frmtNum];
		}
	}
	protected void onBrElement() @safe {
		currTextBlock.next = new TextType(null, currFrmt);
		currTextBlock = currTextBlock.next;
		currTextBlock.flags |= TextType.Flags.newLine;
	}
	protected void onFrontTabElement(dstring[dstring] attributes) @safe {
		closeTextBlockIfNotEmpty();
		currTextBlock.frontTab = attributes["amount"].to!int;
	}
	protected void onImageElement(dstring[dstring] attributes) @safe {
		if (currTextBlock.charLength || currTextBlock.icon) {
			currTextBlock.next = new TextType(null, currFrmt);
			currTextBlock = currTextBlock.next;
		}
	}
	protected void closeFormattingTag() @safe {
		removeTopFromFrmtStack();
		closeTextBlockIfNotEmpty();
	}
	protected final void closeTextBlockIfNotEmpty() @safe {
		if (currTextBlock.charLength) {
			currTextBlock.next = new TextType(null, currFrmt);
			currTextBlock = currTextBlock.next;
			//currTextBlock.flags |= TextType.Flags.newLine;
		}
	}
	protected final void removeTopFromFrmtStack() @safe {
		frmtStack = frmtStack[0..$-1];
		enforce!XMLTextParsingException(frmtStack.length >= 1, "Formatting stack is empty!");
	}
	protected final void testFormatting(ChrFormat frmt) @safe {
		sizediff_t frmtNum = hasChrFormatting(frmt);
		if (frmtNum == -1) {	///New formatting found, add this to overall formatting
			chrFrmt ~= frmt;
			frmtStack ~= frmt;
		} else {				///Already used formatting found, use that instead
			frmtStack ~= chrFrmt[frmtNum];
		}
	}
	protected final sizediff_t hasChrFormatting(ChrFormat frmt) @trusted {
		return countUntil(chrFrmt, frmt);
	}
}
alias TextParser = TextParserTempl!Bitmap8Bit;
public class XMLTextParsingException : PPEException {
	@nogc @safe pure nothrow this(string msg, string file = __FILE__, size_t line = __LINE__, Throwable nextInChain = null)
			{
		super(msg, file, line, nextInChain);
	}

	@nogc @safe pure nothrow this(string msg, Throwable nextInChain, string file = __FILE__, size_t line = __LINE__) {
		super(msg, file, line, nextInChain);
	}
}