Upload

lua/sam/libs/sql/sv_init.lua

@@ -0,0 +1,152 @@
+--[[
+| This file was obtained through the combined efforts
+| of Madbluntz & Plymouth Antiquarian Society.
+|
+| Credits: lifestorm, Gregory Wayne Rossel JR.,
+| 	Maloy, DrPepper10 @ RIP, Atle!
+|
+| Visit for more: https://plymouth.thetwilightzone.ru/
+--]]
+
+if SAM_LOADED then return end
+
+local format, isstring, istable, tonumber = string.format, sam.isstring, sam.istable, tonumber
+
+local config = {}
+
+local SQL, _SQL = {}, nil
+
+function SQL.Print(...)
+	MsgC(
+		Color(255, 255, 255), "(",
+		Color(244, 67, 54), SQL.GetAddonName(),
+		Color(255, 255, 255), " | ",
+		Color(244, 67, 54), SQL.IsMySQL() and "MySQL" or "SQLite",
+		Color(255, 255, 255), ") ",
+		...
+	)
+	Msg("\n")
+end
+
+function SQL.Error(err, trace)
+	SQL.Print(err, trace or debug.traceback("", 2))
+end
+
+function SQL.Connect()
+	return _SQL.Connect(SQL.OnConnected, SQL.OnConnectionFailed, config)
+end
+
+do
+	local in_transaction, old_query = false, nil
+
+	function SQL.Begin()
+		if in_transaction then
+			return SQL.Error("transaction on going!")
+		end
+		in_transaction = true
+
+		SQL.Query, old_query = _SQL.Begin(), SQL.Query
+	end
+
+	function SQL.Commit(callback)
+		if not in_transaction then return end
+
+		in_transaction = false
+		SQL.Query, old_query = old_query, nil
+
+		return _SQL.Commit(callback)
+	end
+end
+
+local gsub = string.gsub
+function SQL.FQuery(query, args, callback, first_row, callback_obj)
+	query = gsub(query, "{(%d)(f?)}", function(i, no_escape)
+		return SQL.Escape(args[tonumber(i)], no_escape ~= "")
+	end)
+
+	return SQL.Query(query, callback, first_row, callback_obj)
+end
+
+do
+	local table_exists = function(data, callback)
+		callback(data and true or false)
+	end
+
+	function SQL.TableExists(name, callback)
+		return SQL.Query(_SQL.TableExistsQuery(name), table_exists, true, callback)
+	end
+end
+
+function SQL.IsMySQL()
+	return config.MySQL == true
+end
+
+do
+	local connected = false
+	function SQL.IsConnected()
+		return connected
+	end
+	function SQL.SetConnected(is_connected)
+		connected = is_connected
+	end
+end
+
+function SQL.Escape(value, no_quotes)
+	local value_type = type(value)
+	if value_type == "string" then
+		return _SQL.EscapeString(value, no_quotes)
+	elseif value_type == "number" then
+		return value
+	else
+		error(
+			format(
+				"%s invalid type '%s' was passed to escape '%s'",
+				"(" .. SQL.GetAddonName() .. " | " .. (SQL.IsMySQL() and "MySQL" or "SQLite") .. ")",
+				value_type,
+				value
+			)
+		)
+	end
+end
+
+function SQL.OnConnected()
+	SQL.SetConnected(true)
+	hook.Call(SQL.GetAddonName() .. ".DatabaseConnected")
+end
+
+function SQL.OnConnectionFailed(error_text)
+	SQL.Error("Failed to connect to the server: " .. error_text)
+	hook.Call(SQL.GetAddonName() .. ".DatabaseConnectionFailed", nil, error_text)
+end
+
+function SQL.SetConfig(new_config)
+	if not istable(new_config) then return end
+	if new_config.MySQL == true then
+		for _, v in ipairs({"Host", "Username", "Password", "Database"}) do
+			if not isstring(new_config[v]) then
+				return SQL.Error(
+					format("config value for '%s' is invalid '%s' needs to be a string!", v, config[v])
+				)
+			end
+		end
+		new_config.Port = tonumber(new_config.Port) or 3306
+		_SQL = sam.load_file("sam/libs/sql/databases/mysql.lua", "sv_")
+	else
+		_SQL = sam.load_file("sam/libs/sql/databases/sqlite.lua", "sv_")
+	end
+
+	SQL.Query = _SQL.Query
+	config = new_config
+end
+
+do
+	local addon_name = "NO NAME"
+	function SQL.SetAddonName(name)
+		addon_name = name
+	end
+	function SQL.GetAddonName()
+		return addon_name
+	end
+end
+
+sam.SQL = SQL