feat(scripting/database): QueryParams

Author: skuzzis

src/plugins/core/scripting.h

@@ -334,6 +334,7 @@ class PluginDatabase
     std::string EscapeString(std::string str);
 
     void QueryLua(std::string query, luabridge::LuaRef callback, lua_State *L);
+    void QueryParamsLua(std::string query, std::map<luabridge::LuaRef, luabridge::LuaRef> params, luabridge::LuaRef callback, lua_State *L);
 };
 
 //////////////////////////////////////////////////////////////

src/plugins/core/scripting/database.cpp

@@ -137,4 +137,28 @@ void PluginDatabase::QueryLua(std::string query, luabridge::LuaRef callback, lua
         dbThreadStarted = true;
         std::thread(DatabaseQueryThread).detach();
     }
+}
+
+void PluginDatabase::QueryParamsLua(std::string query, std::map<luabridge::LuaRef, luabridge::LuaRef> params, luabridge::LuaRef callback, lua_State *L)
+{
+    bool has_ats = (query.find_first_of("@") != std::string::npos);
+    bool has_brace = (query.find_first_of("{") != std::string::npos);
+    bool has_bracket = (query.find_first_of("[") != std::string::npos);
+
+    for (auto it = params.begin(); it != params.end(); ++it)
+    {
+        std::string key = it->first.tostring();
+        std::string value = EscapeString(it->second.tostring());
+
+        if (has_ats)
+            query = replace(query, "@" + key, value);
+
+        if (has_brace)
+            query = replace(query, "{" + key + "}", value);
+
+        if (has_bracket)
+            query = replace(query, "[" + key + "]", value);
+    }
+
+    return QueryLua(query, callback, L);
 }

src/plugins/lua/scripting/database.cpp

@@ -8,6 +8,7 @@ void SetupLuaDatabase(LuaPlugin *plugin, lua_State *state)
         .addFunction("IsConnected", &PluginDatabase::IsConnected)
         .addFunction("EscapeString", &PluginDatabase::EscapeString)
         .addFunction("Query", &PluginDatabase::QueryLua)
+        .addFunction("QueryParams", &PluginDatabase::QueryParamsLua)
         .endClass();
 
     luabridge::setGlobal(state, luabridge::LuaRef(state), "db");