/* AMX Mod X script.
*   Admin Base Plugin
*
* by the AMX Mod X Development Team
*  originally developed by OLO
*
* This file is part of AMX Mod X.
*
*
*  This program is free software; you can redistribute it and/or modify it
*  under the terms of the GNU General Public License as published by the
*  Free Software Foundation; either version 2 of the License, or (at
*  your option) any later version.
*
*  This program is distributed in the hope that it will be useful, but
*  WITHOUT ANY WARRANTY; without even the implied warranty of
*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
*  General Public License for more details.
*
*  You should have received a copy of the GNU General Public License
*  along with this program; if not, write to the Free Software Foundation,
*  Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*  In addition, as a special exception, the author gives permission to
*  link the code of this program with the Half-Life Game Engine ("HL
*  Engine") and Modified Game Libraries ("MODs") developed by Valve,
*  L.L.C ("Valve"). You must obey the GNU General Public License in all
*  respects for all of the code used other than the HL Engine and MODs
*  from Valve. If you modify this file, you may extend this exception
*  to your version of the file, but you are not obligated to do so. If
*  you do not wish to do so, delete this exception statement from your
*  version.
*/

// Uncomment for SQL version
//#define USING_SQL

#include <amxmodx>
#include <amxmisc>
#if defined USING_SQL
#include <dbi>
#endif

#define MAX_ADMINS 64
#define PLUGINNAME	"AMX Mod X"

new g_aPassword[MAX_ADMINS][32]
new g_aName[MAX_ADMINS][32]
new g_aFlags[MAX_ADMINS]
new g_aAccess[MAX_ADMINS]
new g_aNum = 0
new g_cmdLoopback[16]

public plugin_init() {
#if defined USING_SQL
  register_plugin("Admin Base (SQL)",AMXX_VERSION_STR,"AMXX Dev Team")
#else
  register_plugin("Admin Base",AMXX_VERSION_STR,"AMXX Dev Team")
#endif
  register_dictionary("admin.txt")
  register_dictionary("common.txt")
  register_cvar("amx_mode","1")
  register_cvar("amx_password_field","_pw")
  register_cvar("amx_default_access","")

  register_cvar("amx_vote_ratio","0.02")
  register_cvar("amx_vote_time","10")
  register_cvar("amx_vote_answers","1")
  register_cvar("amx_vote_delay","60")
  register_cvar("amx_last_voting","0")
  register_cvar("amx_show_activity","2")
  register_cvar("amx_votekick_ratio","0.40")
  register_cvar("amx_voteban_ratio","0.40")
  register_cvar("amx_votemap_ratio","0.40")

  set_cvar_float("amx_last_voting",0.0)

#if defined USING_SQL
  register_srvcmd("amx_sqladmins","adminSql")
  register_cvar("amx_sql_table","admins")
#endif
  register_cvar("amx_sql_host","127.0.0.1")
  register_cvar("amx_sql_user","root")
  register_cvar("amx_sql_pass","")
  register_cvar("amx_sql_db","amx")

  register_concmd("amx_reloadadmins", "cmdReload", ADMIN_CFG)
  register_concmd("amx_addadmin", "addadminfn", ADMIN_CFG, "<playername> <accessflags> [password] - automatically add specified player as an admin to users.ini")

  format( g_cmdLoopback, 15, "amxauth%c%c%c%c" ,
  	random_num('A','Z') , random_num('A','Z') ,random_num('A','Z'),random_num('A','Z')  )

  register_clcmd( g_cmdLoopback, "ackSignal" )

  remove_user_flags(0,read_flags("z")) // Remove 'user' flag from server rights

  new configsDir[64]
  get_configsdir(configsDir, 63)
  server_cmd("exec %s/amxx.cfg", configsDir) // Execute main configuration file
  server_cmd("exec %s/sql.cfg", configsDir)
#if defined USING_SQL
  server_cmd("amx_sqladmins")
#else
  format(configsDir, 63, "%s/users.ini", configsDir)
  loadSettings(configsDir) // Load admins accounts
#endif
}

