databaseutils.cpp

// SPDX-FileCopyrightText: Copyright (C) 2016 swift Project Community / Contributors
// SPDX-License-Identifier: GPL-3.0-or-later OR LicenseRef-swift-pilot-client-1

#include "core/db/databaseutils.h"

#include <QElapsedTimer>

#include "core/application.h"
#include "core/webdataservices.h"
#include "misc/compressutils.h"
#include "misc/fileutils.h"
#include "misc/logmessage.h"

using namespace swift::misc;
using namespace swift::misc::json;
using namespace swift::misc::aviation;
using namespace swift::misc::simulation;

namespace swift::core::db
{
    const QStringList &CDatabaseUtils::getLogCategories()
    {
        static const QStringList cats { CLogCategories::modelCache(), CLogCategories::modelSetCache() };
        return cats;
    }

    CAircraftModel CDatabaseUtils::consolidateOwnAircraftModelWithDbData(const CAircraftModel &model, bool force,
                                                                         bool *modified)
    {
        bool myModified = false;
        CAircraftModel ownModel = CDatabaseUtils::consolidateModelWithDbData(model, force, &myModified);
        // special case here, as we have some specific values for a local model

        if (myModified)
        {
            ownModel.updateMissingParts(model);
            ownModel.setFileName(model.getFileName());
            myModified = true;
        }
        if (ownModel.getModelType() != CAircraftModel::TypeOwnSimulatorModel)
        {
            ownModel.setModelType(CAircraftModel::TypeOwnSimulatorModel);
            myModified = true;
        }
        if (modified) { *modified = myModified; }
        return ownModel;
    }

    CAircraftModel CDatabaseUtils::consolidateModelWithDbData(const CAircraftModel &model, bool force, bool *modified)
    {
        if (modified) { *modified = false; }
        if (!model.hasModelString()) { return model; }
        if (!hasDbAircraftData()) { return model; }

        CAircraftModel dbModel(sApp->getWebDataServices()->getModelForModelString(model.getModelString()));
        return CDatabaseUtils::consolidateModelWithDbData(model, dbModel, force, modified);
    }

    CAircraftModel CDatabaseUtils::consolidateModelWithDbData(const CAircraftModel &model,
                                                              const CAircraftModel &dbModel, bool force, bool *modified)
    {
        Q_ASSERT_X(sApp, Q_FUNC_INFO, "Missing application object");
        Q_ASSERT_X(sApp->hasWebDataServices(), Q_FUNC_INFO, "No web services");

        if (modified) { *modified = false; }
        if (!model.hasModelString()) { return model; }
        if (!force && model.hasValidDbKey()) { return model; }
        const int distributorOrder = model.getDistributorOrder(); // later restore that order

        if (dbModel.isLoadedFromDb())
        {
            // take the db model as original
            if (modified) { *modified = true; }
            CAircraftModel dbModelModified(dbModel);
            dbModelModified.updateMissingParts(model);
            dbModelModified.setDistributorOrder(distributorOrder);
            dbModelModified.setSimulator(dbModel.getSimulator()); // DB simulator settings have priority
            dbModelModified.setModelLivery(model.getModelLivery()); // keep local livery settings msfs2024
            return dbModelModified;
        }

        // we try our best to update by DB data here
        // since we have no(!) DB model, we update each of it subobjects
        CAircraftModel consolidatedModel(model); // copy over
        if (!consolidatedModel.getLivery().hasValidDbKey())
        {
            const CLivery dbLivery(sApp->getWebDataServices()->smartLiverySelector(consolidatedModel.getLivery()));
            if (dbLivery.hasValidDbKey())
            {
                if (modified) { *modified = true; }
                consolidatedModel.setLivery(dbLivery);
                consolidatedModel.setModelLivery(model.getModelLivery()); // keep local livery settings msfs2024
            }
        }
        if (!consolidatedModel.getAircraftIcaoCode().hasValidDbKey() && consolidatedModel.hasAircraftDesignator())
        {
            // try to find DB aircraft ICAO here
            const CAircraftIcaoCode dbIcao(
                sApp->getWebDataServices()->smartAircraftIcaoSelector(consolidatedModel.getAircraftIcaoCode()));
            if (dbIcao.hasValidDbKey())
            {
                if (modified) { *modified = true; }
                consolidatedModel.setAircraftIcaoCode(dbIcao);
                consolidatedModel.setModelLivery(model.getModelLivery()); // keep local livery settings msfs2024
            }
        }

        const CDistributor dbDistributor(
            sApp->getWebDataServices()->getDistributors().smartDistributorSelector(model.getDistributor(), model));
        if (dbDistributor.isLoadedFromDb())
        {
            if (modified) { *modified = true; }
            consolidatedModel.setDistributor(dbDistributor);
            consolidatedModel.setModelLivery(model.getModelLivery()); // keep local livery settings msfs2024
        }
        consolidatedModel.updateLocalFileNames(model);
        consolidatedModel.setDistributorOrder(distributorOrder);
        return consolidatedModel;
    }

