path: root/fswidget.cpp

#include <QToolBar>
#include <QLabel>
#include <QComboBox>
#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QTreeView>
#include <QStandardItemModel>
#include <QSortFilterProxyModel>
#include <QFileDialog>
#include <QInputDialog>
#include <QMimeDatabase>
#include <QDirIterator>
#include <QJsonObject>
#include <QSqlDatabase>
#include <QElapsedTimer>
#include <QSettings>
#include <QApplication>

#include "fswidget.h"
#include "helper.h"

FSWidget::FSWidget(QWidget *parent) : QWidget(parent) {
    setPalette(qApp->palette());
    setupWidget();
}

void FSWidget::setupWidget(){
    QToolBar *toolbar = new QToolBar;
    QPixmap buttplug(":/butt_plug.png");
    QMatrix rotatematrix;
    rotatematrix.rotate(90);
    QIcon buttplugRight(buttplug.transformed(rotatematrix));
    rotatematrix.rotate(-180);
    QIcon buttplugLeft(buttplug.transformed(rotatematrix));
    QAction *backA = new QAction(buttplugLeft, tr("Prev. dir"), this);
    connect(backA, &QAction::triggered, [=] { advanceDir(-1); });
    toolbar->addAction(backA);
    QAction *forwardA = new QAction(buttplugRight, tr("Next dir"), this);
    connect(forwardA, &QAction::triggered, [=] { advanceDir(1); });
    toolbar->addAction(forwardA);

    QIcon plusIcon = Helper::icon(QColor(255,85,255), '+');
    QIcon minusIcon = Helper::icon(QColor(255,85,255), '-');
    QLabel *dirL = new QLabel(tr("Dir"));
    mDirCB = new QComboBox;
    mDirCB->setSizeAdjustPolicy(QComboBox::AdjustToContents);
    connect(mDirCB, &QComboBox::currentTextChanged, this, &FSWidget::gatherData);
    QAction *addDirA = new QAction(plusIcon, tr("Add dir..."), this);
    connect(addDirA, &QAction::triggered, [=] {
        QString newDir = QFileDialog::getExistingDirectory(this, tr("Select directory"), QDir::homePath());
        insertItem(mDirCB, newDir);
    } );
    QAction *removeDirA = new QAction(minusIcon, tr("Remove dir."), this);
    connect(removeDirA, &QAction::triggered, [=] { removeItem(mDirCB);} );
    QToolBar *dirTB = new QToolBar;
    dirTB->addAction(addDirA);
    dirTB->addAction(removeDirA);

    QLabel *filterL = new QLabel(tr("Filter"));
    mFilterCB = new QComboBox;
    mFilterCB->setInsertPolicy(QComboBox::InsertAlphabetically);
    mFilterCB->setSizeAdjustPolicy(QComboBox::AdjustToContents);

    QAction *addFilterA = new QAction(plusIcon, tr("Add filter..."), this);
    connect(addFilterA, &QAction::triggered, [=] {
        QString filter = QInputDialog::getText(this, tr("Mime filter"), tr("Filter"));
        insertItem(mFilterCB, filter);
    } );
    QAction *removeFilterA = new QAction(minusIcon, tr("Remove filter"), this);
    connect(removeFilterA, &QAction::triggered, [=] { removeItem(mFilterCB); });
    QToolBar *filterTB = new QToolBar;
    filterTB->addAction(addFilterA);
    filterTB->addAction(removeFilterA);

    QHBoxLayout *topWL = new QHBoxLayout;
    topWL->addWidget(dirL);
    topWL->addWidget(mDirCB);
    topWL->addWidget(dirTB);
    topWL->addStretch();
    topWL->addWidget(toolbar);
    topWL->addStretch();
    topWL->addWidget(filterL);
    topWL->addWidget(mFilterCB);
    topWL->addWidget(filterTB);

    mFileView = new QTreeView;
    mFileView->setAlternatingRowColors(true);
    mFileView->setSortingEnabled(true);
    mFileView->setUniformRowHeights(true);
    mFileView->setSelectionBehavior(QAbstractItemView::SelectRows);
    mFileView->setSelectionMode(QAbstractItemView::ExtendedSelection);
    mModel = new QStandardItemModel;
    mProxy = new QSortFilterProxyModel;
    mProxy->setSourceModel(mModel);
    mFileView->setModel(mProxy);

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

    readSettings();
}

