#include "toonzqt/scriptconsole.h"

#include "toonz/scriptengine.h"

#include <qkeyevent></qkeyevent>

#include <qtextblock></qtextblock>

#include <qurl></qurl>

#include <qmimedata></qmimedata>

ScriptConsole::ScriptConsole(QWidget *parent)

    : QTextEdit(parent), m_commandIndex(0) {

  setObjectName("ScriptConsole");

  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();

}