require "shared.init"
require "utils.init"
require "GlobalVar"
require "RedisKeys"
require "ProtocolCode"
require "skynet.manager"
require "utils.MysqlUtil"

skynet = require "skynet"

redisproxy = require("shared.redisproxy")
mysqlproxy = require "shared.mysqlproxy"

SendPacket = function ( ... ) end


local function initRedisDb( ... )
	local function initAutoIncrementUid(tbName, keyName, fieldName)
		if not fieldName then fieldName = "value" end
		local mysqlVal = getDbCfgVal(tbName, keyName, fieldName)
		if not mysqlVal then 
			skynet.error(string.format("get db cfg fail, table %s, key %s, field %s", tbName, keyName, fieldName))
			return
		end
		local redisVal = tonum(redisproxy:hget(tbName, keyName))
		if redisVal < mysqlVal then
			redisproxy:hset(tbName, keyName, mysqlVal)
		end
	end

	initAutoIncrementUid("autoincrement_set", "role")
	initAutoIncrementUid("autoincrement_set", "union")
	initAutoIncrementUid("autoincrement_set", "order")
	initAutoIncrementUid("autoincrement_set", "email")
	initAutoIncrementUid("autoincrement_set", "emailTimestamp")
	initAutoIncrementUid("autoincrement_set", "delay_email")
	initAutoIncrementUid("autoincrement_set", "stopcreate")
	initAutoIncrementUid("autoincrement_set", "maintain")

	--redisproxy:hsetnx("adv_season", "idx", 0)
	--redisproxy:hsetnx("adv_season", "chapter", globalCsv.adv_endless_default_chapter)
	--redisproxy:hsetnx("adv_season", "overTime", 0)
end

-- 初始化服务器数据库以及服务器信息表
local function initServerDatabase()
	local servId = skynet.getenv("servId")
	mysqlproxy:query(string.format("CREATE DATABASE IF NOT EXISTS server_%s DEFAULT CHARSET = utf8mb4 COLLATE utf8mb4_general_ci;", servId))
	mysqlproxy:query(string.format("use server_%s", servId))

	-- 服务器信息表 开服时间
	mysqlproxy:query [[
		CREATE TABLE IF NOT EXISTS `server_info` (
			`key` varchar(45) NOT NULL,
			`int_value` int(11) DEFAULT NULL,
			`str_value` varchar(128) DEFAULT NULL,
			PRIMARY KEY (`key`)
			) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
	]]

	local res = mysqlproxy:query("SELECT * FROM `server_info` where `key` = 'server_start';")
	if not next(res) then
		mysqlproxy:query(string.format("INSERT INTO `server_info`(`key`, `str_value`) VALUES('server_start', '%s');",
			os.date("%Y%m%d", skynet.timex())))
	end
end

local function initAutoIncreUidTable()
	mysqlproxy:query [[
		CREATE TABLE IF NOT EXISTS `autoincrement_set` (
			`key` varchar(45) NOT NULL,
			`value` int(11) DEFAULT NULL,
			PRIMARY KEY (`key`)
			) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
	]]
	local servId = tonumber(skynet.getenv("servId"))
	if servId then
		local tpl = "INSERT INTO `autoincrement_set`(`key`, `value`) values('%s', %d)"
		mysqlproxy:query(string.format(tpl, "role", servId * MAX_ROLE_NUM))
		mysqlproxy:query(string.format(tpl, "union", servId * MAX_ROLE_NUM))
		mysqlproxy:query(string.format(tpl, "order", 0))
		mysqlproxy:query(string.format(tpl, "email", 0))
		mysqlproxy:query(string.format(tpl, "emailTimestamp", 0))
		mysqlproxy:query(string.format(tpl, "delay_email", 0))
		mysqlproxy:query(string.format(tpl, "stopcreate", 0))
		mysqlproxy:query(string.format(tpl, "maintain", 0))
	end
end

local function initAdvSeasonTable()
	mysqlproxy:query [[
		CREATE TABLE IF NOT EXISTS `adv_season` (
			`key` varchar(45) NOT NULL,
			`value` int(11) DEFAULT NULL,
			PRIMARY KEY (`key`)
			) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
	]]
	local servId = tonumber(skynet.getenv("servId"))
	if servId then
		local tpl = "INSERT INTO `adv_season`(`key`, `value`) values('%s', %d)"

		mysqlproxy:query(string.format(tpl, "idx", 0))
		mysqlproxy:query(string.format(tpl, "chapter", globalCsv.adv_endless_default_chapter))
		mysqlproxy:query(string.format(tpl, "overTime", 0))
	end
