#include <QLineEdit>
#include <QPushButton>
#include <QGridLayout>
#include <QVBoxLayout>
#include <QStandardItemModel>
#include <QStandardItem>
#include <QTreeView>
#include <QCompleter>
#include <QDirModel>
#include <QDir>
#include <QFileDialog>
#include <QLabel>
#include <QMimeDatabase>
#include <QMimeType>
#include <QFont>
#include <QSqlDatabase>
#include <QSqlQuery>
#include <QSettings>
#include <QHeaderView>
#include <QApplication>
#include <QCursor>

#include "filewidget.h"
#include "filesorter.h"
#include "helper.h"

FileWidget::FileWidget(QWidget *parent) : QWidget(parent) {
    setupGui();
    gatherData();
}

FileWidget::~FileWidget(){
    writeHeaderData();
    writeSettings();
}

void FileWidget::setupGui(){
    mDir = new QLineEdit;
    QCompleter *fsCompleter = new QCompleter(this);
    QDirModel *dirModel = new QDirModel(this);
    fsCompleter->setModel(dirModel);
    fsCompleter->setCompletionMode(QCompleter::PopupCompletion);
    mDir->setCompleter(fsCompleter);
    connect(mDir, SIGNAL(returnPressed()), this, SLOT(gatherData()));
    mSelDir = new QPushButton(tr("Browse..."));
    connect(mSelDir, SIGNAL(clicked()), this, SLOT(selectDir()));
    QLabel *dirL = new QLabel(tr("&Directory"));
    dirL->setBuddy(mDir);

    mSearchFile = new QLineEdit;
    connect(mSearchFile, SIGNAL(returnPressed()), this, SLOT(searchFile()));
    mSearch = new QPushButton(tr("Search"));
    connect(mSearch, SIGNAL(clicked()), this, SLOT(searchFile()));
    QLabel *searchL = new QLabel(tr("&Search"));
    searchL->setBuddy(mSearchFile);

    QGridLayout *topL = new QGridLayout;
    topL->addWidget(dirL, 0, 0);
    topL->addWidget(mDir, 0, 1);
    topL->addWidget(mSelDir, 0, 2);
    topL->addWidget(searchL, 1, 0);
    topL->addWidget(mSearchFile, 1, 1);
    topL->addWidget(mSearch, 1, 2);

    mFileView = new QTreeView;
    mFileView->setSortingEnabled(true);
    mFileView->setSelectionBehavior(QAbstractItemView::SelectRows);
    mFileView->setSelectionMode(QAbstractItemView::ExtendedSelection);

    mModel = new QStandardItemModel;
    mProxy = new FileSorter;
    mProxy->setSourceModel(mModel);
    mFileView->setModel(mProxy);
    readSettings();

    QSqlDatabase mDb = QSqlDatabase::database("shemovdb");
    if(!mDb.isOpen()){
        mDb.open();
    }
    mQOrigin = QSqlQuery(mDb);
    mQOrigin.prepare("SELECT COUNT(*) FROM files_origin WHERE cmd5sum = :md5");
    mQFiles = QSqlQuery(mDb);
    mQFiles.prepare("SELECT COUNT(*) FROM files WHERE cmd5sum = :md5");

    QVBoxLayout *mainLayout = new QVBoxLayout;
    mainLayout->addLayout(topL);
    mainLayout->addWidget(mFileView);
    setLayout(mainLayout);
}

void FileWidget::selectDir(){
    QString dir = QFileDialog::getExistingDirectory(this, tr("Select directory"), QDir::homePath());
    mDir->setText(QDir::toNativeSeparators(dir));
    gatherData();
}

void FileWidget::gatherData(){
    QDir d(mDir->text());
    QFileInfoList fl = d.entryInfoList(QStringList() << "*", QDir::Files, QDir::Name);
    mModel->clear();
    QStandardItem *root = mModel->invisibleRootItem();
    mModel->setHorizontalHeaderLabels(QStringList() << QChar(0x26A7) << tr("Name") << tr("MIME") << tr("MD5"));
    QBrush redBrush(Qt::red);
    QBrush greenBrush(Qt::darkGreen);

    QMimeDatabase db;
    qApp->setOverrideCursor(Qt::BusyCursor);

    foreach(QFileInfo fi, fl){
        QMimeType mime = db.mimeTypeForFile(fi);
        QStandardItem *i1;
        QStandardItem *i2 = new QStandardItem(fi.fileName());
        i2->setEditable(false);
        QStandardItem *i4;
        if(mime.name().startsWith("video")){
            QString md5 = Helper::md5Sum(fi.absoluteFilePath());
            i4 = new QStandardItem(md5);
            i4->setFont(QFont("courier new"));
            int haveFile = md5Count(md5);
            if(haveFile){
                i1 = new QStandardItem(QIcon(":/huge_bra.png"), QString());
                i1->setData(Present);
                i2->setForeground(greenBrush);
            }else{
                i1 = new QStandardItem(QIcon(":/chastity_belt.png"), QString());
                i1->setData(NotPresent);
                i2->setForeground(redBrush);
            }
        }else{
            i1 = new QStandardItem(QIcon(":/gaping_ass.png"), QString());
            i1->setData(NoVideo);
            i4 = new QStandardItem(tr("n/a"));
            i4->setFont(QFont("courier new"));
        }
        i1->setEditable(false);
        QStandardItem *i3 = new QStandardItem(mime.name());
        i3->setEditable(false);
        root->appendRow(QList<QStandardItem*>() << i1 << i2 << i3 << i4);
    }
    readHeaderData();
    qApp->restoreOverrideCursor();
}

int FileWidget::md5Count(const QString &md5) {
    mQOrigin.bindValue(":md5", md5);
    mQOrigin.exec();
    int retval = 0;
    while(mQOrigin.next()){
        retval = mQOrigin.value(0).toInt();
    }
    if(!retval){
        mQFiles.bindValue(":md5", md5);
        while(mQFiles.next()){
            retval = mQFiles.value(0).toInt();
        }
    }
    return retval;
}

void FileWidget::readHeaderData(){
    QSettings s;
    QByteArray headers = s.value("video/fileheaders").toByteArray();
    if(!headers.isEmpty()){
        mFileView->header()->restoreState(headers);
    }
}

void FileWidget::writeHeaderData(){
    QSettings s;
    s.setValue("video/fileheaders", mFileView->header()->saveState());
}

void FileWidget::readSettings(){
    QSettings s;
    QString dir = s.value("video/searchdir", QDir::toNativeSeparators(QDir::homePath())).toString();
    mDir->setText(dir);
}

void FileWidget::writeSettings(){
    QSettings s;
    s.setValue("video/searchdir", mDir->text());
}

void FileWidget::searchFile(){

}