/*
  This program is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License
  as published by the Free Software Foundation; either version
  2 of the License, or (at your option) any later version.
*/

#include <QSortFilterProxyModel>
#include <QComboBox>
#include <QPushButton>
#include <QLineEdit>
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QInputDialog>
#include <QStringListModel>
#include <QMessageBox>
#include <QSettings>
#include <QHideEvent>
#include <QAction>
#include <QMenu>

#include <algorithm>

#include "mappingtreewidget.h"
#include "mappingtreemodel.h"
#include "smglobals.h"

MappingTreeWidget::MappingTreeWidget(QWidget *parent) : QWidget(parent){
    //setup model
    mModel = static_cast<MappingTreeModel*>(SmGlobals::instance()->model("MappingTree"));
    mProxy = new QSortFilterProxyModel(this);
    mProxy->setSourceModel(mModel);
    mTypesModel = new QStringListModel(this);
    mTypesModel->setStringList(mModel->mappingTypeNames());

    //setup gui
    mTree = new MappingTreeView;
    mTree->setModel(mProxy);
    mTree->setColumnHidden(1, true);
    mTree->setColumnHidden(2, true);
    mTree->setAlternatingRowColors(true);
    mTree->expandAll();
    mTypeBox = new QComboBox;
    mTypeBox->setModel(mTypesModel);
    connect(mTypeBox, SIGNAL(currentIndexChanged(QString)), this, SLOT(typeChanged(QString)));
    mAddType = new QPushButton(tr("Add &type"));
    connect(mAddType, SIGNAL(clicked()), this, SLOT(addType()));
    mDeleteType = new QPushButton(tr("Delete type"));
    connect(mDeleteType, SIGNAL(clicked()), this, SLOT(deleteType()));
    mEdit = new QLineEdit;
    mAddChild = new QPushButton(tr("&Add"));
    connect(mAddChild, SIGNAL(clicked()), this, SLOT(addChild()));
    mDeleteChild = new QPushButton(tr("Delete"));
    connect(mDeleteChild, SIGNAL(clicked()), this, SLOT(deleteChild()));
    QHBoxLayout *typesButtonLayout = new QHBoxLayout;
    typesButtonLayout->addWidget(mDeleteType);
    typesButtonLayout->addWidget(mAddType);
    QHBoxLayout *buttonLayout = new QHBoxLayout;
    buttonLayout->addStretch();
    buttonLayout->addWidget(mDeleteChild);
    buttonLayout->addWidget(mAddChild);

    //select type and populate model
    QSettings s;
    QString lastType = s.value("ui/mappingtreetype").toString();
    if(lastType.isEmpty()){
        if(!mTypesModel->stringList().isEmpty()){
            lastType = mTypesModel->stringList().at(0);
        }
    }
    int typeId = mModel->mappingTypeIdFromName(lastType);
    if(typeId != -1){
        mModel->setType(typeId);
        mTypeBox->setCurrentIndex(mTypeBox->findText(lastType));
        typeChanged(lastType);
    }

    //setup actions
    mDeleteChildA = new QAction(tr("Delete"), this);
    connect(mDeleteChildA, SIGNAL(triggered()), this, SLOT(deleteChild()));
    mTree->addAction(mDeleteChildA);
    mEditChildA = new QAction(tr("Edit..."), this);
    connect(mEditChildA, SIGNAL(triggered()), this, SLOT(editChild()));
    mTree->addAction(mEditChildA);

    //widget layout and tab order
    QVBoxLayout *mainLayout = new QVBoxLayout;
    mainLayout->addWidget(mTypeBox);
    mainLayout->addLayout(typesButtonLayout);
    mainLayout->addWidget(mTree);
    mainLayout->addWidget(mEdit);
    mainLayout->addLayout(buttonLayout);
    setLayout(mainLayout);
    mEdit->setFocus();
    setTabOrder(mEdit, mAddChild);
}

