--  
-- NOQ installer noq_i.lua - as part of the NOQ
--

--
-- Remove this script from game server path after installation
--

--------------------------------------------------------------------------------
color = "^5"
version = "1"
commandprefix = "!"
debug = 1 -- debug 0/1
tablespacer = " " -- use something like " " or "|"
--------------------------------------------------------------------------------
env = nil
con = nil

res = {}

fs_game 		= et.trap_Cvar_Get("fs_game")
homepath 		= et.trap_Cvar_Get("fs_homepath")
scriptpath 		= homepath .. "/" .. fs_game .. "/noq/" -- full qualified path for the NOQ scripts

-------------------------------------------------------------------------------
-- table functions - don't move down or edit!
-------------------------------------------------------------------------------

-- TODO: we use same functions in the noq.lua
-- Find a way to use more centralized

-- The table load 
function table.load( sfile )
   -- catch marker for stringtable
   if string.sub( sfile,-3,-1 ) == "--|" then
	  tables,err = loadstring( sfile )
   else
	  tables,err = loadfile( sfile )
   end
   if err then return _,err
   end
   tables = tables()
   for idx = 1,#tables do
	  local tolinkv,tolinki = {},{}
	  for i,v in pairs( tables[idx] ) do
		 if type( v ) == "table" and tables[v[1]] then
			table.insert( tolinkv,{ i,tables[v[1]] } )
		 end
		 if type( i ) == "table" and tables[i[1]] then
			table.insert( tolinki,{ i,tables[i[1]] } )
		 end
	  end
	  -- link values, first due to possible changes of indices
	  for _,v in ipairs( tolinkv ) do
		 tables[idx][v[1]] = v[2]
	  end
	  -- link indices
	  for _,v in ipairs( tolinki ) do
		 tables[idx][v[2]],tables[idx][v[1]] =  tables[idx][v[1]],nil
	  end
   end
   return tables[1]
end

-- Gets varvalue else null
function getConfig( varname )
	local value = noqvartable[varname]
	
	if value then
	  	return value
	else
		et.G_Print("warning, invalid config value for " .. varname .. "\n")
	  	return "null"
	end
end

et.G_LogPrint("Loading NOQ config from ".. scriptpath.."\n")
noqvartable		= assert(table.load( scriptpath .. "noq_config.cfg"))

--------------------------------------------------------------------------------

-- Handle different dbms
if getConfig("dbms") == "mySQL" then
	require "luasql.mysql"
	env = assert( luasql.mysql() )
	con = assert( env:connect(getConfig("dbname"), getConfig("dbuser"), getConfig("dbpassword"), getConfig("dbhostname"), getConfig("dbport")) )
elseif getConfig("dbms") == "SQLite" then
	require "luasql.sqlite3" 
	env = assert( luasql.sqlite3() )
	-- this opens OR creates a sqlite db - if this file is loaded db is created -fix this?
	con = assert( env:connect( getConfig("dbname") ) )
else
  -- stop script
  error("DBMS not supported.")
end


--------------------------------------------------------------------------------

function et_InitGame( levelTime, randomSeed, restart )
	et.trap_SendServerCommand( -1 ,"chat \"" .. color .. "NOQ install " .. version ) -- keep this message so admins know the script is up & running
	et.RegisterModname( "NOQ install " .. version .. " " .. et.FindSelf() )
end

function et_ConsoleCommand( command )
	if debug == 1 then
	  et.trap_SendServerCommand( -1 ,"chat \"" .. color .. "ConsoleCommand - command: " .. command )
	end
	
	if string.lower(et.trap_Argv(0)) == commandprefix.."sqlcreate" then 
		createTablesDBMS()
	elseif string.lower(et.trap_Argv(0)) == commandprefix.."sqlupdate" then 
		updateTablesDBMS()
	elseif string.lower(et.trap_Argv(0)) == commandprefix.."sqldrop" then 
		dropTablesDBMS()
	elseif string.lower(et.trap_Argv(0)) == commandprefix.."sqlclean" then
		-- drop all tables 
		cleanTablesDBMS()
	end
	-- add more cmds here ...
