openPilotLog/src/classes/jsonhelper.cpp

/*
 *openPilotLog - A FOSS Pilot Logbook Application
 *Copyright (C) 2020-2023 Felix Turowsky
 *
 *This program is free software: you can redistribute it and/or modify
 *it under the terms of the GNU General Public License as published by
 *the Free Software Foundation, either version 3 of the License, or
 *(at your option) any later version.
 *
 *This program is distributed in the hope that it will be useful,
 *but WITHOUT ANY WARRANTY; without even the implied warranty of
 *MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *GNU General Public License for more details.
 *
 *You should have received a copy of the GNU General Public License
 *along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
#include "jsonhelper.h"
#include "src/database/database.h"
#include "src/classes/paths.h"

void JsonHelper::exportDatabase()
{
    for (const auto &table : TABLES){
        QJsonArray array;
        const auto rows = DB->getTable(table);

        for (const auto &row : rows)
            array.append(QJsonObject::fromVariantHash(row));

        QJsonDocument doc(array);
        writeDocToFile(doc, OPL::GLOBALS->getDbTableName(table) + QLatin1String(".json"));
    }
}

void JsonHelper::importDatabase()
{
    TODO << "Another function should do some checking here if data exists etc...";
    TODO << "Make the function take a list of files/fileinfo as arguments";
    // clear tables
    QSqlQuery q;
    // make sure flights is cleared first due to foreign key contstraints
    q.prepare(QStringLiteral("DELETE FROM FLIGHTS"));
    q.exec();
    for (const auto & table : TABLES) {
        const QString table_name = OPL::GLOBALS->getDbTableName(table);
        q.prepare(QLatin1String("DELETE FROM ") + table_name);
        q.exec();
        const auto doc = readFileToDoc(OPL::Paths::filePath(OPL::Paths::Templates,
                                                               table_name + QLatin1String(".json")));
        DB->commit(doc.array(), table);
    }
}

QJsonDocument JsonHelper::readFileToDoc(const QString &file_path)
{
    QFile file(file_path);
    file.open(QIODevice::ReadOnly | QIODevice::Text);
    QString raw = file.readAll();
    file.close();

    QJsonDocument doc = QJsonDocument::fromJson(raw.toUtf8());
    return doc;
}

void JsonHelper::writeDocToFile(const QJsonDocument &doc, const QString &file_name)
{
    QFile out(OPL::Paths::filePath(OPL::Paths::Export,file_name));
    out.open(QFile::WriteOnly);
    out.write(doc.toJson());
    out.close();
}