Updates

1 files changed, 160 insertions, 0 deletions

diff --git a/gamemode/core/database/sv_setup.lua b/gamemode/core/database/sv_setup.lua
new file mode 100644
index 0000000..0d0eb74
--- /dev/null
+++ b/gamemode/core/database/sv_setup.lua
@@ -0,0 +1,160 @@
+--Adds the MySQLite global
+nrequire("sv_mysqlite.lua")
+local config = nrequire("config/sv_sql.lua")
+local data = nrequire("config/sv_newplayer.lua")
+local fn = nrequire("fn.lua")
+local col = nrequire("colortheme.lua")
+local inv = nrequire("inventory/inventory.lua")
+local track = nrequire("sv_invtracker.lua")
+local sql = {}
+
+--Setup the database if it's not already
+local setup_db = [[
+CREATE TABLE IF NOT EXISTS playerdata(SteamID bigint primary key, PlayerData json, MetaData json)]]
+
+--Create a new player
+local create_player_query = [[
+INSERT INTO playerdata (`SteamID`,`PlayerData`,`MetaData`) VALUES(%.0f,%q,%q)]]
+
+--Get a player's data from the database
+local fetch_player_query = [[
+SELECT PlayerData, MetaData FROM playerdata WHERE SteamID=%.0f
+]]
+
+local save_player_query = [[
+UPDATE playerdata SET MetaData=%q PlayerData=%q WHERE SteamID=%.0f
+]]
+
+local s_fmt = function(fmt,...)
+	local args = {...}
+	fn.map(args,MySQLite.SQLStr)
+	return string.format(fmt,unpack(args))
+end
+
+local function q_fai(err,query)
+	MsgC(col.console.red,string.format("Error executing %q, error:%s",query,err))
+end
+
+local function serialize_player(ply)
+	local sdata = {}
+	local invs = {}
+	for k,v in pairs(ply.data.inventories) do
+		invs[k] = {v.Name,v:Serialize()}
+	end
+	sdata.inventories = invs
+	sdata.skills = ply.data.skills
+	sdata.quests = ply.data.quests
+	sdata.prayers = ply.data.prayers
+	return util.TableToJSON(sdata)
+end
+
+local function deserialize_player(ply,str)
+	print("Deseriailizeing player",ply," with ", str)
+	track.ClearInventories(ply)
+	local tbl = util.JSONToTable(str)
+	local invs = tbl.inventories
+	print("Inventories was", invs)
+	PrintTable(invs)
+	for k,v in pairs(invs) do
+		print("Giveing inventory",v[1],v[2])
+		track.GiveInventoryWithData(ply,v[1],v[2])
+	end
+	ply.data.skills = tbl.skills or {}
+	ply.data.quests = tbl.quests or {}
+	ply.data.prayers = tbl.prayers or {}
+	track.SendPlayerData(ply)
+end
+
+local function connect()
+	print("Connecting to the database...")
+	MySQLite.initialize(config)
+end
+hook.Add("DatabaseInitialized","setup_table",function()
+	local setup_success = function(res,li)
+		print("Set up connection to db")
+	end
+	print("Setup query:",setup_db)
+	MySQLite.query(setup_db,setup_success,q_fai)
+end)
+connect()
+
+--Retruns (PlayerData, MetaData) or nil
+function sql.GetPlayerData(ply)
+	local s64 = ply:SteamID64()
+	local q_str = s_fmt(fetch_player_query,s64)
+	local q_suc = function(res,li)
+		print("Got player's data:",res,type(res))
+		if res == nil then
+			print("Was nil, createing player data")
+			sql.CreatePlayerTable(ply)
+		else
+			PrintTable(res)
+			assert(#res == 1,"Not unique!")
+			print("Was unique!")
+			local meta = res[1].MetaData
+			local plyd = res[1].PlayerData
+			local mtbl = util.JSONToTable(meta)
+			print("About to check if we are on the right server")
+			if mtbl.lastserver ~= game.GetIPAddress() then
+				print("Connecting player to ", mtbl.lastserver, " was on ", game.GetIPAddress())
+				ply:ConCommand("connect " .. mtbl.lastserver)
+				return
+			end
+			print("We were on the right server")
+			local _,_,x,y,z = mtbl.lastlocation:find("([%d%.]+) ([%d%.]+) ([%d%.]+)")
+			local vec = {x,y,z}
+			for k,v in pairs(vec) do vec[k] = tonumber(v) end
+			print("setting player pos to")
+			PrintTable(vec)
+			ply:SetPos(Vector(unpack(vec)))
+			deserialize_player(ply,plyd)
+		end
+	end
+	print("doing query",q_str)
+	MySQLite.query(q_str,q_suc,q_fai)
+end
+
+function sql.CreatePlayerTable(ply)
+	print("Createing player table....")
+	local s64 = ply:SteamID64()
+	print("steamid was", s64)
+	local plytbl = data.newdata()
+	local plymet = data.newmeta()
+	local plydata = util.TableToJSON(plytbl)
+	local metdata = util.TableToJSON(plymet)
+	local q_str = s_fmt(create_player_query,s64,plydata,metdata)
+	local q_suc = function(res,li)
+		print("Inserted new player",ply)
+		sql.GetPlayerData(ply)
+	end
+	print("doing query", q_str)
+	MySQLite.query(q_str,q_suc,q_fai)
+end
+
+function sql.SendPlayerToInstance(ply,ls,ll)
+	local s64 = ply:SteamID64()
+	local plydata = serialize_player(ply)
+	local plymeta = util.TableToJSON({
+		lastserver = ls,
+		lastlocation = ll
+	})
+	local q_str = s_fmt(save_player_query,plymeta,plydata,s64)
+	local q_suc = function(res,li)
+		print("Successfully saved player data")
+	end
+	MySQLite.query(q_str,q_suc,q_fai)
+end
+
+concommand.Add("DoQuery",function(ply,cmd,args)
+	if args[1] == "create" then
+		sql.CreatePlayerTable(ply)
+	elseif args[1] == "get" then
+		sql.GetPlayerData(ply)
+	elseif args[1] == "send" then
+		sql.SendPlayerToInstance(ply,args[2],args[3])
+	else
+		error("Command not understood:" .. args[1] .. "!")
+	end
+end)
+
+return sql