Upload

lua/sam/libs/sql/databases/mysql.lua

@@ -0,0 +1,191 @@
+--[[
+| 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/
+--]]
+
+local _SQL = sam.SQL
+local _error = _SQL.Error
+local traceback = debug.traceback
+
+local _mysqloo, database = nil, nil
+
+local SQL = {}
+
+function SQL.Connect(callback, failed_callback, config)
+	if database then
+		local status = database:status()
+		if status == _mysqloo.DATABASE_CONNECTING or status == _mysqloo.DATABASE_CONNECTED then
+			return true
+		end
+	end
+
+	_SQL.SetConnected(false)
+
+	require("mysqloo")
+
+	if not mysqloo then
+		_error("mysqloo module doesn't exist, get it from https://github.com/FredyH/MySQLOO")
+		return false
+	end
+
+	_mysqloo = mysqloo
+
+	database = _mysqloo.connect(
+		config.Host,
+		config.Username,
+		config.Password,
+		config.Database,
+		config.Port
+	)
+
+	function database.onConnected()
+		callback()
+	end
+
+	function database.onConnectionFailed(_, error_text)
+		failed_callback(error_text)
+	end
+
+	database:connect()
+
+	return true
+end
+
+--
+--
+--
+local transaction
+
+local add_transaction = function(query)
+	transaction:addQuery(database:query(query))
+end
+
+function SQL.Begin()
+	transaction = database:createTransaction()
+	return add_transaction
+end
+
+function SQL.Commit(callback)
+	transaction.SQL_traceback = traceback("", 2)
+
+	transaction.onSuccess = callback
+	transaction.onError = transaction_onError
+
+	transaction:start()
+
+	transaction = nil
+end
+--
+--
+--
+
+--
+--
+--
+local on_query_success = function(query, data)
+	if query.SQL_first_row then
+		data = data[1]
+	end
+	query.SQL_callback(data, query.SQL_callback_obj)
+end
+
+local on_query_fail = function(query, error_text)
+	local status = database:status()
+
+	-- https://github.com/Kamshak/LibK/blob/master/lua/libk/server/sv_libk_database.lua#L129
+	if status == _mysqloo.DATABASE_NOT_CONNECTED or status == _mysqloo.DATABASE_CONNECTING or error_text:find("Lost connection to MySQL server during query", 1, true) then
+		_SQL.SetConnected(false)
+		SQL.Query(query.SQL_query_string, query.SQL_callback, query.SQL_first_row, query.SQL_callback_obj)
+	else
+		-- 3cb9b992975d0cc0ba1b28f92ab5d1b700a08080a59b058f1424736060a73552
+		_error("Query error: " .. error_text, query.SQL_traceback)
+	end
+end
+
+function SQL.Query(query, callback, first_row, callback_obj)
+	local status = database:status()
+	if status == _mysqloo.DATABASE_NOT_CONNECTED or status == _mysqloo.DATABASE_INTERNAL_ERROR then
+		_SQL.Connect()
+		database:wait()
+	end
+
+	local query_string = query
+	query = database:query(query)
+
+	query.SQL_query_string = query_string
+
+	if callback then
+		query.onSuccess = on_query_success
+		query.SQL_callback = callback
+		query.SQL_first_row = first_row
+		query.SQL_callback_obj = callback_obj
+	end
+
+	query.SQL_traceback = traceback("", 2)
+	query.onError = on_query_fail
+
+	query:start()
+
+	return query
+end
+
+-- local prepared_set_values = function(prepared_query, values)
+--	 for i = 1, #values do
+--		 local v = values[i]
+--		 local value_type = type(v)
+--		 if value_type == "string" then
+--			 prepared_query:setString(i, v)
+--		 elseif value_type == "number" then
+--			 prepared_query:setNumber(i, v)
+--		 else
+--			 error(
+--				 string.format(
+--					 "%s invalid type '%s' was passed to escape '%s'",
+--					 "(" .. SQL.GetAddonName() .. " | MySQL)",
+--					 value_type,
+--					 v
+--				 )
+--			 )
+--		 end
+--	 end
+-- end
+
+-- function SQL.Prepare(query, callback, first_row, callback_obj)
+--	 local prepared_query = database:prepare(query)
+--	 prepared_query.SetValues = prepared_set_values
+
+--	 if callback then
+--		 prepared_query.onSuccess = on_query_success
+--		 prepared_query.SQL_callback = callback
+--		 prepared_query.SQL_first_row = first_row
+--		 prepared_query.SQL_callback_obj = callback_obj
+--	 end
+
+--	 prepared_query.SQL_traceback = traceback("", 2)
+--	 prepared_query.onError = on_query_fail
+
+--	 return prepared_query
+-- end
+
+--
+--
+--
+
+function SQL.EscapeString(value, no_quotes)
+	if no_quotes then
+		return database:escape(value)
+	else
+		return "'" .. database:escape(value) .. "'"
+	end
+end
+
+function SQL.TableExistsQuery(name)
+	return "SHOW TABLES LIKE " .. SQL.EscapeString(name)
+end
+
+return SQL