    int CDatabaseUtils::consolidateModelsWithDbData(CAircraftModelList &models, bool force)
    {
        return CDatabaseUtils::consolidateModelsWithDbDataAllowsGuiRefresh(models, force, false);
    }

    int CDatabaseUtils::fillInMissingAircraftAndLiveryEntities(CAircraftModelList &models)
    {
        // fill in those entities which have only an id (key), but no data yet
        int c = 0;
        for (CAircraftModel &model : models)
        {
            bool changed = false;
            if (model.getLivery().hasValidDbKey() && !model.getLivery().hasCompleteData())
            {
                const CLivery livery = sApp->getWebDataServices()->getLiveryForDbKey(model.getLivery().getDbKey());
                if (livery.isLoadedFromDb())
                {
                    model.setLivery(livery);
                    changed = true;
                }
            }
            if (model.getAircraftIcaoCode().hasValidDbKey() && !model.getAircraftIcaoCode().hasCompleteData())
            {
                const CAircraftIcaoCode icao =
                    sApp->getWebDataServices()->getAircraftIcaoCodeForDbKey(model.getAircraftIcaoCode().getDbKey());
                if (icao.isLoadedFromDb())
                {
                    model.setAircraftIcaoCode(icao);
                    changed = true;
                }
            }
            if (model.getDistributor().hasValidDbKey() && !model.getDistributor().hasCompleteData())
            {
                const CDistributor distributor =
                    sApp->getWebDataServices()->getDistributorForDbKey(model.getDistributor().getDbKey());
                if (distributor.isLoadedFromDb())
                {
                    model.setDistributor(distributor);
                    changed = true;
                }
            }
            if (changed) { c++; }
        }
        return c;
    }

    CAircraftModelList CDatabaseUtils::consolidateModelsWithSimulatorModelsAllowsGuiRefresh(
        const CAircraftModelList &models, const CAircraftModelList &simulatorModels, QStringList &removedModelStrings,
        bool processEvents)
    {
        if (models.isEmpty() || simulatorModels.isEmpty()) { return models; }

        QElapsedTimer timer;
        timer.start();
        const QSet<QString> allOwnModelsModelStrings = simulatorModels.getModelStringSet();
        CAircraftModelList consolidatedModels;
        removedModelStrings.clear();

        int c = 0;
        for (const CAircraftModel &model : models)
        {
            c++;
            if (processEvents && c % 125 == 0)
            {
                if (!sApp || sApp->isShuttingDown()) { return models; }
                CApplication::processEventsFor(25);
            }

            const QString ms(model.getModelString());
            if (ms.isEmpty()) { continue; }
            if (!allOwnModelsModelStrings.contains(ms))
            {
                removedModelStrings.push_back(ms);
                continue;
            }
            CAircraftModel consolidated = simulatorModels.findFirstByModelStringAliasOrDefault(ms);
            if (consolidated.hasModelString())
            {
                CDatabaseUtils::consolidateModelWithDbData(consolidated, true);
                consolidatedModels.push_back(consolidated);
            }
            else { consolidatedModels.push_back(model); }
        }
        CLogMessage(static_cast<CDatabaseUtils *>(nullptr)).info(u"Consolidated %1 vs. %2 in %3 ms")
            << models.size() << simulatorModels.size() << timer.elapsed() << "ms";
        return consolidatedModels;
    }

