1 files changed, 71 insertions, 0 deletions
diff --git a/sourcemod/scripting/gokz-localdb/db/setup_map.sp b/sourcemod/scripting/gokz-localdb/db/setup_map.sp
new file mode 100644
index 0000000..a02e2d2
--- /dev/null
+++ b/sourcemod/scripting/gokz-localdb/db/setup_map.sp
@@ -0,0 +1,71 @@
+/*
+	Inserts the map information into the database.
+	Retrieves the MapID of the map and stores it in a global variable.
+*/
+
+
+
+void DB_SetupMap()
+{
+	gB_MapSetUp = false;
+	
+	char query[1024];
+	
+	char map[PLATFORM_MAX_PATH];
+	GetCurrentMapDisplayName(map, sizeof(map));
+
+	char escapedMap[PLATFORM_MAX_PATH * 2 + 1];
+	SQL_EscapeString(gH_DB, map, escapedMap, sizeof(escapedMap));
+
+	Transaction txn = SQL_CreateTransaction();
+	
+	// Insert/Update map into database
+	switch (g_DBType)
+	{
+		case DatabaseType_SQLite:
+		{
+			// UPDATE OR IGNORE
+			FormatEx(query, sizeof(query), sqlite_maps_update, escapedMap);
+			txn.AddQuery(query);
+			// INSERT OR IGNORE
+			FormatEx(query, sizeof(query), sqlite_maps_insert, escapedMap);
+			txn.AddQuery(query);
+		}
+		case DatabaseType_MySQL:
+		{
+			// INSERT ... ON DUPLICATE KEY ...
+			FormatEx(query, sizeof(query), mysql_maps_upsert, escapedMap);
+			txn.AddQuery(query);
+		}
+	}
+	// Retrieve mapID of map name
+	FormatEx(query, sizeof(query), sql_maps_findid, escapedMap, escapedMap);
+	txn.AddQuery(query);
+
+	SQL_ExecuteTransaction(gH_DB, txn, DB_TxnSuccess_SetupMap, DB_TxnFailure_Generic, 0, DBPrio_High);
+}
+
+public void DB_TxnSuccess_SetupMap(Handle db, any data, int numQueries, Handle[] results, any[] queryData)
+{
+	switch (g_DBType)
+	{
+		case DatabaseType_SQLite:
+		{
+			if (SQL_FetchRow(results[2]))
+			{
+				gI_DBCurrentMapID = SQL_FetchInt(results[2], 0);
+				gB_MapSetUp = true;
+				Call_OnMapSetup();
+			}
+		}
+		case DatabaseType_MySQL:
+		{
+			if (SQL_FetchRow(results[1]))
+			{
+				gI_DBCurrentMapID = SQL_FetchInt(results[1], 0);
+				gB_MapSetUp = true;
+				Call_OnMapSetup();
+			}
+		}
+	}
+} 
+\ No newline at end of file