swg-launchpad/filescanner.cpp

#include "filescanner.h"
#include "mainwindow.h"
#include <QSettings>
#include <QtConcurrent/QtConcurrent>

FileScanner::FileScanner(MainWindow *main) {
    mainWindow = main;
}

bool FileScanner::checkSwgFolder(const QString& dirString) {
    QDir dir(dirString);

    if (!dir.exists())
        return false;

    QVector<QPair<QString, qint64> > fileListToCheck = MainWindow::getRequiredFiles();

    int fileCountToCheck = fileListToCheck.size() / 3;
    int currentFiles = 0;

   // if (isARequiredFile())

    QStringList files = dir.entryList(QDir::Files | QDir::NoDotAndDotDot);

    //qDebug() << "checking for " << fileCountToCheck << " files";

    for (int i = 0; i < files.size(); ++i) {
        if (isARequiredFile(files.at(i)))
            currentFiles++;
    }

    //qDebug() << "found " << currentFiles << " swg files";

    if (currentFiles < fileCountToCheck)
        return false;

    return true;
}

bool FileScanner::isARequiredFile(const QString& name) {
    QVector<QPair<QString, qint64> > fileListToCheck = MainWindow::getRequiredFiles();

    for (int i = 0; i < fileListToCheck.size(); ++i) {
        const QPair<QString, qint64> data = fileListToCheck.at(i);

        if (name.toLower() == data.first.toLower())
            return true;
    }

    return false;
}

int FileScanner::loadAndBasicCheckFiles(QString swgFolder) {
    QVector<QPair<QString, qint64> > fileListToCheck = MainWindow::getRequiredFiles();

    for (int i = 0; i < fileListToCheck.size() && !mainWindow->doCancelWorkingThreads(); ++i) {
        const QPair<QString, qint64> data = fileListToCheck.at(i);
        QString file = swgFolder + "/" + data.first;

        QFile fileObject(file);

        if (!fileObject.exists()) {
            qDebug() << file << "doesnt exist";
            return -1;
        }

        qint64 fileObjectSize = fileObject.size();

        if (!data.first.contains(".exe") && fileObjectSize != data.second) {
            qDebug() << file << " size mismatch found: " << fileObjectSize << " expected: " << data.second;
            return -1;
        }

        if (!mainWindow->doCancelWorkingThreads())
            emit requiredFileExists(file);
    }

    return 0;
}

void FileScanner::fullScanMultiThreaded(bool ) {
    QFile* file = mainWindow->getRequiredFilesFile();

    QSettings settings;
    QString swgFolder = settings.value("swg_folder").toString();

    while (!file->atEnd()) {
        QByteArray line = file->readLine();
        //process_line(line);

        //QRegExp rx("(\\ |\\,|\\.|\\;|\\t)"); //RegEx for ' ' or ',' or '.' or ':' or '\t'
        QList<QByteArray> query = line.split(';');

        //QListIterator<QByteArray>

        QString name = query.at(0);
        QString size = query.at(1);
        QString md5 = query.at(2).trimmed();

        QString file = swgFolder + "/" + name;

        //std::atomic<int> test;

        QtConcurrent::run(this, &FileScanner::fullScanFile, file, name, size.toLongLong(), md5);
        //emit progressTextChanged(file);
/*
        QFile fileObject(file);

        if (!fileObject.exists()) {
            qDebug() << file << "doesnt exist";
            //return -1;

            filesToDownload.append(patchUrl + name);

            if (!cancelWorkingThreads)
                emit fullScannedFile(name, false);

            continue;
        }

        QCryptographicHash crypto(QCryptographicHash::Md5);
        if (!fileObject.open(QFile::ReadOnly)) {
            qDebug() << "could not open file:" << file;
            //return -1;

            filesToDownload.append(patchUrl + name);

            if (!cancelWorkingThreads)
                emit fullScannedFile(name, false);

            continue;
        }

        //qint64 totalSize = 0;

        while (!fileObject.atEnd() && !cancelWorkingThreads){
            crypto.addData(fileObject.read(8192));
        }

        QByteArray hash = crypto.result();
        QString calculatedHash = hash.toHex().toUpper().trimmed();

        int compareResult = calculatedHash.compare(md5);

        if (compareResult != 0 && !cancelWorkingThreads) {
            qDebug() << "hash mismatch for:" << file << " compare result:" << compareResult;

            qDebug() << "calculated hash of:" << file << " is:" << calculatedHash << " and specified one is:" << md5;
            res = 2;

            filesToDownload.append(patchUrl + name);
        }

        if (!cancelWorkingThreads)
            emit fullScannedFile(name, compareResult == 0);
            */
    }


    /*
    if (!cancelWorkingThreads)
        emit startDownload();
        */

    delete file;
}