    CAircraftModelList
    CDatabaseUtils::updateModelsDirectoriesAllowsGuiRefresh(const CAircraftModelList &models,
                                                            const CAircraftModelList &simulatorModels,
                                                            QStringList &removedModelStrings, bool processEvents)
    {
        if (models.isEmpty() || simulatorModels.isEmpty()) { return models; }

        QElapsedTimer timer;
        timer.start();
        const QSet<QString> allOwnModelsModelStrings = simulatorModels.getModelStringSet();
        CAircraftModelList consolidatedModels;
        removedModelStrings.clear();

        int c = 0;
        for (const CAircraftModel &model : models)
        {
            c++;
            if (processEvents && c % 125 == 0)
            {
                if (!sApp || sApp->isShuttingDown()) { return models; }
                CApplication::processEventsFor(25);
            }

            const QString ms(model.getModelString());
            if (ms.isEmpty()) { continue; }
            if (!allOwnModelsModelStrings.contains(ms))
            {
                removedModelStrings.push_back(ms);
                continue;
            }
            CAircraftModel consolidated = simulatorModels.findFirstByModelStringAliasOrDefault(ms);
            if (consolidated.hasModelString())
            {
                if (consolidated.hasExistingCorrespondingFile())
                {
                    // update by existing one
                    consolidatedModels.push_back(consolidated);
                }
                else
                {
                    // keep
                    consolidatedModels.push_back(model);
                }
            }
            else { consolidatedModels.push_back(model); }
        }
        CLogMessage(static_cast<CDatabaseUtils *>(nullptr)).info(u"Updated directories %1 vs. %2 in %3 ms")
            << models.size() << simulatorModels.size() << timer.elapsed() << "ms";
        return consolidatedModels;
    }

    int CDatabaseUtils::consolidateModelsWithDbDataAllowsGuiRefresh(CAircraftModelList &models, bool force,
                                                                    bool processEvents)
    {
        QElapsedTimer timer;
        timer.start();
        int c = 0;
        if (models.isEmpty()) { return c; }
        for (CAircraftModel &model : models)
        {
            if (!force && model.isLoadedFromDb()) { continue; }
            bool modified = false;
            model = CDatabaseUtils::consolidateModelWithDbData(model, force, &modified);
            if (modified || model.hasValidDbKey())
            {
                c++;
                if (processEvents && c % 125 == 0) { CApplication::processEventsFor(25); }
            }
        }
        CLogMessage(static_cast<CDatabaseUtils *>(nullptr)).info(u"Consolidated %1 models in %2ms")
            << models.size() << timer.elapsed();
        return c;
    }

    int CDatabaseUtils::consolidateModelsWithDbData(const CAircraftModelList &dbModels,
                                                    CAircraftModelList &simulatorModels, bool force)
    {
        if (dbModels.isEmpty() || simulatorModels.isEmpty()) { return 0; }

        QElapsedTimer timer;
        timer.start();
        const QSet<QString> dbModelsModelStrings = dbModels.getModelStringSet();

        int c = 0;
        for (CAircraftModel &model : simulatorModels)
        {
            const QString ms(model.getModelString());
            if (ms.isEmpty()) { continue; }
            if (!dbModelsModelStrings.contains(ms)) { continue; }
            bool modified = false;
            const CAircraftModel consolidated = CDatabaseUtils::consolidateModelWithDbData(
                model, dbModels.findFirstByModelStringAliasOrDefault(ms), force, &modified);
            if (!modified) { continue; }
            model = consolidated;
            c++;
        }
        CLogMessage(static_cast<CDatabaseUtils *>(nullptr)).info(u"Consolidated %1 models in %2 ms")
            << simulatorModels.size() << timer.elapsed();
        return c;
    }

    int CDatabaseUtils::consolidateModelsWithDbModelAndDistributor(CAircraftModelList &models, bool force)
    {
        int c = 0;
        if (models.isEmpty()) { return c; }
        for (CAircraftModel &model : models)
        {
            if (!force && model.hasValidDbKey()) { continue; }
            const CAircraftModel dbModel(sApp->getWebDataServices()->getModelForModelString(model.getModelString()));
            if (dbModel.hasValidDbKey())
            {
                model = dbModel;
                c++;
                continue;
            }
            const CDistributor distributor =
                sApp->getWebDataServices()->smartDistributorSelector(model.getDistributor(), model);
            if (distributor.isLoadedFromDb()) { model.setDistributor(distributor); }
        }
        return c;
    }

    CAircraftModel CDatabaseUtils::consolidateModelWithDbDistributor(const CAircraftModel &model, bool force)
    {
        if (!force && model.getDistributor().isLoadedFromDb()) { return model; }
        const CDistributor distributor =
            sApp->getWebDataServices()->smartDistributorSelector(model.getDistributor(), model);
        if (!distributor.isLoadedFromDb()) { return model; }
        CAircraftModel newModel(model);
        newModel.setDistributor(distributor);
        return newModel;
    }

