added more login logic

src/database/database.cpp

@@ -36,9 +36,29 @@ bool Database::exec(const std::string& sqlQuery){
     return exec(sqlQuery.c_str());
 }
 
-map<string, string> Database::getStrMap(const string& sql){
-    sqlite3_stmt* stmt = prepareStmt(sql);
+sqlite3_stmt* Database::bind(const std::string sql, const Database::QueryData& data) {
+    sqlite3_stmt* stmt = nullptr;
+    if (sqlite3_prepare_v2(m_db, sql.c_str(), -1, &stmt, nullptr) != SQLITE_OK) {
+        CROW_LOG_ERROR << "Failed to prepare statement: " << sqlite3_errmsg(m_db);
+        return nullptr;
+    }
+    for (int i = 0; i < data.size(); i++) {
+        std::visit([stmt, i](auto&& val) {
+            using T = std::decay_t<decltype(val)>;
+            if constexpr (std::is_same_v<T, int64_t>)
+                sqlite3_bind_int64(stmt, i + 1, val);
+            else if constexpr (std::is_same_v<T, double>)
+                sqlite3_bind_double(stmt, i + 1, val);
+            else if constexpr (std::is_same_v<T, std::string>)
+                sqlite3_bind_text(stmt, i + 1, val.c_str(), -1, SQLITE_TRANSIENT);                
+        }, data[i]);
+    }
+    return stmt;
+}
+
+map<string, string> Database::getStrMap(const std::string sql, const Database::QueryData& data){
     map<string, string> map;
+    sqlite3_stmt* stmt = bind(sql, data);
     if (stmt == nullptr)
         return map; 
 
@@ -52,8 +72,8 @@ map<string, string> Database::getStrMap(const string& sql){
     return map;
 }
 
-string Database::getStr(const string& sql){
-    sqlite3_stmt* stmt = prepareStmt(sql);
+string Database::getStr(const std::string sql, const Database::QueryData& data){
+    sqlite3_stmt* stmt = bind(sql, data);
     string str;
     if (stmt == nullptr)
         return str; 

src/database/database.hpp

@@ -3,12 +3,16 @@
 
 #include "sqlite3.h"
 #include <string>
+#include <vector>
+#include <variant>
 #include <optional>
 #include <set>
 #include <map>
 
 class Database {
 
+    typedef std::vector<std::variant<int64_t, std::string>> QueryData;
+
 public:
     Database();
     ~Database();
@@ -24,9 +28,11 @@ public:
 
     std::set<std::string> getStrSet(const std::string& sql);
 
-    string getStr(const string& sql)
+    std::string getStr(const std::string sql, const QueryData& data);
 
-    std::map<std::string, std::string> getStrMap(const std::string& sql);
+    sqlite3_stmt* bind(const std::string sql, const QueryData& data);
+
+    std::map<std::string, std::string> getStrMap(const std::string sql, const QueryData& data);
 
 private:
     sqlite3_stmt* prepareStmt(const std::string& sql);