watch filesystem, compute and match hashes

Signed-off-by: Rachel Powers <508861+Ryex@users.noreply.github.com>
This commit is contained in:
Rachel Powers
2022-10-25 01:19:19 -07:00
parent 028e086960
commit 1598d65824
4 changed files with 212 additions and 4 deletions

View File

@ -1,3 +1,6 @@
#include <qfileinfo.h>
#include <qnamespace.h>
#include "Application.h"
#include "BlockedModsDialog.h"
#include "ui_BlockedModsDialog.h"
#include <QPushButton>
@ -5,20 +8,29 @@
#include <QDesktopServices>
#include <QDebug>
#include <QStandardPaths>
BlockedModsDialog::BlockedModsDialog(QWidget *parent, const QString &title, const QString &text, const QList<BlockedMod> &mods) :
BlockedModsDialog::BlockedModsDialog(QWidget *parent, const QString &title, const QString &text, QList<BlockedMod> &mods) :
QDialog(parent), ui(new Ui::BlockedModsDialog), mods(mods) {
ui->setupUi(this);
auto openAllButton = ui->buttonBox->addButton(tr("Open All"), QDialogButtonBox::ActionRole);
connect(openAllButton, &QPushButton::clicked, this, &BlockedModsDialog::openAll);
connect(&watcher, &QFileSystemWatcher::directoryChanged, this, &BlockedModsDialog::directoryChanged);
hashing_task = shared_qobject_ptr<ConcurrentTask>(new ConcurrentTask(this, "MakeHashesTask", 10));
qDebug() << "Mods List: " << mods;
setupWatch();
scanPaths(true);
this->setWindowTitle(title);
ui->label->setText(text);
ui->textBrowser->setText(body);
update();
}
@ -50,6 +62,110 @@ void BlockedModsDialog::update() {
ui->textBrowser->setText(text);
}
void BlockedModsDialog::directoryChanged(QString path) {
qDebug() << "Directory changed: " << path;
scanPath(path, false);
}
void BlockedModsDialog::setupWatch() {
const QString downloadsFolder = QStandardPaths::writableLocation(QStandardPaths::DownloadLocation);
const QString modsFolder = APPLICATION->settings()->get("CentralModsDir").toString();
watcher.addPath(downloadsFolder);
watcher.addPath(modsFolder);
}
void BlockedModsDialog::scanPaths(bool init) {
for (auto &dir : watcher.directories()) {
scanPath(dir, init);
}
}
void BlockedModsDialog::scanPath(QString path, bool init) {
QDir scan_dir(path);
QDirIterator scan_it(path, QDir::Filter::Files | QDir::Filter::Hidden, QDirIterator::NoIteratorFlags);
while (scan_it.hasNext()) {
QString file = scan_it.next();
if (checked_paths.contains(file)){
continue;
}
if (!checkValidPath(file)) {
continue;
}
auto hash_task = Hashing::createBlockedModHasher(file, ModPlatform::Provider::FLAME, "sha1");
qDebug() << "Creating Hash task for path: " << file;
connect(hash_task.get(), &Task::succeeded, [this, hash_task, file] {
checkMatchHash(hash_task->getResult(), file);
});
connect(hash_task.get(), &Task::failed, [this, hash_task, file] {
qDebug() << "Failed to hash path: " << file;
});
if (init) {
checked_paths.insert(file);
}
hashing_task->addTask(hash_task);
}
hashing_task->start();
}
void BlockedModsDialog::checkMatchHash(QString hash, QString path) {
bool match = false;
qDebug() << "Checking for match on hash: " << hash << " | From path:" << path;
for (auto &mod : mods) {
if (mod.matched) {
continue;
}
if (mod.hash.compare(hash, Qt::CaseInsensitive) == 0) {
mod.matched = true;
mod.localPath = path;
match = true;
qDebug() << "Hash match found: " << mod.name << " " << hash << " | From path:" << path;
break;
}
}
if (match) {
update();
}
}
bool BlockedModsDialog::checkValidPath(QString path) {
QFileInfo file = QFileInfo(path);
QString filename = file.fileName();
for (auto &mod : mods) {
if (mod.name.compare(filename, Qt::CaseInsensitive) == 0) {
qDebug() << "Name match found: " << mod.name << " | From path:" << path;
return true;
}
}
return false;
}
bool BlockedModsDialog::allModsMatched() {
for (auto &mod : mods) {
if (!mod.matched)
return false;
}
return true;
}
QDebug operator<<(QDebug debug, const BlockedMod &m) {
QDebugStateSaver saver(debug);