#include "global_pre.hpp"
#include "TSTextField.hpp"
#include "widgets.hpp"
#include "ui.hpp"
// for gVg
#include "window.hpp"
// for key codes
// // #include <GLFW/glfw3.h>
#include "global_ui.hpp"
using namespace rack;

#include "trowaSoftComponents.hpp"

TSTextField::TSTextField(TextType textType) : TextField() {
	setTextType(textType);
	font = Font::load(assetPlugin(plugin, TROWA_MONOSPACE_FONT));
	fontSize = 14.0f;
	backgroundColor = FORMS_DEFAULT_BG_COLOR;
	color = FORMS_DEFAULT_TEXT_COLOR;
	textOffset = Vec(0, 0);
	borderWidth = 1;
	borderColor = FORMS_DEFAULT_BORDER_COLOR;
	//caretColor = COLOR_TS_RED;// nvgRGBAf(1.0f - color.r, 1.0f - color.g, 1.0f - color.b, 0.70);
	caretColor = nvgRGBAf((color.r + backgroundColor.r) / 2.0, (color.g + backgroundColor.g) / 2.0, (color.b + backgroundColor.b) / 2.0, 0.70);
	return;
}
TSTextField::TSTextField(TextType textType, int maxLength) : TSTextField(textType) {
	this->maxLength = maxLength;
	return;
}


// Taken from Rack's LEDTextField
int TSTextField::getTextPosition(Vec mousePos) {
	bndSetFont(font->handle);
	int textPos = bndIconLabelTextPosition(rack::global_ui->window.gVg, textOffset.x, textOffset.y,
		box.size.x - 2 * textOffset.x, box.size.y - 2 * textOffset.y,
		-1, fontSize, displayStr.c_str(), mousePos.x, mousePos.y);
	bndSetFont(rack::global_ui->window.gGuiFont->handle);
	return textPos;
}

// Draw if visible.
void TSTextField::draw(NVGcontext *vg) {
	if (visible)
	{
		// Draw taken from Rack's LEDTextField and modified for scrolling (my quick & dirty ghetto text scrolling---ONLY truly effective for calculating the width with MONOSPACE font
		// since I don't want to do a bunch of calcs... [lazy]).
		nvgScissor(vg, 0, 0, box.size.x, box.size.y);

		// Background
		nvgBeginPath(vg);
		nvgRoundedRect(vg, 0, 0, box.size.x, box.size.y, 5.0);
		nvgFillColor(vg, backgroundColor);
		nvgFill(vg);

		// Border
		if (borderWidth > 0) {
			nvgStrokeWidth(vg, borderWidth);
			nvgStrokeColor(vg, borderColor);
			nvgStroke(vg);
		}

		// Text
		if (font->handle >= 0) {
			bndSetFont(font->handle);

			//NVGcolor highlightColor = color;
			//highlightColor.a = 0.5;
			int begin = min(cursor, selection);
			int end = (this == RACK_PLUGIN_UI_FOCUSED_WIDGET) ? max(cursor, selection) : -1;

			// Calculate overflow and the displayed text (based on bounding box)
			// Currently the scrolling should work for any font, **BUT** the width calculation is only really good for monospace.
			float txtBounds[4] = { 0,0,0,0 };
			nvgTextAlign(vg, NVG_ALIGN_LEFT);
			nvgFontSize(vg, fontSize);
			nvgFontFaceId(vg, font->handle);
			int maxTextWidth = box.size.x - textOffset.x * 2 - fontSize / 2.0; // There should be a caret
			float estLetterSize = nvgTextBounds(vg, 0, 0, "X", NULL, txtBounds); // Estimate size of a letter (accurate for monospace)
			float nextX = nvgTextBounds(vg, 0, 0, text.c_str(), NULL, txtBounds); // Calculate full string size

			displayStr = text;
			if (nextX > maxTextWidth) {
				int nChars = maxTextWidth / estLetterSize;
				if (nChars < 1)
					nChars = 1;

				if (this == RACK_PLUGIN_UI_FOCUSED_WIDGET) {
					int lastIx = (cursor > nChars) ? cursor : nChars;
					int startIx = clamp(lastIx - nChars, 0, lastIx);
					displayStr = text.substr(startIx, nChars);
					begin -= startIx;
					if (end > -1)
						end -= startIx;
				}
				else {
					displayStr = text.substr(0, nChars);
				}
			}


			// The caret color actually isn't the cursor color (that is hard-coded as nvgRGBf(0.337,0.502,0.761))
			// 

			//void bndIconLabelCaret(NVGcontext *ctx, float x, float y, float w, float h,
			//	int iconid, NVGcolor color, float fontsize, const char *label,
			//	NVGcolor caretcolor, int cbegin, int cend
			bndIconLabelCaret(vg, /*x*/ textOffset.x, /*y*/ textOffset.y,
				/*w*/ box.size.x - 2 * textOffset.x, /*h*/ box.size.y - 2 * textOffset.y,
				/*iconid*/ -1, /*textColor*/ color, /*fontsize*/ fontSize, 
				/*label*/ displayStr.c_str(), 
				/*caretcolor*/ caretColor, /*cbegin*/ begin, /*cend*/ end);

			bndSetFont(rack::global_ui->window.gGuiFont->handle);
		}

		nvgResetScissor(vg);
	}
} // end draw()