public addadminfn(id, level, cid) {
	if (!cmd_access(id, level, cid, 3))
		return PLUGIN_HANDLED

	new arg[33]
	read_argv(1, arg, 32)
	new player = cmd_target(id, arg, 10) // 2 = allow yourself (remove later?) 8 = no bots
	if (!player)
		return PLUGIN_HANDLED

	new flags[64]
	read_argv(2, flags, 63)

	new password[64]
	if (read_argc() == 4)
		read_argv(3, password, 63)

	new steamid[64]
	get_user_authid(player, steamid, 63)

	AddAdmin(id, steamid, flags, password)
	cmdReload(id, ADMIN_CFG, 0)

	new name[32]
	get_user_info(player, "name", name, 31)
	accessUser(player, name)

	return PLUGIN_HANDLED
}

AddAdmin(id, steamid[], accessflags[], password[]) {
	// Make sure that the users.ini file exists.
	new configsDir[64]
	get_configsdir(configsDir, 63)
	format(configsDir, 63, "%s/users.ini", configsDir)

	if (!file_exists(configsDir)) {
		console_print(id, "[%s] File ^"%s^" doesn't exist.", PLUGINNAME, configsDir)
		return
	}

	// Make sure steamid isn't already in file.
	new line = 0, textline[256], len
	const SIZE = 63
	new line_steamid[SIZE + 1], line_password[SIZE + 1], line_accessflags[SIZE + 1], line_flags[SIZE + 1], parsedParams
	// <name|ip|steamid> <password> <access flags> <account flags>
	while ((line = read_file(configsDir, line, textline, 255, len))) {
		if (len == 0 || equal(textline, ";", 1))
			continue // comment line

		parsedParams = parse(textline, line_steamid, SIZE, line_password, SIZE, line_accessflags, SIZE, line_flags, SIZE)
		if (parsedParams != 4)
			continue // Send warning/error?
		if (containi(line_flags, "c") != -1 && equal(line_steamid, steamid)) {
			console_print(id, "[%s] %s already exists!", PLUGINNAME, steamid)
			return
		}
	}

	// If we came here, steamid doesn't exist in users.ini. Add it.
	// Find out what flags we need.
	new flags[64] = "c" // Always use steamid
	new flagslen = strlen(flags)
	if (strlen(password) == 0)
		flagslen += format(flags[flagslen], 63 - flagslen, "e") // add flag to not check password if password wasn't supplied
	new linetoadd[512]
	format(linetoadd, 511, "^"%s^" ^"%s^" ^"%s^" ^"%s^"", steamid, password, accessflags, flags)
	console_print(id, "Adding:^n%s", linetoadd)

	if (!write_file(configsDir, linetoadd))
		console_print(id, "[%s] Failed writing to %s!", PLUGINNAME, configsDir)
}

public plugin_cfg() {
  new configFile[64],curMap[32]
  get_configsdir(configFile,31)
  get_mapname(curMap,31)
  new len = format(configFile,63,"%s/maps/%s.cfg",configFile,curMap)
  if ( file_exists(configFile) )
    set_task(6.1,"delayed_load",0,configFile,len+1)
}

public delayed_load(configFile[])
{
  server_cmd("exec %s",configFile)
}

loadSettings(szFilename[]) {
  if (!file_exists(szFilename)) return 0

  new szText[256], szFlags[32], szAccess[32]
  new a, pos = 0

  while ( g_aNum < MAX_ADMINS && read_file(szFilename,pos++,szText,255,a) )
  {
    if ( szText[0] == ';' ) continue

    if ( parse(szText, g_aName[ g_aNum ] ,31, g_aPassword[ g_aNum ], 31, szAccess,31,szFlags,31 ) < 2 )
        continue

    g_aAccess[ g_aNum ] = read_flags(szAccess)
    g_aFlags[ g_aNum ] = read_flags( szFlags )
    ++g_aNum
  }
  if (g_aNum == 1)
    server_print("[AMXX] %L", LANG_SERVER, "LOADED_ADMIN" )
  else
    server_print("[AMXX] %L", LANG_SERVER, "LOADED_ADMINS", g_aNum )

  return 1
}