FSWidget::~FSWidget(){
    writeSettings();
}

void FSWidget::readSettings(){
    QSettings s;
    QStringList dirs = s.value("fs/dirs").toStringList();
    mDirCB->addItems(dirs);
    QString cDir = s.value("fs/curdir").toString();
    mDirCB->setCurrentText(cDir);
    QStringList filters = s.value("fs/filters").toStringList();
    mFilterCB->addItems(filters);
    QString cFilter = s.value("fs/curfilter").toString();
    mFilterCB->setCurrentText(cFilter);
}

void FSWidget::writeSettings(){
    QSettings s;
    QString curDir = mDirCB->currentText();
    QStringList dirs;
    for(int i = 0 ; i < mDirCB->count(); ++i){
        dirs << mDirCB->itemText(i);
    }
    s.setValue("fs/dirs", dirs);
    s.setValue("fs/curdir", curDir);
    QString curFilter = mFilterCB->currentText();
    QStringList filters;
    for(int i = 0; i < mFilterCB->count(); ++i){
        filters << mFilterCB->itemText(i);
    }
    s.setValue("fs/filters", filters);
    s.setValue("fs/curfilter", curFilter);
}

void FSWidget::insertItem(QComboBox *cb, const QString &text){
    if(text.isEmpty()){
        return;
    }
    int idx = 0;
    while(idx != cb->count()){
        QString curT = cb->itemText(idx);
        if(text == curT){
            return;
        }
        if(cb->itemText(idx) > text){
            break;
        }
        ++idx;
    }
    cb->insertItem(idx, text);
}

void FSWidget::removeItem(QComboBox *cb){
    int toRemove = cb->findText(cb->currentText());
    if(toRemove != -1){
        cb->removeItem(toRemove);
    }
}