// Request focus on this field.
void TSTextField::requestFocus() {
	if (RACK_PLUGIN_UI_FOCUSED_WIDGET) {
		EventDefocus evt;
		RACK_PLUGIN_UI_FOCUSED_WIDGET->onDefocus(evt);
		RACK_PLUGIN_UI_FOCUSED_WIDGET_SET(NULL);
	}
	RACK_PLUGIN_UI_FOCUSED_WIDGET_SET(this);
	{
		EventFocus eFocus;
		onFocus(eFocus);
		cursor = 0;
		selection = text.length();
	}
	return;
} // end requestFocus()

// Remove invalid chars from input.
std::string TSTextField::cleanseString(std::string newText)
{
	if (allowedTextType == TextType::Any)
	{
		return newText.substr(0, maxLength);
	}
	else
	{
		// Remove invalid chars
		std::stringstream cleansedStr;
		// Issue: https://github.com/j4s0n-c/trowaSoft-VCV/issues/5. Changed from string constant (emtpy string "") to string object empty string ("") to older Linux compilers. Thx to @Chaircrusher.
		std::regex_replace(std::ostream_iterator<char>(cleansedStr), newText.begin(), newText.end(), regexInvalidChar, std::string(""));
		return cleansedStr.str().substr(0, maxLength);
	}
} // end cleanseString()

// Remove invalid chars
/** Inserts text at the cursor, replacing the selection if necessary */
void TSTextField::insertText(std::string newText) {
	if (cursor != selection) {
		int begin = min(cursor, selection);
		this->text.erase(begin, std::abs(selection - cursor));
		cursor = selection = begin;
	}
	std::string cleansedStr = cleanseString(newText);
	this->text.insert(cursor, cleansedStr);
	cursor += cleansedStr.size();
	selection = cursor;
	onTextChange();
	return;
} // end insertText()

// On Key
void TSTextField::onText(EventText &e) {
	if (enabled)
	{
		if (e.codepoint < 128) {
			std::string newText(1, (char)e.codepoint);
			//insertText(newText);
			if ((allowedTextType == TextType::Any || regex_match(newText, regexChar)) && text.length() < maxLength)
			{
				insertText(newText);
			}
		}
	}
	e.consumed = true;
	return;
} // end onText()

void TSTextField::setText(std::string text) {
	this->text = cleanseString(text);
	selection = cursor = text.size();
	onTextChange();
}
// When the text changes.
void TSTextField::onTextChange() {
	text = cleanseString(text);
	cursor = clamp(cursor, 0, text.size());
	selection = clamp(selection, 0, text.size());
	//debug("onTextChange() - New cursor: %d", cursor);
	return;
} // end onTextChanged()

