#include "mainwindow.h"
#include "ui_mainwindow.h"

#include <QDebug>
#include <QFileDialog>
#include <QMessageBox>

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    m_ui(new Ui::MainWindow),
    m_contentChanged(false)
{
    m_graphicsView = new GraphWidget(this);

    m_ui->setupUi(this);
    connect(m_ui->actionNew, SIGNAL(activated()),
            this, SLOT(newFile()));
    connect(m_ui->actionOpen, SIGNAL(activated()),
            this, SLOT(openFile()));
    connect(m_ui->actionSave, SIGNAL(activated()),
            this, SLOT(saveFile()));
    connect(m_ui->actionSaveAs, SIGNAL(activated()),
            this, SLOT(saveFileAs()));
    connect(m_ui->actionClose, SIGNAL(activated()),
            this, SLOT(closeFile()));
    connect(m_ui->actionExport, SIGNAL(activated()),
            this, SLOT(exportScene()));

    connect(m_ui->actionQuit, SIGNAL(activated()),
            QApplication::instance(), SLOT(quit()));

    connect(m_ui->actionAbout_QtMindMap, SIGNAL(activated()),
            this, SLOT(about()));
    connect(m_ui->actionKeys, SIGNAL(activated()),
            this, SLOT(keys()));


    setCentralWidget(m_graphicsView);
    m_graphicsView->hide();

    // why can't I do this with qtcreator?
    /// @bug or a feature? no underline here

    m_zoomIn = new QAction(tr("Zoom in (+)"), this);
    m_zoomOut = new QAction(tr("Zoom out (+)"), this);

    m_addNode = new QAction(tr("Add node (ins)"), this);
    connect(m_addNode, SIGNAL(activated()), m_graphicsView, SLOT(insertNode()));
    m_delNode = new QAction(tr("Del node (del)"), this);
    connect(m_delNode, SIGNAL(activated()), m_graphicsView, SLOT(removeNode()));

    m_editNode = new QAction(tr("Edit node (F2)"), this);
    m_scaleUpNode = new QAction(tr("ScaleUp Node (Ctrl +)"), this);
    m_scaleDownNode = new QAction(tr("ScaleDown Node (Ctrl -)"), this);
    m_nodeColor = new QAction(tr("Node color (c)"), this);
    m_nodeTextColor = new QAction(tr("Node textcolor (t)"), this);
    m_addEdge = new QAction(tr("Add edge (a)"), this);
    m_delEdge = new QAction(tr("Del edge (d)"), this);
    m_moveNode = new QAction(tr("Move node (Ctrl cursor)"), this);
    m_moveNode->setDisabled(true);
    m_subtree = new QAction(tr("Apply on subtree (Alt)"), this);
    m_subtree->setDisabled(true);

    m_hintMode = new QAction(tr("Hint mode (f)"), this);
    m_showMainToolbar = new QAction(tr("Main toolbar (Ctrl m)"), this);
    m_showMainToolbar->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_M));
    connect(m_showMainToolbar, SIGNAL(activated()), this, SLOT(showMainToolbar()));

    m_showStatusIconToolbar = new QAction(tr("Status icons (Ctrl i)"), this);

    m_ui->mainToolBar->addAction(m_addNode);
    m_ui->mainToolBar->addAction(m_delNode);
    m_ui->mainToolBar->addAction(m_editNode);
    m_ui->mainToolBar->addAction(m_scaleUpNode);
    m_ui->mainToolBar->addAction(m_scaleDownNode);
    m_ui->mainToolBar->addAction(m_nodeColor);
    m_ui->mainToolBar->addAction(m_nodeTextColor);
    m_ui->mainToolBar->addAction(m_addEdge);
    m_ui->mainToolBar->addAction(m_delEdge);

    m_ui->mainToolBar->addSeparator();
    m_ui->mainToolBar->addAction(m_zoomIn);
    m_ui->mainToolBar->addAction(m_zoomOut);
    m_ui->mainToolBar->addAction(m_hintMode);
    m_ui->mainToolBar->addAction(m_moveNode);
    m_ui->mainToolBar->addAction(m_subtree);
    m_ui->mainToolBar->addAction(m_showMainToolbar);
    m_ui->mainToolBar->addAction(m_showStatusIconToolbar);

    m_ui->mainToolBar->hide();



    m_doIt = new QAction(QIcon(":/applications-system.svg"), "&Do", this);
    m_doIt->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_D));
    connect(m_doIt, SIGNAL(activated()), this, SLOT(insertPicture()));

    m_trash = new QAction(QIcon(":/user-trash-full.svg"), "&Trash", this);
    m_trash->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_T));
    connect(m_trash, SIGNAL(activated()), this, SLOT(insertPicture()));

    m_info = new QAction(QIcon(":/mail-attachment.svg"), "&Refer", this);
    m_info->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_R));
    connect(m_info, SIGNAL(activated()), this, SLOT(insertPicture()));

    m_blocked = new QAction(QIcon(":/dialog-warning.svg"), tr("&Blocked"), this);
    m_blocked->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_B));
    connect(m_blocked, SIGNAL(activated()), this, SLOT(insertPicture()));

    m_question = new QAction(QIcon(":/help-browser.svg"), tr("&How?"), this);
    m_question->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_H));
    connect(m_question, SIGNAL(activated()), this, SLOT(insertPicture()));

    m_postpone = new QAction(QIcon(":/x-office-calendar.svg"), tr("&Postpone"), this);
    m_postpone->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_P));
    connect(m_postpone, SIGNAL(activated()), this, SLOT(insertPicture()));

    m_delegate = new QAction(QIcon(":/system-users.svg"), tr("&Comission"), this);
    m_delegate->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_C));
    connect(m_delegate, SIGNAL(activated()), this, SLOT(insertPicture()));

    m_maybe = new QAction(QIcon(":/dialog-information.svg"), tr("&Maybe"), this);
    m_maybe->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_M));
    connect(m_maybe, SIGNAL(activated()), this, SLOT(insertPicture()));

    m_ui->statusIcons_toolBar->addAction(m_doIt);
    m_ui->statusIcons_toolBar->addAction(m_trash);
    m_ui->statusIcons_toolBar->addAction(m_info);
    m_ui->statusIcons_toolBar->addAction(m_blocked);
    m_ui->statusIcons_toolBar->addAction(m_question);
    m_ui->statusIcons_toolBar->addAction(m_postpone);
    m_ui->statusIcons_toolBar->addAction(m_delegate);
    m_ui->statusIcons_toolBar->addAction(m_maybe);
    m_ui->statusIcons_toolBar->setToolButtonStyle(Qt::ToolButtonTextUnderIcon);
    m_ui->statusIcons_toolBar->hide();
}