void FSWidget::gatherData(const QString &curDir){
    //setup view
    qApp->setOverrideCursor(Qt::BusyCursor);
    mQueryCount = 0;
    int fileCount = 0;
    QElapsedTimer timer;
    timer.start();
    emit message(tr("Gathering data..."));
    mFileView->setSortingEnabled(false);
    mFileView->setRootIsDecorated(false);
    mModel->clear();
    QStandardItem *root = mModel->invisibleRootItem();
    QMimeDatabase mimedb;
    mModel->setHorizontalHeaderLabels(QStringList() << tr("Name") << tr("MIME") << tr("Size") << tr("Duration") << tr("MD5") << tr("Presence"));
    QIcon videoIcon = Helper::icon(QColor(255,85,255), 'M');
    QIcon imageIcon = Helper::icon(QColor(255,85,255), 'P');
    QIcon otherIcon = Helper::icon(QColor(255,85,255), 'O');
    QBrush blackBrush(Qt::black);
    QBrush greenBrush(Qt::darkGreen);
    QBrush blueBrush(Qt::darkBlue);
    QBrush currentBrush = blackBrush;

    //setup database
    QSqlDatabase db = QSqlDatabase::database("treedb");
    QSqlQuery filesMd5Q(db);
    filesMd5Q.prepare("SELECT COUNT(*) FROM files WHERE cmd5sum = :arg");
    QSqlQuery originMd5Q(db);
    originMd5Q.prepare("SELECT COUNT(*) FROM files_origin WHERE cmd5sum = :arg");
    QSqlQuery filesNameQ(db);
    filesNameQ.prepare("SELECT COUNT(*) FROM files WHERE tfilename = :arg");
    QSqlQuery originNameQ(db);
    originNameQ.prepare("SELECT COUNT(*) FROM files WHERE tfilename LIKE :arg");

    QDirIterator it(curDir, QDir::Files);
    mFileView->setUpdatesEnabled(false);
    while(it.hasNext()){
        QFileInfo fi = it.next();
        ++fileCount;
        QMimeType mimeType = mimedb.mimeTypeForFile(fi);
        QString mimeName = mimeType.name();
        qint64 seconds = 0;
        Helper::Duration dur;
        QString durStr(tr("n/a"));
        QLocale l;
        QString size = l.toString(fi.size());
        QString md5 = Helper::md5Sum(fi.absoluteFilePath());
        if(mimeName.startsWith("video")){
            QJsonDocument jDoc = Helper::streamData(fi.absoluteFilePath());
            QJsonObject jObj = jDoc.object();
            QJsonValue durationV = jObj["format"].toObject()["duration"];
            seconds = durationV.toVariant().toDouble();
            dur = Helper::Duration(seconds);
            durStr = dur.toString();
        }
        QList<QStandardItem*> items;
        QFont f("courier new");
        QString presenceStr = tr("None");
        QString likeArg = QString("%1%%").arg(fi.completeBaseName());

        //do this goto dance to execute as few queries as possible
        if(queryCount(filesMd5Q, md5) > 0){
            currentBrush = greenBrush;
            presenceStr = tr("Files: MD5");
            goto ci;
        }
        if(queryCount(originMd5Q, md5)){
            currentBrush = blueBrush;
            presenceStr = tr("Origin: MD5");
            goto ci;
        }
        if(queryCount(filesNameQ, fi.fileName())){
            currentBrush = greenBrush;
            presenceStr = tr("Files: Name");
            goto ci;
        }
        if(queryCount(originNameQ, likeArg)){
            currentBrush = blueBrush;
            presenceStr = tr("Origin: Name");
        }

        ci:
        for(int i = 0; i < 6; ++i){
            QStandardItem *item = new QStandardItem;
            item->setFont(f);
            item->setEditable(false);
            item->setForeground(currentBrush);
            items << item;
        }
        items[0]->setText(fi.fileName());
        if(mimeName.startsWith("video")){
            items[0]->setIcon(videoIcon);
        }else if(mimeName.startsWith("image")){
            items[0]->setIcon(imageIcon);
        }else{
            items[0]->setIcon(otherIcon);
        }
        items[1]->setText(mimeType.name());
        items[2]->setText(size);
        items[2]->setTextAlignment(Qt::AlignRight | Qt::AlignVCenter);
        items[3]->setText(durStr);
        items[4]->setText(md5);
        items[5]->setText(presenceStr);
        root->appendRow(items);
        for(int i = 0; i < 6; ++i){
            mFileView->resizeColumnToContents(i);
        }
    }
    mFileView->setUpdatesEnabled(true);
    mFileView->setSortingEnabled(true);
    qApp->restoreOverrideCursor();
    QString msg = QString(tr("Analyzed %1 files in %2 ms using %3 queries").arg(QString::number(fileCount)).arg(QString::number(timer.elapsed())).arg(QString::number(mQueryCount)));
    emit message(msg);
}

void FSWidget::advanceDir(int by){
    int dirCount = mDirCB->count();
    int nextIdx = mDirCB->currentIndex() + by;
    if(nextIdx >= dirCount){
        nextIdx = 0;
    }
    if(nextIdx < 0){
        nextIdx = dirCount - 1;
    }
    mDirCB->setCurrentIndex(nextIdx);
}

int FSWidget::queryCount(QSqlQuery &q, const QString &arg){
    int retval = -1;
    q.bindValue(":arg", arg);
    q.exec();
    while(q.next()){
        retval = q.value(0).toInt();
    }
    ++mQueryCount;
    return retval;
}