path: root/filewidget.cpp

#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 <QJsonObject>
#include <QJsonDocument>
#include <QItemSelectionModel>
#include <QGroupBox>
#include <QKeyEvent>
#include <QMessageBox>

#include "filewidget.h"
#include "filesorter.h"
#include "filedisplay.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()));
    QPushButton *selDirBtn = new QPushButton(tr("&Browse..."));
    connect(selDirBtn, SIGNAL(clicked()), this, SLOT(selectDir()));
    QGroupBox *dirGB = new QGroupBox(tr("Directory"));
    QHBoxLayout *dirGBL = new QHBoxLayout;
    dirGBL->addWidget(mDir);
    dirGBL->addWidget(selDirBtn);
    dirGB->setLayout(dirGBL);

    mFilter = new QLineEdit;
    connect(mFilter, SIGNAL(returnPressed()), this, SLOT(filter()));
    QPushButton *searchBtn = new QPushButton(tr("Filter"));
    connect(searchBtn, SIGNAL(clicked()), this, SLOT(filter()));
    QPushButton *clearBtn = new QPushButton(tr("&Clear"));
    connect(clearBtn, SIGNAL(clicked()), this, SLOT(clearFilter()));
    QGroupBox *filterGB = new QGroupBox(tr("Filter"));
    QHBoxLayout *filterGBL = new QHBoxLayout;
    filterGBL->addWidget(mFilter);
    filterGBL->addWidget(clearBtn);
    filterGBL->addWidget(searchBtn);
    filterGB->setLayout(filterGBL);

    mFileView = new QTreeView;
    mFileView->setSortingEnabled(true);
    mFileView->setSelectionBehavior(QAbstractItemView::SelectRows);
    mFileView->setSelectionMode(QAbstractItemView::ExtendedSelection);
    connect(mFileView, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(itemSelected(QModelIndex)));
    QGroupBox *filesGB = new QGroupBox(tr("Files"));
    QHBoxLayout *filesGBL = new QHBoxLayout;
    filesGBL->addWidget(mFileView);
    filesGB->setLayout(filesGBL);

    mModel = new QStandardItemModel;
    mProxy = new VideoSorter;
    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");

    mFileDisplay = new FileDisplay(this);

    QVBoxLayout *mainLayout = new QVBoxLayout;
    mainLayout->addWidget(dirGB);
    mainLayout->addWidget(filterGB);
    mainLayout->addWidget(filesGB);
    setLayout(mainLayout);

    connect(mFileView->selectionModel(), SIGNAL(selectionChanged(QItemSelection,QItemSelection)), this, SLOT(fileSelectionChanged(QItemSelection,QItemSelection)));
    mFileView->setFocus();
}

void FileWidget::fileData(const QString &fullPath, const QString &md5){
    mFileDisplay->setFileData(fullPath, md5);
    mFileDisplay->show();
}

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

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

    QMimeDatabase db;
    qApp->setOverrideCursor(Qt::BusyCursor);
    emit statusMessage(QString(tr("Gathering data...")));
    qApp->processEvents();

    foreach(QFileInfo fi, fl){
        QMimeType mime = db.mimeTypeForFile(fi);
        QList<QStandardItem*> fData;
        FileAttrs attr = NotPresent;
        Helper::Duration dur;
        for(int i = 0; i < ColumnCount; ++i){
            QStandardItem *item = new QStandardItem;
            item->setEditable(false);
            if(i == Md5Column){
                item->setFont(QFont("courier new"));
            }
            fData << item;
        }
        fData[NameColumn]->setText(fi.fileName());
        QString md5(tr("n/a"));
        QString duration(tr("n/a"));
        qint64 seconds = 0;
        if(mime.name().startsWith("video")){
            md5 = Helper::md5Sum(fi.absoluteFilePath());
            fData[Md5Column]->setText(md5);
            int haveFile = md5Count(md5);
            if(haveFile){
                fData[IconColumn]->setIcon(QIcon(":/huge_bra.png"));
                fData[NameColumn]->setForeground(greenBrush);
                attr = Present;
            }else{
                fData[IconColumn]->setIcon(QIcon(":/chastity_belt.png"));
                fData[NameColumn]->setForeground(redBrush);
                attr = NotPresent;
            }
            QJsonDocument jDoc = Helper::ffpmegData(fi.absoluteFilePath());
            QJsonObject jObj = jDoc.object();
            QJsonValue durationV = jObj["format"].toObject()["duration"];
            seconds = durationV.toVariant().toDouble();
            dur = Helper::Duration(seconds);
            duration = dur.toString();
        }else if(fi.isDir()){
            if(fi.fileName() == ".."){
                fData[IconColumn]->setIcon(QIcon(":/up_dick.png"));
                attr = ParentDir;
            }else{
                fData[IconColumn]->setIcon(QIcon(":/folder.png"));
                attr = Directory;
            }
        }else{
            fData[IconColumn]->setIcon(QIcon(":/gaping_ass.png"));
            attr = NoVideo;
        }
        fData[DurationColumn]->setText(duration);
        fData[DurationColumn]->setData(seconds, SecondsRole);
        fData[Md5Column]->setText(md5);
        fData[MimeColumn]->setText(mime.name());
        if(mime.name() == "inode/directory"){
            fData[MimeColumn]->setText("directory");
        }
        for(int i = 0; i < ColumnCount; ++i){
            fData[i]->setData(attr, AttrsRole);
            fData[i]->setData(fi.absoluteFilePath(), FullPathRole);
            fData[i]->setData(md5, MD5SumRole);
        }
        root->appendRow(fData);
    }
    readHeaderData();
    qApp->restoreOverrideCursor();
    fileSelectionChanged(QItemSelection(), QItemSelection());
    emit statusMessage(QString(tr("Gathering data... DONE!")));
}

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);
        mQFiles.exec();
        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::filter(){
    mProxy->setFilterRegExp(mFilter->text());
    mProxy->setFilterKeyColumn(NameColumn);
    mProxy->setFilterCaseSensitivity(Qt::CaseInsensitive);
}