    int CDatabaseUtils::consolidateModelsWithDbDistributor(CAircraftModelList &models, bool force)
    {
        int c = 0;
        if (models.isEmpty()) { return c; }
        for (CAircraftModel &model : models)
        {
            if (model.hasValidDbKey() || model.getDistributor().hasValidDbKey()) { continue; }
            model = CDatabaseUtils::consolidateModelWithDbDistributor(model, force);
            if (model.getDistributor().hasValidDbKey()) { c++; }
        }
        return c;
    }

    CAircraftModelList CDatabaseUtils::updateSimulatorForFsFamily(const CAircraftModelList &ownModels,
                                                                  CStatusMessageList *updateInfo, int maxToStash,
                                                                  IProgressIndicator *progressIndicator,
                                                                  bool processEvents)
    {
        if (!sApp || !sApp->getWebDataServices() || sApp->isShuttingDown()) { return {}; }
        const CAircraftModelList dbFsFamilyModels(sApp->getWebDataServices()->getModels().findFsFamilyModels());
        CAircraftModelList stashModels;
        if (dbFsFamilyModels.isEmpty() || ownModels.isEmpty()) { return stashModels; }
        const QSet<QString> dbKeys = dbFsFamilyModels.getModelStringSet();
        const int maxModelsCount = (maxToStash >= 0) ? maxToStash : ownModels.size();
        if (maxModelsCount < 1) { return stashModels; }

        int c = 0; // counter
        for (const CAircraftModel &ownModel : ownModels)
        {
            c++;

            // process events
            if (processEvents && c % 500 == 0)
            {
                if (progressIndicator)
                {
                    const int percentage = c * 100 / maxModelsCount;
                    progressIndicator->updateProgressIndicatorAndProcessEvents(percentage);
                }
                else { CApplication::processEventsFor(10); }
            }

            // values to be skipped
            if (maxToStash >= 0 && stashModels.size() > maxToStash) { break; }
            if (!dbKeys.contains(ownModel.getModelString())) { continue; }
            if (ownModel.matchesSimulatorFlag(CSimulatorInfo::XPLANE)) { continue; }

            // in DB
            CAircraftModel dbModel = dbFsFamilyModels.findFirstByModelStringAliasOrDefault(ownModel.getModelString());
            if (!dbModel.isLoadedFromDb()) { continue; }

            // update simulator and add
            CSimulatorInfo simulator(dbModel.getSimulator());
            const CSimulatorInfo dbModelSimulator(simulator);
            const CSimulatorInfo delta = simulator.add(ownModel.getSimulator());
            if (delta.getSimulator() == CSimulatorInfo::None) { continue; }

            dbModel.setSimulator(simulator);
            stashModels.push_back(dbModel); // changed DB model
            if (updateInfo)
            {
                const CStatusMessage m =
                    CStatusMessage(static_cast<CDatabaseUtils *>(nullptr)).info(u"%1 -> %2 (%3) for '%4'")
                    << dbModelSimulator.toQString(true) << simulator.toQString(true) << delta.toQString(true)
                    << dbModel.getModelStringAndDbKey();
                updateInfo->push_back(m);
            }
        }
        progressIndicator->updateProgressIndicatorAndProcessEvents(100);
        return stashModels;
    }

    QJsonDocument CDatabaseUtils::databaseJsonToQJsonDocument(const QString &content)
    {
        static const QString compressed("swift:");
        if (content.isEmpty()) { return {}; }
        QByteArray byteData;
        if (json::looksLikeJson(content))
        {
            // uncompressed
            byteData = content.toUtf8();
        }
        else if (content.startsWith(compressed) && content.length() > compressed.length() + 3)
        {
            do {
                // "swift:1234:base64encoded
                const int cl = compressed.length();
                const int contentIndex = content.indexOf(':', cl);
                if (contentIndex < cl) break; // should not happen, malformed
                const QString ls = content.mid(cl, contentIndex - cl); // content length
                bool ok {};
                const qint32 size = ls.toInt(&ok);
                if (!ok) break; // malformed size
                if (size < 1) break;

                QByteArray ba;
                ba.append(content.mid(contentIndex).toUtf8());
                ba = QByteArray::fromBase64(ba);
                ba.insert(0, CCompressUtils::lengthHeader(size)); // adding 4 bytes length header
                byteData = qUncompress(ba);
            }
            while (false);
        }

        if (byteData.isEmpty()) { return {}; }
        return QJsonDocument::fromJson(byteData);
    }