lua/sam/libs/sql/databases/sqlite.lua

@@ -0,0 +1,148 @@
+--[[
+| 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/
+--]]
+
+local _SQL = sam.SQL
+local _error = sam.SQL.Error
+local sql_query = sql.Query
+
+local SQL = {}
+
+function SQL.Connect(callback)
+	timer.Simple(0, callback)
+	return true
+end
+
+--
+--
+--
+local transactions
+
+local add_transaction = function(query)
+	table.insert(transactions, query)
+end
+
+function SQL.Begin()
+	transactions = {}
+	sql_query("BEGIN TRANSACTION")
+	return add_transaction
+end
+
+function SQL.Commit(callback)
+	for i = 1, #transactions do
+		if sql_query(transactions[i]) == false then
+			sql_query("ROLLBACK TRANSACTION")
+			transactions = nil
+			_error("Transaction error: " .. sql.LastError())
+			return
+		end
+	end
+
+	transactions = nil
+
+	sql_query("COMMIT TRANSACTION")
+
+	if callback then
+		callback()
+	end
+end
+--
+--
+--
+
+--
+--
+--
+local query_obj = {
+	wait = function() end -- mysqloo has query:wait()
+}
+
+function SQL.Query(query, callback, first_row, callback_obj)
+	local data = sql_query(query)
+	if data == false then
+		_error("Query error: " .. sql.LastError())
+	elseif callback then
+		if data == nil then
+			if not first_row then
+				data = {}
+			end
+		elseif first_row then
+			data = data[1]
+		end
+
+		callback(data, callback_obj)
+	end
+
+	return query_obj
+end
+
+-- local concat = table.concat
+-- local prepared_set_values = function(prepared_query, values)
+--	 for i = 1, prepared_query.args_n do
+--		 prepared_query[prepared_query[-i]] = _SQL.Escape(values[i])
+--	 end
+--	 return concat(prepared_query, "", 1, prepared_query.n)
+-- end
+
+-- local prepared_start = function()
+-- end
+
+-- local sub, find = string.sub, string.find
+-- function SQL.Prepare(query, callback, first_row, callback_obj)
+--	 local prepared_query = {}
+--	 prepared_query.wait = query_obj.wait
+--	 prepared_query.Start = prepared_start
+--	 prepared_query.SetValues = prepared_set_values
+
+--	 local count, args_n = 0, 0
+--	 local pos, start, _end = 0, nil, 0
+--	 while true do
+--		 start, _end = find(query, "?", _end + 1, true)
+
+--		 if not start then
+-- 			break
+-- 		end
+
+--		 if pos ~= start then
+--			 count = count + 1; prepared_query[count] = sub(query, pos, start - 1)
+-- 		end
+
+--		 count = count + 1; prepared_query[count] = "NULL"
+--		 args_n = args_n - 1; prepared_query[args_n] = count
+
+--		 pos = _end + 1
+--	 end
+
+--	 if pos <= #query then
+--		 count = count + 1; prepared_query[count] = sub(query, pos)
+-- 	end
+
+--	 prepared_query.n = count
+--	 prepared_query.args_n = abs(args_n)
+
+--	 return prepared_query
+-- end
+--
+--
+--
+
+local SQLStr = SQLStr
+function SQL.EscapeString(value, no_quotes)
+	return SQLStr(value, no_quotes)
+end
+
+function SQL.TableExistsQuery(name)
+	return "SELECT `name` FROM `sqlite_master` WHERE `name` = " .. SQL.EscapeString(name) .. " AND `type` = 'table'"
+end
+
+--
+--
+--
+
+return SQL