1 files changed, 99 insertions, 0 deletions
diff --git a/sourcemod/scripting/gokz-localdb/db/setup_client.sp b/sourcemod/scripting/gokz-localdb/db/setup_client.sp
new file mode 100644
index 0000000..848be87
--- /dev/null
+++ b/sourcemod/scripting/gokz-localdb/db/setup_client.sp
@@ -0,0 +1,99 @@
+/*
+	Inserts the player into the database, or else updates their information.
+*/
+
+
+
+void DB_SetupClient(int client)
+{
+	if (IsFakeClient(client))
+	{
+		return;
+	}
+	
+	// Setup Client Step 1 - Upsert them into Players Table
+	char query[1024], name[MAX_NAME_LENGTH], nameEscaped[MAX_NAME_LENGTH * 2 + 1], clientIP[16], country[45];
+	
+	int steamID = GetSteamAccountID(client);
+	if (!GetClientName(client, name, MAX_NAME_LENGTH))
+	{
+		LogMessage("Couldn't get name of %L.", client);
+		name = "Unknown";
+	}
+	SQL_EscapeString(gH_DB, name, nameEscaped, MAX_NAME_LENGTH * 2 + 1);
+	if (!GetClientIP(client, clientIP, sizeof(clientIP)))
+	{
+		LogMessage("Couldn't get IP of %L.", client);
+		clientIP = "Unknown";
+	}
+	if (!GeoipCountry(clientIP, country, sizeof(country)))
+	{
+		LogMessage("Couldn't get country of %L (%s).", client, clientIP);
+		country = "Unknown";
+	}
+	
+	DataPack data = new DataPack();
+	data.WriteCell(GetClientUserId(client));
+	data.WriteCell(steamID);
+	
+	Transaction txn = SQL_CreateTransaction();
+	
+	// Insert/Update player into Players table
+	switch (g_DBType)
+	{
+		case DatabaseType_SQLite:
+		{
+			// UPDATE OR IGNORE
+			FormatEx(query, sizeof(query), sqlite_players_update, nameEscaped, country, clientIP, steamID);
+			txn.AddQuery(query);
+			// INSERT OR IGNORE
+			FormatEx(query, sizeof(query), sqlite_players_insert, nameEscaped, country, clientIP, steamID);
+			txn.AddQuery(query);
+		}
+		case DatabaseType_MySQL:
+		{
+			// INSERT ... ON DUPLICATE KEY ...
+			FormatEx(query, sizeof(query), mysql_players_upsert, nameEscaped, country, clientIP, steamID);
+			txn.AddQuery(query);
+		}
+	}
+	
+	FormatEx(query, sizeof(query), sql_players_get_cheater, steamID);
+	txn.AddQuery(query);
+	
+	SQL_ExecuteTransaction(gH_DB, txn, DB_TxnSuccess_SetupClient, DB_TxnFailure_Generic_DataPack, data, DBPrio_High);
+}
+
+public void DB_TxnSuccess_SetupClient(Handle db, DataPack data, int numQueries, Handle[] results, any[] queryData)
+{
+	data.Reset();
+	int client = GetClientOfUserId(data.ReadCell());
+	int steamID = data.ReadCell();
+	delete data;
+	
+	if (client == 0 || !IsClientAuthorized(client))
+	{
+		return;
+	}
+	
+	switch (g_DBType)
+	{
+		case DatabaseType_SQLite:
+		{
+			if (SQL_FetchRow(results[2]))
+			{
+				gB_Cheater[client] = SQL_FetchInt(results[2], 0) == 1;
+			}
+		}
+		case DatabaseType_MySQL:
+		{
+			if (SQL_FetchRow(results[1]))
+			{
+				gB_Cheater[client] = SQL_FetchInt(results[1], 0) == 1;
+			}
+		}
+	}
+	
+	gB_ClientSetUp[client] = true;
+	Call_OnClientSetup(client, steamID, gB_Cheater[client]);
+} 
+\ No newline at end of file