    QJsonDocument CDatabaseUtils::readQJsonDocumentFromDatabaseFile(const QString &filename)
    {
        const QString raw = CFileUtils::readFileToString(filename);
        if (raw.isEmpty()) { return {}; }
        return CDatabaseUtils::databaseJsonToQJsonDocument(raw);
    }

    QJsonObject CDatabaseUtils::readQJsonObjectFromDatabaseFile(const QString &filename)
    {
        const QString raw = CFileUtils::readFileToString(filename);
        if (raw.isEmpty()) { return {}; }

        // allow also compressed format
        const QJsonDocument jsonDoc = CDatabaseUtils::databaseJsonToQJsonDocument(raw);
        return jsonDoc.object();
    }

    QJsonObject CDatabaseUtils::readQJsonObjectFromDatabaseFile(const QString &directory, const QString &filename)
    {
        return CDatabaseUtils::readQJsonObjectFromDatabaseFile(CFileUtils::appendFilePaths(directory, filename));
    }

    bool CDatabaseUtils::hasDbAircraftData()
    {
        return sApp && sApp->hasWebDataServices() && sApp->getWebDataServices()->hasDbAircraftData();
    }

    const QUrlQuery &CDatabaseUtils::getCompressedQuery()
    {
        static const QUrlQuery q("compressed=true");
        return q;
    }

    QHttpPart CDatabaseUtils::getJsonTextMultipart(const QJsonObject &json, bool compress)
    {
        const QByteArray bytes(QJsonDocument(json).toJson(QJsonDocument::Compact));
        return CDatabaseUtils::getJsonTextMultipart(bytes, compress);
    }

    QHttpPart CDatabaseUtils::getJsonTextMultipart(const QString &json, bool compress)
    {
        const QByteArray bytes(json.toUtf8());
        return CDatabaseUtils::getJsonTextMultipart(bytes, compress);
    }

    QHttpPart CDatabaseUtils::getJsonTextMultipart(const QJsonArray &json, bool compress)
    {
        const QByteArray bytes(QJsonDocument(json).toJson(QJsonDocument::Compact));
        return CDatabaseUtils::getJsonTextMultipart(bytes, compress);
    }

    QHttpPart CDatabaseUtils::getJsonTextMultipart(const QByteArray &bytes, bool compress)
    {
        static const QString name("form-data; name=\"swiftjson\"");
        static const QVariant header(name);
        QHttpPart textPart;
        textPart.setHeader(QNetworkRequest::ContentDispositionHeader, header);
        if (compress)
        {
            QByteArray ba = qCompress(bytes);
            ba.remove(0, 4); // remove the non standard header
            textPart.setBody(ba);
        }
        else { textPart.setBody(bytes); }
        return textPart;
    }

    QHttpPart CDatabaseUtils::getMultipartWithDebugFlag()
    {
        QHttpPart textPartDebug;
        textPartDebug.setHeader(QNetworkRequest::ContentDispositionHeader,
                                QVariant("form-data; name=\"XDEBUG_SESSION_START\""));
        textPartDebug.setBody(QString("ECLIPSE_DBGP").toUtf8());
        return textPartDebug;
    }

    ChangedAutoPublishData CDatabaseUtils::autoPublishDataChanged(const QString &modelString,
                                                                  const physical_quantities::CLength &cg,
                                                                  const CSimulatorInfo &simulator)
    {
        ChangedAutoPublishData changed;
        if (!sApp || sApp->isShuttingDown() || !sApp->getWebDataServices()) { return changed; }
        const CAircraftModel model = sApp->getWebDataServices()->getModelForModelString(modelString);
        return CDatabaseUtils::autoPublishDataChanged(model, cg, simulator);
    }

    ChangedAutoPublishData CDatabaseUtils::autoPublishDataChanged(const CAircraftModel &model,
                                                                  const physical_quantities::CLength &cg,
                                                                  const CSimulatorInfo &simulator)
    {
        ChangedAutoPublishData changed;
        changed.modelKnown = model.hasValidDbKey();
        if (!changed.modelKnown) { return changed; }
        changed.changedCG = !(cg == model.getCG());
        changed.changedSim = !(model.getSimulator().matchesAll(simulator));
        return changed;
    }
} // namespace swift::core::db