end

-- 
function cleanTablesDBMS()
	res = assert(con:execute"delete from player")
	et.G_Print(res .. "\n")
	res = assert(con:execute"delete from session")
	et.G_Print(res .. "\n")
	res = assert(con:execute"delete from log")
	et.G_Print(res .. "\n")
end

-- For future versions if the db structure does exist
function updateTablesDBMS()
	-- alter tables ...
end

function dropTablesDBMS()
	res = assert(con:execute"DROP TABLE session")
	et.G_Print(res .. "\n")
	res = assert(con:execute"DROP TABLE player")
	et.G_Print(res .. "\n")
	res = assert(con:execute"DROP TABLE log")
	et.G_Print(res .. "\n")
	res = assert(con:execute"DROP TABLE level")
	et.G_Print(res .. "\n")
	res = assert(con:execute"DROP TABLE version")
	et.G_Print(res .. "\n")
end

function createTablesDBMS()
		-- IMPORTANT NOTES for default field values:
		-- Mandatory fields to create a table are set as NOT NULL
		-- A non existing time is NULL
		-- If you add more fields create usefull default values ...

		et.G_Print(color .. commandprefix.."sqlcreate for ".. getConfig("dbms") .." started\n") 
	
		-- SQLite
		if getConfig("dbms") == "SQLite" then
			
			-- Notes:
			-- We store timestamps as INTEGER - Unix Time, the number of seconds since 1970-01-01 00:00:00 UTC. 
			
			res = assert(con:execute"CREATE TABLE IF NOT EXISTS player ( 		\
				id 		INTEGER 		PRIMARY KEY,		\
				pkey 		TEXT 			UNIQUE NOT NULL,	\
				conname 	TEXT 			NOT NULL,		\
				regname 	TEXT 			DEFAULT '',		\
				netname 	TEXT 			DEFAULT '',		\
				cleanname 	TEXT 			DEFAULT '',		\
				isBot 		INTEGER 		DEFAULT 0,		\
				clan 		TEXT 			DEFAULT '',		\
				level 		INTEGER 		DEFAULT 0,		\
				flags 		TEXT 			DEFAULT '',		\
				user 		TEXT 			DEFAULT '',		\
				password 	TEXT 			DEFAULT '',		\
				email 		TEXT 			DEFAULT '',		\
				xp0 		INTEGER 		DEFAULT 0,		\
				xp1 		INTEGER 		DEFAULT 0,		\
				xp2 		INTEGER 		DEFAULT 0,		\
				xp3 		INTEGER 		DEFAULT 0,		\
				xp4 		INTEGER 		DEFAULT 0,		\
				xp5 		INTEGER 		DEFAULT 0,		\
				xp6 		INTEGER 		DEFAULT 0,		\
				xptot		INTEGER 		DEFAULT 0,		\
				banreason 	TEXT 			DEFAULT '',		\
				bannedby 	TEXT 			DEFAULT '',		\
				banexpire 	DATE 			DEFAULT '1000-01-01 00:00:00',	\
				mutedreason 	TEXT			DEFAULT '',		\
				mutedby 	TEXT 			DEFAULT '',		\
				muteexpire 	DATE 			DEFAULT '1000-01-01 00:00:00',	\
				warnings 	INTEGER 		DEFAULT 0,		\
				suspect 	INTEGER 		DEFAULT 0,		\
				regdate 	DATE 			DEFAULT NULL,		\
				updatedate 	DATE 			DEFAULT CURRENT_DATE,	\
				createdate 	DATE 			DEFAULT CURRENT_DATE)" )
			et.G_Print("CREATE TABLE IF NOT EXISTS player res: " .. res .. "\n")
			
			res = assert(con:execute"CREATE TABLE IF NOT EXISTS log ( 		\
				id		INTEGER			PRIMARY KEY, 		\
				guid1		TEXT			NOT NULL,		\
				guid2		TEXT			DEFAULT NULL,		\
				type		INTEGER			DEFAULT NULL,		\
				textxml		TEXT			DEFAULT NULL,		\
				createdate	DATE			DEFAULT CURRENT_DATE)")
			et.G_Print("CREATE TABLE IF NOT EXISTS log res: " ..  res .. "\n")

			res = assert(con:execute"CREATE TABLE IF NOT EXISTS session ( 		\
				id 		INTEGER 		PRIMARY KEY,		\
				pkey 		INTEGER 		NOT NULL,		\
				slot 		INTEGER 		NOT NULL,		\
				map 		TEXT 			NOT NULL,		\
				ip 		TEXT 			DEFAULT '',		\
				netname 	TEXT 			DEFAULT '',		\
				cleanname 	TEXT 			DEFAULT '',		\
				valid 		INTEGER 		DEFAULT NULL,		\
				start 		DATE 			DEFAULT CURRENT_DATE,	\
				end 		DATE 			DEFAULT NULL,		\
				sptime 		INTEGER 		DEFAULT NULL,		\
				axtime 		INTEGER 		DEFAULT NULL,		\
				altime 		INTEGER 		DEFAULT NULL,		\
				lctime		INTEGER 		DEFAULT NULL,		\
				sstime		INTEGER 		DEFAULT NULL,		\
				xp0 		INTEGER 		DEFAULT 0,		\
				xp1 		INTEGER 		DEFAULT 0,		\
				xp2 		INTEGER 		DEFAULT 0,		\
				xp3 		INTEGER 		DEFAULT 0,		\
				xp4 		INTEGER 		DEFAULT 0,		\
				xp5 		INTEGER 		DEFAULT 0,		\
				xp6 		INTEGER 		DEFAULT 0,		\
				xptot 		INTEGER 		DEFAULT 0,		\
				acc 		REAL			DEFAULT 0.0,		\
				kills 		INTEGER 		DEFAULT 0,		\
				tkills 		INTEGER 		DEFAULT 0,		\
				death 		INTEGER 		DEFAULT 0,		\
				revives		INTEGER 		DEFAULT 0,		\
				uci 		INTEGER 		DEFAULT 0)" )
			et.G_Print("CREATE TABLE IF NOT EXISTS session res: " ..  res .. "\n")

			res = assert(con:execute"CREATE TABLE IF NOT EXISTS level ( 		\
				id		INTEGER 	PRIMARY KEY,			\
				pseudo		TEXT 		UNIQUE NOT NULL,		\
				name		TEXT 		NOT NULL,			\
				greetings	TEXT		DEFAULT '',			\
				flags 		TEXT		NOT NULL)" )
			et.G_Print("CREATE TABLE IF NOT EXISTS level res: " .. res .. "\n")
			
			res = assert(con:execute"CREATE TABLE IF NOT EXISTS version ( 		\
				id 		INTEGER 		PRIMARY KEY,		\
				version 	INTEGER 		NOT NULL UNIQUE )" )
			et.G_Print("CREATE TABLE IF NOT EXISTS version res: " ..  res .. "\n")
		
			-- SQLite needs exra cmds for setting up an index (anybody knows syntax for create table stmd?)
			-- player
			res = assert(con:execute"CREATE INDEX p_regname ON player(regname)" )
			et.G_Print("CREATE INDEX p_regname ON player(regname) res: " ..  res .. "\n")
			res = assert(con:execute"CREATE INDEX p_netname ON player(netname)" )
			et.G_Print("CREATE INDEX p_netname ON player(netname) res: " ..  res .. "\n")
			--log
			res = assert(con:execute"CREATE INDEX l_guid ON log(guid1, guid2)" )
			et.G_Print("CREATE INDEX l_guid ON log(guid1, guid2) res: " ..  res .. "\n")
			-- session
			res = assert(con:execute"CREATE INDEX s_pkey ON session(pkey)" )
			et.G_Print("CREATE INDEX s_pkey ON session(pkey) res: " ..  res .. "\n")
			res = assert(con:execute"CREATE INDEX s_ip ON session(ip)" )
			et.G_Print("CREATE INDEX s_ip ON session(ip) res: " ..  res .. "\n")
			res = assert(con:execute"CREATE INDEX s_end ON session(end)" )
			et.G_Print("CREATE INDEX s_end ON session(end) res: " ..  res .. "\n")

			-- insert data
   			res = assert(con:execute("INSERT INTO version VALUES ( '1', '" .. version .. "' )"))
			et.G_Print("Version res: " .. res .. " - Database version is " .. version .. "\n")
			
			-- TODO: create level entries
			
		-- mySQL	
		elseif getConfig("dbms") == "mySQL" then
		
			res = assert(con:execute"CREATE TABLE player ( \
				id 		INT 			PRIMARY KEY AUTO_INCREMENT,						\
				pkey 		VARCHAR(32) 	UNIQUE  NOT NULL,							\
				conname 	VARCHAR(36) 	NOT NULL,									\
				regname 	VARCHAR(36) 	DEFAULT '',									\
				netname 	VARCHAR(36) 	DEFAULT '',									\
				cleanname 	VARCHAR(36) 	DEFAULT '',									\
				isBot 		BOOLEAN 		DEFAULT 0,									\
				clan 		VARCHAR(20) 	DEFAULT '',									\
				level 		INT 			DEFAULT 0,									\
				flags 		VARCHAR(50) 	DEFAULT '',									\
				user 		VARCHAR(20) 	DEFAULT '',									\
				password 	VARCHAR(32) 	DEFAULT '',									\
				email 		VARCHAR(50) 	DEFAULT '',									\
				xp0 		INT 			DEFAULT 0,									\
				xp1 		INT 			DEFAULT 0,									\
				xp2 		INT 			DEFAULT 0,									\
				xp3 		INT 			DEFAULT 0,									\
				xp4 		INT 			DEFAULT 0,									\
				xp5 		INT 			DEFAULT 0,									\
				xp6 		INT 			DEFAULT 0,									\
				xptot 		INT 			DEFAULT 0,									\
				banreason 	VARCHAR(1024) 	DEFAULT '',									\
				bannedby 	VARCHAR(36) 	DEFAULT '',									\
				banexpire 	DATETIME 		DEFAULT '1000-01-01 00:00:00',				\
				mutedreason VARCHAR(1024)	DEFAULT '',									\
				mutedby 	VARCHAR(36) 	DEFAULT '',									\
				muteexpire 	DATETIME 		DEFAULT '1000-01-01 00:00:00',				\
				warnings 	SMALLINT 		DEFAULT 0,									\
				suspect 	TINYINT 		DEFAULT 0,									\
				regdate 	DATETIME 		DEFAULT NULL,								\
				updatedate 	TIMESTAMP 		DEFAULT '0000-00-00 00:00:00' on update CURRENT_TIMESTAMP,	\
				createdate 	DATETIME 		NOT NULL,									\
				INDEX(`regname`),														\
				INDEX(`netname`)														\
				) ENGINE=InnoDB" )		
			et.G_Print(res .. "\n")

			res = assert(con:execute"CREATE TABLE IF NOT EXISTS log ( 					\
				id		INT			PRIMARY KEY AUTO_INCREMENT, 						\
				guid1		VARCHAR(32)		NOT NULL,									\
				guid2		VARCHAR(32)		DEFAULT NULL,								\
				type		INT			DEFAULT NULL,									\
				textxml		VARCHAR(2056)		DEFAULT NULL,							\
				createdate	DATETIME		NOT NULL,									\
				INDEX(`guid1`),															\
				INDEX(`guid2`)															\
				) ENGINE=InnoDB" )
			et.G_Print(res .. "\n")

			res = assert(con:execute"CREATE TABLE session (								\
				id 		INT 		PRIMARY KEY AUTO_INCREMENT,							\
				pkey 		VARCHAR(32) 	NOT NULL,									\
				slot 		SMALLINT 		NOT NULL,									\
				map 		VARCHAR(36) 	NOT NULL,									\
				ip 		VARCHAR(25) 		DEFAULT '',									\
				netname 	VARCHAR(36) 	DEFAULT '',									\
				cleanname 	VARCHAR(36) 	DEFAULT '',									\
				valid 		BOOLEAN 	DEFAULT NULL,									\
				start 		DATETIME 	NOT NULL,										\
				end 		DATETIME 	DEFAULT NULL,									\
				sptime 		TIME 		DEFAULT NULL,									\
				axtime 		TIME 		DEFAULT NULL,									\
				altime 		TIME 		DEFAULT NULL,									\
				lctime 		TIME 		DEFAULT NULL,									\
				sstime 		TIME 		DEFAULT NULL,									\
				xp0 		INT 		DEFAULT 0,										\
				xp1 		INT 		DEFAULT 0,										\
				xp2 		INT 		DEFAULT 0,										\
				xp3 		INT 		DEFAULT 0,										\
				xp4 		INT 		DEFAULT 0,										\
				xp5 		INT 		DEFAULT 0,										\
				xp6 		INT 		DEFAULT 0,										\
				xptot		INT 		DEFAULT 0,										\
				acc 		DOUBLE (10,2) 	DEFAULT 0.0,								\
				kills 		SMALLINT 	DEFAULT 0,										\
				tkills 		SMALLINT 	DEFAULT 0,										\
				death 		SMALLINT 	DEFAULT 0,										\
				revives 	SMALLINT 	DEFAULT 0,										\
				uci 		TINYINT 	DEFAULT 0,										\
				INDEX(`pkey`),															\
				INDEX(`ip`),															\
				INDEX(`end`)															\
				) ENGINE=InnoDB" )
			et.G_Print(res .. "\n")

			res = assert(con:execute"CREATE TABLE IF NOT EXISTS level ( \
				id 		INT 			PRIMARY KEY AUTO_INCREMENT,		\
				pseudo      	VARCHAR(15) 		UNIQUE NOT NULL,			\
				name		VARCHAR(36)		NOT NULL,				\
				greetings	VARCHAR(150)		DEFAULT '',				\
				flags 		VARCHAR(50)		NOT NULL)" )
			et.G_Print(res .. "\n")
			
			res = assert(con:execute"CREATE TABLE version (				\
				id		INT		PRIMARY KEY AUTO_INCREMENT,		\
				version		INT		NOT NULL UNIQUE			\
				) ENGINE=InnoDB" )
			et.G_Print(res .. "\n")
				
			-- mySQL needs a trigger to add the current date/time to a datetime field
			res = assert(con:execute"CREATE TRIGGER trigger_player_insert 			\
				BEFORE INSERT ON `player` FOR EACH ROW SET NEW.createdate = NOW();" )
			et.G_Print(res .. "\n")
				
			res = assert(con:execute"CREATE TRIGGER trigger_log_insert 			\
				BEFORE INSERT ON `log` FOR EACH ROW SET NEW.createdate = NOW();" )
			et.G_Print(res .. "\n")
				
			-- insert data
			res = assert(con:execute(string.format("INSERT INTO version VALUES ( 1, %s )",version)))
			et.G_Print(res .. "\n")
		end
		
		et.G_Print(color .. commandprefix.."sqlcreate database created version: "..version .."\n")  
end

function shuttdownDBMS()
	if getConfig("dbms") == "mySQL" or getConfig("dbms") == "SQLite" then
		con:close()
		env:close()
	else 
		-- should never happen
		error("DBMS not supported.")
	end
end

function et_ShutdownGame( restart )
	shuttdownDBMS()
end