MappingData MappingTreeWidget::selectedItem() const {
    MappingData retval;
    QModelIndex sel = selected();
    if(!sel.isValid()){
        retval.id = -1;
    }else{
        retval.id = sel.data(MappingTreeModel::IdRole).toInt();
        retval.name = sel.data(MappingTreeModel::NameRole).toString();
        QStringList p;
        QModelIndex parent = sel;
        while(parent.isValid()){
            p << parent.data().toString();
            parent = parent.parent();
        }
        std::reverse(p.begin(), p.end());
        retval.path = p;
    }
    return retval;
}

void MappingTreeWidget::addChild(){
    QModelIndex sel = selected();
    if(!sel.isValid()){
        QMessageBox::critical(this, tr("Error"), tr("No parent item selected!"));
        return;
    }
    QString value = mEdit->text().toLower().trimmed();
    if(value.isEmpty()){
        return;
    }
    QModelIndex real = mProxy->mapToSource(sel);
    mModel->addChild(value, real);
}

void MappingTreeWidget::addType(){
    QString typeName = QInputDialog::getText(this, tr("Enter type name"), tr("Name")).toLower().trimmed();
    if(typeName.isEmpty()){
        return;
    }
    if(mModel->addMappingType(typeName)){
        mTypesModel->setStringList(mModel->mappingTypeNames());
        int idxOf = mTypeBox->findText(typeName);
        if(idxOf != -1){
            mTypeBox->setCurrentIndex(idxOf);
        }
    }
}

void MappingTreeWidget::deleteChild(){
    QModelIndex sel = selected();
    if(!sel.isValid()){
        QMessageBox::critical(this, tr("Error"), tr("No item selected!"));
        return;
    }
    QModelIndex real = mProxy->mapToSource(sel);
    if(mModel->rowCount(real) > 0){
        QString msg = QString(tr("Cannot delete item %1, because it has %2 children!")).arg(real.data().toString()).arg(QString::number(mModel->rowCount(real)));
        QMessageBox::critical(this, tr("Error"), msg);
        return;
    }
    mModel->deleteChild(real);
}

void MappingTreeWidget::deleteType(){
    QString curText = mTypeBox->currentText();
    if(curText.isEmpty()){
        return;
    }
    QString msg = QString(tr("Really delete %1?")).arg(curText);
    int retval = QMessageBox::question(this, tr("Question"), msg, QMessageBox::Yes | QMessageBox::No, QMessageBox::No);
    if(retval == QMessageBox::Yes){
        int typeId = mModel->mappingTypeIdFromName(curText);
        if(typeId != -1){
            if(mModel->deleteMappingType(typeId)){
                mTypesModel->setStringList(mModel->mappingTypeNames());
                if(mTypesModel->stringList().count() > 0){
                    mTypeBox->setCurrentIndex(0);
                }
            }
        }
    }
}

void MappingTreeWidget::hideEvent(QHideEvent *event){
    QString type = mTypeBox->currentText();
    if(!type.isEmpty()){
        QSettings s;
        s.setValue("ui/mappingtreetype", type);
    }
    event->accept();
}

void MappingTreeWidget::typeChanged(QString type){
    int typeId = mModel->mappingTypeIdFromName(type);
    if(typeId != -1){
        mModel->setType(typeId);
        mModel->populate();
        mTree->expandAll();
    }
}

void MappingTreeWidget::editChild(){
    QModelIndex sel = selected();
    if(sel.isValid()){
        mTree->edit(sel);
    }
}

const QModelIndex MappingTreeWidget::selected() const{
    QModelIndexList sel = mTree->selectionModel()->selectedRows();
    if(sel.isEmpty()){
        return QModelIndex();
    }
    return sel.first();
}


MappingTreeView::MappingTreeView(QWidget *parent) : QTreeView(parent) {}

void MappingTreeView::contextMenuEvent(QContextMenuEvent *e){
    QMenu ctxMenu(this);
    ctxMenu.addActions(actions());
    ctxMenu.exec(e->globalPos());
}