MainWindow::~MainWindow()
{
    delete m_ui;
}

void MainWindow::statusBarMsg(const QString &msg)
{
    m_ui->statusBar->showMessage(msg, 5000);
}

void MainWindow::contentChanged(const bool& changed)
{
    if (m_contentChanged == false && changed == true)
    {
        setWindowTitle(windowTitle().prepend("* "));
        m_contentChanged = true;
        if (m_fileName != "untitled")
            m_ui->actionSave->setEnabled(true);
    }
    else if (m_contentChanged == true && changed == false)
    {
        setWindowTitle(windowTitle().remove(0,2));
        m_contentChanged = false;
        m_ui->actionSave->setEnabled(false);
    }
}

void MainWindow::newFile()
{
    if (!closeFile())
        return;

    m_graphicsView->newScene();

    m_ui->actionSave->setEnabled(false);
    m_ui->actionSaveAs->setEnabled(true);
    m_ui->actionClose->setEnabled(true);
    contentChanged(false);
    m_fileName = "untitled";
    setTitle(m_fileName);

    m_graphicsView->setFocus();

    showMainToolbar();
}

void MainWindow::openFile(const QString &fileName)
{
    if (!closeFile())
        return;

    if (fileName.isEmpty())
    {
        QFileDialog dialog(this,
                           tr("Open MindMap"),
                           QDir::homePath(),
                           QString("QtMindMap (*.qmm)"));
        dialog.setAcceptMode(QFileDialog::AcceptOpen);
        dialog.setDefaultSuffix("qmm");

        if (dialog.exec())
        {
            m_fileName = dialog.selectedFiles().first();
        }
        else
        {
            return;
        }
    }
    else
    {
        m_fileName = fileName;
    }

    m_graphicsView->readContentFromXmlFile(m_fileName);

    m_ui->actionSave->setEnabled(true);
    m_ui->actionSaveAs->setEnabled(true);
    m_ui->actionClose->setEnabled(true);
    contentChanged(false);
    setTitle(m_fileName);
    showMainToolbar();
}