#if defined USING_SQL
public adminSql() {
  new host[64],user[32],pass[32],db[128],table[32],error[128],dbType[7]
  dbi_type(dbType, 6)
  get_cvar_string("amx_sql_host",host,63)
  get_cvar_string("amx_sql_user",user,31)
  get_cvar_string("amx_sql_pass",pass,31)
  get_cvar_string("amx_sql_db", db, 127)
  get_cvar_string("amx_sql_table",table,31)

  new Sql:sql = dbi_connect(host,user,pass,db,error,127)
  if (sql <= SQL_FAILED) {
    server_print("[AMXX] %L",LANG_SERVER,"SQL_CANT_CON",error)
  	//backup to users.ini

    new configsDir[64]
    get_configsdir(configsDir, 63)
    format(configsDir, 63, "%s/users.ini", configsDir)
    loadSettings(configsDir) // Load admins accounts

    return PLUGIN_HANDLED
  }

  new Result:Res
  if (equali(dbType,"sqlite")) {
    if (!sqlite_table_exists(sql,table)) {
      dbi_query(sql,"CREATE TABLE %s ( auth TEXT NOT NULL DEFAULT '', password TEXT NOT NULL DEFAULT '', access TEXT NOT NULL DEFAULT '', flags TEXT NOT NULL DEFAULT '' )",table)
    }

    Res = dbi_query(sql,"SELECT auth, password, access, flags FROM %s",table)
  }
  else {
    dbi_query(sql,"CREATE TABLE IF NOT EXISTS `%s` ( `auth` VARCHAR( 32 ) NOT NULL, `password` VARCHAR( 32 ) NOT NULL, `access` VARCHAR( 32 ) NOT NULL, `flags` VARCHAR( 32 ) NOT NULL ) COMMENT = 'AMX Mod X Admins'",table)
    Res = dbi_query(sql,"SELECT `auth`,`password`,`access`,`flags` FROM `%s`",table)
  }

  if (Res == RESULT_FAILED) {
    dbi_error(sql,error,127)
    server_print("[AMXX] %L",LANG_SERVER,"SQL_CANT_LOAD_ADMINS",error)
    dbi_free_result(Res)
    dbi_close(sql)
    return PLUGIN_HANDLED
  }
  else if (Res == RESULT_NONE) {
    server_print("[AMXX] %L",LANG_SERVER,"NO_ADMINS")
    dbi_free_result(Res)
    dbi_close(sql)
    return PLUGIN_HANDLED
  }

  new szFlags[32],szAccess[32]
  g_aNum = 0
  while( dbi_nextrow(Res) > 0 ) {
    dbi_result(Res, "auth", g_aName[g_aNum], 31)
    dbi_result(Res, "password", g_aPassword[g_aNum], 31)
    dbi_result(Res, "access", szAccess, 31)
    dbi_result(Res, "flags", szFlags, 31)

    g_aAccess[ g_aNum ] = read_flags( szAccess )

    g_aFlags[ g_aNum ] = read_flags( szFlags )
    ++g_aNum
  }

  if (g_aNum == 1)
    server_print("[AMXX] %L", LANG_SERVER, "SQL_LOADED_ADMIN" )
  else
    server_print("[AMXX] %L", LANG_SERVER, "SQL_LOADED_ADMINS", g_aNum )
  dbi_free_result(Res)
  dbi_close(sql)
  return PLUGIN_HANDLED
}
#endif

public cmdReload(id,level,cid) {
  if (!cmd_access(id,level,cid,1))
    return PLUGIN_HANDLED

  //strip original flags (patch submitted by mrhunt)
  remove_user_flags(0,read_flags("z"))
#if !defined USING_SQL
  new filename[128]
  get_configsdir(filename,127)
  format(filename,63,"%s/users.ini", filename)

  g_aNum = 0
  loadSettings(filename) // Re-Load admins accounts

  if (id != 0) {
    if (g_aNum == 1)
      console_print(id,"[AMXX] %L", LANG_SERVER, "LOADED_ADMIN" )
    else
      console_print(id,"[AMXX] %L", LANG_SERVER, "LOADED_ADMINS", g_aNum )
  }
#else
  g_aNum = 0
  adminSql()

  if (id != 0) {
    if (g_aNum == 1)
      console_print(id,"[AMXX] %L", LANG_SERVER, "SQL_LOADED_ADMIN" )
    else
      console_print(id,"[AMXX] %L", LANG_SERVER, "SQL_LOADED_ADMINS", g_aNum )
  }
#endif

  return PLUGIN_HANDLED
}