// On key press.
void TSTextField::onKey(EventKey &e) {
	if (!visible)
	{
		// Do not capture the keys.
		e.consumed = false;
		return;
	}
	if (!enabled)
	{
		e.consumed = false; // We are ingoring this.
		return;
	}
	// We can throw invalid chars away in onText(), so we don't have to check here anymore.
	//// Flag if we need to validate/cleanse this character (only if printable and if we are doing validation).
	//bool checkKey = (this->allowedTextType != TextType::Any) && isPrintableKey(e.key);
	switch (e.key) {
		case LGLW_VKEY_TAB/*GLFW_KEY_TAB*/:
			// If we have an event to fire, then do it
			if (windowIsShiftPressed())//(guiIsShiftPressed())
			{
				if (onShiftTabCallback != NULL)
				{			
					onShiftTabCallback(id);
				}
				else if (prevField != NULL)
				{
					TSTextField* fField = prevField;
					if (!fField->visible)
					{
						switch (tabNextHiddenAction)
						{
						case TabFieldHiddenAction::MoveToNextVisibleTabField:
							fField = fField->prevField;
							while (fField != NULL && !fField->visible && fField != this && !fField->canTabToThisEnabled)
								fField = fField->prevField;
							if (fField == this || (fField != NULL && !fField->visible))
								fField = NULL;
							break;
						case TabFieldHiddenAction::ShowHiddenTabToField:
							while (fField != NULL && fField != this && !fField->canTabToThisEnabled)
								fField = fField->prevField;
							if (fField == this || (fField != NULL && !fField->canTabToThisEnabled))
								fField = NULL;
							if (fField != NULL)
								fField->visible = true;
							break;
						case TabFieldHiddenAction::DoNothing:
						default:
							fField = NULL;
							break;
						}
					}
					if (fField != NULL)
					{
						fField->requestFocus();
					}
				} // end if previous field
			}
			else if (onTabCallback != NULL)
			{
				onTabCallback(id);
			}
			else if (nextField != NULL)
			{
				TSTextField* fField = nextField;
				if (!fField->visible)
				{
					switch (tabNextHiddenAction)
					{
					case TabFieldHiddenAction::MoveToNextVisibleTabField:
						fField = fField->nextField;
						while (fField != NULL && !fField->visible && fField != this && !fField->canTabToThisEnabled)
							fField = fField->nextField;
						if (fField == this || (fField != NULL && !fField->visible))
							fField = NULL;
						break;
					case TabFieldHiddenAction::ShowHiddenTabToField:
						while (fField != NULL && fField != this && !fField->canTabToThisEnabled)
							fField = fField->nextField;
						if (fField == this || (fField != NULL && !fField->canTabToThisEnabled))
							fField = NULL;
						if (fField != NULL)
							fField->visible = true;
						break;
					case TabFieldHiddenAction::DoNothing:
					default:
						fField = NULL;
						break;
					}
				}
				if (fField != NULL)
				{
					fField->requestFocus();
				}
			} // end if next field
			break;
		// case GLFW_KEY_KP_ENTER:
		// {
		// 	// Key pad enter should also trigger event action
		// 	EventAction evt;
		// 	onAction(evt);
		// }
		// 	break;
		default:
			// Call base method
			TextField::onKey(e);
			break;
	}
	cursor = clamp(cursor, 0, text.size());
	selection = clamp(selection, 0, text.size());
	e.consumed = true;
	return;
} // end onKey()
//-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-
// isPrintableKey()
// @keyCode : (IN) The key that is pressed.
// @returns: True if the key represents a printable character, false if not.
//-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-
bool isPrintableKey(int key)
{
	bool isPrintable = false;
	switch (key)
	{
      case ' '/*GLFW_KEY_SPACE*/:
      case '\''/*GLFW_KEY_APOSTROPHE*/:
      case ','/*GLFW_KEY_COMMA*/:
      case '-'/*GLFW_KEY_MINUS*/:
      case '.'/*GLFW_KEY_PERIOD*/:
      case '/'/*GLFW_KEY_SLASH*/:
      case '0'/*GLFW_KEY_0*/:
      case '1'/*GLFW_KEY_1*/:
      case '2'/*GLFW_KEY_2*/:
      case '3'/*GLFW_KEY_3*/:
      case '4'/*GLFW_KEY_4*/:
      case '5'/*GLFW_KEY_5*/:
      case '6'/*GLFW_KEY_6*/:
      case '7'/*GLFW_KEY_7*/:
      case '8'/*GLFW_KEY_8*/:
      case '9'/*GLFW_KEY_9*/:
      case ';'/*GLFW_KEY_SEMICOLON*/:
      case '='/*GLFW_KEY_EQUAL*/:
      case 'a'/*GLFW_KEY_A*/:
      case 'b'/*GLFW_KEY_B*/:
      case 'c'/*GLFW_KEY_C*/:
      case 'd'/*GLFW_KEY_D*/:
      case 'e'/*GLFW_KEY_E*/:
      case 'f'/*GLFW_KEY_F*/:
      case 'g'/*GLFW_KEY_G*/:
      case 'h'/*GLFW_KEY_H*/:
      case 'i'/*GLFW_KEY_I*/:
      case 'j'/*GLFW_KEY_J*/:
      case 'k'/*GLFW_KEY_K*/:
      case 'l'/*GLFW_KEY_L*/:
      case 'm'/*GLFW_KEY_M*/:
      case 'n'/*GLFW_KEY_N*/:
      case 'o'/*GLFW_KEY_O*/:
      case 'p'/*GLFW_KEY_P*/:
      case 'q'/*GLFW_KEY_Q*/:
      case 'r'/*GLFW_KEY_R*/:
      case 's'/*GLFW_KEY_S*/:
      case 't'/*GLFW_KEY_T*/:
      case 'u'/*GLFW_KEY_U*/:
      case 'v'/*GLFW_KEY_V*/:
      case 'w'/*GLFW_KEY_W*/:
      case 'x'/*GLFW_KEY_X*/:
      case 'y'/*GLFW_KEY_Y*/:
      case 'z'/*GLFW_KEY_Z*/:
      case '['/*GLFW_KEY_LEFT_BRACKET*/:
      case '\\'/*GLFW_KEY_BACKSLASH*/:
      case ']'/*GLFW_KEY_RIGHT_BRACKET*/:
      case '^'/*GLFW_KEY_GRAVE_ACCENT*/:
      // case GLFW_KEY_WORLD_1:
      // case GLFW_KEY_WORLD_2:
         isPrintable = true;
         break;
	}
	return isPrintable;
} // end isPrintableKey()