#include "toonzqt/scriptconsole.h"
#include "toonz/scriptengine.h"
#include <QKeyEvent>
#include <QTextBlock>
#include <QUrl>
#include <QMimeData>
 
ScriptConsole::ScriptConsole(QWidget *parent)
	: QTextEdit(parent), m_commandIndex(0)
{
	m_prompt = ">> ";
 
	append(m_prompt);
	moveCursor(QTextCursor::EndOfLine);
 
	m_engine = new ScriptEngine();
	connect(m_engine, SIGNAL(evaluationDone()), this, SLOT(onEvaluationDone()));
	connect(m_engine, SIGNAL(output(int, const QString &)), this, SLOT(output(int, const QString &)));
	connect(this, SIGNAL(cursorPositionChanged()), this, SLOT(onCursorPositionChanged()));
}
 
ScriptConsole::~ScriptConsole()
{
	delete m_engine;
}
 
void ScriptConsole::keyPressEvent(QKeyEvent *e)
{
	if (e->modifiers() == Qt::ControlModifier && e->key() == Qt::Key_Y) {
		if (m_engine->isEvaluating()) {
			m_engine->interrupt();
			setTextColor(QColor(255, 127, 0));
			append("Interrupt");
			moveCursor(QTextCursor::EndOfLine);
			setTextColor(Qt::black);
		}
		return;
	}
 
	switch (e->key()) {
	case Qt::Key_Return:
		onReturnKeyPress();
		break;
	case Qt::Key_Up:
		if (m_commandIndex > 0) {
			moveCursor(QTextCursor::End);
			moveCursor(QTextCursor::StartOfBlock);
			moveCursor(QTextCursor::EndOfBlock, QTextCursor::KeepAnchor);
			// the first KeyUp save the current command, possibly not completed yet
			// note: m_currentCommand is the whole line, containing the prompt
			if (m_commandIndex == m_commands.count())
				m_currentCommand = textCursor().selectedText();
			textCursor().insertText(m_prompt + m_commands[--m_commandIndex]);
		}
		break;
	case Qt::Key_Down:
		if (m_commandIndex < m_commands.count()) {
			moveCursor(QTextCursor::End);
			moveCursor(QTextCursor::StartOfBlock);
			moveCursor(QTextCursor::EndOfBlock, QTextCursor::KeepAnchor);
			if (m_commandIndex == m_commands.count() - 1) {
				// the last KeyDown insert back the current command, possibly not completed yet
				textCursor().insertText(m_currentCommand);
				++m_commandIndex;
			} else
				textCursor().insertText(m_prompt + m_commands[++m_commandIndex]);
		}
		break;
	case Qt::Key_Backspace:
	case Qt::Key_Left:
		if (textCursor().positionInBlock() > 3)
			QTextEdit::keyPressEvent(e);
		else
			e->ignore();
		break;
	default:
		QTextEdit::keyPressEvent(e);
	}
}
 
void ScriptConsole::onCursorPositionChanged()
{
	// only the last block (i.e. the current command) is editable
	setReadOnly(textCursor().block().next().isValid());
}
 
bool ScriptConsole::canInsertFromMimeData(const QMimeData *source) const
{
	if (source->hasText()) {
		QString text = source->text();
		if (text.contains("\n"))
			return false;
		else
			return true;
	} else if (source->hasUrls() && source->urls().length() == 1) {
		return true;
	} else
		return false;
}
 
void ScriptConsole::insertFromMimeData(const QMimeData *source)
{
	if (canInsertFromMimeData(source)) {
		if (source->hasText())
			QTextEdit::insertFromMimeData(source);
		else if (source->hasUrls() && source->urls().length() == 1) {
			QUrl url = source->urls()[0];
			QString text = url.toString();
			if (url.isLocalFile())
				text = url.toLocalFile();
			text = "\"" + text.replace("\\", "\\\\").replace("\"", "\\\"") + "\"";
			textCursor().insertText(text);
		}
	}
}
 
void ScriptConsole::onReturnKeyPress()
{
	int promptLength = m_prompt.length();
	QTextCursor cursor = textCursor();
	cursor.movePosition(QTextCursor::StartOfLine);
	cursor.movePosition(QTextCursor::Right, QTextCursor::MoveAnchor, promptLength);
	cursor.movePosition(QTextCursor::EndOfLine, QTextCursor::KeepAnchor);
	QString command = cursor.selectedText();
 
	QTextCharFormat fmt;
	fmt.setForeground(QColor(120, 120, 120));
	cursor.mergeCharFormat(fmt);
	cursor.clearSelection();
 
	if (command.trimmed() != "") {
		int k = m_commands.indexOf(command);
		if (k < 0)
			m_commands.append(command);
		else if (k <= m_commands.length() - 1) {
			m_commands.takeAt(k);
			m_commands.append(command);
		}
		m_commandIndex = m_commands.count();
		// m_currentCommand = m_prompt;
	}
 
	moveCursor(QTextCursor::EndOfLine);
 
	if (command.trimmed() != "") {
		append("");
		cursor.movePosition(QTextCursor::StartOfBlock);
		m_engine->evaluate(command);
	} else {
		append("");
		onEvaluationDone();
	}
}
 
void ScriptConsole::onEvaluationDone()
{
	moveCursor(QTextCursor::End);
	setTextColor(Qt::black);
	textCursor().insertText(m_prompt);
	moveCursor(QTextCursor::EndOfLine);
}
 
void ScriptConsole::output(int type, const QString &value)
{
	moveCursor(QTextCursor::End);
	if (type == ScriptEngine::ExecutionError || type == ScriptEngine::SyntaxError) {
		setTextColor(Qt::red);
	} else if (type == ScriptEngine::UndefinedEvaluationResult || type == ScriptEngine::Warning) {
		setTextColor(QColor(250, 120, 40));
	} else {
		setTextColor(QColor(10, 150, 240));
	}
	textCursor().insertText(value + "\n");
	moveCursor(QTextCursor::EndOfLine);
}
 
void ScriptConsole::executeCommand(const QString &cmd)
{
	moveCursor(QTextCursor::End);
	setTextColor(Qt::black);
	append(m_prompt);
	moveCursor(QTextCursor::EndOfLine);
	textCursor().insertText(cmd);
	moveCursor(QTextCursor::EndOfLine);
	onReturnKeyPress();
}