getAccess(id,name[],authid[],ip[], password[]) {
  new index = -1
  new result = 0
  for (new i = 0; i < g_aNum; ++i) {
    if (g_aFlags[i] & FLAG_AUTHID) {
      if (equal(authid,g_aName[i])) {
        index = i
        break
      }
    }
    else if (g_aFlags[i] & FLAG_IP) {
      new c = strlen( g_aName[i] )
      if ( g_aName[i][ c - 1 ] == '.' ) { /* check if this is not a xxx.xxx. format */
        if (  equal( g_aName[i] , ip , c ) ) {
          index = i
          break
        }
      } /* in other case an IP must just match */
      else  if ( equal(ip,g_aName[i]) ){
        index = i
        break
      }
    }
    else {
      if (g_aFlags[i] & FLAG_TAG) {
        if (contain(name,g_aName[i])!=-1){
          index = i
          break
        }
      }
      else if (equal(name,g_aName[i])) {
        index = i
        break
      }
    }
  }
  if (index != -1) {
    if (g_aFlags[index] & FLAG_NOPASS){
      result |= 8
      new sflags[32]
      get_flags(g_aAccess[index],sflags,31)
      set_user_flags(id,g_aAccess[index])
      log_amx("Login: ^"%s<%d><%s><>^" became an admin (account ^"%s^") (access ^"%s^") (address ^"%s^")",
        name,get_user_userid(id),authid,g_aName[index] ,sflags,ip)
    }
    else if (equal(password,g_aPassword[index])) {
      result |= 12
      set_user_flags(id,g_aAccess[index])
      new sflags[32]
      get_flags(g_aAccess[index],sflags,31)
      log_amx("Login: ^"%s<%d><%s><>^" became an admin (account ^"%s^") (access ^"%s^") (address ^"%s^")",
        name,get_user_userid(id),authid,g_aName[index] ,sflags,ip)
    }
    else {
      result |= 1
      if (g_aFlags[index] & FLAG_KICK){
        result |= 2
        log_amx("Login: ^"%s<%d><%s><>^" kicked due to invalid password (account ^"%s^") (address ^"%s^")",
          name,get_user_userid(id),authid,g_aName[index],ip)
      }
    }
  }
  else if (get_cvar_float("amx_mode")==2.0) {
    result |= 2
  }
  else {
    new defaccess[32]
    get_cvar_string("amx_default_access",defaccess,31)
    if (!strlen(defaccess))
      copy(defaccess, 32, "z")
    new idefaccess = read_flags(defaccess)
    if (idefaccess) {
      result |= 8
      set_user_flags(id,idefaccess)
    }
  }

  return result
}

accessUser( id, name[] = "" ) {
  remove_user_flags(id)
  new userip[32],userauthid[32],password[32],passfield[32],username[32]
  get_user_ip(id,userip,31,1)
  get_user_authid(id,userauthid,31)
  if ( name[0] ) copy( username , 31, name)
  else get_user_name(id,username,31 )
  get_cvar_string("amx_password_field",passfield,31)
  get_user_info(id,passfield,password,31)
  new result = getAccess(id,username,userauthid,userip,password)
  if (result & 1) client_cmd(id,"echo ^"* %L^"",id,"INV_PAS")
  if (result & 2) {
    client_cmd(id,g_cmdLoopback)
    return PLUGIN_HANDLED
  }
  if (result & 4) client_cmd(id,"echo ^"* %L^"",id,"PAS_ACC")
  if (result & 8) client_cmd(id,"echo ^"* %L^"",id,"PRIV_SET")
  return PLUGIN_CONTINUE
}

public client_infochanged(id) {
  if ( !is_user_connected(id) || !get_cvar_num("amx_mode") )
    return PLUGIN_CONTINUE

  new newname[32], oldname[32]
  get_user_name(id,oldname,31)
  get_user_info(id,"name",newname,31)

  if ( !equal(newname,oldname) )
    accessUser( id, newname )

  return PLUGIN_CONTINUE
}

public ackSignal(id) {
  server_cmd("kick #%d ^"%L^"", get_user_userid(id), id, "NO_ENTRY" )
}

public client_authorized(id)
  return get_cvar_num( "amx_mode" ) ? accessUser( id ) : PLUGIN_CONTINUE

public client_putinserver(id)
{
	if (!is_dedicated_server() && id == 1)
		return get_cvar_num("amx_mode") ? accessUser(id) : PLUGIN_CONTINUE
	return PLUGIN_CONTINUE
}