From 7aef973ef1923250ae19a28451b60465edb8da91 Mon Sep 17 00:00:00 2001
From: Kamil <kamilprzybspam@wp.pl>
Date: Sun, 6 Dec 2020 20:29:48 +0100
Subject: [PATCH] Add meta table to DB

---
 src/Database.cpp | 112 +++++++++++++++++++++++++++++++++++++++++++++++
 src/Database.hpp |  25 +++--------
 2 files changed, 118 insertions(+), 19 deletions(-)

diff --git a/src/Database.cpp b/src/Database.cpp
index aad8f24..557e95e 100644
--- a/src/Database.cpp
+++ b/src/Database.cpp
@@ -30,6 +30,7 @@ void Database::open() {
 
     // foreign keys are off by default
     sqlite3_exec(db, "PRAGMA foreign_keys=ON", NULL, NULL, NULL);
+    checkMetaTable();
     createTables();
 
     std::cout << "[INFO] Database in operation ";
@@ -53,7 +54,118 @@ void Database::close() {
     sqlite3_close(db);
 }
 
+void Database::checkMetaTable() {
+
+    // first check if meta table exists
+    const char* sql = R"(
+        SELECT COUNT (*) FROM sqlite_master WHERE type="table" AND name="Meta";
+        )";
+    sqlite3_stmt* stmt;
+    sqlite3_prepare_v2(db, sql, -1, &stmt, 0);
+    int rc = sqlite3_step(stmt);
+    if (rc != SQLITE_ROW) {
+        std::cout << "[FATAL] Failed to check meta table"  << std::endl;
+        terminate(0);
+    }
+
+    int count = sqlite3_column_int(stmt, 0);
+    if (count == 0) {
+        sqlite3_finalize(stmt);
+        return createMetaTable();
+    }
+
+    // check protocol version
+    sql = R"(
+        SELECT "Value" FROM "Meta" WHERE "Key" = "ProtocolVersion";
+        )";
+    sqlite3_prepare_v2(db, sql, -1, &stmt, 0);
+
+    if (sqlite3_step(stmt) != SQLITE_ROW) {
+        sqlite3_finalize(stmt);
+        std::cout << "[FATAL] Failed to check DB Protocol Version" << std::endl;
+        terminate(0);
+    }
+
+    if (sqlite3_column_int(stmt, 0) != PROTOCOL_VERSION) {
+        sqlite3_finalize(stmt);
+        std::cout << "[FATAL] DB Protocol Version doesn't match Server Build" << std::endl;
+        terminate(0);
+    }
+
+    sql = R"(
+        SELECT "Value" FROM "Meta" WHERE "Key" = "DatabaseVersion";
+        )";
+    sqlite3_prepare_v2(db, sql, -1, &stmt, 0);
+    rc = sqlite3_step(stmt);
+    
+    if (rc != SQLITE_ROW) {
+        std::cout << "[FATAL] Failed to check DB Version" << std::endl;
+        sqlite3_finalize(stmt);
+        terminate(0);
+    }
+
+    int dbVersion = sqlite3_column_int(stmt, 0);
+    sqlite3_finalize(stmt);
+
+    if (dbVersion != DATABASE_VERSION) {
+        // we should be handling migrations here in the future
+        std::cout << "[FATAL] DB Version doesn't match Server Build" << std::endl;
+        terminate(0);
+    }
+}
+
+void Database::createMetaTable() {
+    std::lock_guard<std::mutex> lock(dbCrit);
+
+    sqlite3_exec(db, "BEGIN TRANSACTION", NULL, NULL, NULL);
+
+    const char* sql = R"(
+        CREATE TABLE "Meta"(
+        "Key" TEXT NOT NULL UNIQUE,
+        "Value" INTEGER NOT NULL);
+        )";
+    sqlite3_stmt* stmt;
+    sqlite3_prepare_v2(db, sql, -1, &stmt, 0);
+    if (sqlite3_step(stmt) != SQLITE_DONE) {
+        sqlite3_finalize(stmt);
+        sqlite3_exec(db, "ROLLBACK TRANSACTION", NULL, NULL, NULL);
+        std::cout << "[FATAL] Failed to create meta table" << std::endl;
+        terminate(0);
+    }
+
+    sql = R"(
+        INSERT INTO "Meta" ("Key", "Value")
+        VALUES (?, ?);
+        )";
+    sqlite3_prepare_v2(db, sql, -1, &stmt, 0);
+    sqlite3_bind_text(stmt, 1, "ProtocolVersion", -1, 0);
+    sqlite3_bind_int(stmt, 2, PROTOCOL_VERSION);
+
+    if (sqlite3_step(stmt) != SQLITE_DONE) {
+        sqlite3_finalize(stmt);
+        sqlite3_exec(db, "ROLLBACK TRANSACTION", NULL, NULL, NULL);
+        std::cout << "[FATAL] Failed to create meta table" << std::endl;
+        terminate(0);
+    }
+
+    sqlite3_reset(stmt);
+    sqlite3_bind_text(stmt, 1, "DatabaseVersion", -1, 0);
+    sqlite3_bind_int(stmt, 2, DATABASE_VERSION);
+    int rc = sqlite3_step(stmt);
+    sqlite3_finalize(stmt);
+
+    if (rc != SQLITE_DONE) {
+        sqlite3_exec(db, "ROLLBACK TRANSACTION", NULL, NULL, NULL);
+        std::cout << "[FATAL] Failed to create meta table" << std::endl;
+        terminate(0);
+    }
+
+    sqlite3_exec(db, "END TRANSACTION", NULL, NULL, NULL);
+    std::cout << "[INFO] Created new meta table" << std::endl;
+}
+
 void Database::createTables() {
+    std::lock_guard<std::mutex> lock(dbCrit);
 
     char* errMsg = 0;
     char* sql;
diff --git a/src/Database.hpp b/src/Database.hpp
index c300844..b59f7bc 100644
--- a/src/Database.hpp
+++ b/src/Database.hpp
@@ -3,9 +3,11 @@
 #include "Player.hpp"
 #include <string>
 #include <vector>
+#include <map>
+
+#define DATABASE_VERSION 1
 
 namespace Database {
-#pragma region DatabaseStructs
 
     struct Account {
         int AccountID;
@@ -13,11 +15,6 @@ namespace Database {
         int Selected;
         time_t BannedUntil;
     };
-    struct Buddyship {
-        int PlayerAId;
-        int PlayerBId;
-        int16_t Status;
-    };
     struct EmailData {
         int PlayerId;
         int MsgIndex;
@@ -32,21 +29,11 @@ namespace Database {
         uint64_t SendTime;
         uint64_t DeleteTime;
     };
-    struct EmailItem {
-        int PlayerId;
-        int MsgIndex;
-        int Slot;
-        int16_t Type;
-        int16_t Id;
-        int32_t Opt;
-        int32_t TimeLimit;
-    };
-
-
-#pragma endregion DatabaseStructs
-
+    
     void open();
     void close();
+    void checkMetaTable();
+    void createMetaTable();
     void createTables();
     int getTableSize(std::string tableName);