CQtDeployer/tests/units/testutils.cpp

/*
 * Copyright (C) 2018-2023 QuasarApp.
 * Distributed under the lgplv3 software license, see the accompanying
 * Everyone is permitted to copy and distribute verbatim copies
 * of this license document, but changing it is not allowed.
 */

#include "testutils.h"

#include <QDir>
#include <QFileInfo>

TestUtils::TestUtils()
{

}

QSet<QString> TestUtils::getTree(const QString &path, int limit, int depch) {
    QFileInfo info(path);

    if (limit > 0 && depch >= limit) {
        return {};
    }

    if (!info.exists()) {
        return {};
    }

    auto result = QSet<QString>{};

    if (!info.isDir()) {
        result.insert(getFilePath(info.absoluteFilePath()));
        return result;
    }

    QDir dir(info.absoluteFilePath());
    auto list = dir.entryInfoList(QDir::AllEntries | QDir::NoDotAndDotDot);
    for (const auto &i: std::as_const(list)) {
        result.unite(getTree(i.absoluteFilePath(), limit, depch + 1));
    }

    return result;
}

QSet<QString> TestUtils::getFilesSet(const QString &path, int limit, int depch) {
    QFileInfo info(path);

    if (limit > 0 && depch >= limit) {
        return {};
    }

    if (!info.exists()) {
        return {};
    }

    auto result = QSet<QString>{};

    if (!info.isDir()) {
        result.insert(getFilePath(info.fileName()));
        return result;
    }

    QDir dir(info.absoluteFilePath());
    auto list = dir.entryInfoList(QDir::AllEntries | QDir::NoDotAndDotDot);
    for (const auto &i: std::as_const(list)) {
        result.unite(getFilesSet(i.absoluteFilePath(), limit, depch + 1));
    }

    return result;
}

QString TestUtils::getFilePath(const QString& i) {
    auto file = i;
    auto delimiter = file.lastIndexOf("/");
    auto shared = file.indexOf(".so.", Qt::CaseInsensitive);

    if (shared > delimiter) {
        file = file.left(shared);
        file.push_back(".so");
    }

    return file;
}

QSet<QString> TestUtils::createTree(const QStringList &tree) {
    QSet<QString> res;
    for (const auto &i : tree) {
        res.insert(getFilePath(QFileInfo(i).absoluteFilePath()));
    }

    return res;
}

QSet<QString> TestUtils::deployTree(const QStringList &tree) {
    QSet<QString> res;
    for (const auto &i : tree) {
        QFileInfo fInfo(i);
        if (!fInfo.exists() && QDir().mkpath(i)) {
            res.insert(getFilePath(QFileInfo(i).absoluteFilePath()));
        }
    }

    return res;
}

QHash<QString, int> TestUtils::compareTree(const QSet<QString> &leftTree, const QSet<QString> &rightTree) {
    QHash<QString, int> result;

    auto valuel = leftTree - rightTree;
    auto valuer = rightTree - leftTree;

    for(auto &i :std::as_const(valuel)) {
        result.insert(i, 1);
    }

    for(auto &i :std::as_const(valuer)) {
        result.insert(i, -1);
    }

    return result;
}

bool TestUtils::deployFile(const QString &file, const QString &distanation,
                           const QHash<QByteArray, QByteArray> &replaceCase) const {
    QFile f(file);

    if (f.open(QIODevice::ReadOnly)) {
        QFile dist(distanation);
        if (!dist.open(QIODevice::WriteOnly | QIODevice::Truncate)) {
            return false;
        }

        QByteArray data = f.readAll();
        for (auto it = replaceCase.begin(); it != replaceCase.end(); ++it) {
            data.replace(it.key(), it.value());
        }

        return dist.write(data);
    }

    return false;
}