void FileScanner::fullScanFile(const QString& file, const QString& name, qint64, const QString &md5) {
    QFile fileObject(file);

    if (!fileObject.exists()) {
        qDebug() << file << "doesnt exist";
        //return -1;

       // filesToDownload.append(patchUrl + name);

        emit addFileToDownload(MainWindow::patchUrl + name);

      //  if (!cancelWorkingThreads)
            emit fullScannedFile(name, false);

        //fullScanWorkingThreads.deref();
        mainWindow->decrementFullScanWorkingThreads();

        return;
    }

    QCryptographicHash crypto(QCryptographicHash::Md5);
    if (!fileObject.open(QFile::ReadOnly)) {
        qDebug() << "could not open file:" << file;
        //return -1;

       // filesToDownload.append(patchUrl + name);

        emit addFileToDownload(MainWindow::patchUrl + name);

     //   if (!cancelWorkingThreads)
            emit fullScannedFile(name, false);

        //fullScanWorkingThreads.deref();
        mainWindow->decrementFullScanWorkingThreads();

        return;
    }

    //qint64 totalSize = 0;

    while (!fileObject.atEnd() && !mainWindow->doCancelWorkingThreads()){
        crypto.addData(fileObject.read(8192));
    }

    QByteArray hash = crypto.result();
    QString calculatedHash = hash.toHex().toUpper().trimmed();

    int compareResult = calculatedHash.compare(md5);

    if (compareResult != 0 && !mainWindow->doCancelWorkingThreads()) {
        qDebug() << "hash mismatch for:" << file << " compare result:" << compareResult;

        qDebug() << "calculated hash of:" << file << " is:" << calculatedHash << " and specified one is:" << md5;
        //res = 2;

        //filesToDownload.append(patchUrl + name);
        emit addFileToDownload(MainWindow::patchUrl + name);
    }

   // if (!cancelWorkingThreads)
        emit fullScannedFile(name, compareResult == 0);

    //fullScanWorkingThreads.deref();
    mainWindow->decrementFullScanWorkingThreads();
}

int FileScanner::fullScanSingleThreaded(bool ) {
    QFile* file = mainWindow->getRequiredFilesFile();

    QSettings settings;
    QString swgFolder = settings.value("swg_folder").toString();

    int res = 0;

    while (!file->atEnd()) {
        QByteArray line = file->readLine();
        //process_line(line);

        //QRegExp rx("(\\ |\\,|\\.|\\;|\\t)"); //RegEx for ' ' or ',' or '.' or ':' or '\t'
        QList<QByteArray> query = line.split(';');

        //QListIterator<QByteArray>

        QString name = query.at(0);
        QString size = query.at(1);
        QString md5 = query.at(2).trimmed();

        QString file = swgFolder + "/" + name;

        QFile fileObject(file);

        if (!fileObject.exists()) {
            qDebug() << file << "doesnt exist";
            //return -1;

            //filesToDownload.append(MainWindow::patchUrl + name);
            mainWindow->appendToFilesToDownloadStringList(MainWindow::patchUrl + name);

            if (!mainWindow->doCancelWorkingThreads())
                emit fullScannedFile(name, false);

            continue;
        }

        QCryptographicHash crypto(QCryptographicHash::Md5);
        if (!fileObject.open(QFile::ReadOnly)) {
            qDebug() << "could not open file:" << file;
            //return -1;

            //filesToDownload.append(MainWindow::patchUrl + name);
            mainWindow->appendToFilesToDownloadStringList(MainWindow::patchUrl + name);

            if (!mainWindow->doCancelWorkingThreads())
                emit fullScannedFile(name, false);

            continue;
        }

        //qint64 totalSize = 0;

        while (!fileObject.atEnd() && !mainWindow->doCancelWorkingThreads()){
            crypto.addData(fileObject.read(8192));
        }

        QByteArray hash = crypto.result();
        QString calculatedHash = hash.toHex().toUpper().trimmed();

        int compareResult = calculatedHash.compare(md5);

        if (compareResult != 0 && !mainWindow->doCancelWorkingThreads()) {
            qDebug() << "hash mismatch for:" << file << " compare result:" << compareResult;

            qDebug() << "calculated hash of:" << file << " is:" << calculatedHash << " and specified one is:" << md5;
            res = 2;

            //filesToDownload.append(MainWindow::patchUrl + name);
            mainWindow->appendToFilesToDownloadStringList(MainWindow::patchUrl + name);
        }

        if (!mainWindow->doCancelWorkingThreads())
            emit fullScannedFile(name, compareResult == 0);
    }


    /*
    if (!cancelWorkingThreads)
        emit startDownload();
        */

    delete file;

    return res;
}