From 7b6636720cd9f598a52fb9851a7c340c5ba844eb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lie=20Delhaie?= <roboform@hotmail.fr>
Date: Mon, 17 Oct 2022 20:01:27 +0200
Subject: [PATCH] Add backup, fix upgrade v4

---
 main.cpp             | 42 +++++++++++++++++++++++++++++++++++-------
 version4upgrader.cpp |  2 ++
 2 files changed, 37 insertions(+), 7 deletions(-)

diff --git a/main.cpp b/main.cpp
index c05559c..83b98dc 100644
--- a/main.cpp
+++ b/main.cpp
@@ -4,6 +4,7 @@
 #include <QJsonDocument>
 #include <QJsonObject>
 #include <QJsonArray>
+#include <QDateTime>
 #include <iostream>
 
 #include "version2upgrader.h"
@@ -11,25 +12,51 @@
 #include "version4upgrader.h"
 
 #define SAVE_FILENAME "data.json"
+#define BACKUP_FILENAME "backup_%1.json"
 
 
 
-QString get_save_file_path() {
+QString get_save_file_path(QString name) {
     QString path = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation);
     if (!QDir(path).exists()) {
         QDir().mkpath(path);
     }
     path += "/data/";
     if (!QDir(path).exists()) {
-        QDir().mkpath(path);
+        std::cerr << "[ERROR] Chronos data folder not found. Chronos never ran on this computer ?????" << std::endl;
+        exit(-1);
     }
-    path += SAVE_FILENAME;
+    path += name;
     return path;
 }
 
+void make_backup() {
+    std::cout << "[INFO] Making backup..." << std::endl;
+    QFile* file = new QFile(get_save_file_path(SAVE_FILENAME));
+    bool res = file->open(QIODevice::ReadOnly);
+    if (!res) {
+        std::cerr << "[ERROR] Chronos data file not found. Chronos never ran on this computer ?????" << std::endl;
+        exit(-1);
+    }
+    QString backup_file_name = QString(BACKUP_FILENAME).arg(QDateTime::currentSecsSinceEpoch());
+    QFile *f = new QFile(get_save_file_path(backup_file_name));
+    if (f->open(QIODevice::WriteOnly)) {
+        f->write(file->readAll());
+        f->close();
+    }
+    file->close();
+    delete file;
+    delete f;
+    std::cout << "[INFO] Backup was created on " << backup_file_name.toStdString() << std::endl;
+}
+
 QJsonObject open_save() {
-    QFile* file = new QFile(get_save_file_path());
-    file->open(QIODevice::ReadOnly);
+    QFile* file = new QFile(get_save_file_path(SAVE_FILENAME));
+    bool res = file->open(QIODevice::ReadOnly);
+    if (!res) {
+        std::cerr << "[ERROR] Chronos data file not found. Chronos never ran on this computer ?????" << std::endl;
+        exit(-1);
+    }
     auto json = QString(file->readAll());
     file->close();
     delete file;
@@ -38,7 +65,7 @@ QJsonObject open_save() {
 
 void save_to_file(QJsonObject save) {
     QJsonDocument doc(save);
-    QFile *f = new QFile(get_save_file_path());
+    QFile *f = new QFile(get_save_file_path(SAVE_FILENAME));
     if (f->open(QIODevice::WriteOnly)) {
         f->write(doc.toJson());
         f->close();
@@ -48,9 +75,10 @@ void save_to_file(QJsonObject save) {
 
 int main(int argc, char *argv[])
 {
-    std::cout << "Chronos Save Upgrader v2.1" << std::endl;
+    std::cout << "Chronos Save Upgrader v2.1.1" << std::endl;
     std::cout << "by Aurélie Delhaie - https://github.com/mojitaurelie/chronos-save-updater" << std::endl << std::endl;
     QCoreApplication::setApplicationName("Chronos");
+    make_backup();
     QJsonObject save = open_save();
     switch (save["version"].toInt()) {
         case 1: {
diff --git a/version4upgrader.cpp b/version4upgrader.cpp
index 1a02db4..95c20b2 100644
--- a/version4upgrader.cpp
+++ b/version4upgrader.cpp
@@ -29,6 +29,7 @@ QJsonObject Version4Upgrader::UpgradeTemplate(QJsonObject week)
     newTemplate["wednesday"] = UpgradeDayTemplate(week["wednesday"].toObject());
     newTemplate["thurday"] = UpgradeDayTemplate(week["thurday"].toObject());
     newTemplate["friday"] = UpgradeDayTemplate(week["friday"].toObject());
+    newTemplate["timeDelta"] = week["timeDelta"].toInt();
     newTemplate["weekNumber"] = week["weekNumber"].toInt();
     return newTemplate;
 }
@@ -44,6 +45,7 @@ QJsonArray Version4Upgrader::UpgradeWeeks(QJsonArray weeks)
         newWeek["wednesday"] = UpgradeDay(week["wednesday"].toObject());
         newWeek["thurday"] = UpgradeDay(week["thurday"].toObject());
         newWeek["friday"] = UpgradeDay(week["friday"].toObject());
+        newWeek["timeDelta"] = week["timeDelta"].toInt();
         newWeek["weekNumber"] = week["weekNumber"].toInt();
         newWeeks.append(newWeek);
     }