void MainWindow::saveFile()
{
    m_graphicsView->writeContentToXmlFile(m_fileName);
    contentChanged(false);
}

bool MainWindow::saveFileAs()
{
    QFileDialog dialog(this,
                       tr("Save MindMap as"),
                       QDir::homePath(),
                       QString("QtMindMap (*.qmm)"));
    dialog.setAcceptMode(QFileDialog::AcceptSave);
    dialog.setDefaultSuffix("qmm");

    if (dialog.exec())
    {
        m_fileName = dialog.selectedFiles().first();
        setTitle(m_fileName);
        saveFile();
        return true;
    }
    else
    {
        return false; // cancelled
    }
}

bool MainWindow::closeFile()
{
    if (m_contentChanged)
    {
        QMessageBox msgBox(this);
        msgBox.setText("The document has been modified.");
        msgBox.setInformativeText("Do you want to save your changes?");
        msgBox.setStandardButtons(QMessageBox::Save |
                                  QMessageBox::Discard |
                                  QMessageBox::Cancel);
        msgBox.setDefaultButton(QMessageBox::Save);
        int ret = msgBox.exec();

        switch (ret) {
        case QMessageBox::Save:
        {
            if (m_fileName == "untitled")
            {
                if (!saveFileAs())
                        return false;
            }
            else
            {
                saveFile();
            }

            break;
        }
        case QMessageBox::Discard:
            break;
        case QMessageBox::Cancel:
            return false;
        default:
            break;
        }
    }

    m_ui->actionSave->setEnabled(false);
    m_ui->actionSaveAs->setEnabled(false);
    m_ui->actionClose->setEnabled(false);
    m_contentChanged = false;
    setTitle("");
    m_graphicsView->closeScene();
    showMainToolbar(false);
    return true;
}

void MainWindow::exportScene()
{
    QFileDialog dialog(this,
                       tr("Export MindMap to image"),
                       QDir::homePath(),
                       tr("PNG image file (*.png)"));
    dialog.setAcceptMode(QFileDialog::AcceptSave);
    dialog.setDefaultSuffix("png");

     if (dialog.exec())
     {
         m_graphicsView->writeContentToPngFile(dialog.selectedFiles().first());
     }
}

void MainWindow::keys()
{
    QMessageBox msgBox(this);
    msgBox.setWindowTitle(tr("Keys of QtMindMap"));
    msgBox.setText(tr("Usage of keys & mouse buttons"));
    msgBox.setInformativeText(

                QString("<table><tr><td><b>").

                append(tr("Keys")).
                append("</b></td><td></td></tr><tr><td>+,-</td><td>").
                append(tr("zoom in/out of the view")).
                append("</td></tr></tr><tr><td>").
                append(tr("cursors")).
                append("</td><td>").
                append(tr("move view scrollbars")).
                append("</td></tr></tr><tr><td>").
                append(tr("ctrl + cursors")).
                append("</td><td>").
                append(tr("move active node")).
                append("</td></tr></tr><tr><td>Delete</td><td>").
                append(tr("remove active node")).
                append("</td></tr></tr><tr><td>Insert</td><td>").
                append(tr("add new node to active node")).
                append("</td></tr></tr><tr><td>f</td><td>").
                append(tr("enter/leave hint mode (vimperator style select)")).
                append("</td></tr></tr><tr><td>").
                append(tr("numbers, backspace, enter")).
                append("</td><td>").
                append(tr("select numbers in hint mode")).
                append("</td></tr></tr><tr><td>F2</td><td>").
                append(tr("edit active node")).
                append("</td></tr></tr><tr><td>Esc</td><td>").
                append(tr("leaving node editing/adding/deleting mode")).
                append("</td></tr></tr><tr><td>").
                append(tr("editing mode: ctrl+cursors")).
                append("</td><td>").
                append(tr("jump to beginning of next/prev word")).
                append("</td></tr></tr><tr><td>a</td><td>").
                append(tr("add new edge to active node (select the destination)")).
                append("</td></tr></tr><tr><td>d</td><td>").
                append(tr("delete edge of active node (select other end)")).
                append("</td></tr><tr><td><b>").
                append(tr("Mouse")).
                append("</b></td><td></td></tr><tr><td>").
                append(tr("scroll")).
                append("</td><td>").
                append(tr("zoom in/out of the view")).
                append("</td></tr><tr><td>").
                append(tr("click")).
                append("</td><td>").
                append(tr("select node")).
                append("</td></tr><tr><td>").
                append(tr("double clikk")).
                append("</td><td>").
                append(tr("set node editable")).
                append("</td></tr><tr><td>").
                append(tr("drag")).
                append("</td><td>").
                append(tr("move node")).
                append("</td></tr></table>"));

    msgBox.exec();
}