void FileWidget::clearFilter(){
    mFilter->clear();
    filter();
}

void FileWidget::fileSelectionChanged(const QItemSelection &selected, const QItemSelection &deselected){
    Q_UNUSED(selected);
    Q_UNUSED(deselected);
    int count = mFileView->selectionModel()->selectedRows().count();
    int total = mModel->rowCount();
    QString msg = QString("%1/%2").arg(QString::number(count)).arg(QString::number(total));
    emit selectionCountChanged(msg);
    QModelIndexList idxl = mFileView->selectionModel()->selectedRows(DurationColumn);
    qint64 seconds = 0;
    foreach(QModelIndex idx, idxl){
        seconds += idx.data(SecondsRole).toLongLong();
    }
    Helper::Duration dur(seconds);
    emit durationChanged(dur.toString());
}

void FileWidget::itemSelected(const QModelIndex &idx){
    QModelIndex name = idx;
    QString mime = idx.sibling(idx.row(), MimeColumn).data().toString();
    if(name.column() != NameColumn){
        name = idx.sibling(idx.row(), NameColumn);
    }
    int attr = name.data(AttrsRole).toInt();
    if(attr == ParentDir || attr == Directory){
        cd(name.data().toString());
        return;
    }
    if(mime.startsWith("video")){
        fileData(name.data(FullPathRole).toString(), name.data(MD5SumRole).toString());
    }
}

void FileWidget::cd(const QString &to){
    QDir cur(mDir->text());
    cur.cd(to);
    QString newDir = cur.absolutePath();
    mDir->setText(QDir::toNativeSeparators(newDir));
    gatherData();
}

void FileWidget::deleteFiles(){
    QModelIndexList selFiles = mFileView->selectionModel()->selectedRows(NameColumn);
    const QString msg = QString(tr("Really delete %1 file(s)?")).arg(QString::number(selFiles.count()));
    int retval = QMessageBox::question(this, tr("Delete files..."), msg);
    if(retval == QMessageBox::Yes){
        foreach(QModelIndex idx, selFiles){
            QString fp = idx.data(FullPathRole).toString();
            QFile::remove(fp);
        }
        gatherData();
    }
}

void FileWidget::keyPressEvent(QKeyEvent *e){
    if(e->key() == Qt::Key_S && (e->modifiers() & Qt::ControlModifier)){
        mFilter->setFocus();
        mFilter->selectAll();
        goto exit;
    }
    if(e->key() == Qt::Key_D && (e->modifiers() & Qt::ControlModifier)){
        mDir->setFocus();
        mDir->selectAll();
        goto exit;
    }
    if(e->key() == Qt::Key_F && (e->modifiers() & Qt::ControlModifier)){
        mFileView->setFocus();
        goto exit;
    }
    if(mFileView->hasFocus()){
        if(e->key() == Qt::Key_Return){
            QModelIndex idx = mFileView->selectionModel()->selectedRows(NameColumn).at(0);
            itemSelected(idx);
            goto exit;
        }
        if(e->key() == Qt::Key_Delete){
            deleteFiles();
            goto exit;
        }
    }
    QWidget::keyPressEvent(e);

    exit:
    e->accept();
}