/*
  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 <QLabel>
#include <QCheckBox>
#include <QLineEdit>
#include <QPushButton>
#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QGroupBox>
#include <QSortFilterProxyModel>
#include <QSqlDatabase>
#include <QSqlQuery>

#include "smtreeview.h"
#include "smtreeitem.h"
#include "smtreemodel.h"
#include "searchdialog.h"

SearchDialog::SearchDialog(QWidget *parent, Qt::WindowFlags flags) : QDialog(parent, flags){
    // Define GUI items
    setWindowTitle(tr("Search for"));
    mExFilenames = new QCheckBox(tr("Filenames"));
    mExFilenames->setChecked(true);
    mExMeta = new QCheckBox(tr("Metadata"));
    mExMeta->setChecked(true);
    mSearch = new QLineEdit;
    mResult = new SmTreeView;
    mDoSearch = new QPushButton(tr("Search"));
    mClose = new QPushButton(tr("Close"));

    // init Model and View
    const QStringList headers = QStringList() << tr("Match") << tr("Series") << tr("SeriesPartId");
    mModel = new SmTreeModel(headers, this);
    mProxy = new QSortFilterProxyModel(this);
    mProxy->setSourceModel(mModel);
    mResult->setModel(mProxy);
    mResult->setColumnHidden(2, true);
    mResult->setSortingEnabled(true);
    mResult->setEditTriggers(QAbstractItemView::NoEditTriggers);
    connect(mResult, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(searchDoubleclicked()));

    // Layout
    QVBoxLayout *mainLayout = new QVBoxLayout;
    QGroupBox *searchGb = new QGroupBox(tr("Search for:"));
    QVBoxLayout *searchGbL = new QVBoxLayout;
    searchGbL->addWidget(mSearch);
    searchGb->setLayout(searchGbL);
    mainLayout->addWidget(searchGb);
    QGroupBox *excludeGb = new QGroupBox(tr("Include:"));
    QHBoxLayout *excludeGbL = new QHBoxLayout;
    excludeGbL->addWidget(mExFilenames);
    excludeGbL->addStretch();
    excludeGbL->addWidget(mExMeta);
    excludeGb->setLayout(excludeGbL);
    mainLayout->addWidget(excludeGb);
    QGroupBox *resultGb = new QGroupBox(tr("Search result:"));
    QHBoxLayout *resultGbL = new QHBoxLayout;
    resultGbL->addWidget(mResult);
    resultGb->setLayout(resultGbL);
    mainLayout->addWidget(resultGb);
    QHBoxLayout *buttonLayout = new QHBoxLayout;
    mDoSearch = new QPushButton(tr("Search"));
    mClose = new QPushButton(tr("Close"));
    buttonLayout->addStretch();
    buttonLayout->addWidget(mDoSearch);
    buttonLayout->addWidget(mClose);
    mainLayout->addLayout(buttonLayout);
    setLayout(mainLayout);
    setMinimumWidth(800);

    // connect the dots...
    connect(mClose, SIGNAL(clicked()), this, SLOT(hide()));
    connect(mDoSearch, SIGNAL(clicked()), this, SLOT(search()));
    connect(mExFilenames, SIGNAL(clicked()), this, SLOT(disableSearch()));
    connect(mExMeta, SIGNAL(clicked()), this, SLOT(disableSearch()));

}

void SearchDialog::show(){
    mSearch->setFocus();
    mSearch->selectAll();
    QDialog::show();
}

void SearchDialog::search(){
    if(mSearch->text().isEmpty()){
        return;
    }
    QSqlDatabase db = QSqlDatabase::database("treedb");
    SmTreeItem *root = new SmTreeItem(3, 0);
    if(mExMeta->isChecked()){
        QSqlQuery metadataQ(db);
        metadataQ.prepare("SELECT iseriespart_id, tsubject, series.tseries_name, seriesparts.tsubtitle FROM metadata, seriesparts, series WHERE tsubject ~* :re AND metadata.iseriespart_id = seriesparts.iseriesparts_id AND seriesparts.iseries_id = series.iseries_id");
        metadataQ.bindValue(":re", mSearch->text());
        SmTreeItem *metadataItem = new SmTreeItem(QVariantList() << tr("Metadata") << QVariant() << QVariant(), root);
        root->appendChild(metadataItem);
        int ctr = 0;
        metadataQ.exec();
        while(metadataQ.next()){
            ++ctr;
            appendChild(metadataQ.value(0), metadataQ.value(1), metadataQ.value(2), metadataQ.value(3), metadataItem);
        }
        if(ctr == 0){
            appendEmpty(metadataItem);
        }
    }

    if(mExFilenames->isChecked()){
        QSqlQuery filenameQ(db);
        filenameQ.prepare("SELECT tfilename, iseriespart_id, series.tseries_name, seriesparts.tsubtitle FROM files, seriesparts, series WHERE tfilename ~* :re AND files.iseriespart_id =  seriesparts.iseriesparts_id AND seriesparts.iseries_id = series.iseries_id");
        filenameQ.bindValue(":re", mSearch->text());
        SmTreeItem *filenameItem = new SmTreeItem(QVariantList() << tr("Filenames") << QVariant() << QVariant(), root);
        root->appendChild(filenameItem);
        int ctr = 0;
        filenameQ.exec();
        while(filenameQ.next()){
            ++ctr;
            appendChild(filenameQ.value(1), filenameQ.value(0), filenameQ.value(2), filenameQ.value(3), filenameItem);
        }
        if(ctr == 0){
            appendEmpty(filenameItem);
        }
    }
    mModel->setRoot(root);
    mResult->expandAll();
}

void SearchDialog::disableSearch(){
    bool exFnChecked = mExFilenames->isChecked();
    bool exMetaChecked = mExMeta->isChecked();
    if(!exFnChecked && !exMetaChecked){
        mDoSearch->setEnabled(false);
    }else{
        mDoSearch->setEnabled(true);
    }
}

void SearchDialog::appendChild(QVariant id, QVariant subject, QVariant name, QVariant sub, SmTreeItem *parent){
    QString match;
    if(!sub.toString().isEmpty()){
        match = QString("%1 - %2").arg(name.toString()).arg(sub.toString());
    }else{
        match = name.toString();
    }
    QVariantList itemData = QVariantList() << subject << match << id;
    SmTreeItem *retval = new SmTreeItem(itemData, parent);
    parent->appendChild(retval);
}

void SearchDialog::appendEmpty(SmTreeItem *parent){
    SmTreeItem *emptyItem = new SmTreeItem(QVariantList() << tr("no match!") << QVariant() << QVariant(), parent);
    parent->appendChild(emptyItem);
}

void SearchDialog::searchDoubleclicked(){
    QPersistentModelIndex pIdx = mResult->currentIndex();
    QModelIndex idx = mProxy->mapToSource(pIdx);
    SmTreeItem *curItem = static_cast<SmTreeItem*>(idx.internalPointer());
    emit searchResultClicked(curItem->data(2).toInt());
}