#include "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_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, dubclick)"), this);
    connect(m_editNode, SIGNAL(activated()), m_graphicsView, SLOT(editNode()));

    /// @todo pass ctrl
    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);
    connect(m_nodeColor, SIGNAL(activated()), m_graphicsView, SLOT(nodeColor()));
    m_nodeTextColor = new QAction(tr("Node textcolor (t)"), this);
    connect(m_nodeTextColor, SIGNAL(activated()), m_graphicsView, SLOT(nodeTextColor()));
    m_addEdge = new QAction(tr("Add edge (a)"), this);
    connect(m_addEdge, SIGNAL(activated()), m_graphicsView, SLOT(addEdge()));
    m_delEdge = new QAction(tr("Del edge (d)"), this);
    connect(m_delEdge, SIGNAL(activated()), m_graphicsView, SLOT(removeEdge()));
    m_moveNode = new QAction(tr("Move node\n(Ctrl cursor, drag)"), this);
    m_moveNode->setDisabled(true);
    m_subtree = new QAction(tr("Change on wholesubtree\n(Ctrl shift)"), this);
    m_subtree->setDisabled(true);
    m_zoomIn = new QAction(tr("Zoom in (+, scrollup)"), this);
    connect(m_zoomIn, SIGNAL(activated()), m_graphicsView, SLOT(zoomIn()));
    m_zoomOut = new QAction(tr("Zoom out (-, scrolldown)"), this);
    connect(m_zoomOut, SIGNAL(activated()), m_graphicsView, SLOT(zoomOut()));
    m_esc = new QAction(tr("Leave editing,\nedge eadd/remove (esc)"), this);
    connect(m_esc, SIGNAL(activated()), m_graphicsView, SLOT(nodeLostFocus()));
    m_hintMode = new QAction(tr("Hint mode (f)"), this);
    connect(m_hintMode, SIGNAL(activated()), m_graphicsView, SLOT(hintMode()));
    m_showMainToolbar = new QAction(tr("Show main toolbar\n(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("Insert status icons\n(Ctrl i)"), this);
    connect(m_showStatusIconToolbar, SIGNAL(activated()), this, SLOT(showStatusIconToolbar()));

    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_esc);
    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_insertIcon = new QAction(tr("Insert icon:"), this);
    m_insertIcon->setDisabled(true);

    m_doIt = new QAction(QIcon(":/applications-system.svg"), tr("&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"), tr("&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"), tr("&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("ma&Ybe"), this);
    m_maybe->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_Y));
    connect(m_maybe, SIGNAL(activated()), this, SLOT(insertPicture()));

    m_ui->statusIcons_toolBar->addAction(m_insertIcon);
    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 != tr("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);
    m_ui->actionExport->setEnabled(true);
    contentChanged(false);
    m_fileName = tr("untitled");
    setTitle(m_fileName);

    m_graphicsView->setFocus();

    showMainToolbar();
}

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

    QString currFilename(m_fileName);
    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;
    }


    QFileInfo fileInfo(fileName);
    if (!fileInfo.isWritable())
        statusBarMsg(tr("Read-only file!"));

    if (!m_graphicsView->readContentFromXmlFile(m_fileName))
    {
        m_fileName = currFilename;
        return;
    }


    m_ui->actionSaveAs->setEnabled(true);
    m_ui->actionClose->setEnabled(true);
    m_ui->actionExport->setEnabled(true);

    contentChanged(false);

    if (!fileInfo.isWritable())
    {
        m_ui->actionSave->setEnabled(false);
        setTitle(QString("readonly ").append(m_fileName));
    }
    else
    {
        m_ui->actionSave->setEnabled(true);
        setTitle(m_fileName);
    }

    showMainToolbar();
}

void MainWindow::saveFile()
{
    QFileInfo fileInfo(m_fileName);
    if (!fileInfo.isWritable())
    {
        statusBarMsg(tr("Read-only file!"));
        return;
    }

    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(tr("The document has been modified."));
        msgBox.setInformativeText(tr("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 == tr("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_ui->actionExport->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::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::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)
{
    // incative action does not listen to signals
    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);
    }
}