void MainWindow::about()
{
    QMessageBox msgBox(this);
    msgBox.setWindowTitle(tr("About QtMindMap"));
    msgBox.setText(tr("MindMap software written in Qt."));
    msgBox.setTextFormat(Qt::RichText);
    msgBox.setInformativeText(tr("Homepage:").
                              append(" <a href=\"https://gitorious.org/qtmindmap\">https://gitorious.org/qtmindmap</a><br><br>").
                              append(tr("Report bugs to:")).
                              append(" <a href=\"mailto:denes.matetelki@gmail.com\">denes.matetelki@gmail.com</a>"));
    QPixmap pixMap(":/qtmindmap.svg");
    msgBox.setIconPixmap(pixMap.scaled(50,50));
    msgBox.exec();
}

void MainWindow::insertPicture()
{
    QAction *sender = dynamic_cast<QAction*>(QObject::sender());

    /// @note Why QIcon does not store it's fileName? It would be easier:
    //    m_graphicsView->insertPicture(
    //                dynamic_cast<QAction*>(QObject::sender())->icon().name());
    if (sender == m_doIt)
    {
        m_graphicsView->insertPicture(":/applications-system.svg");
    }
    else if (sender == m_trash)
    {
        m_graphicsView->insertPicture(":/user-trash-full.svg");
    }
    else if (sender == m_info)
    {
        m_graphicsView->insertPicture(":/mail-attachment.svg");
    }
    else if (sender == m_blocked)
    {
        m_graphicsView->insertPicture(":/dialog-warning.svg");
    }
    else if (sender == m_question)
    {
        m_graphicsView->insertPicture(":/help-browser.svg");
    }
    else if (sender == m_postpone)
    {
        m_graphicsView->insertPicture(":/x-office-calendar.svg");
    }
    else if (sender == m_delegate)
    {
        m_graphicsView->insertPicture(":/system-users.svg");
    }
    else if (sender == m_maybe)
    {
        m_graphicsView->insertPicture(":/dialog-information.svg");
    }
    else
    {
        return;
    }
}

void MainWindow::addNode() { }
void MainWindow::delNode() { }
void MainWindow::editNode() { }
void MainWindow::scaleUpNode() { }
void MainWindow::scaleDownNode() { }
void MainWindow::nodeColor() { }
void MainWindow::nodeTextColor() { }
void MainWindow::addEdge() { }
void MainWindow::delEdge() { }
void MainWindow::zoomIn() { }
void MainWindow::zoomOut() { }
void MainWindow::hintMode() { }

void MainWindow::showMainToolbar(const bool &show)
{
    m_ui->mainToolBar->setVisible(show ?
                                      !m_ui->mainToolBar->isVisible() :
                                      false);
}

void MainWindow::showStatusIconToolbar(const bool &show)
{
    m_ui->statusIcons_toolBar->setVisible(
                                show ?
                                     !m_ui->statusIcons_toolBar->isVisible() :
                                     false);
}

void MainWindow::closeEvent(QCloseEvent * event)
{
    m_contentChanged && !closeFile() ? event->ignore() : event->accept();
}

void MainWindow::keyPressEvent(QKeyEvent *event)
{
    if (event->modifiers() & Qt::ControlModifier)
    {
        if (event->key() == Qt::Key_M)
        {
            showMainToolbar();
            return;
        }
        if (event->key() == Qt::Key_I)
        {
            showStatusIconToolbar();
            return;
        }
    }

    QMainWindow::keyPressEvent(event);
}

void MainWindow::setTitle(const QString &title)
{
    if (title.isEmpty())
    {
        setWindowTitle("QtMindMap");
    }
    else
    {
        QString t(title);
        t.append(" - QtMindMap");
        setWindowTitle(t);
    }
}