end

local function checkRoleTables()
	local list = {"Role", "Daily", "Activity", "Diner", "Store", "Hero", "RoleIncre", "Rune", "Order", "Email", "Friend"}
	for _, name in ipairs(list) do
		local obj = require("models."..name).new({key = "key"})
		print("check table [" .. name .. "] begin.")
		obj:checkTableSchema()
		print("check table [" .. name .. "] end.")
	end
end

local function createMysqlSp()
	mysqlproxy:query "DROP PROCEDURE IF EXISTS `add_friends`"
	mysqlproxy:query [[
		CREATE PROCEDURE `add_friends`(IN role_id bigint, IN friend_id bigint, IN add_time int)
		BEGIN
			DECLARE t_error INTEGER DEFAULT 0;    
			DECLARE CONTINUE HANDLER FOR SQLEXCEPTION SET t_error=1;    
    
			START TRANSACTION;    
				INSERT INTO `Friend`(`roleid`,`fid`,`addTime`) VALUES(role_id, friend_id, add_time);       
				INSERT INTO `Friend`(`roleid`,`fid`,`addTime`) VALUES(friend_id, role_id, add_time);       
    
			IF t_error = 1 THEN    
				ROLLBACK;    
			ELSE    
				COMMIT;    
			END IF;    
			select t_error; 
		END
	]]

	mysqlproxy:query "DROP PROCEDURE IF EXISTS `del_friends`"
	mysqlproxy:query [[
		CREATE PROCEDURE `del_friends`(IN role_id bigint, IN friend_id bigint)
		BEGIN
			DECLARE t_error INTEGER DEFAULT 0;    
			DECLARE CONTINUE HANDLER FOR SQLEXCEPTION SET t_error=1;    
    
			START TRANSACTION;    
				DELETE FROM `Friend` WHERE `roleid` = role_id AND `fid` = friend_id;       
				DELETE FROM `Friend` WHERE `roleid` = friend_id AND `fid` = role_id;       
    
			IF t_error = 1 THEN    
				ROLLBACK;    
			ELSE    
				COMMIT;    
			END IF;    
			select t_error; 
		END
	]]
end

local steps = {
	[1] = {
		handler = initServerDatabase,
		desc = "initialize server database "
	},
	[2] = {
		handler = initAutoIncreUidTable,
		desc = "initialize auto_increment_uid table "
	},
	[3] = {
		handler = initAdvSeasonTable,
		desc = "initialize adv_season table "
	},
	[4] = {
		handler = checkRoleTables,
		desc = "check role tables "
	},
	[5] = {
		handler = createMysqlSp,
		desc = "create mysql store procedure "
	},
}

local function loadAllUserInfo()
	local maxId = 0
	local sql = "SELECT `id`, `uid`, `name` FROM `Role` WHERE `id` > %d ORDER BY `id` LIMIT 1000;"
	while true do
		local res = mysqlproxy:query(string.format(sql, maxId))
		if not next(res) then
			return
		else
			for _, info in ipairs(res) do
				if info["id"] > maxId then
					maxId = info["id"]
				end
				redisproxy:pipelining(function (red)
					local dbName = string.upper(info["name"])
					red:set(string.format("user:%s", dbName), info["id"])
					red:set(string.format("uid:%s", info["uid"]), dbName)
				end)
			end
		end
	end
end

skynet.start(function ()
	--local new = redisproxy:hsetnx("autoincrement_set", "server_start", os.date("%Y%m%d", skynet.timex())) == 1
	--if not new then
	--	print("server has been initialized...")
	--	skynet.exit()
	--	return
	--end
	csvdb = require "shared.csvdata"
	globalCsv = csvdb["GlobalDefineCsv"]

	for _, action in ipairs(steps) do
		print(action.desc .. "start ...")
		action.handler()
		print(action.desc .. "finished ...")
	end
	initRedisDb()
	loadAllUserInfo()
	skynet.exit()
end)