#include <amxmodx>
#include <fakemeta>
#include <sqlx>
#include <cstrike>
#include <amxmisc>
#include <fun>
#include <admin>
#define VERSION "1.0"
// 密码最短长度
#define PASSWORD_MIN_LEN 6
#define MAX_LOGIN_ATTEMPTS 3
#define BAN_TIME 3
// user_activity 1 - 注册中 2 - 登入中 3 - 输入旧密码 4 - 输入新密码
new user_status[33],user_activity[33],user_len[33],user_password[33][33],user_wrong[33],mysqlx_host,mysqlx_user,mysqlx_db,mysqlx_pass,gmsgScreenFade,user_rest[33]
new Handle:g_hTuple
new bool:user_connect[33]
new const szTables[][] = 
{
 "CREATE TABLE IF NOT EXISTS `register_system` ( `user_id` varchar(64) NOT NULL,`user_password` varchar(128) default NULL,`user_ip` varchar(32) default NULL,PRIMARY KEY (`user_id`) ) TYPE=MyISAM;"
}
public plugin_init()
{
 gmsgScreenFade = get_user_msgid("ScreenFade")
 register_plugin("注册/登入系统", VERSION, "slyh")
 register_clcmd("say", "user_say", ADMIN_ALL)
 register_clcmd("say /login", "user_login", ADMIN_ALL)
 register_event("ScreenFade","Event_ScreenFade","b")
 register_forward(FM_ClientUserInfoChanged, "Fwd_ClientInfoChanged")
 mysqlx_host = register_cvar ("regsys_host", "")
 mysqlx_user = register_cvar ("regsys_user", "")
 mysqlx_pass = register_cvar ("regsys_pass", "")
 mysqlx_db = register_cvar ("regsys_dbname", "register_system")
 MySQLx_Init()
}
public user_say(id,level,cid)
{
 if( ( !cmd_access(id, level, cid, 2) ) || user_activity[id] == 0)
  return PLUGIN_CONTINUE
 new name[32],message[128]
 get_user_name(id,name,31)
 switch(user_activity[id])
 {
  case 1:
  {
   new arg_pass[32]
   read_argv(1, arg_pass, 31)
   new len = strlen(arg_pass);
   if(len < PASSWORD_MIN_LEN || len > 20)
   {
    format(message,127,"^x04[hk-fun.com提醒你] ^x01密码长度最少需有 %d 位, 最多不可超过 20 位, 请输入一个新的密码.",PASSWORD_MIN_LEN)
    client_color(id,id,message)
    client_cmd(id,"messagemode")
    return PLUGIN_HANDLED
   }
   format(message,127,"^x04[hk-fun.com提醒你] ^x01你已成功注册, 请谨记你的密码: %s .",arg_pass)
   access_user(id)
   client_color(id,id,message)
   user_password[id] = arg_pass
   user_status[id] = 3
   user_activity[id] = 0
   set_task(2.5,"Reset_Screen",id)
   set_task(3.0,"chooseteam",id)
   set_task(1.1,"hud_finish",id)
   client_cmd(id,"setinfo ^"_reg^" ^"%s^"",arg_pass)
   new name[32],ip[32]
   get_user_name(id,name,31)
   get_user_ip(id,ip,31,1)
   static szQuery[128];
   formatex( szQuery, 127, "REPLACE INTO `register_system` (`user_id`, `user_password`, `user_ip`) VALUES ('%s', '%s', '%s');",name,arg_pass,ip)
   SQL_ThreadQuery( g_hTuple, "QuerySetData", szQuery);
   return PLUGIN_HANDLED
  }
  case 2:
  {
   new arg_pass[32]
   read_argv(1, arg_pass, 31)
   user_password[id] = arg_pass
   LoadData(id)
   return PLUGIN_HANDLED
  }
  case 3:
  {
   new arg_pass[32]
   read_argv(1, arg_pass, 31)
   user_password[id] = arg_pass
   LoadData(id)
   return PLUGIN_HANDLED
  }
  case 4:
  {
   new arg_pass[32]
   read_argv(1, arg_pass, 31)
   new len = strlen(arg_pass);
   if(len < PASSWORD_MIN_LEN || len > 20)
   {
    format(message,127,"^x04[hk-fun.com提醒你] ^x01密码长度最少需有 %d 位, 最多不可超过 20 位, 请输入一个新的密码.",PASSWORD_MIN_LEN)
    client_color(id,id,message)
    client_cmd(id,"messagemode")
    return PLUGIN_HANDLED
   }
   format(message,127,"^x04[hk-fun.com提醒你] ^x01你已成功更换密码, 请谨记你的密码: %s .",arg_pass)
   client_color(id,id,message)
   user_password[id] = arg_pass
   user_status[id] = 3
   user_activity[id] = 0
   set_task(2.5,"Reset_Screen",id)
   set_task(3.0,"chooseteam",id)
   set_task(1.1,"hud_finish",id)
   new name[32],ip[32]
   get_user_name(id,name,31)
   get_user_ip(id,ip,31,1)
   static szQuery[128];
   formatex( szQuery, 127, "REPLACE INTO `register_system` (`user_id`, `user_password`, `user_ip`) VALUES ('%s', '%s', '%s');",name,arg_pass,ip)
   SQL_ThreadQuery( g_hTuple, "QuerySetData", szQuery);
   return PLUGIN_HANDLED
  }
 }
 return PLUGIN_CONTINUE
}
public QuerySelectData( iFailState, Handle:hQuery, szError[ ], iError, iData[ ], iDataSize, Float:fQueueTime ) 
{ 
 if( iFailState == TQUERY_CONNECT_FAILED 
 || iFailState == TQUERY_QUERY_FAILED ) 
 { 
  log_amx( "%s", szError );
  
  return;
 } 
 else 
 { 
  new id = iData[ 0 ];
  
  new password = SQL_FieldNameToNum(hQuery, "user_password") 
  new pw[44]
  while (SQL_MoreResults(hQuery)) 
  {
   new name[32],message[128]
   get_user_name(id,name,31)
   SQL_ReadResult(hQuery, password, pw, sizeof(pw)-1)
   if(user_connect[id])
   {
    new password[32]
    user_len[id] = strlen(pw)
    user_connect[id] = false
    get_user_info(id, "_reg", password, 31)
    if(equali(pw, password))
    {
    user_status[id] = 3
    access_user(id)
    }
    else
    {
    set_task(1.0, "force_login", id, _, _, "a", 20)
    set_task(20.0,"check_user",id)
    user_rest[id] = 20
    }
   }
   else
   {
    if(!equali(pw, user_password[id]))
    {
     format(message,127,"^x04[hk-fun.com提醒你] ^x01密码错误, 你只有 %d 次机会.",MAX_LOGIN_ATTEMPTS)
     client_color(id,id,message)
     user_wrong[id] += 1
     client_cmd(id,"messagemode")
     if(user_wrong[id] >= MAX_LOGIN_ATTEMPTS)
     {
      format(message,127,"^x04[hk-fun.com提醒你] ^x01由于 ^x03%s^x01 输入了 %d 次错误的密码, 所以已被封禁 %d 分钟.",name,MAX_LOGIN_ATTEMPTS,BAN_TIME)
      client_color(0,id,message)
      server_cmd("banid ^"%d^" ^"#%i^"",BAN_TIME,get_user_userid(id))
      server_cmd("kick #%i ^"由于你输入了 %d 次错误的密码, 所以已被封禁 %d 分钟!^"",get_user_userid(id),MAX_LOGIN_ATTEMPTS,BAN_TIME);
     }
    }
    else
    {
     if(user_activity[id] == 3)
     {
      format(message,127,"^x04[hk-fun.com提醒你] ^x01已认证身份, 请输入新密码.")
      client_color(id,id,message)
      user_activity[id] = 4
      client_cmd(id,"messagemode")
     }
     else
     {
      format(message,127,"^x04[hk-fun.com提醒你] ^x01你已成功登入.")
      client_color(id,id,message)
      user_status[id] = 3
      user_activity[id] = 0
      set_task(2.5,"Reset_Screen",id)
      set_task(3.0,"chooseteam",id)
      set_task(1.1,"hud_finish",id)
      access_user(id)
     }
    }
   }
   SQL_NextRow(hQuery)
  }
 } 
}
public user_login(id)
{
 if(user_status[id] == 3)
 {
  new menu = menu_create("\y欢迎使用登记系统, 你现已登入:", "login_menu_handler")
  menu_additem(menu, "\d注册", "1", 0)
  menu_additem(menu, "\d登录", "2", 0)
  menu_additem(menu, "\w更改密码", "3", 0)
  //menu_additem(menu, "\d设定(暂未开放)", "4", 0)
  menu_setprop(menu, MPROP_EXIT, MEXIT_ALL)
  menu_display(id, menu, 0)
  return PLUGIN_HANDLED
 }
 if(user_len[id] >= PASSWORD_MIN_LEN)
 {
  new menu = menu_create("\y欢迎使用登记系统, 请完成登入程序:", "login_menu_handler")
  menu_additem(menu, "\d注册", "1", 0)
  menu_additem(menu, "\w登录", "2", 0)
  menu_additem(menu, "\d更改密码", "3", 0)
  menu_setprop(menu, MPROP_EXIT, MEXIT_ALL)
  menu_display(id, menu, 0)
  return PLUGIN_HANDLED
 }
 else
 {
  new menu = menu_create("\y欢迎使用登记系统, 请完成注册程序:", "login_menu_handler")
  menu_additem(menu, "\w注册", "1", 0)
  menu_additem(menu, "\d登录", "2", 0)
  menu_additem(menu, "\d更改密码", "3", 0)
  menu_setprop(menu, MPROP_EXIT, MEXIT_ALL)
  menu_display(id, menu, 0)
  return PLUGIN_HANDLED
 }
 return PLUGIN_HANDLED
}
public login_menu_handler(id, menu, item)
{
 if( item == MENU_EXIT )
 {
  menu_destroy(menu)
  return PLUGIN_HANDLED
 }
 new data[6], iName[64]
 new access, callback
 
 menu_item_getinfo(menu, item, access, data,5, iName, 63, callback)
 new key = str_to_num(data)
 switch(key)
 {
  case 1:
  {
   if(user_status[id] != 3 && user_len[id] < PASSWORD_MIN_LEN)
   {
    user_activity[id] = 1
    if(is_user_alive(id))
    {
     user_kill(id)
     cs_set_user_deaths(id, cs_get_user_deaths(id) - 1)
     set_user_frags(id, get_user_frags(id) + 1 )
    }
    cs_set_user_team(id,CS_TEAM_SPECTATOR)
    Fade_To_Black(id)
    client_cmd(id,"messagemode")
    hud(id)
   }
  }
  case 2:
  {
   if(user_status[id] != 3 && user_len[id] >= PASSWORD_MIN_LEN)
   {
    user_activity[id] = 2
    if(is_user_alive(id))
    {
     user_kill(id)
     cs_set_user_deaths(id, cs_get_user_deaths(id) - 1)
     set_user_frags(id, get_user_frags(id) + 1 )
    }
    cs_set_user_team(id,CS_TEAM_SPECTATOR)
    Fade_To_Black(id)
    client_cmd(id,"messagemode")
    hud(id)
   }
  }
  case 3:
  {
   if(user_status[id] == 3)
   {
    user_activity[id] = 3
    if(is_user_alive(id))
    {
     user_kill(id)
     cs_set_user_deaths(id, cs_get_user_deaths(id) - 1)
     set_user_frags(id, get_user_frags(id) + 1 )
    }
    cs_set_user_team(id,CS_TEAM_SPECTATOR)
    Fade_To_Black(id)
    client_cmd(id,"messagemode")
    hud(id)
   }
  }
  case 4:
  {
   //setting_menu(id)
  }
 }
 menu_destroy(menu)
 return PLUGIN_HANDLED
}
public setting_menu(id)
{
 new menu = menu_create("\y系统设定:", "setting_menu_handler")
 menu_additem(menu, "\w返回主目录", "2", 0)
 menu_setprop(menu, MPROP_EXIT, MEXIT_NEVER)
 menu_display(id, menu, 0)
}
public setting_menu_handler(id, menu, item)
{
 if( item == MENU_EXIT )
 {
  menu_destroy(menu)
  return PLUGIN_HANDLED
 }
 new data[6], iName[64]
 new access, callback
 
 menu_item_getinfo(menu, item, access, data,5, iName, 63, callback)
 new key = str_to_num(data)
 switch(key)
 {
  case 2:
  {
   user_login(id)
  }
 }
 menu_destroy(menu)
 return PLUGIN_HANDLED
}
public hud(id)
{
 switch(user_activity[id])
 {
  case 1:
  {
   set_hudmessage(0, 0, 0, -1.0, -1.0, 2, 1.0, 1.0, 0.0, 0.0, -1)
   show_hudmessage(id, "你只要输入密码, 再按Enter, 即完成整个注册过程!^n注意事项:^n1.密码长度不得短于 %d 位^n2.密码长度不得长于 20 位",PASSWORD_MIN_LEN)
   set_task(1.0,"hud",id)
  }
  case 2:
  {
   set_hudmessage(0, 0, 0, -1.0, -1.0, 2, 1.0, 1.0, 0.0, 0.0, -1)
   show_hudmessage(id, "你只要输入密码, 再按Enter, 即完成整个登入过程!^n注意事项:^n你只有 %d 次错误输入密码的机会,^n机会用尽后, 你将会被封禁 %d 分钟!",MAX_LOGIN_ATTEMPTS,BAN_TIME)
   set_task(1.0,"hud",id)
  }
  case 3:
  {
   set_hudmessage(0, 0, 0, -1.0, -1.0, 2, 1.0, 1.0, 0.0, 0.0, -1)
   show_hudmessage(id, "请你输入旧密码以确认身份!")
   set_task(1.0,"hud",id)
  }
  case 4:
  {
   set_hudmessage(0, 0, 0, -1.0, -1.0, 2, 1.0, 1.0, 0.0, 0.0, -1)
   show_hudmessage(id, "你只要新输入密码, 再按Enter, 即完成整个密码更换过程!^n注意事项:^n1.新密码长度不得短于 %d 位^n2.新密码长度不得长于 20 位",PASSWORD_MIN_LEN)
   set_task(1.0,"hud",id)
  }
 }
}
public Event_ScreenFade(id)
{
 if(user_activity[id] > 0)
 {
  Fade_To_Black(id)
 }
}
public Fade_To_Black(id)
{
 message_begin ( MSG_ONE_UNRELIABLE , gmsgScreenFade , _ , id ) ;
 write_short ( ( 1<<3 ) | ( 1<<8 ) | ( 1<<10 ) ) ;
 write_short ( ( 1<<3 ) | ( 1<<8 ) | ( 1<<10 ) ) ;
 write_short ( ( 1<<0 ) | ( 1<<2 ) ) ;
 write_byte ( 0 ) ;
 write_byte ( 0 ) ;
 write_byte ( 0 ) ;
 write_byte ( 255 ) ;
 message_end ( ) ;
}
public Reset_Screen(id)
{
 message_begin ( MSG_ONE_UNRELIABLE , gmsgScreenFade , _ , id ) ;
 write_short ( 1<<2 ) ;
 write_short ( 0 ) ;
 write_short ( 0 ) ;
 write_byte ( 0 ) ;
 write_byte ( 0 ) ;
 write_byte ( 0 ) ;
 write_byte ( 0 ) ;
 message_end ( ) ;
}
public MySQLx_Init()
{
 new szHost[64], szUser[32], szPass[32], szDB[128];
 get_pcvar_string( mysqlx_host, szHost, charsmax( szHost ) );
 get_pcvar_string( mysqlx_user, szUser, charsmax( szUser ) );
 get_pcvar_string( mysqlx_pass, szPass, charsmax( szPass ) );
 get_pcvar_string( mysqlx_db, szDB, charsmax( szDB ) );
 g_hTuple = SQL_MakeDbTuple( szHost, szUser, szPass, szDB );
 for ( new i = 0; i < sizeof szTables; i++ )
 {
  SQL_ThreadQuery( g_hTuple, "QueryCreateTable", szTables[i])
 }
}
public QueryCreateTable( iFailState, Handle:hQuery, szError[ ], iError, iData[ ], iDataSize, Float:fQueueTime ) 
{ 
 if( iFailState == TQUERY_CONNECT_FAILED 
 || iFailState == TQUERY_QUERY_FAILED ) 
 { 
  log_amx( "%s", szError ); 
  
  return;
 } 
}
public LoadData(id)
{
 new name[32]
 get_user_name(id,name,31)
 static szQuery[ 128 ], iData[ 1 ]; 
 formatex( szQuery, 127, "SELECT `user_password` FROM `register_system` WHERE ( `user_id` = '%s' );",name); 
 iData[ 0 ] = id;
 SQL_ThreadQuery( g_hTuple, "QuerySelectData", szQuery, iData, 1 );
}
public QuerySetData( iFailState, Handle:hQuery, szError[ ], iError, iData[ ], iDataSize, Float:fQueueTime ) 
{ 
 if( iFailState == TQUERY_CONNECT_FAILED 
 || iFailState == TQUERY_QUERY_FAILED ) 
 { 
  log_amx( "%s", szError ); 
  
  return;
 } 
}
public plugin_natives()
{
 register_native("get_user_status", "native_get_user_status", 1)
 register_native("get_user_pwlen", "native_get_user_pwlen", 1)
}
public hud_finish(id)
{
 set_hudmessage(0, 0, 0, -1.0, -1.0, 0, 0.0, 2.0, 0.0, 0.0, -1)
 show_hudmessage(id, "你已完成登入/注册/更换密码过程!")
}
public client_connect(id)
{
 user_activity[id] = 0
 user_wrong[id] = 0
 user_status[id] = 0
 user_len[id] = 0
 user_connect[id] = true
 remove_task(id)
 LoadData(id)
}
public client_putinserver(id)
{
 if( ( get_user_flags(id) & ADMIN_KICK ) && ( user_status[id] !=3 ) )
 {
  remove_user_flags(id)
 }
 set_task(1.0,"not_login",id)
}
public messengmode(id)
{
 client_cmd(id,"messagemode")
}
public client_disconnect(id)
{
 user_activity[id] = 0
 user_wrong[id] = 0
 user_status[id] = 0
 user_len[id] = 0
 remove_task(id)
 user_connect[id] = false
}
public Fwd_ClientInfoChanged(id, buffer)
{
 if (!is_user_connected(id))
  return FMRES_IGNORED;
 
 static name[32], val[32];
 get_user_name(id, name, sizeof name - 1)
 
 engfunc(EngFunc_InfoKeyValue, buffer, "name", val, sizeof val - 1);
 
 if(equal(val, name))
  return FMRES_IGNORED;
 
 engfunc(EngFunc_SetClientKeyValue, id, buffer, "name", name);
 
 client_cmd(id, "name ^"%s^"; setinfo name ^"%s^"", name, name);
 
 client_print(id, print_console, "游戏中不允许更换名字.");
 
 return FMRES_SUPERCEDE;
}
public native_get_user_status(id)
{
 return user_status[id]
}
public native_get_user_pwlen(id)
{
 return user_len[id]
}
public chooseteam(id)
{
 client_cmd(id,"chooseteam")
}
public client_color(playerid, colorid, msg[]){
message_begin(playerid?MSG_ONE:MSG_ALL,get_user_msgid("SayText"),{0,0,0},playerid) 
write_byte(colorid)
write_string(msg)
message_end()
}
public not_login(id)
{
 if(user_status[id] != 3)
 {
  set_hudmessage(255, 0, 0, -1.0, 0.7, 0, 0.0, 0.1, 0.0, 0.0)
  show_hudmessage(id,"请说/login注册/登录, 否则你将受以下限制^n失去管理员权力")
  set_task(0.1,"not_login",id)
 }
}
public force_login(id)
{
       if(user_status[id] != 3)
       {
    user_rest[id] -= 1
    set_hudmessage(255, 0, 0, -1.0, -1.0, 0, 0.0, 1.0, 0.0, 0.0)
    show_hudmessage(id,"你的昵称已被使用，请登入或转换另一个名字，否则你将会于 %i 秒后被踢除",user_rest[id]) //这行可以更改
    set_task(1.0,"force_login",id)
       }
}
    
public check_user(id)
{     if(user_status[id] != 3)
     {
  server_cmd("kick #%i ^"对被踢者的提示^"",get_user_userid(id)) // 这行也是可以更改
     }
}